tgadiff.cpp

来自「hl2 source code. Do not use it illegal.」· C++ 代码 · 共 104 行

#include <stdlib.h>
#include <stdio.h>
#include "tgaloader.h"
#include "tgawriter.h"

void Usage( void )
{
	printf( "Usage: tgadiff src1.tga src2.tga diff.tga\n" );
	exit( -1 );
}

int main( int argc, char **argv )
{
	if( argc != 4 )
	{
		Usage();
	}
	const char *pSrcFileName1 = argv[1];
	const char *pSrcFileName2 = argv[2];
	const char *pDstFileName = argv[3];

	int width1, height1;
	ImageFormat imageFormat1;
	float gamma1;
	if( !TGALoader::GetInfo( pSrcFileName1, &width1, &height1, &imageFormat1, &gamma1 ) )
	{
		printf( "error loading %s\n", pSrcFileName1 );
		exit( -1 );
	}

	int width2, height2;
	ImageFormat imageFormat2;
	float gamma2;
	if( !TGALoader::GetInfo( pSrcFileName2, &width2, &height2, &imageFormat2, &gamma2 ) )
	{
		printf( "error loading %s\n", pSrcFileName2 );
		exit( -1 );
	}

	if( width1 != width2 || height1 != height2 || imageFormat1 != imageFormat2 || gamma1 != gamma2 )
	{
		printf( "image geometry/format/gamma different. . can't do diff for %s\n", pDstFileName );
		exit( -1 );
	}

	unsigned char *pImage1 = new unsigned char[ImageLoader::GetMemRequired( width1, height1, imageFormat1, false )];
	unsigned char *pImage2 = new unsigned char[ImageLoader::GetMemRequired( width2, height2, imageFormat2, false )];
	unsigned char *pDiff = new unsigned char[ImageLoader::GetMemRequired( width2, height2, imageFormat1, false )];

	if( !TGALoader::Load( pImage1, pSrcFileName1, width1, height1, imageFormat1, 2.2f, false ) )
	{
		printf( "error loading %s\n", pSrcFileName1 );
		exit( -1 );
	}
	
	if( !TGALoader::Load( pImage2, pSrcFileName2, width2, height2, imageFormat2, 2.2f, false ) )
	{
		printf( "error loading %s\n", pSrcFileName2 );
		exit( -1 );
	}

	int i;
	int sizeInBytes = ImageLoader::SizeInBytes( imageFormat1 );
	bool isDifferent = false;
	for( i = 0; i < width1 * height1 * sizeInBytes; i++ )
	{
		int d;
		d = pImage2[i] - pImage1[i];
		pDiff[i] = d > 0 ? d : -d;
		if( d != 0 )
		{
			isDifferent = true;
		}
	}

	if( !isDifferent )
	{
		printf( "Files are the same %s %s : not generating %s\n", pSrcFileName1, pSrcFileName2, pDstFileName );
		exit( -1 );
	}
	else
	{
		printf( "Generating diff: %s!\n", pDstFileName );
	}

	ImageFormat dstImageFormat;
	if( sizeInBytes == 3 )
	{
		dstImageFormat = IMAGE_FORMAT_RGB888;
	}
	else
	{
		dstImageFormat = IMAGE_FORMAT_RGBA8888;
	}
	
	if( !TGAWriter::Write( pDiff, pDstFileName, width1, height1, dstImageFormat, dstImageFormat ) )
	{
		printf( "error writing %s\n", pDstFileName );
		exit( -1 );
	}
	
	return 0;	
}