mirror of
https://github.com/videolan/vlc.git
synced 2025-01-11 02:08:36 +08:00
* modules/demux/wav.c: proper channels re-ordering for multichannel audio (this should really be implemented as a common facility).
This commit is contained in:
parent
f70500d263
commit
be17d8e9bc
@ -28,9 +28,12 @@
|
|||||||
|
|
||||||
#include <vlc/vlc.h>
|
#include <vlc/vlc.h>
|
||||||
#include <vlc/input.h>
|
#include <vlc/input.h>
|
||||||
|
#include <vlc/aout.h>
|
||||||
|
|
||||||
#include <codecs.h>
|
#include <codecs.h>
|
||||||
|
|
||||||
|
#define MAX_CHANNELS 6
|
||||||
|
|
||||||
/*****************************************************************************
|
/*****************************************************************************
|
||||||
* Module descriptor
|
* Module descriptor
|
||||||
*****************************************************************************/
|
*****************************************************************************/
|
||||||
@ -61,6 +64,10 @@ struct demux_sys_t
|
|||||||
int i_frame_samples;
|
int i_frame_samples;
|
||||||
|
|
||||||
date_t pts;
|
date_t pts;
|
||||||
|
|
||||||
|
uint32_t i_channel_mask;
|
||||||
|
vlc_bool_t b_chan_reorder; /* do we need channel reordering */
|
||||||
|
int pi_chan_table[MAX_CHANNELS];
|
||||||
};
|
};
|
||||||
|
|
||||||
#define __EVEN( x ) ( ( (x)%2 != 0 ) ? ((x)+1) : (x) )
|
#define __EVEN( x ) ( ( (x)%2 != 0 ) ? ((x)+1) : (x) )
|
||||||
@ -71,6 +78,23 @@ static void FrameInfo_IMA_ADPCM( demux_t *, unsigned int *, int * );
|
|||||||
static void FrameInfo_MS_ADPCM ( demux_t *, unsigned int *, int * );
|
static void FrameInfo_MS_ADPCM ( demux_t *, unsigned int *, int * );
|
||||||
static void FrameInfo_PCM ( demux_t *, unsigned int *, int * );
|
static void FrameInfo_PCM ( demux_t *, unsigned int *, int * );
|
||||||
|
|
||||||
|
static const uint32_t pi_channels_in[] =
|
||||||
|
{ WAVE_SPEAKER_FRONT_LEFT, WAVE_SPEAKER_FRONT_RIGHT,
|
||||||
|
WAVE_SPEAKER_FRONT_CENTER, WAVE_SPEAKER_LOW_FREQUENCY,
|
||||||
|
WAVE_SPEAKER_BACK_LEFT, WAVE_SPEAKER_BACK_RIGHT };
|
||||||
|
static const uint32_t pi_channels_out[] =
|
||||||
|
{ AOUT_CHAN_LEFT, AOUT_CHAN_RIGHT,
|
||||||
|
AOUT_CHAN_CENTER, AOUT_CHAN_LFE,
|
||||||
|
AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT };
|
||||||
|
static const uint32_t pi_channels_ordered[] =
|
||||||
|
{ AOUT_CHAN_LEFT, AOUT_CHAN_RIGHT,
|
||||||
|
AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT,
|
||||||
|
AOUT_CHAN_CENTER, AOUT_CHAN_LFE };
|
||||||
|
|
||||||
|
static void CheckReordering( demux_t *, int );
|
||||||
|
static void InterleaveS16( int16_t *, int, int *, int );
|
||||||
|
static void InterleaveFloat32( float *, int, int *, int );
|
||||||
|
|
||||||
/*****************************************************************************
|
/*****************************************************************************
|
||||||
* Open: check file and initializes structures
|
* Open: check file and initializes structures
|
||||||
*****************************************************************************/
|
*****************************************************************************/
|
||||||
@ -101,6 +125,8 @@ static int Open( vlc_object_t * p_this )
|
|||||||
p_demux->pf_control = Control;
|
p_demux->pf_control = Control;
|
||||||
p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
|
p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
|
||||||
p_sys->p_es = NULL;
|
p_sys->p_es = NULL;
|
||||||
|
p_sys->b_chan_reorder = 0;
|
||||||
|
p_sys->i_channel_mask = 0;
|
||||||
|
|
||||||
/* skip riff header */
|
/* skip riff header */
|
||||||
stream_Read( p_demux->s, NULL, 12 ); /* cannot fail as peek succeed */
|
stream_Read( p_demux->s, NULL, 12 ); /* cannot fail as peek succeed */
|
||||||
@ -144,10 +170,27 @@ static int Open( vlc_object_t * p_this )
|
|||||||
if( GetWLE( &p_wf->wFormatTag ) == WAVE_FORMAT_EXTENSIBLE &&
|
if( GetWLE( &p_wf->wFormatTag ) == WAVE_FORMAT_EXTENSIBLE &&
|
||||||
i_size >= sizeof( WAVEFORMATEXTENSIBLE ) )
|
i_size >= sizeof( WAVEFORMATEXTENSIBLE ) )
|
||||||
{
|
{
|
||||||
|
int i, i_channel_mask;
|
||||||
|
|
||||||
wf_tag_to_fourcc( GetWLE( &p_wf_ext->SubFormat ),
|
wf_tag_to_fourcc( GetWLE( &p_wf_ext->SubFormat ),
|
||||||
&p_sys->fmt.i_codec, &psz_name );
|
&p_sys->fmt.i_codec, &psz_name );
|
||||||
i_extended = sizeof( WAVEFORMATEXTENSIBLE ) - sizeof( WAVEFORMATEX );
|
i_extended = sizeof( WAVEFORMATEXTENSIBLE ) - sizeof( WAVEFORMATEX );
|
||||||
p_sys->fmt.i_extra -= i_extended;
|
p_sys->fmt.i_extra -= i_extended;
|
||||||
|
|
||||||
|
i_channel_mask = GetDWLE( &p_wf_ext->dwChannelMask );
|
||||||
|
if( i_channel_mask )
|
||||||
|
{
|
||||||
|
for( i = 0; i < sizeof(pi_channels_in)/sizeof(uint32_t); i++ )
|
||||||
|
{
|
||||||
|
if( i_channel_mask & pi_channels_in[i] )
|
||||||
|
p_sys->i_channel_mask |= pi_channels_out[i];
|
||||||
|
}
|
||||||
|
msg_Dbg( p_demux, "channel mask: %x", p_sys->i_channel_mask );
|
||||||
|
CheckReordering( p_demux, p_sys->fmt.audio.i_channels );
|
||||||
|
}
|
||||||
|
p_sys->fmt.audio.i_physical_channels =
|
||||||
|
p_sys->fmt.audio.i_original_channels =
|
||||||
|
p_sys->i_channel_mask;
|
||||||
}
|
}
|
||||||
|
|
||||||
if( p_sys->fmt.i_extra > 0 )
|
if( p_sys->fmt.i_extra > 0 )
|
||||||
@ -261,6 +304,21 @@ static int Demux( demux_t *p_demux )
|
|||||||
/* set PCR */
|
/* set PCR */
|
||||||
es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_block->i_pts );
|
es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_block->i_pts );
|
||||||
|
|
||||||
|
/* Do the channel reordering */
|
||||||
|
if( p_sys->b_chan_reorder )
|
||||||
|
{
|
||||||
|
if( p_sys->fmt.i_codec == VLC_FOURCC('a','r','a','w') &&
|
||||||
|
p_sys->fmt.audio.i_bitspersample == 16 )
|
||||||
|
InterleaveS16( (int16_t *)p_block->p_buffer,
|
||||||
|
p_block->i_buffer, p_sys->pi_chan_table,
|
||||||
|
p_sys->fmt.audio.i_channels );
|
||||||
|
else if( p_sys->fmt.i_codec == VLC_FOURCC('a','f','l','t') &&
|
||||||
|
p_sys->fmt.audio.i_bitspersample == 32 )
|
||||||
|
InterleaveFloat32( (float *)p_block->p_buffer,
|
||||||
|
p_block->i_buffer, p_sys->pi_chan_table,
|
||||||
|
p_sys->fmt.audio.i_channels );
|
||||||
|
}
|
||||||
|
|
||||||
es_out_Send( p_demux->out, p_sys->p_es, p_block );
|
es_out_Send( p_demux->out, p_sys->p_es, p_block );
|
||||||
|
|
||||||
return 1;
|
return 1;
|
||||||
@ -378,3 +436,81 @@ static void FrameInfo_IMA_ADPCM( demux_t *p_demux, unsigned int *pi_size,
|
|||||||
|
|
||||||
*pi_size = p_sys->fmt.audio.i_blockalign;
|
*pi_size = p_sys->fmt.audio.i_blockalign;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/*****************************************************************************
|
||||||
|
* CheckReordering: Check if we need to do some channel re-ordering
|
||||||
|
* (our channel order is different from the one chosen by Microsoft).
|
||||||
|
*****************************************************************************/
|
||||||
|
static void CheckReordering( demux_t *p_demux, int i_nb_channels )
|
||||||
|
{
|
||||||
|
demux_sys_t *p_sys = p_demux->p_sys;
|
||||||
|
int i, j, k, l;
|
||||||
|
|
||||||
|
p_sys->b_chan_reorder = VLC_FALSE;
|
||||||
|
|
||||||
|
for( i = 0, j = 0;
|
||||||
|
i < (int)(sizeof(pi_channels_out)/sizeof(uint32_t)); i++ )
|
||||||
|
{
|
||||||
|
if( p_sys->i_channel_mask & pi_channels_out[i] )
|
||||||
|
{
|
||||||
|
for( k = 0, l = 0;
|
||||||
|
pi_channels_out[i] != pi_channels_ordered[k]; k++ )
|
||||||
|
{
|
||||||
|
if( p_sys->i_channel_mask & pi_channels_ordered[k] )
|
||||||
|
{
|
||||||
|
l++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
p_sys->pi_chan_table[j] = l;
|
||||||
|
|
||||||
|
j++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for( i = 0; i < i_nb_channels; i++ )
|
||||||
|
{
|
||||||
|
if( p_sys->pi_chan_table[i] != i ) p_sys->b_chan_reorder = VLC_TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
if( p_sys->b_chan_reorder ) msg_Dbg( p_demux, "channel reordering needed");
|
||||||
|
}
|
||||||
|
|
||||||
|
/*****************************************************************************
|
||||||
|
* InterleaveFloat32/S16: change the channel order to the Microsoft one.
|
||||||
|
*****************************************************************************/
|
||||||
|
static void InterleaveFloat32( float *p_buf, int i_buf, int *pi_chan_table,
|
||||||
|
int i_nb_channels )
|
||||||
|
{
|
||||||
|
int i, j;
|
||||||
|
float p_tmp[MAX_CHANNELS];
|
||||||
|
|
||||||
|
for( i = 0; i < i_buf / i_nb_channels / sizeof(float); i++ )
|
||||||
|
{
|
||||||
|
for( j = 0; j < i_nb_channels; j++ )
|
||||||
|
{
|
||||||
|
p_tmp[pi_chan_table[j]] = p_buf[i*i_nb_channels + j];
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy( &p_buf[i*i_nb_channels], p_tmp,
|
||||||
|
i_nb_channels * sizeof(float) );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void InterleaveS16( int16_t *p_buf, int i_buf, int *pi_chan_table,
|
||||||
|
int i_nb_channels )
|
||||||
|
{
|
||||||
|
int i, j;
|
||||||
|
int16_t p_tmp[MAX_CHANNELS];
|
||||||
|
|
||||||
|
for( i = 0; i < i_buf / i_nb_channels / sizeof(int16_t); i++ )
|
||||||
|
{
|
||||||
|
for( j = 0; j < i_nb_channels; j++ )
|
||||||
|
{
|
||||||
|
p_tmp[pi_chan_table[j]] = p_buf[i*i_nb_channels + j];
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy( &p_buf[i*i_nb_channels], p_tmp,
|
||||||
|
i_nb_channels * sizeof(int16_t) );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user