SliceSelection.hpp

#ifndef __FIBEROPERATIONS_SLICE_SELECTION
#define __FIBEROPERATIONS_SLICE_SELECTION

#include <vector/MultiArray.hpp>
#include <vector/MultiCopy.hpp>

namespace Fiber
{

struct  SliceSelection
{
        int     X, Y, Z;

        void setup(int orient)
        {
                switch( orient % 3 )
                {
                default:
                case 2:  X = 0; Y = 1; Z = 2; break;
                case 0:  X = 1; Y = 2; Z = 0; break;
                case 1:  X = 2; Y = 0; Z = 1; break;
                }
        }


        SliceSelection(int o = 0)
        {
                setup(o);
        }
};


template <class Dest, class Source, class Converter>
void    ExtractSlice2D(MultiArray<2, Dest>&DestArray, const MultiArray<3, Source>&SrcArray, 
                       const SliceSelection&O, const index_t&SliceNumber,
                       const Converter&C)
{
index_t DimsX = SrcArray.Size()[O.X],
        DimsY = SrcArray.Size()[O.Y]; 

        assert( DestArray.Size()[0] >= DimsX );
        assert( DestArray.Size()[1] >= DimsY );

        for(index_t y = 0; y<DimsY; y++)
        for(index_t x = 0; x<DimsX; x++)
        {
        MultiIndex<3> P;
                P[O.X] = x; P[O.Y] = y; P[O.Z] = SliceNumber;

        MultiIndex<2> Pdst = MIndex(x,y);

                C.copy( DestArray[ Pdst ], SrcArray[ P ]);
        }
}

template <class Dest, class Source, class Converter>
void    ExtractSlice2D(MultiArray<2, Dest>&DestArray, const MultiArray<3, Source>&SrcArray, 
                       const MultiIndex<3>&SourceOffset,
                       const SliceSelection&O, const index_t&SliceNumber,
                       const Converter&C)
{
MultiIndex<3> Dims = SrcArray.Size();

        for(index_t y = 0; y<Dims[O.Y]; y++)
        for(index_t x = 0; x<Dims[O.X]; x++)
        {
        MultiIndex<3> P = SourceOffset;
                P[O.X] += x; P[O.Y] += y; P[O.Z] += SliceNumber;

        MultiIndex<2> Pdst = MIndex(x,y);

                C.copy( DestArray[ Pdst ], SrcArray[ P ]);
        } 
}


} // namespace Fiber

#endif // __FIBEROPERATIONS_SLICE_SELECTION