]> git.sesse.net Git - kdenlive/blob - src/recmonitor.cpp
Fix firewire capture
[kdenlive] / src / recmonitor.cpp
1 /***************************************************************************
2  *   Copyright (C) 2007 by Jean-Baptiste Mardelle (jb@kdenlive.org)        *
3  *                                                                         *
4  *   This program is free software; you can redistribute it and/or modify  *
5  *   it under the terms of the GNU General Public License as published by  *
6  *   the Free Software Foundation; either version 2 of the License, or     *
7  *   (at your option) any later version.                                   *
8  *                                                                         *
9  *   This program is distributed in the hope that it will be useful,       *
10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
12  *   GNU General Public License for more details.                          *
13  *                                                                         *
14  *   You should have received a copy of the GNU General Public License     *
15  *   along with this program; if not, write to the                         *
16  *   Free Software Foundation, Inc.,                                       *
17  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA          *
18  ***************************************************************************/
19
20
21 #include "recmonitor.h"
22 #include "gentime.h"
23 #include "mltdevicecapture.h"
24 #include "kdenlivesettings.h"
25 #include "managecapturesdialog.h"
26 #include "monitormanager.h"
27 #include "monitor.h"
28 #include "profilesdialog.h"
29
30 #include <KDebug>
31 #include <KLocale>
32 #include <KStandardDirs>
33 #include <KComboBox>
34 #include <KIO/NetAccess>
35 #include <KFileItem>
36 #include <KMessageBox>
37 #include <KApplication>
38 #include <KDiskFreeSpaceInfo>
39 #include <QMouseEvent>
40 #include <QMenu>
41 #include <QToolButton>
42 #include <QFile>
43 #include <QDir>
44
45
46 RecMonitor::RecMonitor(Kdenlive::MONITORID name, MonitorManager *manager, QWidget *parent) :
47     AbstractMonitor(name, manager, parent),
48     m_isCapturing(false),
49     m_didCapture(false),
50     m_isPlaying(false),
51     m_captureDevice(NULL),
52     m_analyse(false)
53 {
54     setupUi(this);
55
56     //video_frame->setAttribute(Qt::WA_PaintOnScreen);
57     device_selector->setCurrentIndex(KdenliveSettings::defaultcapture());
58     connect(device_selector, SIGNAL(currentIndexChanged(int)), this, SLOT(slotVideoDeviceChanged(int)));
59
60     // Video widget holder
61     QVBoxLayout *l = new QVBoxLayout;
62     l->setContentsMargins(0, 0, 0, 0);
63     l->setSpacing(0);
64     l->addWidget(videoBox, 10);
65     video_frame->setLayout(l);
66     createVideoSurface();
67
68     QToolBar *toolbar = new QToolBar(this);
69     QHBoxLayout *layout = new QHBoxLayout;
70     layout->setContentsMargins(0, 0, 0, 0);
71     m_playIcon = KIcon("media-playback-start");
72     m_pauseIcon = KIcon("media-playback-pause");
73
74     m_discAction = toolbar->addAction(KIcon("network-connect"), i18n("Connect"));
75     connect(m_discAction, SIGNAL(triggered()), this, SLOT(slotDisconnect()));
76
77     m_rewAction = toolbar->addAction(KIcon("media-seek-backward"), i18n("Rewind"));
78     connect(m_rewAction, SIGNAL(triggered()), this, SLOT(slotRewind()));
79
80     m_playAction = toolbar->addAction(m_playIcon, i18n("Play"));
81     connect(m_playAction, SIGNAL(triggered()), this, SLOT(slotStartPreview()));
82
83     m_stopAction = toolbar->addAction(KIcon("media-playback-stop"), i18n("Stop"));
84     connect(m_stopAction, SIGNAL(triggered()), this, SLOT(slotStopCapture()));
85     m_stopAction->setEnabled(false);
86     m_fwdAction = toolbar->addAction(KIcon("media-seek-forward"), i18n("Forward"));
87     connect(m_fwdAction, SIGNAL(triggered()), this, SLOT(slotForward()));
88
89     m_recAction = toolbar->addAction(KIcon("media-record"), i18n("Record"));
90     connect(m_recAction, SIGNAL(triggered()), this, SLOT(slotRecord()));
91     m_recAction->setCheckable(true);
92
93     rec_options->setIcon(KIcon("system-run"));
94     QMenu *menu = new QMenu(this);
95     m_addCapturedClip = new QAction(i18n("Add Captured File to Project"), this);
96     m_addCapturedClip->setCheckable(true);
97     m_addCapturedClip->setChecked(true);
98     menu->addAction(m_addCapturedClip);
99
100     rec_audio->setChecked(KdenliveSettings::v4l_captureaudio());
101     rec_video->setChecked(KdenliveSettings::v4l_capturevideo());
102
103     m_previewSettings = new QAction(i18n("Recording Preview"), this);
104     m_previewSettings->setCheckable(true);
105
106
107     rec_options->setMenu(menu);
108     menu->addAction(m_previewSettings);
109
110     toolbar->addSeparator();
111
112     QAction *configAction = toolbar->addAction(KIcon("configure"), i18n("Configure"));
113     connect(configAction, SIGNAL(triggered()), this, SLOT(slotConfigure()));
114     configAction->setCheckable(false);
115
116     layout->addWidget(toolbar);
117     layout->addWidget(&m_logger);
118     layout->addWidget(&m_dvinfo);
119     m_logger.setMaxCount(10);
120     m_logger.setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
121     m_logger.setFrame(false);
122     //m_logger.setInsertPolicy(QComboBox::InsertAtTop);
123
124     m_freeSpace = new KCapacityBar(KCapacityBar::DrawTextInline, this);
125     m_freeSpace->setMaximumWidth(150);
126     QFontMetricsF fontMetrics(font());
127     m_freeSpace->setMaximumHeight(fontMetrics.height() * 1.2);
128     slotUpdateFreeSpace();
129     layout->addWidget(m_freeSpace);
130     connect(&m_spaceTimer, SIGNAL(timeout()), this, SLOT(slotUpdateFreeSpace()));
131     m_spaceTimer.setInterval(30000);
132     m_spaceTimer.setSingleShot(false);
133
134     control_frame_firewire->setLayout(layout);
135     m_displayProcess = new QProcess;
136     m_captureProcess = new QProcess;
137
138     connect(m_captureProcess, SIGNAL(stateChanged(QProcess::ProcessState)), this, SLOT(slotProcessStatus(QProcess::ProcessState)));
139     connect(m_captureProcess, SIGNAL(readyReadStandardError()), this, SLOT(slotReadDvgrabInfo()));
140     
141     QString videoDriver = KdenliveSettings::videodrivername();
142 #if QT_VERSION >= 0x040600
143     QProcessEnvironment env = QProcessEnvironment::systemEnvironment();
144     env.insert("SDL_WINDOWID", QString::number(videoSurface->winId()));
145     if (!videoDriver.isEmpty()) {
146         if (videoDriver == "x11_noaccel") {
147             env.insert("SDL_VIDEO_YUV_HWACCEL", "0");
148             env.insert("SDL_VIDEODRIVER", "x11");
149         } else env.insert("SDL_VIDEODRIVER", videoDriver);
150     }
151     m_displayProcess->setProcessEnvironment(env);
152 #else
153     QStringList env = QProcess::systemEnvironment();
154     env << "SDL_WINDOWID=" + QString::number(videoSurface->winId());
155     if (!videoDriver.isEmpty()) {
156         if (videoDriver == "x11_noaccel") {
157             env << "SDL_VIDEO_YUV_HWACCEL=0";
158             env << "SDL_VIDEODRIVER=x11";
159         } else env << "SDL_VIDEODRIVER=" + videoDriver;
160     }
161     m_displayProcess->setEnvironment(env);
162 #endif
163
164     setenv("SDL_VIDEO_ALLOW_SCREENSAVER", "1", 1);
165
166     kDebug() << "/////// BUILDING MONITOR, ID: " << videoSurface->winId();
167     slotVideoDeviceChanged(device_selector->currentIndex());
168     m_previewSettings->setChecked(KdenliveSettings::enable_recording_preview());
169     connect(m_previewSettings, SIGNAL(triggered(bool)), this, SLOT(slotChangeRecordingPreview(bool)));
170 }
171
172 RecMonitor::~RecMonitor()
173 {
174     m_spaceTimer.stop();
175     delete m_captureProcess;
176     delete m_displayProcess;
177     if (m_captureDevice) delete m_captureDevice;
178 }
179
180 void RecMonitor::mouseDoubleClickEvent(QMouseEvent * event)
181 {
182     if (!KdenliveSettings::openglmonitors() && videoBox && videoBox->isVisible()) {
183         videoBox->switchFullScreen();
184         event->accept();
185     }
186 }
187
188 void RecMonitor::slotSwitchFullScreen()
189 {
190     videoBox->switchFullScreen();
191 }
192
193 void RecMonitor::stop()
194 {
195     slotStopCapture();
196 }
197
198 void RecMonitor::start()
199 {
200     //slotStartPreview(true);
201 }
202
203 void RecMonitor::slotConfigure()
204 {
205     emit showConfigDialog(4, device_selector->currentIndex());
206 }
207
208 void RecMonitor::slotUpdateCaptureFolder(const QString &currentProjectFolder)
209 {
210     if (KdenliveSettings::capturetoprojectfolder()) m_capturePath = currentProjectFolder;
211     else m_capturePath = KdenliveSettings::capturefolder();
212     if (m_captureProcess) m_captureProcess->setWorkingDirectory(m_capturePath);
213     if (m_captureProcess->state() != QProcess::NotRunning) {
214         if (device_selector->currentIndex() == FIREWIRE)
215             KMessageBox::information(this, i18n("You need to disconnect and reconnect in the capture monitor to apply your changes"), i18n("Capturing"));
216         else KMessageBox::information(this, i18n("You need to stop capture before your changes can be applied"), i18n("Capturing"));
217     } else slotVideoDeviceChanged(device_selector->currentIndex());
218
219     // update free space info
220     slotUpdateFreeSpace();
221 }
222
223 void RecMonitor::slotVideoDeviceChanged(int ix)
224 {
225     QString capturefile;
226     QString capturename;
227     m_previewSettings->setEnabled(ix == VIDEO4LINUX || ix == BLACKMAGIC);
228     rec_audio->setVisible(ix == VIDEO4LINUX);
229     rec_video->setVisible(ix == VIDEO4LINUX);
230     m_fwdAction->setVisible(ix == FIREWIRE);
231     m_discAction->setVisible(ix == FIREWIRE);
232     m_rewAction->setVisible(ix == FIREWIRE);
233     m_recAction->setEnabled(ix != FIREWIRE);
234     m_logger.setVisible(ix == BLACKMAGIC);
235     if (m_captureDevice) {
236         // MLT capture still running, abort
237         m_monitorManager->clearScopeSource();
238         m_captureDevice->stop();
239         delete m_captureDevice;
240         m_captureDevice = NULL;
241     }
242
243     // The m_videoBox container has to be shown once before the MLT consumer is build, or preview will fail
244     videoBox->setHidden(ix != VIDEO4LINUX && ix != BLACKMAGIC);
245     videoBox->setHidden(true);
246     switch (ix) {
247     case SCREENGRAB:
248         m_discAction->setEnabled(false);
249         m_rewAction->setEnabled(false);
250         m_fwdAction->setEnabled(false);
251         m_stopAction->setEnabled(false);
252         m_playAction->setEnabled(false);
253         if (KdenliveSettings::rmd_path().isEmpty()) {
254             QString rmdpath = KStandardDirs::findExe("recordmydesktop");
255             if (rmdpath.isEmpty()) video_frame->setPixmap(mergeSideBySide(KIcon("dialog-warning").pixmap(QSize(50, 50)), i18n("Recordmydesktop utility not found,\n please install it for screen grabs")));
256             else KdenliveSettings::setRmd_path(rmdpath);
257         }
258         if (!KdenliveSettings::rmd_path().isEmpty()) video_frame->setPixmap(mergeSideBySide(KIcon("video-display").pixmap(QSize(50, 50)), i18n("Press record button\nto start screen capture\nFiles will be saved in:\n%1", m_capturePath)));
259         //video_frame->setText(i18n("Press record button\nto start screen capture"));
260         break;
261     case VIDEO4LINUX:
262         m_stopAction->setEnabled(false);
263         m_playAction->setEnabled(true);
264         checkDeviceAvailability();
265         break;
266     case BLACKMAGIC:
267         m_stopAction->setEnabled(false);
268         m_playAction->setEnabled(true);
269         capturefile = m_capturePath;
270         if (!capturefile.endsWith("/")) capturefile.append("/");
271         capturename = KdenliveSettings::decklink_filename();
272         capturename.append("xxx.");
273         capturename.append(KdenliveSettings::decklink_extension());
274         capturefile.append(capturename);
275         video_frame->setPixmap(mergeSideBySide(KIcon("camera-photo").pixmap(QSize(50, 50)), i18n("Plug your camcorder and\npress play button\nto start preview.\nFiles will be saved in:\n%1", capturefile)));
276         break;
277     default: // FIREWIRE
278         m_discAction->setEnabled(true);
279         m_stopAction->setEnabled(false);
280         m_playAction->setEnabled(false);
281         m_rewAction->setEnabled(false);
282         m_fwdAction->setEnabled(false);
283
284         // Check that dvgab is available
285         if (KdenliveSettings::dvgrab_path().isEmpty()) {
286             QString dvgrabpath = KStandardDirs::findExe("dvgrab");
287             if (dvgrabpath.isEmpty()) video_frame->setPixmap(mergeSideBySide(KIcon("dialog-warning").pixmap(QSize(50, 50)), i18n("dvgrab utility not found,\n please install it for firewire capture")));
288             else KdenliveSettings::setDvgrab_path(dvgrabpath);
289         } else {
290             // Show capture info
291             capturefile = m_capturePath;
292             if (!capturefile.endsWith("/")) capturefile.append("/");
293             capturename = KdenliveSettings::dvgrabfilename();
294             if (capturename.isEmpty()) capturename = "capture";
295             QString extension;
296             switch (KdenliveSettings::firewireformat()) {
297             case 0:
298                 extension = ".dv";
299                 break;
300             case 1:
301             case 2:
302                 extension = ".avi";
303                 break;
304             case 3:
305                 extension = ".m2t";
306                 break;
307             }
308             capturename.append("xxx" + extension);
309             capturefile.append(capturename);
310             video_frame->setPixmap(mergeSideBySide(KIcon("network-connect").pixmap(QSize(50, 50)), i18n("Plug your camcorder and\npress connect button\nto initialize connection\nFiles will be saved in:\n%1", capturefile)));
311         }
312         break;
313     }
314 }
315
316 void RecMonitor::slotSetInfoMessage(const QString &message)
317 {
318     m_logger.insertItem(0, message);
319 }
320
321 QPixmap RecMonitor::mergeSideBySide(const QPixmap& pix, const QString &txt)
322 {
323     QPainter p;
324     QRect r = QApplication::fontMetrics().boundingRect(QRect(0, 0, video_frame->width(), video_frame->height()), Qt::AlignLeft, txt);
325     int strWidth = r.width();
326     int strHeight = r.height();
327     int pixWidth = pix.width();
328     int pixHeight = pix.height();
329     QPixmap res(strWidth + 8 + pixWidth, qMax(strHeight, pixHeight));
330     res.fill(Qt::transparent);
331     p.begin(&res);
332     p.drawPixmap(0, 0, pix);
333     p.setPen(kapp->palette().text().color());
334     p.drawText(QRect(pixWidth + 8, 0, strWidth, strHeight), 0, txt);
335     p.end();
336     return res;
337 }
338
339
340 void RecMonitor::checkDeviceAvailability()
341 {
342     if (!KIO::NetAccess::exists(KUrl(KdenliveSettings::video4vdevice()), KIO::NetAccess::SourceSide , this)) {
343         m_playAction->setEnabled(false);
344         m_recAction->setEnabled(false);
345         video_frame->setPixmap(mergeSideBySide(KIcon("camera-web").pixmap(QSize(50, 50)), i18n("Cannot read from device %1\nPlease check drivers and access rights.", KdenliveSettings::video4vdevice())));
346     } else {
347         video_frame->setPixmap(mergeSideBySide(KIcon("camera-web").pixmap(QSize(50, 50)), i18n("Press play or record button\nto start video capture\nFiles will be saved in:\n%1", m_capturePath)));
348     }
349 }
350
351 void RecMonitor::slotDisconnect()
352 {
353     if (m_captureProcess->state() == QProcess::NotRunning) {
354         m_captureTime = KDateTime::currentLocalDateTime();
355         kDebug() << "CURRENT TIME: " << m_captureTime.toString();       
356         m_didCapture = false;
357         slotStartPreview(false);
358         m_discAction->setIcon(KIcon("network-disconnect"));
359         m_discAction->setText(i18n("Disconnect"));
360         m_recAction->setEnabled(true);
361         m_stopAction->setEnabled(true);
362         m_playAction->setEnabled(true);
363         m_rewAction->setEnabled(true);
364         m_fwdAction->setEnabled(true);
365     } else {
366         m_captureProcess->write("q", 1);
367         QTimer::singleShot(1000, m_captureProcess, SLOT(kill()));
368         if (m_didCapture) manageCapturedFiles();
369         m_didCapture = false;
370     }
371 }
372
373 void RecMonitor::slotRewind()
374 {
375     m_captureProcess->write("a", 1);
376 }
377
378 void RecMonitor::slotForward()
379 {
380     m_captureProcess->write("z", 1);
381 }
382
383 void RecMonitor::slotStopCapture()
384 {
385     // stop capture
386     if (!m_isCapturing && !m_isPlaying) return;
387     videoBox->setHidden(true);
388     rec_audio->setEnabled(true);
389     rec_video->setEnabled(true);
390     switch (device_selector->currentIndex()) {
391     case FIREWIRE:
392         m_captureProcess->write("\e", 2);
393         m_playAction->setIcon(m_playIcon);
394         m_isPlaying = false;
395         break;
396     case SCREENGRAB:
397         m_captureProcess->write("q\n", 3);
398         m_captureProcess->terminate();
399         video_frame->setText(i18n("Encoding captured video..."));
400         QTimer::singleShot(1000, m_captureProcess, SLOT(kill()));
401         break;
402     case VIDEO4LINUX:
403     case BLACKMAGIC:
404         if (m_captureDevice) {
405             m_captureDevice->stop();
406         }
407         m_previewSettings->setEnabled(true);
408         m_isCapturing = false;
409         m_isPlaying = false;
410         m_playAction->setEnabled(true);
411         m_stopAction->setEnabled(false);
412         m_recAction->setEnabled(true);
413         slotSetInfoMessage(i18n("Capture stopped"));
414         m_isCapturing = false;
415         m_recAction->setChecked(false);
416         if (m_addCapturedClip->isChecked() && !m_captureFile.isEmpty() && QFile::exists(m_captureFile.path())) {
417             emit addProjectClip(m_captureFile);
418             m_captureFile.clear();
419         }
420         break;
421     default:
422         break;
423     }
424 }
425
426 void RecMonitor::slotStartPreview(bool play)
427 {
428     if (m_captureProcess->state() != QProcess::NotRunning) {
429         if (device_selector->currentIndex() == FIREWIRE) {
430             videoBox->setHidden(false);
431             if (m_isPlaying) {
432                 m_captureProcess->write("k", 1);
433                 //captureProcess->write("\e", 2);
434                 m_playAction->setIcon(m_playIcon);
435                 m_isPlaying = false;
436             } else {
437                 m_captureProcess->write("p", 1);
438                 m_playAction->setIcon(m_pauseIcon);
439                 m_isPlaying = true;
440             }
441         }
442         return;
443     }
444     slotActivateMonitor();
445     if (m_isPlaying) return;
446     m_captureArgs.clear();
447     m_displayArgs.clear();
448     m_isPlaying = false;
449     QString capturename = KdenliveSettings::dvgrabfilename();
450     QString path;
451     MltVideoProfile profile;
452     QString producer;
453     QStringList dvargs = KdenliveSettings::dvgrabextra().simplified().split(" ", QString::SkipEmptyParts);
454     int ix = device_selector->currentIndex();
455     videoBox->setHidden(ix != VIDEO4LINUX && ix != BLACKMAGIC && ix != FIREWIRE);
456     switch (ix) {
457     case FIREWIRE:
458         switch (KdenliveSettings::firewireformat()) {
459         case 0:
460             // RAW DV CAPTURE
461             m_captureArgs << "--format" << "raw";
462             m_displayArgs << "-f" << "dv";
463             break;
464         case 1:
465             // DV type 1
466             m_captureArgs << "--format" << "dv1";
467             m_displayArgs << "-f" << "dv";
468             break;
469         case 2:
470             // DV type 2
471             m_captureArgs << "--format" << "dv2";
472             m_displayArgs << "-f" << "dv";
473             break;
474         case 3:
475             // HDV CAPTURE
476             m_captureArgs << "--format" << "hdv";
477             m_displayArgs << "-f" << "mpegts";
478             break;
479         }
480         if (KdenliveSettings::firewireautosplit()) m_captureArgs << "--autosplit";
481         if (KdenliveSettings::firewiretimestamp()) m_captureArgs << "--timestamp";
482         if (!dvargs.isEmpty()) {
483             m_captureArgs << dvargs;
484         }
485         m_captureArgs << "-i";
486         if (capturename.isEmpty()) capturename = "capture";
487         m_captureArgs << capturename << "-";
488
489         m_displayArgs << "-x" << QString::number(video_frame->width()) << "-y" << QString::number(video_frame->height()) << "-noframedrop" << "-";
490
491         m_captureProcess->setStandardOutputProcess(m_displayProcess);
492         m_captureProcess->setWorkingDirectory(m_capturePath);
493         kDebug() << "Capture: Running dvgrab " << m_captureArgs.join(" ");
494
495         m_captureProcess->start(KdenliveSettings::dvgrab_path(), m_captureArgs);
496         if (play) m_captureProcess->write(" ", 1);
497         m_discAction->setEnabled(true);
498         break;
499     case VIDEO4LINUX:
500         path = KStandardDirs::locateLocal("appdata", "profiles/video4linux");
501         buildMltDevice(path);
502         profile = ProfilesDialog::getVideoProfile(path);
503         producer = getV4lXmlPlaylist(profile);
504
505         //producer = QString("avformat-novalidate:video4linux2:%1?width:%2&height:%3&frame_rate:%4").arg(KdenliveSettings::video4vdevice()).arg(profile.width).arg(profile.height).arg((double) profile.frame_rate_num / profile.frame_rate_den);
506         if (!m_captureDevice->slotStartPreview(producer, true)) {
507             // v4l capture failed to start
508             video_frame->setText(i18n("Failed to start Video4Linux,\ncheck your parameters..."));
509             videoBox->setHidden(true);
510
511         } else {
512             m_playAction->setEnabled(false);
513             m_stopAction->setEnabled(true);
514             m_isPlaying = true;
515         }
516
517         break;
518     case BLACKMAGIC:
519         path = KdenliveSettings::current_profile();
520         slotActivateMonitor();
521         buildMltDevice(path);
522         producer = QString("decklink:%1").arg(KdenliveSettings::decklink_capturedevice());
523         if (!m_captureDevice->slotStartPreview(producer)) {
524             // v4l capture failed to start
525             video_frame->setText(i18n("Failed to start Decklink,\ncheck your parameters..."));
526             videoBox->setHidden(true);
527
528         } else {
529             m_playAction->setEnabled(false);
530             m_stopAction->setEnabled(true);
531             m_isPlaying = true;
532         }
533         break;
534     default:
535         break;
536     }
537
538     rec_audio->setEnabled(false);
539     rec_video->setEnabled(false);
540
541     if (device_selector->currentIndex() == FIREWIRE) {
542         kDebug() << "Capture: Running ffplay " << m_displayArgs.join(" ");
543         m_displayProcess->start("ffplay", m_displayArgs);
544         //video_frame->setText(i18n("Initialising..."));
545     } else {
546         // do something when starting screen grab
547     }
548 }
549
550 void RecMonitor::slotRecord()
551 {
552     rec_audio->setEnabled(false);
553     rec_video->setEnabled(false);
554
555     if (m_captureProcess->state() == QProcess::NotRunning && device_selector->currentIndex() == FIREWIRE) {
556         slotStartPreview();
557     }
558     if (m_isCapturing) {
559         // User stopped capture
560         slotStopCapture();
561         return;
562     } else if (device_selector->currentIndex() == FIREWIRE) {
563         m_isCapturing = true;
564         m_didCapture = true;
565         m_captureProcess->write("c\n", 3);
566         m_spaceTimer.start();
567         return;
568     }
569     if (m_captureProcess->state() == QProcess::NotRunning) {
570         m_logger.clear();
571         m_recAction->setChecked(true);
572         QString extension = "mpg";
573         if (device_selector->currentIndex() == SCREENGRAB) extension = "ogv"; //KdenliveSettings::screengrabextension();
574         else if (device_selector->currentIndex() == VIDEO4LINUX) {
575             // TODO: when recording audio only, allow configuration?
576             if (!rec_video->isChecked()) extension = "wav";
577             else extension = KdenliveSettings::v4l_extension();
578         }
579         else if (device_selector->currentIndex() == BLACKMAGIC) extension = KdenliveSettings::decklink_extension();
580         QString path = KUrl(m_capturePath).path(KUrl::AddTrailingSlash) + "capture0000." + extension;
581         int i = 1;
582         while (QFile::exists(path)) {
583             QString num = QString::number(i).rightJustified(4, '0', false);
584             path = KUrl(m_capturePath).path(KUrl::AddTrailingSlash) + "capture" + num + '.' + extension;
585             i++;
586         }
587         m_captureFile = KUrl(path);
588
589         m_captureArgs.clear();
590         m_displayArgs.clear();
591         QString args;
592         QString playlist;
593         QString v4lparameters;
594         MltVideoProfile profile;
595         bool showPreview;
596         QString capturename = KdenliveSettings::dvgrabfilename();
597         if (capturename.isEmpty()) capturename = "capture";
598
599         switch (device_selector->currentIndex()) {
600         case VIDEO4LINUX:
601             slotActivateMonitor();
602             path = KStandardDirs::locateLocal("appdata", "profiles/video4linux");
603             profile = ProfilesDialog::getVideoProfile(path);
604             //m_videoBox->setRatio((double) profile.display_aspect_num / profile.display_aspect_den);
605             buildMltDevice(path);
606             playlist = getV4lXmlPlaylist(profile);
607
608             v4lparameters = KdenliveSettings::v4l_parameters();
609
610             // TODO: when recording audio only, allow param configuration?
611             if (!rec_video->isChecked()) v4lparameters.clear();
612
613             // Add alsa audio capture
614             if (!rec_audio->isChecked()) {
615                 // if we do not want audio, make sure that we don't have audio encoding parameters
616                 // this is required otherwise the MLT avformat consumer will not close properly
617                 if (v4lparameters.contains("acodec")) {
618                     QString endParam = v4lparameters.section("acodec", 1);
619                     int vcodec = endParam.indexOf(" vcodec");
620                     int format = endParam.indexOf(" f=");
621                     int cutPosition = -1;
622                     if (vcodec > -1) {
623                         if (format  > -1) {
624                             cutPosition = qMin(vcodec, format);
625                         }
626                         else cutPosition = vcodec;
627                     }
628                     else if (format  > -1) {
629                         cutPosition = format;
630                     }
631                     else {
632                         // nothing interesting in end params
633                         endParam.clear();
634                     }
635                     if (cutPosition > -1) {
636                         endParam.remove(0, cutPosition);
637                     }
638                     v4lparameters = QString(v4lparameters.section("acodec", 0, 0) + "an=1 " + endParam).simplified();
639                 }
640             }
641
642             showPreview = m_previewSettings->isChecked();
643             if (!rec_video->isChecked()) showPreview = false;
644
645             if (m_captureDevice->slotStartCapture(v4lparameters, m_captureFile.path(), playlist, showPreview)) {
646                 videoBox->setHidden(false);
647                 m_isCapturing = true;
648                 m_recAction->setEnabled(false);
649                 m_stopAction->setEnabled(true);
650                 m_previewSettings->setEnabled(false);
651             }
652             else {
653                 video_frame->setText(i18n("Failed to start Video4Linux,\ncheck your parameters..."));
654                 videoBox->setHidden(true);
655                 m_isCapturing = false;
656             }
657             break;
658
659         case BLACKMAGIC:
660             slotActivateMonitor();
661             path = KdenliveSettings::current_profile();
662             profile = ProfilesDialog::getVideoProfile(path);
663             //m_videoBox->setRatio((double) profile.display_aspect_num / profile.display_aspect_den);
664             buildMltDevice(path);
665
666             playlist = QString("<producer id=\"producer0\" in=\"0\" out=\"99999\"><property name=\"mlt_type\">producer</property><property name=\"length\">100000</property><property name=\"eof\">pause</property><property name=\"resource\">%1</property><property name=\"mlt_service\">decklink</property></producer>").arg(KdenliveSettings::decklink_capturedevice());
667
668             if (m_captureDevice->slotStartCapture(KdenliveSettings::decklink_parameters(), m_captureFile.path(), QString("decklink:%1").arg(KdenliveSettings::decklink_capturedevice()), m_previewSettings->isChecked(), false)) {
669                 videoBox->setHidden(false);
670                 m_isCapturing = true;
671                 slotSetInfoMessage(i18n("Capturing to %1", m_captureFile.fileName()));
672                 m_recAction->setEnabled(false);
673                 m_stopAction->setEnabled(true);
674                 m_previewSettings->setEnabled(false);
675             }
676             else {
677                 video_frame->setText(i18n("Failed to start Decklink,\ncheck your parameters..."));
678                 slotSetInfoMessage(i18n("Failed to start capture"));
679                 videoBox->setHidden(true);
680                 m_isCapturing = false;
681             }
682             break;
683
684         case SCREENGRAB:
685             switch (KdenliveSettings::rmd_capture_type()) {
686             case 0:
687                 // Full screen capture, nothing special to do
688                 break;
689             default:
690                 // Region capture
691                 m_captureArgs << "--width" << QString::number(KdenliveSettings::rmd_width()) << "--height" << QString::number(KdenliveSettings::rmd_height());
692                 if (!KdenliveSettings::rmd_follow_mouse()) {
693                     m_captureArgs << "-x" << QString::number(KdenliveSettings::rmd_offsetx()) << "-y" << QString::number(KdenliveSettings::rmd_offsety());
694                 } else {
695                     m_captureArgs << "--follow-mouse";
696                     if (KdenliveSettings::rmd_hide_frame()) m_captureArgs << "--no-frame";
697                 }
698                 break;
699             }
700             if (KdenliveSettings::rmd_hide_mouse()) m_captureArgs << "--no-cursor";
701             m_isCapturing = true;
702             if (KdenliveSettings::rmd_capture_audio()) {
703                 m_captureArgs << "--freq" << KdenliveSettings::rmd_freq();
704                 m_captureArgs << "--channels" << QString::number(KdenliveSettings::rmd_audio_channels());
705                 if (KdenliveSettings::rmd_use_jack()) {
706                     m_captureArgs << "--use-jack";
707                     QStringList ports = KdenliveSettings::rmd_jackports().split(" ", QString::SkipEmptyParts);
708                     for (int i = 0; i < ports.count(); ++i) {
709                         m_captureArgs << ports.at(i);
710                     }
711                     if (KdenliveSettings::rmd_jack_buffer() > 0.0)
712                         m_captureArgs << "--ring-buffer-size" << QString::number(KdenliveSettings::rmd_jack_buffer());
713                 } else {
714                     if (!KdenliveSettings::rmd_alsadevicename().isEmpty())
715                         m_captureArgs << "--device" << KdenliveSettings::rmd_alsadevicename();
716                     if (KdenliveSettings::rmd_alsa_buffer() > 0)
717                         m_captureArgs << "--buffer-size" << QString::number(KdenliveSettings::rmd_alsa_buffer());
718                 }
719             } else m_captureArgs << "--no-sound";
720
721             if (KdenliveSettings::rmd_fullshots()) m_captureArgs << "--full-shots";
722             m_captureArgs << "--v_bitrate" << QString::number(KdenliveSettings::rmd_bitrate());
723             m_captureArgs << "--v_quality" << QString::number(KdenliveSettings::rmd_quality());
724             m_captureArgs << "--workdir" << KdenliveSettings::currenttmpfolder();
725             m_captureArgs << "--fps" << QString::number(KdenliveSettings::rmd_fps()) << "-o" << m_captureFile.path();
726             m_captureProcess->start(KdenliveSettings::rmd_path(), m_captureArgs);
727             kDebug() << "// RecordMyDesktop params: " << m_captureArgs;
728             break;
729         default:
730             break;
731         }
732
733
734         if (device_selector->currentIndex() == FIREWIRE) {
735             m_isCapturing = true;
736             kDebug() << "Capture: Running ffplay " << m_displayArgs.join(" ");
737             m_displayProcess->start("ffplay", m_displayArgs);
738             video_frame->setText(i18n("Initialising..."));
739         }
740     } else {
741         // stop capture
742         m_displayProcess->kill();
743         //captureProcess->kill();
744         QTimer::singleShot(1000, this, SLOT(slotRecord()));
745     }
746 }
747
748 const QString RecMonitor::getV4lXmlPlaylist(MltVideoProfile profile) {
749
750     QString playlist = QString("<mlt title=\"capture\" LC_NUMERIC=\"C\"><profile description=\"v4l\" width=\"%1\" height=\"%2\" progressive=\"%3\" sample_aspect_num=\"%4\" sample_aspect_den=\"%5\" display_aspect_num=\"%6\" display_aspect_den=\"%7\" frame_rate_num=\"%8\" frame_rate_den=\"%9\" colorspace=\"%10\"/>").arg(profile.width).arg(profile.height).arg(profile.progressive).arg(profile.sample_aspect_num).arg(profile.sample_aspect_den).arg(profile.display_aspect_num).arg(profile.display_aspect_den).arg(profile.frame_rate_num).arg(profile.frame_rate_den).arg(profile.colorspace);
751
752     if (rec_video->isChecked()) {
753         playlist.append(QString("<producer id=\"producer0\" in=\"0\" out=\"999999\"><property name=\"mlt_type\">producer</property><property name=\"length\">1000000</property><property name=\"eof\">loop</property><property name=\"resource\">video4linux2:%1?width:%2&amp;height:%3&amp;frame_rate:%4</property><property name=\"mlt_service\">avformat-novalidate</property></producer><playlist id=\"playlist0\"><entry producer=\"producer0\" in=\"0\" out=\"999999\"/></playlist>").arg(KdenliveSettings::video4vdevice()).arg(profile.width).arg(profile.height).arg((double) profile.frame_rate_num / profile.frame_rate_den));
754     }
755
756     if (rec_audio->isChecked()) {
757         playlist.append(QString("<producer id=\"producer1\" in=\"0\" out=\"999999\"><property name=\"mlt_type\">producer</property><property name=\"length\">1000000</property><property name=\"eof\">loop</property><property name=\"resource\">alsa:%5</property><property name=\"audio_index\">0</property><property name=\"video_index\">-1</property><property name=\"mlt_service\">avformat-novalidate</property></producer><playlist id=\"playlist1\"><entry producer=\"producer1\" in=\"0\" out=\"999999\"/></playlist>").arg(KdenliveSettings::v4l_alsadevicename()));
758     }
759     playlist.append("<tractor id=\"tractor0\" title=\"video0\" global_feed=\"1\" in=\"0\" out=\"999999\">");
760     if (rec_video->isChecked()) playlist.append("<track producer=\"playlist0\"/>");
761     if (rec_audio->isChecked()) playlist.append("<track producer=\"playlist1\"/>");
762     playlist.append("</tractor></mlt>");
763
764     return playlist;
765 }
766
767 /*
768 void RecMonitor::slotStartGrab(const QRect &rect) {
769     rgnGrab->deleteLater();
770     QApplication::restoreOverrideCursor();
771     show();
772     if (rect.isNull()) return;
773     int width = rect.width();
774     int height = rect.height();
775     if (width % 2 != 0) width--;
776     if (height % 2 != 0) height--;
777     QString args = KdenliveSettings::screengrabcapture().replace("%size", QString::number(width) + "x" + QString::number(height)).replace("%offset", "+" + QString::number(rect.x()) + "," + QString::number(rect.y()));
778     if (KdenliveSettings::screengrabenableaudio()) {
779         // also capture audio
780         if (KdenliveSettings::useosscapture()) m_captureArgs << KdenliveSettings::screengrabosscapture().simplified().split(' ');
781         else m_captureArgs << KdenliveSettings::screengrabalsacapture2().simplified().split(' ');
782     }
783     m_captureArgs << args.simplified().split(' ') << KdenliveSettings::screengrabencoding().simplified().split(' ') << m_captureFile.path();
784     m_isCapturing = true;
785     video_frame->setText(i18n("Capturing..."));
786     if (KdenliveSettings::screengrabenableaudio() && !KdenliveSettings::useosscapture()) {
787         QStringList alsaArgs = KdenliveSettings::screengrabalsacapture().simplified().split(' ');
788         alsaProcess->setStandardOutputProcess(captureProcess);
789         kDebug() << "Capture: Running arecord " << alsaArgs.join(" ");
790         alsaProcess->start("arecord", alsaArgs);
791     }
792     kDebug() << "Capture: Running ffmpeg " << m_captureArgs.join(" ");
793     captureProcess->start("ffmpeg", m_captureArgs);
794 }*/
795
796 void RecMonitor::slotProcessStatus(QProcess::ProcessState status)
797 {
798     if (status == QProcess::NotRunning) {
799         m_displayProcess->kill();
800         if (m_isCapturing && device_selector->currentIndex() != FIREWIRE)
801             if (m_addCapturedClip->isChecked() && !m_captureFile.isEmpty() && QFile::exists(m_captureFile.path())) {
802                 emit addProjectClip(m_captureFile);
803                 m_captureFile.clear();
804             }
805         if (device_selector->currentIndex() == FIREWIRE) {
806             m_discAction->setIcon(KIcon("network-connect"));
807             m_discAction->setText(i18n("Connect"));
808             m_playAction->setEnabled(false);
809             m_rewAction->setEnabled(false);
810             m_fwdAction->setEnabled(false);
811             m_recAction->setEnabled(false);
812         }
813         m_isPlaying = false;
814         m_playAction->setIcon(m_playIcon);
815         m_recAction->setChecked(false);
816         m_stopAction->setEnabled(false);
817         device_selector->setEnabled(true);
818         if (m_captureProcess && m_captureProcess->exitStatus() == QProcess::CrashExit) {
819             video_frame->setText(i18n("Capture crashed, please check your parameters"));
820         } else {
821             if (device_selector->currentIndex() != SCREENGRAB) {
822                 video_frame->setText(i18n("Not connected"));
823             } else {
824                 if (m_captureProcess->exitCode() != 0) {
825                     video_frame->setText(i18n("Capture crashed, please check your parameters\nRecordMyDesktop exit code: %1", QString::number(m_captureProcess->exitCode())));
826                 } else {
827                     video_frame->setPixmap(mergeSideBySide(KIcon("video-display").pixmap(QSize(50, 50)), i18n("Press record button\nto start screen capture\nFiles will be saved in:\n%1", m_capturePath)));
828                 }
829             }
830         }
831         m_isCapturing = false;
832
833         m_spaceTimer.stop();
834         // update free space info
835         slotUpdateFreeSpace();
836     } else {
837         if (device_selector->currentIndex() != SCREENGRAB) m_stopAction->setEnabled(true);
838         device_selector->setEnabled(false);
839     }
840 }
841
842 void RecMonitor::manageCapturedFiles()
843 {
844     QString extension;
845     switch (KdenliveSettings::firewireformat()) {
846     case 0:
847         extension = ".dv";
848         break;
849     case 1:
850     case 2:
851         extension = ".avi";
852         break;
853     case 3:
854         extension = ".m2t";
855         break;
856     }
857     QDir dir(m_capturePath);
858     QStringList filters;
859     QString capturename = KdenliveSettings::dvgrabfilename();
860     if (capturename.isEmpty()) capturename = "capture";
861     filters << capturename + "*" + extension;
862     const QStringList result = dir.entryList(filters, QDir::Files, QDir::Time);
863     KUrl::List capturedFiles;
864     foreach(const QString & name, result) {
865         KUrl url = KUrl(dir.filePath(name));
866         if (KIO::NetAccess::exists(url, KIO::NetAccess::SourceSide, this)) {
867             KFileItem file(KFileItem::Unknown, KFileItem::Unknown, url, true);
868             if (file.time(KFileItem::ModificationTime) > m_captureTime) {
869                 // The file was captured in the last batch
870                 if (url.fileName().contains(':')) {
871                     // Several dvgrab options (--timecode,...) use : in the file name, which is
872                     // not supported by MLT, so rename them
873                     QString newUrl = url.directory(KUrl::AppendTrailingSlash) + url.fileName().replace(':', '_');
874                     if (QFile::rename(url.path(), newUrl)) {
875                         url = KUrl(newUrl);
876                     }
877
878                 }
879                 capturedFiles.append(url);
880             }
881         }
882     }
883     kDebug() << "Found : " << capturedFiles.count() << " new capture files";
884     kDebug() << capturedFiles;
885
886     if (capturedFiles.count() > 0) {
887         ManageCapturesDialog *d = new ManageCapturesDialog(capturedFiles, this);
888         if (d->exec() == QDialog::Accepted) {
889             emit addProjectClipList(d->importFiles());
890         }
891         delete d;
892     }
893 }
894
895 // virtual
896 void RecMonitor::mousePressEvent(QMouseEvent *event)
897 {
898     if (m_freeSpace->underMouse()) slotUpdateFreeSpace();
899     else QWidget::mousePressEvent(event);//m_videoBox->mousePressEvent(event);
900 }
901
902 void RecMonitor::slotUpdateFreeSpace()
903 {
904     KDiskFreeSpaceInfo info = KDiskFreeSpaceInfo::freeSpaceInfo(m_capturePath);
905     if (info.isValid() && info.size() > 0) {
906         m_freeSpace->setValue(100 * info.used() / info.size());
907         m_freeSpace->setText(i18n("Free space: %1", KIO::convertSize(info.available())));
908         m_freeSpace->update();
909     }
910 }
911
912 void RecMonitor::refreshRecMonitor(bool visible)
913 {
914     if (visible) {
915         //if (!m_isActive) activateMonitor();
916
917     }
918 }
919
920 void RecMonitor::slotPlay()
921 {
922     /*if (m_isPlaying) slotStopCapture();
923     else slotStartPreview(true);*/
924 }
925
926 void RecMonitor::slotReadDvgrabInfo()
927 {
928     QString data = m_captureProcess->readAllStandardError().simplified();
929     data = data.section('"', 2, 2).simplified();
930     m_dvinfo.setText(data.left(11));
931     m_dvinfo.updateGeometry();
932 }
933
934 AbstractRender *RecMonitor::abstractRender()
935 {
936     return m_captureDevice;
937 }
938
939
940 void RecMonitor::analyseFrames(bool analyse)
941 {
942     m_analyse = analyse;
943     if (m_captureDevice) m_captureDevice->sendFrameForAnalysis = analyse;
944 }
945
946 void RecMonitor::slotDroppedFrames(int dropped)
947 {
948     slotSetInfoMessage(i18n("%1 dropped frames", dropped));
949 }
950
951 void RecMonitor::buildMltDevice(const QString &path)
952 {
953     if (m_captureDevice == NULL) {
954         m_monitorManager->updateScopeSource();
955         m_captureDevice = new MltDeviceCapture(path, videoSurface, this);
956         connect(m_captureDevice, SIGNAL(droppedFrames(int)), this, SLOT(slotDroppedFrames(int)));
957         m_captureDevice->sendFrameForAnalysis = m_analyse;
958         m_monitorManager->updateScopeSource();
959     }
960 }
961
962 void RecMonitor::slotChangeRecordingPreview(bool enable)
963 {
964     KdenliveSettings::setEnable_recording_preview(enable);
965 }
966
967
968 void RecMonitor::slotMouseSeek(int /*eventDelta*/, bool /*fast*/)
969 {
970 }
971
972
973 #include "recmonitor.moc"
974