// TIFFTranslatorTest.cpp
#include "TIFFTranslatorTest.h"
#include <cppunit/Test.h>
#include <cppunit/TestCaller.h>
#include <cppunit/TestSuite.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <image.h>
#include <Translator.h>
#include <TranslatorFormats.h>
#include <TranslatorRoster.h>
#include <Message.h>
#include <View.h>
#include <Rect.h>
#include <File.h>
#include <DataIO.h>
#include <Errors.h>
#include <OS.h>
#include "TranslatorTestAddOn.h"

// Suite
CppUnit::Test *
TIFFTranslatorTest::Suite()
{
	CppUnit::TestSuite *suite = new CppUnit::TestSuite();
	typedef CppUnit::TestCaller<TIFFTranslatorTest> TC;
			
	suite->addTest(
		new TC("TIFFTranslator IdentifyTest",
			&TIFFTranslatorTest::IdentifyTest));

	suite->addTest(
		new TC("TIFFTranslator TranslateTest",
			&TIFFTranslatorTest::TranslateTest));	

#if !TEST_R5
	suite->addTest(
		new TC("TIFFTranslator LoadAddOnTest",
			&TIFFTranslatorTest::LoadAddOnTest));
#endif
		
	return suite;
}		

// setUp
void
TIFFTranslatorTest::setUp()
{
	BTestCase::setUp();
}
	
// tearDown
void
TIFFTranslatorTest::tearDown()
{
	BTestCase::tearDown();
}

void
CheckBits_Tiff(translator_info *pti)
{
	CheckTranslatorInfo(pti, B_TRANSLATOR_BITMAP, B_TRANSLATOR_BITMAP,
		0.7f, 0.6f, "Be Bitmap Format (TIFFTranslator)",
		"image/x-be-bitmap");
}

void
CheckTiff(translator_info *pti, const char *imageType)
{
	CheckTranslatorInfo(pti, B_TIFF_FORMAT, B_TRANSLATOR_BITMAP,
		0.7f, 0.6f, imageType, "image/tiff");
}

// coveniently group path of image with
// the expected Identify() string for that image
struct IdentifyInfo {
	const char *imagePath;
	const char *identifyString;
};

void
IdentifyTests(TIFFTranslatorTest *ptest, BTranslatorRoster *proster,
	const IdentifyInfo *pinfo, int32 len, bool bbits)
{
	translator_info ti;
	printf(" [%d] ", (int) bbits);
	
	for (int32 i = 0; i < len; i++) {
		ptest->NextSubTest();
		BFile file;
		printf(" [%s] ", pinfo[i].imagePath);
		CPPUNIT_ASSERT(file.SetTo(pinfo[i].imagePath, B_READ_ONLY) == B_OK);

		// Identify (output: B_TRANSLATOR_ANY_TYPE)
		ptest->NextSubTest();
		memset(&ti, 0, sizeof(translator_info));
		CPPUNIT_ASSERT(proster->Identify(&file, NULL, &ti) == B_OK);
		if (bbits)
			CheckBits_Tiff(&ti);
		else
			CheckTiff(&ti, pinfo[i].identifyString);
	
		// Identify (output: B_TRANSLATOR_BITMAP)
		ptest->NextSubTest();
		memset(&ti, 0, sizeof(translator_info));
		CPPUNIT_ASSERT(proster->Identify(&file, NULL, &ti, 0, NULL,
			B_TRANSLATOR_BITMAP) == B_OK);
		if (bbits)
			CheckBits_Tiff(&ti);
		else
			CheckTiff(&ti, pinfo[i].identifyString);
	
		// Identify (output: B_TIFF_FORMAT)
		ptest->NextSubTest();
		memset(&ti, 0, sizeof(translator_info));
		CPPUNIT_ASSERT(proster->Identify(&file, NULL, &ti, 0, NULL,
			B_TIFF_FORMAT) == B_OK);
		if (bbits)
			CheckBits_Tiff(&ti);
		else
			CheckTiff(&ti, pinfo[i].identifyString);
	}
}

