1 /***************************************************************************
2 * Copyright (C) 2008 by Jean-Baptiste Mardelle (jb@kdenlive.org) *
3 * Copyright (C) 2011 by Marco Gittler (marco@gitma.de) *
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., *
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA *
19 ***************************************************************************/
22 #include "clipstabilize.h"
25 #include <mlt++/Mlt.h>
26 #include "kdenlivesettings.h"
27 #include <KGlobalSettings>
28 #include <KMessageBox>
29 #include <QtConcurrentRun>
32 #include <KFileDialog>
35 ClipStabilize::ClipStabilize(KUrl::List urls, const QString ¶ms, Mlt::Filter* filter,QWidget * parent) :
36 QDialog(parent), m_urls(urls), m_duration(0),m_profile(NULL),m_consumer(NULL),m_playlist(NULL),m_filter(filter),vbox(NULL)
38 setFont(KGlobalSettings::toolBarFont());
40 setAttribute(Qt::WA_DeleteOnClose);
41 log_text->setHidden(true);
42 setWindowTitle(i18n("Stabilize Clip"));
43 auto_add->setText(i18np("Add clip to project", "Add clips to project", m_urls.count()));
44 m_profile = new Mlt::Profile(KdenliveSettings::current_profile().toUtf8().constData());
47 if (m_urls.count() == 1) {
48 QString fileName = m_urls.at(0).path(); //.section('.', 0, -1);
49 QString newFile = fileName.append(".mlt");
51 source_url->setUrl(m_urls.at(0));
52 dest_url->setMode(KFile::File);
53 dest_url->setUrl(dest);
54 dest_url->fileDialog()->setOperationMode(KFileDialog::Saving);
55 urls_list->setHidden(true);
56 connect(source_url, SIGNAL(textChanged(const QString &)), this, SLOT(slotUpdateParams()));
58 label_source->setHidden(true);
59 source_url->setHidden(true);
60 label_dest->setText(i18n("Destination folder"));
61 dest_url->setMode(KFile::Directory);
62 dest_url->setUrl(KUrl(m_urls.at(0).directory()));
63 dest_url->fileDialog()->setOperationMode(KFileDialog::Saving);
64 for (int i = 0; i < m_urls.count(); i++)
65 urls_list->addItem(m_urls.at(i).path());
67 if (!params.isEmpty()) {
68 label_profile->setHidden(true);
69 profile_list->setHidden(true);
70 //ffmpeg_params->setPlainText(params.simplified());
71 /*if (!description.isEmpty()) {
72 transcode_info->setText(description);
73 } else transcode_info->setHidden(true);*/
76 if (filtername=="videostab"){
78 ls << "shutterangle,type,int,value,0,min,0,max,100,factor,1";
80 }else if (filtername=="videostab2"){
82 ls << "accuracy,type,int,value,4,min,1,max,10,factor,1";
83 ls << "stepsize,type,int,value,6,min,0,max,100,factor,1";
84 ls << "algo,type,int,value,1,min,0,max,1,factor,1";
85 ls << "mincontrast,type,int,value,30,min,0,max,100,factor,100";
86 ls << "show,type,int,value,0,min,0,max,2,factor,1";
87 ls << "smoothing,type,int,value,10,min,0,max,100,factor,1";
88 ls << "maxshift,type,int,value,-1,min,-1,max,1000,factor,1";
89 ls << "maxangle,type,int,value,-1,min,-1,max,1000,factor,1";
90 ls << "crop,type,int,value,0,min,0,max,1,factor,1";
91 ls << "invert,type,int,value,0,min,0,max,1,factor,1";
92 ls << "realtive,type,int,value,1,min,0,max,1,factor,1";
93 ls << "zoom,type,int,value,0,min,-500,max,500,factor,1";
94 ls << "optzoom,type,int,value,1,min,0,max,1,factor,1";
95 ls << "sharpen,type,int,value,8,min,0,max,100,factor,10";
99 connect(button_start, SIGNAL(clicked()), this, SLOT(slotStartStabilize()));
100 connect(buttonBox,SIGNAL(rejected()), this, SLOT(slotAbortStabilize()));
102 m_timer=new QTimer(this);
103 connect(m_timer, SIGNAL(timeout()), this, SLOT(slotShowStabilizeInfo()));
105 vbox=new QVBoxLayout(optionsbox);
106 QHashIterator<QString,QHash<QString,QString> > hi(m_ui_params);
109 QHash<QString,QString> val=hi.value();
110 QLabel *l=new QLabel(hi.key(),this);
111 QSlider* s=new QSlider(Qt::Horizontal,this);
112 if (val["factor"].toInt()>1){
113 s->setTickInterval(val["factor"].toInt()/10);
115 s->setTickPosition(QSlider::TicksBelow);
116 s->setMaximum(val["max"].toInt());
117 s->setMinimum(val["min"].toInt());
118 s->setValue(val["value"].toInt());
119 s->setObjectName(hi.key());
120 connect(s,SIGNAL(sliderMoved(int)),this,SLOT(slotUpdateParams()));
121 QHBoxLayout *hbox=new QHBoxLayout();
124 vbox->addLayout(hbox);
130 ClipStabilize::~ClipStabilize()
132 /*if (m_stabilizeProcess.state() != QProcess::NotRunning) {
133 m_stabilizeProcess.close();
135 if (m_stabilizeRun.isRunning()){
139 m_stabilizeRun.waitForFinished();
141 if (m_profile) free (m_profile);
142 if (m_consumer) free (m_consumer);
143 if (m_playlist) free (m_playlist);
146 void ClipStabilize::slotStartStabilize()
148 if (m_consumer && !m_consumer->is_stopped()) {
152 QStringList parameters;
154 //QString params = ffmpeg_params->toPlainText().simplified();
155 if (urls_list->count() > 0) {
156 source_url->setUrl(m_urls.takeFirst());
157 destination = dest_url->url().path(KUrl::AddTrailingSlash)+ source_url->url().fileName()+".mlt";
158 QList<QListWidgetItem *> matching = urls_list->findItems(source_url->url().path(), Qt::MatchExactly);
159 if (matching.count() > 0) {
160 matching.at(0)->setFlags(Qt::ItemIsSelectable);
161 urls_list->setCurrentItem(matching.at(0));
164 destination = dest_url->url().path();
166 QString s_url = source_url->url().path();
168 if (QFile::exists(destination)) {
169 if (KMessageBox::questionYesNo(this, i18n("File %1 already exists.\nDo you want to overwrite it?", destination )) == KMessageBox::No) return;
171 buttonBox->button(QDialogButtonBox::Abort)->setText(i18n("Abort"));
174 m_playlist= new Mlt::Playlist;
175 Mlt::Filter filter(*m_profile,filtername.toUtf8().data());
176 QHashIterator <QString,QHash<QString,QString> > it(m_ui_params);
177 while (it.hasNext()){
180 it.key().toAscii().data(),
182 (double)(it.value()["value"].toDouble()/it.value()["factor"].toDouble())
186 Mlt::Producer p(*m_profile,s_url.toUtf8().data());
188 m_playlist->append(p);
189 m_playlist->attach(filter);
190 m_consumer= new Mlt::Consumer(*m_profile,"xml",destination.toUtf8().data());
191 m_consumer->set("all","1");
192 m_consumer->set("real_time","-2");
193 m_consumer->connect(*m_playlist);
194 m_stabilizeRun = QtConcurrent::run(this, &ClipStabilize::slotRunStabilize);
196 button_start->setEnabled(false);
202 void ClipStabilize::slotRunStabilize()
210 void ClipStabilize::slotAbortStabilize()
216 slotStabilizeFinished(false);
220 void ClipStabilize::slotShowStabilizeInfo()
223 job_progress->setValue((int) (100.0 * m_consumer->position()/m_playlist->get_out() ));
224 if (m_consumer->position()==m_playlist->get_out()){
226 slotStabilizeFinished(true);
231 void ClipStabilize::slotStabilizeFinished(bool success)
233 buttonBox->button(QDialogButtonBox::Abort)->setText(i18n("Close"));
234 button_start->setEnabled(true);
236 if (m_stabilizeRun.isRunning()){
240 m_stabilizeRun.waitForFinished();
244 log_text->setHtml(log_text->toPlainText() + "<br /><b>" + i18n("Stabilize finished."));
245 if (auto_add->isChecked()) {
247 if (urls_list->count() > 0) {
248 url = KUrl(dest_url->url().path(KUrl::AddTrailingSlash) + source_url->url().fileName()+".mlt");
249 } else url = dest_url->url();
252 if (urls_list->count() > 0 && m_urls.count() > 0) {
253 slotStartStabilize();
255 } else if (auto_close->isChecked()) accept();
257 log_text->setHtml(log_text->toPlainText() + "<br /><b>" + i18n("Stabilizing FAILED!"));
270 void ClipStabilize::slotUpdateParams()
272 for (int i=0;i<vbox->count();i++){
273 QLayout *bo=vbox->itemAt(i)->layout();
274 QWidget* w=bo->itemAt(1)->widget();
275 QString name=w->objectName();
276 if (name !="" && m_ui_params.contains(name)){
277 if (m_ui_params[name]["type"]=="int"){
278 QSlider *s=(QSlider*)w;
279 m_ui_params[name]["value"]=QString::number((double)(s->value()));
285 void ClipStabilize::fillParamaters(QStringList lst)
289 while (!lst.isEmpty()){
290 QString vallist=lst.takeFirst();
291 QStringList cont=vallist.split(",");
292 QString name=cont.takeFirst();
293 while (!cont.isEmpty()){
294 QString valname=cont.takeFirst();
296 if (!cont.isEmpty()){
297 val=cont.takeFirst();
299 m_ui_params[name][valname]=val;
306 #include "clipstabilize.moc"