arrayndwrapper.h

Go to the documentation of this file.

#pragma once

/*+
________________________________________________________________________

 (C) dGB Beheer B.V.; (LICENSE) http://opendtect.org/OpendTect_license.txt
 Author:        Nanne Hemstra
 Date:          October 2007
________________________________________________________________________

-*/


//TODO: Write more info
mClass(Basic) ArrayNDWrapper
{
public:
    void                setDimMap(int srcdim,int targetdim)
                        { dimmap_[srcdim] = targetdim; }

    virtual void        init()          = 0;
    virtual bool        isOK() const    = 0;

protected:
                        ArrayNDWrapper(const ArrayNDInfo& info)
                        { dimmap_.setSize( info.getNDim(), 0 ); }

    TypeSet<int>        dimmap_;
};



template <class T>
mClass(Basic) Array3DWrapper : public Array3D<T>, public ArrayNDWrapper
{
public:
                        Array3DWrapper(ArrayND<T>&);
                        ~Array3DWrapper();

    void                init();
    bool                isOK() const;

    void                set(int,int,int,T);
    T                   get(int,int,int) const;

    const Array3DInfo&  info() const            { return info_; }

protected:

    Array3DInfo&        info_;
    ArrayND<T>&         srcarr_;
};



template <class T>
Array3DWrapper<T>::Array3DWrapper( ArrayND<T>& arr )
    : ArrayNDWrapper(arr.info())
    , srcarr_(arr)
    , info_(*new Array3DInfoImpl(1,1,1))
{
}

template <class T>
Array3DWrapper<T>::~Array3DWrapper()
{ delete &info_; }


template <class T>
void Array3DWrapper<T>::init()
{
    for ( int idx=0; idx<dimmap_.size(); idx++ )
        info_.setSize( dimmap_[idx], srcarr_.info().getSize(idx) );
}


template <class T>
bool Array3DWrapper<T>::isOK() const
{ return srcarr_.info().getNDim() <  3; }


template <class T>
void Array3DWrapper<T>::set( int i0, int i1, int i2, T val )
{
    int pos3d[] = { i0, i1, i2 };
    TypeSet<int> posnd;
    for ( int idx=0; idx<dimmap_.size(); idx++ )
        posnd += pos3d[ dimmap_[idx] ];

    srcarr_.setND( posnd.arr(), val );
}


template <class T>
T Array3DWrapper<T>::get( int i0, int i1, int i2 ) const
{
    int pos3d[] = { i0, i1, i2 };
    TypeSet<int> posnd;
    for ( int idx=0; idx<dimmap_.size(); idx++ )
        posnd += pos3d[ dimmap_[idx] ];

    return srcarr_.getND( posnd.arr() );
}