X-Git-Url: http://git.kpe.io/?a=blobdiff_plain;f=include%2Farray2d.h;h=d397e5f45160ce4824f5245eca3bfed47efe9cad;hb=dca8d386ab6a4e33050b456cff4547d6adfdf249;hp=30b92d474d264b6025bc79bcaa7e7f6cda808f92;hpb=2451ac413848718a1dd666ce6f6464e974680f47;p=ctsim.git diff --git a/include/array2d.h b/include/array2d.h index 30b92d4..d397e5f 100644 --- a/include/array2d.h +++ b/include/array2d.h @@ -1,19 +1,13 @@ /***************************************************************************** ** FILE IDENTIFICATION ** -** Name: array2d.h +** Name: array2d.h ** Purpose: 2-dimension array classes -** Programmer: Kevin Rosenberg -** Date Started: June 2000 +** Programmer: Kevin Rosenberg +** Date Started: June 2000 ** ** 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 -** +** Copyright (c) 1983-2009 Kevin Rosenberg ** ** 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 +26,83 @@ #ifndef ARRAY2D_H #define ARRAY2D_H -#include +#include "ctsupport.h" -template -class Array2d { - private: - unsigned int nx; - unsigned int ny; +template +class Array2d { public: - - T** array_data; - Array2d (unsigned int x, unsigned int y) - { - nx = x; - ny = y; - array_data = new T*[nx]; - - for (unsigned int i = 0; i < nx; i++) - array_data[i] = new T[ny]; - } + : m_nx(x), m_ny(y), array_data(0) + { + 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; - } - - T** getArray (void) const - { return array_data; } + { + deleteArray(); + } + + 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 () const + { return sizeof(T); } + + 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]; + } - unsigned int sizeofPixel (void) const - { return sizeof(T); } + 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 sizeofColumn (void) const - { return (sizeof(T) * ny); } - unsigned int sizeofArray (void) const - { return (sizeof(T) * nx * ny); } + Array2d& operator= (const Array2d& rhs); //assignment operator + Array2d (const Array2d& rhs); // copy constructor };