/*

* 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 <assert.h>

#include "BMCONV.H"

EpocLoader::EpocLoader():

	iPbmBits(NULL)

	{}

EpocLoader::~EpocLoader()

	{

	delete iPbmBits;

	}

int EpocLoader::EpocBitmapCount(char* aFileName, int& aCount, int& isRomFormat)

    {

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

	fstream file(aFileName, ios::in | ios::binary);

#else //!__MSVCDOTNET__

	fstream file(aFileName, ios::in | ios::binary | ios::nocreate);

#endif //__MSVCDOTNET__

	if (file.is_open()==0)

		return Files;

	long int wordbuffer;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	if (wordbuffer==KMultiBitmapRomImageUid)

		{

		// ROM format file - next 4 bytes are the number of bitmaps

		isRomFormat=1;

		}

	else

		{

		if (wordbuffer!=KWriteOnceFileStoreUid)

			return SourceFile;

		isRomFormat=0;

		file.read((char *)&wordbuffer,4);

		if (file.gcount()!=4 || wordbuffer!=KMultiBitmapFileImageUid)

			return SourceFile;

		file.read((char *)&wordbuffer,4);

		if (file.gcount()!=4 || wordbuffer!=0)

			return SourceFile;

		file.read((char *)&wordbuffer,4);

		if (file.gcount()!=4 || wordbuffer!=KMultiBitmapFileImageChecksum)

			return SourceFile;

		file.read((char *)&wordbuffer,4);

		if (file.gcount()!=4)

			return SourceFile;

		file.seekg(wordbuffer,ios::beg);

		}

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	aCount=wordbuffer;

	return NoError;

    }

int EpocLoader::LoadEpocBitmap(char* aFileName,int aIndex)

	{

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

	fstream file(aFileName, ios::in | ios::binary);

#else //!__MSVCDOTNET__

	fstream file(aFileName, ios::in | ios::binary | ios::nocreate);

#endif //__MSVCDOTNET__

	if (file.is_open()==0)

		return Files;

	long int wordbuffer;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	file.close();

	if (wordbuffer==KMultiBitmapRomImageUid)

		return LoadRom(aFileName,aIndex);

	return LoadFile(aFileName,aIndex);

	}

int EpocLoader::LoadFile(char* aFileName,int aIndex)

	{

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

	fstream file(aFileName, ios::in | ios::binary);

#else //!__MSVCDOTNET__

	fstream file(aFileName, ios::in | ios::binary | ios::nocreate);

#endif //__MSVCDOTNET__

	if (file.is_open()==0)

		return Files;

	long int wordbuffer;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4 || wordbuffer!=KWriteOnceFileStoreUid)

		return SourceFile;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4 || wordbuffer!=KMultiBitmapFileImageUid)

		return SourceFile;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4 || wordbuffer!=0)

		return SourceFile;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4 || wordbuffer!=KMultiBitmapFileImageChecksum)

		return SourceFile;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	file.seekg(wordbuffer,ios::beg);

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	int numsources=wordbuffer;

	if (aIndex >= numsources)

		return OutOfRange;

	file.seekg(aIndex*4,ios::cur);

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	file.seekg(wordbuffer,ios::beg);

	int ret = DoLoadFile(file);

	file.close();

	return ret;

	}

int EpocLoader::LoadRom(char* aFileName,int aIndex)

	{

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

	fstream file(aFileName, ios::in | ios::binary);

#else //!__MSVCDOTNET__

	fstream file(aFileName, ios::in | ios::binary | ios::nocreate);

#endif //__MSVCDOTNET__

	if (file.is_open()==0)

		return Files;

	long int wordbuffer;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	if (wordbuffer!=KMultiBitmapRomImageUid)

		return SourceFile;

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	if (aIndex>=wordbuffer)

		return OutOfRange;

	file.seekg(aIndex*4,ios::cur);

	file.read((char *)&wordbuffer,4);

	if (file.gcount()!=4)

		return SourceFile;

	file.seekg(wordbuffer,ios::beg);

	int ret=DoLoadRom(file);

	file.close();

	return ret;

	}

int EpocLoader::DoLoadFile(fstream& aFile)

	{

	long size = sizeof(SEpocBitmapHeader);

	aFile.read((char *)&iPbmHeader,size);

	if (aFile.gcount()!=size)

		return SourceFile;

	iOriginalPbmHeader = iPbmHeader;

	size=iPbmHeader.iBitmapSize-iPbmHeader.iStructSize;

	iPbmBits=new char[size];

	if (iPbmBits==NULL)

		return NoMemory;

	memset(iPbmBits,0xff,size);

	aFile.read(iPbmBits,size);

	aFile.close();

	if (aFile.gcount() != size)

		return SourceFile;

	if (iPbmHeader.iCompression != ENoBitmapCompression)

		{

		return Decompress(size);

		}

	return NoError;

	}

int EpocLoader::DoLoadRom(fstream& aFile)

	{

	Bitmap bmp;

	long size=sizeof(Bitmap);

	aFile.read((char*)&bmp,size);

	if (aFile.gcount() != size)

		return SourceFile;

	size = bmp.iHeader.iBitmapSize - sizeof(SEpocBitmapHeader);

	iPbmBits = new char[size];

	if (iPbmBits == NULL)

		return NoMemory;

	memset(iPbmBits,0xff,size);

	aFile.read(iPbmBits,size);

	if (aFile.gcount() != size)

		return SourceFile;

	iPbmHeader = bmp.iHeader;

	iOriginalPbmHeader = iPbmHeader;

	if (iPbmHeader.iCompression != ENoBitmapCompression)

		{

		return Decompress(size);

		}

	return NoError;

	}

int EpocLoader::Decompress(int aSize)

	{

	int byteWidth = BitmapUtils::ByteWidth(iPbmHeader.iWidthInPixels,iPbmHeader.iBitsPerPixel);

	int expandedsize = byteWidth * iPbmHeader.iHeightInPixels;

	char* newbits = new char[expandedsize];

	if (newbits == NULL)

		return NoMemory;

	memset(newbits,0xff,expandedsize);

	int ret = NoError;

	switch (iPbmHeader.iCompression)

		{

	case EByteRLECompression:

		ret = ExpandByteRLEData(newbits,expandedsize,iPbmBits,aSize);

		break;

	case ETwelveBitRLECompression:

		ret = ExpandTwelveBitRLEData(newbits,expandedsize,iPbmBits,aSize);

		break;

	case ESixteenBitRLECompression:

		ret = ExpandSixteenBitRLEData(newbits,expandedsize,iPbmBits,aSize);

		break;

	case ETwentyFourBitRLECompression:

		ret = ExpandTwentyFourBitRLEData(newbits,expandedsize,iPbmBits,aSize);

		break;

	case EThirtyTwoUBitRLECompression:

		ret = ExpandThirtyTwoUBitRLEData(newbits,expandedsize,iPbmBits,aSize);

		break;

	case EThirtyTwoABitRLECompression:

		ret = ExpandThirtyTwoABitRLEData(newbits,expandedsize,iPbmBits,aSize);

		break;

	default:

		ret = UnknownCompression;

		break;

		}

	delete iPbmBits;

	iPbmBits = newbits;

	iPbmHeader.iCompression = ENoBitmapCompression;

	iPbmHeader.iBitmapSize += expandedsize-aSize;

	return ret;

	}

int EpocLoader::ExpandByteRLEData(char* aDest,int aDestSize,char* aSrce,int aSrceSize)

	{

	char* srcelimit=aSrce+aSrceSize;

	char* destlimit=aDest+aDestSize;

	while(aSrce<srcelimit && aDest<destlimit)

		{

		char count=*aSrce++;

		if (count<0)

			{

			int runLength=-count;

			memcpy(aDest,aSrce,runLength);

			aSrce+=runLength;

			aDest+=runLength;

			}

		else

			{

			char value=*aSrce++;

			while(count>=0)

				{

				*aDest++=value;

				count--;

				}

			}

		}

	if (aSrce!=srcelimit || aDest!=destlimit)

		return DecompressionError;

	return NoError;

	}

int EpocLoader::ExpandTwelveBitRLEData(char* aDest,int aDestSizeInBytes,char* aSrce,int aSrceSizeInBytes)

	{

	unsigned short* srcePtr = (unsigned short*)aSrce;

	unsigned short* destPtr = (unsigned short*)aDest;

	unsigned short* srcePtrLimit = srcePtr + (aSrceSizeInBytes / 2);

	unsigned short* destPtrLimit = destPtr + (aDestSizeInBytes / 2);

	while (srcePtr < srcePtrLimit && destPtr < destPtrLimit)

		{

		unsigned short value = *srcePtr++;

		int runLength = (value >> 12) + 1;

		value &= 0x0fff;

		for (;runLength > 0;runLength--)

			*destPtr++ = value;

		}

	if (srcePtr != srcePtrLimit || destPtr != destPtrLimit)

		return DecompressionError;

	return NoError;

	}

int EpocLoader::ExpandSixteenBitRLEData(char* aDest,int aDestSizeInBytes,char* aSrce,int aSrceSizeInBytes)

	{

	char* srcePtrLimit = aSrce + aSrceSizeInBytes;

	unsigned short* destPtr = (unsigned short*)aDest;

	unsigned short* destPtrLimit = destPtr + (aDestSizeInBytes / 2);

	while (aSrce < srcePtrLimit && destPtr < destPtrLimit)

		{

		int runLength = *aSrce++;

		if (runLength >= 0)

			{

			unsigned short value = *((unsigned short*)(aSrce));

			aSrce += 2;

			for (runLength++; runLength > 0; runLength--)

				*destPtr++ = value;

			}

		else

			{

			runLength = -runLength;

			int byteLength = runLength * 2;

			memcpy(destPtr,aSrce,byteLength);

			aSrce += byteLength;

			destPtr += runLength;

			}

		}

	if (aSrce != srcePtrLimit || destPtr != destPtrLimit)

		return DecompressionError;

	return NoError;

	}

int EpocLoader::ExpandTwentyFourBitRLEData(char* aDest,int aDestSizeInBytes,char* aSrce,int aSrceSizeInBytes)

	{

	char* srcePtrLimit = aSrce + aSrceSizeInBytes;

	char* destPtrLimit = aDest + aDestSizeInBytes;

	while (aSrce < srcePtrLimit && aDest < destPtrLimit)

		{

		int runLength = *aSrce++;

		if (runLength >= 0)

			{

			char component1 = *aSrce++;

			char component2 = *aSrce++;

			char component3 = *aSrce++;

			for (runLength++; runLength > 0; runLength--)

				{

				*aDest++ = component1;

				*aDest++ = component2;

				*aDest++ = component3;

				}

			}

		else

			{

			runLength = -runLength;

			int byteLength = runLength * 3;

			memcpy(aDest,aSrce,byteLength);

			aSrce += byteLength;

			aDest += byteLength;

			}

		}

	if (aSrce != srcePtrLimit || aDest != destPtrLimit)

		return DecompressionError;

	return NoError;

	}

/** The function decodes 24-bit compressed pixel buffer into the 32-bit buffer, where top bytes are unused*/

