git.sesse.net Git - vlc/blob - modules/codec/lpcm.c

   1 /*****************************************************************************
   2  * lpcm.c: lpcm decoder/packetizer module
   3  *****************************************************************************
   4  * Copyright (C) 1999-2008 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Samuel Hocevar <sam@zoy.org>
   8  *          Henri Fallon <henri@videolan.org>
   9  *          Christophe Massiot <massiot@via.ecp.fr>
  10  *          Gildas Bazin <gbazin@videolan.org>
  11  *          Lauren Aimar <fenrir _AT_ videolan _DOT_ org >
  12  *
  13  * This program is free software; you can redistribute it and/or modify
  14  * it under the terms of the GNU General Public License as published by
  15  * the Free Software Foundation; either version 2 of the License, or
  16  * (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU General Public License for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License
  24  * along with this program; if not, write to the Free Software
  25  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  26  *****************************************************************************/
  27
  28 /*****************************************************************************
  29  * Preamble
  30  *****************************************************************************/
  31 #ifdef HAVE_CONFIG_H
  32 # include "config.h"
  33 #endif
  34
  35 #include <vlc_common.h>
  36 #include <vlc_plugin.h>
  37 #include <vlc_codec.h>
  38 #include <vlc_aout.h>
  39 #include <assert.h>
  40
  41 /*****************************************************************************
  42  * Module descriptor
  43  *****************************************************************************/
  44 static int  OpenDecoder   ( vlc_object_t * );
  45 static int  OpenPacketizer( vlc_object_t * );
  46 static void CloseCommon   ( vlc_object_t * );
  47
  48 #ifdef ENABLE_SOUT
  49 static int  OpenEncoder   ( vlc_object_t * );
  50 static void CloseEncoder  ( vlc_object_t * );
  51 static block_t *EncodeFrames( encoder_t *, aout_buffer_t * );
  52 #endif
  53
  54 vlc_module_begin ()
  55
  56     set_category( CAT_INPUT )
  57     set_subcategory( SUBCAT_INPUT_ACODEC )
  58     set_description( N_("Linear PCM audio decoder") )
  59     set_capability( "decoder", 100 )
  60     set_callbacks( OpenDecoder, CloseCommon )
  61
  62     add_submodule ()
  63     set_description( N_("Linear PCM audio packetizer") )
  64     set_capability( "packetizer", 100 )
  65     set_callbacks( OpenPacketizer, CloseCommon )
  66
  67 #ifdef ENABLE_SOUT
  68     add_submodule ()
  69     set_description( N_("Linear PCM audio encoder") )
  70     set_capability( "encoder", 100 )
  71     set_callbacks( OpenEncoder, CloseEncoder )
  72     add_shortcut( "lpcm" )
  73 #endif
  74
  75 vlc_module_end ()
  76
  77
  78 /*****************************************************************************
  79  * decoder_sys_t : lpcm decoder descriptor
  80  *****************************************************************************/
  81 struct decoder_sys_t
  82 {
  83     /* Module mode */
  84     bool b_packetizer;
  85
  86     /*
  87      * Output properties
  88      */
  89     date_t   end_date;
  90
  91     /* */
  92     unsigned i_header_size;
  93     int      i_type;
  94 };
  95
  96 #ifdef ENABLE_SOUT
  97 struct encoder_sys_t
  98 {
  99     int     i_channels;
 100     int     i_rate;
 101
 102     int     i_frame_samples;
 103     uint8_t *p_buffer;
 104     int     i_buffer_used;
 105     int     i_frame_num;
 106 };
 107 #endif
 108
 109 /*
 110  * LPCM DVD header :
 111  * - frame number (8 bits)
 112  * - unknown (16 bits) == 0x0003 ?
 113  * - unknown (4 bits)
 114  * - current frame (4 bits)
 115  * - unknown (2 bits)
 116  * - frequency (2 bits) 0 == 48 kHz, 1 == 32 kHz, 2 == ?, 3 == ?
 117  * - unknown (1 bit)
 118  * - number of channels - 1 (3 bits) 1 == 2 channels
 119  * - start code (8 bits) == 0x80
 120  *
 121  * LPCM DVD-A header (http://dvd-audio.sourceforge.net/spec/aob.shtml)
 122  * - continuity counter (8 bits, clipped to 0x00-0x1f)
 123  * - header size (16 bits)
 124  * - byte pointer to start of first audio frame.
 125  * - unknown (8bits, 0x10 for stereo, 0x00 for surround)
 126  * - sample size (4+4 bits)
 127  * - samplerate (4+4 bits)
 128  * - unknown (8 bits)
 129  * - group assignment (8 bits)
 130  * - unknown (8 bits)
 131  * - padding(variable)
 132  *
 133  * LPCM BD header :
 134  * - unkown (16 bits)
 135  * - number of channels (4 bits)
 136  * - frequency (4 bits)
 137  * - bits per sample (2 bits)
 138  * - unknown (6 bits)
 139  */
 140
 141 #define LPCM_VOB_HEADER_LEN (6)
 142 #define LPCM_AOB_HEADER_LEN (11)
 143 #define LPCM_BD_HEADER_LEN (4)
 144
 145 enum
 146 {
 147     LPCM_VOB,
 148     LPCM_AOB,
 149     LPCM_BD,
 150 };
 151
 152 typedef struct
 153 {
 154     unsigned i_channels;
 155     bool     b_used;
 156     unsigned pi_position[6];
 157 } aob_group_t;
 158
 159 /*****************************************************************************
 160  * Local prototypes
 161  *****************************************************************************/
 162 static void *DecodeFrame  ( decoder_t *, block_t ** );
 163
 164 /* */
 165 static int VobHeader( unsigned *pi_rate,
 166                       unsigned *pi_channels, unsigned *pi_original_channels,
 167                       unsigned *pi_bits,
 168                       const uint8_t *p_header );
 169 static void VobExtract( aout_buffer_t *, block_t *, unsigned i_bits );
 170 /* */
 171 static int AobHeader( unsigned *pi_rate,
 172                       unsigned *pi_channels, unsigned *pi_layout,
 173                       unsigned *pi_bits,
 174                       unsigned *pi_padding,
 175                       aob_group_t g[2],
 176                       const uint8_t *p_header );
 177 static void AobExtract( aout_buffer_t *, block_t *, unsigned i_bits, aob_group_t p_group[2] );
 178 /* */
 179 static int BdHeader( unsigned *pi_rate,
 180                      unsigned *pi_channels, unsigned *pi_original_channels,
 181                      unsigned *pi_bits,
 182                      const uint8_t *p_header );
 183 static void BdExtract( aout_buffer_t *, block_t * );
 184
 185
 186 /*****************************************************************************
 187  * OpenCommon:
 188  *****************************************************************************/
 189 static int OpenCommon( vlc_object_t *p_this, bool b_packetizer )
 190 {
 191     decoder_t *p_dec = (decoder_t*)p_this;
 192     decoder_sys_t *p_sys;
 193     int i_type;
 194     int i_header_size;
 195
 196     switch( p_dec->fmt_in.i_codec )
 197     {
 198     /* DVD LPCM */
 199     case VLC_CODEC_DVD_LPCM:
 200         i_type = LPCM_VOB;
 201         i_header_size = LPCM_VOB_HEADER_LEN;
 202         break;
 203     /* DVD-Audio LPCM */
 204     case VLC_CODEC_DVDA_LPCM:
 205         i_type = LPCM_AOB;
 206         i_header_size = LPCM_AOB_HEADER_LEN;
 207         break;
 208     /* BD LPCM */
 209     case VLC_CODEC_BD_LPCM:
 210         i_type = LPCM_BD;
 211         i_header_size = LPCM_BD_HEADER_LEN;
 212         break;
 213     default:
 214         return VLC_EGENERIC;
 215     }
 216
 217     /* Allocate the memory needed to store the decoder's structure */
 218     if( ( p_dec->p_sys = p_sys = malloc(sizeof(decoder_sys_t)) ) == NULL )
 219         return VLC_ENOMEM;
 220
 221     /* Misc init */
 222     p_sys->b_packetizer = b_packetizer;
 223     date_Set( &p_sys->end_date, 0 );
 224     p_sys->i_type = i_type;
 225     p_sys->i_header_size = i_header_size;
 226
 227     /* Set output properties */
 228     p_dec->fmt_out.i_cat = AUDIO_ES;
 229
 230     if( b_packetizer )
 231     {
 232         switch( i_type )
 233         {
 234         case LPCM_VOB:
 235             p_dec->fmt_out.i_codec = VLC_CODEC_DVD_LPCM;
 236             break;
 237         case LPCM_AOB:
 238             p_dec->fmt_out.i_codec = VLC_CODEC_DVDA_LPCM;
 239             break;
 240         default:
 241             assert(0);
 242         case LPCM_BD:
 243             p_dec->fmt_out.i_codec = VLC_CODEC_BD_LPCM;
 244             break;
 245         }
 246     }
 247     else
 248     {
 249         switch( p_dec->fmt_out.audio.i_bitspersample )
 250         {
 251         case 24:
 252         case 20:
 253             p_dec->fmt_out.i_codec = VLC_CODEC_S24B;
 254             p_dec->fmt_out.audio.i_bitspersample = 24;
 255             break;
 256         default:
 257             p_dec->fmt_out.i_codec = VLC_CODEC_S16B;
 258             p_dec->fmt_out.audio.i_bitspersample = 16;
 259             break;
 260         }
 261     }
 262
 263     /* Set callback */
 264     p_dec->pf_decode_audio = (aout_buffer_t *(*)(decoder_t *, block_t **))
 265         DecodeFrame;
 266     p_dec->pf_packetize    = (block_t *(*)(decoder_t *, block_t **))
 267         DecodeFrame;
 268
 269     return VLC_SUCCESS;
 270 }
 271 static int OpenDecoder( vlc_object_t *p_this )
 272 {
 273     return OpenCommon( p_this, false );
 274 }
 275 static int OpenPacketizer( vlc_object_t *p_this )
 276 {
 277     return OpenCommon( p_this, true );
 278 }
 279
 280 /*****************************************************************************
 281  * DecodeFrame: decodes an lpcm frame.
 282  ****************************************************************************
 283  * Beware, this function must be fed with complete frames (PES packet).
 284  *****************************************************************************/
 285 static void *DecodeFrame( decoder_t *p_dec, block_t **pp_block )
 286 {
 287     decoder_sys_t *p_sys = p_dec->p_sys;
 288     block_t       *p_block;
 289     unsigned int  i_rate = 0, i_original_channels = 0, i_channels = 0, i_bits = 0;
 290     int           i_frame_length;
 291
 292     if( !pp_block || !*pp_block ) return NULL;
 293
 294     p_block = *pp_block;
 295     *pp_block = NULL; /* So the packet doesn't get re-sent */
 296
 297     /* Date management */
 298     if( p_block->i_pts > VLC_TS_INVALID &&
 299         p_block->i_pts != date_Get( &p_sys->end_date ) )
 300     {
 301         date_Set( &p_sys->end_date, p_block->i_pts );
 302     }
 303
 304     if( !date_Get( &p_sys->end_date ) )
 305     {
 306         /* We've just started the stream, wait for the first PTS. */
 307         block_Release( p_block );
 308         return NULL;
 309     }
 310
 311     if( p_block->i_buffer <= p_sys->i_header_size )
 312     {
 313         msg_Err(p_dec, "frame is too short");
 314         block_Release( p_block );
 315         return NULL;
 316     }
 317
 318     int i_ret;
 319     unsigned i_padding = 0;
 320     aob_group_t p_aob_group[2];
 321     switch( p_sys->i_type )
 322     {
 323     case LPCM_VOB:
 324         i_ret = VobHeader( &i_rate, &i_channels, &i_original_channels, &i_bits,
 325                            p_block->p_buffer );
 326         break;
 327     case LPCM_AOB:
 328         i_ret = AobHeader( &i_rate, &i_channels, &i_original_channels, &i_bits, &i_padding,
 329                            p_aob_group,
 330                            p_block->p_buffer );
 331         break;
 332     case LPCM_BD:
 333         i_ret = BdHeader( &i_rate, &i_channels, &i_original_channels, &i_bits,
 334                           p_block->p_buffer );
 335         break;
 336     default:
 337         abort();
 338     }
 339
 340     if( i_ret || p_block->i_buffer <= p_sys->i_header_size + i_padding )
 341     {
 342         msg_Warn( p_dec, "no frame sync or too small frame" );
 343         block_Release( p_block );
 344         return NULL;
 345     }
 346
 347     /* Set output properties */
 348     if( p_dec->fmt_out.audio.i_rate != i_rate )
 349     {
 350         date_Init( &p_sys->end_date, i_rate, 1 );
 351         date_Set( &p_sys->end_date, p_block->i_pts );
 352     }
 353     p_dec->fmt_out.audio.i_rate = i_rate;
 354     p_dec->fmt_out.audio.i_channels = i_channels;
 355     p_dec->fmt_out.audio.i_original_channels = i_original_channels;
 356     p_dec->fmt_out.audio.i_physical_channels = i_original_channels & AOUT_CHAN_PHYSMASK;
 357
 358     i_frame_length = (p_block->i_buffer - p_sys->i_header_size - i_padding) / i_channels * 8 / i_bits;
 359
 360     if( p_sys->b_packetizer )
 361     {
 362         p_block->i_pts = p_block->i_dts = date_Get( &p_sys->end_date );
 363         p_block->i_length =
 364             date_Increment( &p_sys->end_date, i_frame_length ) -
 365             p_block->i_pts;
 366
 367         /* Just pass on the incoming frame */
 368         return p_block;
 369     }
 370     else
 371     {
 372         /* */
 373         if( i_bits == 16 )
 374         {
 375             p_dec->fmt_out.i_codec = VLC_CODEC_S16B;
 376             p_dec->fmt_out.audio.i_bitspersample = 16;
 377         }
 378         else
 379         {
 380             p_dec->fmt_out.i_codec = VLC_CODEC_S24B;
 381             p_dec->fmt_out.audio.i_bitspersample = 24;
 382         }
 383
 384         /* */
 385         aout_buffer_t *p_aout_buffer;
 386         p_aout_buffer = decoder_NewAudioBuffer( p_dec, i_frame_length );
 387         if( !p_aout_buffer )
 388             return NULL;
 389
 390         p_aout_buffer->i_pts = date_Get( &p_sys->end_date );
 391         p_aout_buffer->i_length =
 392             date_Increment( &p_sys->end_date, i_frame_length )
 393             - p_aout_buffer->i_pts;
 394
 395         p_block->p_buffer += p_sys->i_header_size + i_padding;
 396         p_block->i_buffer -= p_sys->i_header_size + i_padding;
 397
 398         switch( p_sys->i_type )
 399         {
 400         case LPCM_VOB:
 401             VobExtract( p_aout_buffer, p_block, i_bits );
 402             break;
 403         case LPCM_AOB:
 404             AobExtract( p_aout_buffer, p_block, i_bits, p_aob_group );
 405             break;
 406         default:
 407             assert(0);
 408         case LPCM_BD:
 409             BdExtract( p_aout_buffer, p_block );
 410             break;
 411         }
 412
 413         block_Release( p_block );
 414         return p_aout_buffer;
 415     }
 416 }
 417
 418 /*****************************************************************************
 419  * CloseCommon : lpcm decoder destruction
 420  *****************************************************************************/
 421 static void CloseCommon( vlc_object_t *p_this )
 422 {
 423     decoder_t *p_dec = (decoder_t*)p_this;
 424     free( p_dec->p_sys );
 425 }
 426
 427 #ifdef ENABLE_SOUT
 428 /*****************************************************************************
 429  * OpenEncoder: lpcm encoder construction
 430  *****************************************************************************/
 431 static int OpenEncoder( vlc_object_t *p_this )
 432 {
 433     encoder_t *p_enc = (encoder_t *)p_this;
 434     encoder_sys_t *p_sys;
 435
 436     /* We only support DVD LPCM yet. */
 437     if( p_enc->fmt_out.i_codec != VLC_CODEC_DVD_LPCM )
 438         return VLC_EGENERIC;
 439
 440     if( p_enc->fmt_in.audio.i_rate != 48000 &&
 441         p_enc->fmt_in.audio.i_rate != 96000 &&
 442         p_enc->fmt_in.audio.i_rate != 44100 &&
 443         p_enc->fmt_in.audio.i_rate != 32000 )
 444     {
 445         msg_Err( p_enc, "DVD LPCM supports only sample rates of 48, 96, 44.1 or 32 kHz" );
 446         return VLC_EGENERIC;
 447     }
 448
 449     if( p_enc->fmt_in.audio.i_channels > 8 )
 450     {
 451         msg_Err( p_enc, "DVD LPCM supports a maximum of eight channels" );
 452         return VLC_EGENERIC;
 453     }
 454
 455     /* Allocate the memory needed to store the encoder's structure */
 456     if( ( p_enc->p_sys = p_sys =
 457           (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
 458         return VLC_ENOMEM;
 459
 460     /* In DVD LCPM, a frame is always 150 PTS ticks. */
 461     p_sys->i_frame_samples = p_enc->fmt_in.audio.i_rate * 150 / 90000;
 462     p_sys->p_buffer = (uint8_t *)malloc(
 463         p_sys->i_frame_samples *
 464         p_enc->fmt_in.audio.i_channels *
 465         p_enc->fmt_in.audio.i_bitspersample);
 466     p_sys->i_buffer_used = 0;
 467     p_sys->i_frame_num = 0;
 468
 469     p_sys->i_channels = p_enc->fmt_in.audio.i_channels;
 470     p_sys->i_rate = p_enc->fmt_in.audio.i_rate;
 471
 472     p_enc->pf_encode_audio = EncodeFrames;
 473     p_enc->fmt_in.i_codec = p_enc->fmt_out.i_codec;
 474
 475     p_enc->fmt_in.audio.i_bitspersample = 16;
 476     p_enc->fmt_in.i_codec = VLC_CODEC_S16B;
 477
 478     p_enc->fmt_out.i_bitrate =
 479         p_enc->fmt_in.audio.i_channels *
 480         p_enc->fmt_in.audio.i_rate *
 481         p_enc->fmt_in.audio.i_bitspersample *
 482         (p_sys->i_frame_samples + LPCM_VOB_HEADER_LEN) /
 483         p_sys->i_frame_samples;
 484
 485     return VLC_SUCCESS;
 486 }
 487
 488 /*****************************************************************************
 489  * CloseEncoder: lpcm encoder destruction
 490  *****************************************************************************/
 491 static void CloseEncoder ( vlc_object_t *p_this )
 492 {
 493     VLC_UNUSED(p_this);
 494 }
 495
 496 /*****************************************************************************
 497  * EncodeFrames: encode zero or more LCPM audio packets
 498  *****************************************************************************/
 499 static block_t *EncodeFrames( encoder_t *p_enc, aout_buffer_t *p_aout_buf )
 500 {
 501     encoder_sys_t *p_sys = p_enc->p_sys;
 502     block_t *p_first_block = NULL, *p_last_block = NULL;
 503
 504     if( !p_aout_buf || !p_aout_buf->i_buffer ) return NULL;
 505
 506     const int i_num_frames = ( p_sys->i_buffer_used + p_aout_buf->i_nb_samples ) /
 507         p_sys->i_frame_samples;
 508     const int i_leftover_samples = ( p_sys->i_buffer_used + p_aout_buf->i_nb_samples ) %
 509         p_sys->i_frame_samples;
 510     const int i_frame_size = p_sys->i_frame_samples * p_sys->i_channels * 2 + LPCM_VOB_HEADER_LEN;
 511     const int i_start_offset = -p_sys->i_buffer_used;
 512
 513     uint8_t i_freq_code = 0;
 514
 515     switch( p_sys->i_rate ) {
 516     case 48000:
 517         i_freq_code = 0;
 518         break;
 519     case 96000:
 520         i_freq_code = 1;
 521         break;
 522     case 44100:
 523         i_freq_code = 2;
 524         break;
 525     case 32000:
 526         i_freq_code = 3;
 527         break;
 528     default:
 529         assert(0);
 530     }
 531
 532     int i_bytes_consumed = 0;
 533
 534     for ( int i = 0; i < i_num_frames; ++i )
 535     {
 536         block_t *p_block = block_New( p_enc, i_frame_size );
 537         if( !p_block )
 538             return NULL;
 539
 540         uint8_t *frame = (uint8_t *)p_block->p_buffer;
 541         frame[0] = 1;  /* one frame in packet */
 542         frame[1] = 0;
 543         frame[2] = 0;  /* no first access unit */
 544         frame[3] = (p_sys->i_frame_num + i) & 0x1f;  /* no emphasis, no mute */
 545         frame[4] = (i_freq_code << 4) | (p_sys->i_channels - 1);
 546         frame[5] = 0x80;  /* neutral dynamic range */
 547
 548         const int i_consume_samples = p_sys->i_frame_samples - p_sys->i_buffer_used;
 549         const int i_kept_bytes = p_sys->i_buffer_used * p_sys->i_channels * 2;
 550         const int i_consume_bytes = i_consume_samples * p_sys->i_channels * 2;
 551
 552         memcpy( frame + 6, p_sys->p_buffer, i_kept_bytes );
 553         memcpy( frame + 6 + i_kept_bytes, p_aout_buf->p_buffer + i_bytes_consumed, i_consume_bytes );
 554
 555         p_sys->i_frame_num++;
 556         p_sys->i_buffer_used = 0;
 557         i_bytes_consumed += i_consume_bytes;
 558
 559         p_block->i_dts = p_block->i_pts = p_aout_buf->i_pts +
 560             (i * p_sys->i_frame_samples + i_start_offset) * CLOCK_FREQ / p_sys->i_rate;
 561         p_block->i_length = p_sys->i_frame_samples * CLOCK_FREQ / p_sys->i_rate;
 562
 563         if( !p_first_block )
 564             p_first_block = p_last_block = p_block;
 565         else
 566             p_last_block = p_last_block->p_next = p_block;
 567     }
 568
 569     memcpy( p_sys->p_buffer,
 570             p_aout_buf->p_buffer + i_bytes_consumed,
 571             i_leftover_samples * p_sys->i_channels * 2 );
 572     p_sys->i_buffer_used = i_leftover_samples;
 573
 574     return p_first_block;
 575 }
 576 #endif
 577
 578 /*****************************************************************************
 579  *
 580  *****************************************************************************/
 581 static int VobHeader( unsigned *pi_rate,
 582                       unsigned *pi_channels, unsigned *pi_original_channels,
 583                       unsigned *pi_bits,
 584                       const uint8_t *p_header )
 585 {
 586     const uint8_t i_header = p_header[4];
 587
 588     switch( (i_header >> 4) & 0x3 )
 589     {
 590     case 0:
 591         *pi_rate = 48000;
 592         break;
 593     case 1:
 594         *pi_rate = 96000;
 595         break;
 596     case 2:
 597         *pi_rate = 44100;
 598         break;
 599     case 3:
 600         *pi_rate = 32000;
 601         break;
 602     }
 603
 604     *pi_channels = (i_header & 0x7) + 1;
 605     switch( *pi_channels - 1 )
 606     {
 607     case 0:
 608         *pi_original_channels = AOUT_CHAN_CENTER;
 609         break;
 610     case 1:
 611         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
 612         break;
 613     case 2:
 614         /* This is unsure. */
 615         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_LFE;
 616         break;
 617     case 3:
 618         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
 619                                | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
 620         break;
 621     case 4:
 622         /* This is unsure. */
 623         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
 624                                | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
 625                                | AOUT_CHAN_LFE;
 626         break;
 627     case 5:
 628         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
 629                                | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
 630                                | AOUT_CHAN_CENTER | AOUT_CHAN_LFE;
 631         break;
 632     case 6:
 633         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
 634                                | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
 635                                | AOUT_CHAN_CENTER | AOUT_CHAN_MIDDLELEFT
 636                                | AOUT_CHAN_MIDDLERIGHT;
 637         break;
 638     case 7:
 639         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
 640                                | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
 641                                | AOUT_CHAN_CENTER | AOUT_CHAN_MIDDLELEFT
 642                                | AOUT_CHAN_MIDDLERIGHT | AOUT_CHAN_LFE;
 643         break;
 644     }
 645
 646     switch( (i_header >> 6) & 0x3 )
 647     {
 648     case 2:
 649         *pi_bits = 24;
 650         break;
 651     case 1:
 652         *pi_bits = 20;
 653         break;
 654     case 0:
 655     default:
 656         *pi_bits = 16;
 657         break;
 658     }
 659
 660     /* Check frame sync and drop it. */
 661     if( p_header[5] != 0x80 )
 662         return -1;
 663     return 0;
 664 }
 665
 666 static const unsigned p_aob_group1[21][6] = {
 667     { AOUT_CHAN_CENTER, 0 },
 668     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 669     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 670     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 671     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 672     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 673     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 674     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 675     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 676     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 677     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 678     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 679     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, 0 },
 680     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_CENTER,   0 },
 681     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_CENTER,   0 },
 682     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_CENTER,   0 },
 683     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_CENTER,   0 },
 684     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_CENTER,   0 },
 685     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT, 0  },
 686     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT, 0  },
 687     { AOUT_CHAN_LEFT,   AOUT_CHAN_RIGHT, AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT, 0  },
 688 };
 689 static const unsigned p_aob_group2[21][6] = {
 690     { 0 },
 691     { 0 },
 692     { AOUT_CHAN_REARCENTER, 0 },
 693     { AOUT_CHAN_REARLEFT,   AOUT_CHAN_REARRIGHT,    0 },
 694     { AOUT_CHAN_LFE,        0 },
 695     { AOUT_CHAN_LFE,        AOUT_CHAN_REARCENTER,   0 },
 696     { AOUT_CHAN_LFE,        AOUT_CHAN_REARLEFT,     AOUT_CHAN_REARRIGHT,    0 },
 697     { AOUT_CHAN_CENTER,     0 },
 698     { AOUT_CHAN_CENTER,     AOUT_CHAN_REARCENTER, 0 },
 699     { AOUT_CHAN_CENTER,     AOUT_CHAN_REARLEFT,   AOUT_CHAN_REARRIGHT,    0 },
 700     { AOUT_CHAN_CENTER,     AOUT_CHAN_LFE,        0 },
 701     { AOUT_CHAN_CENTER,     AOUT_CHAN_LFE,        AOUT_CHAN_REARCENTER,   0 },
 702     { AOUT_CHAN_CENTER,     AOUT_CHAN_LFE,        AOUT_CHAN_REARLEFT,     AOUT_CHAN_REARRIGHT,    0 },
 703     { AOUT_CHAN_REARCENTER, 0 },
 704     { AOUT_CHAN_REARLEFT,   AOUT_CHAN_REARRIGHT,    0 },
 705     { AOUT_CHAN_LFE,        0 },
 706     { AOUT_CHAN_LFE,        AOUT_CHAN_REARCENTER,   0 },
 707     { AOUT_CHAN_LFE,        AOUT_CHAN_REARLEFT,     AOUT_CHAN_REARRIGHT,    0 },
 708     { AOUT_CHAN_LFE,        0 },
 709     { AOUT_CHAN_CENTER,     0 },
 710     { AOUT_CHAN_CENTER,     AOUT_CHAN_LFE,          0 },
 711 };
 712
 713 static int AobHeader( unsigned *pi_rate,
 714                       unsigned *pi_channels, unsigned *pi_layout,
 715                       unsigned *pi_bits,
 716                       unsigned *pi_padding,
 717                       aob_group_t g[2],
 718                       const uint8_t *p_header )
 719 {
 720     const unsigned i_header_size = GetWBE( &p_header[1] );
 721     if( i_header_size + 3 < LPCM_AOB_HEADER_LEN )
 722         return VLC_EGENERIC;
 723
 724     *pi_padding = 3+i_header_size - LPCM_AOB_HEADER_LEN;
 725
 726     const int i_index_size_g1 = (p_header[6] >> 4) & 0x0f;
 727     const int i_index_size_g2 = (p_header[6]     ) & 0x0f;
 728     const int i_index_rate_g1 = (p_header[7] >> 4) & 0x0f;
 729     const int i_index_rate_g2 = (p_header[7]     ) & 0x0f;
 730     const int i_assignment     = p_header[9];
 731
 732     /* Validate */
 733     if( i_index_size_g1 > 0x02 ||
 734         ( i_index_size_g2 != 0x0f && i_index_size_g2 > 0x02 ) )
 735         return VLC_EGENERIC;
 736     if( (i_index_rate_g1 & 0x07) > 0x02 ||
 737         ( i_index_rate_g2 != 0x0f && (i_index_rate_g1 & 0x07) > 0x02 ) )
 738         return VLC_EGENERIC;
 739     if( i_assignment > 20 )
 740         return VLC_EGENERIC;
 741
 742     /* */
 743     *pi_bits = 16 + 4 * i_index_size_g1;
 744     if( i_index_rate_g1 & 0x08 )
 745         *pi_rate = 44100 << (i_index_rate_g1 & 0x07);
 746     else
 747         *pi_rate = 48000 << (i_index_rate_g1 & 0x07);
 748
 749
 750     /* Group1 */
 751     unsigned i_channels1 = 0;
 752     unsigned i_layout1 = 0;
 753     for( int i = 0; p_aob_group1[i_assignment][i] != 0; i++ )
 754     {
 755         i_channels1++;
 756         i_layout1 |= p_aob_group1[i_assignment][i];
 757     }
 758     /* Group2 */
 759     unsigned i_channels2 = 0;
 760     unsigned i_layout2 = 0;
 761     if( i_index_size_g2 != 0x0f && i_index_rate_g2 != 0x0f )
 762     {
 763         for( int i = 0; p_aob_group2[i_assignment][i] != 0; i++ )
 764         {
 765             i_channels2++;
 766             i_layout2 |= p_aob_group2[i_assignment][i];
 767         }
 768         assert( (i_layout1 & i_layout2) == 0 );
 769     }
 770     /* It is enabled only when presents and compatible wih group1 */
 771     const bool b_group2_used = i_index_size_g1 == i_index_size_g2 &&
 772                                i_index_rate_g1 == i_index_rate_g2;
 773
 774     /* */
 775     *pi_channels = i_channels1 + ( b_group2_used ? i_channels2 : 0 );
 776     *pi_layout   = i_layout1   | ( b_group2_used ? i_layout2   : 0 );
 777
 778     /* */
 779     for( unsigned i = 0; i < 2; i++ )
 780     {
 781         const unsigned *p_aob = i == 0 ? p_aob_group1[i_assignment] :
 782                                          p_aob_group2[i_assignment];
 783         g[i].i_channels = i == 0 ? i_channels1 :
 784                                    i_channels2;
 785
 786         g[i].b_used = i == 0 || b_group2_used;
 787         if( !g[i].b_used )
 788             continue;
 789         for( unsigned j = 0; j < g[i].i_channels; j++ )
 790         {
 791             g[i].pi_position[j] = 0;
 792             for( int k = 0; pi_vlc_chan_order_wg4[k] != 0; k++ )
 793             {
 794                 const unsigned i_channel = pi_vlc_chan_order_wg4[k];
 795                 if( i_channel == p_aob[j] )
 796                     break;
 797                 if( (*pi_layout) & i_channel )
 798                     g[i].pi_position[j]++;
 799             }
 800         }
 801     }
 802     return VLC_SUCCESS;
 803 }
 804
 805 static int BdHeader( unsigned *pi_rate,
 806                      unsigned *pi_channels, unsigned *pi_original_channels,
 807                      unsigned *pi_bits,
 808                      const uint8_t *p_header )
 809 {
 810     const uint32_t h = GetDWBE( p_header );
 811     switch( ( h & 0xf000) >> 12 )
 812     {
 813     case 1:
 814         *pi_channels = 1;
 815         *pi_original_channels = AOUT_CHAN_CENTER;
 816         break;
 817     case 3:
 818         *pi_channels = 2;
 819         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
 820         break;
 821     case 4:
 822         *pi_channels = 3;
 823         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER;
 824         break;
 825     case 5:
 826         *pi_channels = 3;
 827         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_REARCENTER;
 828         break;
 829     case 6:
 830         *pi_channels = 4;
 831         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
 832                                 AOUT_CHAN_REARCENTER;
 833         break;
 834     case 7:
 835         *pi_channels = 4;
 836         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
 837                                 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
 838         break;
 839     case 8:
 840         *pi_channels = 5;
 841         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
 842                                 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
 843         break;
 844     case 9:
 845         *pi_channels = 6;
 846         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
 847                                 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
 848                                 AOUT_CHAN_LFE;
 849         break;
 850     case 10:
 851         *pi_channels = 7;
 852         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
 853                                 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
 854                                 AOUT_CHAN_MIDDLELEFT | AOUT_CHAN_MIDDLERIGHT;
 855         break;
 856     case 11:
 857         *pi_channels = 8;
 858         *pi_original_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER |
 859                                 AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT |
 860                                 AOUT_CHAN_MIDDLELEFT | AOUT_CHAN_MIDDLERIGHT |
 861                                 AOUT_CHAN_LFE;
 862         break;
 863
 864     default:
 865         return -1;
 866     }
 867     switch( (h >> 6) & 0x03 )
 868     {
 869     case 1:
 870         *pi_bits = 16;
 871         break;
 872     case 2: /* 20 bits but samples are stored on 24 bits */
 873     case 3: /* 24 bits */
 874         *pi_bits = 24;
 875         break;
 876     default:
 877         return -1;
 878     }
 879     switch( (h >> 8) & 0x0f )
 880     {
 881     case 1:
 882         *pi_rate = 48000;
 883         break;
 884     case 4:
 885         *pi_rate = 96000;
 886         break;
 887     case 5:
 888         *pi_rate = 192000;
 889         break;
 890     default:
 891         return -1;
 892     }
 893     return 0;
 894 }
 895
 896 static void VobExtract( aout_buffer_t *p_aout_buffer, block_t *p_block,
 897                         unsigned i_bits )
 898 {
 899     uint8_t *p_out = p_aout_buffer->p_buffer;
 900
 901     /* 20/24 bits LPCM use special packing */
 902     if( i_bits == 24 )
 903     {
 904         while( p_block->i_buffer / 12 )
 905         {
 906             /* Sample 1 */
 907             p_out[0] = p_block->p_buffer[0];
 908             p_out[1] = p_block->p_buffer[1];
 909             p_out[2] = p_block->p_buffer[8];
 910             /* Sample 2 */
 911             p_out[3] = p_block->p_buffer[2];
 912             p_out[4] = p_block->p_buffer[3];
 913             p_out[5] = p_block->p_buffer[9];
 914             /* Sample 3 */
 915             p_out[6] = p_block->p_buffer[4];
 916             p_out[7] = p_block->p_buffer[5];
 917             p_out[8] = p_block->p_buffer[10];
 918             /* Sample 4 */
 919             p_out[9] = p_block->p_buffer[6];
 920             p_out[10] = p_block->p_buffer[7];
 921             p_out[11] = p_block->p_buffer[11];
 922
 923             p_block->i_buffer -= 12;
 924             p_block->p_buffer += 12;
 925             p_out += 12;
 926         }
 927     }
 928     else if( i_bits == 20 )
 929     {
 930         while( p_block->i_buffer / 10 )
 931         {
 932             /* Sample 1 */
 933             p_out[0] = p_block->p_buffer[0];
 934             p_out[1] = p_block->p_buffer[1];
 935             p_out[2] = p_block->p_buffer[8] & 0xF0;
 936             /* Sample 2 */
 937             p_out[3] = p_block->p_buffer[2];
 938             p_out[4] = p_block->p_buffer[3];
 939             p_out[5] = p_block->p_buffer[8] << 4;
 940             /* Sample 3 */
 941             p_out[6] = p_block->p_buffer[4];
 942             p_out[7] = p_block->p_buffer[5];
 943             p_out[8] = p_block->p_buffer[9] & 0xF0;
 944             /* Sample 4 */
 945             p_out[9] = p_block->p_buffer[6];
 946             p_out[10] = p_block->p_buffer[7];
 947             p_out[11] = p_block->p_buffer[9] << 4;
 948
 949             p_block->i_buffer -= 10;
 950             p_block->p_buffer += 10;
 951             p_out += 12;
 952         }
 953     }
 954     else
 955     {
 956         assert( i_bits == 16 );
 957         memcpy( p_out, p_block->p_buffer, p_block->i_buffer );
 958     }
 959 }
 960 static void AobExtract( aout_buffer_t *p_aout_buffer,
 961                         block_t *p_block, unsigned i_bits, aob_group_t p_group[2] )
 962 {
 963     const unsigned i_channels = p_group[0].i_channels +
 964                                 ( p_group[1].b_used ? p_group[1].i_channels : 0 );
 965     uint8_t *p_out = p_aout_buffer->p_buffer;
 966
 967     while( p_block->i_buffer > 0 )
 968     {
 969         for( int i = 0; i < 2; i++ )
 970         {
 971             const aob_group_t *g = &p_group[1-i];
 972             const unsigned int i_group_size = 2 * g->i_channels * i_bits / 8;
 973
 974             if( p_block->i_buffer < i_group_size )
 975             {
 976                 p_block->i_buffer = 0;
 977                 break;
 978             }
 979             for( unsigned n = 0; n < 2; n++ )
 980             {
 981                 for( unsigned j = 0; j < g->i_channels && g->b_used; j++ )
 982                 {
 983                     const int i_src = n * g->i_channels + j;
 984                     const int i_dst = n * i_channels + g->pi_position[j];
 985
 986                     if( i_bits == 24 )
 987                     {
 988                         p_out[3*i_dst+0] = p_block->p_buffer[2*i_src+0];
 989                         p_out[3*i_dst+1] = p_block->p_buffer[2*i_src+1];
 990                         p_out[3*i_dst+2] = p_block->p_buffer[4*g->i_channels+i_src];
 991                     }
 992                     else if( i_bits == 20 )
 993                     {
 994                         p_out[3*i_dst+0] = p_block->p_buffer[2*i_src+0];
 995                         p_out[3*i_dst+1] = p_block->p_buffer[2*i_src+1];
 996                         if( n == 0 )
 997                             p_out[3*i_dst+2] = (p_block->p_buffer[4*g->i_channels+i_src]     ) & 0xf0;
 998                         else
 999                             p_out[3*i_dst+2] = (p_block->p_buffer[4*g->i_channels+i_src] << 4) & 0xf0;
1000                     }
1001                     else
1002                     {
1003                         assert( i_bits == 16 );
1004                         p_out[2*i_dst+0] = p_block->p_buffer[2*i_src+0];
1005                         p_out[2*i_dst+1] = p_block->p_buffer[2*i_src+1];
1006                     }
1007                 }
1008             }
1009
1010             p_block->i_buffer -= i_group_size;
1011             p_block->p_buffer += i_group_size;
1012         }
1013         /* */
1014         p_out += (i_bits == 16 ? 2 : 3) * i_channels * 2;
1015
1016     }
1017 }
1018 static void BdExtract( aout_buffer_t *p_aout_buffer, block_t *p_block )
1019 {
1020     memcpy( p_aout_buffer->p_buffer, p_block->p_buffer, p_block->i_buffer );
1021 }
1022