X-Git-Url: http://git.kpe.io/?a=blobdiff_plain;f=include%2Farray2d.h;h=8c323f4034ca45f3b1449b3ba3f2b50c7db3c2ed;hb=9f7d379c2b03f3c09907cfd885072433e0428042;hp=30b92d474d264b6025bc79bcaa7e7f6cda808f92;hpb=2451ac413848718a1dd666ce6f6464e974680f47;p=ctsim.git diff --git a/include/array2d.h b/include/array2d.h index 30b92d4..8c323f4 100644 --- a/include/array2d.h +++ b/include/array2d.h @@ -9,11 +9,7 @@ ** This is part of the CTSim program ** Copyright (C) 1983-2000 Kevin Rosenberg ** -** $Id: array2d.h,v 1.3 2000/06/09 01:35:33 kevin Exp $ -** $Log: array2d.h,v $ -** Revision 1.3 2000/06/09 01:35:33 kevin -** Convert MPI structure to C++ class -** +** $Id: array2d.h,v 1.8 2000/09/02 05:10:39 kevin Exp $ ** ** This program is free software; you can redistribute it and/or modify ** it under the terms of the GNU General Public License (version 2) as @@ -32,53 +28,83 @@ #ifndef ARRAY2D_H #define ARRAY2D_H -#include +#include "ctsupport.h" + template class Array2d { - private: - unsigned int nx; - unsigned int ny; - public: - - T** array_data; - Array2d (unsigned int x, unsigned int y) + : m_nx(x), m_ny(y), array_data(0) { - nx = x; - ny = y; - array_data = new T*[nx]; - - for (unsigned int i = 0; i < nx; i++) - array_data[i] = new T[ny]; + allocArray(); } + Array2d () + : m_nx(0), m_ny(0), array_data(0) + {} ~Array2d () { - for (int i = 0; i < nx; i++) - delete array_data[i]; - delete array_data; + deleteArray(); } - T** getArray (void) const + void initSetSize (unsigned int x, unsigned int y) + { + m_nx = x; + m_ny = y; + deleteArray(); + allocArray(); + } + + T** getArray () const { return array_data; } T* getColumn (unsigned int x) const - { return array_data[x]; } + { return (array_data ? array_data[x] : NULL); } T getPoint (unsigned int x, unsigned int y) const - { return (array_data[x][y]); } + { return (array_data ? array_data[x][y] : NULL); } - unsigned int sizeofPixel (void) const + unsigned int sizeofPixel () const { return sizeof(T); } - unsigned int sizeofColumn (void) const - { return (sizeof(T) * ny); } + unsigned int sizeofColumn () const + { return (sizeof(T) * m_ny); } + + unsigned int sizeofArray () const + { return (sizeof(T) * m_nx * m_ny); } + + + private: + unsigned int m_nx; + unsigned int m_ny; + T** array_data; + + void allocArray () + { + if (array_data) + deleteArray(); + + array_data = new T*[m_nx]; + + for (unsigned int i = 0; i < m_nx; i++) + array_data[i] = new T[m_ny]; + } + + void deleteArray () + { + if (array_data) { + for (unsigned int i = 0; i < m_nx; i++) + delete array_data [i]; + delete array_data; + array_data = NULL; + } + } + - unsigned int sizeofArray (void) const - { return (sizeof(T) * nx * ny); } + Array2d& operator= (const Array2d& rhs); //assignment operator + Array2d (const Array2d& rhs); // copy constructor };