Inject VUI data in SPS header if missing

2021-06-13 21:29:32 +02:00
parent 30f7742f51
commit 8e32c8e6f4
10 changed files with 478 additions and 84 deletions
@@ -0,0 +1,20 @@
+#ifndef SUNSHINE_CBS_H
+#define SUNSHINE_CBS_H
+
+#include "utility.h"
+
+struct AVPacket;
+struct AVCodecContext;
+namespace cbs {
+
+util::buffer_t<std::uint8_t> read_sps(const AVPacket *packet, int codec_id);
+util::buffer_t<std::uint8_t> make_sps_h264(const AVCodecContext *ctx);
+util::buffer_t<std::uint8_t> make_sps_hevc(const AVCodecContext *ctx);
+
+/**
+ * Check if SPS->VUI is present
+ */
+bool validate_sps(const AVPacket *packet, int codec_id);
+} // namespace cbs
+
+#endif