]> git.sesse.net Git - vlc/blobdiff - modules/codec/cc.h
Split SPS/PPS parsing and annex b conversion to a reusable utility function
[vlc] / modules / codec / cc.h
index 145f3c909dc6e93db6f4bcc493058f9f62b1aadb..500973fa2a6634c6e3454d664035e55689d66a02 100644 (file)
  *****************************************************************************/
 
 #ifndef _CC_H
-#define _C_H 1
+#define _CC_H 1
+
+#include <vlc_bits.h>
 
 /* CC have a maximum rate of 9600 bit/s (per field?) */
 #define CC_MAX_DATA_SIZE (2 * 3*600)
+enum
+{
+    CC_PAYLOAD_NONE,
+    CC_PAYLOAD_GA94,
+    CC_PAYLOAD_DVD,
+    CC_PAYLOAD_REPLAYTV,
+    CC_PAYLOAD_SCTE20,
+};
 typedef struct
 {
     /* Which channel are present */
@@ -34,6 +44,10 @@ typedef struct
     /* */
     bool b_reorder;
 
+    /* */
+    int i_payload_type;
+    int i_payload_other_count;
+
     /* CC data per field
      *  byte[x+0]: field (0/1)
      *  byte[x+1]: cc data 1
@@ -43,18 +57,6 @@ typedef struct
     uint8_t p_data[CC_MAX_DATA_SIZE];
 } cc_data_t;
 
-static inline int cc_Channel( int i_field, const uint8_t p_data[2] )
-{
-    const uint8_t d = p_data[0] & 0x7f;
-    if( i_field != 0 && i_field != 1 )
-        return -1;
-    if( d == 0x14 )
-        return 2*i_field + 0;
-    else if( d == 0x1c )
-        return 2*i_field + 1;
-    /* unknown(middle of a command)  or not cc channel */
-    return -1;
-}
 static inline void cc_Init( cc_data_t *c )
 {
     int i;
@@ -63,6 +65,8 @@ static inline void cc_Init( cc_data_t *c )
         c-> pb_present[i] = false; 
     c->i_data = 0;
     c->b_reorder = false;
+    c->i_payload_type = CC_PAYLOAD_NONE;
+    c->i_payload_other_count = 0;
 }
 static inline void cc_Exit( cc_data_t *c )
 {
@@ -73,7 +77,21 @@ static inline void cc_Flush( cc_data_t *c )
 {
     c->i_data = 0;
 }
-static inline void cc_Extract( cc_data_t *c, const uint8_t *p_src, int i_src )
+
+static inline void cc_AppendData( cc_data_t *c, int i_field, const uint8_t cc[2] )
+{
+    if( i_field == 0 || i_field == 1 )
+    {
+        c->pb_present[2*i_field+0] =
+        c->pb_present[2*i_field+1] = true;
+    }
+
+    c->p_data[c->i_data++] = i_field;
+    c->p_data[c->i_data++] = cc[0];
+    c->p_data[c->i_data++] = cc[1];
+}
+
+static inline void cc_Extract( cc_data_t *c, bool b_top_field_first, const uint8_t *p_src, int i_src )
 {
     static const uint8_t p_cc_ga94[4] = { 0x47, 0x41, 0x39, 0x34 };
     static const uint8_t p_cc_dvd[4] = { 0x43, 0x43, 0x01, 0xf8 };
@@ -81,14 +99,59 @@ static inline void cc_Extract( cc_data_t *c, const uint8_t *p_src, int i_src )
     static const uint8_t p_cc_replaytv4b[2] = { 0xcc, 0x02 };
     static const uint8_t p_cc_replaytv5a[2] = { 0x99, 0x02 };
     static const uint8_t p_cc_replaytv5b[2] = { 0xaa, 0x02 };
-    //static const uint8_t p_afd_start[4] = { 0x44, 0x54, 0x47, 0x31 };
+    static const uint8_t p_cc_scte20[2] = { 0x03, 0x81 };
+    static const uint8_t p_cc_scte20_old[2] = { 0x03, 0x01 };
 
     if( i_src < 4 )
         return;
 
+    int i_payload_type;
     if( !memcmp( p_cc_ga94, p_src, 4 ) && i_src >= 5+1+1+1 && p_src[4] == 0x03 )
     {
-        /* Extract CC from DVB/ATSC TS */
+        /* CC from DVB/ATSC TS */
+        i_payload_type = CC_PAYLOAD_GA94;
+    }
+    else if( !memcmp( p_cc_dvd, p_src, 4 ) && i_src > 4+1 )
+    {
+        i_payload_type = CC_PAYLOAD_DVD;
+    }
+    else if( i_src >= 2+2 + 2+2 &&
+             ( ( !memcmp( p_cc_replaytv4a, &p_src[0], 2 ) && !memcmp( p_cc_replaytv4b, &p_src[4], 2 ) ) ||
+               ( !memcmp( p_cc_replaytv5a, &p_src[0], 2 ) && !memcmp( p_cc_replaytv5b, &p_src[4], 2 ) ) ) )
+    {
+        i_payload_type = CC_PAYLOAD_REPLAYTV;
+    }
+    else if( ( !memcmp( p_cc_scte20, p_src, 2 ) ||
+               !memcmp( p_cc_scte20_old, p_src, 2 ) ) && i_src > 2 )
+    {
+        i_payload_type = CC_PAYLOAD_SCTE20;
+    }
+    else
+    {
+#if 0
+#define V(x) ( ( x < 0x20 || x >= 0x7f ) ? '?' : x )
+        fprintf( stderr, "-------------- unknown user data " );
+        for( int i = 0; i < i_src; i++ )
+            fprintf( stderr, "%2.2x ", p_src[i] );
+        for( int i = 0; i < i_src; i++ )
+            fprintf( stderr, "%c ", V(p_src[i]) );
+        fprintf( stderr, "\n" );
+#undef V
+#endif
+        return;
+    }
+
+    if( c->i_payload_type != CC_PAYLOAD_NONE && c->i_payload_type != i_payload_type )
+    {
+        c->i_payload_other_count++;
+        if( c->i_payload_other_count < 50 )
+            return;
+    }
+    c->i_payload_type        = i_payload_type;
+    c->i_payload_other_count = 0;
+
+    if( i_payload_type == CC_PAYLOAD_GA94 )
+    {
         /* cc_data()
          *          u1 reserved(1)
          *          u1 process_cc_data_flag
@@ -126,7 +189,6 @@ static inline void cc_Extract( cc_data_t *c, const uint8_t *p_src, int i_src )
         for( i = 0; i < i_count_cc; i++, cc += 3 )
         {
             int i_field = cc[0] & 0x03;
-            int i_channel;
             if( ( cc[0] & 0xfc ) != 0xfc )
                 continue;
             if( i_field != 0 && i_field != 1 )
@@ -134,19 +196,12 @@ static inline void cc_Extract( cc_data_t *c, const uint8_t *p_src, int i_src )
             if( c->i_data + 3 > CC_MAX_DATA_SIZE )
                 continue;
 
-            i_channel = cc_Channel( i_field, &cc[1] );
-            if( i_channel >= 0 && i_channel < 4 )
-                c->pb_present[i_channel] = true;
-
-            c->p_data[c->i_data++] = i_field;
-            c->p_data[c->i_data++] = cc[1];
-            c->p_data[c->i_data++] = cc[2];
+            cc_AppendData( c, i_field, &cc[1] );
         }
         c->b_reorder = true;
     }
-    else if( !memcmp( p_cc_dvd, p_src, 4 ) && i_src > 4+1 )
+    else if( i_payload_type == CC_PAYLOAD_DVD )
     {
-        /* DVD CC */
         const int b_truncate = p_src[4] & 0x01;
         const int i_field_first = (p_src[4] & 0x80) ? 0 : 1;
         const int i_count_cc2 = (p_src[4] >> 1) & 0xf;
@@ -161,7 +216,6 @@ static inline void cc_Extract( cc_data_t *c, const uint8_t *p_src, int i_src )
             for( j = 0; j < 2; j++, cc += 3 )
             {
                 const int i_field = j == i_field_first ? 0 : 1;
-                int i_channel;
 
                 if( b_truncate && i == i_count_cc2 - 1 && j == 1 )
                     break;
@@ -170,21 +224,13 @@ static inline void cc_Extract( cc_data_t *c, const uint8_t *p_src, int i_src )
                 if( c->i_data + 3 > CC_MAX_DATA_SIZE )
                     continue;
 
-                i_channel = cc_Channel( i_field, &cc[1] );
-                if( i_channel >= 0 && i_channel < 4 )
-                    c->pb_present[i_channel] = true;
-                c->p_data[c->i_data++] = i_field;
-                c->p_data[c->i_data++] = cc[1];
-                c->p_data[c->i_data++] = cc[2];
+                cc_AppendData( c, i_field, &cc[1] );
             }
         }
         c->b_reorder = false;
     }
-    else if( i_src >= 2+2 + 2+2 &&
-             ( ( !memcmp( p_cc_replaytv4a, &p_src[0], 2 ) && !memcmp( p_cc_replaytv4b, &p_src[4], 2 ) ) ||
-               ( !memcmp( p_cc_replaytv5a, &p_src[0], 2 ) && !memcmp( p_cc_replaytv5b, &p_src[4], 2 ) ) ) )
+    else if( i_payload_type == CC_PAYLOAD_REPLAYTV )
     {
-        /* ReplayTV CC */
         const uint8_t *cc = &p_src[0];
         int i;
         if( c->i_data + 2*3 > CC_MAX_DATA_SIZE )
@@ -193,25 +239,43 @@ static inline void cc_Extract( cc_data_t *c, const uint8_t *p_src, int i_src )
         for( i = 0; i < 2; i++, cc += 4 )
         {
             const int i_field = i == 0 ? 1 : 0;
-            int i_channel = cc_Channel( i_field, &cc[2] );
-            if( i_channel >= 0 && i_channel < 4 )
-                c->pb_present[i_channel] = true;
-            c->p_data[c->i_data++] = i_field;
-            c->p_data[c->i_data++] = cc[2];
-            c->p_data[c->i_data++] = cc[3];
+
+            cc_AppendData( c, i_field, &cc[2] );
         }
         c->b_reorder = false;
     }
     else
     {
-#if 0
-#define V(x) ( ( x < 0x20 || x >= 0x7f ) ? '?' : x )
-        fprintf( stderr, "-------------- unknown user data %2.2x %2.2x %2.2x %2.2x %c%c%c%c\n",
-                 p_src[0], p_src[1], p_src[2], p_src[3],
-                 V(p_src[0]), V(p_src[1]), V(p_src[2]), V(p_src[3]) );
-#undef V
-#endif
-        /* TODO DVD CC, ... */
+        bs_t s;
+        bs_init( &s, &p_src[2], i_src - 2 );
+        const int i_cc_count = bs_read( &s, 5 );
+        for( int i = 0; i < i_cc_count; i++ )
+        {
+            bs_skip( &s, 2 );
+            const int i_field_idx = bs_read( &s, 2 );
+            bs_skip( &s, 5 );
+            uint8_t cc[2];
+            for( int j = 0; j < 2; j++ )
+            {
+                cc[j] = 0;
+                for( int k = 0; k < 8; k++ )
+                    cc[j] |= bs_read( &s, 1 ) << k;
+            }
+            bs_skip( &s, 1 );
+
+            if( i_field_idx == 0 )
+                continue;
+            if( c->i_data + 2*3 > CC_MAX_DATA_SIZE )
+                continue;
+
+            /* 1,2,3 -> 0,1,0. I.E. repeated field 3 is merged with field 1 */
+            int i_field = ((i_field_idx - 1) & 1);
+            if (!b_top_field_first)
+                i_field ^= 1;
+
+            cc_AppendData( c, i_field, &cc[0] );
+        }
+        c->b_reorder = true;
     }
 }