MediaInfo: Export h264 ref_frames

bilibili · Dec 28, 2017 · 3d81d15 · 3d81d15
1 parent a8f9da1
commit 3d81d15
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/src/core/media-info.js b/src/core/media-info.js
@@ -37,6 +37,7 @@ class MediaInfo {
         this.fps = null;
         this.profile = null;
         this.level = null;
+        this.refFrames = null;
         this.chromaFormat = null;
         this.sarNum = null;
         this.sarDen = null;
@@ -63,6 +64,7 @@ class MediaInfo {
                                  this.fps != null &&
                                  this.profile != null &&
                                  this.level != null &&
+                                 this.refFrames != null &&
                                  this.chromaFormat != null &&
                                  this.sarNum != null &&
                                  this.sarDen != null);

diff --git a/src/demux/flv-demuxer.js b/src/demux/flv-demuxer.js
@@ -951,6 +951,7 @@ class FLVDemuxer {
             mi.fps = meta.frameRate.fps;
             mi.profile = meta.profile;
             mi.level = meta.level;
+            mi.refFrames = config.ref_frames;
             mi.chromaFormat = config.chroma_format_string;
             mi.sarNum = meta.sarRatio.width;
             mi.sarDen = meta.sarRatio.height;

diff --git a/src/demux/sps-parser.js b/src/demux/sps-parser.js
@@ -99,7 +99,7 @@ class SPSParser {
                 gb.readSEG();  // offset_for_ref_frame
             }
         }
-        gb.readUEG();  // max_num_ref_frames
+        let ref_frames = gb.readUEG();  // max_num_ref_frames
         gb.readBits(1);  // gaps_in_frame_num_value_allowed_flag
 
         let pic_width_in_mbs_minus1 = gb.readUEG();
@@ -198,6 +198,7 @@ class SPSParser {
             profile_string: profile_string,  // baseline, high, high10, ...
             level_string: level_string,  // 3, 3.1, 4, 4.1, 5, 5.1, ...
             bit_depth: bit_depth,  // 8bit, 10bit, ...
+            ref_frames: ref_frames,
             chroma_format: chroma_format,  // 4:2:0, 4:2:2, ...
             chroma_format_string: SPSParser.getChromaFormatString(chroma_format),