add:录像支持mkv

QtScrcpy/dialog.cpp

@@ -74,6 +74,10 @@ void Dialog::initUI()
     ui->videoSizeBox->addItem("1080");
     ui->videoSizeBox->addItem("native");
     ui->videoSizeBox->setCurrentIndex(1);
+
+    ui->formatBox->addItem("mp4");
+    ui->formatBox->addItem("mkv");
+    ui->videoSizeBox->setCurrentIndex(0);
 }
 
 void Dialog::on_updateDevice_clicked()
@@ -92,8 +96,9 @@ void Dialog::on_startServerBtn_clicked()
         QString fileDir(ui->recordPathEdt->text().trimmed());
         if (!fileDir.isEmpty()) {
             QDateTime dateTime = QDateTime::currentDateTime();
-            QString fileName = dateTime.toString("_yyyyMMdd_hhmmss.zzz");
-            fileName = windowTitle() + fileName + ".mp4";
+            QString fileName = dateTime.toString("_yyyyMMdd_hhmmss_zzz");
+            QString ext = ui->formatBox->currentText().trimmed();
+            fileName = windowTitle() + fileName + "." + ext;
             QDir dir(fileDir);
             absFilePath = dir.absoluteFilePath(fileName);
         }

QtScrcpy/dialog.ui

@@ -6,19 +6,19 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>502</width>
+    <width>639</width>
     <height>600</height>
    </rect>
   </property>
   <property name="minimumSize">
    <size>
-    <width>502</width>
+    <width>520</width>
     <height>0</height>
    </size>
   </property>
   <property name="maximumSize">
    <size>
-    <width>502</width>
+    <width>639</width>
     <height>16777215</height>
    </size>
   </property>
@@ -158,6 +158,20 @@
       <string>Config</string>
      </property>
      <layout class="QGridLayout" name="gridLayout_3">
+      <item row="1" column="0">
+       <widget class="QLabel" name="label_5">
+        <property name="text">
+         <string>record save path:</string>
+        </property>
+       </widget>
+      </item>
+      <item row="0" column="4">
+       <widget class="QLabel" name="label_6">
+        <property name="text">
+         <string>format：</string>
+        </property>
+       </widget>
+      </item>
       <item row="0" column="3">
        <widget class="QComboBox" name="videoSizeBox">
         <property name="toolTip">
@@ -165,6 +179,13 @@
         </property>
        </widget>
       </item>
+      <item row="0" column="0">
+       <widget class="QLabel" name="label_3">
+        <property name="text">
+         <string>bit rate:</string>
+        </property>
+       </widget>
+      </item>
       <item row="0" column="1">
        <widget class="QComboBox" name="bitRateBox">
         <property name="toolTip">
@@ -182,28 +203,17 @@
         </property>
        </widget>
       </item>
-      <item row="0" column="0">
-       <widget class="QLabel" name="label_3">
-        <property name="text">
-         <string>bit rate:</string>
-        </property>
-       </widget>
-      </item>
-      <item row="1" column="0">
-       <widget class="QLabel" name="label_5">
-        <property name="text">
-         <string>record save path:</string>
-        </property>
-       </widget>
+      <item row="0" column="5">
+       <widget class="QComboBox" name="formatBox"/>
       </item>
-      <item row="1" column="1" colspan="2">
+      <item row="1" column="1" colspan="3">
        <widget class="QLineEdit" name="recordPathEdt">
         <property name="readOnly">
          <bool>true</bool>
         </property>
        </widget>
       </item>
-      <item row="1" column="3">
+      <item row="1" column="4" colspan="2">
        <widget class="QPushButton" name="selectRecordPathBtn">
         <property name="text">
          <string>select path</string>

QtScrcpy/recorder/recorder.cpp

@@ -1,4 +1,5 @@
 #include <QDebug>
+#include <QFileInfo>
 
 #include "recorder.h"
 
@@ -16,6 +17,7 @@ static const AVRational SCRCPY_TIME_BASE = {1, 1000000}; // timestamps in us
 
 Recorder::Recorder(const QString& fileName)
     : m_fileName(fileName)
+    , m_format(guessRecordFormat(fileName))
 {
 
 }
@@ -30,11 +32,18 @@ void Recorder::setFrameSize(const QSize &declaredFrameSize)
     m_declaredFrameSize = declaredFrameSize;
 }
 
