test: add transcode

tests/transcode/main.cpp

@@ -0,0 +1,170 @@
+/******************************************************************************
+    transcode:  this file is part of QtAV examples
+    Copyright (C) 2015 Wang Bin <[email protected]>
+
+*   This file is part of QtAV
+
+    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 3 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, see <http://www.gnu.org/licenses/>.
+******************************************************************************/
+
+#include <QCoreApplication>
+#include <QtDebug>
+#include <QtCore/QDir>
+#include <QtCore/QElapsedTimer>
+#include <QtCore/QStringList>
+#include <QtAV>
+#include <QtAV/VideoEncoder.h>
+#include <QtAV/AVMuxer.h>
+
+using namespace QtAV;
+
+int main(int argc, char *argv[])
+{
+    QCoreApplication a(argc, argv);
+    QString file = "test.avi";
+    int idx = a.arguments().indexOf("-i");
+    if (idx > 0)
+        file = a.arguments().at(idx + 1);
+    QString decName("FFmpeg");
+    idx = a.arguments().indexOf("-d:v");
+    if (idx > 0)
+        decName = a.arguments().at(idx + 1);
+    QString outFile("/tmp/out.mp4");
+    idx = a.arguments().indexOf("-o");
+    if (idx > 0)
+        outFile = a.arguments().at(idx + 1);
+    QDir().mkpath(outFile.left(outFile.lastIndexOf('/')+1));
+
+    QString cv("libx264");
+    idx = a.arguments().indexOf("-c:v");
+    if (idx > 0)
+        cv = a.arguments().at(idx + 1);
+
+    QString fmt;
+    idx = a.arguments().indexOf("-f");
+    if (idx > 0)
+        fmt = a.arguments().at(idx + 1);
+
+
+    QString opt;
+    QVariantHash decopt;
+    idx = decName.indexOf(":");
+    if (idx > 0) {
+        opt = decName.right(decName.size() - idx -1);
+        decName = decName.left(idx);
+        QStringList opts(opt.split(";"));
+        QVariantHash subopt;
+        foreach (QString o, opts) {
+            idx = o.indexOf(":");
+            subopt[o.left(idx)] = o.right(o.size() - idx - 1);
+        }
+        decopt[decName] = subopt;
+    }
+    qDebug() << decopt;
+
+    VideoDecoderId cid = VideoDecoderFactory::id(decName.toStdString());
+    if (cid <= 0) {
+        qWarning("Can not find decoder: %s", decName.toUtf8().constData());
+        return 1;
+    }
+    VideoDecoder *dec = VideoDecoderFactory::create(cid);
+    if (!decopt.isEmpty())
+        dec->setOptions(decopt);
+    AVDemuxer demux;
+    demux.setMedia(file);
+    if (!demux.load()) {
+        qWarning("Failed to load file: %s", file.toUtf8().constData());
+        return 1;
+    }
+
+    dec->setCodecContext(demux.videoCodecContext());
+    dec->prepare();
+    dec->open();
+    QElapsedTimer timer;
+    timer.start();
+    int count = 0;
+    int vstream = demux.videoStream();
+    VideoEncoder *venc = VideoEncoder::create("FFmpeg");
+    venc->setCodecName(cv);
+    //venc->setCodecName("png");
+    venc->setBitRate(1024*1024);
+    //venc->setPixelFormat(VideoFormat::Format_RGBA32);
+    AVMuxer mux;
+    //mux.setMedia("/Users/wangbin/Movies/m3u8/bbb%05d.ts");
+    //mux.setMedia("/Users/wangbin/Movies/img2/bbb%05d.png");
+    mux.setMedia(outFile);
+    QVariantHash muxopt, avfopt;
+    avfopt["segment_time"] = 4;
+    avfopt["segment_list_size"] = 0;
+    avfopt["segment_list"] = outFile.left(outFile.lastIndexOf('/')+1) + "index.m3u8";// "//Users/wangbin/Movies/m3u8/bbb.m3u8";
+    avfopt["segment_format"] = "mpegts";
+    muxopt["avformat"] = avfopt;
+    qreal fps = 0;
+    while (!demux.atEnd()) {
+        if (!demux.readFrame())
+            continue;
+        if (demux.stream() != vstream)
+            continue;
+        const Packet pkt = demux.packet();
+        if (dec->decode(pkt)) {
+            VideoFrame frame = dec->frame(); // why is faster to call frame() for hwdec? no frame() is very slow for VDA
+            if (!frame)
+                continue;
+            if (!venc->isOpen()) {
+                venc->setWidth(frame.width());
+                venc->setHeight(frame.height());
+                if (!venc->open()) {
+                    qWarning("failed to open encoder");
+                    return 1;
+                }
+            }
+            if (!mux.isOpen()) {
+                mux.copyProperties(venc);
+                mux.setOptions(muxopt);
+                if (!fmt.isEmpty())
+                    mux.setFormat(fmt);
+                //mux.setFormat("segment");
+               // mux.setFormat("image2");
+                if (!mux.open()) {
+                    qWarning("failed to open muxer");
+                    return 1;
+                }
+                //mux.setOptions(muxopt);
+            }
+            if (frame.pixelFormat() != venc->pixelFormat())
+                frame = frame.to(venc->pixelFormat());
+            if (venc->encode(frame)) {
+                Packet pkt(venc->encoded());
+                mux.writeVideo(pkt);
+                count++;
+                if (count%20 == 0) {
+                    fps = qreal(count*1000)/qreal(timer.elapsed());
+                }
+                printf("decode count: %d, fps: %.2f frame size: %dx%d %d\r", count, fps, frame.width(), frame.height(), frame.data().size());fflush(0);
+            }
+        }
+    }
+    // get delayed frames
+    while (venc->encode()) {
+        qDebug("encode delayed frames...\r");
+        Packet pkt(venc->encoded());
+        mux.writeVideo(pkt);
+    }
+    qint64 elapsed = timer.elapsed();
+    int msec = elapsed/1000LL+1;
+    qDebug("decoded frames: %d, time: %d, average speed: %d", count, msec, count/msec);
+    venc->close();
+    mux.close();
+    return 0;
+}

tests/transcode/transcode.pro

@@ -0,0 +1,9 @@
+CONFIG -= app_bundle
+TEMPLATE = app
+TARGET = transcode
+
+PROJECTROOT = $$PWD/../..
+include($$PROJECTROOT/src/libQtAV.pri)
+preparePaths($$OUT_PWD/../../out)
+
+SOURCES += main.cpp