/****************************************************************************

**

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the Qt Mobility Components.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "s60audiocapturesession.h"

#include <QtCore/qdebug.h>

#include <QtCore/qurl.h>

#include <QDir>

#include <Mda\Common\Audio.h>

#include <Mda\Common\Resource.h>

#include <Mda\Client\Utility.h>

#include <MdaAudioSampleEditor.h>

#include <mmf\common\mmfcontrollerpluginresolver.h>

#include <mmf\common\mmfcontroller.h>

#include <BADESCA.H>

#include <BAUTILS.H>

#include <coemain.h>  

_LIT(KAudioDummyFile, "c:\\data\\temp\\temp.wav");

S60AudioCaptureSession::S60AudioCaptureSession(QObject *parent):

    QObject(parent)

    , m_recorderUtility(NULL)    

    , m_captureState(ENotInitialized)    

    , m_controllerIdMap(QHash<QString, ControllerData>())    

    , m_audioCodeclist(QHash<QString, CodecData>())    

    , m_error(QMediaRecorder::NoError) 

{

    TRAPD(err, initializeSessionL());

    setError(err);

}

void S60AudioCaptureSession::initializeSessionL()

{    

    m_recorderUtility = CMdaAudioRecorderUtility::NewL(*this, 0, 80,(TMdaPriorityPreference) 0x5210001);        

    updateAudioContainersL();    

    populateAudioCodecsDataL();

    setDefaultSettings();

}

void S60AudioCaptureSession::setError(TInt aError)

{

    if (aError == KErrNone)

        return;

    m_error = aError;

    QMediaRecorder::Error recorderError = fromSymbianErrorToMultimediaError(m_error);

    // TODO: fix to user friendly string at some point

    // These error string are only dev usable

    QString symbianError; 

    symbianError.append("Symbian:");

    symbianError.append(QString::number(m_error));

    emit error(recorderError, symbianError);

}

QMediaRecorder::Error S60AudioCaptureSession::fromSymbianErrorToMultimediaError(int error)

{

    switch(error) {

    case KErrNoMemory:

    case KErrNotFound:

    case KErrBadHandle:

    case KErrAbort:    

    case KErrCorrupt:

    case KErrGeneral:    

    case KErrPathNotFound:        

    case KErrUnknown:

    case KErrNotReady:

    case KErrInUse:

    case KErrAccessDenied:

    case KErrLocked:

    case KErrPermissionDenied:

    case KErrAlreadyExists:        

        return QMediaRecorder::ResourceError;    

    case KErrNotSupported:   

    case KErrArgument:        

        return QMediaRecorder::FormatError;   

    case KErrNone:

    default:

        return QMediaRecorder::NoError;

    }

}

S60AudioCaptureSession::~S60AudioCaptureSession()

{       

    delete m_recorderUtility;    

}

QAudioFormat S60AudioCaptureSession::format() const

{

    return m_format;

}

bool S60AudioCaptureSession::setFormat(const QAudioFormat &format)

{		

    m_format = format;

    return true;

}

QStringList S60AudioCaptureSession::supportedAudioCodecs() const

{   

    return m_audioCodeclist.keys();    

}

QStringList S60AudioCaptureSession::supportedAudioContainers() const

{

    return m_controllerIdMap.keys();

}

QString S60AudioCaptureSession::codecDescription(const QString &codecName)

{

    if (m_audioCodeclist.keys().contains(codecName))

        return m_audioCodeclist.value(codecName).codecDescription;

    else

        return QString();

}

QString S60AudioCaptureSession::audioContainerDescription(const QString &containerName)

{   

    if (m_controllerIdMap.keys().contains(containerName))

        return m_controllerIdMap.value(containerName).destinationFormatDescription;

    else

        return QString();

}

bool S60AudioCaptureSession::setAudioCodec(const QString &codecName)

{    

    QStringList codecs = supportedAudioCodecs();

    if(codecs.contains(codecName)) {

        m_format.setCodec(codecName);

        return true;

    }    

    return false;

}

bool S60AudioCaptureSession::setAudioContainer(const QString &containerMimeType) 

{    

    QStringList containers = supportedAudioContainers();

    if(containers.contains(containerMimeType)) { 

        m_container = containerMimeType;

        return true;

    }

    return false;

}

QString S60AudioCaptureSession::audioCodec() const

{   

    return m_format.codec();

}

QString S60AudioCaptureSession::audioContainer() const

{

    return m_container;

}

QUrl S60AudioCaptureSession::outputLocation() const

{

    return m_sink;

}

bool S60AudioCaptureSession::setOutputLocation(const QUrl& sink)

{

    QString filename = QDir::toNativeSeparators(m_sink.toString());

    TPtrC16 path(reinterpret_cast<const TUint16*>(filename.utf16()));         

    TRAPD(err, BaflUtils::EnsurePathExistsL(CCoeEnv::Static()->FsSession(),path));

    if (err==KErrNone) {

        m_sink = sink;

        setError(err);

        return true;

    }else {

        setError(err);

        return false;

    }

}

qint64 S60AudioCaptureSession::position() const

{    

    if ((m_captureState != ERecording) || !m_recorderUtility) 

        return 0;

    return m_recorderUtility->Duration().Int64() / 1000;        

}

QString S60AudioCaptureSession::initializeSinkL()

{    

    QString sink = QDir::toNativeSeparators(m_sink.toString());    

    int index = sink.lastIndexOf('.');

    if (index != -1)

        sink.chop(sink.length()-index);

    sink.append(m_controllerIdMap.value(m_container).fileExtension);    

    return sink;    

}

void S60AudioCaptureSession::record()

{   

    if (!m_recorderUtility)

        return;

    if (m_captureState == EInitialized || m_captureState == ERecordComplete) {

        QString filename = initializeSinkL();

        TPtrC16 sink(reinterpret_cast<const TUint16*>(filename.utf16()));        

        TUid controllerUid(TUid::Uid(m_controllerIdMap.value(m_container).controllerUid));

        TUid formatUid(TUid::Uid(m_controllerIdMap.value(m_container).destinationFormatUid));        

        TRAPD(err, 

        if (m_container != "audio/amr")   

            m_recorderUtility->OpenFileL(sink, controllerUid, KNullUid, formatUid);

        else   

            m_recorderUtility->OpenFileL(sink);

        );       

        setError(err);

    }else if (m_captureState == EPaused) {        

        m_recorderUtility->SetPosition(m_pausedPosition);

        TRAPD(error, m_recorderUtility->RecordL());

        setError(error);

        m_captureState = ERecording;

        emit stateChanged(m_captureState);

    }        

}

void S60AudioCaptureSession::setDefaultSettings()

{

    // Setting AMR to default format if supported

    if (m_controllerIdMap.count() > 0) {

        if ( m_controllerIdMap.contains("audio/amr"))

            m_container = QString("audio/amr");

        else

            m_container = m_controllerIdMap.keys()[0];

    }

    if (m_audioCodeclist.keys().count() > 0) {

        if (m_audioCodeclist.keys().contains("AMR")) {            

            m_format.setSampleSize(8);

            m_format.setChannels(1);

            m_format.setFrequency(8000);

            m_format.setSampleType(QAudioFormat::SignedInt);            

            m_format.setCodec("AMR");

        }else

            m_format.setCodec(m_audioCodeclist.keys()[0]);

    }

}

void S60AudioCaptureSession::pause()

{  

    if (!m_recorderUtility)

        return;

    m_pausedPosition = m_recorderUtility->Position();

    m_recorderUtility->Stop();

    m_captureState = EPaused;

    emit stateChanged(m_captureState);    

}

void S60AudioCaptureSession::stop()

{    

    if (!m_recorderUtility)

        return;

    m_recorderUtility->Stop();

    m_recorderUtility->Close();        

    m_captureState = ERecordComplete;

    emit stateChanged(m_captureState);   

}

void S60AudioCaptureSession::setCaptureDevice(const QString &deviceName)

{

    m_captureDevice = deviceName;

}

void S60AudioCaptureSession::MoscoStateChangeEvent(CBase* aObject,

        TInt aPreviousState, TInt aCurrentState, TInt aErrorCode)

{ 	    

    if (aErrorCode==KErrNone) {

	    TRAPD(err, MoscoStateChangeEventL(aObject, aPreviousState, aCurrentState, NULL));

	    setError(err);

	}

	setError(aErrorCode);

}

void S60AudioCaptureSession::MoscoStateChangeEventL(CBase* aObject,

        TInt aPreviousState, TInt aCurrentState, TInt aErrorCode)

{    

	if (aObject != m_recorderUtility)

	    return;

		switch(aCurrentState) {

        case CMdaAudioClipUtility::EOpen: {            

            if(aPreviousState == CMdaAudioClipUtility::ENotReady) {                

                if (m_captureState == EInitializing) {                    

                    retrieveSupportedAudioSampleRatesL();                   

                    m_recorderUtility->Close();                    

                    m_captureState = EInitialized;

                    emit stateChanged(m_captureState);

                }else {         

                    applyAudioSettingsL();                    

                    m_recorderUtility->SetGain(m_recorderUtility->MaxGain());                 

                    m_recorderUtility->RecordL();

                    m_captureState = EOpenCompelete;                    

                    emit stateChanged(m_captureState);

                    }

                break;

            }

        }		

        case CMdaAudioClipUtility::ENotReady: {            

            m_captureState = EInitialized;            

            emit stateChanged(m_captureState);

            break;

        }

        case CMdaAudioClipUtility::ERecording: {            

            m_captureState = ERecording;            

            emit stateChanged(m_captureState);

            break;

        }

        default: {            

            break;

        }   

		}    

}

void S60AudioCaptureSession::updateAudioContainersL()

{

    CMMFControllerPluginSelectionParameters* pluginParameters = 

    	CMMFControllerPluginSelectionParameters::NewLC();

	CMMFFormatSelectionParameters* formatParameters = 

		CMMFFormatSelectionParameters::NewLC();

	pluginParameters->SetRequiredRecordFormatSupportL(*formatParameters);

	RArray<TUid> ids;

	CleanupClosePushL(ids);

	User::LeaveIfError(ids.Append(KUidMediaTypeAudio));  

	pluginParameters->SetMediaIdsL(ids, 

		CMMFPluginSelectionParameters::EAllowOnlySuppliedMediaIds);

	RMMFControllerImplInfoArray controllers; 

	CleanupResetAndDestroyPushL(controllers);

	//Get all audio record controllers/formats that are supported

	pluginParameters->ListImplementationsL(controllers);

	for (TInt index=0; index<controllers.Count(); index++) {

		const RMMFFormatImplInfoArray& recordFormats = 

			controllers[index]->RecordFormats();

		for (TInt j=0; j<recordFormats.Count(); j++) {

			const CDesC8Array& mimeTypes = recordFormats[j]->SupportedMimeTypes();

			const CDesC8Array& fileExtensions = recordFormats[j]->SupportedFileExtensions();

			TInt mimeCount = mimeTypes.Count();

			TInt fileExtCount = fileExtensions.Count();

			if (mimeCount > 0 && fileExtCount > 0) {

                TPtrC8 extension = fileExtensions[0];

                TPtrC8 mimeType = mimeTypes[0];             

                QString type = QString::fromUtf8((char *)mimeType.Ptr(), mimeType.Length());          

				if (type != "audio/mp4" && type != "audio/basic") {

                    ControllerData data;

                    data.controllerUid = controllers[index]->Uid().iUid;

                    data.destinationFormatUid = recordFormats[j]->Uid().iUid;

                    data.destinationFormatDescription = QString::fromUtf16(

                            recordFormats[j]->DisplayName().Ptr(),

                            recordFormats[j]->DisplayName().Length());  

                    data.fileExtension = QString::fromUtf8((char *)extension.Ptr(), extension.Length());  

                    m_controllerIdMap[type] = data;

				}

			}

		}

	}	

	CleanupStack::PopAndDestroy(4);//controllers, ids, formatParameters, pluginParameters

}

void S60AudioCaptureSession::retrieveSupportedAudioSampleRatesL()

{       

    if (!m_recorderUtility || m_captureState != EInitializing) 

        return;

    m_supportedSampleRates.clear();

    RArray<TUint> supportedSampleRates;

    CleanupClosePushL(supportedSampleRates);

    m_recorderUtility->GetSupportedSampleRatesL(supportedSampleRates);        

    for (TInt j = 0; j < supportedSampleRates.Count(); j++ )             

        m_supportedSampleRates.append(supportedSampleRates[j]);                                

    CleanupStack::PopAndDestroy(&supportedSampleRates);    

}

QList<int> S60AudioCaptureSession::supportedAudioSampleRates(const QAudioEncoderSettings &settings) const

{

    QList<int> supportedSampleRates;

    if (!settings.codec().isEmpty()) {

        if (settings.codec() == "AMR")             

            supportedSampleRates.append(8000);

        else

            supportedSampleRates = m_supportedSampleRates; 

    }else 

        supportedSampleRates = m_supportedSampleRates;           

    return supportedSampleRates;

}

void S60AudioCaptureSession::populateAudioCodecsDataL()

{        

    if (!m_recorderUtility)  

        return;

    if (m_controllerIdMap.contains("audio/amr")) {       

        CodecData data;              

        data.codecDescription = QString("GSM AMR Codec");

        m_audioCodeclist[QString("AMR")]=data;

    }   

    if (m_controllerIdMap.contains("audio/basic")) {

        CodecData data;           

        data.fourCC = KMMFFourCCCodeALAW;                   

        data.codecDescription = QString("Sun/Next ""Au"" audio codec");

        m_audioCodeclist[QString("AULAW")]=data;            

    }

    if (m_controllerIdMap.contains("audio/wav")) {            

        CodecData data;           

        data.fourCC = KMMFFourCCCodePCM16;                   

        data.codecDescription = QString("Pulse code modulation");

        m_audioCodeclist[QString("PCM")]=data;

    }    

    if (m_controllerIdMap.contains("audio/mp4")) {            

        CodecData data;           

        data.fourCC = KMMFFourCCCodeAAC;                   

        data.codecDescription = QString("Advanced Audio Codec");

        m_audioCodeclist[QString("AAC")]=data;

    }        

    // default samplerates

    m_supportedSampleRates << 96000 << 88200 << 64000 << 48000 << 44100 << 32000 << 24000 << 22050 << 16000 << 12000 << 11025 << 8000;

    m_captureState = EInitialized;

    if (m_controllerIdMap.contains("audio/wav") && m_captureState != EInitialized) {            

        TMdaFileClipLocation location;   

        location.iName = KAudioDummyFile();

        TMdaWavClipFormat format;

        m_captureState = EInitializing;

        m_recorderUtility->OpenL(&location, &format);             

    }      

}

void S60AudioCaptureSession::applyAudioSettingsL()

{      

    if (!m_recorderUtility)

        return;

    TFourCC fourCC = m_audioCodeclist.value(m_format.codec()).fourCC;    

    //set destination datatype

    RArray<TFourCC> supportedDataTypes;

    CleanupClosePushL(supportedDataTypes);

    m_recorderUtility->GetSupportedDestinationDataTypesL(supportedDataTypes);    

    for (TInt k = 0; k < supportedDataTypes.Count(); k++ ) {        

        if (supportedDataTypes[k].FourCC() == fourCC.FourCC()) {            

            m_recorderUtility->SetDestinationDataTypeL(supportedDataTypes[k]);

            break;

        }

    }

    CleanupStack::PopAndDestroy(&supportedDataTypes);    

    RArray<TUint> supportedSampleRates;

    CleanupClosePushL(supportedSampleRates);

    m_recorderUtility->GetSupportedSampleRatesL(supportedSampleRates);    

    for (TInt i = 0; i < supportedSampleRates.Count(); i++ ) {

        TUint supportedSampleRate = supportedSampleRates[i];

        if (supportedSampleRate == m_format.frequency()) { 

            m_recorderUtility->SetDestinationSampleRateL(m_format.frequency());

            break;

        }        

    }

    CleanupStack::PopAndDestroy(&supportedSampleRates);

    RArray<TUint> supportedChannels;

    CleanupClosePushL(supportedChannels);

    m_recorderUtility->GetSupportedNumberOfChannelsL(supportedChannels);