/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies). 

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:

*

*/

#include "BMCONV.H"

#include <stdlib.h>

#include <ctype.h>

#include <sys/stat.h>

extern TRgb* color256Palette;

extern char* color256InversePalette;

static inline bool CanCopy16bppData(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (256 + 1) + //(2 * 128) bytes data + 1 leading byte

		                            (aDataSize % 128 ? (aDataSize % 128) * 2 + 1 : 0);

	return aDestNew <= aDestEnd;

	}

static inline bool CanWrite16bppValue(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (2 + 1) + //2 bytes data + 1 leading byte

		                            (aDataSize % 128 ? 2 + 1 : 0);

	return aDestNew <= aDestEnd;

	}

static inline bool CanCopy8bppData(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (128 + 1) + //128 bytes data + 1 leading byte

		                            (aDataSize % 128 ? (aDataSize % 128) + 1 : 0);

	return aDestNew <= aDestEnd;

	}

static inline bool CanWrite8bppValue(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (1 + 1) + //1 byte data + 1 leading byte

		                            (aDataSize % 128 ? 1 + 1 : 0);

	return aDestNew <= aDestEnd;

	}

static inline bool CanCopy24bppData(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (384 + 1) + //(128 * 3) bytes data + 1 leading byte

		                            (aDataSize % 128 ? (aDataSize % 128) * 3 + 1 : 0);

	return aDestNew <= aDestEnd;

	}

static inline bool CanWrite24bppValue(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (3 + 1) + //3 bytes data + 1 leading byte

		                            (aDataSize % 128 ? 3 + 1 : 0);

	return aDestNew <= aDestEnd;

	}

static inline bool CanCopy32bppData(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (512 + 1) + //(128 * 4) bytes data + 1 leading byte

		                            (aDataSize % 128 ? (aDataSize % 128) * 4 + 1 : 0);

	return aDestNew <= aDestEnd;

	}

static inline bool CanWrite32bppValue(const char* aDest, const char* aDestEnd, int aDataSize)

	{

	const char* aDestNew = 	aDest + (aDataSize / 128) * (4 + 1) + //4 bytes data + 1 leading byte

		                            (aDataSize % 128 ? 4 + 1 : 0);

	return aDestNew <= aDestEnd;

	}

BitmapCompiler::BitmapCompiler(TSourceFile* aSourcefilenames,int aNumSources):

	iSourcefilenames(aSourcefilenames),

	iPbmSources(NULL),

	iNumSources(aNumSources)

	{}

BitmapCompiler::~BitmapCompiler()

	{

	iDestFile.close();

	if(iPbmSources)

		for(int count=0;count<iNumSources;count++)

			delete iPbmSources[count];

	delete [] iPbmSources;

	iPbmSources = NULL;

	}

int BitmapCompiler::Compile(TStoreType aSt,int aCompress,char* aDestfilename,char* aHeaderFilename,char* aPaletteFilename)

	{

	int ret = LoadPalette(aPaletteFilename);

	if (ret)

		return ret;

	iDestFile.open(aDestfilename, ios::out | ios::binary);

	if(iDestFile.is_open()==0)

		return DestFile;

	if (iNumSources < 1)

		return SourceFile;

	if (iNumSources==1 && (iSourcefilenames[0].FileName()[0]==OPTCHAR || iSourcefilenames[0].FileName()[0]==ALTERNATE_OPTCHAR) && iSourcefilenames[0].FileName()[1]=='m')

		{

		// Source is an existing MBM file, presumably because we want to convert

		// a filestore into a ROM image.

		if (aHeaderFilename)

			{

			cout << "Header file generation is not permitted with /m or -m\n";

			aHeaderFilename = NULL;

			}

		iSourcefilenames[0].FileName()+=2;

		EpocLoader pl;

		int romFormat;

		ret = pl.EpocBitmapCount(iSourcefilenames[0].FileName(), iNumSources, romFormat);

		if (!ret)

			ret = LoadPbmFile(iSourcefilenames[0].FileName());

		}

	else

		{

		// Read the usual list of [OPT]bmp_n source bitmaps.

		ret = LoadSourcefiles();

		}

	if (ret)

		return ret;

	ret = CreateHeader(aHeaderFilename);

	if (ret)

		return ret;

	if (aSt == ERomStore || aSt == ECompressedRomStore)

		ret = RomImage(aSt == ECompressedRomStore);

	else

		ret = FileImage(aCompress);

	return ret;

	};

