/*  This file is part of the KDE project.

    Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).

    This library is free software: you can redistribute it and/or modify

    it under the terms of the GNU Lesser General Public License as published by

    the Free Software Foundation, either version 2.1 or 3 of the License.

    This library is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License

    along with this library.  If not, see <http://www.gnu.org/licenses/>.

*/

#include <QtCore/QEvent>

#include "mediaobject.h"

#include "backendheader.h"

#include "videowidget.h"

#include "videoframe.h"

#include "audiooutput.h"

#include "quicktimevideoplayer.h"

#include "quicktimemetadata.h"

#include "audiograph.h"

#include "mediaobjectaudionode.h"

#include "quicktimeaudioplayer.h"

QT_BEGIN_NAMESPACE

namespace Phonon

{

namespace QT7

{

MediaObject::MediaObject(QObject *parent) : MediaNode(AudioSource | VideoSource, parent)

{

    m_owningMediaObject = this;

    m_state = Phonon::LoadingState;

    m_videoPlayer = new QuickTimeVideoPlayer();

    m_audioPlayer = new QuickTimeAudioPlayer();

    m_nextVideoPlayer = new QuickTimeVideoPlayer();

    m_nextAudioPlayer = new QuickTimeAudioPlayer();

    m_mediaObjectAudioNode = new MediaObjectAudioNode(m_audioPlayer, m_nextAudioPlayer);

    setAudioNode(m_mediaObjectAudioNode);

    m_audioGraph = new AudioGraph(this);

    m_tickInterval = 0;

    m_prefinishMark = 0;

    m_currentTime = 0;

    m_transitionTime = 0;

    m_percentageLoaded = 0;

    m_waitNextSwap = false;

    m_autoplayTitles = true;

    m_audioEffectCount = 0;

    m_audioOutputCount = 0;

    m_videoEffectCount = 0;

    m_videoOutputCount = 0;

    m_audioSystem = AS_Unset;

    m_errorType = Phonon::NoError;

    m_tickTimer = 0;

    m_videoTimer = 0;

    m_audioTimer = 0;

    m_rapidTimer = 0;

#if QT_ALLOW_QUICKTIME

    m_displayLink = 0;

    m_pendingDisplayLinkEvent = false;

#endif

    checkForError();

}

MediaObject::~MediaObject()

{

    // m_mediaObjectAudioNode is owned by super class.

#if QT_ALLOW_QUICKTIME

    stopDisplayLink();

#endif

    m_audioPlayer->unsetVideoPlayer();

    m_nextAudioPlayer->unsetVideoPlayer();

    delete m_videoPlayer;

    delete m_nextVideoPlayer;

    checkForError();

}

bool MediaObject::setState(Phonon::State state)

{

    Phonon::State prevState = m_state;

    m_state = state;

    if (prevState != m_state){

        emit stateChanged(m_state, prevState);

        if (m_state != state){

            // End-application did something

            // upon  receiving the signal.

            return false;

        }

    }

    return true;

}

void MediaObject::inspectAudioGraphRecursive(AudioConnection *connection, int &effectCount, int &outputCount)

{

    if ((connection->m_sink->m_description & (AudioSource | AudioSink)) == (AudioSource | AudioSink))

        ++effectCount;

	else if (connection->m_sink->m_description & AudioSink)

    	++outputCount;

    for (int i=0; i<connection->m_sink->m_audioSinkList.size(); ++i)

        inspectAudioGraphRecursive(connection->m_sink->m_audioSinkList[i], effectCount, outputCount);

}

void MediaObject::inspectVideoGraphRecursive(MediaNode *node, int &effectCount, int &outputCount)

{

    if ((node->m_description & (VideoSource | VideoSink)) == (VideoSource | VideoSink))

        ++effectCount;

	else if (node->m_description & VideoSink)

    	++outputCount;

    for (int i=0; i<node->m_videoSinkList.size(); ++i)

        inspectVideoGraphRecursive(node->m_videoSinkList[i], effectCount, outputCount);

}

void MediaObject::inspectGraph()

{

    // Inspect the graph to check wether there are any

    // effects or outputs connected. This will have

    // influence on the audio system and video system that ends up beeing used:

    int prevVideoOutputCount = m_videoOutputCount;

    m_audioEffectCount = 0;

    m_audioOutputCount = 0;

    m_videoEffectCount = 0;

    m_videoOutputCount = 0;

    AudioConnection rootConnection(this);

    inspectAudioGraphRecursive(&rootConnection, m_audioEffectCount, m_audioOutputCount);

    inspectVideoGraphRecursive(this, m_videoEffectCount, m_videoOutputCount);

	if (m_videoOutputCount != prevVideoOutputCount){

	    MediaNodeEvent e1(MediaNodeEvent::VideoOutputCountChanged, &m_videoOutputCount);

	    notify(&e1);

    }

}

void MediaObject::setupAudioSystem()

{

    // Select which audio system to use:

    AudioSystem newAudioSystem = AS_Unset;

    if (!m_audioOutputCount || !m_videoPlayer->canPlayMedia()){

        newAudioSystem = AS_Silent;

    } else if (m_audioEffectCount == 0){

        newAudioSystem = AS_Video;

    } else if (QSysInfo::MacintoshVersion < QSysInfo::MV_10_4){

        newAudioSystem = AS_Video;

        SET_ERROR("Audio effects are not supported for Mac OS 10.3 and below", NORMAL_ERROR);

    } else if (m_videoPlayer->isDrmProtected()){

        newAudioSystem = AS_Video;

        SET_ERROR("Audio effects are not supported for DRM protected media", NORMAL_ERROR);

    } else if (m_audioGraph->graphCannotPlay()){

        newAudioSystem = AS_Video;

        SET_ERROR("Audio effects are not supported for the current codec", NORMAL_ERROR);

#ifdef QUICKTIME_C_API_AVAILABLE

    } else {

        newAudioSystem = AS_Graph;

    }

#else

    } else {

        newAudioSystem = AS_Video;

        SET_ERROR("Audio effects are not supported for the 64-bit version of the Phonon QT7 backend", NORMAL_ERROR);

    }

#endif

