/*

* 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 "msmmengine.h"

#include "msmmnodebase.h"

#include <usb/usblogger.h>

CMsmmEngine::~CMsmmEngine()

    {

    if (iDataEntrys)

        {

        delete iDataEntrys;

        }

    }

CMsmmEngine* CMsmmEngine::NewL()

    {

    CMsmmEngine* self = CMsmmEngine::NewLC();

    CleanupStack::Pop(self);

    return self;

    }

CMsmmEngine* CMsmmEngine::NewLC()

    {

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

    CleanupStack::PushL(self);

    self->ConstructL();

    return self;

    }

void CMsmmEngine::AddUsbMsDeviceL(const TUSBMSDeviceDescription& aDevice)

    {

    TUsbMsDevice* device = SearchDevice(aDevice.iDeviceId);

    if (!device)

        {

        device = new (ELeave) TUsbMsDevice(aDevice);

        iDataEntrys->AddChild(device);

        }

    }

TUsbMsInterface* CMsmmEngine::AddUsbMsInterfaceL(TInt aDeviceId, TUint8 aInterfaceNumber,

        TInt32 aInterfaceToken)

    {

    TUsbMsDevice* device = SearchDevice(aDeviceId);

    if (!device)

        {

        User::Leave(KErrArgument);

        }

    TUsbMsInterface* interface = 

        SearchInterface(device, aInterfaceNumber);

    if (interface)

        {

        User::Leave(KErrAlreadyExists);

        }

    else

        {

        interface = AddUsbMsInterfaceNodeL(device, aInterfaceNumber, aInterfaceToken);

        }

    return interface;

    }

void CMsmmEngine::AddUsbMsLogicalUnitL(TInt aDeviceId,

        TInt aInterfaceNumber, TInt aLogicalUnitNumber, TText aDrive)

    {

    TUsbMsDevice* device = SearchDevice(aDeviceId);

    if (!device)

        {

        User::Leave(KErrArgument); // A proper device node can't be found

        }

    TUsbMsInterface* interface = SearchInterface(device, aInterfaceNumber);

    if (interface)

        {

        AddUsbMsLogicalUnitNodeL(interface, aLogicalUnitNumber, aDrive);

        }

    else

        {

        User::Leave(KErrArgument); // A proper interface node can't be found

        }

    }

void CMsmmEngine::RemoveUsbMsNode(TMsmmNodeBase* aNodeToBeRemoved)

    {

    delete aNodeToBeRemoved;

    }

TUsbMsDevice* CMsmmEngine::SearchDevice(TInt aDeviceId) const

    {

    return static_cast<TUsbMsDevice*>(

            iDataEntrys->SearchInChildren(aDeviceId));

    }

TUsbMsInterface* CMsmmEngine::SearchInterface(TMsmmNodeBase* aDevice, 

        TInt aInterfaceNumber) const

    {

    return static_cast<TUsbMsInterface*>(

            aDevice->SearchInChildren(aInterfaceNumber));

    }

CMsmmEngine::CMsmmEngine()

    {

    }

void CMsmmEngine::ConstructL()

    {

    // Create the root of the whole node tree

    iDataEntrys = new (ELeave) TMsmmNodeBase(0x0);

    }

TUsbMsInterface* CMsmmEngine::AddUsbMsInterfaceNodeL(TUsbMsDevice* iParent,

        TInt aInterfaceNumber, TInt aInterfaceToken)

    {

    TUsbMsInterface* interface = new (ELeave) TUsbMsInterface(

            aInterfaceNumber, aInterfaceToken);

    iParent->AddChild(interface);

    return interface;

    }

TUsbMsLogicalUnit* CMsmmEngine::AddUsbMsLogicalUnitNodeL(

        TUsbMsInterface* iParent, TInt aLogicalUnitNumber, 

        TText aDrive)

    {

    TUsbMsLogicalUnit* logicalUnit = new (ELeave) TUsbMsLogicalUnit(

            aLogicalUnitNumber, aDrive);

    iParent->AddChild(logicalUnit);

    return logicalUnit;

    }

// End of file