?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
chromecast: refactor, make place for SPUs
[vlc.git] / modules / demux / asf / asf.c
blobd17fb5e2dbb1f7bbfd4499923d293e184f14bb79
1 /*****************************************************************************
2 * asf.c : ASF demux module
3 *****************************************************************************
4 * Copyright © 2002-2004, 2006-2008, 2010 VLC authors and VideoLAN
5 *
6 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
7 *
8 * This program is free software; you can redistribute it and/or modify it
9 * under the terms of the GNU Lesser General Public License as published by
10 * the Free Software Foundation; either version 2.1 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public License
19 * along with this program; if not, write to the Free Software Foundation,
20 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
21 *****************************************************************************/
22
23 /*****************************************************************************
24 * Preamble
25 *****************************************************************************/
26
27 #ifdef HAVE_CONFIG_H
28 # include "config.h"
29 #endif
30
31 #include <vlc_common.h>
32 #include <vlc_plugin.h>
33 #include <vlc_demux.h>
34 #include <vlc_dialog.h>
35
36 #include <vlc_meta.h> /* vlc_meta_Set*, vlc_meta_New */
37 #include <vlc_access.h> /* GET_PRIVATE_ID_STATE */
38 #include <vlc_codecs.h> /* VLC_BITMAPINFOHEADER, WAVEFORMATEX */
39 #include <vlc_input.h>
40 #include <vlc_vout.h>
41
42 #include <limits.h>
43
44 #include "asfpacket.h"
45 #include "libasf.h"
46 #include "assert.h"
47
48 /* TODO
49 * - add support for the newly added object: language, bitrate,
50 * extended stream properties.
51 */
52
53 /*****************************************************************************
54 * Module descriptor
55 *****************************************************************************/
56 static int Open ( vlc_object_t * );
57 static void Close ( vlc_object_t * );
58
59 vlc_module_begin ()
60 set_category( CAT_INPUT )
61 set_subcategory( SUBCAT_INPUT_DEMUX )
62 set_description( N_("ASF/WMV demuxer") )
63 set_capability( "demux", 200 )
64 set_callbacks( Open, Close )
65 add_shortcut( "asf", "wmv" )
66 vlc_module_end ()
67
68
69 /*****************************************************************************
70 * Local prototypes
71 *****************************************************************************/
72 static int Demux ( demux_t * );
73 static int Control( demux_t *, int i_query, va_list args );
74
75 #define MAX_ASF_TRACKS (ASF_MAX_STREAMNUMBER + 1)
76 #define ASF_PREROLL_FROM_CURRENT -1
77
78 /* callbacks for packet parser */
79 static void Packet_UpdateTime( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number,
80 mtime_t i_time );
81 static void Packet_SetSendTime( asf_packet_sys_t *p_packetsys, mtime_t i_time);
82 static bool Block_Dequeue( demux_t *p_demux, mtime_t i_nexttime );
83 static asf_track_info_t * Packet_GetTrackInfo( asf_packet_sys_t *p_packetsys,
84 uint8_t i_stream_number );
85 static bool Packet_DoSkip( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number, bool b_packet_keyframe );
86 static void Packet_Enqueue( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number, block_t **pp_frame );
87 static void Packet_SetAR( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number,
88 uint8_t i_ratio_x, uint8_t i_ratio_y );
89
90 typedef struct
91 {
92 int i_cat;
93
94 es_out_id_t *p_es;
95 es_format_t *p_fmt; /* format backup for video changes */
96 bool b_selected;
97
98 mtime_t i_time; /* track time*/
99
100 asf_track_info_t info;
101
102 struct
103 {
104 block_t *p_first;
105 block_t **pp_last;
106 } queue;
107
108 } asf_track_t;
109
110 struct demux_sys_t
111 {
112 mtime_t i_time; /* s */
113 mtime_t i_sendtime;
114 mtime_t i_length; /* length of file file */
115 uint64_t i_bitrate; /* global file bitrate */
116 bool b_eos; /* end of current stream */
117 bool b_eof; /* end of current media */
118
119 asf_object_root_t *p_root;
120 asf_object_file_properties_t *p_fp;
121
122 unsigned int i_track;
123 asf_track_t *track[MAX_ASF_TRACKS]; /* track number is stored on 7 bits */
124
125 uint64_t i_data_begin;
126 uint64_t i_data_end;
127
128 bool b_index;
129 bool b_canfastseek;
130 uint8_t i_seek_track;
131 uint8_t i_access_selected_track[ES_CATEGORY_COUNT]; /* mms, depends on access algorithm */
132 unsigned int i_wait_keyframe;
133
134 mtime_t i_preroll_start;
135
136 asf_packet_sys_t packet_sys;
137
138 vlc_meta_t *meta;
139 };
140
141 static int DemuxInit( demux_t * );
142 static void DemuxEnd( demux_t * );
143
144 static void FlushQueue( asf_track_t * );
145 static void FlushQueues( demux_t *p_demux );
146
147 /*****************************************************************************
148 * Open: check file and initializes ASF structures
149 *****************************************************************************/
150 static int Open( vlc_object_t * p_this )
151 {
152 demux_t *p_demux = (demux_t *)p_this;
153 demux_sys_t *p_sys;
154 guid_t guid;
155 const uint8_t *p_peek;
156
157 /* A little test to see if it could be a asf stream */
158 if( vlc_stream_Peek( p_demux->s, &p_peek, 16 ) < 16 ) return VLC_EGENERIC;
159
160 ASF_GetGUID( &guid, p_peek );
161 if( !guidcmp( &guid, &asf_object_header_guid ) ) return VLC_EGENERIC;
162
163 /* Set p_demux fields */
164 p_demux->pf_demux = Demux;
165 p_demux->pf_control = Control;
166 p_demux->p_sys = p_sys = calloc( 1, sizeof( demux_sys_t ) );
167
168 /* Load the headers */
169 if( DemuxInit( p_demux ) )
170 {
171 free( p_sys );
172 return VLC_EGENERIC;
173 }
174
175 p_sys->packet_sys.p_demux = p_demux;
176 p_sys->packet_sys.pf_doskip = Packet_DoSkip;
177 p_sys->packet_sys.pf_send = Packet_Enqueue;
178 p_sys->packet_sys.pf_gettrackinfo = Packet_GetTrackInfo;
179 p_sys->packet_sys.pf_updatetime = Packet_UpdateTime;
180 p_sys->packet_sys.pf_updatesendtime = Packet_SetSendTime;
181 p_sys->packet_sys.pf_setaspectratio = Packet_SetAR;
182
183 return VLC_SUCCESS;
184 }
185
186 /*****************************************************************************
187 * Demux: read packet and send them to decoders
188 *****************************************************************************/
189 #define CHUNK (CLOCK_FREQ / 10)
190 static int Demux( demux_t *p_demux )
191 {
192 demux_sys_t *p_sys = p_demux->p_sys;
193
194 for( int i=0; i<ES_CATEGORY_COUNT; i++ )
195 {
196 if ( p_sys->i_access_selected_track[i] > 0 )
197 {
198 es_out_Control( p_demux->out, ES_OUT_SET_ES_STATE,
199 p_sys->track[p_sys->i_access_selected_track[i]]->p_es, true );
200 p_sys->i_access_selected_track[i] = 0;
201 }
202 }
203
204 /* Get selected tracks, especially for computing PCR */
205 for( int i=0; i<MAX_ASF_TRACKS; i++ )
206 {
207 asf_track_t *tk = p_sys->track[i];
208 if ( !tk ) continue;
209 if ( tk->p_es )
210 es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, & tk->b_selected );
211 else
212 tk->b_selected = false;
213 }
214
215 while( !p_sys->b_eos && ( p_sys->i_sendtime - p_sys->i_time - CHUNK < 0 ||
216 ( p_sys->i_sendtime - p_sys->i_time - CHUNK ) /
217 UINT64_C( 1000 ) < p_sys->p_fp->i_preroll ) )
218 {
219 /* Read and demux a packet */
220 if( DemuxASFPacket( &p_sys->packet_sys,
221 p_sys->p_fp->i_min_data_packet_size,
222 p_sys->p_fp->i_max_data_packet_size ) <= 0 )
223 {
224 p_sys->b_eos = true;
225 /* Check if we have concatenated files */
226 const uint8_t *p_peek;
227 if( vlc_stream_Peek( p_demux->s, &p_peek, 16 ) == 16 )
228 {
229 guid_t guid;
230
231 ASF_GetGUID( &guid, p_peek );
232 p_sys->b_eof = !guidcmp( &guid, &asf_object_header_guid );
233 if( !p_sys->b_eof )
234 msg_Warn( p_demux, "found a new ASF header" );
235 }
236 else
237 p_sys->b_eof = true;
238 }
239
240 if ( p_sys->i_time == -1 )
241 p_sys->i_time = p_sys->i_sendtime;
242 }
243
244 if( p_sys->b_eos || ( p_sys->i_sendtime - p_sys->i_time - CHUNK >= 0 &&
245 ( p_sys->i_sendtime - p_sys->i_time - CHUNK ) /
246 UINT64_C( 1000 ) >= p_sys->p_fp->i_preroll ) )
247 {
248 bool b_data = Block_Dequeue( p_demux, p_sys->i_time + CHUNK );
249
250 p_sys->i_time += CHUNK;
251 es_out_SetPCR( p_demux->out, VLC_TS_0 + p_sys->i_time );
252 #ifdef ASF_DEBUG
253 msg_Dbg( p_demux, "Demux Loop Setting PCR to %"PRId64, VLC_TS_0 + p_sys->i_time );
254 #endif
255 if ( !b_data && p_sys->b_eos )
256 {
257 /* We end this stream */
258 if( !p_sys->b_eof )
259 {
260 DemuxEnd( p_demux );
261
262 /* And we prepare to read the next one */
263 if( DemuxInit( p_demux ) )
264 {
265 msg_Err( p_demux, "failed to load the new header" );
266 vlc_dialog_display_error( p_demux,
267 _("Could not demux ASF stream"), "%s",
268 _("VLC failed to load the ASF header.") );
269 return VLC_DEMUXER_EOF;
270 }
271 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
272 }
273 else
274 return VLC_DEMUXER_EOF;
275 }
276 }
277
278 return 1;
279 }
280
281 /*****************************************************************************
282 * Close: frees unused data
283 *****************************************************************************/
284 static void Close( vlc_object_t * p_this )
285 {
286 demux_t *p_demux = (demux_t *)p_this;
287
288 DemuxEnd( p_demux );
289
290 free( p_demux->p_sys );
291 }
292
293 /*****************************************************************************
294 * WaitKeyframe: computes the number of frames to wait for a keyframe
295 *****************************************************************************/
296 static void WaitKeyframe( demux_t *p_demux )
297 {
298 demux_sys_t *p_sys = p_demux->p_sys;
299 if ( ! p_sys->i_seek_track )
300 {
301 for ( int i=0; i<MAX_ASF_TRACKS; i++ )
302 {
303 asf_track_t *tk = p_sys->track[i];
304 if ( tk && tk->info.p_sp && tk->i_cat == VIDEO_ES && tk->b_selected )
305 {
306 p_sys->i_seek_track = tk->info.p_sp->i_stream_number;
307 break;
308 }
309 }
310 }
311
312 if ( p_sys->i_seek_track )
313 {
314 /* Skip forward at least 1 min */
315 asf_track_t *tk = p_sys->track[p_sys->i_seek_track];
316 if ( tk->info.p_esp && tk->info.p_esp->i_average_time_per_frame )
317 {
318 /* 1 min if fastseek, otherwise 5 sec */
319 /* That's a guess for bandwidth */
320 uint64_t i_maxwaittime = ( p_sys->b_canfastseek ) ? 600000000 : 50000000;
321 i_maxwaittime /= tk->info.p_esp->i_average_time_per_frame;
322 p_sys->i_wait_keyframe = __MIN( i_maxwaittime, UINT_MAX );
323 }
324 else
325 {
326 p_sys->i_wait_keyframe = ( p_sys->b_canfastseek ) ? 25 * 30 : 25 * 5;
327 }
328 }
329 else
330 {
331 p_sys->i_wait_keyframe = 0;
332 }
333
334 }
335
336 /*****************************************************************************
337 * SeekIndex: goto to i_date or i_percent
338 *****************************************************************************/
339 static int SeekPercent( demux_t *p_demux, int i_query, va_list args )
340 {
341 demux_sys_t *p_sys = p_demux->p_sys;
342
343 WaitKeyframe( p_demux );
344
345 msg_Dbg( p_demux, "seek with percent: waiting %i frames", p_sys->i_wait_keyframe );
346 return demux_vaControlHelper( p_demux->s, __MIN( INT64_MAX, p_sys->i_data_begin ),
347 __MIN( INT64_MAX, p_sys->i_data_end ),
348 __MIN( INT64_MAX, p_sys->i_bitrate ),
349 __MIN( INT16_MAX, p_sys->p_fp->i_min_data_packet_size ),
350 i_query, args );
351 }
352
353 static int SeekIndex( demux_t *p_demux, mtime_t i_date, float f_pos )
354 {
355 demux_sys_t *p_sys = p_demux->p_sys;
356 asf_object_index_t *p_index;
357
358 msg_Dbg( p_demux, "seek with index: %i seconds, position %f",
359 i_date >= 0 ? (int)(i_date/1000000) : -1, f_pos );
360
361 if( i_date < 0 )
362 i_date = p_sys->i_length * f_pos;
363
364 p_sys->i_preroll_start = i_date - (int64_t) p_sys->p_fp->i_preroll;
365 if ( p_sys->i_preroll_start < 0 ) p_sys->i_preroll_start = 0;
366
367 p_index = ASF_FindObject( p_sys->p_root, &asf_object_simple_index_guid, 0 );
368
369 uint64_t i_entry = p_sys->i_preroll_start * 10 / p_index->i_index_entry_time_interval;
370 if( i_entry >= p_index->i_index_entry_count )
371 {
372 msg_Warn( p_demux, "Incomplete index" );
373 return VLC_EGENERIC;
374 }
375
376 WaitKeyframe( p_demux );
377
378 uint64_t i_offset = (uint64_t)p_index->index_entry[i_entry].i_packet_number *
379 p_sys->p_fp->i_min_data_packet_size;
380
381 if ( vlc_stream_Seek( p_demux->s, i_offset + p_sys->i_data_begin ) == VLC_SUCCESS )
382 {
383 es_out_Control( p_demux->out, ES_OUT_SET_NEXT_DISPLAY_TIME, VLC_TS_0 + i_date );
384 return VLC_SUCCESS;
385 }
386 else return VLC_EGENERIC;
387 }
388
389 static void SeekPrepare( demux_t *p_demux )
390 {
391 demux_sys_t *p_sys = p_demux->p_sys;
392
393 p_sys->b_eof = false;
394 p_sys->b_eos = false;
395 p_sys->i_time = -1;
396 p_sys->i_sendtime = -1;
397 p_sys->i_preroll_start = ASFPACKET_PREROLL_FROM_CURRENT;
398
399 for( int i = 0; i < MAX_ASF_TRACKS ; i++ )
400 {
401 asf_track_t *tk = p_sys->track[i];
402 if( tk )
403 {
404 FlushQueue( tk );
405 tk->i_time = -1;
406 }
407 }
408
409 es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
410 }
411
412 /*****************************************************************************
413 * Control:
414 *****************************************************************************/
415 static int Control( demux_t *p_demux, int i_query, va_list args )
416 {
417 demux_sys_t *p_sys = p_demux->p_sys;
418 vlc_meta_t *p_meta;
419 int64_t i64, *pi64;
420 int i;
421 double f, *pf;
422
423 switch( i_query )
424 {
425 case DEMUX_GET_LENGTH:
426 pi64 = va_arg( args, int64_t * );
427 *pi64 = p_sys->i_length;
428 return VLC_SUCCESS;
429
430 case DEMUX_GET_TIME:
431 pi64 = va_arg( args, int64_t * );
432 if( p_sys->i_time < 0 ) return VLC_EGENERIC;
433 *pi64 = p_sys->i_time;
434 return VLC_SUCCESS;
435
436 case DEMUX_SET_TIME:
437 if ( p_sys->p_fp &&
438 ! ( p_sys->p_fp->i_flags & ASF_FILE_PROPERTIES_SEEKABLE ) )
439 return VLC_EGENERIC;
440
441 SeekPrepare( p_demux );
442
443 if( p_sys->b_index && p_sys->i_length > 0 )
444 {
445 va_list acpy;
446 va_copy( acpy, args );
447 i64 = va_arg( acpy, int64_t );
448 va_end( acpy );
449
450 if( !SeekIndex( p_demux, i64, -1 ) )
451 return VLC_SUCCESS;
452 }
453 return SeekPercent( p_demux, i_query, args );
454
455 case DEMUX_SET_ES:
456 {
457 i = va_arg( args, int );
458 int i_ret;
459 if ( i >= 0 )
460 {
461 msg_Dbg( p_demux, "Requesting access to enable stream %d", i );
462 i_ret = vlc_stream_Control( p_demux->s,
463 STREAM_SET_PRIVATE_ID_STATE, i, true );
464 }
465 else
466 { /* i contains -1 * es_category */
467 msg_Dbg( p_demux, "Requesting access to disable stream %d", i );
468 i_ret = vlc_stream_Control( p_demux->s,
469 STREAM_SET_PRIVATE_ID_STATE, i,
470 false );
471 }
472
473 if ( i_ret == VLC_SUCCESS )
474 {
475 asf_track_t *tk;
476 if( i >= 0 )
477 {
478 tk = p_sys->track[i];
479 }
480 else
481 {
482 for( int j = 0; j < MAX_ASF_TRACKS ; j++ )
483 {
484 tk = p_sys->track[j];
485 if( !tk || !tk->p_fmt || tk->i_cat != -1 * i )
486 continue;
487 FlushQueue( tk );
488 tk->i_time = -1;
489 }
490 }
491
492 p_sys->i_seek_track = 0;
493 if ( ( tk && tk->i_cat == VIDEO_ES ) || i == -1 * VIDEO_ES )
494 WaitKeyframe( p_demux );
495 }
496 return i_ret;
497 }
498
499 case DEMUX_GET_POSITION:
500 if( p_sys->i_time < 0 ) return VLC_EGENERIC;
501 if( p_sys->i_length > 0 )
502 {
503 pf = va_arg( args, double * );
504 *pf = p_sys->i_time / (double)p_sys->i_length;
505 return VLC_SUCCESS;
506 }
507 return demux_vaControlHelper( p_demux->s,
508 __MIN( INT64_MAX, p_sys->i_data_begin ),
509 __MIN( INT64_MAX, p_sys->i_data_end ),
510 __MIN( INT64_MAX, p_sys->i_bitrate ),
511 __MIN( INT16_MAX, p_sys->p_fp->i_min_data_packet_size ),
512 i_query, args );
513
514 case DEMUX_SET_POSITION:
515 if ( p_sys->p_fp &&
516 ! ( p_sys->p_fp->i_flags & ASF_FILE_PROPERTIES_SEEKABLE ) )
517 return VLC_EGENERIC;
518
519 SeekPrepare( p_demux );
520
521 if( p_sys->b_index && p_sys->i_length > 0 )
522 {
523 va_list acpy;
524 va_copy( acpy, args );
525 f = va_arg( acpy, double );
526 va_end( acpy );
527
528 if( !SeekIndex( p_demux, -1, f ) )
529 return VLC_SUCCESS;
530 }
531 return SeekPercent( p_demux, i_query, args );
532
533 case DEMUX_GET_META:
534 p_meta = va_arg( args, vlc_meta_t * );
535 vlc_meta_Merge( p_meta, p_sys->meta );
536 return VLC_SUCCESS;
537
538 case DEMUX_CAN_SEEK:
539 if ( p_sys->p_fp &&
540 ! ( p_sys->p_fp->i_flags & ASF_FILE_PROPERTIES_SEEKABLE ) )
541 {
542 bool *pb_bool = va_arg( args, bool * );
543 *pb_bool = false;
544 return VLC_SUCCESS;
545 }
546 /* fall through */
547 default:
548 return demux_vaControlHelper( p_demux->s,
549 __MIN( INT64_MAX, p_sys->i_data_begin ),
550 __MIN( INT64_MAX, p_sys->i_data_end),
551 __MIN( INT64_MAX, p_sys->i_bitrate ),
552 ( p_sys->p_fp ) ? __MIN( INT_MAX, p_sys->p_fp->i_min_data_packet_size ) : 1,
553 i_query, args );
554 }
555 }
556
557 /*****************************************************************************
558 *
559 *****************************************************************************/
560 static void Packet_SetAR( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number,
561 uint8_t i_ratio_x, uint8_t i_ratio_y )
562 {
563 demux_t *p_demux = p_packetsys->p_demux;
564 demux_sys_t *p_sys = p_demux->p_sys;
565 asf_track_t *tk = p_sys->track[i_stream_number];
566
567 if ( !tk->p_fmt || (tk->p_fmt->video.i_sar_num == i_ratio_x && tk->p_fmt->video.i_sar_den == i_ratio_y ) )
568 return;
569
570 /* Only apply if origin pixel size >= 1x1, due to broken yacast */
571 if ( tk->p_fmt->video.i_height * i_ratio_x > tk->p_fmt->video.i_width * i_ratio_y )
572 {
573 vout_thread_t *p_vout = input_GetVout( p_demux->p_input );
574 if ( p_vout )
575 {
576 msg_Info( p_demux, "Changing aspect ratio to %i/%i", i_ratio_x, i_ratio_y );
577 vout_ChangeAspectRatio( p_vout, i_ratio_x, i_ratio_y );
578 vlc_object_release( p_vout );
579 }
580 }
581 tk->p_fmt->video.i_sar_num = i_ratio_x;
582 tk->p_fmt->video.i_sar_den = i_ratio_y;
583 }
584
585 static void Packet_SetSendTime( asf_packet_sys_t *p_packetsys, mtime_t i_time )
586 {
587 demux_t *p_demux = p_packetsys->p_demux;
588 demux_sys_t *p_sys = p_demux->p_sys;
589
590 p_sys->i_sendtime = i_time;
591 }
592
593 static void Packet_UpdateTime( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number,
594 mtime_t i_time )
595 {
596 demux_t *p_demux = p_packetsys->p_demux;
597 demux_sys_t *p_sys = p_demux->p_sys;
598 asf_track_t *tk = p_sys->track[i_stream_number];
599
600 if ( tk )
601 tk->i_time = i_time;
602 }
603
604 static asf_track_info_t * Packet_GetTrackInfo( asf_packet_sys_t *p_packetsys,
605 uint8_t i_stream_number )
606 {
607 demux_t *p_demux = p_packetsys->p_demux;
608 demux_sys_t *p_sys = p_demux->p_sys;
609 asf_track_t *tk = p_sys->track[i_stream_number];
610
611 if (!tk)
612 return NULL;
613 else
614 return & tk->info;
615 }
616
617 static bool Packet_DoSkip( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number, bool b_packet_keyframe )
618 {
619 demux_t *p_demux = p_packetsys->p_demux;
620 demux_sys_t *p_sys = p_demux->p_sys;
621 const asf_track_t *tk = p_sys->track[i_stream_number];
622
623 if( tk == NULL )
624 {
625 msg_Warn( p_demux, "undeclared stream[Id 0x%x]", i_stream_number );
626 return true;
627 }
628
629 if( p_sys->i_wait_keyframe )
630 {
631 if ( i_stream_number == p_sys->i_seek_track )
632 {
633 if ( !b_packet_keyframe )
634 {
635 p_sys->i_wait_keyframe--;
636 return true;
637 }
638 else
639 p_sys->i_wait_keyframe = 0;
640 }
641 else
642 return true;
643 }
644
645 if( !tk->p_es )
646 return true;
647
648 return false;
649 }
650
651 static void Packet_Enqueue( asf_packet_sys_t *p_packetsys, uint8_t i_stream_number, block_t **pp_frame )
652 {
653 demux_t *p_demux = p_packetsys->p_demux;
654 demux_sys_t *p_sys = p_demux->p_sys;
655 asf_track_t *tk = p_sys->track[i_stream_number];
656 if ( !tk )
657 return;
658
659 block_t *p_gather = block_ChainGather( *pp_frame );
660 if( p_gather )
661 {
662 block_ChainLastAppend( & tk->queue.pp_last, p_gather );
663 #ifdef ASF_DEBUG
664 msg_Dbg( p_demux, " enqueue packet dts %"PRId64" pts %"PRId64" pcr %"PRId64, p_gather->i_dts, p_gather->i_pts, p_sys->i_time );
665 #endif
666 }
667
668 *pp_frame = NULL;
669 }
670
671 static bool Block_Dequeue( demux_t *p_demux, mtime_t i_nexttime )
672 {
673 demux_sys_t *p_sys = p_demux->p_sys;
674 bool b_tracks_have_data = false;
675 for( int i = 0; i < MAX_ASF_TRACKS; i++ )
676 {
677 asf_track_t *tk = p_sys->track[i];
678 if (!tk)
679 continue;
680 b_tracks_have_data |= (tk->queue.p_first != NULL);
681 while( tk->queue.p_first && tk->queue.p_first->i_dts <= i_nexttime )
682 {
683 block_t *p_block = tk->queue.p_first;
684 tk->queue.p_first = p_block->p_next;
685 if( tk->queue.p_first == NULL )
686 tk->queue.pp_last = &tk->queue.p_first;
687 else
688 p_block->p_next = NULL;
689
690 if( p_sys->i_time < VLC_TS_0 )
691 {
692 es_out_SetPCR( p_demux->out, VLC_TS_0 + p_sys->i_time );
693 #ifdef ASF_DEBUG
694 msg_Dbg( p_demux, " dequeue setting PCR to %"PRId64, VLC_TS_0 + p_sys->i_time );
695 #endif
696 }
697
698 #ifdef ASF_DEBUG
699 msg_Dbg( p_demux, " sending packet dts %"PRId64" pts %"PRId64" pcr %"PRId64, p_block->i_dts, p_block->i_pts, p_sys->i_time );
700 #endif
701 es_out_Send( p_demux->out, tk->p_es, p_block );
702 }
703 }
704 return b_tracks_have_data;
705 }
706
707 /*****************************************************************************
708 *
709 *****************************************************************************/
710 typedef struct asf_es_priorities_t
711 {
712 uint16_t *pi_stream_numbers;
713 uint16_t i_count;
714 } asf_es_priorities_t;
715
716 /* Fills up our exclusion list */
717 static void ASF_fillup_es_priorities_ex( demux_sys_t *p_sys, void *p_hdr,
718 asf_es_priorities_t *p_prios )
719 {
720 /* Find stream exclusions */
721 asf_object_advanced_mutual_exclusion_t *p_mutex =
722 ASF_FindObject( p_hdr, &asf_object_advanced_mutual_exclusion, 0 );
723 if (! p_mutex ) return;
724
725 #if ( UINT_MAX > SIZE_MAX / 2 )
726 if ( p_sys->i_track > (size_t)SIZE_MAX / sizeof(uint16_t) )
727 return;
728 #endif
729 p_prios->pi_stream_numbers = vlc_alloc( p_sys->i_track, sizeof(uint16_t) );
730 if ( !p_prios->pi_stream_numbers ) return;
731
732 if ( p_mutex->i_stream_number_count )
733 {
734 /* Just set highest prio on highest in the group */
735 for ( uint16_t i = 1; i < p_mutex->i_stream_number_count; i++ )
736 {
737 if ( p_prios->i_count > p_sys->i_track || i > p_sys->i_track ) break;
738 p_prios->pi_stream_numbers[ p_prios->i_count++ ] = p_mutex->pi_stream_number[ i ];
739 }
740 }
741 }
742
743 /* Fills up our bitrate exclusion list */
744 static void ASF_fillup_es_bitrate_priorities_ex( demux_sys_t *p_sys, void *p_hdr,
745 asf_es_priorities_t *p_prios )
746 {
747 /* Find bitrate exclusions */
748 asf_object_bitrate_mutual_exclusion_t *p_bitrate_mutex =
749 ASF_FindObject( p_hdr, &asf_object_bitrate_mutual_exclusion_guid, 0 );
750 if (! p_bitrate_mutex ) return;
751
752 #if ( UINT_MAX > SIZE_MAX / 2 )
753 if ( p_sys->i_track > (size_t)SIZE_MAX / sizeof(uint16_t) )
754 return;
755 #endif
756 p_prios->pi_stream_numbers = vlc_alloc( p_sys->i_track, sizeof( uint16_t ) );
757 if ( !p_prios->pi_stream_numbers ) return;
758
759 if ( p_bitrate_mutex->i_stream_number_count )
760 {
761 /* Just remove < highest */
762 for ( uint16_t i = 1; i < p_bitrate_mutex->i_stream_number_count; i++ )
763 {
764 if ( p_prios->i_count > p_sys->i_track || i > p_sys->i_track ) break;
765 p_prios->pi_stream_numbers[ p_prios->i_count++ ] = p_bitrate_mutex->pi_stream_numbers[ i ];
766 }
767 }
768
769 }
770
771 #define GET_CHECKED( target, getter, maxtarget, temp ) \
772 {\
773 temp i_temp = getter;\
774 if ( i_temp > maxtarget ) {\
775 msg_Warn( p_demux, "rejecting stream %u : " #target " overflow", i_stream );\
776 es_format_Clean( &fmt );\
777 goto error;\
778 } else {\
779 target = i_temp;\
780 }\
781 }
782
783 static int DemuxInit( demux_t *p_demux )
784 {
785 demux_sys_t *p_sys = p_demux->p_sys;
786
787 /* init context */
788 p_sys->i_time = -1;
789 p_sys->i_sendtime = -1;
790 p_sys->i_length = 0;
791 p_sys->b_eos = false;
792 p_sys->b_eof = false;
793 p_sys->i_bitrate = 0;
794 p_sys->p_root = NULL;
795 p_sys->p_fp = NULL;
796 p_sys->b_index = 0;
797 p_sys->i_track = 0;
798 p_sys->i_seek_track = 0;
799 p_sys->i_wait_keyframe = 0;
800 for( int i = 0; i < MAX_ASF_TRACKS; i++ )
801 {
802 p_sys->track[i] = NULL;
803 }
804 p_sys->i_data_begin = 0;
805 p_sys->i_data_end = 0;
806 p_sys->i_preroll_start = 0;
807 p_sys->meta = NULL;
808
809 /* Now load all object ( except raw data ) */
810 vlc_stream_Control( p_demux->s, STREAM_CAN_FASTSEEK,
811 &p_sys->b_canfastseek );
812 if( !(p_sys->p_root = ASF_ReadObjectRoot(p_demux->s, p_sys->b_canfastseek)) )
813 {
814 msg_Warn( p_demux, "ASF plugin discarded (not a valid file)" );
815 return VLC_EGENERIC;
816 }
817 p_sys->p_fp = p_sys->p_root->p_fp;
818
819 if( p_sys->p_fp->i_min_data_packet_size != p_sys->p_fp->i_max_data_packet_size )
820 {
821 msg_Warn( p_demux, "ASF plugin discarded (invalid file_properties object)" );
822 goto error;
823 }
824
825 if ( ASF_FindObject( p_sys->p_root->p_hdr,
826 &asf_object_content_encryption_guid, 0 ) != NULL
827 || ASF_FindObject( p_sys->p_root->p_hdr,
828 &asf_object_extended_content_encryption_guid, 0 ) != NULL
829 || ASF_FindObject( p_sys->p_root->p_hdr,
830 &asf_object_advanced_content_encryption_guid, 0 ) != NULL )
831 {
832 vlc_dialog_display_error( p_demux, _("Could not demux ASF stream"), "%s",
833 ("DRM protected streams are not supported.") );
834 goto error;
835 }
836
837 p_sys->i_track = ASF_CountObject( p_sys->p_root->p_hdr,
838 &asf_object_stream_properties_guid );
839 if( p_sys->i_track == 0 )
840 {
841 msg_Warn( p_demux, "ASF plugin discarded (cannot find any stream!)" );
842 goto error;
843 }
844 msg_Dbg( p_demux, "found %u streams", p_sys->i_track );
845
846 /* check if index is available */
847 asf_object_index_t *p_index = ASF_FindObject( p_sys->p_root,
848 &asf_object_simple_index_guid, 0 );
849 const bool b_index = p_index && p_index->i_index_entry_count;
850
851 /* Find the extended header if any */
852 asf_object_t *p_hdr_ext = ASF_FindObject( p_sys->p_root->p_hdr,
853 &asf_object_header_extension_guid, 0 );
854
855 asf_object_language_list_t *p_languages = NULL;
856 asf_es_priorities_t fmt_priorities_ex = { NULL, 0 };
857 asf_es_priorities_t fmt_priorities_bitrate_ex = { NULL, 0 };
858
859 if( p_hdr_ext )
860 {
861 p_languages = ASF_FindObject( p_hdr_ext, &asf_object_language_list, 0 );
862
863 ASF_fillup_es_priorities_ex( p_sys, p_hdr_ext, &fmt_priorities_ex );
864 ASF_fillup_es_bitrate_priorities_ex( p_sys, p_hdr_ext, &fmt_priorities_bitrate_ex );
865 }
866
867 const bool b_mms = !strncasecmp( p_demux->psz_url, "mms:", 4 );
868 bool b_dvrms = false;
869
870 if( b_mms )
871 {
872 es_out_Control( p_demux->out, ES_OUT_SET_ES_CAT_POLICY,
873 VIDEO_ES, ES_OUT_ES_POLICY_EXCLUSIVE );
874 }
875
876 for( unsigned i_stream = 0; i_stream < p_sys->i_track; i_stream++ )
877 {
878 asf_track_t *tk;
879 asf_object_stream_properties_t *p_sp;
880 asf_object_extended_stream_properties_t *p_esp;
881 bool b_access_selected;
882
883 p_sp = ASF_FindObject( p_sys->p_root->p_hdr,
884 &asf_object_stream_properties_guid,
885 i_stream );
886 p_esp = NULL;
887
888 tk = p_sys->track[p_sp->i_stream_number] = malloc( sizeof( asf_track_t ) );
889 if (!tk)
890 goto error;
891 memset( tk, 0, sizeof( asf_track_t ) );
892
893 tk->i_time = -1;
894 tk->info.p_sp = p_sp;
895 tk->p_es = NULL;
896 tk->info.p_esp = NULL;
897 tk->info.p_frame = NULL;
898 tk->info.i_cat = UNKNOWN_ES;
899 tk->queue.p_first = NULL;
900 tk->queue.pp_last = &tk->queue.p_first;
901
902 if ( !b_mms )
903 {
904 /* Check (not mms) if this track is selected (ie will receive data) */
905 if( !vlc_stream_Control( p_demux->s, STREAM_GET_PRIVATE_ID_STATE,
906 (int) p_sp->i_stream_number,
907 &b_access_selected ) &&
908 !b_access_selected )
909 {
910 tk->i_cat = UNKNOWN_ES;
911 msg_Dbg( p_demux, "ignoring not selected stream(ID:%u) (by access)",
912 p_sp->i_stream_number );
913 continue;
914 }
915 }
916
917 /* Find the associated extended_stream_properties if any */
918 if( p_hdr_ext )
919 {
920 int i_ext_stream = ASF_CountObject( p_hdr_ext,
921 &asf_object_extended_stream_properties_guid );
922 for( int i = 0; i < i_ext_stream; i++ )
923 {
924 asf_object_t *p_tmp =
925 ASF_FindObject( p_hdr_ext,
926 &asf_object_extended_stream_properties_guid, i );
927 if( p_tmp->ext_stream.i_stream_number == p_sp->i_stream_number )
928 {
929 p_esp = &p_tmp->ext_stream;
930 tk->info.p_esp = p_esp;
931 break;
932 }
933 }
934 }
935
936 /* Check for DVR-MS */
937 if( p_esp )
938 for( uint16_t i=0; i<p_esp->i_payload_extension_system_count && !b_dvrms; i++ )
939 b_dvrms = guidcmp( &p_esp->p_ext[i].i_extension_id, &asf_dvr_sampleextension_timing_rep_data_guid );
940
941 es_format_t fmt;
942
943 if( guidcmp( &p_sp->i_stream_type, &asf_object_stream_type_audio ) &&
944 p_sp->i_type_specific_data_length >= sizeof( WAVEFORMATEX ) - 2 )
945 {
946 uint8_t *p_data = p_sp->p_type_specific_data;
947 int i_format;
948
949 es_format_Init( &fmt, AUDIO_ES, 0 );
950 i_format = GetWLE( &p_data[0] );
951 wf_tag_to_fourcc( i_format, &fmt.i_codec, NULL );
952
953 GET_CHECKED( fmt.audio.i_channels, GetWLE( &p_data[2] ),
954 255, uint16_t );
955 GET_CHECKED( fmt.audio.i_rate, GetDWLE( &p_data[4] ),
956 UINT_MAX, uint32_t );
957 GET_CHECKED( fmt.i_bitrate, GetDWLE( &p_data[8] ) * 8,
958 UINT_MAX, uint32_t );
959 fmt.audio.i_blockalign = GetWLE( &p_data[12] );
960 fmt.audio.i_bitspersample = GetWLE( &p_data[14] );
961
962 if( p_sp->i_type_specific_data_length > sizeof( WAVEFORMATEX ) &&
963 i_format != WAVE_FORMAT_MPEGLAYER3 &&
964 i_format != WAVE_FORMAT_MPEG )
965 {
966 GET_CHECKED( fmt.i_extra, __MIN( GetWLE( &p_data[16] ),
967 p_sp->i_type_specific_data_length -
968 sizeof( WAVEFORMATEX ) ),
969 INT_MAX, uint32_t );
970 fmt.p_extra = malloc( fmt.i_extra );
971 memcpy( fmt.p_extra, &p_data[sizeof( WAVEFORMATEX )],
972 fmt.i_extra );
973 }
974 msg_Dbg( p_demux, "added new audio stream (codec:%4.4s(0x%x),ID:%d)",
975 (char*)&fmt.i_codec, GetWLE( p_data ), p_sp->i_stream_number );
976 }
977 else if( guidcmp( &p_sp->i_stream_type,
978 &asf_object_stream_type_video ) &&
979 p_sp->i_type_specific_data_length >= 11 +
980 sizeof( VLC_BITMAPINFOHEADER ) )
981 {
982 uint8_t *p_data = &p_sp->p_type_specific_data[11];
983
984 es_format_Init( &fmt, VIDEO_ES,
985 VLC_FOURCC( p_data[16], p_data[17],
986 p_data[18], p_data[19] ) );
987
988 GET_CHECKED( fmt.video.i_width, GetDWLE( p_data + 4 ),
989 UINT_MAX, uint32_t );
990 GET_CHECKED( fmt.video.i_height, GetDWLE( p_data + 8 ),
991 UINT_MAX, uint32_t );
992 fmt.video.i_visible_width = fmt.video.i_width;
993 fmt.video.i_visible_height = fmt.video.i_height;
994
995 if( p_esp && p_esp->i_average_time_per_frame > 0 )
996 {
997 fmt.video.i_frame_rate = 10000000;
998 GET_CHECKED( fmt.video.i_frame_rate_base,
999 p_esp->i_average_time_per_frame,
1000 UINT_MAX, uint64_t );
1001 }
1002
1003 if( fmt.i_codec == VLC_FOURCC( 'D','V','R',' ') )
1004 {
1005 /* DVR-MS special ASF */
1006 fmt.i_codec = VLC_CODEC_MPGV;
1007 }
1008
1009 if( p_sp->i_type_specific_data_length > 11 +
1010 sizeof( VLC_BITMAPINFOHEADER ) )
1011 {
1012 GET_CHECKED( fmt.i_extra, __MIN( GetDWLE( p_data ),
1013 p_sp->i_type_specific_data_length - 11 -
1014 sizeof( VLC_BITMAPINFOHEADER ) ),
1015 UINT_MAX, uint32_t );
1016 fmt.p_extra = malloc( fmt.i_extra );
1017 memcpy( fmt.p_extra, &p_data[sizeof( VLC_BITMAPINFOHEADER )],
1018 fmt.i_extra );
1019 }
1020
1021 /* Look for an aspect ratio */
1022 if( p_sys->p_root->p_metadata )
1023 {
1024 asf_object_metadata_t *p_meta = p_sys->p_root->p_metadata;
1025 unsigned int i_aspect_x = 0, i_aspect_y = 0;
1026 uint32_t i;
1027 for( i = 0; i < p_meta->i_record_entries_count; i++ )
1028 {
1029 if( !p_meta->record[i].psz_name )
1030 continue;
1031 if( !strcmp( p_meta->record[i].psz_name, "AspectRatioX" ) )
1032 {
1033 if( (!i_aspect_x && !p_meta->record[i].i_stream) ||
1034 p_meta->record[i].i_stream ==
1035 p_sp->i_stream_number )
1036 GET_CHECKED( i_aspect_x, p_meta->record[i].i_val,
1037 UINT_MAX, uint64_t );
1038 }
1039 if( !strcmp( p_meta->record[i].psz_name, "AspectRatioY" ) )
1040 {
1041 if( (!i_aspect_y && !p_meta->record[i].i_stream) ||
1042 p_meta->record[i].i_stream ==
1043 p_sp->i_stream_number )
1044 GET_CHECKED( i_aspect_y, p_meta->record[i].i_val,
1045 UINT_MAX, uint64_t );
1046 }
1047 }
1048
1049 if( i_aspect_x && i_aspect_y )
1050 {
1051 fmt.video.i_sar_num = i_aspect_x;
1052 fmt.video.i_sar_den = i_aspect_y;
1053 }
1054 }
1055
1056 /* If there is a video track then use the index for seeking */
1057 p_sys->b_index = b_index;
1058
1059 msg_Dbg( p_demux, "added new video stream(codec:%4.4s,ID:%d)",
1060 (char*)&fmt.i_codec, p_sp->i_stream_number );
1061 }
1062 else if( guidcmp( &p_sp->i_stream_type, &asf_object_stream_type_binary ) &&
1063 p_sp->i_type_specific_data_length >= 64 )
1064 {
1065 guid_t i_major_media_type;
1066 ASF_GetGUID( &i_major_media_type, p_sp->p_type_specific_data );
1067 msg_Dbg( p_demux, "stream(ID:%d) major type " GUID_FMT, p_sp->i_stream_number,
1068 GUID_PRINT(i_major_media_type) );
1069
1070 guid_t i_media_subtype;
1071 ASF_GetGUID( &i_media_subtype, &p_sp->p_type_specific_data[16] );
1072 msg_Dbg( p_demux, "stream(ID:%d) subtype " GUID_FMT, p_sp->i_stream_number,
1073 GUID_PRINT(i_media_subtype) );
1074
1075 //uint32_t i_fixed_size_samples = GetDWBE( &p_sp->p_type_specific_data[32] );
1076 //uint32_t i_temporal_compression = GetDWBE( &p_sp->p_type_specific_data[36] );
1077 //uint32_t i_sample_size = GetDWBE( &p_sp->p_type_specific_data[40] );
1078
1079 guid_t i_format_type;
1080 ASF_GetGUID( &i_format_type, &p_sp->p_type_specific_data[44] );
1081 msg_Dbg( p_demux, "stream(ID:%d) format type " GUID_FMT, p_sp->i_stream_number,
1082 GUID_PRINT(i_format_type) );
1083
1084 //uint32_t i_format_data_size = GetDWBE( &p_sp->p_type_specific_data[60] );
1085 uint8_t *p_data = p_sp->p_type_specific_data + 64;
1086 unsigned int i_data = p_sp->i_type_specific_data_length - 64;
1087
1088 msg_Dbg( p_demux, "Ext stream header detected. datasize = %d", p_sp->i_type_specific_data_length );
1089 if( guidcmp( &i_major_media_type, &asf_object_extended_stream_type_audio ) &&
1090 i_data >= sizeof( WAVEFORMATEX ) - 2)
1091 {
1092 uint16_t i_format;
1093 es_format_Init( &fmt, AUDIO_ES, 0 );
1094
1095 i_format = GetWLE( &p_data[0] );
1096 if( i_format == 0 )
1097 fmt.i_codec = VLC_CODEC_A52;
1098 else
1099 wf_tag_to_fourcc( i_format, &fmt.i_codec, NULL );
1100
1101 GET_CHECKED( fmt.audio.i_channels, GetWLE( &p_data[2] ),
1102 255, uint16_t );
1103 GET_CHECKED( fmt.audio.i_rate, GetDWLE( &p_data[4] ),
1104 UINT_MAX, uint32_t );
1105 GET_CHECKED( fmt.i_bitrate, GetDWLE( &p_data[8] ) * 8,
1106 UINT_MAX, uint32_t );
1107 fmt.audio.i_blockalign = GetWLE( &p_data[12] );
1108 fmt.audio.i_bitspersample = GetWLE( &p_data[14] );
1109
1110 if( p_sp->i_type_specific_data_length > sizeof( WAVEFORMATEX ) &&
1111 i_format != WAVE_FORMAT_MPEGLAYER3 &&
1112 i_format != WAVE_FORMAT_MPEG && i_data >= 19 )
1113 {
1114 GET_CHECKED( fmt.i_extra, __MIN( GetWLE( &p_data[16] ),
1115 p_sp->i_type_specific_data_length -
1116 sizeof( WAVEFORMATEX ) ),
1117 INT_MAX, uint32_t );
1118 fmt.p_extra = malloc( fmt.i_extra );
1119 if ( fmt.p_extra )
1120 memcpy( fmt.p_extra, &p_data[sizeof( WAVEFORMATEX )], fmt.i_extra );
1121 else
1122 fmt.i_extra = 0;
1123 }
1124
1125 msg_Dbg( p_demux, "added new audio stream (codec:%4.4s(0x%x),ID:%d)",
1126 (char*)&fmt.i_codec, i_format, p_sp->i_stream_number );
1127 }
1128 else
1129 {
1130 es_format_Init( &fmt, UNKNOWN_ES, 0 );
1131 }
1132 }
1133 else
1134 {
1135 es_format_Init( &fmt, UNKNOWN_ES, 0 );
1136 }
1137
1138 if( b_dvrms )
1139 {
1140 fmt.i_original_fourcc = VLC_FOURCC( 'D','V','R',' ');
1141 fmt.b_packetized = false;
1142 }
1143
1144 if( fmt.i_codec == VLC_CODEC_MP4A )
1145 fmt.b_packetized = false;
1146
1147 tk->i_cat = tk->info.i_cat = fmt.i_cat;
1148 if( fmt.i_cat != UNKNOWN_ES )
1149 {
1150 if( p_esp && p_languages &&
1151 p_esp->i_language_index < p_languages->i_language &&
1152 p_languages->ppsz_language[p_esp->i_language_index] )
1153 {
1154 fmt.psz_language = strdup( p_languages->ppsz_language[p_esp->i_language_index] );
1155 char *p;
1156 if( fmt.psz_language && (p = strchr( fmt.psz_language, '-' )) )
1157 *p = '\0';
1158 }
1159
1160 /* Set our priority so we won't get multiple videos */
1161 int i_priority = ES_PRIORITY_SELECTABLE_MIN;
1162 for( uint16_t i = 0; i < fmt_priorities_ex.i_count; i++ )
1163 {
1164 if ( fmt_priorities_ex.pi_stream_numbers[i] == p_sp->i_stream_number )
1165 {
1166 i_priority = ES_PRIORITY_NOT_DEFAULTABLE;
1167 break;
1168 }
1169 }
1170 for( uint16_t i = 0; i < fmt_priorities_bitrate_ex.i_count; i++ )
1171 {
1172 if ( fmt_priorities_bitrate_ex.pi_stream_numbers[i] == p_sp->i_stream_number )
1173 {
1174 i_priority = ES_PRIORITY_NOT_DEFAULTABLE;
1175 break;
1176 }
1177 }
1178 fmt.i_priority = i_priority;
1179
1180 if ( i_stream <= INT_MAX )
1181 fmt.i_id = i_stream;
1182 else
1183 msg_Warn( p_demux, "Can't set fmt.i_id to match stream id %u", i_stream );
1184
1185 if ( fmt.i_cat == VIDEO_ES )
1186 {
1187 /* Backup our video format */
1188 tk->p_fmt = malloc( sizeof( es_format_t ) );
1189 if ( tk->p_fmt )
1190 es_format_Copy( tk->p_fmt, &fmt );
1191 }
1192
1193 fmt.i_id = tk->info.p_sp->i_stream_number;
1194
1195 tk->p_es = es_out_Add( p_demux->out, &fmt );
1196
1197 if( !vlc_stream_Control( p_demux->s, STREAM_GET_PRIVATE_ID_STATE,
1198 (int) p_sp->i_stream_number,
1199 &b_access_selected ) &&
1200 b_access_selected )
1201 {
1202 p_sys->i_access_selected_track[fmt.i_cat] = p_sp->i_stream_number;
1203 }
1204
1205 }
1206 else
1207 {
1208 msg_Dbg( p_demux, "ignoring unknown stream(ID:%d)",
1209 p_sp->i_stream_number );
1210 }
1211
1212 es_format_Clean( &fmt );
1213 }
1214
1215 free( fmt_priorities_ex.pi_stream_numbers );
1216 free( fmt_priorities_bitrate_ex.pi_stream_numbers );
1217
1218 p_sys->i_data_begin = p_sys->p_root->p_data->i_object_pos + 50;
1219 if( p_sys->p_root->p_data->i_object_size != 0 )
1220 { /* local file */
1221 p_sys->i_data_end = p_sys->p_root->p_data->i_object_pos +
1222 p_sys->p_root->p_data->i_object_size;
1223 p_sys->i_data_end = __MIN( (uint64_t)stream_Size( p_demux->s ), p_sys->i_data_end );
1224 }
1225 else
1226 { /* live/broacast */
1227 p_sys->i_data_end = 0;
1228 }
1229
1230 /* go to first packet */
1231 if( vlc_stream_Seek( p_demux->s, p_sys->i_data_begin ) != VLC_SUCCESS )
1232 goto error;
1233
1234 /* try to calculate movie time */
1235 if( p_sys->p_fp->i_data_packets_count > 0 )
1236 {
1237 uint64_t i_count;
1238 uint64_t i_size = stream_Size( p_demux->s );
1239
1240 if( p_sys->i_data_end > 0 && i_size > p_sys->i_data_end )
1241 {
1242 i_size = p_sys->i_data_end;
1243 }
1244
1245 /* real number of packets */
1246 i_count = ( i_size - p_sys->i_data_begin ) /
1247 p_sys->p_fp->i_min_data_packet_size;
1248
1249 /* calculate the time duration in micro-s */
1250 p_sys->i_length = (mtime_t)p_sys->p_fp->i_play_duration / 10 *
1251 (mtime_t)i_count /
1252 (mtime_t)p_sys->p_fp->i_data_packets_count - p_sys->p_fp->i_preroll * 1000;
1253 if( p_sys->i_length < 0 )
1254 p_sys->i_length = 0;
1255
1256 if( p_sys->i_length > 0 )
1257 {
1258 p_sys->i_bitrate = 8 * i_size * 1000000 / p_sys->i_length;
1259 }
1260 }
1261
1262 /* Create meta information */
1263 p_sys->meta = vlc_meta_New();
1264
1265 asf_object_content_description_t *p_cd;
1266 if( ( p_cd = ASF_FindObject( p_sys->p_root->p_hdr,
1267 &asf_object_content_description_guid, 0 ) ) )
1268 {
1269 if( p_cd->psz_title && *p_cd->psz_title )
1270 {
1271 vlc_meta_SetTitle( p_sys->meta, p_cd->psz_title );
1272 }
1273 if( p_cd->psz_artist && *p_cd->psz_artist )
1274 {
1275 vlc_meta_SetArtist( p_sys->meta, p_cd->psz_artist );
1276 }
1277 if( p_cd->psz_copyright && *p_cd->psz_copyright )
1278 {
1279 vlc_meta_SetCopyright( p_sys->meta, p_cd->psz_copyright );
1280 }
1281 if( p_cd->psz_description && *p_cd->psz_description )
1282 {
1283 vlc_meta_SetDescription( p_sys->meta, p_cd->psz_description );
1284 }
1285 if( p_cd->psz_rating && *p_cd->psz_rating )
1286 {
1287 vlc_meta_SetRating( p_sys->meta, p_cd->psz_rating );
1288 }
1289 }
1290 asf_object_extended_content_description_t *p_ecd;
1291 if( ( p_ecd = ASF_FindObject( p_sys->p_root->p_hdr,
1292 &asf_object_extended_content_description, 0 ) ) )
1293 {
1294 for( int i = 0; i < p_ecd->i_count; i++ )
1295 {
1296
1297 #define set_meta( name, vlc_type ) \
1298 if( p_ecd->ppsz_name[i] && !strncmp( p_ecd->ppsz_name[i], name, strlen(name) ) ) \
1299 vlc_meta_Set( p_sys->meta, vlc_type, p_ecd->ppsz_value[i] );
1300
1301 set_meta( "WM/AlbumTitle", vlc_meta_Album )
1302 else set_meta( "WM/TrackNumber", vlc_meta_TrackNumber )
1303 else set_meta( "WM/Year", vlc_meta_Date )
1304 else set_meta( "WM/Genre", vlc_meta_Genre )
1305 else set_meta( "WM/Genre", vlc_meta_Genre )
1306 else set_meta( "WM/AlbumArtist", vlc_meta_AlbumArtist )
1307 else set_meta( "WM/Publisher", vlc_meta_Publisher )
1308 else set_meta( "WM/PartOfSet", vlc_meta_DiscNumber )
1309 else if( p_ecd->ppsz_value[i] != NULL && p_ecd->ppsz_name[i] &&
1310 *p_ecd->ppsz_value[i] != '\0' && /* no empty value */
1311 *p_ecd->ppsz_value[i] != '{' && /* no guid value */
1312 *p_ecd->ppsz_name[i] != '{' ) /* no guid name */
1313 vlc_meta_AddExtra( p_sys->meta, p_ecd->ppsz_name[i], p_ecd->ppsz_value[i] );
1314 /* TODO map WM/Composer, WM/Provider, WM/PartOfSet, PeakValue, AverageLevel */
1315 #undef set_meta
1316 }
1317 }
1318
1319 /// \tood Fix Child meta for ASF tracks
1320 #if 0
1321 for( i_stream = 0, i = 0; i < MAX_ASF_TRACKS; i++ )
1322 {
1323 asf_object_codec_list_t *p_cl = ASF_FindObject( p_sys->p_root->p_hdr,
1324 &asf_object_codec_list_guid, 0 );
1325
1326 if( p_sys->track[i] )
1327 {
1328 vlc_meta_t *tk = vlc_meta_New();
1329 TAB_APPEND( p_sys->meta->i_track, p_sys->meta->track, tk );
1330
1331 if( p_cl && i_stream < p_cl->i_codec_entries_count )
1332 {
1333 if( p_cl->codec[i_stream].psz_name &&
1334 *p_cl->codec[i_stream].psz_name )
1335 {
1336 vlc_meta_Add( tk, VLC_META_CODEC_NAME,
1337 p_cl->codec[i_stream].psz_name );
1338 }
1339 if( p_cl->codec[i_stream].psz_description &&
1340 *p_cl->codec[i_stream].psz_description )
1341 {
1342 vlc_meta_Add( tk, VLC_META_CODEC_DESCRIPTION,
1343 p_cl->codec[i_stream].psz_description );
1344 }
1345 }
1346 i_stream++;
1347 }
1348 }
1349 #endif
1350
1351 p_sys->packet_sys.pi_preroll = &p_sys->p_fp->i_preroll;
1352 p_sys->packet_sys.pi_preroll_start = &p_sys->i_preroll_start;
1353
1354 return VLC_SUCCESS;
1355
1356 error:
1357 DemuxEnd( p_demux );
1358 return VLC_EGENERIC;
1359 }
1360
1361 /*****************************************************************************
1362 * FlushQueues: flushes tail packets and send queues
1363 *****************************************************************************/
1364 static void FlushQueue( asf_track_t *tk )
1365 {
1366 if( tk->info.p_frame )
1367 {
1368 block_ChainRelease( tk->info.p_frame );
1369 tk->info.p_frame = NULL;
1370 }
1371 if( tk->queue.p_first )
1372 {
1373 block_ChainRelease( tk->queue.p_first );
1374 tk->queue.p_first = NULL;
1375 tk->queue.pp_last = &tk->queue.p_first;
1376 }
1377 }
1378
1379 static void FlushQueues( demux_t *p_demux )
1380 {
1381 demux_sys_t *p_sys = p_demux->p_sys;
1382 for ( unsigned int i = 0; i < MAX_ASF_TRACKS; i++ )
1383 {
1384 asf_track_t *tk = p_sys->track[i];
1385 if( !tk )
1386 continue;
1387 FlushQueue( tk );
1388 }
1389 }
1390
1391 /*****************************************************************************
1392 *
1393 *****************************************************************************/
1394 static void DemuxEnd( demux_t *p_demux )
1395 {
1396 demux_sys_t *p_sys = p_demux->p_sys;
1397
1398 if( p_sys->p_root )
1399 {
1400 ASF_FreeObjectRoot( p_demux->s, p_sys->p_root );
1401 p_sys->p_root = NULL;
1402 }
1403 if( p_sys->meta )
1404 {
1405 vlc_meta_Delete( p_sys->meta );
1406 p_sys->meta = NULL;
1407 }
1408
1409 FlushQueues( p_demux );
1410
1411 for( int i = 0; i < MAX_ASF_TRACKS; i++ )
1412 {
1413 asf_track_t *tk = p_sys->track[i];
1414
1415 if( tk )
1416 {
1417 if( tk->p_es )
1418 {
1419 es_out_Del( p_demux->out, tk->p_es );
1420 }
1421 if ( tk->p_fmt )
1422 {
1423 es_format_Clean( tk->p_fmt );
1424 free( tk->p_fmt );
1425 }
1426 free( tk );
1427 }
1428 p_sys->track[i] = 0;
1429 }
1430 }
1431
VLC Development tree mirror