X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=thumbnailer%2Fwestleypreview.cpp;h=b3212da8edac1e366cdc37b493349ab362dafb7a;hb=1404e8f1d631c81399e0e92ffce7dfac0195fabb;hp=af1a58c0d3dbc6924370fbaf330d6eaae66aff5b;hpb=a97b0c01710b048ff8e8640378457f410f77321c;p=kdenlive

diff --git a/thumbnailer/westleypreview.cpp b/thumbnailer/westleypreview.cpp
index af1a58c0..b3212da8 100644
--- a/thumbnailer/westleypreview.cpp
+++ b/thumbnailer/westleypreview.cpp
@@ -18,185 +18,145 @@
     Boston, MA 02110-1301, USA.
  ***************************************************************************/
 
+#include "westleypreview.h"
+
 #include <qfile.h>
-#include <qpixmap.h>
 #include <qimage.h>
 #include <QtCore/QVarLengthArray>
 
 #include <kstandarddirs.h>
-#include <kmimetype.h>
-#include <QDir>
-#include <qpainter.h>
 #include <krandomsequence.h>
 #include <qdatetime.h>
-#include <qregexp.h>
-#include <QProcess>
+#include <QColor>
 #include <kdebug.h>
 #include <ktempdir.h>
 #include <kurl.h>
-#include <math.h>
 #include <qfileinfo.h>
-#include <kcodecs.h>
 #include <KTemporaryFile>
 
 #include <unistd.h>
 
-#include "westleypreview.h"
 
 #define DBG_AREA
 
 //#include "config.h"