int BitmapCompiler::AllocatePbmSourcesArray()

	{

	iPbmSources = new SEpocBitmapHeader*[iNumSources];

	if(iPbmSources == NULL)

		return NoMemory;

	memset(iPbmSources,0xff,sizeof(SEpocBitmapHeader*)*iNumSources);

	int count;

	for(count=0;count<iNumSources;count++)

		iPbmSources[count]=NULL;

	return NoError;

	}

int BitmapCompiler::LoadSourcefiles()

	{

	int ret = AllocatePbmSourcesArray();

	if (ret)

		return ret;

	for(int count=0;count<iNumSources;count++)

		{

		bool useAlpha = false;

		int bpp = 2;

		TBitmapColor color = EMonochromeBitmap;

		BitmapLoader bmpload;

		if(iSourcefilenames[count].FileName()[0]==OPTCHAR || iSourcefilenames[count].FileName()[0]==ALTERNATE_OPTCHAR)

			{

			iSourcefilenames[count].FileName()++;

			if(iSourcefilenames[count].FileName()[0]=='c')

				{

				color = EColorBitmap;

				iSourcefilenames[count].FileName()++;

				}

			bpp=iSourcefilenames[count].FileName()[0]-'0';

			iSourcefilenames[count].FileName()++;

			int next=iSourcefilenames[count].FileName()[0]-'0';

			if(next==2 || next==6 || next==4)

				{

				bpp=bpp*10+next;

				iSourcefilenames[count].FileName()++;

				}

			if(iSourcefilenames[count].FileName()[0]=='a')

				{

				useAlpha = true;

				iSourcefilenames[count].FileName()++;

				if (color!=EColorBitmap || bpp!=32)

					return Bpp;

				}

			if (color == EColorBitmap && bpp!=4 && bpp!=8 && bpp!=12 && bpp!=16 && bpp!=24 && bpp!=32)

				return Bpp;

			if (color == EMonochromeBitmap && bpp!=1 && bpp!=2 && bpp!=4 && bpp!=8)

				return Bpp;

			}

		if (color == EColorBitmap && useAlpha)

			color = EColorBitmapAlpha;

		ret = bmpload.LoadBitmap(iSourcefilenames[count],bpp,color,iPbmSources[count]);

		if (ret)

			return ret;

		}

	return NoError;

	}

int BitmapCompiler::LoadPbmFile(char* aPbmFilename)

	{

	int ret = AllocatePbmSourcesArray();

	if (ret)

		return ret;

	for(int count=0;count<iNumSources;count++)

		{

		EpocLoader pl;

		ret = pl.LoadEpocBitmap(aPbmFilename,count);

		if (!ret)

			ret = pl.DupBitmap(iPbmSources[count]);

		if (ret)

			return ret;

		}

	return NoError;

	}

int BitmapCompiler::RomImage(bool aCompress)

	{

	int count=0;

	if (aCompress)

		{

		for(; count<iNumSources; count++)

			{

			int ret = CompressBitmap(iPbmSources[count]);

			if (ret)

				return ret;

			}

		}

	int ret = WriteRomheader();

	if (ret)

		return ret;

	for(count=0;count<iNumSources;count++)

		{

		ret = WriteRombitmap(iPbmSources[count]);

		if (ret)

			return ret;

		}

	return NoError;

	}

int BitmapCompiler::FileImage(int aCompress)

	{

	int count = 0;

	if(aCompress)

		{

		for(;count<iNumSources;count++)

			{

			int ret = CompressBitmap(iPbmSources[count]);

			if(ret)

				return ret;

			}

		}

	int ret = WriteFileheader();

	if (ret)

		return ret;

	for (count=0;count<iNumSources;count++)

		{

		ret = WriteFilebitmap(iPbmSources[count]);

		if (ret)

			return ret;

		}

	ret = WriteHeadStream();

	return ret;

	}

