// Copyright (c) 1997-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 <e32base.h>

#include "JpegTypes.h"

void TDataUnitProcessor::SetBufferPtrUtil(CRgbBufferPtr& aBufferPtrUtil)

    {

    iBufferPtrUtil = &aBufferPtrUtil;

    }

// TMonoProcessor

FORCEDINLINE TInt16 MonoPixel(CRgbBufferPtr::TConstRgbBufferPtr aRgbBuffer)

    {

    return TInt16(

        			  (

        			  	(CRgbBufferPtr::Red(aRgbBuffer)<<1) + 

        			    CRgbBufferPtr::Green(aRgbBuffer) +

        			    (CRgbBufferPtr::Green(aRgbBuffer)<<2) + 

        			    CRgbBufferPtr::Blue(aRgbBuffer)

        			   )>>3

        	    );

    }

TDataUnit* TMonoProcessor::Process(CRgbBufferPtr::TConstRgbBufferPtr aRgbBuffer)

	{

	TInt16* YPtr = iMCU.iCoeff;

	const TInt16* YPtrLimit = YPtr + KJpgDCTBlockSize;

	const TInt KLineStride = iBufferPtrUtil->NextLineOffset();

	do

		{

		TInt p=0;

		do 

		    {		

		    *YPtr++ = MonoPixel( CRgbBufferPtr::ShiftPtr(aRgbBuffer, p++) );

		    *YPtr++ = MonoPixel( CRgbBufferPtr::ShiftPtr(aRgbBuffer, p++) );

		    } while (p < KJpgDCTBlockWidth);

		aRgbBuffer += KLineStride;

		} while (YPtr < YPtrLimit);

	return &iMCU;

	}

// T444Processor

TDataUnit* T444Processor::Process(CRgbBufferPtr::TConstRgbBufferPtr aRgbBuffer)

	{

	TInt16* YPtr  = iMCU[0].iCoeff;

	TInt16* CbPtr = iMCU[1].iCoeff;

	TInt16* CrPtr = iMCU[2].iCoeff;

	const TInt16* YPtrLimit = YPtr + KJpgDCTBlockSize;

	const TInt KLineStride = iBufferPtrUtil->NextLineOffset();

	do

		{

		TInt p=0;

		do 

		    {

    		TYCbCr::RGBtoYCbCr(YPtr++, CbPtr++, CrPtr++, CRgbBufferPtr::ShiftPtr(aRgbBuffer, p++) );

		    } while (p < KJpgDCTBlockWidth);

		aRgbBuffer += KLineStride;

		} while (YPtr < YPtrLimit);

	return iMCU;

	}

// T422Processor

TDataUnit* T422Processor::Process(CRgbBufferPtr::TConstRgbBufferPtr aRgbBuffer)

	{

	TInt16* CbPtr = iMCU[2].iCoeff;

	ASSERT( CbPtr + KJpgDCTBlockSize == iMCU[3].iCoeff); // we assume that V comp is next to U one

	const TInt halfWidth = KJpgDCTBlockWidth >> 1;

	DoProcess(iMCU[0].iCoeff,CbPtr, aRgbBuffer);

	DoProcess(iMCU[1].iCoeff,CbPtr + halfWidth, CRgbBufferPtr::ShiftPtr(aRgbBuffer,KJpgDCTBlockWidth) );

	return iMCU;

	}

void T422Processor::DoProcess(TInt16* aYPtr,TInt16* aCbPtr, CRgbBufferPtr::TConstRgbBufferPtr aRgbBuffer)

	{

	const TInt KLineStride = iBufferPtrUtil->NextLineOffset();

	const TInt16* YPtrLimit = aYPtr + KJpgDCTBlockSize;

	do

		{

		TYCbCr::RGBtoYCbCr2PixUVSum(aYPtr,aCbPtr, aRgbBuffer);

		TYCbCr::RGBtoYCbCr2PixUVSum(aYPtr+2,aCbPtr+1, CRgbBufferPtr::ShiftPtr(aRgbBuffer,2));

		TYCbCr::RGBtoYCbCr2PixUVSum(aYPtr+4,aCbPtr+2,CRgbBufferPtr::ShiftPtr(aRgbBuffer,4));

		TYCbCr::RGBtoYCbCr2PixUVSum(aYPtr+6,aCbPtr+3,CRgbBufferPtr::ShiftPtr(aRgbBuffer,6));

		aYPtr  += KJpgDCTBlockWidth;

		aCbPtr += KJpgDCTBlockWidth;

		aRgbBuffer += KLineStride;

		} while (aYPtr < YPtrLimit);

	}

// T420Processor

TDataUnit* T420Processor::Process(CRgbBufferPtr::TConstRgbBufferPtr aRgbBuffer)

	{

	TInt16* CbPtr = iMCU[4].iCoeff;

	ASSERT( CbPtr + KJpgDCTBlockSize == iMCU[5].iCoeff); // we assume that V comp is next to U one

	const TInt halfWidth = KJpgDCTBlockWidth >> 1;

	const TInt halfSize = KJpgDCTBlockSize >> 1;

	DoProcess(iMCU[0].iCoeff,CbPtr, aRgbBuffer);

	DoProcess(iMCU[1].iCoeff,CbPtr + halfWidth, CRgbBufferPtr::ShiftPtr(aRgbBuffer, KJpgDCTBlockWidth));

	aRgbBuffer += (iBufferPtrUtil->NextLineOffset() << 3);

	CbPtr += halfSize;

	DoProcess(iMCU[2].iCoeff,CbPtr, aRgbBuffer);

	DoProcess(iMCU[3].iCoeff,CbPtr + halfWidth, CRgbBufferPtr::ShiftPtr(aRgbBuffer, KJpgDCTBlockWidth));

	return iMCU;

	}

void T420Processor::DoProcess(TInt16* aYPtr,TInt16* aCbPtr, CRgbBufferPtr::TConstRgbBufferPtr  aRgbBuffer)

	{

	const TInt16* YPtrLimit = aYPtr + KJpgDCTBlockSize;

	const TInt KLineStride = iBufferPtrUtil->NextLineOffset();

	const TInt KDoubleLineStride = KLineStride << 1;

	do

		{

		TYCbCr::RGBtoYCbCr4PixUVSum(aYPtr, aCbPtr, aRgbBuffer, KLineStride);

		TYCbCr::RGBtoYCbCr4PixUVSum(aYPtr+2, aCbPtr+1, CRgbBufferPtr::ShiftPtr(aRgbBuffer,2), KLineStride);

        TYCbCr::RGBtoYCbCr4PixUVSum(aYPtr+4, aCbPtr+2, CRgbBufferPtr::ShiftPtr(aRgbBuffer,4), KLineStride);

        TYCbCr::RGBtoYCbCr4PixUVSum(aYPtr+6, aCbPtr+3, CRgbBufferPtr::ShiftPtr(aRgbBuffer,6), KLineStride);

		aRgbBuffer 	+= KDoubleLineStride;

		aCbPtr 		+= KJpgDCTBlockWidth;

		aYPtr 		+= KJpgDCTBlockWidth << 1;

		} while (aYPtr < YPtrLimit);

	}