-extern "C"
-{
-    KDE_EXPORT ThumbCreator *new_creator()
-    {
-        return new WestleyPreview;
-    }
+extern "C" {
+KDE_EXPORT ThumbCreator *new_creator() {
+    return new MltPreview;
+}
 }
 
-WestleyPreview::WestleyPreview()
+MltPreview::MltPreview() :
+    QObject(),
+    ThumbCreator()
 {
+    Mlt::Factory::init();
 }
 
-WestleyPreview::~WestleyPreview()
+MltPreview::~MltPreview()
 {
-    delete rand;
-    delete inigoprocess;
+    Mlt::Factory::close();
 }
 
-bool WestleyPreview::startAndWaitProcess(const QStringList &args) {
-    kDebug(DBG_AREA) << "westleypreview: starting process with args: " << args << endl;
-    inigoprocess->start( args.join(" ") );
-    if(! inigoprocess->waitForStarted() ) {
-        kDebug(DBG_AREA) << "westleypreview: PROCESS NOT STARTED!!! exiting\n";
+
+bool MltPreview::create(const QString &path, int width, int height, QImage &img)
+{
+    Mlt::Profile *profile = new Mlt::Profile("dv_pal");
+    Mlt::Producer *producer = new Mlt::Producer(*profile, path.toUtf8().data());
+
+
+    if (producer->is_blank()) {
+        delete producer;
+        delete profile;
         return false;
     }
-    if(! inigoprocess->waitForFinished() ) {
-        kDebug(DBG_AREA) << "westleypreview: PROCESS DIDN'T FINISH!! exiting\n";
-        inigoprocess->close();
-        return false;
+    int frame = 75;
+    uint variance = 10;
+    int ct = 1;
+
+    //img = getFrame(producer, frame, width, height);
+
+    while (variance <= 40 && ct < 4) {
+        img = getFrame(producer, frame, width, height);
+        variance = imageVariance(img);
+        frame += 100 * ct;
+        ct++;
     }
-  kDebug() << "westleypreview: process started and ended correctly\n";
-  return true;
+
+    delete producer;
+    delete profile;
+    return (img.isNull() == false);
 }
 
-bool WestleyPreview::create(const QString &path, int width, int height, QImage &img)
+QImage MltPreview::getFrame(Mlt::Producer *producer, int framepos, int /*width*/, int height)
 {
-    QFileInfo fi(path);
-    /*if (fi.suffix().trimmed() != "westley" && fi.suffix().trimmed() != "kdenlive")
-    {
-        kDebug(DBG_AREA) << "westleypreview: matched extension " << fi.suffix().prepend('.') << "; exiting.\n";
-        return false;
-    }*/
-    playerBin=KStandardDirs::findExe("inigo");
-    if(playerBin.isEmpty())
-    {
-	kDebug(DBG_AREA) << "westleypreview: inigo not found, exiting.\n";
-        return false;
+    QImage result;
+    if (producer == NULL) {
+        return result;
     }
 
-    fileinfo.seconds=0;
-    fileinfo.fps=0;
-
-    rand=new KRandomSequence(QDateTime::currentDateTime().toTime_t());
-    inigoprocess=new QProcess();
-    KUrl furl(path);
-    kDebug(DBG_AREA) << "videopreview: url=" << furl << "; local:" << furl.isLocalFile() << endl;
-    fileinfo.towidth=width;
-    fileinfo.toheight=height;
-    QPixmap pix;
-//    if(furl.isLocalFile())
-//    {
-    QStringList args;
-    //TODO: modify inigo so that it can return some infos about a westley clip (duration, track number,fps,...) 
-    // without actually playing the file
-/*
-    args << playerBin << QString("\"" + path + "\"") << "-file-info";
-
-    kDebug(DBG_AREA) << "videopreview: starting process: --_" << " " << args.join(" ") << "_--\n";
-    if (! startAndWaitProcess(args) ) return NULL;
-
-    QString information=QString(inigoprocess->readAllStandardOutput() );
-    QRegExp findInfos("ID_VIDEO_FPS=([\\d]*).*ID_LENGTH=([\\d]*).*");
-    if(findInfos.indexIn( information) == -1 )
-    {
-        kDebug(DBG_AREA) << "videopreview: No information found, exiting\n";
-        return NULL;
+    producer->seek(framepos);
+    Mlt::Frame *frame = producer->get_frame();
+    if (frame == NULL) {
+        return result;
     }
-    fileinfo.seconds =findInfos.cap(2).toInt();
-    fileinfo.fps=findInfos.cap(1).toInt();
-    */
-    fileinfo.seconds = 250;
-    fileinfo.fps = 25;
-    
-    //kDebug(DBG_AREA) << "videopreview: find length=" << fileinfo.seconds << ", fps=" << fileinfo.fps << endl;
-
-    const int LASTTRY = 3;
-    for(int i=0; i<=LASTTRY; i++)
-    {
-        pix=getFrame(path);
-        if(!pix.isNull()) {
-            uint variance=imageVariance(pix.toImage()/*.bits(),( (width+ 7) & ~0x7), width, height, 1 */);
-            kDebug(DBG_AREA) << "videopreview: " << QFileInfo(path).fileName() << " frame variance: " << variance << "; " << 
-                    ((variance<=40 && ( i!=LASTTRY-1))? "!!!DROPPING!!!" : "GOOD :-)" ) << endl;
-            if(variance>40 || i==LASTTRY-1 ) break;
-        }
-    }
-    if(pix.isNull() )
-    {
-        return false;
+
+    mlt_image_format format = mlt_image_rgb24a;
+    height = 200;
+    double ar = frame->get_double("aspect_ratio");
+    if (ar == 0.0) ar = 1.33;
+    int calculated_width = (int)((double) height * ar);
+    uint8_t *data = frame->get_image(format, calculated_width, height, 0);
+    QImage image((uchar *)data, calculated_width, height, QImage::Format_ARGB32);
+
+    if (!image.isNull()) {
+        result = image.rgbSwapped().convertToFormat(QImage::Format_RGB32);
     }
-    img = pix.toImage();
-    return true;
-}
 
-QPixmap WestleyPreview::getFrame(const QString &path)
-{
-    QStringList args;
-    #define START ((fileinfo.seconds*15)/100)
-    #define END ((fileinfo.seconds*70)/100)
-    args.clear();
-    args << playerBin << "\"" + path + "\"";
-    if(fileinfo.towidth>fileinfo.toheight) fileinfo.toheight=-2; else fileinfo.towidth=-2;
-//     switch( flags ){
-//         case random
-//     }
-    unsigned long start=(unsigned long)(START+(rand->getDouble() * (END - START) ) );
-    args << QString("in=%1").arg(start)<< QString("out=%1").arg(start)<<"-consumer";
-
-    KTemporaryFile temp;
-    temp.setSuffix(".png");
-    temp.open();
-    args << QString("avformat:%1").arg(temp.fileName()) << "vframes=1"<< "f=rawvideo"<< "vcodec=png"<< QString("s=%1x%2").arg(fileinfo.towidth).arg(fileinfo.toheight);
-    if (! startAndWaitProcess(args) ) return NULL;
-    QPixmap retpix(temp.fileName());
-    temp.close();
-    return retpix;
+    delete frame;
+    return result;
 }
 
 
-uint WestleyPreview::imageVariance(QImage image )
+uint MltPreview::imageVariance(const QImage &image)
 {
-    uint delta=0;
-    uint avg=0;
-    uint bytes=image.numBytes();
-    uint STEPS=bytes/2;
+    if (image.isNull()) return 0;
+    uint delta = 0;
+    uint avg = 0;
+    uint bytes = image.numBytes();
+    if (bytes == 0) return 0;
+    uint STEPS = bytes / 2;
     QVarLengthArray<uchar> pivot(STEPS);
     kDebug(DBG_AREA) << "Using " << STEPS << " steps\n";
-    uchar *bits=image.bits();
+    const uchar *bits=image.bits();
     // First pass: get pivots and taking average
     for( uint i=0; i<STEPS ; i++ ){
-        pivot[i]=bits[i*(bytes/STEPS)];
+        pivot[i] = bits[2 * i];
+#if QT_VERSION >= 0x040700
+        avg+=pivot.at(i);
+#else
         avg+=pivot[i];
+#endif
     }
     avg=avg/STEPS;
     // Second Step: calculate delta (average?)
     for (uint i=0; i<STEPS; i++)
     {
-        int curdelta=abs(int(avg-pivot[i]));
+#if QT_VERSION >= 0x040700
+        int curdelta=abs(int(avg - pivot.at(i)));
+#else
+        int curdelta=abs(int(avg - pivot[i]));
+#endif
         delta+=curdelta;
     }
-    return delta/STEPS;
+    return delta / STEPS;
 }
-#include "westleypreview.moc"
 
+ThumbCreator::Flags MltPreview::flags() const
+{
+    return None;
+}
+
+
+
+#include "westleypreview.moc"