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

// GZIP output stream functions

//

#include "xmlenggzipbufferoutputstream.h"

#include <ezstream.h>

#include <ezcompressor.h>

    CXmlEngGZIPBufferOutputStream* CXmlEngGZIPBufferOutputStream::NewLC( RBuf8& aOutputBuffer )

	    {

	    CXmlEngGZIPBufferOutputStream* gzbos = new (ELeave) CXmlEngGZIPBufferOutputStream( aOutputBuffer );

	    CleanupStack::PushL( gzbos );

	    gzbos->ConstructL();

	    return gzbos;	

	    }

    CXmlEngGZIPBufferOutputStream* CXmlEngGZIPBufferOutputStream::NewL( RBuf8& aOutputBuffer )

	    {

	    CXmlEngGZIPBufferOutputStream* gzbos = CXmlEngGZIPBufferOutputStream::NewLC( aOutputBuffer );

	    CleanupStack::Pop( gzbos );

	    return gzbos;	

	    }

   CXmlEngGZIPBufferOutputStream::CXmlEngGZIPBufferOutputStream( RBuf8& aOutputBuffer ) : iOutputString(aOutputBuffer)

        {

        iKeepGoing = ETrue;

        }

   void CXmlEngGZIPBufferOutputStream::ConstructL()

        {

        if(!iOutputString.MaxSize())

            {

            iOutputString.ReAllocL(KOutputBufferSize);

            }

        //we can't create compressor here, because we heve no input at this moment

        iOutputDescriptor.CreateL( KOutputBufferSize );

        iInputDescriptor.CreateL( KDefaultInputBufferSize );

        iOldInputDescriptor.CreateL( KDefaultInputBufferSize );

        }

   CXmlEngGZIPBufferOutputStream::~CXmlEngGZIPBufferOutputStream()

        {

        delete iCompressor;

        iOutputDescriptor.Close();

        iInputDescriptor.Close();

        iOldInputDescriptor.Close();

        }

	TInt CXmlEngGZIPBufferOutputStream::Write( const TDesC8& aBuffer )

		{

		TRAP ( iError, WriteL( aBuffer ) );

		if ( iError )

		    {

	        return -1;

		    }

		return aBuffer.Size();

		}

	void CXmlEngGZIPBufferOutputStream::WriteL( const TDesC8& aBuffer )

		{

		iNoInputNeeded = ETrue;

		if ( !iCompressor )

		    {

		    const TInt newLength = aBuffer.Size();

		    if ( newLength > iInputDescriptor.Size() )

		        {

		        iInputDescriptor.ReAllocL( newLength );

		        }

		    iInputDescriptor.Copy( aBuffer );

		    iCompressor = CEZCompressor::NewL( *this );

		    }

		else

		    {

		    iInputDescriptor.Swap( iOldInputDescriptor );

		    const TInt newLength = aBuffer.Length();

		    if ( newLength > iInputDescriptor.MaxLength() )

		        {

		        iInputDescriptor.ReAllocL( newLength );

		        }

		    iInputDescriptor.Copy( aBuffer );		    

		    while ( iNoInputNeeded )

		        {

		        iCompressor->DeflateL();

		        }		        

		    }       		

		}

	TInt CXmlEngGZIPBufferOutputStream::Close()

		{

		iCloseInvoked = ETrue;

        if ( iError )

            {

            return -1;

            }		

		TRAP ( iError, CloseL() );

		if ( iError )

		    {

	        return -1;

		    }		

		return KErrNone;

		}

	void CXmlEngGZIPBufferOutputStream::CloseL()

		{

		while ( iKeepGoing )

		    {

		    iKeepGoing = iCompressor->DeflateL();

		    }

		}

	TInt CXmlEngGZIPBufferOutputStream::CheckError()

	    {

	    return iError;

	    }

    void CXmlEngGZIPBufferOutputStream::InitializeL( CEZZStream& aZStream )

    	{

    	aZStream.SetInput( iInputDescriptor );

    	aZStream.SetOutput( iOutputDescriptor );

    	iNoInputNeeded = EFalse;

    	}

    void CXmlEngGZIPBufferOutputStream::NeedInputL( CEZZStream& aZStream )

    	{

        if ( iCloseInvoked )

            {

            aZStream.SetInput( KNullDesC8 );

            }

        else        

            {

        	aZStream.SetInput( iInputDescriptor );

        	iNoInputNeeded = EFalse;

        	}

    	}

    void CXmlEngGZIPBufferOutputStream::NeedOutputL( CEZZStream& aZStream )

    	{

    	TPtrC8 od = aZStream.OutputDescriptor();

    	while(TRUE)

    	    {

    	    if (od.Size() > (iOutputString.MaxSize() - iOutputString.Size()))

		        {

		        iOutputString.ReAllocL( iOutputString.MaxSize()*2 );

		        }

		    else

		        {

		        break;

		        }    	    

    	    }

    	iOutputString.Append( od );

    	aZStream.SetOutput( iOutputDescriptor );

    	}

    void CXmlEngGZIPBufferOutputStream::FinalizeL( CEZZStream& aZStream )

        {

        TPtrC8 od = aZStream.OutputDescriptor();

    	while(TRUE)

    	    {

    	    if (od.Size() > (iOutputString.MaxSize() - iOutputString.Size()))

		        {

		        iOutputString.ReAllocL( iOutputString.MaxSize()*2 );

		        }

		    else

		        {

		        break;

		        }    	    

    	    }

    	iOutputString.Append( od );

    	}