1 /* $Id: decdca.c,v 1.14 2005/03/03 17:21:57 titer Exp $
3 This file is part of the HandBrake source code.
4 Homepage: <http://handbrake.fr/>.
5 It may be used under the terms of the GNU General Public License. */
10 struct hb_work_private_s
28 /* max frame size of the 16 bits version is 16384 */
29 /* max frame size of the 14 bits version is 18726 */
34 int out_discrete_channels;
38 static int decdcaInit( hb_work_object_t *, hb_job_t * );
39 static int decdcaWork( hb_work_object_t *, hb_buffer_t **, hb_buffer_t ** );
40 static void decdcaClose( hb_work_object_t * );
41 static int decdcaBSInfo( hb_work_object_t *, const hb_buffer_t *,
44 hb_work_object_t hb_decdca =
55 /***********************************************************************
57 **********************************************************************/
58 static hb_buffer_t * Decode( hb_work_object_t * w );
60 /***********************************************************************
62 ***********************************************************************
63 * Allocate the work object, initialize libdca
64 **********************************************************************/
65 static int decdcaInit( hb_work_object_t * w, hb_job_t * job )
67 hb_work_private_t * pv = calloc( 1, sizeof( hb_work_private_t ) );
68 hb_audio_t * audio = w->audio;
73 pv->list = hb_list_init();
74 pv->state = dca_init( 0 );
76 /* Decide what format we want out of libdca
77 work.c has already done some of this deduction for us in do_job() */
79 pv->flags_out = HB_AMIXDOWN_GET_DCA_FORMAT(audio->config.out.mixdown);
81 /* pass the number of channels used into the private work data */
82 /* will only be actually used if we're not doing AC3 passthru */
83 pv->out_discrete_channels = HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT(audio->config.out.mixdown);
90 /***********************************************************************
92 ***********************************************************************
94 **********************************************************************/
95 static void decdcaClose( hb_work_object_t * w )
97 hb_work_private_t * pv = w->private_data;
98 dca_free( pv->state );
99 hb_list_empty( &pv->list );
101 w->private_data = NULL;
104 /***********************************************************************
106 ***********************************************************************
107 * Add the given buffer to the data we already have, and decode as much
109 **********************************************************************/
110 static int decdcaWork( hb_work_object_t * w, hb_buffer_t ** buf_in,
111 hb_buffer_t ** buf_out )
113 hb_work_private_t * pv = w->private_data;
116 if ( (*buf_in)->size <= 0 )
118 /* EOF on input stream - send it downstream & say that we're done */
124 hb_list_add( pv->list, *buf_in );
127 /* If we got more than a frame, chain raw buffers */
128 *buf_out = buf = Decode( w );
131 buf->next = Decode( w );
138 /***********************************************************************
140 ***********************************************************************
142 **********************************************************************/
143 static hb_buffer_t * Decode( hb_work_object_t * w )
145 hb_work_private_t * pv = w->private_data;
151 /* Get a frame header if don't have one yet */
154 while( hb_list_bytes( pv->list ) >= 14 )
156 /* We have 14 bytes, check if this is a correct header */
157 hb_list_seebytes( pv->list, pv->frame, 14 );
158 pv->size = dca_syncinfo( pv->state, pv->frame, &pv->flags_in, &pv->rate,
159 &pv->bitrate, &pv->frame_length );
165 hb_log( "dca_syncinfo ok" );
175 hb_log( "dca_syncinfo failed" );
179 /* Try one byte later */
180 hb_list_getbytes( pv->list, pv->frame, 1, NULL, NULL );
185 hb_list_bytes( pv->list ) < pv->size )
191 /* Get the whole frame */
192 hb_list_getbytes( pv->list, pv->frame, pv->size, &pts, &pos );
195 dca_frame( pv->state, pv->frame, &pv->flags_out, &pv->level, 0 );
197 /* find out how many blocks are in this frame */
198 num_blocks = dca_blocks_num( pv->state );
200 /* num_blocks blocks per frame, 256 samples per block, channelsused channels */
201 buf = hb_buffer_init( num_blocks * 256 * pv->out_discrete_channels * sizeof( float ) );
202 buf->start = pts + ( pos / pv->size ) * num_blocks * 256 * 90000 / pv->rate;
203 buf->stop = buf->start + num_blocks * 256 * 90000 / pv->rate;
205 for( i = 0; i < num_blocks; i++ )
207 dca_sample_t * samples_in;
210 dca_block( pv->state );
211 samples_in = dca_samples( pv->state );
212 samples_out = ((float *) buf->data) + 256 * pv->out_discrete_channels * i;
215 for( j = 0; j < 256; j++ )
217 for ( k = 0; k < pv->out_discrete_channels; k++ )
219 samples_out[(pv->out_discrete_channels*j)+k] = samples_in[(256*k)+j] * 16384;
230 static int decdcaBSInfo( hb_work_object_t *w, const hb_buffer_t *b,
231 hb_work_info_t *info )
233 int i, flags, rate, bitrate, frame_length;
234 dca_state_t * state = dca_init( 0 );
236 memset( info, 0, sizeof(*info) );
238 /* since DCA frames don't line up with MPEG ES frames scan the
239 * entire frame for an DCA sync pattern. */
240 for ( i = 0; i < b->size - 7; ++i )
242 if( dca_syncinfo( state, &b->data[i], &flags, &rate, &bitrate,
248 if ( i >= b->size - 7 )
250 /* didn't find DCA sync */
257 info->bitrate = bitrate;
260 if ( ( flags & DCA_CHANNEL_MASK) == DCA_DOLBY )
262 info->flags |= AUDIO_F_DOLBY;
265 switch( flags & DCA_CHANNEL_MASK )
269 info->channel_layout = HB_INPUT_CH_LAYOUT_MONO;
274 case DCA_STEREO_SUMDIFF:
275 case DCA_STEREO_TOTAL:
276 info->channel_layout = HB_INPUT_CH_LAYOUT_STEREO;
280 info->channel_layout = HB_INPUT_CH_LAYOUT_3F2R;
284 info->channel_layout = HB_INPUT_CH_LAYOUT_3F1R;
288 info->channel_layout = HB_INPUT_CH_LAYOUT_3F;
291 info->channel_layout = HB_INPUT_CH_LAYOUT_2F1R;
294 info->channel_layout = HB_INPUT_CH_LAYOUT_2F2R;
297 info->channel_layout = HB_INPUT_CH_LAYOUT_4F2R;
301 info->channel_layout = HB_INPUT_CH_LAYOUT_STEREO;
306 info->channel_layout |= HB_INPUT_CH_LAYOUT_HAS_LFE;