void
TIFFTranslatorTest::IdentifyTest()
{
	// Init
	NextSubTest();
	status_t result = B_ERROR;
	BTranslatorRoster *proster = new BTranslatorRoster();
	CPPUNIT_ASSERT(proster);
	CPPUNIT_ASSERT(proster->AddTranslators(
		"/boot/home/config/add-ons/Translators/TIFFTranslator") == B_OK);
	BFile wronginput("../src/tests/kits/translation/data/images/image.jpg",
		B_READ_ONLY);
	CPPUNIT_ASSERT(wronginput.InitCheck() == B_OK);
		
	// Identify (bad input, output types)
	NextSubTest();
	translator_info ti;
	memset(&ti, 0, sizeof(translator_info));
	result = proster->Identify(&wronginput, NULL, &ti, 0,
		NULL, B_TRANSLATOR_TEXT);
	CPPUNIT_ASSERT(result == B_NO_TRANSLATOR);
	CPPUNIT_ASSERT(ti.type == 0 && ti.translator == 0);
	
	// Identify (wrong type of input data)
	NextSubTest();
	memset(&ti, 0, sizeof(translator_info));
	result = proster->Identify(&wronginput, NULL, &ti);
	CPPUNIT_ASSERT(result == B_NO_TRANSLATOR);
	CPPUNIT_ASSERT(ti.type == 0 && ti.translator == 0);
	
	// Identify (successfully identify the following files)
	const IdentifyInfo aBitsPaths[] = {
		{ "/boot/home/resources/tiff/beer.bits", "" },
		{ "/boot/home/resources/tiff/blocks.bits", "" }
	};
	const IdentifyInfo aTiffPaths[] = {
		{ "/boot/home/resources/tiff/beer_rgb_nocomp.tif",
			"TIFF image (Little, RGB, None)" },
		{ "/boot/home/resources/tiff/beer_rgb_nocomp_big.tif",
			"TIFF image (Big, RGB, None)" },
		{ "/boot/home/resources/tiff/blocks_rgb_nocomp.tif",
			"TIFF image (Little, RGB, None)" },
		{ "/boot/home/resources/tiff/hills_bw_huffman.tif",
			"TIFF image (Little, Mono, Huffman)" },
		{ "/boot/home/resources/tiff/hills_cmyk_nocomp.tif",
			"TIFF image (Little, CMYK, None)" },
		{ "/boot/home/resources/tiff/hills_cmyk_nocomp_big.tif",
			"TIFF image (Big, CMYK, None)" },
		{ "/boot/home/resources/tiff/hills_rgb_nocomp.tif",
			"TIFF image (Little, RGB, None)" },
		{ "/boot/home/resources/tiff/hills_rgb_packbits.tif",
			"TIFF image (Little, RGB, PackBits)" },
		{ "/boot/home/resources/tiff/homes_bw_fax.tif",
			"TIFF image (Little, Mono, Group 3)" },
		{ "/boot/home/resources/tiff/homes_bw_huffman.tif",
			"TIFF image (Little, Mono, Huffman)" },
		{ "/boot/home/resources/tiff/homes_bw_nocomp.tif",
			"TIFF image (Little, Mono, None)" },
		{ "/boot/home/resources/tiff/homes_bw_nocomp_big.tif",
			"TIFF image (Big, Mono, None)" },
		{ "/boot/home/resources/tiff/homes_bw_packbits.tif",
			"TIFF image (Little, Mono, PackBits)" },
		{ "/boot/home/resources/tiff/homes_cmap4_nocomp.tif",
			"TIFF image (Little, Palette, None)" },
		{ "/boot/home/resources/tiff/homes_cmap4_nocomp_big.tif",
			"TIFF image (Big, Palette, None)" },
		{ "/boot/home/resources/tiff/homes_cmap4_packbits.tif",
			"TIFF image (Little, Palette, PackBits)" },
		{ "/boot/home/resources/tiff/homes_gray8_nocomp.tif",
			"TIFF image (Little, Gray, None)" },
		{ "/boot/home/resources/tiff/homes_gray8_nocomp_big.tif",
			"TIFF image (Big, Gray, None)" },
		{ "/boot/home/resources/tiff/homes_gray8_packbits.tif",
			"TIFF image (Little, Gray, PackBits)" },
		{ "/boot/home/resources/tiff/logo_cmap4_nocomp.tif",
			"TIFF image (Little, Palette, None)" },
		{ "/boot/home/resources/tiff/logo_cmap4_nocomp_big.tif",
			"TIFF image (Big, Palette, None)" },
		{ "/boot/home/resources/tiff/logo_cmap4_packbits.tif",
			"TIFF image (Little, Palette, PackBits)" },
		{ "/boot/home/resources/tiff/logo_cmap8_nocomp.tif",
			"TIFF image (Little, Palette, None)" },
		{ "/boot/home/resources/tiff/logo_cmap8_nocomp_big.tif",
			"TIFF image (Big, Palette, None)" },
		{ "/boot/home/resources/tiff/logo_cmap8_packbits.tif",
			"TIFF image (Little, Palette, PackBits)" },
		{ "/boot/home/resources/tiff/logo_cmyk_nocomp.tif",
			"TIFF image (Little, CMYK, None)" },
		{ "/boot/home/resources/tiff/vsmall_cmap4_nocomp.tif",
			"TIFF image (Little, Palette, None)" },
		{ "/boot/home/resources/tiff/vsmall_rgb_nocomp.tif",
			"TIFF image (Little, RGB, None)" },
		{ "/boot/home/resources/tiff/backup_help.tif",
			"TIFF image (Big, Mono, Group 3)" }
	};

	IdentifyTests(this, proster, aTiffPaths,
		sizeof(aTiffPaths) / sizeof(IdentifyInfo), false);
	IdentifyTests(this, proster, aBitsPaths,
		sizeof(aBitsPaths) / sizeof(IdentifyInfo), true);
	
	delete proster;
	proster = NULL;
}

