artificiallight.cc

torch tracking code, it is a good code

const char *help = "\
progname: artificial_light.cc\n\
code2html: This program reads a pgm image (grayscale) and artificially relight it.\n\
\n\
version: Torch3 vision2.0, 2005\n\
(c) Guillaume Heusch ([e-mail])\n";

#include "ImageGray.h"
#include "DiskXFile.h"
#include "CmdLine.h"
#include "ipArtiLight.h"

using namespace Torch;

int main(int argc, char **argv)
{
  
  char *image_filename;
  char *out_filename;

  bool verbose;
  int delta;
  

  // --------------------- COMMAND LINE -------------------------------------------------------------------------
  CmdLine cmd;
  cmd.setBOption("write log", false);
  cmd.info(help);

  cmd.addText("\nArguments:");
  cmd.addSCmdArg("image_filename", &image_filename, "input image filename");
  cmd.addSCmdArg("out_filename", &out_filename, "processed image filename");

  cmd.addText("\nOptions:");
  cmd.addBCmdOption("-verbose", &verbose, false, "verbose");
  cmd.addICmdOption("-delta", &delta, 80, "illumination delta (in pixels)");

  cmd.read(argc, argv);

  Allocator *allocator = new Allocator;


  // ----------------------- LOAD IMAGE --------------------------------------------------------------------------
  DiskXFile *image_file = NULL;
  Image *image_in = NULL;

  image_in = new(allocator) ImageGray();
  image_in->setBOption("verbose", verbose);
	
  image_file = new(allocator) DiskXFile(image_filename, "r");
  image_in->loadXFile(image_file);

  if(verbose)
    {
      print("Image info:\n");
      print("   width = %d\n", image_in->width);
      print("   height = %d\n", image_in->height);
      print("   format = %s (%d)\n", image_in->coding, image_in->n_planes);
    }
  
  

  // ---------------------- PROCESSING IMAGE MACHINE ------------------------------------------------------------	
  ipCore *artiLight = NULL;
  artiLight = new(allocator) ipArtiLight(delta, image_in->width, image_in->height, "gray");
  artiLight->setBOption("verbose", verbose);
  artiLight->process(image_in);


  // ----------------------- SAVE IMAGE(S)  --------------------------------------------------------------------------
  Image *image_out = NULL;
  image_out = new(allocator) ImageGray();
  image_out->setBOption("verbose", verbose);

  image_out->copyFrom(image_in->width, image_in->height, artiLight->seq_out->frames[0], "gray", 255);
 
  image_out->updatePixmapFromData();
  image_out->save(out_filename);

  // ---------------------- CLEAN UP -----------------------------------------------------------------------------
  delete allocator;
  	
  return(0);	
}