// Copyright (c) 2003-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:

// Name        : MessageWriter.cpp

// Part of     : compressor

// Interface   : 

// SigComp message writer

// Version     : 1.0

//

#include <e32base.h>

#include <badesca.h>

#include "MessageWriter.h"

/** Message delimeter, appended whe streaming protocol is enabled */

static const TUint16 KMessageDelimeter = 0xffff;

CMessageWriter::CMessageWriter(CBufBase* aBuffer, TBool aStreamBasedProtocol)

                :iBuffer(aBuffer), iStreamBasedProtocol(aStreamBasedProtocol),

                iCount(0)

    {

    }

void CMessageWriter::WriteByteL(TInt aByte)

    {

    TUint8 tmp;

    if(iStreamBasedProtocol)

        {

        if(aByte == 0xFF)

            {

            iCount++;

            }

        else

            {

            if(iCount)

                {

                iCount--;

                tmp = 0xFF;

                iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

                tmp = static_cast<TUint8>(iCount);

                iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

                for(TInt i=0; i<(iCount); i++)

                    {

                    tmp = 0xFF;

                    iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

                    }

                iCount = 0;

                tmp = static_cast<TUint8>(aByte);

                iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

                }

            else

                {

                tmp = static_cast<TUint8>(aByte);

                iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

                }

            }

        }

    else

        {

        tmp = static_cast<TUint8>(aByte);

        iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

        }

    }

void CMessageWriter::WriteBlockL(const TDesC8& aBlock)

    {

    for(TInt i=0; i<aBlock.Length(); i++)

        {

        WriteByteL(aBlock[i]);

        }

    }

void CMessageWriter::FlushAndTerminateL()

    {

    if(iStreamBasedProtocol)

        {

        if(iCount)

            {

            TUint8 tmp;

// this code is not covered, because there is no way for deflate compressor

// to output FF at the end of compressed message, the EOS character encoding

// flushes all FFs in WriteByteL()

// I left the code there, as it may be usefull for other compressors.

            iCount--;

            tmp = 0xFF;

            iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

            tmp = static_cast<TUint8>(iCount);

            iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

            for(TInt i=0; i<(iCount); i++)

                {

                tmp = 0xFF;

                iBuffer->InsertL(iBuffer->Size(), &tmp, 1);

                }

            iCount = 0;

            }

        iBuffer->InsertL(iBuffer->Size(), &KMessageDelimeter, 2);

        }

    }