testSTBReader.cpp

Go to the documentation of this file.

#include <iostream>
#include "DGtal/base/Common.h"
#include "ConfigTest.h"
#include "DGtalCatch.h"
#include "DGtal/helpers/StdDefs.h"
#include "DGtal/io/readers/STBReader.h"
#include "DGtal/images/ImageSelector.h"
 
#include "DGtal/io/writers/PPMWriter.h"

 
using namespace std;
using namespace DGtal;

// Functions for testing class STBReader.
 
TEST_CASE( "Testing STBReader" )
{
  SECTION("Testing feature io/readers of STBReader (Grayscale PNG)")
  {
    std::string filename = testPath + "samples/contourS.png";
    typedef ImageSelector < Z2i::Domain, Color>::Type Image;
    Image image = STBReader<Image>::import( filename );
    CAPTURE(image);
 
    PPMWriter<Image>::exportPPM("testwriter.ppm", image);
    CHECK( image.isValid());
  }
 
  SECTION("Testing feature io/readers of STBReader (PPM color)")
  {
    std::string filename = testPath + "samples/color64.";
    typedef ImageSelector < Z2i::Domain, Color>::Type Image;
 
    Image image = STBReader<Image>::import( filename+"ppm" );
    PPMWriter<Image>::exportPPM("testwriterColor.ppm", image);
    CHECK( image.isValid());
  }
 
  SECTION("Testing all file formats")
  {
    std::string filename = testPath + "samples/color64.";
    typedef ImageSelector < Z2i::Domain, Color>::Type Image;
 
    Image imagePPM = STBReader<Image>::import( filename+"ppm" );
    CHECK( imagePPM.isValid());
    Image imageJPG = STBReader<Image>::import( filename+"jpg" );
    CHECK( imageJPG.isValid());
    Image imageTGA = STBReader<Image>::import( filename+"tga" );
    CHECK( imageTGA.isValid());
    Image imageBMP = STBReader<Image>::import( filename+"bmp" );
    CHECK( imageBMP.isValid());
    Image imagePNG = STBReader<Image>::import( filename+"png" );
    CHECK( imagePNG.isValid());
    Image imageGIF = STBReader<Image>::import( filename+"gif" );
    CHECK( imageGIF.isValid());
 
    //For lossless compression formats
    CHECK( imageBMP == imagePPM );
    CHECK( imageTGA == imagePPM );
    CHECK( imagePNG == imagePPM );
  }
}