// coveniently group path of tiff image with
// path of bits image that it should translate to
struct TranslatePaths {
	const char *tiffPath;
	const char *bitsPath;
};

void
TranslateTests(TIFFTranslatorTest *ptest, BTranslatorRoster *proster,
	const TranslatePaths *paths, int32 len)
{
	// Perform translations on every file in the array
	for (int32 i = 0; i < len; i++) {
		// Setup input files	
		ptest->NextSubTest();
		BFile tiff_file, bits_file;
		CPPUNIT_ASSERT(tiff_file.SetTo(paths[i].tiffPath, B_READ_ONLY) == B_OK);
		CPPUNIT_ASSERT(bits_file.SetTo(paths[i].bitsPath, B_READ_ONLY) == B_OK);
		printf(" [%s] ", paths[i].tiffPath);
		
		BMallocIO mallio, dmallio;
		
		// Convert to B_TRANSLATOR_ANY_TYPE (should be B_TRANSLATOR_BITMAP)
		ptest->NextSubTest();
		CPPUNIT_ASSERT(mallio.Seek(0, SEEK_SET) == 0);
		CPPUNIT_ASSERT(mallio.SetSize(0) == B_OK);
		CPPUNIT_ASSERT(proster->Translate(&tiff_file, NULL, NULL, &mallio,
			B_TRANSLATOR_ANY_TYPE) == B_OK);
		CPPUNIT_ASSERT(CompareStreams(mallio, bits_file) == true);
		
		// Convert to B_TRANSLATOR_BITMAP
		ptest->NextSubTest();
		CPPUNIT_ASSERT(mallio.Seek(0, SEEK_SET) == 0);
		CPPUNIT_ASSERT(mallio.SetSize(0) == B_OK);
		CPPUNIT_ASSERT(proster->Translate(&tiff_file, NULL, NULL, &mallio,
			B_TRANSLATOR_BITMAP) == B_OK);
		CPPUNIT_ASSERT(CompareStreams(mallio, bits_file) == true);
		
		// Convert bits mallio to B_TRANSLATOR_BITMAP dmallio
		/* Not Supported Yet
		ptest->NextSubTest();
		CPPUNIT_ASSERT(dmallio.Seek(0, SEEK_SET) == 0);
		CPPUNIT_ASSERT(dmallio.SetSize(0) == B_OK);
		CPPUNIT_ASSERT(proster->Translate(&mallio, NULL, NULL, &dmallio,
			B_TRANSLATOR_BITMAP) == B_OK);
		CPPUNIT_ASSERT(CompareStreams(dmallio, bits_file) == true);
		*/
		
		// Convert to B_TIFF_FORMAT
		/* Not Supported Yet
		ptest->NextSubTest();
		CPPUNIT_ASSERT(mallio.Seek(0, SEEK_SET) == 0);
		CPPUNIT_ASSERT(mallio.SetSize(0) == B_OK);
		CPPUNIT_ASSERT(proster->Translate(&tiff_file, NULL, NULL, &mallio,
			B_TIFF_FORMAT) == B_OK);
		CPPUNIT_ASSERT(CompareStreams(mallio, tiff_file) == true);
		*/
		
		// Convert TIFF mallio to B_TRANSLATOR_BITMAP dmallio
		/* Not Ready Yet
		ptest->NextSubTest();
		CPPUNIT_ASSERT(dmallio.Seek(0, SEEK_SET) == 0);
		CPPUNIT_ASSERT(dmallio.SetSize(0) == B_OK);
		CPPUNIT_ASSERT(proster->Translate(&mallio, NULL, NULL, &dmallio,
			B_TRANSLATOR_BITMAP) == B_OK);
		CPPUNIT_ASSERT(CompareStreams(dmallio, bits_file) == true);
		*/
		
		// Convert TIFF mallio to B_TIFF_FORMAT dmallio
		/* Not Ready Yet
		ptest->NextSubTest();
		CPPUNIT_ASSERT(dmallio.Seek(0, SEEK_SET) == 0);
		CPPUNIT_ASSERT(dmallio.SetSize(0) == B_OK);
		CPPUNIT_ASSERT(proster->Translate(&mallio, NULL, NULL, &dmallio,
			B_TIFF_FORMAT) == B_OK);
		CPPUNIT_ASSERT(CompareStreams(dmallio, tiff_file) == true);
		*/
	}
}

