X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fkthumb.cpp;h=c6189ae7c88b35fb6292550664f88335e73de4e9;hb=209c8ebecef7e5681336e5a5c180ee9a8f7521da;hp=09f8802a6ff0f0a07911b14957ca11b52fe89eee;hpb=e9e42d4ed596ccad15f4bbe5e5da0580b73fd4ff;p=kdenlive diff --git a/src/kthumb.cpp b/src/kthumb.cpp index 09f8802a..c6189ae7 100644 --- a/src/kthumb.cpp +++ b/src/kthumb.cpp @@ -46,6 +46,7 @@ KThumb::KThumb(ClipManager *clipManager, KUrl url, const QString &id, const QStr m_url(url), m_thumbFile(), m_dar(1), + m_ratio(1), m_producer(NULL), m_clipManager(clipManager), m_id(id), @@ -56,32 +57,36 @@ KThumb::KThumb(ClipManager *clipManager, KUrl url, const QString &id, const QStr KThumb::~KThumb() { - m_requestedThumbs.clear(); + if (m_producer) m_clipManager->stopThumbs(m_id); m_intraFramesQueue.clear(); if (m_audioThumbProducer.isRunning()) { m_stopAudioThumbs = true; m_audioThumbProducer.waitForFinished(); slotAudioThumbOver(); } - m_future.waitForFinished(); m_intra.waitForFinished(); } void KThumb::setProducer(Mlt::Producer *producer) { - m_requestedThumbs.clear(); + if (m_producer) m_clipManager->stopThumbs(m_id); m_intraFramesQueue.clear(); - m_future.waitForFinished(); m_intra.waitForFinished(); + m_mutex.lock(); m_producer = producer; // FIXME: the profile() call leaks an object, but trying to free // it leads to a double-free in Profile::~Profile() - if (producer) m_dar = producer->profile()->dar(); + if (producer) { + Mlt::Profile *profile = producer->profile(); + m_dar = profile->dar(); + m_ratio = (double) profile->width() / profile->height(); + } + m_mutex.unlock(); } void KThumb::clearProducer() { - setProducer(NULL); + if (m_producer) setProducer(NULL); } bool KThumb::hasProducer() const @@ -97,8 +102,6 @@ void KThumb::updateThumbUrl(const QString &hash) void KThumb::updateClipUrl(KUrl url, const QString &hash) { m_url = url; - //if (m_producer) - //m_producer->set("resource", url.path().toUtf8().constData()); m_thumbFile = m_clipManager->projectFolder() + "/thumbs/" + hash + ".thumb"; } @@ -109,31 +112,32 @@ QPixmap KThumb::getImage(KUrl url, int width, int height) return getImage(url, 0, width, height); } -void KThumb::extractImage(int frame, int frame2) +void KThumb::extractImage(QList frames) { if (!KdenliveSettings::videothumbnails() || m_producer == NULL) return; - if (frame != -1 && !m_requestedThumbs.contains(frame)) m_requestedThumbs.append(frame); - if (frame2 != -1 && !m_requestedThumbs.contains(frame2)) m_requestedThumbs.append(frame2); - if (!m_future.isRunning()) m_future = QtConcurrent::run(this, &KThumb::doGetThumbs); + m_clipManager->requestThumbs(m_id, frames); } -void KThumb::doGetThumbs() + +void KThumb::getThumb(int frame) { const int theight = KdenliveSettings::trackheight(); - const int twidth = FRAME_SIZE;//(int)(theight * m_dar + 0.5); + const int swidth = (int)(theight * m_ratio + 0.5); + const int dwidth = (int)(theight * m_dar + 0.5); - while (!m_requestedThumbs.isEmpty()) { - int frame = m_requestedThumbs.takeFirst(); - if (frame != -1) { - QImage img = getFrame(m_producer, frame, twidth, theight); - emit thumbReady(frame, img); - } - } + QImage img = getProducerFrame(frame, swidth, dwidth, theight); + emit thumbReady(frame, img); } QPixmap KThumb::extractImage(int frame, int width, int height) { - return QPixmap::fromImage(getFrame(m_producer, frame, width, height)); + if (m_producer == NULL) { + QPixmap p(width, height); + p.fill(Qt::black); + return p; + } + QImage img = getProducerFrame(frame, (int) (height * m_ratio + 0.5), width, height); + return QPixmap::fromImage(img); } //static @@ -146,58 +150,92 @@ QPixmap KThumb::getImage(KUrl url, int frame, int width, int height) //""); //Mlt::Producer producer(profile, "xml-string", tmp); Mlt::Producer *producer = new Mlt::Producer(profile, url.path().toUtf8().constData()); - - pix = QPixmap::fromImage(getFrame(producer, frame, width, height)); + double swidth = (double) profile.width() / profile.height(); + pix = QPixmap::fromImage(getFrame(producer, frame, (int) (height * swidth + 0.5), width, height)); delete producer; return pix; } + +QImage KThumb::getProducerFrame(int framepos, int frameWidth, int displayWidth, int height) +{ + if (m_producer == NULL || !m_producer->is_valid()) { + QImage p(displayWidth, height, QImage::Format_ARGB32_Premultiplied); + p.fill(QColor(Qt::red).rgb()); + return p; + } + if (m_producer->is_blank()) { + QImage p(displayWidth, height, QImage::Format_ARGB32_Premultiplied); + p.fill(QColor(Qt::black).rgb()); + return p; + } + m_mutex.lock(); + m_producer->seek(framepos); + Mlt::Frame *frame = m_producer->get_frame(); + QImage p = getFrame(frame, frameWidth, displayWidth, height); + delete frame; + m_mutex.unlock(); + return p; +} + //static -QImage KThumb::getFrame(Mlt::Producer *producer, int framepos, int width, int height) +QImage KThumb::getFrame(Mlt::Producer *producer, int framepos, int frameWidth, int displayWidth, int height) { - QImage p(width, height, QImage::Format_ARGB32_Premultiplied); if (producer == NULL || !producer->is_valid()) { - p.fill(Qt::red); + QImage p(displayWidth, height, QImage::Format_ARGB32_Premultiplied); + p.fill(QColor(Qt::red).rgb()); return p; } - if (producer->is_blank()) { - p.fill(Qt::black); + QImage p(displayWidth, height, QImage::Format_ARGB32_Premultiplied); + p.fill(QColor(Qt::black).rgb()); return p; } producer->seek(framepos); Mlt::Frame *frame = producer->get_frame(); - if (!frame) { - kDebug() << "///// BROKEN FRAME"; - p.fill(Qt::red); + QImage p = getFrame(frame, frameWidth, displayWidth, height); + delete frame; + return p; +} + + +//static +QImage KThumb::getFrame(Mlt::Frame *frame, int frameWidth, int displayWidth, int height) +{ + QImage p(displayWidth, height, QImage::Format_ARGB32_Premultiplied); + if (frame == NULL || !frame->is_valid()) { + p.fill(QColor(Qt::red).rgb()); return p; } - /*Mlt::Producer parentProd(producer->parent()); - Mlt::Service service(parentProd.get_service()); - mlt_service_lock(service.get_service());*/ - int ow = width; + int ow = frameWidth; int oh = height; mlt_image_format format = mlt_image_rgb24a; - uint8_t *data = frame->get_image(format, ow, oh, 0); - QImage image((uchar *)data, ow, oh, QImage::Format_ARGB32_Premultiplied); - //mlt_service_unlock(service.get_service()); + const uchar* imagedata = frame->get_image(format, ow, oh); + QImage image(ow, oh, QImage::Format_ARGB32_Premultiplied); + memcpy(image.bits(), imagedata, ow * oh * 4);//.byteCount()); + + //const uchar* imagedata = frame->get_image(format, ow, oh); + //QImage image(imagedata, ow, oh, QImage::Format_ARGB32_Premultiplied); + if (!image.isNull()) { - if (ow > (2 * width)) { + if (ow > (2 * displayWidth)) { // there was a scaling problem, do it manually - QImage scaled = image.scaled(width, height); - p = scaled.rgbSwapped(); - } else p = image.rgbSwapped(); + image = image.scaled(displayWidth, height).rgbSwapped(); + } else { + image = image.scaled(displayWidth, height, Qt::IgnoreAspectRatio).rgbSwapped(); + } + p.fill(QColor(Qt::black).rgb()); + QPainter painter(&p); + painter.drawImage(p.rect(), image); + painter.end(); } else - p.fill(Qt::red); - - delete frame; + p.fill(QColor(Qt::red).rgb()); return p; } - //static uint KThumb::imageVariance(QImage image ) { @@ -328,7 +366,7 @@ void KThumb::getAudioThumbs(int channel, double frame, double frameLength, int a return; } - QMap > storeIn; + audioByteArray storeIn; //FIXME: Hardcoded!!! m_frequency = 48000; m_channels = channel; @@ -378,8 +416,8 @@ void KThumb::getAudioThumbs(int channel, double frame, double frameLength, int a void KThumb::slotCreateAudioThumbs() { - Mlt::Profile prof((char*) KdenliveSettings::current_profile().toUtf8().data()); - Mlt::Producer producer(prof, m_url.path().toUtf8().data()); + Mlt::Profile prof((char*) KdenliveSettings::current_profile().toUtf8().constData()); + Mlt::Producer producer(prof, m_url.path().toUtf8().constData()); if (!producer.is_valid()) { kDebug() << "++++++++ INVALID CLIP: " << m_url.path(); return; @@ -399,7 +437,10 @@ void KThumb::slotCreateAudioThumbs() int last_val = 0; int val = 0; - //kDebug() << "for " << m_frame << " " << m_frameLength << " " << m_producer.is_valid(); + mlt_audio_format m_audioFormat = mlt_audio_pcm; + double framesPerSecond = mlt_producer_get_fps(producer.get_producer()); + Mlt::Frame *mlt_frame; + for (int z = (int) m_frame; z < (int)(m_frame + m_frameLength) && producer.is_valid(); z++) { if (m_stopAudioThumbs) break; val = (int)((z - m_frame) / (m_frame + m_frameLength) * 100.0); @@ -408,13 +449,10 @@ void KThumb::slotCreateAudioThumbs() last_val = val; } producer.seek(z); - Mlt::Frame *mlt_frame = producer.get_frame(); + mlt_frame = producer.get_frame(); if (mlt_frame && mlt_frame->is_valid()) { - double m_framesPerSecond = mlt_producer_get_fps(producer.get_producer()); - int m_samples = mlt_sample_calculator(m_framesPerSecond, m_frequency, mlt_frame_get_position(mlt_frame->get_frame())); - mlt_audio_format m_audioFormat = mlt_audio_pcm; + int m_samples = mlt_sample_calculator(framesPerSecond, m_frequency, mlt_frame_get_position(mlt_frame->get_frame())); qint16* m_pcm = static_cast(mlt_frame->get_audio(m_audioFormat, m_frequency, m_channels, m_samples)); - for (int c = 0; c < m_channels; c++) { QByteArray m_array; m_array.resize(m_arrayWidth); @@ -449,33 +487,39 @@ void KThumb::askForAudioThumbs(const QString &id) } #if KDE_IS_VERSION(4,5,0) -void KThumb::queryIntraThumbs(int start, int end) +void KThumb::queryIntraThumbs(QList missingFrames) { - for (int i = start; i <= end; i++) { + foreach (int i, missingFrames) { if (!m_intraFramesQueue.contains(i)) m_intraFramesQueue.append(i); } qSort(m_intraFramesQueue); - if (!m_intra.isRunning()) m_intra = QtConcurrent::run(this, &KThumb::slotGetIntraThumbs); + if (!m_intra.isRunning()) { + m_intra = QtConcurrent::run(this, &KThumb::slotGetIntraThumbs); + } } void KThumb::slotGetIntraThumbs() { - int theight = KdenliveSettings::trackheight(); - int twidth = FRAME_SIZE; - QString path = m_url.path() + "%"; - QImage img; + const int theight = KdenliveSettings::trackheight(); + const int frameWidth = (int)(theight * m_ratio + 0.5); + const int displayWidth = (int)(theight * m_dar + 0.5); + QString path = m_url.path() + "_"; + bool addedThumbs = false; while (!m_intraFramesQueue.isEmpty()) { int pos = m_intraFramesQueue.takeFirst(); if (!m_clipManager->pixmapCache->contains(path + QString::number(pos))) { - m_clipManager->pixmapCache->insertImage(path + QString::number(pos), getFrame(m_producer, pos, twidth, theight)); + if (m_clipManager->pixmapCache->insertImage(path + QString::number(pos), getProducerFrame(pos, frameWidth, displayWidth, theight))) { + addedThumbs = true; + } + else kDebug()<<"// INSERT FAILD FOR: "<pixmapCache->findImage(path, &img);