X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fclipstabilize.cpp;h=6e0122ced3a761d613b9d38f89ec0e634b878e92;hb=4c3e738cacd82522bee069346966a6617161cb4f;hp=5c6db9cddbde9462d5d592ebbadaf9c3d66e60d4;hpb=3ffb17dcb7cdd3e1f387c99fd1fb200e11c446af;p=kdenlive diff --git a/src/clipstabilize.cpp b/src/clipstabilize.cpp index 5c6db9cd..6e0122ce 100644 --- a/src/clipstabilize.cpp +++ b/src/clipstabilize.cpp @@ -20,30 +20,51 @@ #include "clipstabilize.h" +#include "doubleparameterwidget.h" #include #include #include "kdenlivesettings.h" #include #include +#include +#include +#include +#include #include - -ClipStabilize::ClipStabilize(KUrl::List urls, const QString ¶ms, QWidget * parent) : - QDialog(parent), m_urls(urls), m_duration(0) +ClipStabilize::ClipStabilize(KUrl::List urls, const QString ¶ms, 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 ¶ms, 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 > hi(m_ui_params); + while(hi.hasNext()){ + hi.next(); + QHash 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 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 > 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::slotShowStabilizeInfo() +void ClipStabilize::slotAbortStabilize() { - 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); - } + if (m_consumer) + { + m_timer->stop(); + m_consumer->stop(); + slotStabilizeFinished(false); } - 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(); +} + +void ClipStabilize::slotShowStabilizeInfo() +{ + 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); } - else progress = (int) time.toDouble(); - kDebug()<<"// PROGRESS: "<setValue((int) (100.0 * progress / m_duration)); } - //kDebug() << "//LOG: " << log; - log_text->setPlainText(log); } -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() + "
" + i18n("Transcoding finished.")); + if (success) { + log_text->setHtml(log_text->toPlainText() + "
" + 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() + "
" + i18n("Transcoding FAILED!")); + log_text->setHtml(log_text->toPlainText() + "
" + 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); + for (int i=0;icount();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" ; + } + } } - if (urls_list->count() == 0) { - QString newFile = ffmpeg_params->toPlainText().simplified().section(' ', -1).replace("%1", fileName); - dest_url->setUrl(KUrl(newFile)); +} + +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"