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

// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the License "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:

// e32test/multimedia/t_camera_bitmap.cpp

// This is a basic Windows bitmap file writer, that can be used for converting YUV422 data into

// RGB format and dumping it to a Windows .bmp file, for manual examination.

// 

//

#include <e32test.h>

#include <f32file.h>

#include "t_camera_bitmap.h"

#define CLIP(a) if (a < 0) a = 0; else if (a > 255) a = 255;

/**

Converts a RGB565 buffer into 24 bit RGB format in a second buffer.

@param	aDest		Pointer to the buffer into which to place the 24 bit RGB data

@param	aSource		Pointer to the buffer containing the RGB565 data to be converted

@param	aWidth		The width of the data in the buffer in pixels

@param	aHeight		The height of the data in the buffer in pixels

*/

void RBitmap::RGBToRGB(TUint8* aDest, TUint8* aSource, TInt aWidth, TInt aHeight)

	{

	TUint16* source = (TUint16*) aSource;

	TUint16 pixel;

	for (TInt y = 0; y < aHeight; ++y)

		{

		for (TInt x = 0; x < aWidth; ++x)

			{

			pixel = *source++;

			*aDest++ = (TUint8) ((pixel & 0xf800) >> 8);

			*aDest++ = (TUint8) ((pixel & 0x07e0) >> 3);

			*aDest++ = (TUint8) ((pixel & 0x001f) << 3);

			}

		}

	}

/**

Converts a YUV422 buffer into 24 bit RGB format in a second buffer.

@param	aDest		Pointer to the buffer into which to place the 24 bit RGB data

@param	aSource		Pointer to the buffer containing the YUV422 data to be converted

@param	aWidth		The width of the data in the buffer in pixels

@param	aHeight		The height of the data in the buffer in pixels

*/

void RBitmap::YUVToRGB(TUint8* aDest, TUint8* aSource, TInt aWidth, TInt aHeight)

	{

	TInt y, u, v, r, g, b;

	aDest += ((aWidth * 3) * (aHeight - 1));

	for (TInt l = 0; l < aHeight; ++l)

		{

		for (TInt x = 0; x < (aWidth / 2); ++x)

			{

			u = (aSource[0] - 128);

			v = (aSource[2] - 128);

			y = (aSource[3] - 16);

			r = ((298 * y + 409 * u) / 256);

			g = ((298 * y - 100 * v - 208 * u) / 256);

			b = ((298 * y + 516 * v) / 256);

			CLIP(r);

			CLIP(g);

			CLIP(b);

			*aDest++ = (TUint8) r;

			*aDest++ = (TUint8) g;

			*aDest++ = (TUint8) b;

			y = (aSource[1] - 16);

			r = ((298 * y + 409 * u) / 256);

			g = ((298 * y - 100 * v - 208 * u) / 256);

			b = ((298 * y + 516 * v) / 256);

			CLIP(r);

			CLIP(g);

			CLIP(b);

			*aDest++ = (TUint8) r;

			*aDest++ = (TUint8) g;

			*aDest++ = (TUint8) b;

			aSource += 4;

			}

		aDest -= (aWidth * 3 * 2);

		}

	}

/**

Converts a 32 bit long from whatever the host format is into little endian format in a user supplied buffer.

@param	aBuffer		Pointer to the buffer into which to write the value

@param	aLong		The value to be written

*/

void RBitmap::PutLong(TUint8* aBuffer, TInt aLong)

	{

	*aBuffer++ = (TUint8) (aLong & 0xff);

	*aBuffer++ = (TUint8) ((aLong >> 8) & 0xff);

	*aBuffer++ = (TUint8) ((aLong >> 16) & 0xff);

	*aBuffer++ = (TUint8) ((aLong >> 24) & 0xff);

	}

/**

Converts a 16 bit short from whatever the host format is into little endian format in a user supplied buffer.

@param	aBuffer		Pointer to the buffer into which to write the value

@param	aShort		The value to be written

*/

void RBitmap::PutShort(TUint8* aBuffer, TInt16 aShort)

	{

	*aBuffer++ = (TUint8) (aShort & 0xff);

	*aBuffer++ = (TUint8) ((aShort >> 8) & 0xff);

	}

/**

Writes a standard Windows .bmp header to a file, including the standard .bmp file header, followed

by a V3 DIB header.

@param	aFile		A reference to the file to which to write the header

@param	aWidth		The width of the bitmap in pixels

@param	aHeight		The height of the bitmap in pixels

@return	KErrNone if write was successful, otherwise one of the other system wide error codes

*/

