1 /***************************************************************************
2 Copyright (C) 2006-2008
3 by Marco Gulino <marco@kmobiletools.org>
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the
17 Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18 Boston, MA 02110-1301, USA.
19 ***************************************************************************/
24 #include <QtCore/QVarLengthArray>
26 #include <kstandarddirs.h>
27 #include <krandomsequence.h>
28 #include <qdatetime.h>
33 #include <qfileinfo.h>
34 #include <KTemporaryFile>
38 #include "westleypreview.h"
44 KDE_EXPORT ThumbCreator *new_creator() {
45 return new WestleyPreview;
49 WestleyPreview::WestleyPreview()
50 : m_rand(0), m_inigoprocess(0) {
53 WestleyPreview::~WestleyPreview() {
55 delete m_inigoprocess;
58 bool WestleyPreview::startAndWaitProcess(const QStringList &args) {
59 kDebug(DBG_AREA) << "westleypreview: starting process with args: " << args << endl;
60 m_inigoprocess->start(args.join(" "));
61 if (! m_inigoprocess->waitForStarted()) {
62 kDebug(DBG_AREA) << "westleypreview: PROCESS NOT STARTED!!! exiting\n";
65 if (! m_inigoprocess->waitForFinished()) {
66 kDebug(DBG_AREA) << "westleypreview: PROCESS DIDN'T FINISH!! exiting\n";
67 m_inigoprocess->close();
70 kDebug() << "westleypreview: process started and ended correctly\n";
74 bool WestleyPreview::create(const QString &path, int width, int height, QImage &img) {
76 /*if (fi.suffix().trimmed() != "westley" && fi.suffix().trimmed() != "kdenlive")
78 kDebug(DBG_AREA) << "westleypreview: matched extension " << fi.suffix().prepend('.') << "; exiting.\n";
81 playerBin = KStandardDirs::findExe("inigo");
82 if (playerBin.isEmpty()) {
83 kDebug(DBG_AREA) << "westleypreview: inigo not found, exiting.\n";
90 m_rand = new KRandomSequence(QDateTime::currentDateTime().toTime_t());
91 m_inigoprocess = new QProcess();
93 kDebug(DBG_AREA) << "videopreview: url=" << furl << "; local:" << furl.isLocalFile() << endl;
94 fileinfo.towidth = width;
95 fileinfo.toheight = height;
97 // if(furl.isLocalFile())
100 //TODO: modify inigo so that it can return some infos about a westley clip (duration, track number,fps,...)
101 // without actually playing the file
103 args << playerBin << QString("\"" + path + "\"") << "-file-info";
105 kDebug(DBG_AREA) << "videopreview: starting process: --_" << " " << args.join(" ") << "_--\n";
106 if (! startAndWaitProcess(args) ) return NULL;
108 QString information=QString(inigoprocess->readAllStandardOutput() );
109 QRegExp findInfos("ID_VIDEO_FPS=([\\d]*).*ID_LENGTH=([\\d]*).*");
110 if(findInfos.indexIn( information) == -1 )
112 kDebug(DBG_AREA) << "videopreview: No information found, exiting\n";
115 fileinfo.seconds =findInfos.cap(2).toInt();
116 fileinfo.fps=findInfos.cap(1).toInt();
118 fileinfo.seconds = 250;
121 //kDebug(DBG_AREA) << "videopreview: find length=" << fileinfo.seconds << ", fps=" << fileinfo.fps << endl;
123 const int LASTTRY = 3;
124 for (int i = 0; i <= LASTTRY; i++) {
125 pix = getFrame(path);
127 uint variance = imageVariance(pix.toImage()/*.bits(),( (width+ 7) & ~0x7), width, height, 1 */);
128 kDebug(DBG_AREA) << "videopreview: " << QFileInfo(path).fileName() << " frame variance: " << variance << "; " <<
129 ((variance <= 40 && (i != LASTTRY - 1)) ? "!!!DROPPING!!!" : "GOOD :-)") << endl;
130 if (variance > 40 || i == LASTTRY - 1) break;
140 QPixmap WestleyPreview::getFrame(const QString &path) {
142 #define START ((fileinfo.seconds*15)/100)
143 #define END ((fileinfo.seconds*70)/100)
145 args << playerBin << "\"" + path + "\"";
146 if (fileinfo.towidth > fileinfo.toheight) fileinfo.toheight = -2;
147 else fileinfo.towidth = -2;
151 unsigned long start = (unsigned long)(START + (m_rand->getDouble() * (END - START)));
152 args << QString("in=%1").arg(start) << QString("out=%1").arg(start) << "-consumer";
155 temp.setSuffix(".png");
157 args << QString("avformat:%1").arg(temp.fileName()) << "vframes=1" << "f=rawvideo" << "vcodec=png" << QString("s=%1x%2").arg(fileinfo.towidth).arg(fileinfo.toheight);
158 if (! startAndWaitProcess(args)) return NULL;
159 QPixmap retpix(temp.fileName());
165 uint WestleyPreview::imageVariance(QImage image) {
168 uint bytes = image.numBytes();
169 uint STEPS = bytes / 2;
170 QVarLengthArray<uchar> pivot(STEPS);
171 kDebug(DBG_AREA) << "Using " << STEPS << " steps\n";
172 uchar *bits = image.bits();
173 // First pass: get pivots and taking average
174 for (uint i = 0; i < STEPS ; i++) {
175 pivot[i] = bits[i*(bytes/STEPS)];
179 // Second Step: calculate delta (average?)
180 for (uint i = 0; i < STEPS; i++) {
181 int curdelta = abs(int(avg - pivot[i]));
184 return delta / STEPS;
187 ThumbCreator::Flags WestleyPreview::flags() const {
191 #include "westleypreview.moc"