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
 RCS:           $Id$
________________________________________________________________________
 
-*/
 
 
//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() );
}