summaryrefslogtreecommitdiff
path: root/src/videopreview/videopreview.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/videopreview/videopreview.cpp')
-rw-r--r--src/videopreview/videopreview.cpp648
1 files changed, 648 insertions, 0 deletions
diff --git a/src/videopreview/videopreview.cpp b/src/videopreview/videopreview.cpp
new file mode 100644
index 0000000..4e4151c
--- /dev/null
+++ b/src/videopreview/videopreview.cpp
@@ -0,0 +1,648 @@
+/* smplayer, GUI front-end for mplayer.
+ Copyright (C) 2006-2009 Ricardo Villalba <rvm@escomposlinux.org>
+
+ This program is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program; if not, write to the Free Software
+ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+*/
+
+#include "videopreview.h"
+#include "videopreviewconfigdialog.h"
+#include <QProcess>
+#include <QRegExp>
+#include <QDir>
+#include <QTime>
+#include <QProgressDialog>
+#include <QWidget>
+#include <QGridLayout>
+#include <QLabel>
+#include <QScrollArea>
+#include <QDialogButtonBox>
+#include <QPushButton>
+#include <QPainter>
+#include <QFileDialog>
+#include <QMessageBox>
+#include <QSettings>
+#include <QApplication>
+#include <QPixmapCache>
+#include <QImageWriter>
+#include <QImageReader>
+
+#include <cmath>
+
+#define RENAME_PICTURES 0
+
+VideoPreview::VideoPreview(QString mplayer_path, QWidget * parent) : QWidget(parent, Qt::Window)
+{
+ setMplayerPath(mplayer_path);
+
+ set = 0; // settings
+ save_last_directory = true;
+
+ prop.input_video.clear();
+ prop.dvd_device.clear();
+ prop.n_cols = 4;
+ prop.n_rows = 4;
+ prop.initial_step = 20;
+ prop.max_width = 800;
+ prop.aspect_ratio = 0;
+ prop.display_osd = true;
+ prop.extract_format = JPEG;
+
+ output_dir = "smplayer_preview";
+ full_output_dir = QDir::tempPath() +"/"+ output_dir;
+
+ progress = new QProgressDialog(this);
+ progress->setMinimumDuration(0);
+ connect( progress, SIGNAL(canceled()), this, SLOT(cancelPressed()) );
+
+ w_contents = new QWidget(this);
+ QPalette p = w_contents->palette();
+ p.setColor(w_contents->backgroundRole(), Qt::white);
+ p.setColor(w_contents->foregroundRole(), Qt::black);
+ w_contents->setPalette(p);
+
+ info = new QLabel(this);
+
+ foot = new QLabel(this);
+ foot->setAlignment(Qt::AlignRight);
+
+ grid_layout = new QGridLayout;
+ grid_layout->setSpacing(2);
+
+ QVBoxLayout * l = new QVBoxLayout;
+ l->setSizeConstraint(QLayout::SetFixedSize);
+ l->addWidget(info);
+ l->addLayout(grid_layout);
+ l->addWidget(foot);
+
+ w_contents->setLayout(l);
+
+ scroll_area = new QScrollArea(this);
+ scroll_area->setWidgetResizable(true);
+ scroll_area->setAlignment(Qt::AlignCenter);
+ scroll_area->setWidget( w_contents );
+
+ button_box = new QDialogButtonBox(QDialogButtonBox::Close | QDialogButtonBox::Save, Qt::Horizontal, this);
+ connect( button_box, SIGNAL(rejected()), this, SLOT(close()) );
+ connect( button_box->button(QDialogButtonBox::Save), SIGNAL(clicked()), this, SLOT(saveImage()) );
+
+ QVBoxLayout * my_layout = new QVBoxLayout;
+ my_layout->addWidget(scroll_area);
+ my_layout->addWidget(button_box);
+ setLayout(my_layout);
+
+ retranslateStrings();
+
+ QList<QByteArray> r_formats = QImageReader::supportedImageFormats();
+ QString read_formats;
+ for (int n=0; n < r_formats.count(); n++) {
+ read_formats.append(r_formats[n]+" ");
+ }
+ qDebug("VideoPreview::VideoPreview: supported formats for reading: %s", read_formats.toUtf8().constData());
+
+ QList<QByteArray> w_formats = QImageWriter::supportedImageFormats();
+ QString write_formats;
+ for (int n=0; n < w_formats.count(); n++) {
+ write_formats.append(w_formats[n]+" ");
+ }
+ qDebug("VideoPreview::VideoPreview: supported formats for writing: %s", write_formats.toUtf8().constData());
+
+ toggleInfoAct = new QAction(this);
+ toggleInfoAct->setCheckable(true);
+ toggleInfoAct->setChecked(true);
+ toggleInfoAct->setShortcut( QKeySequence("Ctrl+H") );
+ connect( toggleInfoAct, SIGNAL(toggled(bool)), this, SLOT(showInfo(bool)) );
+ addAction(toggleInfoAct);
+}
+
+VideoPreview::~VideoPreview() {
+ if (set) saveSettings();
+}
+
+void VideoPreview::retranslateStrings() {
+ progress->setWindowTitle(tr("Video preview"));
+ progress->setCancelButtonText( tr("Cancel") );
+
+ foot->setText("<i>"+ tr("Generated by SMPlayer") +" </i>");
+}
+
+void VideoPreview::setMplayerPath(QString mplayer_path) {
+ mplayer_bin = mplayer_path;
+ QFileInfo fi(mplayer_bin);
+ if (fi.exists() && fi.isExecutable() && !fi.isDir()) {
+ mplayer_bin = fi.absoluteFilePath();
+ }
+
+ qDebug("VideoPreview::setMplayerPath: mplayer_bin: '%s'", mplayer_bin.toUtf8().constData());
+}
+
+void VideoPreview::setSettings(QSettings * settings) {
+ set = settings;
+ loadSettings();
+}
+
+void VideoPreview::clearThumbnails() {
+ for (int n=0; n < label_list.count(); n++) {
+ grid_layout->removeWidget( label_list[n] );
+ delete label_list[n];
+ }
+ label_list.clear();
+ info->clear();
+}
+
+QString VideoPreview::framePicture() {
+ if (prop.extract_format == PNG)
+ return "00000005.png";
+ else
+ return "00000005.jpg";
+}
+
+bool VideoPreview::createThumbnails() {
+ clearThumbnails();
+ error_message.clear();
+
+ button_box->setEnabled(false);
+
+ bool result = extractImages();
+
+ progress->close();
+
+ if ((result == false) && (!error_message.isEmpty())) {
+ QMessageBox::critical(this, tr("Error"),
+ tr("The following error has occurred while creating the thumbnails:")+"\n"+ error_message );
+ }
+
+ button_box->setEnabled(true);
+
+ // Adjust size
+ //resize( w_contents->sizeHint() );
+
+ cleanDir(full_output_dir);
+ return result;
+}
+
+bool VideoPreview::extractImages() {
+ VideoInfo i = getInfo(mplayer_bin, prop.input_video);
+ int length = i.length;
+
+ if (length == 0) {
+ if (error_message.isEmpty()) error_message = tr("The length of the video is 0");
+ return false;
+ }
+
+ // Create a temporary directory
+ QDir d(QDir::tempPath());
+ if (!d.exists(output_dir)) {
+ if (!d.mkpath(output_dir)) {
+ qDebug("VideoPreview::extractImages: error: can't create '%s'", full_output_dir.toUtf8().constData());
+ error_message = tr("The temporary directory (%1) can't be created").arg(full_output_dir);
+ return false;
+ }
+ }
+
+ displayVideoInfo(i);
+
+ // Let's begin
+ run.thumbnail_width = 0;
+
+ int num_pictures = prop.n_cols * prop.n_rows;
+ length -= prop.initial_step;
+ int s_step = length / num_pictures;
+
+ int current_time = prop.initial_step;
+
+ canceled = false;
+ progress->setLabelText(tr("Creating thumbnails..."));
+ progress->setRange(0, num_pictures-1);
+ progress->show();
+
+ double aspect_ratio = i.aspect;
+ if (prop.aspect_ratio != 0) aspect_ratio = prop.aspect_ratio;
+
+ for (int n = 0; n < num_pictures; n++) {
+ qDebug("VideoPreview::extractImages: getting frame %d of %d...", n+1, num_pictures);
+ progress->setValue(n);
+ qApp->processEvents();
+
+ if (canceled) return false;
+
+ if (!runMplayer(current_time, aspect_ratio)) return false;
+
+ QString frame_picture = full_output_dir + "/" + framePicture();
+ if (!QFile::exists(frame_picture)) {
+ error_message = tr("The file %1 doesn't exist").arg(frame_picture);
+ return false;
+ }
+
+#if RENAME_PICTURES
+ QString extension = (extractFormat()==PNG) ? "png" : "jpg";
+ QString output_file = output_dir + QString("/picture_%1.%2").arg(current_time, 8, 10, QLatin1Char('0')).arg(extension);
+ d.rename(output_dir + "/" + framePicture(), output_file);
+#else
+ QString output_file = output_dir + "/" + framePicture();
+#endif
+
+ if (!addPicture(QDir::tempPath() +"/"+ output_file, n, current_time)) {
+ return false;
+ }
+
+ current_time += s_step;
+ }
+
+ return true;
+}
+
+bool VideoPreview::runMplayer(int seek, double aspect_ratio) {
+ QStringList args;
+ args << "-nosound";
+
+ if (prop.extract_format == PNG) {
+ args << "-vo"
+ << "png:outdir=\""+full_output_dir+"\"";
+ } else {
+ args << "-vo"
+ << "jpeg:outdir=\""+full_output_dir+"\"";
+ }
+
+ args << "-frames" << "6" << "-ss" << QString::number(seek);
+
+ if (aspect_ratio != 0) {
+ args << "-aspect" << QString::number(aspect_ratio) << "-zoom";
+ }
+
+ if (!prop.dvd_device.isEmpty()) {
+ args << "-dvd-device" << prop.dvd_device;
+ }
+
+ /*
+ if (display_osd) {
+ args << "-vf" << "expand=osd=1" << "-osdlevel" << "2";
+ }
+ */
+
+ args << prop.input_video;
+
+ QString command = mplayer_bin + " ";
+ for (int n = 0; n < args.count(); n++) command = command + args[n] + " ";
+ qDebug("VideoPreview::runMplayer: command: %s", command.toUtf8().constData());
+
+ QProcess p;
+ p.start(mplayer_bin, args);
+ if (!p.waitForFinished()) {
+ qDebug("VideoPreview::runMplayer: error running process");
+ error_message = tr("The mplayer process didn't run");
+ return false;
+ }
+
+ return true;
+}
+
+
+bool VideoPreview::addPicture(const QString & filename, int num, int time) {
+ int row = num / prop.n_cols;
+ int col = num % prop.n_cols;
+
+ qDebug("VideoPreview::addPicture: %d (row: %d col: %d) file: '%s'", num, row, col, filename.toUtf8().constData());
+
+ QPixmapCache::clear();
+ QPixmap picture;
+ if (!picture.load(filename)) {
+ qDebug("VideoPreview::addPicture: can't load file");
+ error_message = tr("The file %1 can't be loaded").arg(filename);
+ return false;
+ }
+
+ if (run.thumbnail_width == 0) {
+ int spacing = grid_layout->horizontalSpacing() * (prop.n_cols-1);
+ if (spacing < 0) spacing = 0;
+ qDebug("VideoPreview::addPicture: spacing: %d", spacing);
+ run.thumbnail_width = (prop.max_width - spacing) / prop.n_cols;
+ if (run.thumbnail_width > picture.width()) run.thumbnail_width = picture.width();
+ qDebug("VideoPreview::addPicture: thumbnail_width set to %d", run.thumbnail_width);
+ }
+
+ QPixmap scaled_picture = picture.scaledToWidth(run.thumbnail_width, Qt::SmoothTransformation);
+
+ // Add current time text
+ if (prop.display_osd) {
+ QString stime = QTime().addSecs(time).toString("hh:mm:ss");
+ QFont font("Arial");
+ font.setBold(true);
+ QPainter painter(&scaled_picture);
+ painter.setPen( Qt::white );
+ painter.setFont(font);
+ painter.drawText(scaled_picture.rect(), Qt::AlignRight | Qt::AlignBottom, stime);
+ }
+
+ QLabel * l = new QLabel(this);
+ label_list.append(l);
+ l->setPixmap(scaled_picture);
+ //l->setPixmap(picture);
+ grid_layout->addWidget(l, row, col);
+
+ return true;
+}
+
+void VideoPreview::displayVideoInfo(const VideoInfo & i) {
+ // Display info about the video
+ QTime t = QTime().addSecs(i.length);
+
+ QString aspect = QString::number(i.aspect);
+ if (fabs(1.77 - i.aspect) < 0.1) aspect = "16:9";
+ else
+ if (fabs(1.33 - i.aspect) < 0.1) aspect = "4:3";
+ else
+ if (fabs(2.35 - i.aspect) < 0.1) aspect = "2.35:1";
+
+ QString no_info = tr("No info");
+
+ QString fps = (i.fps==0 || i.fps==1000) ? no_info : QString("%1").arg(i.fps);
+ QString video_bitrate = (i.video_bitrate==0) ? no_info : tr("%1 kbps").arg(i.video_bitrate/1000);
+ QString audio_bitrate = (i.audio_bitrate==0) ? no_info : tr("%1 kbps").arg(i.audio_bitrate/1000);
+ QString audio_rate = (i.audio_rate==0) ? no_info : tr("%1 Hz").arg(i.audio_rate);
+
+ info->setText(
+ "<b><font size=+1>" + i.filename +"</font></b>"
+ "<table cellspacing=4 cellpadding=4><tr>"
+ "<td>" +
+ tr("Size: %1 MB").arg(i.size / (1024*1024)) + "<br>" +
+ tr("Resolution: %1x%2").arg(i.width).arg(i.height) + "<br>" +
+ tr("Length: %1").arg(t.toString("hh:mm:ss")) +
+ "</td>"
+ "<td>" +
+ tr("Video format: %1").arg(i.video_format) + "<br>" +
+ tr("Frames per second: %1").arg(fps) + "<br>" +
+ tr("Aspect ratio: %1").arg(aspect) + //"<br>" +
+ "</td>"
+ "<td>" +
+ tr("Video bitrate: %1").arg(video_bitrate) + "<br>" +
+ tr("Audio bitrate: %1").arg(audio_bitrate) + "<br>" +
+ tr("Audio rate: %1").arg(audio_rate) + //"<br>" +
+ "</td>"
+ "</tr></table>"
+ );
+ setWindowTitle( tr("Video preview") + " - " + i.filename );
+}
+
+void VideoPreview::cleanDir(QString directory) {
+ QStringList filter;
+ if (prop.extract_format == PNG) {
+ filter.append("*.png");
+ } else {
+ filter.append("*.jpg");
+ }
+
+ QDir d(directory);
+ QStringList l = d.entryList( filter, QDir::Files, QDir::Unsorted);
+
+ for (int n = 0; n < l.count(); n++) {
+ qDebug("VideoPreview::cleanDir: deleting '%s'", l[n].toUtf8().constData());
+ d.remove(l[n]);
+ }
+ qDebug("VideoPreview::cleanDir: removing directory '%s'", directory.toUtf8().constData());
+ d.rmpath(directory);
+}
+
+VideoInfo VideoPreview::getInfo(const QString & mplayer_path, const QString & filename) {
+ VideoInfo i;
+
+ if (filename.isEmpty()) {
+ error_message = tr("No filename");
+ return i;
+ }
+
+ QFileInfo fi(filename);
+ if (fi.exists()) {
+ i.filename = fi.fileName();
+ i.size = fi.size();
+ }
+
+ QRegExp rx("^ID_(.*)=(.*)");
+
+ QProcess p;
+ p.setProcessChannelMode( QProcess::MergedChannels );
+
+ QStringList args;
+ args << "-vo" << "null" << "-ao" << "null" << "-frames" << "1" << "-identify" << "-nocache" << "-noquiet" << filename;
+
+ if (!prop.dvd_device.isEmpty()) {
+ args << "-dvd-device" << prop.dvd_device;
+ }
+
+ p.start(mplayer_path, args);
+
+ if (p.waitForFinished()) {
+ QByteArray line;
+ while (p.canReadLine()) {
+ line = p.readLine().trimmed();
+ qDebug("VideoPreview::getInfo: '%s'", line.constData());
+ if (rx.indexIn(line) > -1) {
+ QString tag = rx.cap(1);
+ QString value = rx.cap(2);
+ qDebug("VideoPreview::getInfo: tag: '%s', value: '%s'", tag.toUtf8().constData(), value.toUtf8().constData());
+
+ if (tag == "LENGTH") i.length = (int) value.toDouble();
+ else
+ if (tag == "VIDEO_WIDTH") i.width = value.toInt();
+ else
+ if (tag == "VIDEO_HEIGHT") i.height = value.toInt();
+ else
+ if (tag == "VIDEO_FPS") i.fps = value.toDouble();
+ else
+ if (tag == "VIDEO_ASPECT") {
+ i.aspect = value.toDouble();
+ if ((i.aspect == 0) && (i.width != 0) && (i.height != 0)) {
+ i.aspect = (double) i.width / i.height;
+ }
+ }
+ else
+ if (tag == "VIDEO_BITRATE") i.video_bitrate = value.toInt();
+ else
+ if (tag == "AUDIO_BITRATE") i.audio_bitrate = value.toInt();
+ else
+ if (tag == "AUDIO_RATE") i.audio_rate = value.toInt();
+ else
+ if (tag == "VIDEO_FORMAT") i.video_format = value;
+ }
+ }
+ } else {
+ qDebug("VideoPreview::getInfo: error: process didn't start");
+ error_message = tr("The mplayer process didn't start while trying to get info about the video");
+ }
+
+ qDebug("VideoPreview::getInfo: filename: '%s'", i.filename.toUtf8().constData());
+ qDebug("VideoPreview::getInfo: resolution: '%d x %d'", i.width, i.height);
+ qDebug("VideoPreview::getInfo: length: '%d'", i.length);
+ qDebug("VideoPreview::getInfo: size: '%d'", (int) i.size);
+
+ return i;
+}
+
+void VideoPreview::showInfo(bool visible) {
+ qDebug("VideoPreview::showInfo: %d", visible);
+ info->setShown(visible);
+ foot->setShown(visible);
+}
+
+void VideoPreview::saveImage() {
+ qDebug("VideoPreview::saveImage");
+
+ // Proposed name
+ QString proposed_name = "";
+ if (save_last_directory) proposed_name = last_directory;
+
+ QFileInfo fi(prop.input_video);
+ if (fi.exists()) {
+ if (!save_last_directory) proposed_name = fi.absolutePath();
+ QString extension = (extractFormat()==PNG) ? "png" : "jpg";
+ proposed_name += "/"+ fi.completeBaseName() +"_preview."+ extension;
+ }
+
+ // Formats
+ QList<QByteArray> w_formats = QImageWriter::supportedImageFormats();
+ QString write_formats;
+ for (int n=0; n < w_formats.count(); n++) {
+ write_formats.append("*."+w_formats[n]+" ");
+ }
+ if (write_formats.isEmpty()) {
+ // Shouldn't happen!
+ write_formats = "*.png *.jpg";
+ }
+
+ QString filename = QFileDialog::getSaveFileName(this, tr("Save file"),
+ proposed_name, tr("Images") +" ("+ write_formats +")");
+
+ if (!filename.isEmpty()) {
+ QPixmap image = QPixmap::grabWidget(w_contents);
+ if (!image.save(filename)) {
+ // Failed!!!
+ qDebug("VideoPreview::saveImage: error saving '%s'", filename.toUtf8().constData());
+ QMessageBox::warning(this, tr("Error saving file"),
+ tr("The file couldn't be saved") );
+ } else {
+ last_directory = QFileInfo(filename).absolutePath();
+ }
+ }
+}
+
+bool VideoPreview::showConfigDialog(QWidget * parent) {
+ VideoPreviewConfigDialog d(parent);
+
+ d.setVideoFile( videoFile() );
+ d.setDVDDevice( DVDDevice() );
+ d.setCols( cols() );
+ d.setRows( rows() );
+ d.setInitialStep( initialStep() );
+ d.setMaxWidth( maxWidth() );
+ d.setDisplayOSD( displayOSD() );
+ d.setAspectRatio( aspectRatio() );
+ d.setFormat( extractFormat() );
+ d.setSaveLastDirectory( save_last_directory );
+
+ if (d.exec() == QDialog::Accepted) {
+ setVideoFile( d.videoFile() );
+ setDVDDevice( d.DVDDevice() );
+ setCols( d.cols() );
+ setRows( d.rows() );
+ setInitialStep( d.initialStep() );
+ setMaxWidth( d.maxWidth() );
+ setDisplayOSD( d.displayOSD() );
+ setAspectRatio( d.aspectRatio() );
+ setExtractFormat(d.format() );
+ save_last_directory = d.saveLastDirectory();
+
+ return true;
+ }
+
+ return false;
+}
+
+void VideoPreview::saveSettings() {
+ qDebug("VideoPreview::saveSettings");
+
+ set->beginGroup("videopreview");
+
+ set->setValue("columns", cols());
+ set->setValue("rows", rows());
+ set->setValue("initial_step", initialStep());
+ set->setValue("max_width", maxWidth());
+ set->setValue("osd", displayOSD());
+ set->setValue("format", extractFormat());
+ set->setValue("save_last_directory", save_last_directory);
+
+ if (save_last_directory) {
+ set->setValue("last_directory", last_directory);
+ }
+
+ set->setValue("filename", videoFile());
+ set->setValue("dvd_device", DVDDevice());
+
+ set->setValue("show_info", toggleInfoAct->isChecked());
+
+ set->endGroup();
+}
+
+void VideoPreview::loadSettings() {
+ qDebug("VideoPreview::loadSettings");
+
+ set->beginGroup("videopreview");
+
+ setCols( set->value("columns", cols()).toInt() );
+ setRows( set->value("rows", rows()).toInt() );
+ setInitialStep( set->value("initial_step", initialStep()).toInt() );
+ setMaxWidth( set->value("max_width", maxWidth()).toInt() );
+ setDisplayOSD( set->value("osd", displayOSD()).toBool() );
+ setExtractFormat( (ExtractFormat) set->value("format", extractFormat()).toInt() );
+ save_last_directory = set->value("save_last_directory", save_last_directory).toBool();
+ last_directory = set->value("last_directory", last_directory).toString();
+
+ setVideoFile( set->value("filename", videoFile()).toString() );
+ setDVDDevice( set->value("dvd_device", DVDDevice()).toString() );
+
+ toggleInfoAct->setChecked(set->value("show_info", true).toBool());
+
+ set->endGroup();
+}
+
+void VideoPreview::adjustWindowSize() {
+ qDebug("VideoPreview::adjustWindowSize: window size: %d %d", width(), height());
+ qDebug("VideoPreview::adjustWindowSize: scroll_area size: %d %d", scroll_area->width(), scroll_area->height());
+
+ int diff_width = width() - scroll_area->maximumViewportSize().width();
+ int diff_height = height() - scroll_area->maximumViewportSize().height();
+
+ qDebug("VideoPreview::adjustWindowSize: diff_width: %d diff_height: %d", diff_width, diff_height);
+
+ QSize new_size = w_contents->size() + QSize( diff_width, diff_height);
+
+ qDebug("VideoPreview::adjustWindowSize: new_size: %d %d", new_size.width(), new_size.height());
+
+ resize(new_size);
+}
+
+void VideoPreview::cancelPressed() {
+ canceled = true;
+}
+
+// Language change stuff
+void VideoPreview::changeEvent(QEvent *e) {
+ if (e->type() == QEvent::LanguageChange) {
+ retranslateStrings();
+ } else {
+ QWidget::changeEvent(e);
+ }
+}
+
+#include "moc_videopreview.cpp"
+