gamma.cc

torch tracking code, it is a good code

const char *help = "\
progname: gamma.cc\n\
code2html: This program reads a ppm image and perform gamma correction.\n\
version: Torch3 vision2.0, 2005\n\
(c) Sebastien Marcel ([e-mail])\n";

#include "ImageRgb.h"
#include "ImageGray.h"
#include "ipGamma.h"
#include "DiskXFile.h"
#include "xtprobeImageDiskXFile.h"
#include "CmdLine.h"

using namespace Torch;

int main(int argc, char **argv)
{
	char *image_filename;
	real gamma;
	bool verbose;
  
	// Construct the command line
  	CmdLine cmd;
	cmd.setBOption("write log", false);
	
  	// Put the help line at the beginning
  	cmd.info(help);

  	cmd.addText("\nArguments:");
  	cmd.addSCmdArg("image filename", &image_filename, "image filename");
  	cmd.addText("\nOptions:");
  	cmd.addBCmdOption("-verbose", &verbose, false, "verbose");
  	cmd.addRCmdOption("-gamma", &gamma, 1.2, "gamma correction");

	cmd.read(argc, argv);


	Image *image_in = NULL;

	image_in = new ImageRgb();
	image_in->setBOption("verbose", verbose);
	
	ImageDiskXFile *image_loader = new xtprobeImageDiskXFile(image_filename, "r");
	image_in->loadImageXFile(image_loader);
	delete image_loader;

	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);
	}

	ipCore *imachine = NULL;

	imachine = new ipGamma(image_in->width, image_in->height, "rgb", gamma);
	imachine->setBOption("verbose", verbose);
	imachine->process(image_in);

	Image *image_out = new ImageRgb();
        image_out->setBOption("verbose", verbose);

        image_out->copyFrom(image_in->width, image_in->height, imachine->seq_out->frames[0], "rgb");
		
	image_out->save("gamma.ppm");
		
	delete imachine;
	delete image_in;
	delete image_out;

	return(0);
}