testImage2D.cpp

#include <iostream>
#include <fstream>
#include <sstream>
#include "Image2D.hpp"
#include "Image2DReader.hpp"
#include "Image2DWriter.hpp"
 
int main( int argc, char** argv )
{
  typedef unsigned char GrayLevel;
  typedef Image2D<GrayLevel> GrayLevelImage2D;
  typedef Image2DReader<GrayLevel> GrayLevelImage2DReader;
  typedef Image2DWriter<GrayLevel> GrayLevelImage2DWriter;
 
  if ( argc < 3 ) 
    { 
      std::cerr << "Usage: testImage2D <input.pgm> <output.pgm> <kernel_size>" << std::endl;
      return 0;
    }
  int kernel = (argc > 3) ? atoi( argv[ 3 ] ) : 1;
  GrayLevelImage2D img;
  std::ifstream input( argv[1] ); // récupère le 1er argument.
  bool ok = GrayLevelImage2DReader::read( img, input );
  if ( !ok )
    {
      std::cerr << "Error reading input file." << std::endl;
       return 1;
    }
  input.close();
 
  GrayLevelImage2D img2( img.w(), img.h() );
  std::vector<GrayLevel> values;
  for ( GrayLevelImage2D::Iterator outIt = img2.begin(), outItE = img2.end();
        outIt != outItE; ++outIt )
      {
        std::pair<int,int> p = img2.position( outIt );
        int xmin = std::max( p.first-kernel, 0 );
        int ymin = std::max( p.second-kernel, 0 );
        int xmax = std::min( p.first+kernel, img.w()-1 );
        int ymax = std::min( p.second+kernel, img.h()-1 );
        GrayLevelImage2D::SimpleImageConstIterator it2d = img.start2d( xmin, ymin );
        GrayLevelImage2D::SimpleImageConstIterator it2dE = img.start2d( xmax, ymax );
        for ( GrayLevelImage2D::SimpleImageConstIterator it2dRow = it2d; 
              it2dRow.y <= it2dE.y; 
              ++it2dRow.y )
          for ( GrayLevelImage2D::SimpleImageConstIterator it2dCol = it2dRow; 
                it2dCol.x <= it2dE.x; ++it2dCol.x )
            values.push_back( *it2dCol );
        int mid = values.size() / 2;
        std::nth_element( values.begin(), values.begin() + mid, values.end() );
        *outIt = values[ mid ];
        values.clear();
      }
  std::ofstream output( argv[2] ); // récupère le 2ème argument.
  bool ok2 = GrayLevelImage2DWriter::write( img2, output, false );
  if ( !ok2 )
    {
      std::cerr << "Error writing output file." << std::endl;
       return 1;
    }
  output.close();
 
 
  return 0;
}