]> git.sesse.net Git - kdenlive/blobdiff - src/clipstabilize.cpp
fix warning
[kdenlive] / src / clipstabilize.cpp
index 5c6db9cddbde9462d5d592ebbadaf9c3d66e60d4..ccf54846db6141259a8c571b285e64462b1138e9 100644 (file)
 
 
 #include "clipstabilize.h"
+#include "doubleparameterwidget.h"
 
 #include <KDebug>
 #include <mlt++/Mlt.h>
 #include "kdenlivesettings.h"
 #include <KGlobalSettings>
 #include <KMessageBox>
+#include <KColorScheme>
+#include <QtConcurrentRun>
+#include <QTimer>
+#include <QSlider>
 #include <KFileDialog>
 
-
-ClipStabilize::ClipStabilize(KUrl::List urls, const QString &params, QWidget * parent) :
-        QDialog(parent), m_urls(urls), m_duration(0)
+ClipStabilize::ClipStabilize(KUrl::List urls, const QString &params, Mlt::Filter* filter,QWidget * parent) :
+        QDialog(parent), m_profile(NULL),m_consumer(NULL),m_playlist(NULL),m_urls(urls),m_duration(0),m_filter(filter),vbox(NULL)
 {
     setFont(KGlobalSettings::toolBarFont());
     setupUi(this);
-    setAttribute(Qt::WA_DeleteOnClose);
+    if (filter) setAttribute(Qt::WA_DeleteOnClose);
     log_text->setHidden(true);
     setWindowTitle(i18n("Stabilize Clip"));
     auto_add->setText(i18np("Add clip to project", "Add clips to project", m_urls.count()));
-       profile=Mlt::Profile(KdenliveSettings::current_profile().toUtf8().constData());
-       filtername=params;
+    m_profile = new Mlt::Profile(KdenliveSettings::current_profile().toUtf8().constData());
+    filtername = params;
+
+
+    QPalette p = palette();
+    KColorScheme scheme(p.currentColorGroup(), KColorScheme::View, KSharedConfig::openConfig(KdenliveSettings::colortheme()));
+    QColor dark_bg = scheme.shade(KColorScheme::DarkShade);
+    QColor selected_bg = scheme.decoration(KColorScheme::FocusColor).color();
+    QColor hover_bg = scheme.decoration(KColorScheme::HoverColor).color();
+    QColor light_bg = scheme.shade(KColorScheme::LightShade);
+
+    QString stylesheet(QString("QProgressBar:horizontal {border: 1px solid %1;border-radius:0px;border-top-left-radius: 4px;border-bottom-left-radius: 4px;border-right: 0px;background:%4;padding: 0px;text-align:left center}\
+                               QProgressBar:horizontal#dragOnly {background: %1} QProgressBar:horizontal:hover#dragOnly {background: %3} QProgressBar:horizontal:hover {border: 1px solid %3;border-right: 0px;}\
+                               QProgressBar::chunk:horizontal {background: %1;} QProgressBar::chunk:horizontal:hover {background: %3;}\
+                               QProgressBar:horizontal[inTimeline=\"true\"] { border: 1px solid %2;border-right: 0px;background: %4;padding: 0px;text-align:left center } QProgressBar::chunk:horizontal[inTimeline=\"true\"] {background: %2;}\
+                               QAbstractSpinBox#dragBox {border: 1px solid %1;border-top-right-radius: 4px;border-bottom-right-radius: 4px;padding-right:0px;} QAbstractSpinBox::down-button#dragBox {width:0px;padding:0px;}\
+                               QAbstractSpinBox::up-button#dragBox {width:0px;padding:0px;} QAbstractSpinBox[inTimeline=\"true\"]#dragBox { border: 1px solid %2;} QAbstractSpinBox:hover#dragBox {border: 1px solid %3;} ")
+                       .arg(dark_bg.name()).arg(selected_bg.name()).arg(hover_bg.name()).arg(light_bg.name()));
+       setStyleSheet(stylesheet);
 
     if (m_urls.count() == 1) {
         QString fileName = m_urls.at(0).path(); //.section('.', 0, -1);
-        QString newFile = params.section(' ', -1).replace("%1", fileName);
+        QString newFile = fileName.append(".mlt");
         KUrl dest(newFile);
         source_url->setUrl(m_urls.at(0));
         dest_url->setMode(KFile::File);
@@ -64,152 +85,248 @@ ClipStabilize::ClipStabilize(KUrl::List urls, const QString &params, QWidget * p
     if (!params.isEmpty()) {
         label_profile->setHidden(true);
         profile_list->setHidden(true);
-        ffmpeg_params->setPlainText(params.simplified());
+        //ffmpeg_params->setPlainText(params.simplified());
         /*if (!description.isEmpty()) {
             transcode_info->setText(description);
         } else transcode_info->setHidden(true);*/
     } 
 
+       if (filtername=="videostab"){
+               QStringList ls;
+               ls << "shutterangle,type,int,value,0,min,0,max,100,tooltip,Angle that Images could be maximum rotated";
+               fillParameters(ls);
+       }else if (filtername=="videostab2"){
+               QStringList ls;
+               ls << "accuracy,type,int,value,4,min,1,max,10,tooltip,Accuracy of Shakiness detection";
+               ls << "shakiness,type,int,value,4,min,1,max,10,tooltip,How shaky is the Video";
+               ls << "stepsize,type,int,value,6,min,0,max,100,tooltip,Stepsize of Detection process minimum around";
+               ls << "algo,type,bool,value,1,min,0,max,1,tooltip,0 = Bruteforce 1 = small measurement fields";
+               ls << "mincontrast,type,double,value,0.3,min,0,max,1,factor,1,decimals,2,tooltip,Below this Contrast Field is discarded";
+               ls << "show,type,int,value,0,min,0,max,2,tooltip,0 = draw nothing. 1 or 2 show fields and transforms";
+               ls << "smoothing,type,int,value,10,min,0,max,100,tooltip,number of frames for lowpass filtering";
+               ls << "maxshift,type,int,value,-1,min,-1,max,1000,tooltip,max number of pixels to shift";
+               ls << "maxangle,type,int,value,-1,min,-1,max,1000,tooltip,max anglen to rotate (in rad)";
+               ls << "crop,type,bool,value,0,min,0,max,1,tooltip,0 = keep border  1 = black background";
+               ls << "invert,type,bool,value,0,min,0,max,1,tooltip,invert transform";
+               ls << "realtive,type,bool,value,1,min,0,max,1,tooltip,0 = absolute transform  1= relative";
+               ls << "zoom,type,int,value,0,min,-500,max,500,tooltip,additional zoom during transform";
+               ls << "optzoom,type,bool,value,1,min,0,max,1,tooltip,use optimal zoom (calulated from transforms)";
+               ls << "sharpen,type,double,value,0.8,min,0,max,1,decimals,1,tooltip,sharpen transformed image";
+               fillParameters(ls);
+
+       }
     connect(button_start, SIGNAL(clicked()), this, SLOT(slotStartStabilize()));
+       connect(buttonBox,SIGNAL(rejected()), this, SLOT(slotAbortStabilize()));
 
-    m_stabilizeProcess.setProcessChannelMode(QProcess::MergedChannels);
-    connect(&m_stabilizeProcess, SIGNAL(readyReadStandardOutput()), this, SLOT(slotShowStabilizeInfo()));
-    connect(&m_stabilizeProcess, SIGNAL(finished(int, QProcess::ExitStatus)), this, SLOT(slotStabilizeFinished(int, QProcess::ExitStatus)));
+       m_timer=new QTimer(this);
+       connect(m_timer, SIGNAL(timeout()), this, SLOT(slotShowStabilizeInfo()));
 
+       vbox=new QVBoxLayout(optionsbox);
+       QHashIterator<QString,QHash<QString,QString> > hi(m_ui_params);
+       while(hi.hasNext()){
+               hi.next();
+               QHash<QString,QString> val=hi.value();
+               if (val["type"]=="int" || val["type"]=="double"){
+                       DoubleParameterWidget *dbl=new DoubleParameterWidget(hi.key(), val["value"].toDouble(),
+                                       val["min"].toDouble(),val["max"].toDouble(),val["value"].toDouble(),
+                                       "",0/*id*/,""/*suffix*/,val["decimals"]!=""?val["decimals"].toInt():0,this);
+                       dbl->setObjectName(hi.key());
+                       dbl->setToolTip(val["tooltip"]);
+                       connect(dbl,SIGNAL(valueChanged(double)),this,SLOT(slotUpdateParams()));
+                       vbox->addWidget(dbl);
+               }else if (val["type"]=="bool"){
+                       QCheckBox *ch=new QCheckBox(hi.key(),this);
+                       ch->setCheckState(val["value"] == "0" ? Qt::Unchecked : Qt::Checked);
+                       ch->setObjectName(hi.key());
+                       connect(ch, SIGNAL(stateChanged(int)) , this,SLOT(slotUpdateParams()));
+                       ch->setToolTip(val["tooltip"]);
+                       vbox->addWidget(ch);
+               
+               }
+       }
     adjustSize();
 }
 
 ClipStabilize::~ClipStabilize()
 {
-    if (m_stabilizeProcess.state() != QProcess::NotRunning) {
+    /*if (m_stabilizeProcess.state() != QProcess::NotRunning) {
         m_stabilizeProcess.close();
-    }
+    }*/
+       if (m_stabilizeRun.isRunning()){
+               if (m_consumer){
+                       m_consumer->stop();
+               }
+               m_stabilizeRun.waitForFinished();
+       }
+       if (m_profile) free (m_profile);
+       if (m_consumer) free (m_consumer);
+       if (m_playlist) free (m_playlist);
 }
 
 void ClipStabilize::slotStartStabilize()
 {
-    if (m_stabilizeProcess.state() != QProcess::NotRunning) {
+    if (m_consumer && !m_consumer->is_stopped()) {
         return;
     }
     m_duration = 0;
     QStringList parameters;
     QString destination;
-    QString params = ffmpeg_params->toPlainText().simplified();
+    //QString params = ffmpeg_params->toPlainText().simplified();
     if (urls_list->count() > 0) {
         source_url->setUrl(m_urls.takeFirst());
-        destination = dest_url->url().path(KUrl::AddTrailingSlash) + source_url->url().fileName();
+        destination = dest_url->url().path(KUrl::AddTrailingSlash)+ source_url->url().fileName()+".mlt";
         QList<QListWidgetItem *> matching = urls_list->findItems(source_url->url().path(), Qt::MatchExactly);
         if (matching.count() > 0) {
             matching.at(0)->setFlags(Qt::ItemIsSelectable);
             urls_list->setCurrentItem(matching.at(0));
         }
     } else {
-        destination = dest_url->url().path().section('.', 0, -2);
+        destination = dest_url->url().path();
     }
-    QString extension = params.section("%1", 1, 1).section(' ', 0, 0);
     QString s_url = source_url->url().path();
 
-    parameters << "-i" << s_url;
-    if (QFile::exists(destination + extension)) {
-        if (KMessageBox::questionYesNo(this, i18n("File %1 already exists.\nDo you want to overwrite it?", destination + extension)) == KMessageBox::No) return;
-        parameters << "-y";
+    if (QFile::exists(destination)) {
+                       if (KMessageBox::questionYesNo(this, i18n("File %1 already exists.\nDo you want to overwrite it?", destination )) == KMessageBox::No) return;
     }
-    foreach(QString s, params.split(' '))
-    parameters << s.replace("%1", destination);
     buttonBox->button(QDialogButtonBox::Abort)->setText(i18n("Abort"));
 
-    //kDebug() << "/// FFMPEG ARGS: " << parameters;
+       if (m_profile){
+               m_playlist= new Mlt::Playlist;
+               Mlt::Filter filter(*m_profile,filtername.toUtf8().data());
+               QHashIterator <QString,QHash<QString,QString> > it(m_ui_params);
+               while (it.hasNext()){
+                       it.next();
+                       filter.set(
+                                       it.key().toAscii().data(),
+                                       QString::number(it.value()["value"].toDouble()).toAscii().data());
+               }
+               Mlt::Producer p(*m_profile,s_url.toUtf8().data());
+               if (p.is_valid()) {
+                       m_playlist->append(p);
+                       m_playlist->attach(filter);
+                       m_consumer= new Mlt::Consumer(*m_profile,"xml",destination.toUtf8().data());
+                       m_consumer->set("all",1);
+                       m_consumer->set("real_time",-2);
+                       m_consumer->connect(*m_playlist);
+                       m_stabilizeRun = QtConcurrent::run(this, &ClipStabilize::slotRunStabilize);
+                       m_timer->start(500);
+                       button_start->setEnabled(false);
+               }
+       }
+
+}
 
-       Mlt::Producer producer(profile,s_url.toUtf8().data());
-       Mlt::Filter filter(profile,filtername.toUtf8().data());
-       producer.attach(filter);
-       Mlt::Consumer xmlout(profile,"xml");
-       xmlout.connect(producer);
-       xmlout.run();
-    //m_stabilizeProcess.start("ffmpeg", parameters);
-    button_start->setEnabled(false);
+void ClipStabilize::slotRunStabilize()
+{
+       if (m_consumer)
+       {
+               m_consumer->run();
+       }
+}
 
+void ClipStabilize::slotAbortStabilize()
+{
+       if (m_consumer)
+       {
+               m_timer->stop();
+               m_consumer->stop();
+               slotStabilizeFinished(false);
+       }
 }
 
 void ClipStabilize::slotShowStabilizeInfo()
 {
-    QString log = QString(m_stabilizeProcess.readAll());
-    int progress;
-    if (m_duration == 0) {
-        if (log.contains("Duration:")) {
-            QString data = log.section("Duration:", 1, 1).section(',', 0, 0).simplified();
-            QStringList numbers = data.split(':');
-            m_duration = numbers.at(0).toInt() * 3600 + numbers.at(1).toInt() * 60 + numbers.at(2).toDouble();
-            log_text->setHidden(true);
-            job_progress->setHidden(false);
-        }
-        else {
-            log_text->setHidden(false);
-            job_progress->setHidden(true);
-        }
-    }
-    else if (log.contains("time=")) {
-        QString time = log.section("time=", 1, 1).simplified().section(' ', 0, 0);
-        if (time.contains(':')) {
-            QStringList numbers = time.split(':');
-            progress = numbers.at(0).toInt() * 3600 + numbers.at(1).toInt() * 60 + numbers.at(2).toDouble();
-        }
-        else progress = (int) time.toDouble();
-        kDebug()<<"// PROGRESS: "<<progress<<", "<<m_duration;
-        job_progress->setValue((int) (100.0 * progress / m_duration));
-    }
-    //kDebug() << "//LOG: " << log;
-    log_text->setPlainText(log);
+       if (m_playlist){
+        job_progress->setValue((int) (100.0 * m_consumer->position()/m_playlist->get_out() ));
+               if (m_consumer->position()==m_playlist->get_out()){
+                       m_timer->stop();
+                       slotStabilizeFinished(true);
+               }
+       }
 }
 
-void ClipStabilize::slotStabilizeFinished(int exitCode, QProcess::ExitStatus exitStatus)
+void ClipStabilize::slotStabilizeFinished(bool success)
 {
     buttonBox->button(QDialogButtonBox::Abort)->setText(i18n("Close"));
     button_start->setEnabled(true);
     m_duration = 0;
+    if (m_stabilizeRun.isRunning()){
+        if (m_consumer){
+            m_consumer->stop();
+        }
+        m_stabilizeRun.waitForFinished();
+    }
 
-    if (exitCode == 0 && exitStatus == QProcess::NormalExit) {
-        log_text->setHtml(log_text->toPlainText() + "<br /><b>" + i18n("Transcoding finished."));
+    if (success) {
+        log_text->setHtml(log_text->toPlainText() + "<br /><b>" + i18n("Stabilize finished."));
         if (auto_add->isChecked()) {
             KUrl url;
             if (urls_list->count() > 0) {
-                QString params = ffmpeg_params->toPlainText().simplified();
-                QString extension = params.section("%1", 1, 1).section(' ', 0, 0);
-                url = KUrl(dest_url->url().path(KUrl::AddTrailingSlash) + source_url->url().fileName() + extension);
+                               url = KUrl(dest_url->url().path(KUrl::AddTrailingSlash) + source_url->url().fileName()+".mlt");
             } else url = dest_url->url();
             emit addClip(url);
         }
         if (urls_list->count() > 0 && m_urls.count() > 0) {
-            m_stabilizeProcess.close();
             slotStartStabilize();
             return;
         } else if (auto_close->isChecked()) accept();
     } else {
-        log_text->setHtml(log_text->toPlainText() + "<br /><b>" + i18n("Transcoding FAILED!"));
+        log_text->setHtml(log_text->toPlainText() + "<br /><b>" + i18n("Stabilizing FAILED!"));
     }
+       if (m_playlist){
+               free(m_playlist);
+               m_playlist=NULL;
+       }
+       if (m_consumer){
+               free(m_consumer);
+               m_consumer=NULL;
+       }
 
-    m_stabilizeProcess.close();
 }
 
-void ClipStabilize::slotUpdateParams(int ix)
+void ClipStabilize::slotUpdateParams()
 {
-    QString fileName = source_url->url().path();
-    if (ix != -1) {
-        QString params = profile_list->itemData(ix).toString();
-        ffmpeg_params->setPlainText(params.simplified());
-        QString desc = profile_list->itemData(ix, Qt::UserRole + 1).toString();
-        if (!desc.isEmpty()) {
-            transcode_info->setText(desc);
-            transcode_info->setHidden(false);
-        } else transcode_info->setHidden(true);
-    }
-    if (urls_list->count() == 0) {
-        QString newFile = ffmpeg_params->toPlainText().simplified().section(' ', -1).replace("%1", fileName);
-        dest_url->setUrl(KUrl(newFile));
-    }
+       for (int i=0;i<vbox->count();i++){
+               QWidget* w=vbox->itemAt(i)->widget();
+               QString name=w->objectName();
+               if (name !="" && m_ui_params.contains(name)){
+                       if (m_ui_params[name]["type"]=="int" || m_ui_params[name]["type"]=="double"){
+                               DoubleParameterWidget *dbl=(DoubleParameterWidget*)w;
+                               m_ui_params[name]["value"]=QString::number((double)(dbl->getValue()));
+                       }else if (m_ui_params[name]["type"]=="bool"){
+                               QCheckBox *ch=(QCheckBox*)w;
+                               m_ui_params[name]["value"]= ch->checkState() == Qt::Checked ? "1" : "0" ;
+                       }
+               }
+       }
+}
 
+bool ClipStabilize::autoAddClip() const
+{
+    return auto_add->isChecked();
 }
 
+void ClipStabilize::fillParameters(QStringList lst)
+{
+
+       m_ui_params.clear();
+       while (!lst.isEmpty()){
+               QString vallist=lst.takeFirst();
+               QStringList cont=vallist.split(",");
+               QString name=cont.takeFirst();
+               while (!cont.isEmpty()){
+                       QString valname=cont.takeFirst();
+                       QString val;
+                       if (!cont.isEmpty()){
+                               val=cont.takeFirst();
+                       }
+                       m_ui_params[name][valname]=val;
+               }
+       }
+
+}
+
+
 #include "clipstabilize.moc"