SQCSimulator2023/SQCSim2021/array3d.h
MarcEricMartel de7945038c :)
2021-11-27 14:14:05 -05:00

56 lines
1.3 KiB
C++

#ifndef ARRAY3D_H__
#define ARRAY3D_H__
#include "define.h"
template <class T>
class Array3d {
public:
Array3d(int x, int y, int z);
~Array3d();
Array3d(const Array3d& array);
void Set(int x, int y, int z, T type);
T Get(int x, int y, int z) const;
void Reset(T type);
private:
int m_x, m_y, m_z;
T* m_array;
int To1dIndex(int x, int y, int z) const;
};
template <class T>
Array3d<T>::Array3d(int x, int y, int z) : m_x(x), m_y(y), m_z(z) { m_array = new T[m_x * m_y * m_z]; }
template <class T>
Array3d<T>::~Array3d() { delete[] m_array; }
template <class T>
Array3d<T>::Array3d(const Array3d<T>& array) : m_x(array.m_x), m_y(array.m_y), m_z(array.m_z) {
m_array = new T[m_x * m_y * m_z];
for (int i = 0; i < m_x * m_y * m_z; ++i)
m_array[i] = array.m_array[i];
}
template <class T>
void Array3d<T>::Set(int x, int y, int z, T type) {
m_array[To1dIndex(x, y, z)] = type;
}
template <class T>
T Array3d<T>::Get(int x, int y, int z) const { return m_array[To1dIndex(x, y, z)]; }
template <class T>
void Array3d<T>::Reset(T type) {
for (int i = 0; i < m_x * m_y * m_z; ++i)
m_array[i] = type;
}
template <class T>
int Array3d<T>::To1dIndex(int x, int y, int z) const { return x + (z * m_x) + (y * m_z * m_x); }
#endif // ARRAY3D_H__