void
TIFFTranslatorTest::TranslateTest()
{
	// Init
	NextSubTest();
	status_t result = B_ERROR;
	off_t filesize = -1;
	BTranslatorRoster *proster = new BTranslatorRoster();
	CPPUNIT_ASSERT(proster);
	CPPUNIT_ASSERT(proster->AddTranslators(
		"/boot/home/config/add-ons/Translators/TIFFTranslator") == B_OK);
	BFile wronginput("../src/tests/kits/translation/data/images/image.jpg",
		B_READ_ONLY);
	CPPUNIT_ASSERT(wronginput.InitCheck() == B_OK);
	BFile output("/tmp/tiff_test.out", B_WRITE_ONLY | 
		B_CREATE_FILE | B_ERASE_FILE);
	CPPUNIT_ASSERT(output.InitCheck() == B_OK);
	
	// Translate (bad input, output types)
	NextSubTest();
	result = proster->Translate(&wronginput, NULL, NULL, &output,
		B_TRANSLATOR_TEXT);
	CPPUNIT_ASSERT(result == B_NO_TRANSLATOR);
	CPPUNIT_ASSERT(output.GetSize(&filesize) == B_OK);
	CPPUNIT_ASSERT(filesize == 0);
	
	// Translate (wrong type of input data)
	NextSubTest();
	result = proster->Translate(&wronginput, NULL, NULL, &output,
		B_TIFF_FORMAT);
	CPPUNIT_ASSERT(result == B_NO_TRANSLATOR);
	CPPUNIT_ASSERT(output.GetSize(&filesize) == B_OK);
	CPPUNIT_ASSERT(filesize == 0);
	
	// Translate (wrong type of input, B_TRANSLATOR_ANY_TYPE output)
	NextSubTest();
	result = proster->Translate(&wronginput, NULL, NULL, &output,
		B_TRANSLATOR_ANY_TYPE);
	CPPUNIT_ASSERT(result == B_NO_TRANSLATOR);
	CPPUNIT_ASSERT(output.GetSize(&filesize) == B_OK);
	CPPUNIT_ASSERT(filesize == 0);
	
	// Translate TIFF images to bits
	const TranslatePaths aPaths[] = {
		{ "/boot/home/resources/tiff/beer_rgb_nocomp.tif",
			"/boot/home/resources/tiff/beer.bits" },
		{ "/boot/home/resources/tiff/beer_rgb_nocomp_big.tif",
			"/boot/home/resources/tiff/beer.bits" },
		{ "/boot/home/resources/tiff/blocks_rgb_nocomp.tif",
			"/boot/home/resources/tiff/blocks.bits" },
		{ "/boot/home/resources/tiff/hills_bw_huffman.tif",
			"/boot/home/resources/tiff/hills_bw.bits" },
		{ "/boot/home/resources/tiff/hills_cmyk_nocomp.tif",
			"/boot/home/resources/tiff/hills_cmyk.bits" },
		{ "/boot/home/resources/tiff/hills_cmyk_nocomp_big.tif",
			"/boot/home/resources/tiff/hills_cmyk.bits" },
		{ "/boot/home/resources/tiff/hills_rgb_nocomp.tif",
			"/boot/home/resources/tiff/hills_rgb.bits" },
		{ "/boot/home/resources/tiff/hills_rgb_packbits.tif",
			"/boot/home/resources/tiff/hills_rgb.bits" },
		{ "/boot/home/resources/tiff/homes_bw_fax.tif",
			"/boot/home/resources/tiff/homes_bw.bits" },
		{ "/boot/home/resources/tiff/homes_bw_huffman.tif",
			"/boot/home/resources/tiff/homes_bw.bits" },
		{ "/boot/home/resources/tiff/homes_bw_nocomp.tif",
			"/boot/home/resources/tiff/homes_bw.bits" },
		{ "/boot/home/resources/tiff/homes_bw_nocomp_big.tif",
			"/boot/home/resources/tiff/homes_bw.bits" },
		{ "/boot/home/resources/tiff/homes_bw_packbits.tif",
			"/boot/home/resources/tiff/homes_bw.bits" },
		{ "/boot/home/resources/tiff/homes_cmap4_nocomp.tif",
			"/boot/home/resources/tiff/homes_cmap4.bits" },
		{ "/boot/home/resources/tiff/homes_cmap4_nocomp_big.tif",
			"/boot/home/resources/tiff/homes_cmap4.bits" },
		{ "/boot/home/resources/tiff/homes_cmap4_packbits.tif",
			"/boot/home/resources/tiff/homes_cmap4.bits" },
		{ "/boot/home/resources/tiff/homes_gray8_nocomp.tif",
			"/boot/home/resources/tiff/homes_gray8.bits" },
		{ "/boot/home/resources/tiff/homes_gray8_nocomp_big.tif",
			"/boot/home/resources/tiff/homes_gray8.bits" },
		{ "/boot/home/resources/tiff/homes_gray8_packbits.tif",
			"/boot/home/resources/tiff/homes_gray8.bits" },
		{ "/boot/home/resources/tiff/logo_cmap4_nocomp.tif",
			"/boot/home/resources/tiff/logo_cmap4.bits" },
		{ "/boot/home/resources/tiff/logo_cmap4_nocomp_big.tif",
			"/boot/home/resources/tiff/logo_cmap4.bits" },
		{ "/boot/home/resources/tiff/logo_cmap4_packbits.tif",
			"/boot/home/resources/tiff/logo_cmap4.bits" },
		{ "/boot/home/resources/tiff/logo_cmap8_nocomp.tif",
			"/boot/home/resources/tiff/logo_rgb.bits" },
		{ "/boot/home/resources/tiff/logo_cmap8_nocomp_big.tif",
			"/boot/home/resources/tiff/logo_rgb.bits" },
		{ "/boot/home/resources/tiff/logo_cmap8_packbits.tif",
			"/boot/home/resources/tiff/logo_rgb.bits" },
		{ "/boot/home/resources/tiff/logo_cmyk_nocomp.tif",
			"/boot/home/resources/tiff/logo_cmyk.bits" },
		{ "/boot/home/resources/tiff/vsmall_cmap4_nocomp.tif",
			"/boot/home/resources/tiff/vsmall.bits" },
		{ "/boot/home/resources/tiff/vsmall_rgb_nocomp.tif",
			"/boot/home/resources/tiff/vsmall.bits" },
		{ "/boot/home/resources/tiff/backup_help.tif",
			"/boot/home/resources/tiff/backup_help.bits" }
	};
	
	TranslateTests(this, proster, aPaths,
		sizeof(aPaths) / sizeof(TranslatePaths));
	
	delete proster;
	proster = NULL;
}

