One for the common payload lenght validation and one for sei prefix
and suffix parsing.
---
 libavcodec/hevc_sei.c | 57 +++++++++++++++++++++++++++++++--------------------
 1 file changed, 35 insertions(+), 22 deletions(-)

diff --git a/libavcodec/hevc_sei.c b/libavcodec/hevc_sei.c
index 8727b65..ade97e8 100644
--- a/libavcodec/hevc_sei.c
+++ b/libavcodec/hevc_sei.c
@@ -118,6 +118,39 @@ static int decode_nal_sei_display_orientation(HEVCContext 
*s)
     return 0;
 }
 
+static int decode_nal_sei_prefix(HEVCContext *s, int type, int size)
+{
+    GetBitContext *gb = &s->HEVClc.gb;
+
+    switch (type) {
+    case 256:  // Mismatched value from HM 8.1
+    case SEI_TYPE_DECODED_PICTURE_HASH:
+        return decode_nal_sei_decoded_picture_hash(s);
+    case SEI_TYPE_FRAME_PACKING:
+        return decode_nal_sei_frame_packing_arrangement(s);
+    case SEI_TYPE_DISPLAY_ORIENTATION:
+        return decode_nal_sei_display_orientation(s);
+    default:
+        av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", type);
+        skip_bits_long(gb, 8 * size);
+        return 0;
+    }
+}
+
+static int decode_nal_sei_suffix(HEVCContext *s, int type, int size)
+{
+    GetBitContext *gb = &s->HEVClc.gb;
+
+    switch (type) {
+    case SEI_TYPE_DECODED_PICTURE_HASH:
+        return decode_nal_sei_decoded_picture_hash(s);
+    default:
+        av_log(s->avctx, AV_LOG_DEBUG, "Skipped SUFFIX SEI %d\n", type);
+        skip_bits_long(gb, 8 * size);
+        return 0;
+    }
+}
+
 static int decode_nal_sei_message(HEVCContext *s)
 {
     GetBitContext *gb = &s->HEVClc.gb;
@@ -150,30 +183,10 @@ static int decode_nal_sei_message(HEVCContext *s)
     }
 
     if (s->nal_unit_type == NAL_SEI_PREFIX) {
-        switch (payload_type) {
-        case 256:  // Mismatched value from HM 8.1
-        case SEI_TYPE_DECODED_PICTURE_HASH:
-            return decode_nal_sei_decoded_picture_hash(s);
-        case SEI_TYPE_FRAME_PACKING:
-            return decode_nal_sei_frame_packing_arrangement(s);
-        case SEI_TYPE_DISPLAY_ORIENTATION:
-            return decode_nal_sei_display_orientation(s);
-        default:
-            av_log(s->avctx, AV_LOG_DEBUG, "Skipped PREFIX SEI %d\n", 
payload_type);
-            skip_bits_long(gb, 8 * payload_size);
-            return 0;
-        }
+        return decode_nal_sei_prefix(s, payload_type, payload_size);
     } else { /* nal_unit_type == NAL_SEI_SUFFIX */
-        switch (payload_type) {
-        case SEI_TYPE_DECODED_PICTURE_HASH:
-            return decode_nal_sei_decoded_picture_hash(s);
-        default:
-            av_log(s->avctx, AV_LOG_DEBUG, "Skipped SUFFIX SEI %d\n", 
payload_type);
-            skip_bits_long(gb, 8 * payload_size);
-            return 0;
-        }
+        return decode_nal_sei_suffix(s, payload_type, payload_size);
     }
-    return 0;
 }
 
 static int more_rbsp_data(GetBitContext *gb)
-- 
2.3.2

_______________________________________________
libav-devel mailing list
[email protected]
https://lists.libav.org/mailman/listinfo/libav-devel

Reply via email to