TensorIndex.h

Go to the documentation of this file.

#ifndef TensorIndex_h
#define TensorIndex_h
template <int d1, int d2 = 1, int d3 = 1, int d4 = 1>
class TensorIndex {
public:
  enum { SIZE = d1 * d2 * d3 * d4 };
  enum { LEN1 = d1 };
  enum { LEN2 = d2 };
  enum { LEN3 = d3 };
  enum { LEN4 = d4 };

  static int indexOf(const int i) { return i; }
  static int indexOf(const int i, const int j) { return j * d1 + i; }
  static int indexOf(const int i, const int j, const int k) { return (k * d2 + j) * d1 + i; }
  static int indexOf(const int i, const int j, const int k, const int l) { return ((l * d3 + k) * d2 + j) * d1 + i; }
};

template <int d1, int d2 = 1, int d3 = 1, int d4 = 1>
class TensorIndex_base1 {
public:
  enum { SIZE = d1 * d2 * d3 * d4 };
  enum { LEN1 = d1 };
  enum { LEN2 = d2 };
  enum { LEN3 = d3 };
  enum { LEN4 = d4 };

  static int indexOf(const int i) { return i - 1; }
  static int indexOf(const int i, const int j) { return j * d1 + i - 1 - d1; }
  static int indexOf(const int i, const int j, const int k) { return (k * d2 + j) * d1 + i - 1 - d1 - d1 * d2; }
  static int indexOf(const int i, const int j, const int k, const int l) {
    return ((l * d3 + k) * d2 + j) * d1 + i - 1 - d1 - d1 * d2 - d1 * d2 * d3;
  }
};
#endif