|| || || || || || || ||

---

ArrayPool.h

Go to the documentation of this file.

#ifndef Impala_Core_Array_ArrayPool_h
#define Impala_Core_Array_ArrayPool_h

#include "Core/Array/Arrays.h"
#ifdef PX_HORUS_USED
#include "Ops/Array/PxHorus/PxSystem.h"
#include "mpi.h"
#endif

namespace Impala
{
namespace Core
{
namespace Array
{

class ArrayPool
{
public:

    ArrayPool(int width, int height, int nrViewX, int nrViewY)
    {

#ifdef PX_HORUS_USED
        int sizes[2];
        sizes[0] = width;
        sizes[1] = height;
        MPI_Bcast(sizes, 2, MPI_INT, PxRootCPU(), MPI_COMM_WORLD);
        mWidth  = sizes[0];
        mHeight = sizes[1];
#else
        mWidth = width;
        mHeight = height;
#endif

        mNrViewX = nrViewX;
        mNrViewY = nrViewY;
        mArV3UI8 = new Array2dVec3UInt8*[mNrViewX * mNrViewY];
        for (int i=0 ; i<mNrViewX*mNrViewY ; i++)
            mArV3UI8[i] = ArrayCreate<Array2dVec3UInt8>(mWidth, mHeight);
        mArV3R64 = 0;
        mNrV3R64 = 0;
        mArSR64 = 0;
        mNrSR64 = 0;
        mArC64 = 0;
        mNrC64 = 0;
    }

    ~ArrayPool()
    {
        int i;
        for (i=0 ; i<mNrViewX*mNrViewY ; i++)
            delete mArV3UI8[i];
        delete mArV3UI8;

        for (i=0 ; i<mNrV3R64 ; i++)
            delete mArV3R64[i];
        delete mArV3R64;

        for (i=0 ; i<mNrSR64 ; i++)
            delete mArSR64[i];
        delete mArSR64;

        for (i=0 ; i<mNrC64 ; i++)
            delete mArC64[i];
        delete mArC64;
    }

    int 
    Width()
    {
        return mWidth;
    }
    
    int
    Height()
    {
        return mHeight;
    }

    Array2dVec3UInt8*
    GetV3UI8(int viewX, int viewY)
    {
        return mArV3UI8[viewY * mNrViewX + viewX];
    }

    UInt8*
    GetV3UI8PB(int viewX, int viewY)
    {
        return mArV3UI8[viewY * mNrViewX + viewX]->PB();
    }

    void
    AllocV3R64(int nrIm, int bw = 0, int bh = 0)
    {
        mNrV3R64 = nrIm;
        mArV3R64 = new Array2dVec3Real64*[mNrV3R64];
        for (int i=0 ; i<mNrV3R64 ; i++)
            mArV3R64[i] = ArrayCreate<Array2dVec3Real64>(mWidth, mHeight,
                                                         bw, bh);
    }

    Array2dVec3Real64*
    GetV3R64(int nr)
    {
        return mArV3R64[nr];
    }

    void
    AllocSR64(int nrIm, int bw = 0, int bh = 0)
    {
        mNrSR64 = nrIm;
        mArSR64 = new Array2dScalarReal64*[mNrSR64];
        for (int i=0 ; i<mNrSR64 ; i++)
            mArSR64[i] = ArrayCreate<Array2dScalarReal64>(mWidth, mHeight,
                                                          bw, bh);
    }

    Array2dScalarReal64*
    GetSR64(int nr)
    {
        return mArSR64[nr];
    }

    void
    AllocC64(int nrIm, int bw = 0, int bh = 0)
    {
        mNrC64 = nrIm;
        mArC64 = new Array2dComplex64*[mNrC64];
        for (int i=0 ; i<mNrC64 ; i++)
            mArC64[i] = ArrayCreate<Array2dComplex64>(mWidth, mHeight,
                                                      bw, bh);
    }

    Array2dComplex64*
    GetC64(int nr)
    {
        return mArC64[nr];
    }

private:

    int mWidth;
    int mHeight;
    int mNrViewX;
    int mNrViewY;

    Array2dVec3UInt8**    mArV3UI8;
    Array2dVec3Real64**   mArV3R64;
    int                   mNrV3R64;
    Array2dScalarReal64** mArSR64;
    int                   mNrSR64;
    Array2dComplex64**    mArC64;
    int                   mNrC64;
};

} // namespace Array
} // namespace Core
} // namespace Impala

#endif

---