// 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:

//

/**

 @file

 @internalComponent

 */

#include <e32err.h>

#include <mtp/mtpprotocolconstants.h>

#include "mtpservicecommon.h"

#include "rmtpframework.h"

#include "cmtpdatacodegenerator.h"

#include "cmtpservicemgr.h"

const TUint16 KUndenfinedStartCode = EMTPCodeUndefined1Start + 1;

const TUint16 KUndenfinedEndCode = EMTPCodeUndefined1End;

CMTPDataCodeGenerator* CMTPDataCodeGenerator::NewL()

    {

    CMTPDataCodeGenerator* self = new (ELeave) CMTPDataCodeGenerator();

    CleanupStack::PushL ( self );

    self->ConstructL ();

    CleanupStack::Pop ( self );

    return self;

    }

CMTPDataCodeGenerator::~CMTPDataCodeGenerator()

    {

    iSingletons.Close();

    }

void CMTPDataCodeGenerator::ConstructL()

    {

    iSingletons.OpenL ();

    }

CMTPDataCodeGenerator::CMTPDataCodeGenerator() :

    iUndefinedNextCode(KUndenfinedStartCode ),

    iVendorExtFormatCode(EMTPFormatCodeVendorExtDynamicStart)

    {

    }

TInt CMTPDataCodeGenerator::IncServiceIDResource( const TUint aServiceType, TUint& aServiceID )

    {

    if ( iUndefinedNextCode >= KUndenfinedEndCode )

        return KErrOverflow;

    switch ( aServiceType )

       {

       case EMTPServiceTypeNormal:

           {

           aServiceID = ( ( ++iUndefinedNextCode ) | KNormalServiceTypeMask );

           }

           break;

       case EMTPServiceTypeAbstract:

           {

           aServiceID = ( (++iUndefinedNextCode) | KAbstrackServiceTypeMask );

           }

           break;

       default:

           {

           }

       }

    return KErrNone;

    }

void CMTPDataCodeGenerator::DecServiceIDResource()

    {

    iUndefinedNextCode--;

    }

TBool CMTPDataCodeGenerator::IsValidServiceType( const TUint aServiceType ) const

    {

    return ( (EMTPServiceTypeNormal == aServiceType) || (EMTPServiceTypeAbstract == aServiceType) );

    }

TInt CMTPDataCodeGenerator::AllocateServiceID(const TMTPTypeGuid& aPGUID, const TUint aServiceType, TUint& aServiceID )

    {

    if( !IsValidServiceType(aServiceType) )

        return KErrArgument;

    TInt err(KErrNone);

    TUint retID (KInvliadServiceID);

    if( iSingletons.ServiceMgr().IsSupportedService(aPGUID) )

        {

        if( iSingletons.ServiceMgr().ServiceTypeOfSupportedService(aPGUID) != aServiceType )

            return KErrArgument;

        err = iSingletons.ServiceMgr().GetServiceId(aPGUID , retID);

        if( KErrNone != err )

            {

            if((err = IncServiceIDResource( aServiceType, retID )) != KErrNone)

                return err;

            err = iSingletons.ServiceMgr().EnableService( aPGUID, retID );

            if( KErrNone != err )

                {

                DecServiceIDResource();

                return err;

                }

            }

        }

    else

        {

        if((err = IncServiceIDResource( aServiceType, retID )) != KErrNone)

            return err;

        }

   aServiceID = retID;

   iSingletons.ServiceMgr().InsertServiceId( retID );

    return KErrNone;

    }

TInt CMTPDataCodeGenerator::AllocateServicePropertyCode( const TMTPTypeGuid& aServicePGUID, const TMTPTypeGuid& aPKNamespace, const TUint aPKID, TUint16& aServicePropertyCode )

    {

    TUint retID = KInvliadU16DataCode;

    if( iSingletons.ServiceMgr().IsSupportedService(aServicePGUID) )

        {

        TInt err = iSingletons.ServiceMgr().GetServicePropertyCode( aServicePGUID, aPKNamespace, aPKID, retID );

        if( KErrNone != err )

            return err;

        if(retID == KInvliadU16DataCode)

           {

           if ( iUndefinedNextCode >= KUndenfinedEndCode )

               return KErrOverflow;

           retID = ++iUndefinedNextCode;

           iSingletons.ServiceMgr().SetServicePropertyCode( aServicePGUID, aPKNamespace, aPKID, retID);

           }

        }

    else

        {

        if ( iUndefinedNextCode >= KUndenfinedEndCode )

            return KErrOverflow;

        retID = ++iUndefinedNextCode;

        }

    aServicePropertyCode = retID;

    return KErrNone;

    }

TInt CMTPDataCodeGenerator::AllocateServiceFormatCode( const TMTPTypeGuid& aServicePGUID, const TMTPTypeGuid& aGUID, TUint16& aServiceFormatCode )

    {

    TUint retID = KInvliadU16DataCode;

    if( iSingletons.ServiceMgr().IsSupportedService(aServicePGUID) )

        {

        TInt err = iSingletons.ServiceMgr().GetServiceFormatCode( aServicePGUID, aGUID, retID );

        if( KErrNone != err )

            return err;

        if(retID == KInvliadU16DataCode)

           {

           if ( iVendorExtFormatCode > EMTPFormatCodeVendorExtDynamicEnd )

               return KErrOverflow;

           retID = ++iVendorExtFormatCode;

           iSingletons.ServiceMgr().SetServiceFormatCode( aServicePGUID, aGUID, retID);

           }

        }

    else

        {

        if ( iVendorExtFormatCode > EMTPFormatCodeVendorExtDynamicEnd )

            return KErrOverflow;

        retID = ++iVendorExtFormatCode;

        }

    aServiceFormatCode = retID;

    return KErrNone;

    }

TInt CMTPDataCodeGenerator::AllocateServiceMethodFormatCode( const TMTPTypeGuid& aServicePGUID, const TMTPTypeGuid& aGUID, TUint16& aMethodFormatCode )

    {

    TUint retID = KInvliadU16DataCode;

    if( iSingletons.ServiceMgr().IsSupportedService(aServicePGUID) )

        {

        TInt err = iSingletons.ServiceMgr().GetServiceMethodCode( aServicePGUID, aGUID, retID );

        if( KErrNone != err )

            return err;

        if(retID == KInvliadU16DataCode)

           {

           if ( iUndefinedNextCode > KUndenfinedEndCode )

               return KErrOverflow;

           retID = ++iUndefinedNextCode;

           iSingletons.ServiceMgr().SetServiceMethodCode( aServicePGUID, aGUID, retID);

           }

        }

    else

        {

        if ( iUndefinedNextCode > KUndenfinedEndCode )

            return KErrOverflow;

        retID = ++iUndefinedNextCode;

        }

    aMethodFormatCode = retID;

    return KErrNone;

    }