Commit 77f8e25e authored by Laurent Aimar's avatar Laurent Aimar

* all : a demux for aac stream ( raw aac with adts headers ). For now

it has a capability equal to 0 because it conflicts with PS, so you need
to force it.
parent ac354f20
.dep
*.lo
*.o.*
*.lo.*
aac_SOURCES = demux.c
/*****************************************************************************
* demux.c : Raw aac Stream input module for vlc
*****************************************************************************
* Copyright (C) 2001 VideoLAN
* $Id: demux.c,v 1.1 2002/08/12 16:59:15 fenrir Exp $
*
* Authors: Laurent Aimar <fenrir@via.ecp.fr>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
*****************************************************************************/
/*****************************************************************************
* Preamble
*****************************************************************************/
#include <stdlib.h> /* malloc(), free() */
#include <string.h>
#include <vlc/vlc.h>
#include <vlc/input.h>
#include <sys/types.h>
/*****************************************************************************
* Local prototypes
*****************************************************************************/
static int Activate ( vlc_object_t * );
static int Demux ( input_thread_t * );
/*****************************************************************************
* Module descriptor
*****************************************************************************/
vlc_module_begin();
set_description( _("AAC stream demux" ) );
set_capability( "demux", 0 );
set_callbacks( Activate, NULL );
add_shortcut( "aac" );
vlc_module_end();
/*****************************************************************************
* Definitions of structures and functions used by this plugins
*****************************************************************************/
/* XXX set this to 0 to avoid problem with PS XXX */
/* but with some file or web radio will failed to detect */
/* it's you to choose */
#define AAC_MAXTESTPOS 0
static int i_aac_samplerate[16] =
{
96000, 88200, 64000, 48000, 44100, 32000,
24000, 22050, 16000, 12000, 11025, 8000,
7350, 0, 0, 0
};
typedef struct adts_header_s
{
int i_id;
int i_framelength;
int i_layer;
int i_protection_absent;
int i_profile;
int i_samplerate_index;
int i_private_bit;
int i_channel_configuration;
int i_original;
int i_home;
int i_emphasis;
int i_copyright_identification_bit;
int i_copyright_identification_start;
int i_aac_frame_length;
int i_adts_buffer_fullness;
int i_no_raw_data_blocks_in_frame;
int i_crc_check;
} adts_header_t;
/* Not yet used */
typedef struct adif_header_s
{
int b_copyright_id_present;
u8 i_copyright_id[10];
int b_original_copy;
int i_home;
int i_bitstream_type;
int i_bitrate;
int i_num_program_config_elements;
int adif_buffer_fullness;
// program_config_element
} adif_header_t;
struct demux_sys_t
{
mtime_t i_pts;
es_descriptor_t *p_es;
int b_adif_header;
adif_header_t adif_header;
int b_adts_header;
adts_header_t adts_header;
/* extracted information */
int i_samplerate_index;
int i_object_type;
int i_samplerate;
int i_channels;
int i_framelength;
int i_aac_frame_length;
};
/****************************************************************************
* bit_* : function to get a bitstream from a memory buffer
****************************************************************************
*
****************************************************************************/
typedef struct bit_s
{
u8 *p_buffer;
int i_buffer;
int i_mask;
} bit_t;
static void bit_init( bit_t *p_bit,
u8 *p_buffer,
int i_buffer )
{
p_bit->p_buffer = p_buffer;
p_bit->i_buffer = i_buffer;
p_bit->i_mask = 0x80;
}
static u32 bit_get( bit_t *p_bit )
{
u32 i_bit;
if( p_bit->i_buffer <= 0 )
{
return( 0 );
}
i_bit = ( p_bit->p_buffer[0]&p_bit->i_mask ) ? 1 : 0;
p_bit->i_mask >>= 1;
if( !p_bit->i_mask )
{
p_bit->p_buffer++;
p_bit->i_buffer--;
p_bit->i_mask = 0x80;
}
return( i_bit );
}
static u32 bit_gets( bit_t *p_bit, int i_count )
{
u32 i_bits;
i_bits = 0;
for( ; i_count > 0; i_count-- )
{
i_bits = ( i_bits << 1 )|bit_get( p_bit );
}
return( i_bits );
}
/*****************************************************************************
* Function to manipulate stream easily
*****************************************************************************
*
* SkipBytes : skip bytes :) not yet uoptimised ( read bytes to be skipped )
*
* ReadPes : read data and make a PES
*
*****************************************************************************/
static int SkipBytes( input_thread_t *p_input, int i_size )
{
data_packet_t *p_data;
int i_read;
while( i_size > 0 )
{
i_read = input_SplitBuffer(p_input, &p_data, __MIN( i_size, 1024 ) );
if( i_read <= 0 )
{
return( 0 );
}
input_DeletePacket( p_input->p_method_data, p_data );
i_size -= i_read;
}
return( 1 );
}
static int ReadPES( input_thread_t *p_input,
pes_packet_t **pp_pes,
int i_size )
{
pes_packet_t *p_pes;
*pp_pes = NULL;
if( !(p_pes = input_NewPES( p_input->p_method_data )) )
{
msg_Err( p_input, "cannot allocate new PES" );
return( 0 );
}
while( i_size > 0 )
{
data_packet_t *p_data;
int i_read;
if( (i_read = input_SplitBuffer( p_input,
&p_data,
__MIN( i_size, 1024 ) ) ) <= 0 )
{
input_DeletePES( p_input->p_method_data, p_pes );
return( 0 );
}
if( !p_pes->p_first )
{
p_pes->p_first = p_data;
p_pes->i_nb_data = 1;
p_pes->i_pes_size = i_read;
}
else
{
p_pes->p_last->p_next = p_data;
p_pes->i_nb_data++;
p_pes->i_pes_size += i_read;
}
p_pes->p_last = p_data;
i_size -= i_read;
}
*pp_pes = p_pes;
return( 1 );
}
/*****************************************************************************
* GetADIF: find and load adif header if present
*****************************************************************************/
static int GetADIF( input_thread_t *p_input,
adif_header_t *p_adif )
{
u8 *p_peek;
int i_size;
if( ( i_size = input_Peek( p_input, &p_peek, 60 ) ) < 60 )
{
/* it's false, min size is 11 byte but easier ;) */
return( 0 );
}
if( ( p_peek[0] != 'A' )||( p_peek[1] != 'D' )||
( p_peek[2] != 'I' )||( p_peek[3] != 'F' ) )
{
return( 0 );
}
/* we now that we have an adif header */
// return( 1 );
return( 0 ); /* need some work */
}
/*****************************************************************************
* GetADTS: find and load adts header
*****************************************************************************/
#define ADTS_HEADERSIZE 10 /* 8+2 for crc */
static int GetADTS( input_thread_t *p_input,
adts_header_t *p_adts,
int i_max_pos,
int *pi_skip )
{
u8 *p_peek;
int i_size;
bit_t bit;
if( ( i_size = input_Peek( p_input, &p_peek, i_max_pos + ADTS_HEADERSIZE ) )
< ADTS_HEADERSIZE )
{
return( 0 );
}
*pi_skip = 0;
for( ; ; )
{
if( i_size < ADTS_HEADERSIZE )
{
return( 0 );
}
if( ( p_peek[0] != 0xff )||
( ( p_peek[1]&0xf6 ) != 0xf0 ) ) /* Layer == 0 */
{
p_peek++;
i_size--;
*pi_skip += 1;
continue;
}
/* we have find an adts header, load it */
break;
}
bit_init( &bit, p_peek, i_size );
bit_gets( &bit, 12 ); /* synchro bits */
p_adts->i_id = bit_get( &bit );
p_adts->i_layer = bit_gets( &bit, 2);
p_adts->i_protection_absent = bit_get( &bit );
p_adts->i_profile = bit_gets( &bit, 2 );
p_adts->i_samplerate_index = bit_gets( &bit, 4);
p_adts->i_private_bit = bit_get( &bit );
p_adts->i_channel_configuration = bit_gets( &bit, 3);
p_adts->i_original = bit_get( &bit );
p_adts->i_home = bit_get( &bit );
if( p_adts->i_id == 0 )
{
p_adts->i_emphasis = bit_gets( &bit, 2);
}
p_adts->i_copyright_identification_bit = bit_get( &bit );
p_adts->i_copyright_identification_start = bit_get( &bit );
p_adts->i_aac_frame_length = bit_gets( &bit, 13 );
p_adts->i_adts_buffer_fullness = bit_gets( &bit, 11 );
p_adts->i_no_raw_data_blocks_in_frame = bit_gets( &bit, 2 );
if( p_adts->i_protection_absent == 0 )
{
p_adts->i_crc_check = bit_gets( &bit, 16 );
}
return( 1 );
}
static void ExtractConfiguration( demux_sys_t *p_aac )
{
if( p_aac->b_adif_header )
{
}
if( p_aac->b_adts_header )
{
p_aac->i_samplerate_index = p_aac->adts_header.i_samplerate_index;
p_aac->i_object_type = p_aac->adts_header.i_profile;
p_aac->i_samplerate = i_aac_samplerate[p_aac->i_samplerate_index];
p_aac->i_channels = p_aac->adts_header.i_channel_configuration;
if( p_aac->i_channels > 6 )
{
/* I'm not sure of that, got from faad */
p_aac->i_channels = 2;
}
p_aac->i_aac_frame_length = p_aac->adts_header.i_aac_frame_length;
p_aac->i_framelength = 1024;
}
/* FIXME FIXME for LD, but LD = ?
if( p_aac->i_object_type == LD )
{
p_aac->i_framelength /= 2;
}
*/
}
/****************************************************************************
* SkipID3Tag : check if an ID3 header is present and skip it
****************************************************************************
*
* Author : Sigmund Augdal
*
****************************************************************************/
static int SkipID3Tag( input_thread_t *p_input )
{
int i_count;
byte_t *p_peek;
byte_t version, revision;
int b_footer;
int i_size;
/* get 10 byte id3 header */
if( ( i_count = input_Peek( p_input, &p_peek, 10 ) ) < 10 )
{
msg_Err( p_input, "cannot peek()" );
return( -1 );
}
if ( !( (p_peek[0] == 0x49) && (p_peek[1] == 0x44) && (p_peek[2] == 0x33)))
{
return( 0 );
}
version = p_peek[3]; /* These may become usfull later, */
revision = p_peek[4]; /* but we ignore them for now */
b_footer = p_peek[5] & 0x10;
i_size = (p_peek[6] << 21) +
(p_peek[7] << 14) +
(p_peek[8] << 7) +
p_peek[9]; /* Is this safe? */
if ( b_footer )
{
i_size += 10;
}
i_size += 10;
msg_Dbg( p_input, "ID3 tag found, skiping %d bytes", i_size );
if ( input_Peek( p_input, &p_peek, i_size ) < i_size )
{
msg_Err( p_input, "cannot peek()" );
return( -1 );
}
p_input->p_current_data += i_size; /* seek passed end of ID3 tag */
return (0);
}
/****************************************************************************
* CheckPS : check if this stream could be some ps,
* yes it's ugly ... but another idea ?
*
* XXX it seems that aac stream always match ...
*
****************************************************************************/
static int CheckPS( input_thread_t *p_input )
{
u8 *p_peek;
int i_size = input_Peek( p_input, &p_peek, 8196 );
while( i_size > 4 )
{
if( ( p_peek[0] == 0 ) && ( p_peek[1] == 0 )&&
( p_peek[2] == 1 ) && ( p_peek[3] >= 0xb9 ) )
{
return( 1 ); /* Perhaps some ps stream */
}
p_peek++;
i_size--;
}
return( 0 );
}
/*****************************************************************************
* Activate: initializes AAC demux structures
*****************************************************************************/
static int Activate( vlc_object_t * p_this )
{
input_thread_t * p_input = (input_thread_t *)p_this;
demux_sys_t * p_aac;
input_info_category_t * p_category;
int i_skip;
int b_forced;
/* Set the demux function */
p_input->pf_demux = Demux;
/* Initialize access plug-in structures. */
if( p_input->i_mtu == 0 )
{
/* Improve speed. */
p_input->i_bufsize = INPUT_DEFAULT_BUFSIZE;
}
b_forced = ( ( *p_input->psz_demux )&&
( !strncmp( p_input->psz_demux, "aac", 10 ) ) ) ? 1 : 0;
/* check if it can be a ps stream */
if( !b_forced && CheckPS( p_input ) )
{
return( -1 );
}
/* skip possible id3 header */
if ( SkipID3Tag( p_input ) )
{
return -1;
}
/* allocate p_aac */
if( !( p_aac = malloc( sizeof( demux_sys_t ) ) ) )
{
msg_Err( p_input, "out of memory" );
return( -1 );
}
memset( p_aac, 0, sizeof( demux_sys_t ) );
/* Now check for adif/adts header */
i_skip = 0;
if( GetADIF( p_input, &p_aac->adif_header ) )
{
p_aac->b_adif_header = 1;
msg_Err( p_input,
"found ADIF header (unsupported)" );
free( p_aac );
return( -1 );
}
else
if( GetADTS( p_input,
&p_aac->adts_header,
b_forced ? 8000 : 0,
&i_skip ) )
{
p_aac->b_adts_header = 1;
msg_Info( p_input,
"found ADTS header" );
}
else
{
msg_Warn( p_input,
"AAC module discarded (no header found)" );
free( p_aac );
return( -1 );
}
ExtractConfiguration( p_aac );
vlc_mutex_lock( &p_input->stream.stream_lock );
if( input_InitStream( p_input, 0 ) == -1)
{
msg_Err( p_input, "cannot init stream" );
return( -1 );
}
if( input_AddProgram( p_input, 0, 0) == NULL )
{
msg_Err( p_input, "cannot add program" );
return( -1 );
}
p_input->stream.pp_programs[0]->b_is_ok = 0;
p_input->stream.p_selected_program = p_input->stream.pp_programs[0];
/* create our ES */
p_aac->p_es = input_AddES( p_input,
p_input->stream.p_selected_program,
1, /* id */
0 );
if( !p_aac->p_es )
{
vlc_mutex_unlock( &p_input->stream.stream_lock );
msg_Err( p_input, "out of memory" );
return( -1 );
}
p_aac->p_es->i_stream_id = 1;
p_aac->p_es->i_fourcc = VLC_FOURCC( 'm', 'p', '4', 'a' );
p_aac->p_es->i_cat = AUDIO_ES;
input_SelectES( p_input, p_aac->p_es );
p_input->stream.p_selected_program->b_is_ok = 1;
vlc_mutex_unlock( &p_input->stream.stream_lock );
vlc_mutex_lock( &p_input->stream.stream_lock );
if( p_aac->b_adif_header )
{
p_input->stream.i_mux_rate = 0 / 50;
}
if( p_aac->b_adts_header )
{
p_input->stream.i_mux_rate = 0 / 50;
}
vlc_mutex_unlock( &p_input->stream.stream_lock );
#if 0
/* FIXME FIXME FIXME FIXME FIXME FIXME FIXME */
/* if i don't do that, it don't work correctly but why ??? */
if( p_input->stream.b_seekable )
{
p_input->pf_seek( p_input, 0 );
input_AccessReinit( p_input );
}
/* FIXME FIXME FIXME FIXME FIXME FIXME FIXME */
#endif
/* all is ok :)) */
if( p_aac->b_adif_header )
{
}
if( p_aac->b_adts_header )
{
msg_Dbg( p_input,
"audio AAC MPEG-%d layer-%d %d channels %dHz",
p_aac->adts_header.i_id == 1 ? 2 : 4,
4 - p_aac->adts_header.i_layer, /* it's always 4 in fact */
p_aac->i_channels,
p_aac->i_samplerate );
vlc_mutex_lock( &p_input->stream.stream_lock );
p_category = input_InfoCategory( p_input, "aac" );
input_AddInfo( p_category, "input type", "MPEG-%d AAC",
p_aac->adts_header.i_id == 1 ? 2 : 4 );
input_AddInfo( p_category, "layer", "%d",
4 - p_aac->adts_header.i_layer );
input_AddInfo( p_category, "channels", "%d",
p_aac->i_channels );
input_AddInfo( p_category, "sample rate", "%dHz",
p_aac->i_samplerate );
vlc_mutex_unlock( &p_input->stream.stream_lock );
}
p_input->p_demux_data = p_aac;
return( 0 );
}
/*****************************************************************************
* Demux: reads and demuxes data packets
*****************************************************************************
* Returns -1 in case of error, 0 in case of EOF, 1 otherwise
*****************************************************************************/
static int Demux( input_thread_t * p_input )
{
int i_skip;
int i_found;
pes_packet_t *p_pes;
demux_sys_t *p_aac = p_input->p_demux_data;
/* Get a frame */
if( p_aac->b_adif_header )
{
i_skip = 0;
return( -1 ); /* FIXME */
}
else
if( p_aac->b_adts_header )
{
i_found = GetADTS( p_input,
&p_aac->adts_header,
8000,
&i_skip );
}
else
{
return( -1 );
}
ExtractConfiguration( p_aac );
/* skip garbage bytes */
if( i_skip > 0 )
{
msg_Dbg( p_input,
"skipping %d bytes",
i_skip );
SkipBytes( p_input, i_skip );
}
if( !i_found )
{
msg_Info( p_input, "can't find next frame" );
return( 0 );
}
input_ClockManageRef( p_input,
p_input->stream.p_selected_program,
p_aac->i_pts );
if( !ReadPES( p_input, &p_pes, p_aac->i_aac_frame_length ) )
{
msg_Err( p_input,
"cannot read data" );
return( -1 );
}
p_pes->i_dts =
p_pes->i_pts = input_ClockGetTS( p_input,
p_input->stream.p_selected_program,
p_aac->i_pts );
if( !p_aac->p_es->p_decoder_fifo )
{
msg_Err( p_input, "no audio decoder" );
input_DeletePES( p_input->p_method_data, p_pes );
return( -1 ); /* perhaps not, it's my choice */
}
else
{
input_DecodePES( p_aac->p_es->p_decoder_fifo, p_pes );
}
/* Update date information */
p_aac->i_pts += (mtime_t)90000 *
(mtime_t)p_aac->i_framelength /
(mtime_t)p_aac->i_samplerate;
return( 1 );
}
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment