1 /*****************************************************************************
2 * pes.c: PES packetizer used by the MPEG multiplexers
3 *****************************************************************************
4 * Copyright (C) 2001, 2002 the VideoLAN team
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 * Eric Petit <titer@videolan.org>
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
28 #include <sys/types.h>
39 #include <vlc_block.h>
45 #include <vlc_codecs.h>
49 static inline int PESHeader( uint8_t *p_hdr, mtime_t i_pts, mtime_t i_dts,
50 int i_es_size, es_format_t *p_fmt,
51 int i_stream_id, int i_private_id,
52 vlc_bool_t b_mpeg2, vlc_bool_t b_data_alignment,
58 /* For PES_PRIVATE_STREAM_1 there is an extra header after the
60 /* i_private_id != -1 because TS use 0xbd without private_id */
61 if( i_stream_id == PES_PRIVATE_STREAM_1 && i_private_id != -1 )
64 if( ( i_private_id & 0xf0 ) == 0x80 )
70 bits_initwrite( &bits, 50, p_hdr );
73 bits_write( &bits, 24, 0x01 );
74 bits_write( &bits, 8, i_stream_id );
77 case PES_PROGRAM_STREAM_MAP:
79 case PES_PRIVATE_STREAM_2:
82 case PES_PROGRAM_STREAM_DIRECTORY:
83 case PES_DSMCC_STREAM:
84 case PES_ITU_T_H222_1_TYPE_E_STREAM:
85 /* add pes data size */
86 bits_write( &bits, 16, i_es_size );
88 return( bits.i_data );
91 /* arg, a little more difficult */
96 if( i_pts > 0 && i_dts > 0 &&
97 ( i_pts != i_dts || ( p_fmt->i_cat == VIDEO_ES &&
98 p_fmt->i_codec != VLC_FOURCC('m','p','g','v') ) ) )
101 if ( !i_header_size ) i_header_size = 0xa;
106 if ( !i_header_size ) i_header_size = 0x5;
111 if ( !i_header_size ) i_header_size = 0x0;
114 bits_write( &bits, 16, i_es_size + i_extra + 3
115 + i_header_size ); // size
116 bits_write( &bits, 2, 0x02 ); // mpeg2 id
117 bits_write( &bits, 2, 0x00 ); // pes scrambling control
118 bits_write( &bits, 1, 0x00 ); // pes priority
119 bits_write( &bits, 1, b_data_alignment ); // data alignement indicator
120 bits_write( &bits, 1, 0x00 ); // copyright
121 bits_write( &bits, 1, 0x00 ); // original or copy
123 bits_write( &bits, 2, i_pts_dts ); // pts_dts flags
124 bits_write( &bits, 1, 0x00 ); // escr flags
125 bits_write( &bits, 1, 0x00 ); // es rate flag
126 bits_write( &bits, 1, 0x00 ); // dsm trick mode flag
127 bits_write( &bits, 1, 0x00 ); // additional copy info flag
128 bits_write( &bits, 1, 0x00 ); // pes crc flag
129 bits_write( &bits, 1, 0x00 ); // pes extension flags
130 bits_write( &bits, 8, i_header_size ); // header size -> pts and dts
133 if( i_pts_dts & 0x02 )
135 bits_write( &bits, 4, i_pts_dts ); // '0010' or '0011'
136 bits_write( &bits, 3, i_pts >> 30 );
137 bits_write( &bits, 1, 0x01 ); // marker
138 bits_write( &bits, 15, i_pts >> 15 );
139 bits_write( &bits, 1, 0x01 ); // marker
140 bits_write( &bits, 15, i_pts );
141 bits_write( &bits, 1, 0x01 ); // marker
142 i_header_size -= 0x5;
145 if( i_pts_dts & 0x01 )
147 bits_write( &bits, 4, 0x01 ); // '0001'
148 bits_write( &bits, 3, i_dts >> 30 );
149 bits_write( &bits, 1, 0x01 ); // marker
150 bits_write( &bits, 15, i_dts >> 15 );
151 bits_write( &bits, 1, 0x01 ); // marker
152 bits_write( &bits, 15, i_dts );
153 bits_write( &bits, 1, 0x01 ); // marker
154 i_header_size -= 0x5;
156 while ( i_header_size )
158 bits_write( &bits, 8, 0xff );
166 if( i_pts > 0 && i_dts > 0 &&
167 ( i_pts != i_dts || p_fmt->i_cat == VIDEO_ES ) )
169 bits_write( &bits, 16, i_es_size + i_extra + 10 /* + stuffing */ );
174 bits_write( &bits, 16, i_es_size + i_extra + 5 /* + stuffing */ );
179 bits_write( &bits, 16, i_es_size + i_extra + 1 /* + stuffing */);
183 /* FIXME: Now should be stuffing */
185 /* No STD_buffer_scale and STD_buffer_size */
188 if( i_pts_dts & 0x02 )
190 bits_write( &bits, 4, i_pts_dts ); // '0010' or '0011'
191 bits_write( &bits, 3, i_pts >> 30 );
192 bits_write( &bits, 1, 0x01 ); // marker
193 bits_write( &bits, 15, i_pts >> 15 );
194 bits_write( &bits, 1, 0x01 ); // marker
195 bits_write( &bits, 15, i_pts );
196 bits_write( &bits, 1, 0x01 ); // marker
199 if( i_pts_dts & 0x01 )
201 bits_write( &bits, 4, 0x01 ); // '0001'
202 bits_write( &bits, 3, i_dts >> 30 );
203 bits_write( &bits, 1, 0x01 ); // marker
204 bits_write( &bits, 15, i_dts >> 15 );
205 bits_write( &bits, 1, 0x01 ); // marker
206 bits_write( &bits, 15, i_dts );
207 bits_write( &bits, 1, 0x01 ); // marker
211 bits_write( &bits, 8, 0x0F );
216 /* now should be stuffing */
217 /* and then pes data */
220 if( i_stream_id == PES_PRIVATE_STREAM_1 && i_private_id != -1 )
222 bits_write( &bits, 8, i_private_id );
223 if( ( i_private_id&0xf0 ) == 0x80 )
225 bits_write( &bits, 24, 0 ); // ac3
229 return( bits.i_data );
233 int E_( EStoPES )( sout_instance_t *p_sout, block_t **pp_pes, block_t *p_es,
234 es_format_t *p_fmt, int i_stream_id,
235 int b_mpeg2, int b_data_alignment, int i_header_size,
239 mtime_t i_pts, i_dts, i_length;
244 int i_private_id = -1;
246 uint8_t header[50]; // PES header + extra < 50 (more like 17)
252 /* HACK for private stream 1 in ps */
253 if( ( i_stream_id >> 8 ) == PES_PRIVATE_STREAM_1 )
255 i_private_id = i_stream_id & 0xff;
256 i_stream_id = PES_PRIVATE_STREAM_1;
259 if( p_fmt->i_codec == VLC_FOURCC( 'm', 'p','4', 'v' ) &&
260 p_es->i_flags & BLOCK_FLAG_TYPE_I )
262 /* For MPEG4 video, add VOL before I-frames */
263 p_es = block_Realloc( p_es, p_fmt->i_extra, p_es->i_buffer );
265 memcpy( p_es->p_buffer, p_fmt->p_extra, p_fmt->i_extra );
268 i_pts = p_es->i_pts <= 0 ? 0 : p_es->i_pts * 9 / 100; // 90000 units clock
269 i_dts = p_es->i_dts <= 0 ? 0 : p_es->i_dts * 9 / 100; // 90000 units clock
271 i_size = p_es->i_buffer;
272 p_data = p_es->p_buffer;
274 *pp_pes = p_pes = NULL;
277 memset( header, 0, 50 );
282 i_pes_payload = __MIN( i_size, (i_max_pes_size ?
283 i_max_pes_size : PES_PAYLOAD_SIZE_MAX) );
284 i_pes_header = PESHeader( header, i_pts, i_dts, i_pes_payload,
285 p_fmt, i_stream_id, i_private_id, b_mpeg2,
286 b_data_alignment, i_header_size );
287 i_dts = 0; // only first PES has a dts/pts
292 p_es = block_Realloc( p_es, i_pes_header, p_es->i_buffer );
293 p_data = p_es->p_buffer+i_pes_header;
294 /* reuse p_es for first frame */
295 *pp_pes = p_pes = p_es;
296 /* don't touch i_dts, i_pts, i_length as are already set :) */
301 p_pes->p_next = block_New( p_sout, i_pes_header + i_pes_payload );
302 p_pes = p_pes->p_next;
307 if( i_pes_payload > 0 )
309 p_sout->p_libvlc->pf_memcpy( p_pes->p_buffer + i_pes_header,
310 p_data, i_pes_payload );
316 memcpy( p_pes->p_buffer, header, i_pes_header );
318 i_size -= i_pes_payload;
319 p_data += i_pes_payload;
320 p_pes->i_buffer = i_pes_header + i_pes_payload;
322 } while( i_size > 0 );
324 /* Now redate all pes */
325 i_dts = (*pp_pes)->i_dts;
326 i_length = (*pp_pes)->i_length / i_pes_count;
327 for( p_pes = *pp_pes; p_pes != NULL; p_pes = p_pes->p_next )
329 p_pes->i_dts = i_dts;
330 p_pes->i_length = i_length;