freeman2img.cpp

#include <iostream>

#include "DGtal/base/Common.h"
#include "DGtal/helpers/StdDefs.h"

//image
#include "DGtal/io/readers/PointListReader.h"
#include "DGtal/io/writers/GenericWriter.h"

//contour
#include "DGtal/geometry/curves/FreemanChain.h"
#include "DGtal/topology/helpers/Surfaces.h"
#include "DGtal/topology/SurfelSetPredicate.h"

//STL
#include <vector>
#include <string>

#include "CLI11.hpp"


using namespace DGtal;


int main( int argc, char** argv )
{

  // parse command line using CLI ----------------------------------------------
  CLI::App app;
  std::string inputFileName;
  std::string outputFileName="result.pgm";
  unsigned int border {0};
  std::vector<int> space;

  app.description("Transform one or several freeman chains into an grayscale image file by filling its interior areas.\nThe transformation can fill shapes with hole by using the freemanchain orientation.  The interior is considered on the left according to a freeman chain move, i.e. a clockwise oriented contour represents a hole in the shape. Basic example:\n \t freeman2img  -i  inputChain.fc -o contourDisplay.pgm -b 5");    
  app.add_option("-i,--input,1", inputFileName, "Input freeman chain file name." )
    ->required()
    ->check(CLI::ExistingFile);
  app.add_option("-b,--border",border, "add a border in the resulting image (used only in the automatic mode i.e when --space is not used.");
  app.add_option("-o,--output,2", outputFileName, "the output fileName", true);
  app.add_option("-s,--space", space, "Define the space from its bounding box (lower and upper coordinates) else the space is automatically defined from the freemanchain bounding boxes." )
    ->expected(4);

  
  app.get_formatter()->column_width(40);
  CLI11_PARSE(app, argc, argv);
  // END parse command line using CLI ----------------------------------------------

    
  typedef KhalimskySpaceND<2, int>::Cell Cell;
  typedef KhalimskySpaceND<2, int>::SCell SCell;
  typedef FreemanChain<Z2i::Integer> FreemanChain;
  typedef DGtal::KhalimskySpaceND< 2, int > KSpace;
  typedef DGtal::ImageContainerBySTLVector<Z2i::Domain, unsigned char> Image2D ; 
      
    
  std::vector< FreemanChain > vectFcs =  PointListReader< Z2i::Point >::getFreemanChainsFromFile<Z2i::Integer> (inputFileName);    
  int minx=std::numeric_limits<int>::max();
  int miny=std::numeric_limits<int>::max();
  int maxx=std::numeric_limits<int>::min();
  int maxy=std::numeric_limits<int>::min();

  
  if(space.size()!=4){
      for(std::vector< FreemanChain >::const_iterator it = vectFcs.begin(); it!= vectFcs.end(); it++){
        FreemanChain fc = *it;
        int t_minx=std::numeric_limits<int>::max();
        int t_miny=std::numeric_limits<int>::max();
        int t_maxx=std::numeric_limits<int>::min();
        int t_maxy=std::numeric_limits<int>::min();
        fc.computeBoundingBox(t_minx, t_miny, t_maxx, t_maxy);
        minx = t_minx > minx? minx: t_minx;
        miny = t_miny > miny? miny: t_miny;
        maxx = t_maxx < maxx? maxx: t_maxx;
        maxy = t_maxy < maxy? maxy: t_maxy;
      }
      minx-=border; miny-=border; maxx+=border;   maxy+=border;
    }else{
      minx = space[0];
      miny = space[1];
      maxx = space[2];
      maxy = space[3];      
    }
    KSpace aKSpace;
    aKSpace.init(Z2i::Point(minx, miny), Z2i::Point(maxx, maxy), true);
    std::set<SCell> boundarySCell;
    std::set<Cell> interiorCell;
    for(std::vector< FreemanChain >::const_iterator it = vectFcs.begin(); it!= vectFcs.end(); it++){
      FreemanChain fc = *it;
      FreemanChain::getInterPixelLinels(aKSpace, fc, boundarySCell, true);
    }
    
    Image2D imageResult (Z2i::Domain(Z2i::Point(minx, miny), Z2i::Point(maxx, maxy))); 
    Surfaces<KSpace>::uFillInterior(aKSpace, functors::SurfelSetPredicate<std::set<SCell>,SCell>(boundarySCell), imageResult, 255, false, false );  
    imageResult >> outputFileName;

}