]> git.sesse.net Git - kdenlive/blobdiff - src/kthumb.cpp
cleanup
[kdenlive] / src / kthumb.cpp
index 12b8403202b5a6c1f0d5c6eb6abf71d5c7d1bd66..ec3fea708c6922e6d26aaf40b181d7f3f27efc23 100644 (file)
  *                                                                         *
  ***************************************************************************/
 
-#include <qapplication.h>
+#include <qxml.h>
+#include <QImage>
+#include <QApplication>
+#include <QCryptographicHash>
 
 #include <kio/netaccess.h>
 #include <kdebug.h>
 
 #include <mlt++/Mlt.h>
 
-#include <qxml.h>
-#include <qimage.h>
-
-#include <QThread>
-#include <QApplication>
-#include <QCryptographicHash>
-
 #include "clipmanager.h"
 #include "renderer.h"
 #include "kthumb.h"
 #include "kdenlivesettings.h"
 #include "events.h"
 
+
 void MyThread::init(QObject *parent, KUrl url, QString target, double frame, double frameLength, int frequency, int channels, int arrayWidth) {
     stop_me = false;
     m_parent = parent;
@@ -123,22 +120,13 @@ void MyThread::run() {
     m_isWorking = false;
     if (stop_me) {
         f.remove();
-        QApplication::postEvent(m_parent, new ProgressEvent(-1, (QEvent::Type)10005));
-
     }
-    QApplication::postEvent(m_parent, new ProgressEvent(0, (QEvent::Type)10005));
+    QApplication::postEvent(m_parent, new ProgressEvent(-1, (QEvent::Type)10005));
 
 }
 
 
-#define _S(a)           (a)>255 ? 255 : (a)<0 ? 0 : (a)
-#define _R(y,u,v) (0x2568*(y)                          + 0x3343*(u)) /0x2000
-#define _G(y,u,v) (0x2568*(y) - 0x0c92*(v) - 0x1a1e*(u)) /0x2000
-#define _B(y,u,v) (0x2568*(y) + 0x40cf*(v))                                          /0x2000
-
-KThumb::KThumb(ClipManager *clipManager, KUrl url, int width, int height, QObject * parent, const char *name): QObject(parent), m_clipManager(clipManager), m_url(url), m_width(width), m_height(height) {
-
-    m_profile = new Mlt::Profile((char*) KdenliveSettings::current_profile().data());
+KThumb::KThumb(ClipManager *clipManager, KUrl url, QObject * parent, const char *name): QObject(parent), m_clipManager(clipManager), m_url(url), m_producer(NULL), m_dar(1) {
     QCryptographicHash context(QCryptographicHash::Sha1);
     context.addData((KFileItem(m_url, "text/plain", S_IFREG).timeString() + m_url.fileName()).toAscii().data());
     m_thumbFile = KGlobal::dirs()->saveLocation("tmp" , "kdenlive") + context.result().toHex() + ".thumb";
@@ -146,106 +134,127 @@ KThumb::KThumb(ClipManager *clipManager, KUrl url, int width, int height, QObjec
 }
 
 KThumb::~KThumb() {
-    if (m_profile) delete m_profile;
     if (thumbProducer.isRunning()) thumbProducer.exit();
 }
 
+void KThumb::setProducer(Mlt::Producer *producer) {
+    m_producer = producer;
+    m_dar = producer->profile()->dar();
+}
+
+void KThumb::updateClipUrl(KUrl url) {
+    m_url = url;
+}
 
 //static
 QPixmap KThumb::getImage(KUrl url, int width, int height) {
     if (url.isEmpty()) return QPixmap();
-    /*QPixmap pix(width, height);
-    kDebug() << "+++++++++++  GET THMB IMG FOR: " << url;
-    char *tmp = Render::decodedString(url.path());
-    Mlt::Profile prof((char*) KdenliveSettings::current_profile().data());
-    Mlt::Producer m_producer(prof, tmp);
-    delete[] tmp;
-
-    if (m_producer.is_blank()) {
-        pix.fill(Qt::black);
-        return pix;
-    }
-    Mlt::Frame * m_frame;
-    mlt_image_format format = mlt_image_rgb24a;
-    Mlt::Filter m_convert(prof, "avcolour_space");
-    m_convert.set("forced", mlt_image_rgb24a);
-    m_producer.attach(m_convert);
-    //m_producer.seek(frame);
-    m_frame = m_producer.get_frame();
-    if (m_frame && m_frame->is_valid()) {
-        uint8_t *thumb = m_frame->get_image(format, width, height);
-        QImage image(thumb, width, height, QImage::Format_ARGB32);
-        if (!image.isNull()) {
-            pix = pix.fromImage(image);
-        } else pix.fill(Qt::black);
-    }
-    if (m_frame) delete m_frame;
-    return pix;*/
     return getImage(url, 0, width, height);
 }
 
 void KThumb::extractImage(int frame, int frame2) {
-    if (m_url.isEmpty()) return;
+    if (m_url.isEmpty() || !KdenliveSettings::videothumbnails() || m_producer == NULL) return;
 
-    char *tmp = Render::decodedString(m_url.path());
-    Mlt::Producer m_producer(*m_profile, tmp);
-    delete[] tmp;
-
-    QPixmap pix(m_width, m_height);
-    if (m_producer.is_blank()) {
-        QPixmap pix(m_width, m_height);
+    int twidth = (int)(KdenliveSettings::trackheight() * m_dar);
+    if (m_producer->is_blank()) {
+        QPixmap pix(twidth, KdenliveSettings::trackheight());
         pix.fill(Qt::black);
         emit thumbReady(frame, pix);
         return;
     }
     if (frame != -1) {
-        QPixmap pix = getFrame(&m_producer, frame, m_width, m_height);
+        QPixmap pix = getFrame(m_producer, frame, twidth, KdenliveSettings::trackheight());
         emit thumbReady(frame, pix);
     }
     if (frame2 != -1) {
-        QPixmap pix = getFrame(&m_producer, frame2, m_width, m_height);
+        QPixmap pix = getFrame(m_producer, frame2, twidth , KdenliveSettings::trackheight());
         emit thumbReady(frame2, pix);
     }
+}
 
+QPixmap KThumb::extractImage(int frame, int width, int height) {
+    return getFrame(m_producer, frame, width, height);
 }
 
+//static
 QPixmap KThumb::getImage(KUrl url, int frame, int width, int height) {
     Mlt::Profile profile((char*) KdenliveSettings::current_profile().data());
     QPixmap pix(width, height);
     if (url.isEmpty()) return pix;
+
     char *tmp = Render::decodedString(url.path());
-    Mlt::Producer producer(profile, tmp);
+    //"<westley><playlist><producer resource=\"" + url.path() + "\" /></playlist></westley>");
+    //Mlt::Producer producer(profile, "westley-xml", tmp);
+    Mlt::Producer *producer = new Mlt::Producer(profile, tmp);
     delete[] tmp;
 
-    if (producer.is_blank()) {
-
+    if (producer->is_blank()) {
         pix.fill(Qt::black);
+        delete producer;
         return pix;
     }
-    return getFrame(&producer, frame, width, height);
-
+    pix = getFrame(producer, frame, width, height);
+    delete producer;
+    return pix;
 }
 
-QPixmap KThumb::getFrame(Mlt::Producer* producer, int frame, int width, int height) {
+//static
+/*
+QPixmap KThumb::getImage(QDomElement xml, int frame, int width, int height) {
     Mlt::Profile profile((char*) KdenliveSettings::current_profile().data());
-    Mlt::Filter m_convert(profile, "avcolour_space");
-    m_convert.set("forced", mlt_image_rgb24a);
-    producer->attach(m_convert);
-
-    producer->seek(frame);
-    Mlt::Frame * m_frame = producer->get_frame();
-    mlt_image_format format = mlt_image_rgb24a;
     QPixmap pix(width, height);
-    if (m_frame && m_frame->is_valid()) {
-        uint8_t *thumb = m_frame->get_image(format, width, height);
-        QImage image(thumb, width, height, QImage::Format_ARGB32);
-        if (!image.isNull()) {
-            pix = pix.fromImage(image);
-        } else pix.fill(Qt::black);
+    QDomDocument doc;
+    QDomElement westley = doc.createElement("westley");
+    QDomElement play = doc.createElement("playlist");
+    doc.appendChild(westley);
+    westley.appendChild(play);
+    play.appendChild(doc.importNode(xml, true));
+    char *tmp = Render::decodedString(doc.toString());
+    Mlt::Producer producer(profile, "westley-xml", tmp);
+    delete[] tmp;
+
+    if (producer.is_blank()) {
+        pix.fill(Qt::black);
+        return pix;
     }
-    if (m_frame) delete m_frame;
-    return pix;
+    return getFrame(producer, frame, width, height);
+}*/
 
+//static
+QPixmap KThumb::getFrame(Mlt::Producer *producer, int framepos, int width, int height) {
+    kDebug() << "//REQUESTING FRAME: " << framepos;
+    if (producer == NULL) return QPixmap();
+    producer->seek(framepos);
+    Mlt::Frame *frame = producer->get_frame();
+    if (!frame) {
+        kDebug() << "///// BROKEN FRAME";
+        return QPixmap();
+    }
+    kDebug() << "///// FRAME exists";
+    mlt_image_format format = mlt_image_yuv422;
+    int frame_width = 0;
+    int frame_height = 0;
+    frame->set("normalised_height", height);
+    frame->set("normalised_width", width);
+    QPixmap pix(width, height);
+    kDebug() << "///// FRAME exists 2";
+    uint8_t *data = frame->get_image(format, frame_width, frame_height, 0);
+    kDebug() << "///// FRAME exists 2a";
+    uint8_t *new_image = (uint8_t *)mlt_pool_alloc(frame_width * (frame_height + 1) * 4);
+    kDebug() << "///// FRAME exists 2b";
+    mlt_convert_yuv422_to_rgb24a((uint8_t *)data, new_image, frame_width * frame_height);
+    kDebug() << "///// FRAME exists 2c";
+    QImage image((uchar *)new_image, frame_width, frame_height, QImage::Format_ARGB32);
+    kDebug() << "///// FRAME exists 3";
+    if (!image.isNull()) {
+        pix = pix.fromImage(image.rgbSwapped());
+    } else
+        pix.fill(Qt::black);
+
+    mlt_pool_release(new_image);
+    delete frame;
+    kDebug() << "//REQUESTING FRAME " << framepos << " OVER";
+    return pix;
 }
 /*
 void KThumb::getImage(KUrl url, int frame, int width, int height)
@@ -375,7 +384,7 @@ void KThumb::getAudioThumbs(int channel, double frame, double frameLength, int a
 void KThumb::customEvent(QEvent * event) {
     if (event->type() == 10005) {
         ProgressEvent* p = (ProgressEvent*) event;
-        m_clipManager->setThumbsProgress(m_url, p->value());
+        m_clipManager->setThumbsProgress(i18n("Creating thumbnail for %1", m_url.fileName()), p->value());
     }
 }