a2d/include/ad/a2dvec.h at 646a8de5128b1bba96df9bf96c10d931401acb27 · gjkennedy/a2d · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#ifndef A2D_VEC_H
#define A2D_VEC_H

#include "../a2ddefs.h"

namespace A2D {

template <typename T, int N>
class Vec {
 public:
  typedef T type;
  static const ADObjType obj_type = ADObjType::VECTOR;
  static const index_t ncomp = N;

  A2D_FUNCTION Vec() {
    for (int i = 0; i < N; i++) {
      V[i] = 0.0;
    }
  }
  template <typename T2>
  A2D_FUNCTION Vec(const T2* vals) {
    for (int i = 0; i < N; i++) {
      V[i] = vals[i];
    }
  }
  template <typename T2>
  A2D_FUNCTION Vec(const Vec<T2, N>& src) {
    for (int i = 0; i < N; i++) {
      V[i] = src(i);
    }
  }
  A2D_FUNCTION void zero() {
    for (int i = 0; i < N; i++) {
      V[i] = 0.0;
    }
  }
  template <typename T2>
  A2D_FUNCTION void copy(const Vec<T2, N>& vec) {
    for (int i = 0; i < N; i++) {
      V[i] = vec(i);
    }
  }
  template <class IdxType>
  A2D_FUNCTION T& operator()(const IdxType i) {
    return V[i];
  }
  template <class IdxType>
  A2D_FUNCTION const T& operator()(const IdxType i) const {
    return V[i];
  }

  A2D_FUNCTION T* get_data() { return V; }
  A2D_FUNCTION const T* get_data() const { return V; }

  template <typename I>
  A2D_FUNCTION T& operator[](const I i) {
    return V[i];
  }
  template <typename I>
  A2D_FUNCTION const T& operator[](const I i) const {
    return V[i];
  }

 private:
  T V[N];
};

template <typename T>
struct is_a2d_vector : std::false_type {};

template <typename U, int N>
struct is_a2d_vector<Vec<U, N>> : std::true_type {};

}  // namespace A2D

#endif  // A2D_VEC_H