PNMReader.ih

// IMPLEMENTATION of inline methods.

#include <cstdlib>
#include <iostream>
#include <fstream>
#include <sstream>



// Implementation of inline methods                                          //


// Implementation of inline functions and external operators                 //

template <typename TImageContainer>
inline
TImageContainer 
DGtal::PNMReader<TImageContainer>::importPGM(const std::string & aFilename, bool topbotomOrder ) throw(DGtal::IOException)
{
  ifstream infile;
  DGtal::IOException dgtalio;
  BOOST_STATIC_ASSERT( (ImageContainer::Domain::dimension == 2));
  try 
    {
      infile.open (aFilename.c_str(), ifstream::in);
    }
  catch( ... )
    {
      trace.error() << "PNMReader : can't open " << aFilename << endl;
      throw dgtalio;
    }
  bool isASCIImode = false;
  string str;
  getline( infile, str );
  if ( ! infile.good() )
    {
      trace.error() << "PNMReader : can't read " << aFilename << endl;
      throw dgtalio;
    }
  if ( str != "P5" &&  str != "P2")
    {
      trace.error() << "PNMReader : No P5 or P2 format in " << aFilename << endl;
      throw dgtalio;
    }
  if(str== "P2")
    {
      isASCIImode=true;
    }
  do
    {
      getline( infile, str );
      if ( ! infile.good() )
        {
          trace.error() << "PNMReader : Invalid format in " << aFilename << endl;
          throw dgtalio;
        } 
    }
  while ( str[ 0 ] == '#' || str=="");
  istringstream str_in( str );
  unsigned int w, h;
  str_in >> w >> h;
  
   
  typename TImageContainer::Point firstPoint;
  typename TImageContainer::Point lastPoint;
      
  firstPoint = TImageContainer::Point::zero;
  lastPoint[0] = w-1;
  lastPoint[1] = h-1;
  
  typename TImageContainer::Domain domain(firstPoint,lastPoint);
  TImageContainer image(domain);

  getline( infile, str );
  istringstream str2_in( str );
  unsigned int max_value;
  str2_in >> max_value;  

  if ( ! infile.good() )
    {
      trace.error() << "PNMReader : Invalid format in " << aFilename << endl;
      throw dgtalio;
    } 
  
  unsigned int nb_read = 0;
  if(!isASCIImode)
    infile >> noskipws;
  else
    infile >> skipws;
  
  for(unsigned int y=0; y <h; y++)
    for(unsigned int x=0; x <w; x++)
      {
          typename TImageContainer::Point pt;
          if (topbotomOrder){
            pt[0]=x; pt[1]=h-1-y;
          }else{
            pt[0]=x; pt[1]=y;
          }
          
          
          if(!isASCIImode)
            {
              unsigned char c; 
              infile >> c;
              if ( infile.good() )
                { 
                  ++nb_read;
                  image.setValue( pt, c);
                }
            }
          else
            {
              int c; 
              infile >> c;
              if ( infile.good() )
                {
                  ++nb_read;
                  image.setValue( pt, c);
                }
            }
      }
  if ( infile.fail() || infile.bad() )
    {
      trace.error() << "# nbread=" << nb_read << endl;
      throw dgtalio;
    }
  infile >> skipws;
  return  image;
}



template <typename TImageContainer>
inline
TImageContainer 
DGtal::PNMReader<TImageContainer>::importPGM3D(const std::string & aFilename ) throw(DGtal::IOException)
{
  ifstream infile;
  DGtal::IOException dgtalio;
  BOOST_STATIC_ASSERT( (ImageContainer::Domain::dimension == 3));
  try 
    {
      infile.open (aFilename.c_str(), ifstream::in);
    }
  catch( ... )
    {
      trace.error() << "PNMReader : can't open " << aFilename << endl;
      throw dgtalio;
    }
 
  string str;
  getline( infile, str );
  if ( ! infile.good() ) {
    trace.error() << "PNMReader : can't read " << aFilename << endl;
    throw dgtalio;
  }
  if ( str != "P3d" &&  str != "P3D"){
    trace.error() << "PNMReader : No P3d format in " << aFilename << endl;
    throw dgtalio;
  }
  
  do
    {
      getline( infile, str );
      if ( ! infile.good() ){
        trace.error() << "PNMReader : Invalid format in " << aFilename << endl;
        throw dgtalio;
      } 
    }
  while ( str[ 0 ] == '#' || str=="");
  istringstream str_in( str );
  unsigned int w, h, e;
  str_in >> w >> h >> e ;

   
  typename TImageContainer::Point firstPoint;
  typename TImageContainer::Point lastPoint;
      
  firstPoint = TImageContainer::Point::zero;
  lastPoint[0] = w-1;
  lastPoint[1] = h-1;
  lastPoint[2] = e-1;

  typename TImageContainer::Domain domain(firstPoint,lastPoint);
  TImageContainer image(domain);

  getline( infile, str );
  istringstream str2_in( str );
  int max_value;
  str2_in >> max_value;  

  if ( ! infile.good() ){
    trace.error() << "PNMReader : Invalid format in " << aFilename << endl;
    throw dgtalio;
  } 
  unsigned int nb_read = 0;

  infile >> noskipws;
  for(unsigned int z=0; z <e; z++){
    for(unsigned int y=0; y <h; y++){
      for(unsigned int x=0; x <w; x++){
        typename TImageContainer::Point pt;
        pt[0]=x; pt[1]=y; pt[2]=z;
        char c; 
        infile >> c;
        if ( infile.good() ){ 
          ++nb_read;
          image.setValue( pt, c);
        }
      }
    }
  }
  if ( infile.fail() || infile.bad() )
    {
      trace.error() << "# nbread=" << nb_read << endl;
      throw dgtalio;
    }
  infile >> skipws;
  return  image;
}



//                                                                           //