TInt RBitmap::WriteHeader(RFile& aFile, TInt aWidth, TInt aHeight)

	{

	TBuf8<14> header(14);

	TUint8* buffer = (TUint8*) header.Ptr();

	header.Fill(0);

	header[0] = 'B';

	header[1] = 'M';

	PutLong((buffer + 2), (14 + 40 + (aWidth * aHeight * 3)));

	PutLong((buffer + 10), (14 + 40));

	TInt r = aFile.Write(header);

	if (r == KErrNone)

		{

		TBuf8<40> bitmapInfoHeader(40);

		TUint8* buffer = (TUint8*) bitmapInfoHeader.Ptr();

		bitmapInfoHeader.Fill(0);

		PutLong(buffer, 40);

		PutLong((buffer + 4), aWidth);

		PutLong((buffer + 8), aHeight);

		PutShort((buffer + 12), 1);

		PutShort((buffer + 14), 24);

		PutLong((buffer + 20), (aWidth * aHeight * 3));

		r = aFile.Write(bitmapInfoHeader);

		}

	return r;

	}

/**

Converts a YUV422 or RGB565 buffer into 24 bit RGB format and writes it to a file.

@param	aFile			A reference to the file to which to write the data

@param	aBuffer			A pointer to the buffer containing the data to be converted and written

@param	aPixelFormat	UID specifying the format of the source data

@param	aWidth			The width of the data in the buffer in pixels

@param	aHeight			The height of the data in the buffer in pixels

@return	KErrNone if write was successful, otherwise one of the other system wide error codes

*/

TInt RBitmap::WriteBitmapData(RFile& aFile, TUint8* aBuffer, SDevCamPixelFormat aPixelFormat, TInt aWidth, TInt aHeight)

	{

	TInt length = (aWidth * aHeight * 3);

	TUint8* rgbBuffer = new TUint8[length];

	TInt r = KErrNone;

	if (rgbBuffer)

		{

		if (aPixelFormat.iPixelFormat == EUidPixelFormatYUV_422Interleaved)

			{

			YUVToRGB(rgbBuffer, aBuffer, aWidth, aHeight);

			}

		else if (aPixelFormat.iPixelFormat == EUidPixelFormatRGB_565)

			{

			RGBToRGB(rgbBuffer, aBuffer, aWidth, aHeight);

			}

		else

			{

			r = KErrNotSupported;

			}

		if (r == KErrNone)

			{

			TPtr8 buffer(rgbBuffer, length, length);

			r = aFile.Write(buffer);

			}

		delete [] rgbBuffer;

		}

	else

		{

		r = KErrNoMemory;

		}

	return r;

	}

/**

Converts a YUV422 or RGB565 buffer into 24 bit RGB format and writes it to a Windows .bmp file.

@param	aFileName		A reference to the fully qualified name of the file to write the .bmp file to

@param	aBuffer			A pointer to the buffer containing the data to be converted and written

@param	aPixelFormat	UID specifying the format of the source data

@param	aWidth			The width of the data in the buffer in pixels

@param	aHeight			The height of the data in the buffer in pixels

@return	KErrNone if write was successful, otherwise one of the other system wide error codes

*/

TInt RBitmap::WriteBMP(const TDesC& aFileName, TUint8* aBuffer, SDevCamPixelFormat aPixelFormat, TInt aWidth, TInt aHeight)

	{

	TInt r;

	RFile file;

	RFs fs;

	if ((r = fs.Connect()) == KErrNone)

		{

		if ((r = file.Replace(fs, aFileName, EFileWrite)) == KErrNone)

			{

			if ((r = WriteHeader(file, aWidth, aHeight)) == KErrNone)

				{

				r = WriteBitmapData(file, aBuffer, aPixelFormat, aWidth, aHeight);

				}

			file.Close();

			// If anything went wrong, delete the file so that we do not leave partial files that

			// might cause confusion

			if (r != KErrNone)

				{

				fs.Delete(aFileName);

				}

			}

		fs.Close();

		}

	return r;

	}

/**

Dumps a buffer straight to disk, without any kind of processing.

@param	aFileName		A reference to the fully qualified name of the file to write the file to

@param	aBuffer			A pointer to the buffer containing the data to be converted written

@param	aSize			The size of the buffer to be written, in bytes

@return	KErrNone if write was successful, otherwise one of the other system wide error codes

*/

TInt RBitmap::WriteBuffer(const TDesC& aFileName, TUint8* aBuffer, TInt aSize)

	{

	TInt r;

	RFile file;

	RFs fs;

	if ((r = fs.Connect()) == KErrNone)

		{

		if ((r = file.Replace(fs, aFileName, EFileWrite)) == KErrNone)

			{

			TPtrC8 buffer(aBuffer, aSize);

			r = file.Write(buffer);

			file.Close();

			// If anything went wrong, delete the file so that we do not leave partial files that

			// might cause confusion

			if (r != KErrNone)

				{

				fs.Delete(aFileName);

				}

			}

		fs.Close();

		}

	return r;

	}