/*  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/QVector>

#include <QtCore/QTimerEvent>

#include <QtCore/QTimer>

#include <QtCore/QTime>

#include <QtCore/QLibrary>

#ifndef Q_CC_MSVC

#include <dshow.h>

#endif

#include <objbase.h>

#include <initguid.h>

#include <qnetwork.h>

#include <evcode.h>

#include "mediaobject.h"

#include "videowidget.h"

#include "audiooutput.h"

#include <QtCore/QDebug>

#define TIMER_INTERVAL 16 //... ms for the timer that polls the current state (we use the multimedia timer)

#define PRELOAD_TIME 2000 // 2 seconds to load a source

QT_BEGIN_NAMESPACE

namespace Phonon

{

    namespace DS9

    {

        typedef BOOL (WINAPI* LPAMGETERRORTEXT)(HRESULT, WCHAR *, DWORD);

        //first the definition of the WorkerThread class

        WorkerThread::WorkerThread()

          : QThread(), m_finished(false), m_currentWorkId(1)

        {

        }

        WorkerThread::~WorkerThread()

        {

        }

        void WorkerThread::run()

        {

            while (m_finished == false) {

                HANDLE handles[FILTER_COUNT +1];

                handles[0] = m_waitCondition;

                int count = 1;

                for(int i = 0; i < FILTER_COUNT; ++i) {

                    if (m_graphHandle[i].graph) {

                        handles[count++] = m_graphHandle[i].handle;

                    }

                }

                DWORD result = ::WaitForMultipleObjects(count, handles, FALSE, INFINITE);

                if (result == WAIT_OBJECT_0) {

                    handleTask();

                } else {

                    //this is the event management

                    const Graph &graph = m_graphHandle[result - WAIT_OBJECT_0 - 1].graph;

                    long eventCode;

                    LONG_PTR param1, param2;

                    ComPointer<IMediaEvent> mediaEvent(graph, IID_IMediaEvent);

                    mediaEvent->GetEvent(&eventCode, &param1, &param2, 0);

                    emit eventReady(graph, eventCode, param1);

                    mediaEvent->FreeEventParams(eventCode, param1, param2);

                }

            }

        }

        //wants to know as soon as the state is set

        void WorkerThread::addStateChangeRequest(Graph graph, OAFilterState state, QList<Filter> decoders)

        {

            QMutexLocker locker(&m_mutex);

            bool found = false;

            //we try to see if there is already an attempt to change the state and we remove it

            for(int i = 0; !found && i < m_queue.size(); ++i) {

                const Work &w = m_queue.at(i);

                if (w.graph == graph && w.task == ChangeState) {

                    found = true;

                    m_queue.removeAt(i);

                }

            }

            //now let's create the new task

            Work w;

            w.task = ChangeState;

            w.id = m_currentWorkId++;

            w.graph = graph;

            w.state = state;

            w.decoders = decoders;

            m_queue.enqueue(w);

            m_waitCondition.set();

        }

        quint16 WorkerThread::addSeekRequest(Graph graph, qint64 time)

        {

            QMutexLocker locker(&m_mutex);

            bool found = false;

            //we try to see if there is already an attempt to seek and we remove it

            for(int i = 0; !found && i < m_queue.size(); ++i) {

                const Work &w = m_queue.at(i);

                if (w.graph == graph && w.task == Seek) {

                    found = true;

                    m_queue.removeAt(i);

                }

            }

            Work w;

            w.task = Seek;

            //we create a new graph

            w.graph = graph;

            w.id = m_currentWorkId++;

            w.time = time;

            m_queue.enqueue(w);

            m_waitCondition.set();

            return w.id;

        }

        quint16 WorkerThread::addUrlToRender(const QString &url)

        {

            QMutexLocker locker(&m_mutex);

            Work w;

            w.task = Render;

            //we create a new graph

            w.graph = Graph(CLSID_FilterGraph, IID_IGraphBuilder);

            w.url = url;

            w.url.detach();

            w.id = m_currentWorkId++;

            m_queue.enqueue(w);

            m_waitCondition.set();

            return w.id;

        }

        quint16 WorkerThread::addFilterToRender(const Filter &filter)

        {

            QMutexLocker locker(&m_mutex);

            Work w;

            w.task = Render;

            //we create a new graph

            w.graph = Graph(CLSID_FilterGraph, IID_IGraphBuilder);

            w.filter = filter;

            w.graph->AddFilter(filter, 0);

            w.id = m_currentWorkId++;

            m_queue.enqueue(w);

            m_waitCondition.set();

            return w.id;

        }

        void WorkerThread::replaceGraphForEventManagement(Graph newGraph, Graph oldGraph)

        {

            QMutexLocker locker(&m_mutex);

            Work w;

            w.task = ReplaceGraph;

            w.graph = newGraph;

            w.oldGraph = oldGraph;

            m_queue.enqueue(w);

            m_waitCondition.set();

        }

        void WorkerThread::handleTask()

        {

            QMutexLocker locker(Backend::directShowMutex);

            {

                QMutexLocker locker(&m_mutex);

                if (m_finished || m_queue.isEmpty()) {

                    return;

                }

                m_currentWork = m_queue.dequeue();

                //we ensure to have the wait condition in the right state

                if (m_queue.isEmpty()) {

                    m_waitCondition.reset();

                } else {

                    m_waitCondition.set();

                }

            }

            HRESULT hr = S_OK;

            if (m_currentWork.task == ReplaceGraph) {

                int index = -1;

                for(int i = 0; i < FILTER_COUNT; ++i) {

                    if (m_graphHandle[i].graph == m_currentWork.oldGraph) {

                        m_graphHandle[i].graph = Graph();

                        index = i;

                        break;

                    } else if (index == -1 && m_graphHandle[i].graph == 0) {

                        //this is the first available slot

                        index = i;

                    }

                }

                Q_ASSERT(index != -1);

                //add the new graph

                HANDLE h;

                if (SUCCEEDED(ComPointer<IMediaEvent>(m_currentWork.graph, IID_IMediaEvent)

                    ->GetEventHandle(reinterpret_cast<OAEVENT*>(&h)))) {

                    m_graphHandle[index].graph = m_currentWork.graph;

                    m_graphHandle[index].handle = h;

                }

            } else if (m_currentWork.task == Render) {

                if (m_currentWork.filter) {

                    //let's render pins

                    const QList<OutputPin> outputs = BackendNode::pins(m_currentWork.filter, PINDIR_OUTPUT);

                    for (int i = 0; SUCCEEDED(hr) && i < outputs.count(); ++i) {

                        hr = m_currentWork.graph->Render(outputs.at(i));

                    }

                } else if (!m_currentWork.url.isEmpty()) {

                    //let's render a url (blocking call)

                    hr = m_currentWork.graph->RenderFile(reinterpret_cast<const wchar_t *>(m_currentWork.url.utf16()), 0);

                }

                if (hr != E_ABORT) {

                    emit asyncRenderFinished(m_currentWork.id, hr, m_currentWork.graph);

                }

            } else if (m_currentWork.task == Seek) {

                //that's a seekrequest

                ComPointer<IMediaSeeking> mediaSeeking(m_currentWork.graph, IID_IMediaSeeking);

                qint64 newtime = m_currentWork.time * 10000;

                hr = mediaSeeking->SetPositions(&newtime, AM_SEEKING_AbsolutePositioning,

                    0, AM_SEEKING_NoPositioning);

                emit asyncSeekingFinished(m_currentWork.id, newtime / 10000);

                hr = E_ABORT; //to avoid emitting asyncRenderFinished

            } else if (m_currentWork.task == ChangeState) {

                //remove useless decoders

                QList<Filter> unused;

                for (int i = 0; i < m_currentWork.decoders.count(); ++i) {

                    const Filter &filter = m_currentWork.decoders.at(i);

                    bool used = false;

                    const QList<OutputPin> pins = BackendNode::pins(filter, PINDIR_OUTPUT);

                    for( int i = 0; i < pins.count(); ++i) {

                        InputPin input;

                        if (pins.at(i)->ConnectedTo(input.pparam()) == S_OK) {

                            used = true;

                        }

                    }

                    if (!used) {

                        unused += filter;

                    }

                }

                //we can get the state

                for (int i = 0; i < unused.count(); ++i) {

                    //we should remove this filter from the graph

                    m_currentWork.graph->RemoveFilter(unused.at(i));

                }

                //we can get the state

                ComPointer<IMediaControl> mc(m_currentWork.graph, IID_IMediaControl);

                //we change the state here

                switch(m_currentWork.state)

                {

                case State_Stopped:

                    mc->Stop();

                    break;

                case State_Paused:

                    mc->Pause();

                    break;

                case State_Running:

                    mc->Run();

                    break;

                }

                OAFilterState s;

                //blocking call

                HRESULT hr = mc->GetState(INFINITE, &s);

                if (SUCCEEDED(hr)) {

                    if (s == State_Stopped) {

                        emit stateReady(m_currentWork.graph, Phonon::StoppedState);

                    } else if (s == State_Paused) {

                        emit stateReady(m_currentWork.graph, Phonon::PausedState);

                    } else /*if (s == State_Running)*/ {

                        emit stateReady(m_currentWork.graph, Phonon::PlayingState);

                    }

                }

            }

            {

                QMutexLocker locker(&m_mutex);

                m_currentWork = Work(); //reinitialize

            }

        }

        void WorkerThread::abortCurrentRender(qint16 renderId)

        {

            QMutexLocker locker(&m_mutex);

            if (m_currentWork.id == renderId) {

                m_currentWork.graph->Abort();

            }

            bool found = false;

            for(int i = 0; !found && i < m_queue.size(); ++i) {

                const Work &w = m_queue.at(i);

                if (w.id == renderId) {

                    found = true;

                    m_queue.removeAt(i);

                    if (m_queue.isEmpty()) {

                        m_waitCondition.reset();

                    }

                }

            }

        }

        //tells the thread to stop processing

        void WorkerThread::signalStop()

        {

            QMutexLocker locker(&m_mutex);

            m_queue.clear();

            if (m_currentWork.graph) {

                //in case we're currently rendering something

                m_currentWork.graph->Abort();

            }

            m_finished = true;

            m_waitCondition.set();

        }

        MediaObject::MediaObject(QObject *parent) : BackendNode(parent),

            transactionState(Phonon::StoppedState),

            m_errorType(Phonon::NoError),

            m_state(Phonon::LoadingState),

            m_nextState(Phonon::StoppedState),

            m_prefinishMark(0),

            m_tickInterval(0),

            m_buffering(false),

            m_oldHasVideo(false),

            m_prefinishMarkSent(false),

            m_aboutToFinishSent(false),

            m_nextSourceReadyToStart(false),

