ImageClipWorkflow.cpp 5.34 KB
Newer Older
1 2 3
/*****************************************************************************
 * ImageClipWorkflow.cpp : Will extract a frame from an image
 *****************************************************************************
Ludovic Fauvet's avatar
Ludovic Fauvet committed
4
 * Copyright (C) 2008-2010 VideoLAN
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 *
 * Authors: Hugo Beauzee-Luyssen <hugo@vlmc.org>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
 *****************************************************************************/

#include "ImageClipWorkflow.h"
24
#include "Clip.h"
25
#include "ClipHelper.h"
26
#include "MainWorkflow.h"
27
#include "Media.h"
28
#include "VLCMediaPlayer.h"
29
#include "VLCMedia.h"
30
#include "Workflow/Types.h"
31

32 33 34
#include <QReadWriteLock>
#include <QtDebug>

35 36
ImageClipWorkflow::ImageClipWorkflow( ClipHelper *ch ) :
        ClipWorkflow( ch ),
37
        m_buffer( NULL )
38
{
39 40 41 42
    //This is used to queue the media player stopping, as it can't be asked for
    //from vlc's input thread (well it can but it will deadlock)
    connect( this, SIGNAL( computedFinished() ),
             this, SLOT( stopComputation() ), Qt::QueuedConnection );
43 44
    m_effectsLock = new QReadWriteLock;
    m_effectFrame = new Workflow::Frame;
45 46
}

47 48 49
ImageClipWorkflow::~ImageClipWorkflow()
{
    stop();
50 51 52
    qDeleteAll( m_filters );
    delete m_effectFrame;
    delete m_effectsLock;
53 54
}

55 56
void
ImageClipWorkflow::initVlcOutput()
57 58 59
{
    char    buffer[32];

60 61 62 63 64 65
    m_vlcMedia->addOption( ":no-audio" );
    m_vlcMedia->addOption( ":no-sout-audio" );
    m_vlcMedia->addOption( ":sout=#transcode{}:smem" );
    m_vlcMedia->setVideoDataCtx( this );
    m_vlcMedia->setVideoLockCallback( reinterpret_cast<void*>( getLockCallback() ) );
    m_vlcMedia->setVideoUnlockCallback( reinterpret_cast<void*>( getUnlockCallback() ) );
66
    m_vlcMedia->addOption( ":sout-transcode-vcodec=RV32" );
67 68 69 70 71 72 73 74 75 76
    m_vlcMedia->addOption( ":sout-smem-time-sync" );

    sprintf( buffer, ":sout-transcode-width=%i",
             MainWorkflow::getInstance()->getWidth() );
    m_vlcMedia->addOption( buffer );
    sprintf( buffer, ":sout-transcode-height=%i",
             MainWorkflow::getInstance()->getHeight() );
    m_vlcMedia->addOption( buffer );
    sprintf( buffer, ":sout-transcode-fps=%f", (float)Clip::DefaultFPS );
    m_vlcMedia->addOption( buffer );
77
    sprintf( buffer, ":fake-duration=%d", 1000 );
78
    m_vlcMedia->addOption( buffer );
79
    sprintf( buffer, ":fake-fps=%f", m_clipHelper->clip()->getMedia()->fps() );
80
    m_vlcMedia->addOption( buffer );
81 82 83

    m_effectFrame->resize( MainWorkflow::getInstance()->getWidth(),
                            MainWorkflow::getInstance()->getHeight() );
84
}
85 86

void*
Hugo Beauzee-Luyssen's avatar
Hugo Beauzee-Luyssen committed
87
ImageClipWorkflow::getLockCallback() const
88 89 90 91 92
{
    return reinterpret_cast<void*>( &ImageClipWorkflow::lock );
}

void*
Hugo Beauzee-Luyssen's avatar
Hugo Beauzee-Luyssen committed
93
ImageClipWorkflow::getUnlockCallback() const
94 95 96 97
{
    return reinterpret_cast<void*>( &ImageClipWorkflow::unlock );
}

98
Workflow::OutputBuffer*
99 100 101 102
ImageClipWorkflow::getOutput( ClipWorkflow::GetMode )
{
    QMutexLocker    lock( m_renderLock );

103
    QReadLocker     lock2( m_effectsLock );
104 105 106
    qint64          currentFrame = MainWorkflow::getInstance()->getCurrentFrame( false );
    quint32 *buff = EffectsEngine::applyFilters( m_filters, m_buffer, currentFrame,
                                                currentFrame * 1000.0 / clip()->getMedia()->fps() );
107 108 109 110 111
    if ( buff != NULL )
    {
        m_effectFrame->setBuffer( buff );
        return m_effectFrame;
    }
112
    return m_buffer;
113 114 115
}

void
116
ImageClipWorkflow::lock(ImageClipWorkflow *cw, void **pp_ret, int )
117 118 119 120
{
    cw->m_renderLock->lock();
    if ( cw->m_buffer == NULL )
    {
121
        cw->m_buffer = new Workflow::Frame( MainWorkflow::getInstance()->getWidth(),
122
                                            MainWorkflow::getInstance()->getHeight() );
123
    }
124
    *pp_ret = cw->m_buffer->buffer();
125 126 127
}

void
128
ImageClipWorkflow::unlock(ImageClipWorkflow *cw, void*, int, int, int, int, qint64 )
129 130 131 132 133
{
    cw->m_renderLock->unlock();
    cw->emit computedFinished();
}

134
quint32
135 136 137 138 139 140 141 142 143
ImageClipWorkflow::getNbComputedBuffers() const
{
    QMutexLocker    lock( m_renderLock );

    if ( m_buffer != NULL )
        return 1;
    return 0;
}

144
quint32
145 146 147 148 149 150 151 152 153 154 155
ImageClipWorkflow::getMaxComputedBuffers() const
{
    return 1;
}

void
ImageClipWorkflow::stopComputation()
{
    m_mediaPlayer->stop();
}

156 157 158 159
void
ImageClipWorkflow::flushComputedBuffers()
{
}
160 161 162 163 164 165 166 167 168 169 170 171 172 173 174

EffectsEngine::EffectHelper*
ImageClipWorkflow::appendEffect( Effect *effect, qint64 start, qint64 end )
{
    if ( effect->type() != Effect::Filter )
    {
        qWarning() << "VideoClipWorkflow does not handle non filter effects.";
        return NULL;
    }
    EffectInstance      *effectInstance = effect->createInstance();
    QWriteLocker        lock( m_effectsLock );
    EffectsEngine::EffectHelper *ret = new EffectsEngine::EffectHelper( effectInstance, start, end );
    m_filters.push_back( ret );
    return ret;
}