    if (newAudioSystem == m_audioSystem)

        return;

    // Enable selected audio system:

    m_audioSystem = newAudioSystem;

    switch (newAudioSystem){

        case AS_Silent:

            m_audioGraph->stop();

            m_videoPlayer->enableAudio(false);

            m_nextVideoPlayer->enableAudio(false);

            m_audioPlayer->enableAudio(false);

            m_nextAudioPlayer->enableAudio(false);

        break;

        case AS_Graph:

            if (m_state == Phonon::PausedState)

                m_audioGraph->prepare();

            else

                m_audioGraph->start();

            // Starting the graph can lead to a recursive call

            // telling us that we must direct audio through

            // video. If that has happened, we must not proceed:

            if (m_audioSystem != AS_Graph)

                return;

            m_videoPlayer->enableAudio(false);

            m_nextVideoPlayer->enableAudio(false);

            m_audioPlayer->enableAudio(true);

            m_audioPlayer->seek(m_videoPlayer->currentTime());

            m_nextAudioPlayer->enableAudio(true);

            m_audioPlayer->seek(m_videoPlayer->currentTime());

            m_nextAudioPlayer->seek(m_nextVideoPlayer->currentTime());

        break;

        case AS_Video:

        case AS_Unset:

            m_audioGraph->stop();

            m_videoPlayer->enableAudio(true);

            m_nextVideoPlayer->enableAudio(true);

            m_audioPlayer->enableAudio(false);

            m_nextAudioPlayer->enableAudio(false);

            m_videoPlayer->seek(m_audioPlayer->currentTime());

            m_nextVideoPlayer->seek(m_nextAudioPlayer->currentTime());

        break;

    }

}

void MediaObject::setSource(const MediaSource &source)

{

    IMPLEMENTED;

	PhononAutoReleasePool pool;

    setState(Phonon::LoadingState);

    // Save current state for event/signal handling below:

    bool prevHasVideo = m_videoPlayer->hasVideo();

    qint64 prevTotalTime = totalTime();

    int prevTrackCount = m_videoPlayer->trackCount();

    m_waitNextSwap = false;

    // Cancel cross-fade if any:

    m_nextVideoPlayer->pause();

    m_nextAudioPlayer->pause();

    m_mediaObjectAudioNode->cancelCrossFade();

    // Set new source:

    m_audioPlayer->unsetVideoPlayer();

    m_videoPlayer->setMediaSource(source);

    m_audioPlayer->setVideoPlayer(m_videoPlayer);

    m_audioGraph->updateStreamSpecifications();

    m_nextAudioPlayer->unsetVideoPlayer();

    m_nextVideoPlayer->unsetCurrentMediaSource();

    m_currentTime = 0;

    // Emit/notify information about the new source:

    QRect videoRect = m_videoPlayer->videoRect();

    MediaNodeEvent e1(MediaNodeEvent::VideoFrameSizeChanged, &videoRect);

    notify(&e1);

    // Clear video widgets:

    VideoFrame emptyFrame;

    updateVideo(emptyFrame);

    emit currentSourceChanged(source);

    emit metaDataChanged(m_videoPlayer->metaData());

    if (prevHasVideo != m_videoPlayer->hasVideo())

        emit hasVideoChanged(m_videoPlayer->hasVideo());

    if (prevTotalTime != totalTime())

        emit totalTimeChanged(totalTime());

    if (prevTrackCount != m_videoPlayer->trackCount())

        emit availableTitlesChanged(m_videoPlayer->trackCount());

    if (checkForError())

        return;

    if (!m_videoPlayer->isDrmAuthorized())

        SET_ERROR("This computer is not authorized to play current media (DRM protected).", FATAL_ERROR)

    if (checkForError())

        return;

    if (!m_videoPlayer->canPlayMedia())

        SET_ERROR("Cannot play media.", FATAL_ERROR)

    // The state might have changed from LoadingState

    // as a response to an error state change. So we

    // need to check it before stopping: 

    if (m_state == Phonon::LoadingState)

        stop();

    setupAudioSystem();

    checkForError();

}

void MediaObject::setNextSource(const MediaSource &source)

{

    IMPLEMENTED;

    m_nextAudioPlayer->unsetVideoPlayer();

    m_nextVideoPlayer->setMediaSource(source);

    m_nextAudioPlayer->setVideoPlayer(m_nextVideoPlayer);

    checkForError();

}

void MediaObject::swapCurrentWithNext(qint32 transitionTime)

{

	PhononAutoReleasePool pool;

    setState(Phonon::LoadingState);

    // Save current state for event/signal handling below:

    bool prevHasVideo = m_videoPlayer->hasVideo();

    qint64 prevTotalTime = totalTime();

    int prevTrackCount = m_videoPlayer->trackCount();

    qSwap(m_audioPlayer, m_nextAudioPlayer);

    qSwap(m_videoPlayer, m_nextVideoPlayer);

    m_mediaObjectAudioNode->startCrossFade(transitionTime);

    m_audioGraph->updateStreamSpecifications();

    m_waitNextSwap = false;

    m_currentTime = 0;

    // Emit/notify information about the new source:

    QRect videoRect = m_videoPlayer->videoRect();

    MediaNodeEvent e1(MediaNodeEvent::VideoFrameSizeChanged, &videoRect);

    notify(&e1);

    emit currentSourceChanged(m_videoPlayer->mediaSource());

    emit metaDataChanged(m_videoPlayer->metaData());

    if (prevHasVideo != m_videoPlayer->hasVideo())

        emit hasVideoChanged(m_videoPlayer->hasVideo());

    if (prevTotalTime != totalTime())

        emit totalTimeChanged(totalTime());

    if (prevTrackCount != m_videoPlayer->trackCount())

        emit availableTitlesChanged(m_videoPlayer->trackCount());

    if (checkForError())

        return;

    if (!m_videoPlayer->isDrmAuthorized())

        SET_ERROR("This computer is not authorized to play current media (DRM protected).", FATAL_ERROR)

    if (checkForError())

        return;

    if (!m_videoPlayer->canPlayMedia())

        SET_ERROR("Cannot play next media.", FATAL_ERROR)

    setupAudioSystem();

    checkForError();

    if (m_state == Phonon::LoadingState){

        if (setState(Phonon::PlayingState))

            play_internal();

        checkForError();

    }

}

#if QT_ALLOW_QUICKTIME

static CVReturn displayLinkCallback(CVDisplayLinkRef /*displayLink*/,

								 const CVTimeStamp */*inNow*/,

								 const CVTimeStamp */*inOutputTime*/,

								 CVOptionFlags /*flagsIn*/,

								 CVOptionFlags */*flagsOut*/,

                                 void *userData)

{

    MediaObject *mediaObject = static_cast<MediaObject *>(userData);

    mediaObject->displayLinkEvent();

    return kCVReturnSuccess;

}

void MediaObject::displayLinkEvent()

{

    // This function is called from a

    // thread != gui thread. So we post the event.

    // But we need to make sure that we don't post faster

    // than the event loop can eat:

    m_displayLinkMutex.lock();

    bool pending = m_pendingDisplayLinkEvent;

    m_pendingDisplayLinkEvent = true;

    m_displayLinkMutex.unlock();

    if (!pending)

        qApp->postEvent(this, new QEvent(QEvent::User), Qt::HighEventPriority);

}

void MediaObject::startDisplayLink()

{

    if (m_displayLink)

        return;

    OSStatus err = CVDisplayLinkCreateWithCGDisplay(kCGDirectMainDisplay, &m_displayLink);

    if (err != noErr)

        goto fail;

    err = CVDisplayLinkSetCurrentCGDisplay(m_displayLink, kCGDirectMainDisplay);

    if (err != noErr)

        goto fail;

    err = CVDisplayLinkSetOutputCallback(m_displayLink, displayLinkCallback, this);

    if (err != noErr)

        goto fail;

    err = CVDisplayLinkStart(m_displayLink);

    if (err != noErr)

        goto fail;

    return;

fail:

    stopDisplayLink();

}

void MediaObject::stopDisplayLink()

{

    if (!m_displayLink)

        return;

    CVDisplayLinkStop(m_displayLink);

    CFRelease(m_displayLink);

    m_displayLink = 0;

}

#endif

void MediaObject::restartAudioVideoTimers()

{

    if (m_videoTimer)

        killTimer(m_videoTimer);

    if (m_audioTimer)

        killTimer(m_audioTimer);

#if QT_ALLOW_QUICKTIME

    // We prefer to use a display link as timer if available, since

    // it is more steady, and results in better and smoother frame drawing:

    startDisplayLink();

    if (!m_displayLink){

        float fps = m_videoPlayer->staticFps();

        long videoUpdateFrequency = fps ? long(1000.0f / fps) : 0.001;

        m_videoTimer = startTimer(videoUpdateFrequency);

    }

#else

    float fps = m_videoPlayer->staticFps();

    long videoUpdateFrequency = fps ? long(1000.0f / fps) : 0.001;

    m_videoTimer = startTimer(videoUpdateFrequency);

#endif

    long audioUpdateFrequency = m_audioPlayer->regularTaskFrequency();

    m_audioTimer = startTimer(audioUpdateFrequency);

    updateVideoFrames();

    updateAudioBuffers();

}

void MediaObject::play_internal()

{

    // Play main audio/video:

    m_videoPlayer->play();

    m_audioPlayer->play();

    updateLipSynch(0);

    // Play old audio/video to finish cross-fade:

    if (m_nextVideoPlayer->currentTime() > 0){

        m_nextVideoPlayer->play();

        m_nextAudioPlayer->play();

    }

    restartAudioVideoTimers();

    if (!m_rapidTimer)

        m_rapidTimer = startTimer(100);

}

void MediaObject::pause_internal()

{

    m_audioGraph->stop();

    m_audioPlayer->pause();

    m_nextAudioPlayer->pause();

    m_videoPlayer->pause();

    m_nextVideoPlayer->pause();

    killTimer(m_rapidTimer);

    killTimer(m_videoTimer);

    killTimer(m_audioTimer);

    m_rapidTimer = 0;

    m_videoTimer = 0;

    m_audioTimer = 0;

#if QT_ALLOW_QUICKTIME

    stopDisplayLink();

#endif

    if (m_waitNextSwap)

        m_swapTimeLeft = m_swapTime.msecsTo(QTime::currentTime());

}

void MediaObject::play()

{

    IMPLEMENTED;

    if (m_state == Phonon::PlayingState)

        return;

    if (m_waitNextSwap){

        // update swap time after pause:

        m_swapTime = QTime::currentTime();

        m_swapTime.addMSecs(m_swapTimeLeft);

        setState(Phonon::PlayingState);

        return;

    }

    if (m_currentTime == m_videoPlayer->duration())

        return;

    if (!m_videoPlayer->canPlayMedia())

        return;

    if (!setState(Phonon::PlayingState))

        return;

    if (m_audioSystem == AS_Graph){

        m_audioGraph->start();

        m_mediaObjectAudioNode->setMute(true);

    }

	// Inform the graph that we are about to play:

	bool playing = true;

    MediaNodeEvent e1(MediaNodeEvent::MediaPlaying, &playing);

    notify(&e1);

	// Start to play:

    play_internal();

    m_mediaObjectAudioNode->setMute(false);

    checkForError();

}

void MediaObject::pause()

{

    IMPLEMENTED;

    if (m_state == Phonon::PausedState)

        return;

    if (!setState(Phonon::PausedState))

        return;

    pause_internal();

	// Inform the graph that we are no longer playing:

	bool playing = false;

    MediaNodeEvent e1(MediaNodeEvent::MediaPlaying, &playing);

    notify(&e1);

	// But be prepared:

    if (m_audioSystem == AS_Graph)

        m_audioGraph->prepare();

    checkForError();

}

void MediaObject::stop()

{

    IMPLEMENTED;

    if (m_state == Phonon::StoppedState)

        return;

    if (!setState(Phonon::StoppedState))

        return;

    m_waitNextSwap = false;

    m_nextVideoPlayer->unsetCurrentMediaSource();

    m_nextAudioPlayer->unsetVideoPlayer();

    pause_internal();

    seek(0);

    checkForError();

}