?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
mux:ps: fix variables that are mtime_t
[vlc.git] / modules / mux / mpeg / ps.c
blobfe89ad06bf6d46a971cb72764545120c7074ccb1
1 /*****************************************************************************
2 * ps.c: MPEG PS (ISO/IEC 13818-1) / MPEG SYSTEM (ISO/IEC 1172-1)
3 * multiplexer module for vlc
4 *****************************************************************************
5 * Copyright (C) 2001, 2002 VLC authors and VideoLAN
6 * $Id$
7 *
8 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
9 * Eric Petit <titer@videolan.org>
10 * Gildas Bazin <gbazin@videolan.org>
11 *
12 * This program is free software; you can redistribute it and/or modify it
13 * under the terms of the GNU Lesser General Public License as published by
14 * the Free Software Foundation; either version 2.1 of the License, or
15 * (at your option) any later version.
16 *
17 * This program is distributed in the hope that it will be useful,
18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 * GNU Lesser General Public License for more details.
21 *
22 * You should have received a copy of the GNU Lesser General Public License
23 * along with this program; if not, write to the Free Software Foundation,
24 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25 *****************************************************************************/
26
27 /*****************************************************************************
28 * Preamble
29 *****************************************************************************/
30
31 #ifdef HAVE_CONFIG_H
32 # include "config.h"
33 #endif
34
35 #include <vlc_common.h>
36 #include <vlc_plugin.h>
37 #include <vlc_sout.h>
38 #include <vlc_block.h>
39
40 #include "bits.h"
41 #include "pes.h"
42
43 #include <vlc_iso_lang.h>
44
45 /*****************************************************************************
46 * Module descriptor
47 *****************************************************************************/
48 #define DTS_TEXT N_("DTS delay (ms)")
49 #define DTS_LONGTEXT N_("Delay the DTS (decoding time " \
50 "stamps) and PTS (presentation timestamps) of the data in the " \
51 "stream, compared to the SCRs. This allows for some buffering inside " \
52 "the client decoder.")
53
54 #define PES_SIZE_TEXT N_("PES maximum size")
55 #define PES_SIZE_LONGTEXT N_("Set the maximum allowed PES "\
56 "size when producing the MPEG PS streams.")
57
58 static int Open ( vlc_object_t * );
59 static void Close ( vlc_object_t * );
60
61 #define SOUT_CFG_PREFIX "sout-ps-"
62
63 vlc_module_begin ()
64 set_description( N_("PS muxer") )
65 set_shortname( "MPEG-PS" )
66 set_category( CAT_SOUT )
67 set_subcategory( SUBCAT_SOUT_MUX )
68 set_capability( "sout mux", 50 )
69 add_shortcut( "ps", "mpeg1", "dvd" )
70 set_callbacks( Open, Close )
71
72 add_integer( SOUT_CFG_PREFIX "dts-delay", 200, DTS_TEXT,
73 DTS_LONGTEXT, true )
74 add_integer( SOUT_CFG_PREFIX "pes-max-size", PES_PAYLOAD_SIZE_MAX,
75 PES_SIZE_TEXT, PES_SIZE_LONGTEXT, true )
76 vlc_module_end ()
77
78 /*****************************************************************************
79 * Exported prototypes
80 *****************************************************************************/
81 static int Control ( sout_mux_t *, int, va_list );
82 static int AddStream( sout_mux_t *, sout_input_t * );
83 static void DelStream( sout_mux_t *, sout_input_t * );
84 static int Mux ( sout_mux_t * );
85
86 /*****************************************************************************
87 * Local prototypes
88 *****************************************************************************/
89
90 static void MuxWritePackHeader ( sout_mux_t *, block_t **, mtime_t );
91 static void MuxWriteSystemHeader( sout_mux_t *, block_t **, mtime_t );
92 static void MuxWritePSM ( sout_mux_t *, block_t **, mtime_t );
93
94 static void StreamIdInit ( bool *id, int i_range );
95 static int StreamIdGet ( bool *id, int i_id_min, int i_id_max );
96 static void StreamIdRelease ( bool *id, int i_id_min, int i_id );
97
98 typedef struct ps_stream_s
99 {
100 int i_stream_id;
101 int i_stream_type;
102 int i_max_buff_size; /* used in system header */
103
104 /* Language is iso639-2T */
105 uint8_t lang[3];
106 mtime_t i_dts;
107
108 } ps_stream_t;
109
110 typedef struct
111 {
112 /* Which id are unused */
113 bool stream_id_mpga[16]; /* 0xc0 -> 0xcf */
114 bool stream_id_mpgv[16]; /* 0xe0 -> 0xef */
115 bool stream_id_a52[8]; /* 0x80 -> 0x87 <- FIXME I'm not sure */
116 bool stream_id_spu[32]; /* 0x20 -> 0x3f */
117 bool stream_id_dts[8]; /* 0x88 -> 0x8f */
118 bool stream_id_lpcm[16]; /* 0xa0 -> 0xaf */
119
120 int i_audio_bound;
121 int i_video_bound;
122 int i_pes_count;
123 int i_system_header;
124 mtime_t i_dts_delay;
125 int i_rate_bound; /* units of 50 bytes/second */
126
127 int64_t i_instant_bitrate;
128 int64_t i_instant_size;
129 mtime_t i_instant_dts;
130
131 bool b_mpeg2;
132
133 int i_pes_max_size;
134
135 int i_psm_version;
136 uint32_t crc32_table[256];
137 } sout_mux_sys_t;
138
139 static const char *const ppsz_sout_options[] = {
140 "dts-delay", "pes-max-size", NULL
141 };
142
143 /*****************************************************************************
144 * Open:
145 *****************************************************************************/
146 static int Open( vlc_object_t *p_this )
147 {
148 sout_mux_t *p_mux = (sout_mux_t*)p_this;
149 sout_mux_sys_t *p_sys;
150 vlc_value_t val;
151
152 msg_Info( p_mux, "Open" );
153 config_ChainParse( p_mux, SOUT_CFG_PREFIX, ppsz_sout_options, p_mux->p_cfg );
154
155 p_mux->pf_control = Control;
156 p_mux->pf_addstream = AddStream;
157 p_mux->pf_delstream = DelStream;
158 p_mux->pf_mux = Mux;
159 p_mux->p_sys = p_sys = malloc( sizeof( sout_mux_sys_t ) );
160
161 /* Init free stream id */
162 StreamIdInit( p_sys->stream_id_a52, 8 );
163 StreamIdInit( p_sys->stream_id_dts, 8 );
164 StreamIdInit( p_sys->stream_id_mpga, 16 );
165 StreamIdInit( p_sys->stream_id_mpgv, 16 );
166 StreamIdInit( p_sys->stream_id_lpcm, 16 );
167 StreamIdInit( p_sys->stream_id_spu, 32 );
168
169 p_sys->i_audio_bound = 0;
170 p_sys->i_video_bound = 0;
171 p_sys->i_system_header = 0;
172 p_sys->i_pes_count = 0;
173
174 p_sys->i_psm_version = 0;
175
176 p_sys->i_instant_bitrate = 0;
177 p_sys->i_instant_size = 0;
178 p_sys->i_instant_dts = 0;
179 p_sys->i_rate_bound = 0;
180 p_sys->b_mpeg2 = !(p_mux->psz_mux && !strcmp( p_mux->psz_mux, "mpeg1" ));
181
182 var_Get( p_mux, SOUT_CFG_PREFIX "dts-delay", &val );
183 p_sys->i_dts_delay = (int64_t)val.i_int * 1000;
184
185 var_Get( p_mux, SOUT_CFG_PREFIX "pes-max-size", &val );
186 p_sys->i_pes_max_size = (int64_t)val.i_int;
187
188 /* Initialise CRC32 table */
189 if( p_sys->b_mpeg2 )
190 {
191 uint32_t i, j, k;
192
193 for( i = 0; i < 256; i++ )
194 {
195 k = 0;
196 for( j = (i << 24) | 0x800000; j != 0x80000000; j <<= 1 )
197 k = (k << 1) ^ (((k ^ j) & 0x80000000) ? 0x04c11db7 : 0);
198
199 p_sys->crc32_table[i] = k;
200 }
201 }
202
203 return VLC_SUCCESS;
204 }
205
206 /*****************************************************************************
207 * Close:
208 *****************************************************************************/
209 static void Close( vlc_object_t * p_this )
210 {
211 sout_mux_t *p_mux = (sout_mux_t*)p_this;
212 sout_mux_sys_t *p_sys = p_mux->p_sys;
213
214 block_t *p_end;
215
216 msg_Info( p_mux, "Close" );
217
218 p_end = block_Alloc( 4 );
219 if( p_end )
220 {
221 p_end->p_buffer[0] = 0x00; p_end->p_buffer[1] = 0x00;
222 p_end->p_buffer[2] = 0x01; p_end->p_buffer[3] = 0xb9;
223
224 sout_AccessOutWrite( p_mux->p_access, p_end );
225 }
226
227 free( p_sys );
228 }
229
230 /*****************************************************************************
231 * Control:
232 *****************************************************************************/
233 static int Control( sout_mux_t *p_mux, int i_query, va_list args )
234 {
235 VLC_UNUSED(p_mux);
236 bool *pb_bool;
237 char **ppsz;
238
239 switch( i_query )
240 {
241 case MUX_CAN_ADD_STREAM_WHILE_MUXING:
242 pb_bool = va_arg( args, bool * );
243 *pb_bool = true;
244 return VLC_SUCCESS;
245
246 case MUX_GET_ADD_STREAM_WAIT:
247 pb_bool = va_arg( args, bool * );
248 *pb_bool = false;
249 return VLC_SUCCESS;
250
251 case MUX_GET_MIME:
252 ppsz = va_arg( args, char ** );
253 *ppsz = strdup( "video/mpeg" );
254 return VLC_SUCCESS;
255
256 default:
257 return VLC_EGENERIC;
258 }
259 }
260
261 /*****************************************************************************
262 * AddStream:
263 *****************************************************************************/
264 static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input )
265 {
266 sout_mux_sys_t *p_sys = p_mux->p_sys;
267 ps_stream_t *p_stream;
268
269
270 msg_Dbg( p_mux, "adding input codec=%4.4s",
271 (char*)&p_input->p_fmt->i_codec );
272
273 p_input->p_sys = p_stream = malloc( sizeof( ps_stream_t ) );
274 if( unlikely(p_input->p_sys == NULL) )
275 return VLC_ENOMEM;
276 p_stream->i_stream_type = 0x81;
277 p_stream->i_dts = -1;
278
279 /* Init this new stream */
280 switch( p_input->p_fmt->i_codec )
281 {
282 case VLC_CODEC_MPGV:
283 case VLC_CODEC_MP2V:
284 case VLC_CODEC_MP1V:
285 p_stream->i_stream_id =
286 StreamIdGet( p_sys->stream_id_mpgv, 0xe0, 0xef );
287 p_stream->i_stream_type = 0x02; /* ISO/IEC 13818 Video */
288 break;
289 case VLC_CODEC_MP4V:
290 p_stream->i_stream_id =
291 StreamIdGet( p_sys->stream_id_mpgv, 0xe0, 0xef );
292 p_stream->i_stream_type = 0x10;
293 break;
294 case VLC_CODEC_H264:
295 p_stream->i_stream_id =
296 StreamIdGet( p_sys->stream_id_mpgv, 0xe0, 0xef );
297 p_stream->i_stream_type = 0x1b;
298 break;
299 case VLC_CODEC_DVD_LPCM:
300 p_stream->i_stream_id =
301 0xbd00 | StreamIdGet( p_sys->stream_id_lpcm, 0xa0, 0xaf );
302 break;
303 case VLC_CODEC_DTS:
304 p_stream->i_stream_id =
305 0xbd00 | StreamIdGet( p_sys->stream_id_dts, 0x88, 0x8f );
306 break;
307 case VLC_CODEC_A52:
308 p_stream->i_stream_id =
309 0xbd00 | StreamIdGet( p_sys->stream_id_a52, 0x80, 0x87 );
310 break;
311 case VLC_CODEC_MPGA:
312 p_stream->i_stream_id =
313 StreamIdGet( p_sys->stream_id_mpga, 0xc0, 0xcf );
314 p_stream->i_stream_type = 0x03; /* ISO/IEC 11172 Audio */
315 break;
316 case VLC_CODEC_MP4A:
317 p_stream->i_stream_id =
318 StreamIdGet( p_sys->stream_id_mpga, 0xc0, 0xcf );
319 p_stream->i_stream_type = 0x0f;
320 break;
321 case VLC_CODEC_SPU:
322 p_stream->i_stream_id =
323 0xbd00 | StreamIdGet( p_sys->stream_id_spu, 0x20, 0x3f );
324 break;
325 default:
326 goto error;
327 }
328
329 if( p_stream->i_stream_id < 0 ) goto error;
330
331 if( p_input->p_fmt->i_cat == AUDIO_ES )
332 {
333 p_sys->i_audio_bound++;
334 p_stream->i_max_buff_size = 4 * 1024;
335 }
336 else if( p_input->p_fmt->i_cat == VIDEO_ES )
337 {
338 p_sys->i_video_bound++;
339 p_stream->i_max_buff_size = 400 * 1024; /* FIXME -- VCD uses 46, SVCD
340 uses 230, ffmpeg has 230 with a note that it is small */
341 }
342 else
343 { /* FIXME -- what's valid for not audio or video? */
344 p_stream->i_max_buff_size = 4 * 1024;
345 }
346
347 /* Try to set a sensible default value for the instant bitrate */
348 p_sys->i_instant_bitrate += p_input->p_fmt->i_bitrate + 1000/* overhead */;
349
350 /* FIXME -- spec requires an upper limit rate boundary in the system header;
351 our codecs are VBR; using 2x nominal rate, convert to 50 bytes/sec */
352 p_sys->i_rate_bound += p_input->p_fmt->i_bitrate * 2 / (8 * 50);
353 p_sys->i_psm_version++;
354
355 p_stream->lang[0] = p_stream->lang[1] = p_stream->lang[2] = 0;
356 if( p_input->p_fmt->psz_language )
357 {
358 char *psz = p_input->p_fmt->psz_language;
359 const iso639_lang_t *pl = NULL;
360
361 if( strlen( psz ) == 2 )
362 {
363 pl = GetLang_1( psz );
364 }
365 else if( strlen( psz ) == 3 )
366 {
367 pl = GetLang_2B( psz );
368 if( !strcmp( pl->psz_iso639_1, "??" ) )
369 {
370 pl = GetLang_2T( psz );
371 }
372 }
373 if( pl && strcmp( pl->psz_iso639_1, "??" ) )
374 {
375 p_stream->lang[0] = pl->psz_iso639_2T[0];
376 p_stream->lang[1] = pl->psz_iso639_2T[1];
377 p_stream->lang[2] = pl->psz_iso639_2T[2];
378
379 msg_Dbg( p_mux, " - lang=%c%c%c",
380 p_stream->lang[0], p_stream->lang[1], p_stream->lang[2] );
381 }
382 }
383 return VLC_SUCCESS;
384
385 error:
386 free( p_stream );
387 return VLC_EGENERIC;
388 }
389
390 /*****************************************************************************
391 * DelStream:
392 *****************************************************************************/
393 static void DelStream( sout_mux_t *p_mux, sout_input_t *p_input )
394 {
395 sout_mux_sys_t *p_sys = p_mux->p_sys;
396 ps_stream_t *p_stream =(ps_stream_t*)p_input->p_sys;
397
398 msg_Dbg( p_mux, "removing input" );
399 switch( p_input->p_fmt->i_codec )
400 {
401 case VLC_CODEC_MPGV:
402 StreamIdRelease( p_sys->stream_id_mpgv, 0xe0,
403 p_stream->i_stream_id );
404 break;
405 case VLC_CODEC_DVD_LPCM:
406 StreamIdRelease( p_sys->stream_id_lpcm, 0xa0,
407 p_stream->i_stream_id&0xff );
408 break;
409 case VLC_CODEC_DTS:
410 StreamIdRelease( p_sys->stream_id_dts, 0x88,
411 p_stream->i_stream_id&0xff );
412 break;
413 case VLC_CODEC_A52:
414 StreamIdRelease( p_sys->stream_id_a52, 0x80,
415 p_stream->i_stream_id&0xff );
416 break;
417 case VLC_CODEC_MPGA:
418 StreamIdRelease( p_sys->stream_id_mpga, 0xc0,
419 p_stream->i_stream_id );
420 break;
421 case VLC_CODEC_SPU:
422 StreamIdRelease( p_sys->stream_id_spu, 0x20,
423 p_stream->i_stream_id&0xff );
424 break;
425 default:
426 /* Never reached */
427 break;
428 }
429
430 if( p_input->p_fmt->i_cat == AUDIO_ES )
431 {
432 p_sys->i_audio_bound--;
433 }
434 else if( p_input->p_fmt->i_cat == VIDEO_ES )
435 {
436 p_sys->i_video_bound--;
437 }
438
439 /* Try to set a sensible default value for the instant bitrate */
440 p_sys->i_instant_bitrate -= (p_input->p_fmt->i_bitrate + 1000);
441 /* rate_bound is in units of 50 bytes/second */
442 p_sys->i_rate_bound -= (p_input->p_fmt->i_bitrate * 2)/(8 * 50);
443
444 p_sys->i_psm_version++;
445
446 free( p_stream );
447 }
448
449 /*****************************************************************************
450 * Mux: Call each time there is new data for at least one stream
451 *****************************************************************************/
452 static int Mux( sout_mux_t *p_mux )
453 {
454 sout_mux_sys_t *p_sys = p_mux->p_sys;
455
456 for( ;; )
457 {
458 sout_input_t *p_input;
459 ps_stream_t *p_stream;
460
461 block_t *p_ps, *p_data;
462
463 mtime_t i_dts;
464
465 /* Choose which stream to mux */
466 int i_stream = sout_MuxGetStream( p_mux, 1, &i_dts );
467 if( i_stream < 0 )
468 {
469 return VLC_SUCCESS;
470 }
471
472 p_input = p_mux->pp_inputs[i_stream];
473 p_stream = (ps_stream_t*)p_input->p_sys;
474 p_ps = NULL;
475
476 p_stream->i_dts = i_dts;
477
478 /* Write regulary PackHeader */
479 if( p_sys->i_pes_count % 30 == 0)
480 {
481 mtime_t i_mindts = INT64_MAX;
482 for( size_t i=0; i<p_mux->i_nb_inputs; i++ )
483 {
484 ps_stream_t *p_s = (ps_stream_t*)p_input->p_sys;
485 if( p_input->p_fmt->i_cat == SPU_ES && p_mux->i_nb_inputs > 1 )
486 continue;
487 if( p_s->i_dts >= 0 && i_mindts > p_s->i_dts )
488 i_mindts = p_s->i_dts;
489 }
490
491 if( i_mindts > p_sys->i_instant_dts )
492 {
493 /* Update the instant bitrate every second or so */
494 if( p_sys->i_instant_size &&
495 i_dts - p_sys->i_instant_dts > 1*CLOCK_FREQ )
496 {
497 int64_t i_instant_bitrate = p_sys->i_instant_size * 8000000 /
498 ( i_dts - p_sys->i_instant_dts );
499
500 p_sys->i_instant_bitrate += i_instant_bitrate;
501 p_sys->i_instant_bitrate /= 2;
502
503 p_sys->i_instant_size = 0;
504 p_sys->i_instant_dts = i_dts;
505 }
506 else if( !p_sys->i_instant_size )
507 {
508 p_sys->i_instant_dts = i_dts;
509 }
510
511 MuxWritePackHeader( p_mux, &p_ps, i_dts );
512 }
513 }
514
515 /* Write regulary SystemHeader */
516 if( p_sys->i_pes_count % 300 == 0 )
517 {
518 block_t *p_pk;
519
520 MuxWriteSystemHeader( p_mux, &p_ps, i_dts );
521
522 /* For MPEG1 streaming, set HEADER flag */
523 for( p_pk = p_ps; p_pk != NULL; p_pk = p_pk->p_next )
524 {
525 p_pk->i_flags |= BLOCK_FLAG_HEADER;
526 }
527 }
528
529 /* Write regulary ProgramStreamMap */
530 if( p_sys->b_mpeg2 && p_sys->i_pes_count % 300 == 0 )
531 {
532 MuxWritePSM( p_mux, &p_ps, i_dts );
533 }
534
535 /* Get and mux a packet */
536 p_data = block_FifoGet( p_input->p_fifo );
537 EStoPES ( &p_data, p_input->p_fmt, p_stream->i_stream_id,
538 p_sys->b_mpeg2, 0, 0, p_sys->i_pes_max_size, 0 );
539
540 block_ChainAppend( &p_ps, p_data );
541
542 /* Get size of output data so we can calculate the instant bitrate */
543 for( p_data = p_ps; p_data; p_data = p_data->p_next )
544 {
545 p_sys->i_instant_size += p_data->i_buffer;
546 }
547
548 sout_AccessOutWrite( p_mux->p_access, p_ps );
549
550 /* Increase counter */
551 p_sys->i_pes_count++;
552 }
553
554 return VLC_SUCCESS;
555 }
556
557 /*****************************************************************************
558 *
559 *****************************************************************************/
560 static void StreamIdInit( bool *id, int i_range )
561 {
562 int i;
563
564 for( i = 0; i < i_range; i++ )
565 {
566 id[i] = true;
567 }
568 }
569 static int StreamIdGet( bool *id, int i_id_min, int i_id_max )
570 {
571 int i;
572
573 for( i = 0; i <= i_id_max - i_id_min; i++ )
574 {
575 if( id[i] )
576 {
577 id[i] = false;
578
579 return i_id_min + i;
580 }
581 }
582 return -1;
583 }
584 static void StreamIdRelease( bool *id, int i_id_min, int i_id )
585 {
586 id[i_id - i_id_min] = true;
587 }
588
589 static void MuxWritePackHeader( sout_mux_t *p_mux, block_t **p_buf,
590 mtime_t i_dts )
591 {
592 sout_mux_sys_t *p_sys = p_mux->p_sys;
593 bits_buffer_t bits;
594 block_t *p_hdr;
595 mtime_t i_scr;
596 int i_mux_rate;
597
598 i_scr = (i_dts - p_sys->i_dts_delay) * 9 / 100;
599
600 p_hdr = block_Alloc( 18 );
601 if( !p_hdr )
602 return;
603 p_hdr->i_pts = p_hdr->i_dts = i_dts;
604 bits_initwrite( &bits, 14, p_hdr->p_buffer );
605 bits_write( &bits, 32, 0x01ba );
606
607 /* The spec specifies that the mux rate must be rounded upwards */
608 i_mux_rate = (p_sys->i_instant_bitrate + 8 * 50 - 1 ) / (8 * 50);
609
610 if( p_sys->b_mpeg2 )
611 {
612 bits_write( &bits, 2, 0x01 );
613 }
614 else
615 {
616 bits_write( &bits, 4, 0x02 );
617 }
618
619 bits_write( &bits, 3, ( i_scr >> 30 )&0x07 );
620 bits_write( &bits, 1, 1 ); // marker
621 bits_write( &bits, 15, ( i_scr >> 15 )&0x7fff );
622 bits_write( &bits, 1, 1 ); // marker
623 bits_write( &bits, 15, i_scr&0x7fff );
624 bits_write( &bits, 1, 1 ); // marker
625
626 if( p_sys->b_mpeg2 )
627 {
628 bits_write( &bits, 9, 0 ); // src extension
629 }
630 bits_write( &bits, 1, 1 ); // marker
631
632 bits_write( &bits, 22, i_mux_rate);
633 bits_write( &bits, 1, 1 ); // marker
634
635 if( p_sys->b_mpeg2 )
636 {
637 bits_write( &bits, 1, 1 ); // marker
638 bits_write( &bits, 5, 0x1f ); // reserved
639 bits_write( &bits, 3, 0 ); // stuffing bytes
640 }
641
642 p_hdr->i_buffer = p_sys->b_mpeg2 ? 14: 12;
643
644 block_ChainAppend( p_buf, p_hdr );
645 }
646
647 static void MuxWriteSystemHeader( sout_mux_t *p_mux, block_t **p_buf,
648 mtime_t i_dts )
649 {
650 sout_mux_sys_t *p_sys = p_mux->p_sys;
651 block_t *p_hdr;
652 bits_buffer_t bits;
653 bool b_private;
654 int i_rate_bound;
655
656 int i_nb_private, i_nb_stream;
657 int i;
658
659 /* Count the number of private stream */
660 for( i = 0, i_nb_private = 0; i < p_mux->i_nb_inputs; i++ )
661 {
662 ps_stream_t *p_stream;
663
664 p_stream = (ps_stream_t*)p_mux->pp_inputs[i]->p_sys;
665
666 if( ( p_stream->i_stream_id&0xff00 ) == 0xbd00 )
667 {
668 i_nb_private++;
669 }
670 }
671
672 /* Private stream are declared only one time */
673 i_nb_stream = p_mux->i_nb_inputs -
674 ( i_nb_private > 0 ? i_nb_private - 1 : 0 );
675
676 p_hdr = block_Alloc( 12 + i_nb_stream * 3 );
677 if( !p_hdr )
678 return;
679 p_hdr->i_dts = p_hdr->i_pts = i_dts;
680
681 /* The spec specifies that the reported rate_bound must be upper limit */
682 i_rate_bound = (p_sys->i_rate_bound);
683
684 bits_initwrite( &bits, 12 + i_nb_stream * 3, p_hdr->p_buffer );
685 bits_write( &bits, 32, 0x01bb );
686 bits_write( &bits, 16, 12 - 6 + i_nb_stream * 3 );
687 bits_write( &bits, 1, 1 ); // marker bit
688 bits_write( &bits, 22, i_rate_bound);
689 bits_write( &bits, 1, 1 ); // marker bit
690
691 bits_write( &bits, 6, p_sys->i_audio_bound );
692 bits_write( &bits, 1, 0 ); // fixed flag
693 bits_write( &bits, 1, 0 ); // CSPS flag
694 bits_write( &bits, 1, 0 ); // system audio lock flag
695 bits_write( &bits, 1, 0 ); // system video lock flag
696
697 bits_write( &bits, 1, 1 ); // marker bit
698
699 bits_write( &bits, 5, p_sys->i_video_bound );
700 bits_write( &bits, 1, 1 ); // packet rate restriction flag (1 for mpeg1)
701 bits_write( &bits, 7, 0xff ); // reserved bits
702
703 /* stream_id table */
704 for( i = 0, b_private = false; i < p_mux->i_nb_inputs; i++ )
705 {
706 sout_input_t *p_input;
707 ps_stream_t *p_stream;
708
709 p_input = p_mux->pp_inputs[i];
710 p_stream = (ps_stream_t *)p_input->p_sys;
711
712 if( ( p_stream->i_stream_id&0xff00 ) == 0xbd00 )
713 {
714 if( b_private )
715 {
716 continue;
717 }
718 b_private = true;
719 /* Write stream id */
720 bits_write( &bits, 8, 0xbd );
721 }
722 else
723 {
724 /* Write stream id */
725 bits_write( &bits, 8, p_stream->i_stream_id&0xff );
726 }
727
728 bits_write( &bits, 2, 0x03 ); /* reserved */
729 if( p_input->p_fmt->i_cat == AUDIO_ES )
730 {
731 bits_write( &bits, 1, 0 );
732 bits_write( &bits, 13, p_stream->i_max_buff_size / 128 );
733 }
734 else if( p_input->p_fmt->i_cat == VIDEO_ES )
735 {
736 bits_write( &bits, 1, 1 );
737 bits_write( &bits, 13, p_stream->i_max_buff_size / 1024);
738 }
739 else
740 {
741 /* FIXME -- the scale of 0 means do a /128 */
742 bits_write( &bits, 1, 0 );
743 bits_write( &bits, 13, p_stream->i_max_buff_size / 128 );
744 }
745 }
746
747 block_ChainAppend( p_buf, p_hdr );
748 }
749
750 static void MuxWritePSM( sout_mux_t *p_mux, block_t **p_buf, mtime_t i_dts )
751 {
752 sout_mux_sys_t *p_sys = p_mux->p_sys;
753 block_t *p_hdr;
754 bits_buffer_t bits;
755 int i, i_psm_size = 16, i_es_map_size = 0;
756
757 for( i = 0; i < p_mux->i_nb_inputs; i++ )
758 {
759 sout_input_t *p_input = p_mux->pp_inputs[i];
760 ps_stream_t *p_stream = p_input->p_sys;
761
762 i_es_map_size += 4;
763 if( p_stream->lang[0] != 0 ) i_es_map_size += 6;
764 }
765
766 i_psm_size += i_es_map_size;
767
768 p_hdr = block_Alloc( i_psm_size );
769 if( !p_hdr )
770 return;
771 p_hdr->i_dts = p_hdr->i_pts = i_dts;
772
773 memset( p_hdr->p_buffer, 0, p_hdr->i_buffer );
774 bits_initwrite( &bits, i_psm_size, p_hdr->p_buffer );
775 bits_write( &bits, 32, 0x01bc );
776 bits_write( &bits, 16, i_psm_size - 6 );
777 bits_write( &bits, 1, 1 ); /* current_next_indicator */
778 bits_write( &bits, 2, 0xF ); /* reserved */
779 bits_write( &bits, 5, p_sys->i_psm_version );
780 bits_write( &bits, 7, 0xFF ); /* reserved */
781 bits_write( &bits, 1, 1 ); /* marker */
782
783 bits_write( &bits, 16, 0 ); /* program_stream_info_length */
784 /* empty */
785
786 bits_write( &bits, 16, i_es_map_size ); /* elementary_stream_map_length */
787 for( i = 0; i < p_mux->i_nb_inputs; i++ )
788 {
789 sout_input_t *p_input = p_mux->pp_inputs[i];
790 ps_stream_t *p_stream = p_input->p_sys;
791
792 bits_write( &bits, 8, p_stream->i_stream_type ); /* stream_type */
793 bits_write( &bits, 8, p_stream->i_stream_id ); /* elementary_stream_id */
794
795 /* ISO639 language descriptor */
796 if( p_stream->lang[0] != 0 )
797 {
798 bits_write( &bits, 16, 6 ); /* elementary_stream_info_length */
799
800 bits_write( &bits, 8, 0x0a ); /* descriptor_tag */
801 bits_write( &bits, 8, 4 ); /* descriptor_length */
802
803 bits_write( &bits, 8, p_stream->lang[0] );
804 bits_write( &bits, 8, p_stream->lang[1] );
805 bits_write( &bits, 8, p_stream->lang[2] );
806 bits_write( &bits, 8, 0 ); /* audio type: 0x00 undefined */
807 }
808 else
809 {
810 bits_write( &bits, 16, 0 ); /* elementary_stream_info_length */
811 }
812 }
813
814 /* CRC32 */
815 {
816 uint32_t i_crc = 0xffffffff;
817 for( i = 0; (size_t)i < p_hdr->i_buffer; i++ )
818 i_crc = (i_crc << 8) ^
819 p_sys->crc32_table[((i_crc >> 24) ^ p_hdr->p_buffer[i]) & 0xff];
820
821 bits_write( &bits, 32, i_crc );
822 }
823
824 block_ChainAppend( p_buf, p_hdr );
825 }
VLC Development tree mirror