git.sesse.net Git - vlc/blob - include/input_ext-dec.h

   1 /*****************************************************************************
   2  * input_ext-dec.h: structures exported to the VideoLAN decoders
   3  *****************************************************************************
   4  * Copyright (C) 1999, 2000 VideoLAN
   5  * $Id: input_ext-dec.h,v 1.10 2001/01/10 19:22:10 massiot Exp $
   6  *
   7  * Authors:
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  22  *****************************************************************************/
  23
  24 /* Structures exported to the decoders */
  25
  26 /*****************************************************************************
  27  * data_packet_t
  28  *****************************************************************************
  29  * Describe a data packet.
  30  *****************************************************************************/
  31 typedef struct data_packet_s
  32 {
  33     /* Nothing before this line, the code relies on that */
  34     byte_t *                p_buffer;                     /* raw data packet */
  35
  36     /* Decoders information */
  37     byte_t *                p_payload_start;
  38                                   /* start of the PES payload in this packet */
  39     byte_t *                p_payload_end;                    /* guess ? :-) */
  40     boolean_t               b_discard_payload;  /* is the packet messed up ? */
  41
  42     /* Used to chain the TS packets that carry data for a same PES or PSI */
  43     struct data_packet_s *  p_next;
  44 } data_packet_t;
  45
  46 /*****************************************************************************
  47  * pes_packet_t
  48  *****************************************************************************
  49  * Describes an PES packet, with its properties, and pointers to the TS packets
  50  * containing it.
  51  *****************************************************************************/
  52 typedef struct pes_packet_s
  53 {
  54     /* PES properties */
  55     boolean_t               b_messed_up;  /* At least one of the data packets
  56                                            * has a questionable content      */
  57     boolean_t               b_data_alignment;  /* used to find the beginning of
  58                                                 * a video or audio unit      */
  59     boolean_t               b_discontinuity; /* This packet doesn't follow the
  60                                               * previous one                 */
  61
  62     mtime_t                 i_pts;/* the PTS for this packet (zero if unset) */
  63     mtime_t                 i_dts;/* the DTS for this packet (zero if unset) */
  64
  65     int                     i_pes_size;    /* size of the current PES packet */
  66
  67     /* Pointers to packets (packets are then linked by the p_prev and
  68        p_next fields of the data_packet_t struct) */
  69     data_packet_t *         p_first;      /* The first packet containing this
  70                                            * PES (used by decoders). */
  71 } pes_packet_t;
  72
  73 /*****************************************************************************
  74  * decoder_fifo_t
  75  *****************************************************************************
  76  * This rotative FIFO contains PES packets that are to be decoded.
  77  *****************************************************************************/
  78 typedef struct decoder_fifo_s
  79 {
  80     /* Thread structures */
  81     vlc_mutex_t             data_lock;                     /* fifo data lock */
  82     vlc_cond_t              data_wait;     /* fifo data conditional variable */
  83
  84     /* Data */
  85     pes_packet_t *          buffer[FIFO_SIZE + 1];
  86     int                     i_start;
  87     int                     i_end;
  88
  89     /* Communication interface between input and decoders */
  90     boolean_t               b_die;          /* the decoder should return now */
  91     boolean_t               b_error;      /* the decoder is in an error loop */
  92     void *                  p_packets_mgt;   /* packets management services
  93                                               * data (netlist...)            */
  94     void                 (* pf_delete_pes)( void *, pes_packet_t * );
  95                                      /* function to use when releasing a PES */
  96 } decoder_fifo_t;
  97
  98 /* Macros to manage a decoder_fifo_t structure. Please remember to take
  99  * data_lock before using them. */
 100 #define DECODER_FIFO_ISEMPTY( fifo )  ( (fifo).i_start == (fifo).i_end )
 101 #define DECODER_FIFO_ISFULL( fifo )   ( ( ((fifo).i_end + 1 - (fifo).i_start)\
 102                                           & FIFO_SIZE ) == 0 )
 103 #define DECODER_FIFO_START( fifo )    ( (fifo).buffer[ (fifo).i_start ] )
 104 #define DECODER_FIFO_INCSTART( fifo ) ( (fifo).i_start = ((fifo).i_start + 1)\
 105                                                          & FIFO_SIZE )
 106 #define DECODER_FIFO_END( fifo )      ( (fifo).buffer[ (fifo).i_end ] )
 107 #define DECODER_FIFO_INCEND( fifo )   ( (fifo).i_end = ((fifo).i_end + 1) \
 108                                                        & FIFO_SIZE )
 109
 110 /*****************************************************************************
 111  * bit_fifo_t : bit fifo descriptor
 112  *****************************************************************************
 113  * This type describes a bit fifo used to store bits while working with the
 114  * input stream at the bit level.
 115  *****************************************************************************/
 116 typedef u32         WORD_TYPE;        /* only u32 is supported at the moment */
 117
 118 typedef struct bit_fifo_s
 119 {
 120     /* This unsigned integer allows us to work at the bit level. This buffer
 121      * can contain 32 bits, and the used space can be found on the MSb's side
 122      * and the available space on the LSb's side. */
 123     WORD_TYPE           buffer;
 124
 125     /* Number of bits available in the bit buffer */
 126     int                 i_available;
 127
 128 } bit_fifo_t;
 129
 130 /*****************************************************************************
 131  * bit_stream_t : bit stream descriptor
 132  *****************************************************************************
 133  * This type, based on a PES stream, includes all the structures needed to
 134  * handle the input stream like a bit stream.
 135  *****************************************************************************/
 136 typedef struct bit_stream_s
 137 {
 138     /*
 139      * Input structures
 140      */
 141     /* The decoder fifo contains the data of the PES stream */
 142     decoder_fifo_t *        p_decoder_fifo;
 143
 144     /* Function to jump to the next data packet */
 145     void                 (* pf_next_data_packet)( struct bit_stream_s * );
 146
 147     /* Callback to the decoder used when changing data packets ; set
 148      * to NULL if your decoder doesn't need it. */
 149     void                 (* pf_bitstream_callback)( struct bit_stream_s *,
 150                                                     boolean_t b_new_pes );
 151     /* Optional argument to the callback */
 152     void *                  p_callback_arg;
 153
 154     /*
 155      * Byte structures
 156      */
 157     /* Current data packet (in the current PES packet of the PES stream) */
 158     data_packet_t *         p_data;
 159     /* Pointer to the next byte that is to be read (in the current TS packet) */
 160     byte_t *                p_byte;
 161     /* Pointer to the last byte that is to be read (in the current TS packet */
 162     byte_t *                p_end;
 163
 164     /*
 165      * Bit structures
 166      */
 167     bit_fifo_t              fifo;
 168 } bit_stream_t;
 169
 170 /*****************************************************************************
 171  * Inline functions used by the decoders to read bit_stream_t
 172  *****************************************************************************/
 173
 174 /*
 175  * Philosophy of the first implementation : the bit buffer is first filled by
 176  * NeedBits, then the buffer can be read via p_bit_stream->fifo.buffer, and
 177  * unnecessary bits are dumped with a DumpBits() call.
 178  */
 179
 180 /*****************************************************************************
 181  * GetByte : reads the next byte in the input stream
 182  *****************************************************************************/
 183 static __inline__ byte_t GetByte( bit_stream_t * p_bit_stream )
 184 {
 185     /* Are there some bytes left in the current data packet ? */
 186     /* could change this test to have a if (! (bytes--)) instead */
 187     if ( p_bit_stream->p_byte >= p_bit_stream->p_end )
 188     {
 189         /* no, switch to next data packet */
 190         p_bit_stream->pf_next_data_packet( p_bit_stream );
 191     }
 192
 193     return( *(p_bit_stream->p_byte++) );
 194 }
 195
 196 /*****************************************************************************
 197  * NeedBits : reads i_bits new bits in the bit stream and stores them in the
 198  *            bit buffer
 199  *****************************************************************************
 200  * - i_bits must be less or equal 32 !
 201  * - There is something important to notice with that function : if the number
 202  * of bits available in the bit buffer when calling NeedBits() is greater than
 203  * 24 (i_available > 24) but less than the number of needed bits
 204  * (i_available < i_bits), the byte returned by GetByte() will be shifted with
 205  * a negative value and the number of bits available in the bit buffer will be
 206  * set to more than 32 !
 207  *****************************************************************************/
 208 static __inline__ void NeedBits( bit_stream_t * p_bit_stream, int i_bits )
 209 {
 210     while ( p_bit_stream->fifo.i_available < i_bits )
 211     {
 212         p_bit_stream->fifo.buffer |= ((WORD_TYPE)GetByte( p_bit_stream ))
 213                                      << (sizeof(WORD_TYPE) - 8
 214                                             - p_bit_stream->fifo.i_available);
 215         p_bit_stream->fifo.i_available += 8;
 216     }
 217 }
 218
 219 /*****************************************************************************
 220  * DumpBits : removes i_bits bits from the bit buffer
 221  *****************************************************************************
 222  * - i_bits <= i_available
 223  * - i_bits < 32 (because (u32 << 32) <=> (u32 = u32))
 224  *****************************************************************************/
 225 static __inline__ void DumpBits( bit_stream_t * p_bit_stream, int i_bits )
 226 {
 227     p_bit_stream->fifo.buffer <<= i_bits;
 228     p_bit_stream->fifo.i_available -= i_bits;
 229 }
 230
 231
 232 /*
 233  * Philosophy of the second implementation : WORD_LENGTH (usually 32) bits
 234  * are read at the same time, thus minimizing the number of p_byte changes.
 235  * Bits are read via GetBits() or ShowBits. This is slightly faster. Be
 236  * aware that if, in the forthcoming functions, i_bits > 24, the data have to
 237  * be already aligned on an 8-bit boundary, or wrong results will be
 238  * returned.
 239  */
 240
 241 #if (WORD_TYPE != u32)
 242 #   error Not supported word
 243 #endif
 244
 245 /*
 246  * This is stolen from the livid source who stole it from the kernel
 247  * FIXME: The macro swab32 for little endian machines does
 248  *        not seem to work correctly
 249  */
 250
 251 #if defined(SYS_BEOS)
 252 #   define swab32(x) B_BENDIAN_TO_HOST_INT32(x)
 253 #else
 254 #   ifdef WORDS_BIG_ENDIAN
 255 #       define swab32(x) (x)
 256 #   else
 257 #       if defined (HAVE_X86_BSWAP)
 258 static __inline__ const u32 __i386_swab32( u32 x )
 259 {
 260     __asm__("bswap %0" : "=r" (x) : "0" (x));
 261     return x;
 262 }
 263 #           define swab32(x) __i386_swab32(x)
 264 #       else
 265 #           define swab32(x)                                                 \
 266             ( ( (u32)(((u8*)&x)[0]) << 24 ) | ( (u32)(((u8*)&x)[1]) << 16 ) |\
 267               ( (u32)(((u8*)&x)[2]) << 8 )  | ( (u32)(((u8*)&x)[3])) )
 268 #       endif
 269 #   endif
 270 #endif
 271
 272 /*****************************************************************************
 273  * ShowBits : return i_bits bits from the bit stream
 274  *****************************************************************************/
 275 static __inline__ WORD_TYPE ShowWord( bit_stream_t * p_bit_stream )
 276 {
 277     if( p_bit_stream->p_byte <= p_bit_stream->p_end - sizeof(WORD_TYPE) )
 278     {
 279         return( swab32( *((WORD_TYPE *)p_bit_stream->p_byte) ) );
 280     }
 281
 282     p_bit_stream->pf_next_data_packet( p_bit_stream );
 283     return( swab32( *((WORD_TYPE *)p_bit_stream->p_byte) ) );
 284 }
 285
 286 static __inline__ WORD_TYPE ShowBits( bit_stream_t * p_bit_stream, int i_bits )
 287 {
 288     if( p_bit_stream->fifo.i_available >= i_bits )
 289     {
 290         return( p_bit_stream->fifo.buffer >> (8 * sizeof(WORD_TYPE) - i_bits) );
 291     }
 292
 293     return( (p_bit_stream->fifo.buffer |
 294             (ShowWord( p_bit_stream ) >> p_bit_stream->fifo.i_available))
 295                     >> (8 * sizeof(WORD_TYPE) - i_bits) );
 296 }
 297
 298 /*****************************************************************************
 299  * GetWord : returns the next word to be read
 300  *****************************************************************************/
 301 static __inline__ WORD_TYPE GetWord( bit_stream_t * p_bit_stream )
 302 {
 303     if( p_bit_stream->p_byte <= p_bit_stream->p_end - sizeof(WORD_TYPE) )
 304     {
 305         return( swab32( *(((WORD_TYPE *)p_bit_stream->p_byte)++) ) );
 306     }
 307     else
 308     {
 309         p_bit_stream->pf_next_data_packet( p_bit_stream );
 310         return( swab32( *(((WORD_TYPE *)p_bit_stream->p_byte)++) ) );
 311     }
 312 }
 313
 314 /*****************************************************************************
 315  * RemoveBits : removes i_bits bits from the bit buffer
 316  *****************************************************************************/
 317 static __inline__ void RemoveBits( bit_stream_t * p_bit_stream, int i_bits )
 318 {
 319     p_bit_stream->fifo.i_available -= i_bits;
 320
 321     if( p_bit_stream->fifo.i_available >= 0 )
 322     {
 323         p_bit_stream->fifo.buffer <<= i_bits;
 324         return;
 325     }
 326     p_bit_stream->fifo.buffer = GetWord( p_bit_stream )
 327                             << ( -p_bit_stream->fifo.i_available );
 328     p_bit_stream->fifo.i_available += sizeof(WORD_TYPE) * 8;
 329 }
 330
 331 /*****************************************************************************
 332  * RemoveBits32 : removes 32 bits from the bit buffer (and as a side effect,
 333  *                refill it). This should be faster than RemoveBits, though
 334  *                RemoveBits will work, too.
 335  *****************************************************************************/
 336 static __inline__ void RemoveBits32( bit_stream_t * p_bit_stream )
 337 {
 338     p_bit_stream->fifo.buffer = GetWord( p_bit_stream )
 339                         << (32 - p_bit_stream->fifo.i_available);
 340 }
 341
 342 /*****************************************************************************
 343  * GetBits : returns i_bits bits from the bit stream and removes them
 344  *****************************************************************************/
 345 static __inline__ WORD_TYPE GetBits( bit_stream_t * p_bit_stream, int i_bits )
 346 {
 347     u32             i_result;
 348
 349     p_bit_stream->fifo.i_available -= i_bits;
 350     if( p_bit_stream->fifo.i_available >= 0 )
 351     {
 352         i_result = p_bit_stream->fifo.buffer >> (8 * sizeof(WORD_TYPE) - i_bits);
 353         p_bit_stream->fifo.buffer <<= i_bits;
 354         return( i_result );
 355     }
 356
 357     i_result = p_bit_stream->fifo.buffer >> (8 * sizeof(WORD_TYPE) - i_bits);
 358     p_bit_stream->fifo.buffer = GetWord( p_bit_stream );
 359     i_result |= p_bit_stream->fifo.buffer
 360                              >> (8 * sizeof(WORD_TYPE)
 361                                      + p_bit_stream->fifo.i_available);
 362     p_bit_stream->fifo.buffer <<= ( -p_bit_stream->fifo.i_available );
 363     p_bit_stream->fifo.i_available += sizeof(WORD_TYPE) * 8;
 364
 365     return( i_result );
 366 }
 367
 368 /*****************************************************************************
 369  * GetBits32 : returns 32 bits from the bit stream and removes them
 370  *****************************************************************************/
 371 static __inline__ WORD_TYPE GetBits32( bit_stream_t * p_bit_stream )
 372 {
 373     WORD_TYPE               i_result;
 374
 375     i_result = p_bit_stream->fifo.buffer;
 376     p_bit_stream->fifo.buffer = GetWord( p_bit_stream );
 377     i_result |= p_bit_stream->fifo.buffer
 378                              >> (p_bit_stream->fifo.i_available);
 379     p_bit_stream->fifo.buffer <<= (8 * sizeof(WORD_TYPE)
 380                                     - p_bit_stream->fifo.i_available);
 381
 382     return( i_result );
 383 }
 384
 385 /*****************************************************************************
 386  * RealignBits : realigns the bit buffer on an 8-bit boundary
 387  *****************************************************************************/
 388 static __inline__ void RealignBits( bit_stream_t * p_bit_stream )
 389 {
 390     p_bit_stream->fifo.buffer <<= (p_bit_stream->fifo.i_available & 0x7);
 391     p_bit_stream->fifo.i_available &= ~0x7;
 392 }
 393
 394
 395 /*
 396  * Philosophy of the third implementation : the decoder asks for n bytes,
 397  * and we will copy them in its buffer.
 398  */
 399
 400 /*****************************************************************************
 401  * GetChunk : reads a large chunk of data
 402  *****************************************************************************
 403  * The position in the stream must be byte-aligned, if unsure call
 404  * RealignBits(). p_buffer must to a buffer at least as big as i_buf_len
 405  * otherwise your code will crash.
 406  *****************************************************************************/
 407 static __inline__ void GetChunk( bit_stream_t * p_bit_stream,
 408                                  byte_t * p_buffer, size_t i_buf_len )
 409 {
 410     ptrdiff_t           i_available;
 411
 412     if( (i_available = p_bit_stream->p_end - p_bit_stream->p_byte)
 413             >= i_buf_len )
 414     {
 415         memcpy( p_buffer, p_bit_stream->p_byte, i_buf_len );
 416         p_bit_stream->p_byte += i_buf_len;
 417     }
 418     else
 419     {
 420         do
 421         {
 422             memcpy( p_buffer, p_bit_stream->p_byte, i_available );
 423             p_bit_stream->p_byte = p_bit_stream->p_end;
 424             p_buffer += i_available;
 425             i_buf_len -= i_available;
 426             p_bit_stream->pf_next_data_packet( p_bit_stream );
 427         }
 428         while( (i_available = p_bit_stream->p_end - p_bit_stream->p_byte)
 429                 <= i_buf_len );
 430
 431         if( i_buf_len )
 432         {
 433             memcpy( p_buffer, p_bit_stream->p_byte, i_buf_len );
 434             p_bit_stream->p_byte += i_buf_len;
 435         }
 436     }
 437 }
 438
 439
 440 /*
 441  * Communication interface between input and decoders
 442  */
 443
 444 /*****************************************************************************
 445  * decoder_config_t
 446  *****************************************************************************
 447  * Standard pointers given to the decoders as a toolbox.
 448  *****************************************************************************/
 449 typedef struct decoder_config_s
 450 {
 451     u16                     i_id;
 452     u8                      i_type;         /* type of the elementary stream */
 453
 454     struct stream_ctrl_s *  p_stream_ctrl;
 455     struct decoder_fifo_s * p_decoder_fifo;
 456     void                 (* pf_init_bit_stream)( struct bit_stream_s *,
 457                                                  struct decoder_fifo_s * );
 458 } decoder_config_t;
 459
 460 /*****************************************************************************
 461  * vdec_config_t
 462  *****************************************************************************
 463  * Pointers given to video decoders threads.
 464  *****************************************************************************/
 465 struct vout_thread_s;
 466
 467 typedef struct vdec_config_s
 468 {
 469     struct vout_thread_s *  p_vout;
 470
 471     struct picture_s *   (* pf_create_picture)( struct vout_thread_s *,
 472                                                 int i_type, int i_width,
 473                                                 int i_height );
 474     void                 (* pf_destroy_picture)( struct vout_thread_s *,
 475                                                  struct picture_s * );
 476     void                 (* pf_display_picture)( struct vout_thread_s *,
 477                                                  struct picture_s * );
 478     void                 (* pf_date_picture)( struct vout_thread_s *,
 479                                               struct picture_s *, mtime_t date );
 480     void                 (* pf_link_picture)( struct vout_thread_s *,
 481                                               struct picture_s *, mtime_t date );
 482     void                 (* pf_unlink_picture)( struct vout_thread_s *,
 483                                                 struct picture_s *, mtime_t date );
 484     struct subpicture_s *(* pf_create_subpicture)( struct vout_thread_s *,
 485                                                    int i_type, int i_size );
 486     void                 (* pf_destroy_subpicture)( struct vout_thread_s *,
 487                                                     struct subpicture_s * );
 488     void                 (* pf_display_subpicture)( struct vout_thread_s *,
 489                                                     struct subpicture_s * );
 490
 491     decoder_config_t        decoder_config;
 492 } vdec_config_t;
 493
 494 /*****************************************************************************
 495  * adec_config_t
 496  *****************************************************************************
 497  * Pointers given to audio decoders threads.
 498  *****************************************************************************/
 499 struct aout_thread_s;
 500
 501 typedef struct adec_config_s
 502 {
 503     struct aout_thread_s *  p_aout;
 504
 505     struct aout_fifo_s * (* pf_create_fifo)( struct aout_thread_s *,
 506                                             struct aout_fifo_s * );
 507     void                 (* pf_destroy_fifo)( struct aout_thread_s *);
 508
 509     decoder_config_t        decoder_config;
 510 } adec_config_t;
 511
 512
 513 /*
 514  * Communication interface between decoders and input
 515  */
 516
 517 /*****************************************************************************
 518  * decoder_capabilities_t
 519  *****************************************************************************
 520  * Structure returned by a call to GetCapabilities() of the decoder.
 521  *****************************************************************************/
 522 typedef struct decoder_capabilities_s
 523 {
 524     int                     i_dec_type;
 525     u8                      i_stream_type;   /* == i_type in es_descriptor_t */
 526     int                     i_weight; /* for a given stream type, the decoder
 527                                        * with higher weight will be spawned  */
 528
 529     vlc_thread_t         (* pf_create_thread)( void * );
 530 } decoder_capabilities_t;
 531
 532 /* Decoder types */
 533 #define NONE_D              0
 534 #define VIDEO_D             1
 535 #define AUDIO_D             2