int BitmapCompiler::WriteRomheader()

	{

	iDestFile.write((char*)&KMultiBitmapRomImageUid,4);

	iDestFile.write((char*)&iNumSources,4);

	int byteswritten=8+(iNumSources*4);

	for (int count=0;count<iNumSources;count++)

		{

		iDestFile.write((char*)&byteswritten,4);

		byteswritten+=(((iPbmSources[count]->iBitmapSize)+3)/4*4)-sizeof(SEpocBitmapHeader)+sizeof(Bitmap);

		}

	return NoError;

	}

int BitmapCompiler::WriteRombitmap(SEpocBitmapHeader* aPbm)

	{

	if (aPbm->iPaletteEntries != 0)

	    return PaletteSupportNotImplemented;

	int bitmapsize = aPbm->iBitmapSize-sizeof(SEpocBitmapHeader)+sizeof(Bitmap);

	bitmapsize = ((bitmapsize+3)/4)*4;

	char* buffer = new char[bitmapsize];

	if (buffer == NULL)

		return NoMemory;

	memset(buffer,0xff,bitmapsize);

	Bitmap* bmp = (Bitmap*)buffer;

	bmp->iUid=KCBitwiseBitmapUid;

	TBitmapColor color = aPbm->iColor;

	switch(aPbm->iBitsPerPixel)

		{ // for corresponding enums, see TDisplayMode in <gdi.h>

	case 1:

		bmp->iDispMode=1;

		break;

	case 2:

		bmp->iDispMode=2;

		break;

	case 4:

		if (color == EMonochromeBitmap)

			bmp->iDispMode=3;

		else

			bmp->iDispMode=5;

		break;

	case 8:

		if (color == EMonochromeBitmap)

			bmp->iDispMode=4;

		else

			bmp->iDispMode=6;

		break;

	case 12:

		bmp->iDispMode=10;

		break;

	case 16:

		bmp->iDispMode=7;

		break;

	case 24:

		bmp->iDispMode=8;

		break;

	case 32:

		if (color == EColorBitmapAlpha)

			bmp->iDispMode=12; // Color16MA

		else

			bmp->iDispMode=11; // Color16MU

		break;

	default:

		delete [] buffer;

		return SourceFile;

		}

	bmp->iHeap = NULL;

	bmp->iPile = NULL;

	bmp->iHeader = *aPbm;

	bmp->iByteWidth = BitmapUtils::ByteWidth(bmp->iHeader.iWidthInPixels,bmp->iHeader.iBitsPerPixel);

	bmp->iDataOffset = sizeof(Bitmap);

	CopyTail(buffer + sizeof(Bitmap), aPbm, aPbm->iBitmapSize, sizeof(SEpocBitmapHeader));

	iDestFile.write(buffer,bitmapsize);

	delete [] buffer;

	return NoError;

	}

int BitmapCompiler::WriteFileheader()

	{

	int zero=0;

	iDestFile.write((char*)&KWriteOnceFileStoreUid,4);

	iDestFile.write((char*)&KMultiBitmapFileImageUid,4);

	iDestFile.write((char*)&zero,4);

	iDestFile.write((char*)&KMultiBitmapFileImageChecksum,4);

	int byteswritten=16;

	for(int count=0;count<iNumSources;count++)

		{

		byteswritten+=iPbmSources[count]->iBitmapSize;

		}

	byteswritten+=4;

	iDestFile.write((char*)&byteswritten,4);

	return NoError;

	}

int BitmapCompiler::WriteHeadStream()

	{

	iDestFile.write((char*)&iNumSources,4);

	int byteswritten=20;

	for(int count=0;count<iNumSources;count++)

		{

		iDestFile.write((char*)&byteswritten,4);

		byteswritten+=iPbmSources[count]->iBitmapSize;

		}

	return NoError;

	}