#if !TEST_R5

// The input formats that this translator is supposed to support
translation_format gTIFFInputFormats[] = {
	{
		B_TRANSLATOR_BITMAP,
		B_TRANSLATOR_BITMAP,
		0.7f, // quality
		0.6f, // capability
		"image/x-be-bitmap",
		"Be Bitmap Format (TIFFTranslator)"
	},
	{
		B_TIFF_FORMAT,
		B_TRANSLATOR_BITMAP,
		0.7f,
		0.6f,
		"image/tiff",
		"TIFF image"
	}
};

// The output formats that this translator is supposed to support
translation_format gTIFFOutputFormats[] = {
	{
		B_TRANSLATOR_BITMAP,
		B_TRANSLATOR_BITMAP,
		0.7f, // quality
		0.6f, // capability
		"image/x-be-bitmap",
		"Be Bitmap Format (TIFFTranslator)"
	},
	{
		B_TIFF_FORMAT,
		B_TRANSLATOR_BITMAP,
		0.7f,
		0.6f,
		"image/tiff",
		"TIFF image"
	}
};

void
TIFFTranslatorTest::LoadAddOnTest()
{
	TranslatorLoadAddOnTest("/boot/home/config/add-ons/Translators/TIFFTranslator",
		this,
		gTIFFInputFormats, sizeof(gTIFFInputFormats) / sizeof(translation_format),
		gTIFFOutputFormats, sizeof(gTIFFOutputFormats) / sizeof(translation_format),
		B_TRANSLATION_MAKE_VER(1,0,0));
}

#endif // #if !TEST_R5