QmitkFFmpegWriter.cpp

Go to the documentation of this file.

/*============================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

============================================================================*/

#include "QmitkFFmpegWriter.h"
#include <QMessageBox>
#include <mitkExceptionMacro.h>
#include <mitkLogMacros.h>

QmitkFFmpegWriter::QmitkFFmpegWriter(QObject *parent)
  : QObject(parent), m_Process(new QProcess(this)), m_Framerate(0), m_IsRunning(false)
{
  this->connect(m_Process, SIGNAL(error(QProcess::ProcessError)), this, SLOT(OnProcessError(QProcess::ProcessError)));

  this->connect(
    m_Process, SIGNAL(finished(int, QProcess::ExitStatus)), this, SLOT(OnProcessFinished(int, QProcess::ExitStatus)));
}

QmitkFFmpegWriter::~QmitkFFmpegWriter()
{
}

QString QmitkFFmpegWriter::GetFFmpegPath() const
{
  return m_FFmpegPath;
}

void QmitkFFmpegWriter::SetFFmpegPath(const QString &path)
{
  m_FFmpegPath = path;
}

QSize QmitkFFmpegWriter::GetSize() const
{
  return m_Size;
}

void QmitkFFmpegWriter::SetSize(const QSize &size)
{
  m_Size = size;
}

void QmitkFFmpegWriter::SetSize(int width, int height)
{
  m_Size = QSize(width, height);
}

int QmitkFFmpegWriter::GetFramerate() const
{
  return m_Framerate;
}

void QmitkFFmpegWriter::SetFramerate(int framerate)
{
  m_Framerate = framerate;
}

QString QmitkFFmpegWriter::GetOutputPath() const
{
  return m_OutputPath;
}

void QmitkFFmpegWriter::SetOutputPath(const QString &path)
{
  m_OutputPath = path;
}

void QmitkFFmpegWriter::Start()
{
  if (m_FFmpegPath.isEmpty())
    mitkThrow() << "FFmpeg/Libav path is empty!";

  if (m_Size.isNull())
    mitkThrow() << "Invalid video frame size!";

  if (m_Framerate <= 0)
    mitkThrow() << "Invalid framerate!";

  if (m_OutputPath.isEmpty())
    mitkThrow() << "Output path is empty!";

  m_Process->start(m_FFmpegPath,
                   QStringList() << "-y"
                                 << "-f"
                                 << "rawvideo"
                                 << "-pix_fmt"
                                 << "rgb24"
                                 << "-s"
                                 << QString("%1x%2").arg(m_Size.width()).arg(m_Size.height())
                                 << "-r"
                                 << QString("%1").arg(m_Framerate)
                                 << "-i"
                                 << "-"
                                 << "-vf"
                                 << "vflip"
                                 << "-pix_fmt"
                                 << "yuv420p"
                                 << "-crf"
                                 << "18"
                                 << m_OutputPath);

  m_Process->waitForStarted();
  m_IsRunning = true;
}

bool QmitkFFmpegWriter::IsRunning() const
{
  return m_IsRunning;
}

void QmitkFFmpegWriter::WriteFrame(const unsigned char *frame)
{
  if (frame == nullptr || !m_Process->isOpen())
    return;

  m_Process->write(reinterpret_cast<const char *>(frame), m_Size.width() * m_Size.height() * 3);
  m_Process->waitForBytesWritten();
}

void QmitkFFmpegWriter::Stop()
{
  m_IsRunning = false;
  m_Process->closeWriteChannel();
}

void QmitkFFmpegWriter::OnProcessError(QProcess::ProcessError error)
{
  m_IsRunning = false;

  MITK_ERROR << QString::fromLatin1(m_Process->readAllStandardError()).toStdString();

  switch (error)
  {
    case QProcess::FailedToStart:
      mitkThrow() << "FFmpeg/Libav failed to start!";

    case QProcess::Crashed:
      mitkThrow() << "FFmpeg/Libav crashed!";

    case QProcess::Timedout:
      mitkThrow() << "FFmpeg/Libav timed out!";

    case QProcess::WriteError:
      mitkThrow() << "Could not write to FFmpeg/Libav!";

    case QProcess::ReadError:
      mitkThrow() << "Could not read from FFmpeg/Libav!";

    default:
      mitkThrow() << "An unknown error occurred!";
  }
}

void QmitkFFmpegWriter::OnProcessFinished(int exitCode, QProcess::ExitStatus exitStatus)
{
  m_IsRunning = false;

  if (exitStatus != QProcess::CrashExit)
  {
    if (exitCode != 0)
    {
      m_Process->close();
      mitkThrow() << QString("FFmpeg/Libav exit code: %1").arg(exitCode).toStdString().c_str();
    }
  }

  m_Process->close();
}