PGMReader.ih

/**
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as
 *  published by the Free Software Foundation, either version 3 of the
 *  License, or  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 **/
 
/**
 * @file PGMReader.ih
 * @author Bertrand Kerautret (\c kerautre@loria.fr )
 * LORIA (CNRS, UMR 7503), University of Nancy, France
 *
 * @date 2011/04/29
 *
 * Implementation of inline methods defined in PGMReader.h
 *
 * This file is part of the DGtal library.
 */
 
///////////////////////////////////////////////////////////////////////////////
// IMPLEMENTATION of inline methods.
///////////////////////////////////////////////////////////////////////////////
 
//////////////////////////////////////////////////////////////////////////////
#include <cstdlib>
#include <iostream>
#include <fstream>
#include <sstream>
 
#include "DGtal/io/Color.h"
//////////////////////////////////////////////////////////////////////////////
 
 
 
///////////////////////////////////////////////////////////////////////////////
// Implementation of inline methods                                          //
 
 
///////////////////////////////////////////////////////////////////////////////
// Implementation of inline functions and external operators                 //
 
template <typename TImageContainer, typename TFunctor>
inline
TImageContainer 
DGtal::PGMReader<TImageContainer, TFunctor>::importPGM(const std::string & aFilename, 
                                                       const TFunctor &aFunctor,
                                                       bool topbotomOrder )
{
  std::ifstream infile;
  DGtal::IOException dgtalio;
  BOOST_STATIC_ASSERT( (ImageContainer::Domain::dimension == 2));
  try 
    {
      infile.open (aFilename.c_str(), std::ifstream::in | std::ifstream::binary);
    }
  catch( ... )
    {
      trace.error() << "PGMReader : can't open " << aFilename << std::endl;
      throw dgtalio;
    }
  bool isASCIImode = false;
  std::string str;
  getline( infile, str );
  if ( ! infile.good() )
    {
      trace.error() << "PGMReader : can't read " << aFilename << std::endl;
      throw dgtalio;
    }
  str = str.substr(0, 2);
  if ( str != "P5" &&  str != "P2")
    {
      trace.error() << "PGMReader : No P5 or P2 format in " << aFilename << std::endl;
      throw dgtalio;
    }
  if(str== "P2")
    {
      isASCIImode=true;
    }
  do
    {
      getline( infile, str );
      if ( ! infile.good() )
        {
          trace.error() << "PGMReader : Invalid format in " << aFilename << std::endl;
          throw dgtalio;
        } 
    }
  while ( str[ 0 ] == '#' || str=="");
  std::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 );
  std::istringstream str2_in( str );
  unsigned int max_value;
  str2_in >> max_value;  
 
  if ( ! infile.good() )
    {
      trace.error() << "PGMReader : Invalid format in " << aFilename << std::endl;
      throw dgtalio;
    } 
  
  unsigned int nb_read = 0;
  if(!isASCIImode)
    infile >> std::noskipws;
  else
    infile >> std::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, aFunctor(c));
                }
            }
          else
            {
              int c; 
              infile >> c;
              if ( infile.good() )
                {
                  ++nb_read;
                  image.setValue( pt, aFunctor(c));
                }
            }
      }
  if ( infile.fail() || infile.bad() )
    {
      trace.error() << "# nbread=" << nb_read << std::endl;
      throw dgtalio;
    }
  infile >> std::skipws;
  return  image;
}
 
 
 
template <typename TImageContainer, typename TFunctor>
inline
TImageContainer 
DGtal::PGMReader<TImageContainer,TFunctor>::importPGM3D(const std::string & aFilename, 
                                               const TFunctor &aFunctor)
{
  std::ifstream infile;
  DGtal::IOException dgtalio;
  BOOST_STATIC_ASSERT( (ImageContainer::Domain::dimension == 3));
  try 
    {
      infile.open (aFilename.c_str(), std::ifstream::in | std::ifstream::binary);
    }
  catch( ... )
    {
      trace.error() << "PGMReader : can't open " << aFilename << std::endl;
      throw dgtalio;
    }
 
  std::string str;
  getline( infile, str );
  if ( ! infile.good() ) {
    trace.error() << "PGMReader : can't read " << aFilename << std::endl;
    throw dgtalio;
  }
  
  if ( str.compare( 0, 2, "P2" ) != 0  && str.compare( 0, 2, "P3" ) != 0 && str.compare( 0, 2, "P5" ) != 0 ){
    trace.error() << "PGMReader : Wrong format in " << aFilename << std::endl;
    throw dgtalio;
  }
 
  bool isASCIImode = ( str.compare( 0, 2, "P2" ) == 0 );
  if(!isASCIImode)
    infile >> std::noskipws;
  else
    infile >> std::skipws;
  
  do
    {
      getline( infile, str );
      if ( ! infile.good() ){
        trace.error() << "PGMReader : Invalid format in " << aFilename << std::endl;
        throw dgtalio;
      } 
    }
  while ( str[ 0 ] == '#' || str=="");
  std::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 );
  std::istringstream str2_in( str );
  int max_value;
  str2_in >> max_value;  
 
  if ( ! infile.good() ){
    trace.error() << "PGMReader : Invalid format in " << aFilename << std::endl;
    throw dgtalio;
  } 
  unsigned int nb_read = 0;
  
  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;
 
  if(!isASCIImode)
            {
              unsigned char c; 
              infile >> c;
              if ( infile.good() )
                { 
                  ++nb_read;
                  image.setValue( pt, aFunctor(c));
                }
            }
          else
            {
              int c; 
              infile >> c;
              if ( infile.good() )
                {
                  ++nb_read;
                  image.setValue( pt, aFunctor(c));
                }
            }
      }
    }
  }
  if ( infile.fail() || infile.bad() )
    {
      trace.error() << "# nbread=" << nb_read << std::endl;
      throw dgtalio;
    }
  infile >> std::skipws;
  return  image;
}
 
 
 
//                                                                           //
///////////////////////////////////////////////////////////////////////////////