rtpdec_h264: Prepare h264_handle_packet_stap_a for sharing with hevc
authorMartin Storsjö <martin@martin.st>
Thu, 19 Feb 2015 19:40:48 +0000 (21:40 +0200)
committerMartin Storsjö <martin@martin.st>
Sat, 21 Feb 2015 09:32:36 +0000 (11:32 +0200)
Add a parameter for skipping a number of bytes at the start of each nal.

Signed-off-by: Martin Storsjö <martin@martin.st>
libavformat/rtpdec_h264.c

index bc563f2..e6c8c26 100644 (file)
@@ -177,7 +177,8 @@ static int sdp_parse_fmtp_config_h264(AVFormatContext *s,
 }
 
 static int h264_handle_packet_stap_a(AVFormatContext *ctx, AVPacket *pkt,
 }
 
 static int h264_handle_packet_stap_a(AVFormatContext *ctx, AVPacket *pkt,
-                                     const uint8_t *buf, int len)
+                                     const uint8_t *buf, int len,
+                                     int start_skip)
 {
     int pass         = 0;
     int total_length = 0;
 {
     int pass         = 0;
     int total_length = 0;
@@ -189,6 +190,9 @@ static int h264_handle_packet_stap_a(AVFormatContext *ctx, AVPacket *pkt,
         const uint8_t *src = buf;
         int src_len        = len;
 
         const uint8_t *src = buf;
         int src_len        = len;
 
+        src     += start_skip;
+        src_len -= start_skip;
+
         while (src_len > 2) {
             uint16_t nal_size = AV_RB16(src);
 
         while (src_len > 2) {
             uint16_t nal_size = AV_RB16(src);
 
@@ -215,8 +219,8 @@ static int h264_handle_packet_stap_a(AVFormatContext *ctx, AVPacket *pkt,
             }
 
             // eat what we handled
             }
 
             // eat what we handled
-            src     += nal_size;
-            src_len -= nal_size;
+            src     += nal_size + start_skip;
+            src_len -= nal_size + start_skip;
         }
 
         if (pass == 0) {
         }
 
         if (pass == 0) {
@@ -304,7 +308,7 @@ static int h264_handle_packet(AVFormatContext *ctx, PayloadContext *data,
         // consume the STAP-A NAL
         buf++;
         len--;
         // consume the STAP-A NAL
         buf++;
         len--;
-        result = h264_handle_packet_stap_a(ctx, pkt, buf, len);
+        result = h264_handle_packet_stap_a(ctx, pkt, buf, len, 0);
         break;
 
     case 25:                   // STAP-B
         break;
 
     case 25:                   // STAP-B