]> git.sesse.net Git - kdenlive/blobdiff - src/lib/audio/audioEnvelope.cpp
Use QLatin1String
[kdenlive] / src / lib / audio / audioEnvelope.cpp
index 7849372d1a5ce067d715d3a0dcfd59d1a297a3c7..6a5da33bcd973b738e5152f023057e9ffbb15946 100644 (file)
 #include "audioEnvelope.h"
 
 #include "audioStreamInfo.h"
+#include <QDebug>
 #include <QImage>
 #include <QTime>
 #include <cmath>
 #include <iostream>
 
-AudioEnvelope::AudioEnvelope(Mlt::Producer *producer) :
+AudioEnvelope::AudioEnvelope(const QString &url, Mlt::Producer *producer, int offset, int length) :
     m_envelope(NULL),
-    m_producer(producer),
+    m_offset(offset),
+    m_length(length),
     m_envelopeSize(producer->get_length()),
+    m_envelopeMax(0),
+    m_envelopeMean(0),
+    m_envelopeStdDev(0),
     m_envelopeStdDevCalculated(false),
     m_envelopeIsNormalized(false)
 {
+    // make a copy of the producer to avoid audio playback issues
+    QString path = QString::fromUtf8(producer->get("resource"));
+    if (path == QLatin1String("<playlist>") || path == QLatin1String("<tractor>") || path ==QLatin1String( "<producer>"))
+       path = url;
+    m_producer = new Mlt::Producer(*(producer->profile()), path.toUtf8().constData());
+    if (!m_producer || !m_producer->is_valid()) {
+       qDebug()<<"// Cannot create envelope for producer: "<<path;
+    }
     m_info = new AudioInfo(m_producer);
+
+    Q_ASSERT(m_offset >= 0);
+    if (m_length > 0) {
+        Q_ASSERT(m_length+m_offset <= m_envelopeSize);
+        m_envelopeSize = m_length;
+    }
 }
 
 AudioEnvelope::~AudioEnvelope()
@@ -32,6 +51,7 @@ AudioEnvelope::~AudioEnvelope()
         delete[] m_envelope;
     }
     delete m_info;
+    delete m_producer;
 }
 
 
@@ -47,6 +67,10 @@ int AudioEnvelope::envelopeSize() const
 {
     return m_envelopeSize;
 }
+int64_t AudioEnvelope::maxValue() const
+{
+    return m_envelopeMax;
+}
 
 
 
@@ -71,9 +95,10 @@ void AudioEnvelope::loadEnvelope()
 
     QTime t;
     t.start();
-    m_producer->seek(0);
+    int count = 0;
+    m_producer->seek(m_offset);
     m_producer->set_speed(1.0); // This is necessary, otherwise we don't get any new frames in the 2nd run.
-    for (int i = 0; i < m_envelopeSize; i++) {
+    for (int i = 0; i < m_envelopeSize; ++i) {
 
         frame = m_producer->get_frame(i);
         position = mlt_frame_get_position(frame->get_frame());
@@ -96,6 +121,11 @@ void AudioEnvelope::loadEnvelope()
 //                  << "-" << m_producer->get_in() << "+" << m_producer->get_out() << " ";
 
         delete frame;
+
+        count++;
+        if (m_length > 0 && count > m_length) {
+            break;
+        }
     }
     m_envelopeMean /= m_envelopeSize;
     std::cout << "Calculating the envelope (" << m_envelopeSize << " frames) took "
@@ -113,7 +143,7 @@ int64_t AudioEnvelope::loadStdDev()
         }
 
         m_envelopeStdDev = 0;
-        for (int i = 0; i < m_envelopeSize; i++) {
+        for (int i = 0; i < m_envelopeSize; ++i) {
             m_envelopeStdDev += sqrt((m_envelope[i]-m_envelopeMean)*(m_envelope[i]-m_envelopeMean)/m_envelopeSize);
         }
         m_envelopeStdDevCalculated = true;
@@ -132,7 +162,7 @@ void AudioEnvelope::normalizeEnvelope(bool clampTo0)
 
         m_envelopeMax = 0;
         int64_t newMean = 0;
-        for (int i = 0; i < m_envelopeSize; i++) {
+        for (int i = 0; i < m_envelopeSize; ++i) {
 
             m_envelope[i] -= m_envelopeMean;
 
@@ -161,6 +191,10 @@ QImage AudioEnvelope::drawEnvelope()
 
     QImage img(m_envelopeSize, 400, QImage::Format_ARGB32);
     img.fill(qRgb(255,255,255));
+
+    if (m_envelopeMax == 0)
+        return img;
+
     double fy;
     for (int x = 0; x < img.width(); x++) {
         fy = m_envelope[x]/double(m_envelopeMax) * img.height();