#ifndef QT_NO_PHONON_MEDIACONTROLLER

            m_autoplayTitles(true),

            m_currentTitle(0),

#endif //QT_NO_PHONON_MEDIACONTROLLER

            m_targetTick(INFINITE)

        {

            for(int i = 0; i < FILTER_COUNT; ++i) {

                m_graphs[i] = new MediaGraph(this, i);

            }

            connect(&m_thread, SIGNAL(stateReady(Graph,Phonon::State)),

                               SLOT(slotStateReady(Graph,Phonon::State)));

            connect(&m_thread, SIGNAL(eventReady(Graph,long,long)),

                               SLOT(handleEvents(Graph,long,long)));

            connect(&m_thread, SIGNAL(asyncRenderFinished(quint16,HRESULT,Graph)),

                SLOT(finishLoading(quint16,HRESULT,Graph)));

            connect(&m_thread, SIGNAL(asyncSeekingFinished(quint16,qint64)),

                SLOT(finishSeeking(quint16,qint64)));

            //really special case

            m_mediaObject = this;

            m_thread.start();

        }

        MediaObject::~MediaObject()

        {

            //be sure to finish the timer first

            m_tickTimer.stop();

            //we finish the worker thread here

            m_thread.signalStop();

            m_thread.wait();

            //and then we delete the graphs

            for (int i = 0; i < FILTER_COUNT; ++i) {

                delete m_graphs[i];

            }

        }

        WorkerThread *MediaObject::workerThread()

        {

            return &m_thread;

        }

        MediaGraph *MediaObject::currentGraph() const

        {

            return m_graphs[0];

        }

        MediaGraph *MediaObject::nextGraph() const

        {

            return m_graphs[FILTER_COUNT - 1];

        }

        //utility function to save the graph to a file

        void MediaObject::timerEvent(QTimerEvent *e)

        {

            if (e->timerId() == m_tickTimer.timerId()) {

                const qint64 current = currentTime();

                const qint64 total = totalTime();

                if ( m_tickInterval != 0 && current > m_targetTick) {

                    updateTargetTick();

                    emit tick(current);

                }

                //check that the title hasn't changed

#ifndef QT_NO_PHONON_MEDIACONTROLLER

                if (m_autoplayTitles && m_currentTitle < _iface_availableTitles() - 1) {

                    if (current >= total) {

                        //we go to the next title

                        _iface_setCurrentTitle(m_currentTitle + 1, false);

                        emit tick(current);

                    }

                    return;

                }

#endif //QT_NO_PHONON_MEDIACONTROLLER

                if (total) {

                    const qint64 remaining = total - current;

                    if (m_transitionTime < 0 && m_nextSourceReadyToStart) {

                        if (remaining < -m_transitionTime + TIMER_INTERVAL/2) {

                            //we need to switch graphs to run the next source in the queue (with cross-fading)

                            switchToNextSource();

                            return;

                        } else if (current < -m_transitionTime) {

                            //we are currently crossfading

                            for (int i = 0; i < m_audioOutputs.count(); ++i) {

                                m_audioOutputs.at(i)->setCrossFadingProgress( currentGraph()->index(), qMin( qreal(1.), qreal(current) / qreal(-m_transitionTime)));

                            }

                        }

                    }

                    if (m_prefinishMark > 0 && !m_prefinishMarkSent && remaining < m_prefinishMark + TIMER_INTERVAL/2) {

#ifdef GRAPH_DEBUG

                        qDebug() << "DS9: emit prefinishMarkReached" << remaining << QTime::currentTime().toString();

#endif

                        m_prefinishMarkSent = true;

                        emit prefinishMarkReached( remaining );

                    }

                    if (!m_aboutToFinishSent && remaining < PRELOAD_TIME - m_transitionTime  + TIMER_INTERVAL/2) {

                        //let's take a 2 seconds time to actually load the next file

#ifdef GRAPH_DEBUG

                        qDebug() << "DS9: emit aboutToFinish" << remaining << QTime::currentTime().toString();

#endif

                        m_aboutToFinishSent = true;

                        emit aboutToFinish();

                    }

                } else {

                    //total is 0: the stream is probably live (endless)

                }

                if (m_buffering) {

                    ComPointer<IAMNetworkStatus> status(currentGraph()->realSource(), IID_IAMNetworkStatus);

                    if (status) {

                        long l;

                        status->get_BufferingProgress(&l);

                        emit bufferStatus(l);

#ifdef GRAPH_DEBUG

                        qDebug() << "emit bufferStatus(" << l << ")";

#endif

                    }

                }

            }

        }

        void MediaObject::switchToNextSource()

        {

            m_prefinishMarkSent = false;

            m_aboutToFinishSent = false;

            m_nextSourceReadyToStart = false;

            m_oldHasVideo = currentGraph()->hasVideo();

            qSwap(m_graphs[0], m_graphs[1]); //swap the graphs

            if (m_transitionTime >= 0)

                m_graphs[1]->stop(); //make sure we stop the previous graph

            if (currentGraph()->mediaSource().type() != Phonon::MediaSource::Invalid &&

                catchComError(currentGraph()->renderResult())) {

                    setState(Phonon::ErrorState);

                    return;

            }

            //we need to play the next media

            play();

            //we tell the video widgets to switch now to the new source

#ifndef QT_NO_PHONON_VIDEO

            for (int i = 0; i < m_videoWidgets.count(); ++i) {

                m_videoWidgets.at(i)->setCurrentGraph(currentGraph()->index());

            }

#endif //QT_NO_PHONON_VIDEO

            emit currentSourceChanged(currentGraph()->mediaSource());

            emit metaDataChanged(currentGraph()->metadata());

            if (nextGraph()->hasVideo() != currentGraph()->hasVideo()) {

                emit hasVideoChanged(currentGraph()->hasVideo());

            }

            emit tick(0);

            emit totalTimeChanged(totalTime());

#ifndef QT_NO_PHONON_MEDIACONTROLLER

            setTitles(currentGraph()->titles());