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
30 /* max frame size of the 16 bits version is 16384 */
31 /* max frame size of the 14 bits version is 18726 */
36 int out_discrete_channels;
40 static int decdcaInit( hb_work_object_t *, hb_job_t * );
41 static int decdcaWork( hb_work_object_t *, hb_buffer_t **, hb_buffer_t ** );
42 static void decdcaClose( hb_work_object_t * );
43 static int decdcaBSInfo( hb_work_object_t *, const hb_buffer_t *,
46 hb_work_object_t hb_decdca =
57 /***********************************************************************
59 **********************************************************************/
60 static hb_buffer_t * Decode( hb_work_object_t * w );
62 /***********************************************************************
64 ***********************************************************************
65 * Allocate the work object, initialize libdca
66 **********************************************************************/
67 static int decdcaInit( hb_work_object_t * w, hb_job_t * job )
69 hb_work_private_t * pv = calloc( 1, sizeof( hb_work_private_t ) );
70 hb_audio_t * audio = w->audio;
75 pv->list = hb_list_init();
76 pv->state = dca_init( 0 );
78 /* Decide what format we want out of libdca
79 work.c has already done some of this deduction for us in do_job() */
81 pv->flags_out = HB_AMIXDOWN_GET_DCA_FORMAT(audio->config.out.mixdown);
82 if ( audio->config.out.codec == HB_ACODEC_LAME )
83 pv->flags_out |= DCA_ADJUST_LEVEL;
85 /* pass the number of channels used into the private work data */
86 /* will only be actually used if we're not doing AC3 passthru */
87 pv->out_discrete_channels = HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT(audio->config.out.mixdown);
94 /***********************************************************************
96 ***********************************************************************
98 **********************************************************************/
99 static void decdcaClose( hb_work_object_t * w )
101 hb_work_private_t * pv = w->private_data;
102 dca_free( pv->state );
103 hb_list_empty( &pv->list );
105 w->private_data = NULL;
108 /***********************************************************************
110 ***********************************************************************
111 * Add the given buffer to the data we already have, and decode as much
113 **********************************************************************/
114 static int decdcaWork( hb_work_object_t * w, hb_buffer_t ** buf_in,
115 hb_buffer_t ** buf_out )
117 hb_work_private_t * pv = w->private_data;
120 if ( (*buf_in)->size <= 0 )
122 /* EOF on input stream - send it downstream & say that we're done */
128 if ( (*buf_in)->start < -1 && pv->next_pts == 0 )
130 // discard buffers that start before video time 0
135 hb_list_add( pv->list, *buf_in );
138 /* If we got more than a frame, chain raw buffers */
139 *buf_out = buf = Decode( w );
142 buf->next = Decode( w );
149 /***********************************************************************
151 ***********************************************************************
153 **********************************************************************/
154 static hb_buffer_t * Decode( hb_work_object_t * w )
156 hb_work_private_t * pv = w->private_data;
158 hb_audio_t * audio = w->audio;
164 /* Get a frame header if don't have one yet */
167 while( hb_list_bytes( pv->list ) >= 14 )
169 /* We have 14 bytes, check if this is a correct header */
170 hb_list_seebytes( pv->list, pv->frame, 14 );
171 pv->size = dca_syncinfo( pv->state, pv->frame, &pv->flags_in, &pv->rate,
172 &pv->bitrate, &pv->frame_length );
178 hb_log( "dca_syncinfo ok" );
188 hb_log( "dca_syncinfo failed" );
192 /* Try one byte later */
193 hb_list_getbytes( pv->list, pv->frame, 1, NULL, NULL );
197 if( !pv->sync || hb_list_bytes( pv->list ) < pv->size )
203 /* Get the whole frame */
204 hb_list_getbytes( pv->list, pv->frame, pv->size, &upts, &upos );
208 if ( pts != pv->last_buf_pts )
210 pv->last_buf_pts = pts;
214 // spec says that the PTS is the start time of the first frame
215 // that starts in the PES frame so we only use the PTS once then
216 // get the following frames' PTS from the frame length.
220 // mkv files typically use a 1ms timebase which results in a lot of
221 // truncation error in their timestamps. Also, TSMuxer or something
222 // in the m2ts-to-mkv toolchain seems to take a very casual attitude
223 // about time - timestamps seem to randomly offset by ~40ms for a few
224 // seconds then recover. So, if the pts we got is within 50ms of the
225 // pts computed from the data stream, use the data stream pts.
226 if ( pts == -1 || ( pv->next_pts && fabs( pts - pv->next_pts ) < 50.*90. ) )
231 double frame_dur = (double)(pv->frame_length & ~0xFF) / (double)pv->rate * 90000.;
233 /* DCA passthrough: don't decode the DCA frame */
234 if( audio->config.out.codec == HB_ACODEC_DCA_PASS )
236 buf = hb_buffer_init( pv->size );
237 memcpy( buf->data, pv->frame, pv->size );
239 pv->next_pts = pts + frame_dur;
240 buf->stop = pv->next_pts;
246 dca_frame( pv->state, pv->frame, &pv->flags_out, &pv->level, 0 );
248 /* find out how many blocks are in this frame */
249 num_blocks = dca_blocks_num( pv->state );
251 /* num_blocks blocks per frame, 256 samples per block, channelsused channels */
252 int nsamp = num_blocks * 256;
253 buf = hb_buffer_init( nsamp * pv->out_discrete_channels * sizeof( float ) );
256 pv->next_pts = pts + (double)nsamp / (double)pv->rate * 90000.;
257 buf->stop = pv->next_pts;
259 for( i = 0; i < num_blocks; i++ )
261 dca_sample_t * samples_in;
264 dca_block( pv->state );
265 samples_in = dca_samples( pv->state );
266 samples_out = ((float *) buf->data) + 256 * pv->out_discrete_channels * i;
269 for( j = 0; j < 256; j++ )
271 for ( k = 0; k < pv->out_discrete_channels; k++ )
273 samples_out[(pv->out_discrete_channels*j)+k] = samples_in[(256*k)+j] * 16384;
284 static int decdcaBSInfo( hb_work_object_t *w, const hb_buffer_t *b,
285 hb_work_info_t *info )
287 int i, flags, rate, bitrate, frame_length;
288 dca_state_t * state = dca_init( 0 );
290 memset( info, 0, sizeof(*info) );
292 /* since DCA frames don't line up with MPEG ES frames scan the
293 * entire frame for an DCA sync pattern. */
294 for ( i = 0; i < b->size - 7; ++i )
296 if( dca_syncinfo( state, &b->data[i], &flags, &rate, &bitrate,
302 if ( i >= b->size - 7 )
304 /* didn't find DCA sync */
311 info->bitrate = bitrate;
314 if ( ( flags & DCA_CHANNEL_MASK) == DCA_DOLBY )
316 info->flags |= AUDIO_F_DOLBY;
319 switch( flags & DCA_CHANNEL_MASK )
323 info->channel_layout = HB_INPUT_CH_LAYOUT_MONO;
328 case DCA_STEREO_SUMDIFF:
329 case DCA_STEREO_TOTAL:
330 info->channel_layout = HB_INPUT_CH_LAYOUT_STEREO;
334 info->channel_layout = HB_INPUT_CH_LAYOUT_3F2R;
338 info->channel_layout = HB_INPUT_CH_LAYOUT_3F1R;
342 info->channel_layout = HB_INPUT_CH_LAYOUT_3F;
345 info->channel_layout = HB_INPUT_CH_LAYOUT_2F1R;
348 info->channel_layout = HB_INPUT_CH_LAYOUT_2F2R;
351 info->channel_layout = HB_INPUT_CH_LAYOUT_4F2R;
355 info->channel_layout = HB_INPUT_CH_LAYOUT_STEREO;
360 info->channel_layout |= HB_INPUT_CH_LAYOUT_HAS_LFE;