[vlc-devel] [RFC PATCH] h264_nal: add h264_get_spspps

[Thomas Guillem]

It fetches the SPS and PPS pointers from an Annex B buffer.
---
 modules/packetizer/h264_nal.c | 65 +++++++++++++++++++++++++++++++++++++++++++
 modules/packetizer/h264_nal.h |  6 ++++
 2 files changed, 71 insertions(+)

diff --git a/modules/packetizer/h264_nal.c b/modules/packetizer/h264_nal.c
index 2a375c2..d1f9ddf 100644
--- a/modules/packetizer/h264_nal.c
+++ b/modules/packetizer/h264_nal.c
@@ -22,6 +22,8 @@
 
 #include <limits.h>
 
+static const uint8_t annexb_startcode[] = { 0x00, 0x00, 0x00, 0x01 };
+
 int convert_sps_pps( decoder_t *p_dec, const uint8_t *p_buf,
                      uint32_t i_buf_size, uint8_t *p_out_buf,
                      uint32_t i_out_buf_size, uint32_t *p_sps_pps_size,
@@ -138,6 +140,69 @@ void convert_h264_to_annexb( uint8_t *p_buf, uint32_t i_len,
     }
 }
 
+int h264_get_spspps( uint8_t *p_buf, size_t i_buf,
+                     uint8_t **pp_sps, size_t *p_sps_size,
+                     uint8_t **pp_pps, size_t *p_pps_size )
+{
+    uint8_t *p_sps = NULL, *p_pps = NULL;
+    size_t i_sps_size = 0, i_pps_size = 0;
+    int i_nal_type = NAL_UNKNOWN;
+
+    while( true )
+    {
+        int i_inc = 0;
+
+        if( i_buf > 5 && memcmp( p_buf, annexb_startcode, 4 ) == 0 )
+        {
+            i_nal_type = p_buf[4] & 0x1F;
+
+            if( i_nal_type == NAL_UNKNOWN || i_nal_type > 18
+             || (i_nal_type >= 10 && i_nal_type <= 12))
+                return -1;
+
+            /* size of startcode + nal_type */
+            i_inc = 5;
+
+            if (i_nal_type >= 1 && i_nal_type <= 5)
+                break;
+
+            /* pointer to the beginning of the sps/pps  */
+            if( i_nal_type == NAL_SPS && !p_sps )
+                p_sps = p_buf;
+            if( i_nal_type == NAL_PPS && !p_pps )
+                p_pps = p_buf;
+        } else {
+            i_inc = 1;
+        }
+        if( !i_sps_size && p_sps && i_nal_type != NAL_SPS )
+            i_sps_size = p_buf - p_sps;
+        if( !i_pps_size && p_pps && i_nal_type != NAL_PPS )
+            i_pps_size = p_buf - p_pps;
+
+        i_buf -= i_inc;
+        p_buf += i_inc;
+
+        if( i_buf == 0 )
+        {
+            if( !i_sps_size && p_sps )
+                i_sps_size = p_buf - p_sps;
+            if( !i_pps_size && p_pps )
+                i_pps_size = p_buf - p_pps;
+            break;
+        }
+        if( i_nal_type == NAL_UNKNOWN )
+            return -1;
+    }
+    if( ( !p_sps || !i_sps_size ) && ( !p_pps || !i_pps_size ) )
+        return -1;
+    *pp_sps = p_sps;
+    *p_sps_size = i_sps_size;
+    *pp_pps = p_pps;
+    *p_pps_size = i_pps_size;
+
+    return 0;
+}
+
 bool h264_get_profile_level(const es_format_t *p_fmt, size_t *p_profile,
                             size_t *p_level, size_t *p_nal_size)
 {
diff --git a/modules/packetizer/h264_nal.h b/modules/packetizer/h264_nal.h
index 6358fe6..bc346cf 100644
--- a/modules/packetizer/h264_nal.h
+++ b/modules/packetizer/h264_nal.h
@@ -82,6 +82,12 @@ void convert_h264_to_annexb( uint8_t *p_buf, uint32_t i_len,
                              size_t i_nal_size,
                              struct H264ConvertState *state );
 
+/* Get the SPS/PPS pointers from an Annex B buffer
+ * Returns 0 if a SPS and/or a PPS is found */
+int h264_get_spspps( uint8_t *p_buf, size_t i_buf,
+                     uint8_t **pp_sps, size_t *p_sps_size,
+                     uint8_t **pp_pps, size_t *p_pps_size );
+
 /* Get level and Profile */
 bool h264_get_profile_level(const es_format_t *p_fmt, size_t *p_profile,
                             size_t *p_level, size_t *p_nal_size);
-- 
2.1.3