testPNMRawWriter.cpp

#include <iostream>
#include "DGtal/base/Common.h"
#include "DGtal/kernel/SpaceND.h"
#include "DGtal/kernel/domains/HyperRectDomain.h"
#include "DGtal/images/ImageSelector.h"
#include "DGtal/io/colormaps/GrayscaleColorMap.h"
#include "DGtal/io/colormaps/HueShadeColorMap.h"
#include "DGtal/io/colormaps/GradientColorMap.h"
#include "DGtal/io/colormaps/ColorBrightnessColorMap.h"
#include "DGtal/io/colormaps/RandomColorMap.h"
#include "DGtal/io/colormaps/BasicColorToScalarFunctors.h"
#include "DGtal/base/BasicFunctors.h"

#include "DGtal/io/writers/PGMWriter.h"
#include "DGtal/io/writers/PPMWriter.h"
#include "DGtal/io/readers/PNMReader.h"
#include "DGtal/io/writers/RawWriter.h"
#include "DGtal/io/boards/Board2D.h"

using namespace std;
using namespace DGtal;

// Functions for testing class PNMWriter.

bool testPNMWriter()
{
  
  trace.beginBlock ( "Testing block ..." );

  typedef SpaceND<2> TSpace;
  typedef TSpace::Point Point;
  typedef HyperRectDomain<TSpace> Domain;
  typedef HueShadeColorMap<unsigned char> Hue;
  typedef RandomColorMap  Rand;
  typedef HueShadeColorMap<unsigned char,2> HueTwice;
  typedef GrayscaleColorMap<unsigned char> Gray;
  // Gradient using the "Jet" preset.
  typedef GradientColorMap<unsigned char, CMAP_JET > Jet;
  // Gradient from black to red.
  const int BlackColor = DGTAL_RGB2INT(0,0,0);
  const int RedColor = DGTAL_RGB2INT(255,0,0);
  typedef GradientColorMap< unsigned char, CMAP_CUSTOM, BlackColor, RedColor > RedShade1;
  // Gradient from black to red, using a ColorBrightnessColorMap.
  typedef ColorBrightnessColorMap< unsigned char, RedColor > RedShade2;

  Point a ( 1, 1);
  Point b ( 16, 16);
  typedef ImageSelector<Domain, unsigned char>::Type Image;
  Image image(Domain(a,b));
  for(unsigned int i=0 ; i < 256; i++)
    image[i] = i;

  PPMWriter<Image,Hue>::exportPPM("export-hue.ppm",image, Hue(0,255) );
  PPMWriter<Image,HueTwice>::exportPPM("export-hue-twice.ppm",image,HueTwice(0,255));
  PGMWriter<Image>::exportPGM("export-hue-twice.pgm",image);
  PPMWriter<Image,Gray>::exportPPM("export-gray.ppm",image, Gray(0,255));
  PPMWriter<Image,Jet>::exportPPM("export-jet.ppm",image,Jet(0,255));
  PPMWriter<Image,RedShade1>::exportPPM("export-red1.ppm",image,RedShade1(0,255));
  PPMWriter<Image,RedShade2>::exportPPM("export-red2.ppm",image,RedShade2(0,255));

  //TestingFunctor
  typedef Composer< Jet, BasicColorToScalarFunctors::RedChannel, unsigned char> RedFunctor;
  RedFunctor redFunctor( Jet(0,255), BasicColorToScalarFunctors::RedChannel() ) ;
  PGMWriter<Image, RedFunctor>::exportPGM("export-jet-red.pgm",image, redFunctor);
  

  //test Raw export
  RawWriter<Image>::exportRaw8("export-hue-twice.raw",image);

  //test Image export with libboard
  Board2D  board;
  board.setUnit(LibBoard::Board::UCentimeter);
  Display2DFactory::drawImage<HueTwice>(board, image, (unsigned char)0, (unsigned char)255);
  board.saveSVG("export-hue-twice.svg");

  trace.endBlock();
  
  return true;
}


bool testRWIssue254()
{
   trace.beginBlock ( "Testing R/W on PPM Writer (issue 254) ..." );

  typedef SpaceND<2> TSpace;
  typedef TSpace::Point Point;
  typedef HyperRectDomain<TSpace> Domain;
  typedef GrayscaleColorMap<unsigned char> Gray;

  Point a ( 0, 0);
  Point b ( 15, 15);
  typedef ImageSelector<Domain, unsigned char>::Type Image;
  Domain domain(a,b); 
  Image image(domain);
  for(unsigned int i=0 ; i < 256; i++)
    image[i] = i;

  PGMWriter<Image>::exportPGM("export-gray-first.pgm",image);
  
  Image imageRead = PNMReader<Image>::importPGM("export-gray-first.pgm");

  PGMWriter<Image>::exportPGM("export-gray-second.pgm",imageRead);

  trace.info() << image<<std::endl;
  trace.info() << imageRead<<std::endl;
  
  
  bool ok = true;
  for(Image::Domain::ConstIterator it = image.domain().begin(),
        itend = image.domain().end();
      it != itend;
      ++it)
    ok = (image(*it) == imageRead(*it));
    
  trace.endBlock();
  return ok;
}

// Standard services - public :

int main( int argc, char** argv )
{
  trace.beginBlock ( "Testing class PNMWriter" );
  trace.info() << "Args:";
  for ( int i = 0; i < argc; ++i )
    trace.info() << " " << argv[ i ];
  trace.info() << endl;
  
  bool res = testPNMWriter() && testRWIssue254(); // && ... other tests
  trace.emphase() << ( res ? "Passed." : "Error." ) << endl;
  trace.endBlock();
  return res ? 0 : 1;
}

//                                                                           //