testImageSpanIterators.cpp

#include <cstdio>
#include <cmath>
#include <iostream>

#include "DGtal/base/Common.h"
#include "DGtal/kernel/SpaceND.h"
#include "DGtal/kernel/domains/HyperRectDomain.h"
#include "DGtal/images/ImageContainerBySTLVector.h"

using namespace DGtal;
using namespace std;


bool testSpanIterators()
{
    typedef SpaceND<3> Space3Type;
    typedef Space3Type::Point Point;
    typedef HyperRectDomain<Space3Type> TDomain;
    typedef ImageContainerBySTLVector<TDomain, double> TContainerV;
    
    const int t[ ] = { 0, 0, 0};
    const int t2[ ] = { 5, 5, 5};
    const int t3[ ] = { 0, 0, 0};
    Point a ( t );
    Point b ( t2 );
    Point c( t3 );

    trace.beginBlock("Test of Concepts");
    TContainerV myImageV ( TDomain(a,b ));

    double cpt=0;
    //Image Construction 
    for ( TContainerV::Iterator it = myImageV.begin();
            it != myImageV.end();
            ++it)
    {
      (*it) = cpt ;
      cpt++;
    }


    //We process a 1D slice
    trace.info()<<"Slice dim 0 ";
    for ( TContainerV::SpanIterator it = myImageV.spanBegin(c,0), itend = myImageV.spanEnd(c,0);
    it != itend;
    ++it)
      trace.info() << (*it)<<" ";
    trace.info() << endl;
    
    //We process a 1D slice
    trace.info()<<"Slice dim 1 ";
    for ( TContainerV::SpanIterator it = myImageV.spanBegin(c,1), itend = myImageV.spanEnd(c,1);
    it != itend;
    ++it)
      trace.info() << (*it)<<" ";
    trace.info() << endl;
  
    //We process a 1D slice
    trace.info()<<"Slice dim 2 ";
    for ( TContainerV::SpanIterator it = myImageV.spanBegin(c,2), itend = myImageV.spanEnd(c,2);
    it != itend;
    ++it)
      trace.info() << (*it)<<" ";
    trace.info() << endl;
  

    //We process a 1D slice to set a value
    trace.info()<<"SetValue Slice dim 1 "<<endl;
    for ( TContainerV::SpanIterator it = myImageV.spanBegin(c,1), itend = myImageV.spanEnd(c,1);
    it != itend;
    ++it)
      myImageV.setValue(it, 12);
    
    //We check the a 1D slice
    trace.info()<<"Check Slice dim 1 ";
    for ( TContainerV::SpanIterator it = myImageV.spanBegin(c,1), itend = myImageV.spanEnd(c,1);
    it != itend;
    ++it)
      trace.info() << (*it)<<" ";
    trace.info() << endl;
  

  
    trace.endBlock();

    return true;

}



int main()
{

    if ( testSpanIterators())
        return 0;
    else
        return 1;
}