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"
projects / kdenlive / blobdiff