int BitmapCompiler::WriteFilebitmap(SEpocBitmapHeader* aPbm)

	{

	if (aPbm->iPaletteEntries != 0)

		return PaletteSupportNotImplemented;

	int dataSize = aPbm->iBitmapSize - sizeof(SEpocBitmapHeader);

	iDestFile.write((char*)(aPbm),sizeof(SEpocBitmapHeader));

	// WritePalette()

	iDestFile.write(((char*)(aPbm)) + sizeof(SEpocBitmapHeader),dataSize);

	return NoError;

	}

void BitmapCompiler::WritePalette()

	{

	for (int index = 0; index < 256; index++)

		{

		iDestFile.write((char*)(&iPalette[index]),3);

		}

	}

int BitmapCompiler::LoadPalette(char* aPaletteFilename)

	{

	if (!aPaletteFilename)

		{

		iDefaultPalette = 1;

		return NoError;

		}

	iDefaultPalette = 0;

	color256Palette = iPalette;

	color256InversePalette = iInversePalette;

	struct stat fileinfo;

	if (stat(aPaletteFilename,&fileinfo)==-1)

		return CommandFile;

	int filesize = fileinfo.st_size;

	if (filesize == 0)

		return PaletteFile;

#if defined(__MSVCDOTNET__) || defined(__TOOLS2__) || defined(__CW32__)

	fstream paletteFile(aPaletteFilename, ios::in | ios::binary);

#else

	fstream paletteFile(aPaletteFilename, ios::in | ios::binary | ios::nocreate);

#endif

	if(!paletteFile.is_open())

		return PaletteFile;

	char* paletteData = new char[filesize];

	if (!paletteData)

		return NoMemory;

	memset(paletteData,0,filesize);

	paletteFile.read(paletteData,filesize);

	paletteFile.close();

	char* dataPtr = (char*)paletteData;

	char* dataPtrLimit = dataPtr + filesize;

	for (int index = 0; index < 256; index++)

		{

		char hexBuf[10];

		int ret = ReadHexString(hexBuf,dataPtr,dataPtrLimit);

		if (ret)

			{

			delete paletteData;

			return ret;

			}

		int red = HexToInt(hexBuf[8],hexBuf[9]);

		int green = HexToInt(hexBuf[6],hexBuf[7]);

		int blue = HexToInt(hexBuf[4],hexBuf[5]);

		iPalette[index] = TRgb(red,green,blue);

		}

	delete paletteData;

	CalculateInversePalette();

	return NoError;

	}

void BitmapCompiler::CalculateInversePalette()

	{

	for (int index = 0; index < 4096; index++)

		{

		TRgb color = TRgb((index & 0x00f) * 17,((index & 0x0f0) >> 4) * 17,((index & 0xf00) >> 8) * 17);

		int nearest = 0;

		int distance = iPalette[0].Difference(color);

		for (int paletteIndex = 0; paletteIndex < 256; paletteIndex++)

			{

			TRgb paletteColor = iPalette[paletteIndex];

			if (paletteColor == color)

				{

				nearest = paletteIndex;

				break;

				}

			int paletteDistance = paletteColor.Difference(color);

			if (paletteDistance < distance)

				{

				nearest = paletteIndex;

				distance = paletteDistance;

				}

			}

		iInversePalette[index] = (char)nearest;

		TRgb palColor = iPalette[nearest];

		color = palColor;

		}

	}

int BitmapCompiler::CreateHeader(char* aHeaderFilename)

	{

	if (!aHeaderFilename)

		return NoError;

	fstream hf(aHeaderFilename,ios::out);

	if(!hf.is_open())

		return DestFile;

	char* basename = strrchr(aHeaderFilename,'\\');

	if (basename==0)

		basename = aHeaderFilename;

	else

		basename++;	// skip directory separator

	char unadornedFile[256];

	strcpy(unadornedFile, UnadornedName(aHeaderFilename));

	hf << "// " << basename << "\n";

	hf << "// Generated by BitmapCompiler\n";

	hf << "// Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies). All rights reserved.\n";

	hf << "//\n\n";

	hf << "enum TMbm" << unadornedFile << "\n";

	hf << "\t{\n";

	for (int count=0;count<iNumSources;count++)