PPMWriter.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 PPMWriter.ih
 * @author David Coeurjolly (\c david.coeurjolly@liris.cnrs.fr )
 * Laboratoire d'InfoRmatique en Image et Systèmes d'information - LIRIS (CNRS, UMR 5205), CNRS, France
 *
 * @date 2010/07/22
 *
 * Implementation of inline methods defined in PPMWriter.h
 *
 * This file is part of the DGtal library.
 */
 
 
//////////////////////////////////////////////////////////////////////////////
#include <cstdlib>
#include <fstream>
#include "DGtal/io/Color.h"
#include "DGtal/kernel/domains/HyperRectDomain.h"
//////////////////////////////////////////////////////////////////////////////
 
///////////////////////////////////////////////////////////////////////////////
// IMPLEMENTATION of inline methods.
///////////////////////////////////////////////////////////////////////////////
 
 
namespace DGtal {
 
  template<typename I,typename C>
  bool
  PPMWriter<I,C>::exportPPM(const std::string & filename, const I & aImage,
                            const Functor & aFunctor, bool topbotomOrder)
  {
    BOOST_STATIC_ASSERT(I::Domain::dimension == 2);
 
    std::ofstream out;
    typename I::Domain domain = aImage.domain();
    typename I::Domain::Point p = I::Domain::Point::diagonal(1);
    typename I::Domain::Vector size =  (domain.upperBound() - domain.lowerBound()) + p;
 
    typename I::Value val;
    DGtal::Color col;
    out.open(filename.c_str(), std::ios::out | std::ios::binary);
 
    //PPM format
    out << "P3"<<std::endl;
    out << "#DGtal PNM Writer"<<std::endl<<std::endl;
    out << size[0]<<" "<< size[1]<<std::endl;
    out << "255" <<std::endl;
    
    if(!topbotomOrder)
      {
        //We scan the domain instead of the image becaus we cannot
        //trust the image container Iterator
        for(typename I::Domain::ConstIterator it = domain.begin(), itend=domain.end();
            it!=itend;
            ++it)
          {
            val = aImage( (*it) );
            col = aFunctor( val );
            out << (int)col.red()<<" "<<(int)col.green()<<" "<<(int)col.blue()<<" ";
          }
      }
    else
      {
        typename I::Domain::Point ptUpper=  domain.upperBound();
        
        for(typename HyperRectDomain<typename I::Domain::Space>::ConstSubRange::ConstReverseIterator itY = domain.subRange(1, ptUpper).rbegin(),
              itYend=domain.subRange(1, ptUpper).rend();  itY!=itYend; ++itY)
          {    
            typename I::Domain::Point ptUpperY=  *itY;
            
            for(typename HyperRectDomain<typename I::Domain::Space>::ConstSubRange::ConstIterator it = domain.subRange(0, ptUpperY).begin(), 
                  itend=domain.subRange(0, ptUpperY).end();
                it!=itend;
                ++it)
              {
                val = aImage( (*it) );
                col = aFunctor( val );
                out << (int)col.red()<<" "<<(int)col.green()<<" "<<(int)col.blue()<<" ";            
              }
          }
      }
    
    out.close(); 
  
    ///@todo catch IOerror excpetion
    return true;
  }
  
  template<typename I,typename C>
bool
PPMWriter<I,C>::exportPPM3D(const std::string & filename, const I & aImage,
                            const Functor&aFunctor)
{
  BOOST_STATIC_ASSERT(I::Domain::dimension == 3);
  
  std::ofstream out;
  typename I::Domain domain(aImage.lowerBound(), aImage.upperBound());
  typename I::Domain::Point p = I::Domain::Point::diagonal(1);
  typename I::Domain::Vector size =  (domain.upperBound() - domain.lowerBound()) + p;
 
  typename I::Value val;
  DGtal::Color col;
  
  out.open(filename.c_str(), std::ios::out | std::ios::binary);
  
  //PPM format
  out << "P3-3D"<<std::endl;
  out << "#DGtal PNM Writer"<<std::endl<<std::endl;
  out << size[0]<<" "<< size[1]<<" "<< size[2]<<std::endl;
  out << "255" <<std::endl;
  
  //We scan the domain instead of the image becaus we cannot
  //trust the image container Iterator
  for(typename I::Domain::ConstIterator it = domain.begin(), itend=domain.end();
      it!=itend;
      ++it)
    {
      
      val = aImage( (*it) );
      col = aFunctor( val );
      out << (int)col.red()<<" "<<(int)col.green()<<" "<<(int)col.blue()<<" ";  }
  
  out.close(); 
  
  ///@todo catch IOerror excpetion
  return true;
}
 
 
}//namespace