image_loader.h

Go to the documentation of this file.

#ifndef MMVL_IMAGE_LOADER_HH_INCLUDED
#define MMVL_IMAGE_LOADER_HH_INCLUDED

#include <string>
#include <iostream>
#include <iomanip>
#include <sstream>
#include <fstream>
#include <image.h>
#include <image_fileinput.h>
#include "mimasexception.h"

namespace mimas{
  namespace hf{
    template<typename T>
    class image_loader{
    public:
      image<T> &image_IL;
      
      class set{
      public:
  std::string basename;
  int nb_images; 
  
  int image_nb;
  std::string extension;
  int size;
  int first_image_number;
  
  set(std::string basename, int size = 4, std::string extension = ".pgm",
      int first_image_number = 0, int nb_images = 0)
    :basename(basename), nb_images(nb_images), extension(extension),
     size(size),first_image_number(first_image_number){} 
      };
      
      //carefull, if it is not used as a policy you need to load the first image by
      //a call to next() which is not very satisfying in the sense that when
      //the class is instanciated it is not directly usable.
      image_loader(image<T> &image, set param)
        :image_IL(image), basename(param.basename), nb_images(param.nb_images),
  image_nb(param.first_image_number),extension(param.extension),size(param.size)
      {
  next(); //load the first image so the object is ready to use 
      }

      //"Normal" constructor
      image_loader(image<T> &image, std::string basename, int size, std::string extension, int first_image_number = 0, int nb_images = 0)
        :image_IL(image), basename(basename), nb_images(nb_images),
  image_nb(first_image_number),extension(extension),size(size)
      {
  next(); //load the first image so the object is ready to use 
      }


      //debug code
      std::string fileName(void)
      {
  std::ostringstream str;
  str << basename << std::setw( size ) << std::setfill( '0' ) <<  image_nb
      << extension;
#ifndef NDEBUG
  std::cerr<< "File name is: " << str.str() <<std::endl; 
#endif
  return str.str();
      }

      void next(void) throw(mimasexception)
      {
  //put an ending condition
  //if nb_images equal 0 we do not consider their is a maximal number of images.
  //An exception will be raised when the filename does not correspond to any file
  if(image_nb < nb_images || nb_images == 0)
    {
      std::ifstream cimage_file(fileName().c_str(), std::ios::binary );
      if (cimage_file.fail())
        MMERROR(false , mimasexception, ,
          "failed to open file");  
      cimage_file >> image_IL;
      image_nb++;
      return;
    }
  else
    MMERROR(false , mimasexception, ,
       "max number of image to read reached");
      }

    private:

      std::string basename;
      int nb_images; 

      int image_nb;
      std::string extension;
      int size;
    };

  }
}

#endif

---