git.sesse.net Git - kdenlive/blob - 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 <QMouseEvent>
  22 #include <QStylePainter>
  23 #include <QMenu>
  24 #include <QToolButton>
  25 #include <QFile>
  26 #include <QDir>
  27
  28 #include <KDebug>
  29 #include <KLocale>
  30 #include <KStandardDirs>
  31 #include <KComboBox>
  32
  33 #include "gentime.h"
  34 #include "kdenlivesettings.h"
  35 #include "recmonitor.h"
  36
  37 RecMonitor::RecMonitor(QString name, QWidget *parent)
  38         : QWidget(parent), m_name(name), m_isActive(false), m_isCapturing(false), m_isPlaying(false) {
  39     ui.setupUi(this);
  40
  41     ui.video_frame->setAttribute(Qt::WA_PaintOnScreen);
  42     ui.device_selector->setCurrentIndex(KdenliveSettings::defaultcapture());
  43     connect(ui.device_selector, SIGNAL(currentIndexChanged(int)), this, SLOT(slotVideoDeviceChanged(int)));
  44
  45
  46
  47     QToolBar *toolbar = new QToolBar(name, this);
  48     QHBoxLayout *layout = new QHBoxLayout;
  49
  50     m_playIcon = KIcon("media-playback-start");
  51     m_pauseIcon = KIcon("media-playback-pause");
  52
  53     m_discAction = toolbar->addAction(KIcon("network-connect"), i18n("Connect"));
  54     connect(m_discAction, SIGNAL(triggered()), this, SLOT(slotDisconnect()));
  55
  56     m_rewAction = toolbar->addAction(KIcon("media-seek-backward"), i18n("Rewind"));
  57     connect(m_rewAction, SIGNAL(triggered()), this, SLOT(slotRewind()));
  58
  59     m_playAction = toolbar->addAction(m_playIcon, i18n("Play"));
  60     connect(m_playAction, SIGNAL(triggered()), this, SLOT(slotStartCapture()));
  61
  62     m_stopAction = toolbar->addAction(KIcon("media-playback-stop"), i18n("Stop"));
  63     connect(m_stopAction, SIGNAL(triggered()), this, SLOT(slotStopCapture()));
  64     m_stopAction->setEnabled(false);
  65     m_fwdAction = toolbar->addAction(KIcon("media-seek-forward"), i18n("Forward"));
  66     connect(m_fwdAction, SIGNAL(triggered()), this, SLOT(slotForward()));
  67
  68     m_recAction = toolbar->addAction(KIcon("media-record"), i18n("Record"));
  69     connect(m_recAction, SIGNAL(triggered()), this, SLOT(slotRecord()));
  70     m_recAction->setCheckable(true);
  71
  72     layout->addWidget(toolbar);
  73     ui.control_frame_firewire->setLayout(layout);
  74
  75     slotVideoDeviceChanged(ui.device_selector->currentIndex());
  76     displayProcess = new QProcess;
  77     captureProcess = new QProcess;
  78
  79     connect(captureProcess, SIGNAL(stateChanged(QProcess::ProcessState)), this, SLOT(slotProcessStatus(QProcess::ProcessState)));
  80
  81     QStringList env = QProcess::systemEnvironment();
  82     env << "SDL_WINDOWID=" + QString::number(ui.video_frame->winId());
  83     displayProcess->setEnvironment(env);
  84
  85     kDebug() << "/////// BUILDING MONITOR, ID: " << ui.video_frame->winId();
  86 }
  87
  88 QString RecMonitor::name() const {
  89     return m_name;
  90 }
  91
  92 void RecMonitor::slotVideoDeviceChanged(int ix) {
  93
  94     if (ix == 1) {
  95         m_discAction->setEnabled(false);
  96         m_rewAction->setEnabled(false);
  97         m_fwdAction->setEnabled(false);
  98         m_recAction->setEnabled(true);
  99         m_stopAction->setEnabled(false);
 100         m_playAction->setEnabled(true);
 101     } else {
 102         m_discAction->setEnabled(true);
 103         m_recAction->setEnabled(false);
 104         m_stopAction->setEnabled(false);
 105         m_playAction->setEnabled(false);
 106         m_rewAction->setEnabled(false);
 107         m_fwdAction->setEnabled(false);
 108     }
 109 }
 110
 111 void RecMonitor::slotDisconnect() {
 112     if (captureProcess->state() == QProcess::NotRunning) {
 113         slotStartCapture(false);
 114         m_discAction->setIcon(KIcon("network-disconnect"));
 115         m_discAction->setText(i18n("Disonnect"));
 116         m_recAction->setEnabled(true);
 117         m_stopAction->setEnabled(true);
 118         m_playAction->setEnabled(true);
 119         m_rewAction->setEnabled(true);
 120         m_fwdAction->setEnabled(true);
 121     } else {
 122         captureProcess->write("q", 1);
 123         m_discAction->setIcon(KIcon("network-connect"));
 124         m_discAction->setText(i18n("Connect"));
 125         m_recAction->setEnabled(false);
 126         m_stopAction->setEnabled(false);
 127         m_playAction->setEnabled(false);
 128         m_rewAction->setEnabled(false);
 129         m_fwdAction->setEnabled(false);
 130     }
 131 }
 132
 133 void RecMonitor::slotRewind() {
 134     captureProcess->write("a", 1);
 135 }
 136
 137 void RecMonitor::slotForward() {
 138     captureProcess->write("z", 1);
 139 }
 140
 141 void RecMonitor::slotStopCapture() {
 142     // stop capture
 143     if (ui.device_selector->currentIndex() == 0) {
 144         captureProcess->write("\e", 2);
 145         m_playAction->setIcon(m_playIcon);
 146         m_isPlaying = false;
 147     } else {
 148         if (m_isCapturing && ui.autoaddbox->isChecked()) emit addProjectClip(m_captureFile);
 149         captureProcess->kill();
 150         displayProcess->kill();
 151     }
 152 }
 153
 154 void RecMonitor::slotStartCapture(bool play) {
 155
 156     /*
 157     *captureProcess<<"dvgrab";
 158
 159     bool isHdv = false;
 160
 161     switch (m_recPanel->capture_format->currentItem()){
 162         case 0:
 163       *captureProcess<<"--format"<<"dv1";
 164      break;
 165         case 1:
 166       *captureProcess<<"--format"<<"dv2";
 167      break;
 168         case 3:
 169       *captureProcess<<"--format"<<"hdv";
 170      isHdv = true;
 171      break;
 172         default:
 173             *captureProcess<<"--format"<<"raw";
 174      break;
 175     }
 176
 177     if (KdenliveSettings::autosplit()) *captureProcess<<"--autosplit";
 178     if (KdenliveSettings::timestamp()) *captureProcess<<"--timestamp";
 179     *captureProcess<<"-i"<<"capture"<<"-";*/
 180
 181     /*
 182         QStringList captureArgs;
 183         captureArgs<<"--format"<<"hdv"<<"-i"<<"capture"<<"-";
 184         QStringList displayArgs;
 185
 186         displayArgs<<"-f"<<"mpegts"<<"-x"<<QString::number(ui.video_frame->width())<<"-y"<<QString::number(ui.video_frame->height())<<"-";
 187
 188         captureProcess->setStandardOutputProcess(displayProcess);
 189         ui.video_frame->setScaledContents(false);
 190         captureProcess->start("dvgrab",captureArgs);
 191         displayProcess->start("ffplay",  displayArgs);*/
 192
 193
 194     if (captureProcess->state() != QProcess::NotRunning) {
 195         if (ui.device_selector->currentIndex() == 0) {
 196             if (m_isPlaying) {
 197                 captureProcess->write("k", 1);
 198                 //captureProcess->write("\e", 2);
 199                 m_playAction->setIcon(m_playIcon);
 200                 m_isPlaying = false;
 201             } else {
 202                 captureProcess->write("p", 1);
 203                 m_playAction->setIcon(m_pauseIcon);
 204                 m_isPlaying = true;
 205             }
 206         }
 207         return;
 208     }
 209     m_captureArgs.clear();
 210     m_displayArgs.clear();
 211     m_isPlaying = false;
 212
 213     if (ui.device_selector->currentIndex() == 0) {
 214         m_captureArgs << "--format" << "hdv" << "-i" << "capture" << "-";
 215         m_displayArgs << "-f" << "mpegts" << "-x" << QString::number(ui.video_frame->width()) << "-y" << QString::number(ui.video_frame->height()) << "-";
 216     } else {
 217         if (!KdenliveSettings::video4adevice().isEmpty()) m_captureArgs << "-f" << KdenliveSettings::video4aformat() << "-i" << KdenliveSettings::video4adevice();
 218         m_captureArgs << "-f" << "video4linux2" << "-s" << QString::number(KdenliveSettings::video4width()) + "x" + QString::number(KdenliveSettings::video4height()) << "-r" << QString::number(KdenliveSettings::video4rate()) << "-i" << KdenliveSettings::video4vdevice() << "-f" << KdenliveSettings::video4vformat() << "-";
 219         m_displayArgs << "-f" << KdenliveSettings::video4vformat() << "-x" << QString::number(ui.video_frame->width()) << "-y" << QString::number(ui.video_frame->height()) << "-";
 220     }
 221
 222     captureProcess->setStandardOutputProcess(displayProcess);
 223     if (ui.device_selector->currentIndex() == 0) {
 224         captureProcess->setWorkingDirectory(KdenliveSettings::capturefolder());
 225         captureProcess->start("dvgrab", m_captureArgs);
 226         if (play) captureProcess->write(" ", 1);
 227         m_discAction->setEnabled(true);
 228     } else captureProcess->start("ffmpeg", m_captureArgs);
 229     displayProcess->start("ffplay", m_displayArgs);
 230     ui.video_frame->setText(i18n("Initialising..."));
 231
 232 }
 233
 234 void RecMonitor::slotRecord() {
 235     if (captureProcess->state() == QProcess::NotRunning && ui.device_selector->currentIndex() == 0) {
 236         slotStartCapture();
 237     }
 238     if (m_isCapturing) {
 239         if (ui.device_selector->currentIndex() == 0) {
 240             captureProcess->write("\e", 2);
 241             m_playAction->setIcon(m_playIcon);
 242             m_isCapturing = false;
 243             m_isPlaying = false;
 244             m_recAction->setChecked(false);
 245         } else {
 246             slotStopCapture();
 247             QTimer::singleShot(1000, this, SLOT(slotStartCapture()));
 248         }
 249         return;
 250     } else if (ui.device_selector->currentIndex() == 0) {
 251         m_isCapturing = true;
 252         captureProcess->write("c\n", 3);
 253         return;
 254     }
 255     if (captureProcess->state() == QProcess::NotRunning) {
 256         m_recAction->setChecked(true);
 257
 258         QString path = KdenliveSettings::capturefolder() + "/capture0000.mpg";
 259         int i = 1;
 260         while (QFile::exists(path)) {
 261             QString num = QString::number(i);
 262             if (i < 10) num.prepend("000");
 263             else if (i < 100) num.prepend("00");
 264             else num.prepend("0");
 265             path = KdenliveSettings::capturefolder() + "/capture" + num + ".mpg";
 266             i++;
 267         }
 268
 269         m_captureFile = KUrl(path);
 270
 271         m_isCapturing = true;
 272         m_captureArgs.clear();
 273         m_displayArgs.clear();
 274
 275         if (ui.device_selector->currentIndex() == 0) {
 276             m_captureArgs << "--format" << "hdv" << "-i" << "capture" << "-";
 277             m_displayArgs << "-f" << "mpegts" << "-x" << QString::number(ui.video_frame->width()) << "-y" << QString::number(ui.video_frame->height()) << "-";
 278         } else {
 279             if (!KdenliveSettings::video4adevice().isEmpty()) m_captureArgs << "-f" << KdenliveSettings::video4aformat() << "-i" << KdenliveSettings::video4adevice();
 280             m_captureArgs << "-f" << "video4linux2" << "-s" << QString::number(KdenliveSettings::video4width()) + "x" + QString::number(KdenliveSettings::video4height()) << "-r" << QString::number(KdenliveSettings::video4rate()) << "-i" << KdenliveSettings::video4vdevice() << "-y" << "-f" << KdenliveSettings::video4vformat() << m_captureFile.path() << "-f" << KdenliveSettings::video4vformat() << "-";
 281             m_displayArgs << "-f" << KdenliveSettings::video4vformat() << "-x" << QString::number(ui.video_frame->width()) << "-y" << QString::number(ui.video_frame->height()) << "-";
 282         }
 283
 284         captureProcess->setStandardOutputProcess(displayProcess);
 285         //ui.video_frame->setScaledContents(false);
 286         if (ui.device_selector->currentIndex() == 0) captureProcess->start("dvgrab", m_captureArgs);
 287         else captureProcess->start("ffmpeg", m_captureArgs);
 288         displayProcess->start("ffplay", m_displayArgs);
 289         ui.video_frame->setText(i18n("Initialising..."));
 290     } else {
 291         // stop capture
 292         displayProcess->kill();
 293         captureProcess->kill();
 294         QTimer::singleShot(1000, this, SLOT(slotRecord()));
 295     }
 296 }
 297
 298 void RecMonitor::slotProcessStatus(QProcess::ProcessState status) {
 299     if (status == QProcess::NotRunning) {
 300         m_isCapturing = false;
 301         m_isPlaying = false;
 302         m_playAction->setIcon(m_playIcon);
 303         m_recAction->setChecked(false);
 304         m_stopAction->setEnabled(false);
 305         ui.device_selector->setEnabled(true);
 306         ui.video_frame->setText(i18n("Not connected"));
 307     } else {
 308         m_stopAction->setEnabled(true);
 309         ui.device_selector->setEnabled(false);
 310     }
 311 }
 312
 313 // virtual
 314 void RecMonitor::mousePressEvent(QMouseEvent * event) {
 315     slotPlay();
 316 }
 317
 318 void RecMonitor::activateRecMonitor() {
 319     //if (!m_isActive) m_monitorManager->activateRecMonitor(m_name);
 320 }
 321
 322 void RecMonitor::stop() {
 323     m_isActive = false;
 324
 325 }
 326
 327 void RecMonitor::start() {
 328     m_isActive = true;
 329
 330 }
 331
 332 void RecMonitor::refreshRecMonitor(bool visible) {
 333     if (visible) {
 334         //if (!m_isActive) m_monitorManager->activateRecMonitor(m_name);
 335
 336     }
 337 }
 338
 339 void RecMonitor::slotPlay() {
 340
 341     //if (!m_isActive) m_monitorManager->activateRecMonitor(m_name);
 342
 343 }
 344
 345
 346 #include "recmonitor.moc"