int EpocLoader::ExpandThirtyTwoUBitRLEData(char* aDest,int aDestSizeInBytes,char* aSrce,int aSrceSizeInBytes)

	{

	char* srcePtrLimit = aSrce + aSrceSizeInBytes;

	char* destPtrLimit = aDest + aDestSizeInBytes;

	while (aSrce < srcePtrLimit && aDest < destPtrLimit)

		{

		int runLength = *aSrce++;

		if (runLength >= 0)

			{

			char component1 = *aSrce++;

			char component2 = *aSrce++;

			char component3 = *aSrce++;

			for (runLength++; runLength > 0; runLength--)

				{

				*aDest++ = component1;

				*aDest++ = component2;

				*aDest++ = component3;

				aDest++;

				}

			}

		else 

			{

			runLength = -runLength;

			for(int ii = 0; ii < runLength; ii++)

				{

				memcpy(aDest,aSrce,3);

				aSrce += 3;

				aDest += 4;

				}

			}

		}

	if (aSrce != srcePtrLimit || aDest != destPtrLimit)

		return DecompressionError;

	return NoError;

	}

/** The function decodes 32-bit compressed pixel buffer into the 32-bit buffer, where top bytes are alpha channel*/

int EpocLoader::ExpandThirtyTwoABitRLEData(char* aDest,int aDestSizeInBytes,char* aSrce,int aSrceSizeInBytes)

	{

	char* srcePtrLimit = aSrce + aSrceSizeInBytes;

	char* destPtrLimit = aDest + aDestSizeInBytes;

	while (aSrce < srcePtrLimit && aDest < destPtrLimit)

		{

		int runLength = *aSrce++;

		if (runLength >= 0)

			{

			char component1 = *aSrce++;

			char component2 = *aSrce++;

			char component3 = *aSrce++;

			char component4 = *aSrce++;

			for (runLength++; runLength > 0; runLength--)

				{

				*aDest++ = component1;

				*aDest++ = component2;

				*aDest++ = component3;

				*aDest++ = component4;

				}

			}

		else 

			{

			runLength = -runLength;

			memcpy(aDest,aSrce,4*runLength);

			aSrce += 4*runLength;

			aDest += 4*runLength;

			}

		}

	if (aSrce != srcePtrLimit || aDest != destPtrLimit)

		return DecompressionError;

	return NoError;

	}