+void Recorder::setFormat(Recorder::RecorderFormat format)
+{
+    m_format = format;
+}
+
 bool Recorder::open(AVCodec *inputCodec)
 {
-    const AVOutputFormat* mp4 = findMp4Muxer();
-    if (!mp4) {
-        qCritical("Could not find mp4 muxer");
+    QString formatName = recorderGetFormatName(m_format);
+    Q_ASSERT(!formatName.isEmpty());
+    const AVOutputFormat* format = findMuxer(formatName.toUtf8());
+    if (!format) {
+        qCritical("Could not find muxer");
         return false;
     }
 
@@ -49,7 +58,7 @@ bool Recorder::open(AVCodec *inputCodec)
     // still expects a pointer-to-non-const (it has not be updated accordingly)
     // <https://github.com/FFmpeg/FFmpeg/commit/0694d8702421e7aff1340038559c438b61bb30dd>
 
-    m_formatCtx->oformat = (AVOutputFormat*)mp4;
+    m_formatCtx->oformat = (AVOutputFormat*)format;
 
     AVStream* outStream = avformat_new_stream(m_formatCtx, inputCodec);
     if (!outStream) {
@@ -115,7 +124,7 @@ bool Recorder::write(AVPacket *packet)
     return av_write_frame(m_formatCtx, packet) >= 0;
 }
 
-const AVOutputFormat *Recorder::findMp4Muxer()
+const AVOutputFormat *Recorder::findMuxer(const char* name)
 {
 #if LIBAVFORMAT_VERSION_INT >= AV_VERSION_INT(58, 9, 100)
     void* opaque = Q_NULLPTR;
@@ -127,8 +136,8 @@ const AVOutputFormat *Recorder::findMp4Muxer()
 #else
         outFormat = av_oformat_next(outFormat);
 #endif
-        // until null or with name "mp4"
-    } while (outFormat && strcmp(outFormat->name, "mp4"));
+        // until null or with name "name"
+    } while (outFormat && strcmp(outFormat->name, name));
     return outFormat;
 }
 
@@ -167,3 +176,29 @@ void Recorder::recorderRescalePacket(AVPacket *packet)
     AVStream *ostream = m_formatCtx->streams[0];
     av_packet_rescale_ts(packet, SCRCPY_TIME_BASE, ostream->time_base);
 }
+
+QString Recorder::recorderGetFormatName(Recorder::RecorderFormat format)
+{
+    switch (format) {
+    case RECORDER_FORMAT_MP4: return "mp4";
+    case RECORDER_FORMAT_MKV: return "matroska";
+    default: return "";
+    }
+}
+
+Recorder::RecorderFormat Recorder::guessRecordFormat(const QString &fileName)
+{
+    if (4 > fileName.length()) {
+        return Recorder::RECORDER_FORMAT_NULL;
+    }
+    QFileInfo fileInfo = QFileInfo(fileName);
+    QString ext = fileInfo.suffix();
+    if (0 == ext.compare("mp4")) {
+        return Recorder::RECORDER_FORMAT_MP4;
+    }
+    if (0 == ext.compare("mkv")) {
+        return Recorder::RECORDER_FORMAT_MKV;
+    }
+
+    return Recorder::RECORDER_FORMAT_NULL;
+}

QtScrcpy/recorder/recorder.h

@@ -11,24 +11,34 @@ extern "C"
 class Recorder
 {
 public:
+    enum RecorderFormat {
+        RECORDER_FORMAT_NULL = 0,
+        RECORDER_FORMAT_MP4,
+        RECORDER_FORMAT_MKV,
+    };
+
     Recorder(const QString& fileName);
     virtual ~Recorder();
 
     void setFrameSize(const QSize& declaredFrameSize);
+    void setFormat(Recorder::RecorderFormat format);
     bool open(AVCodec* inputCodec);
     void close();
     bool write(AVPacket* packet);
 
 private:
-    const AVOutputFormat* findMp4Muxer();
+    const AVOutputFormat* findMuxer(const char* name);
     bool recorderWriteHeader(AVPacket* packet);
     void recorderRescalePacket(AVPacket *packet);
+    QString recorderGetFormatName(Recorder::RecorderFormat format);
+    RecorderFormat guessRecordFormat(const QString& fileName);
 
 private:
     QString m_fileName = "";
     AVFormatContext* m_formatCtx = Q_NULLPTR;
     QSize m_declaredFrameSize;
     bool m_headerWritten = false;
+    RecorderFormat m_format = RECORDER_FORMAT_NULL;
 };
 
 #endif // RECORDER_H

QtScrcpy/videoform.cpp

@@ -13,11 +13,11 @@
 #include <QMessageBox>
 
 #include "videoform.h"
+#include "recorder.h"
 #include "ui_videoform.h"
 #include "iconhelper.h"
 #include "toolform.h"
 #include "controlevent.h"
-#include "recorder.h"
 #include "mousetap/mousetap.h"
 
 VideoForm::VideoForm(const QString& serial, quint16 maxSize, quint32 bitRate, const QString& fileName, QWidget *parent) :

QtScrcpy/videoform.h

@@ -44,7 +44,7 @@ private:
     void initUI();
     void initSignals();
     void showToolFrom(bool show = true);
-    void postKeyCodeClick(AndroidKeycode keycode);
+    void postKeyCodeClick(AndroidKeycode keycode);    
 
 protected:
     void mousePressEvent(QMouseEvent *event);