Implement get video and audio codec context

hasbi93 · Sep 9, 2023 · 320864d · 320864d
1 parent a13c9c0
commit 320864d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/zm_ffmpeg_input.h b/src/zm_ffmpeg_input.h
@@ -38,6 +38,8 @@ class FFmpeg_Input {
       return ( audio_stream_id >= 0 ) ? input_format_context->streams[audio_stream_id] : nullptr;
     }
     AVFormatContext *get_format_context() { return input_format_context; };
+    AVCodecContext *get_video_codec_context() { return ( video_stream_id >= 0 ) ? streams[video_stream_id].context : nullptr; };
+    AVCodecContext *get_audio_codec_context() { return ( audio_stream_id >= 0 ) ? streams[audio_stream_id].context : nullptr; };
 
   private:
     typedef struct {