TRgb EpocLoader::GetPixel(int aXCoord,int aYCoord)

	{

	unsigned char col;

	aXCoord%=iPbmHeader.iWidthInPixels;

	aYCoord%=iPbmHeader.iHeightInPixels;

	int byteWidth = BitmapUtils::ByteWidth(iPbmHeader.iWidthInPixels,iPbmHeader.iBitsPerPixel);

	int yOffset = aYCoord * byteWidth;

	switch(iPbmHeader.iBitsPerPixel)

		{

		case 1:

			col = iPbmBits[yOffset + (aXCoord / 8)];

			col >>= (aXCoord&7);

			return TRgb::Gray2(col & 1);

		case 2:

			col = iPbmBits[yOffset + (aXCoord / 4)];

			col = (unsigned char)(col>>(2*(aXCoord%4)));

			return TRgb::Gray4(col & 3);

		case 4:

			col = iPbmBits[yOffset + (aXCoord / 2)];

			if (aXCoord & 1)

				col >>= 4;

			col &= 0xf;

			if (iPbmHeader.iColor==EColorBitmap)

				return TRgb::Color16(col);

			else

				return TRgb::Gray16(col);

		case 8:

			col=iPbmBits[yOffset + aXCoord];

			if (iPbmHeader.iColor==EColorBitmap)

				return TRgb::Color256(col);

			else

				return TRgb::Gray256(col);

		case 12:

		case 16:

			{

			unsigned short* shortPtr = (unsigned short*)&iPbmBits[yOffset + (aXCoord * 2)];

			if (iPbmHeader.iBitsPerPixel == 12)

				return TRgb::Color4K(*shortPtr);

			else

				return TRgb::Color64K(*shortPtr);

			}

		case 24:

			{

			char* pixelPtr = iPbmBits + yOffset + (aXCoord * 3);

			TRgb pixelColor;

			pixelColor.iBlue = *pixelPtr++;

			pixelColor.iGreen = *pixelPtr++;

			pixelColor.iRed = *pixelPtr;

			return pixelColor;

			}

		case 32:

			{

			char* pixelPtr = iPbmBits + yOffset + (aXCoord * 4);

			TRgb pixelColor;

			pixelColor.iBlue = *pixelPtr++;

			pixelColor.iGreen = *pixelPtr++;