/*

* 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 "mifconv_mbmgenerator_pbmcomp.h"

#include "mifconv_mbmgenerator_bmtopbm.h"

#include "mifconv_util.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(MifConvSourceFileList aSourcefilenames):

	iSourcefilenames(aSourcefilenames),

	iPbmSources(NULL),

	iNumSources(aSourcefilenames.size())

	{}

BitmapCompiler::~BitmapCompiler()

	{

	iDestFile.close();

	if(iPbmSources)

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

			delete iPbmSources[count];

	delete [] iPbmSources;

	iPbmSources = NULL;

	}

int BitmapCompiler::Compile(const MifConvString& aDestfilename, const MifConvString& aPaletteFilename)

	{    

	int ret = LoadPalette(aPaletteFilename.c_str());

	if (ret)

		return ret;

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

	if(iDestFile.is_open()==0)

		return DestFile;

	if (iNumSources < 1)

		return SourceFile;

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

	ret = LoadSourcefiles();

	if (ret)

		return ret;

	return FileImage();

	};

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()

	{

    //Initialize iPbmSources.

	int ret = AllocatePbmSourcesArray();

	if (ret)

		return ret;

    int count = 0;	

    for( MifConvSourceFileList::iterator i = iSourcefilenames.begin(); i != iSourcefilenames.end(); ++i, ++count )

        {

        bool useAlpha = false;

        int bpp = 2;

        TBitmapColor color = EMonochromeBitmap;

        BitmapLoader bmpload;

        int loc = 0;

        if(i->DepthString().length() > 0)

            {

            if(i->DepthString()[loc]=='c')

                {

                color = EColorBitmap;

                loc++;

                }

            bpp=i->DepthString()[loc]-'0';

            loc++;

            int next=i->DepthString()[loc]-'0';

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

                {

                bpp=bpp*10+next;

                loc++;

                }

            if(i->DepthString()[loc]=='a')

                {

                useAlpha = true;

                loc++;

                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(*i,bpp,color,iPbmSources[count]);

        // If error.

        if (ret)

            return ret;

        }

    return NoError;

	}

int BitmapCompiler::FileImage()

	{    

	int count = 0;

	// Compress

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

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

	return NoError;

	}

int BitmapCompiler::LoadPalette(const char* aPaletteFilename)

	{

	if (!aPaletteFilename)

		{

		iDefaultPalette = 1;

		return NoError;

		}

	iDefaultPalette = 0;

	color256Palette = iPalette;

	color256InversePalette = iInversePalette;

	struct stat fileinfo;

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

		return PaletteFile;

	int filesize = fileinfo.st_size;

	if (filesize == 0)

		return PaletteFile;

	fstream paletteFile(aPaletteFilename, FILE_IN_BINARY_NOCREATE_FLAGS);

	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::CompressBitmap(SEpocBitmapHeader*& aPbm)

	{

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

	char* newbits=new char[originalsize+sizeof(SEpocBitmapHeader)];

	if(!newbits)

		return NoMemory;

	memset(newbits,0xff,sizeof(SEpocBitmapHeader)+originalsize);

	memcpy(newbits,aPbm,sizeof(SEpocBitmapHeader));

	char* newbitsptr=newbits+sizeof(SEpocBitmapHeader);

	char* oldbits=((char*)(aPbm))+sizeof(SEpocBitmapHeader);

	TBitmapfileCompression compression = ENoBitmapCompression;

	int ret = NoCompression;

	if (aPbm->iBitsPerPixel <= 8)

		{

		compression = EByteRLECompression;

		ret = CompressByteData(newbitsptr,oldbits,originalsize);

		}

	else if (aPbm->iBitsPerPixel == 12)

		{

		compression = ETwelveBitRLECompression;

		ret = CompressTwelveBitData(newbitsptr,oldbits,originalsize);

		}

	else if (aPbm->iBitsPerPixel == 16)

		{

		compression = ESixteenBitRLECompression;

		ret = CompressSixteenBitData(newbitsptr,oldbits,originalsize);

		}

	else if (aPbm->iBitsPerPixel == 24)

		{

		compression = ETwentyFourBitRLECompression;

		ret = CompressTwentyFourBitData(newbitsptr,oldbits,originalsize);

		}

	else if (aPbm->iBitsPerPixel == 32 && (aPbm->iColor == EColorBitmap))

		{

		compression = EThirtyTwoUBitRLECompression;

		ret = CompressThirtyTwoUBitData(newbitsptr,oldbits,originalsize);

		}

	else if (aPbm->iBitsPerPixel == 32 && (aPbm->iColor == EColorBitmapAlpha))

		{

		compression = EThirtyTwoABitRLECompression;

		ret = CompressThirtyTwoABitData(newbitsptr,oldbits,originalsize);

		}

	if(ret)

		{

		delete [] newbits;

		if(ret>0)

			return ret;

		return NoError;

		}

	delete aPbm;

	aPbm = (SEpocBitmapHeader*)newbits;

	aPbm->iBitmapSize = newbitsptr-(newbits+sizeof(SEpocBitmapHeader))+sizeof(SEpocBitmapHeader);

	aPbm->iCompression = compression;

	return NoError;

	}

int BitmapCompiler::CompressByteData(char*& aDest,char* aSrce,int aSize)

	{

	const char* destEnd = aDest + aSize;

	char* bytepointer=aSrce;

	char* limitpointer=bytepointer+aSize;

	int margin = (aSize>>6);

	char* limitmargin = limitpointer - ((margin > 4) ? margin : 4);

	char* cutoffpointer=aDest+(aSize>>1)+(aSize>>2);

	int ret=NoError;

	char* oldbytepointer=NULL;

	while(bytepointer<limitmargin)

		{

		char value=*bytepointer;

		if(*(bytepointer+1)==value && *(bytepointer+2)==value)

			{

			oldbytepointer=bytepointer;

			bytepointer+=3;

			while ( ( bytepointer < limitmargin ) && ( *bytepointer == value ) )

				bytepointer++;

			ret = WriteCompressedByteValues(aDest,value,bytepointer-oldbytepointer, destEnd);

			if(ret) return ret;

			}

		else

			{

			oldbytepointer=bytepointer;

			while((bytepointer<limitmargin) && (*(bytepointer+1)!=value || *(bytepointer+2)!=value))

				{

				bytepointer++;

				value=*bytepointer;

				}

			ret = WriteCompressedByteData(aDest,oldbytepointer,bytepointer-oldbytepointer, destEnd);

			if(ret) return ret;

			}

		if(aDest>cutoffpointer)

			return NoCompression;

		}

	int remaining = limitpointer-bytepointer;

	if(remaining > 0)

		{

		if (aDest + remaining > cutoffpointer)

			return NoCompression;

		ret=WriteCompressedByteData(aDest,bytepointer,remaining, destEnd);

		if(ret) return ret;

		}

	return NoError;

	}

int BitmapCompiler::WriteCompressedByteData(char*& aDest,char* aData,int aLength, const char* aDestEnd)

	{

	if(!CanCopy8bppData(aDest, aDestEnd, aLength))

		return NoCompression;

	while(aLength>128)

		{

		*aDest++=-128;

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

			*aDest++=*aData++;

		aLength-=128;

		}

	if(aLength>128 || aLength<1) return CompressionError;

	*aDest++=char(-aLength);

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

		*aDest++=*aData++;

	return NoError;

	}

int BitmapCompiler::WriteCompressedByteValues(char*& aDest,char aValue,int aLength, const char* aDestEnd)

	{

	if (aLength < 1)

		return CompressionError;

	if(!CanWrite8bppValue(aDest, aDestEnd, aLength))

		return NoCompression;

	while (aLength > 128)

		{

		*aDest++ = 127;

		*aDest++ = aValue;

		aLength -= 128;

		}

	*aDest++ = char(aLength-1);

	*aDest++ = aValue;

	return NoError;

	}

int BitmapCompiler::CompressTwelveBitData(char*& aDest,char* aSrce,int aSizeInBytes)

	{

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

	unsigned short* srceLimitPtr = srcePtr + (aSizeInBytes / 2);

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

	while (srcePtr < srceLimitPtr)

		{

		unsigned short* runStartPtr = srcePtr;

		unsigned short value = *srcePtr;

		do

			srcePtr++;

		while (srcePtr < srceLimitPtr && *srcePtr == value);

		WriteCompressedTwelveBitData(dest,*runStartPtr,srcePtr - runStartPtr); // Ignore error return as 12bpp compression never fails

		}

	aDest = (char*)dest;

	return NoError;

	}

int BitmapCompiler::WriteCompressedTwelveBitData(unsigned short*& aDest,unsigned short aData,int aLength)

	{

	// The run length l is stored as l-1 in the top 4 bits of each 16-bit word (between 1 and 16)

	aData &= 0x0fff;

	unsigned short maxLengthData = (unsigned short)(aData | 0xf000);

	while(aLength>16)

		{

		*aDest++ = maxLengthData;

		aLength -= 16;

		}