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 ***************************************************************************/
21 #include "westleypreview.h"
25 #include <QtCore/QVarLengthArray>
27 #include <kstandarddirs.h>
28 #include <krandomsequence.h>
29 #include <qdatetime.h>
34 #include <qfileinfo.h>
35 #include <KTemporaryFile>
45 KDE_EXPORT ThumbCreator *new_creator() {
46 return new MltPreview;
50 MltPreview::MltPreview() :
57 MltPreview::~MltPreview()
59 Mlt::Factory::close();
63 bool MltPreview::create(const QString &path, int width, int height, QImage &img)
65 Mlt::Profile *profile = new Mlt::Profile("dv_pal");
66 char *tmp = (char *) qstrdup(path.toUtf8().data());
67 Mlt::Producer *producer = new Mlt::Producer(*profile, tmp);
71 if (producer->is_blank()) {
79 //img = getFrame(producer, frame, width, height);
81 while (variance <= 40 && ct < 4) {
82 img = getFrame(producer, frame, width, height);
83 variance = imageVariance(img);
90 return (img.isNull() == false);
93 QImage MltPreview::getFrame(Mlt::Producer *producer, int framepos, int /*width*/, int height)
96 if (producer == NULL) {
100 producer->seek(framepos);
101 Mlt::Frame *frame = producer->get_frame();
106 mlt_image_format format = mlt_image_rgb24a;
108 double ar = frame->get_double("aspect_ratio");
109 if (ar == 0.0) ar = 1.33;
110 int calculated_width = (int)((double) height * ar);
111 uint8_t *data = frame->get_image(format, calculated_width, height, 0);
112 QImage image((uchar *)data, calculated_width, height, QImage::Format_ARGB32);
114 if (!image.isNull()) {
115 result = image.rgbSwapped().convertToFormat(QImage::Format_RGB32);
123 uint MltPreview::imageVariance(QImage image)
125 if (image.isNull()) return 0;
128 uint bytes = image.numBytes();
129 uint STEPS = bytes / 2;
130 QVarLengthArray<uchar> pivot(STEPS);
131 kDebug(DBG_AREA) << "Using " << STEPS << " steps\n";
132 uchar *bits = image.bits();
133 // First pass: get pivots and taking average
134 for (uint i = 0; i < STEPS ; i++) {
135 pivot[i] = bits[i*(bytes/STEPS)];
139 // Second Step: calculate delta (average?)
140 for (uint i = 0; i < STEPS; i++) {
141 int curdelta = abs(int(avg - pivot[i]));
144 return delta / STEPS;
147 ThumbCreator::Flags MltPreview::flags() const