pes.c 11.4 KB
Newer Older
1
/*****************************************************************************
2
 * pes.c: PES packetizer used by the MPEG multiplexers
3
4
 *****************************************************************************
 * Copyright (C) 2001, 2002 VideoLAN
5
 * $Id$
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
 *          Eric Petit <titer@videolan.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <string.h>
#include <errno.h>
#include <fcntl.h>

#include <vlc/vlc.h>
#include <vlc/input.h>
#include <vlc/sout.h>

#ifdef HAVE_UNISTD_H
#   include <unistd.h>
#endif

#include "codecs.h"
#include "pes.h"
#include "bits.h"

gbazin's avatar
   
gbazin committed
47
static inline int PESHeader( uint8_t *p_hdr, mtime_t i_pts, mtime_t i_dts,
48
49
                             int i_es_size, es_format_t *p_fmt,
                             int i_stream_id, int i_private_id,
50
51
                             vlc_bool_t b_mpeg2, vlc_bool_t b_data_alignment,
                             int i_header_size )
52
53
{
    bits_buffer_t bits;
54
55
56
57
58
59
60
61
    int     i_extra = 0;

    /* For PES_PRIVATE_STREAM_1 there is an extra header after the
       pes header */
    /* i_private_id != -1 because TS use 0xbd without private_id */
    if( i_stream_id == PES_PRIVATE_STREAM_1 && i_private_id != -1 )
    {
        i_extra = 1;
gbazin's avatar
   
gbazin committed
62
        if( ( i_private_id & 0xf0 ) == 0x80 )
63
64
65
66
67
        {
            i_extra += 3;
        }
    }

68

69
    bits_initwrite( &bits, 50, p_hdr );
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

    /* add start code */
    bits_write( &bits, 24, 0x01 );
    bits_write( &bits, 8, i_stream_id );
    switch( i_stream_id )
    {
        case PES_PROGRAM_STREAM_MAP:
        case PES_PADDING:
        case PES_PRIVATE_STREAM_2:
        case PES_ECM:
        case PES_EMM:
        case PES_PROGRAM_STREAM_DIRECTORY:
        case PES_DSMCC_STREAM:
        case PES_ITU_T_H222_1_TYPE_E_STREAM:
            /* add pes data size  */
            bits_write( &bits, 16, i_es_size );
            bits_align( &bits );
            return( bits.i_data );

        default:
            /* arg, a little more difficult */
91
            if( b_mpeg2 )
92
            {
gbazin's avatar
   
gbazin committed
93
                int i_pts_dts;
94

95
                if( i_pts > 0 && i_dts > 0 &&
96
97
                    ( i_pts != i_dts || ( p_fmt->i_cat == VIDEO_ES &&
                      p_fmt->i_codec != VLC_FOURCC('m','p','g','v') ) ) )
98
99
                {
                    i_pts_dts = 0x03;
100
                    if ( !i_header_size ) i_header_size = 0xa;
101
                }
gbazin's avatar
   
gbazin committed
102
                else if( i_pts > 0 )
103
104
                {
                    i_pts_dts = 0x02;
105
                    if ( !i_header_size ) i_header_size = 0x5;
106
                }
gbazin's avatar
   
gbazin committed
107
108
109
                else
                {
                    i_pts_dts = 0x00;
110
                    if ( !i_header_size ) i_header_size = 0x0;
gbazin's avatar
   
gbazin committed
111
                }
112

113
114
                bits_write( &bits, 16, i_es_size + i_extra + 3
                             + i_header_size ); // size
115
116
117
                bits_write( &bits, 2, 0x02 ); // mpeg2 id
                bits_write( &bits, 2, 0x00 ); // pes scrambling control
                bits_write( &bits, 1, 0x00 ); // pes priority
118
                bits_write( &bits, 1, b_data_alignment ); // data alignement indicator
119
120
121
122
123
124
125
126
127
128
                bits_write( &bits, 1, 0x00 ); // copyright
                bits_write( &bits, 1, 0x00 ); // original or copy

                bits_write( &bits, 2, i_pts_dts ); // pts_dts flags
                bits_write( &bits, 1, 0x00 ); // escr flags
                bits_write( &bits, 1, 0x00 ); // es rate flag
                bits_write( &bits, 1, 0x00 ); // dsm trick mode flag
                bits_write( &bits, 1, 0x00 ); // additional copy info flag
                bits_write( &bits, 1, 0x00 ); // pes crc flag
                bits_write( &bits, 1, 0x00 ); // pes extention flags
129
                bits_write( &bits, 8, i_header_size ); // header size -> pts and dts
130
131

                /* write pts */
gbazin's avatar
   
gbazin committed
132
133
134
135
136
137
138
139
140
                if( i_pts_dts & 0x02 )
                {
                    bits_write( &bits, 4, i_pts_dts ); // '0010' or '0011'
                    bits_write( &bits, 3, i_pts >> 30 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_pts >> 15 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_pts );
                    bits_write( &bits, 1, 0x01 ); // marker
141
                    i_header_size -= 0x5;
gbazin's avatar
   
gbazin committed
142
                }
143
144
145
146
147
148
149
150
151
152
                /* write i_dts */
                if( i_pts_dts & 0x01 )
                {
                    bits_write( &bits, 4, 0x01 ); // '0001'
                    bits_write( &bits, 3, i_dts >> 30 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_dts >> 15 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_dts );
                    bits_write( &bits, 1, 0x01 ); // marker
153
154
155
156
157
158
                    i_header_size -= 0x5;
                }
                while ( i_header_size )
                {
                    bits_write( &bits, 8, 0xff );
                    i_header_size--;
159
160
                }
            }
161
162
            else /* MPEG1 */
            {
163
164
                int i_pts_dts;

165
166
                if( i_pts > 0 && i_dts > 0 &&
                    ( i_pts != i_dts || p_fmt->i_cat == VIDEO_ES ) )
167
                {
168
                    bits_write( &bits, 16, i_es_size + i_extra + 10 /* + stuffing */ );
169
170
                    i_pts_dts = 0x03;
                }
gbazin's avatar
   
gbazin committed
171
                else if( i_pts > 0 )
172
                {
173
                    bits_write( &bits, 16, i_es_size + i_extra + 5 /* + stuffing */ );
174
175
176
177
                    i_pts_dts = 0x02;
                }
                else
                {
178
                    bits_write( &bits, 16, i_es_size + i_extra + 1 /* + stuffing */);
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
                    i_pts_dts = 0x00;
                }

                /* FIXME: Now should be stuffing */

                /* No STD_buffer_scale and STD_buffer_size */

                /* write pts */
                if( i_pts_dts & 0x02 )
                {
                    bits_write( &bits, 4, i_pts_dts ); // '0010' or '0011'
                    bits_write( &bits, 3, i_pts >> 30 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_pts >> 15 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_pts );
                    bits_write( &bits, 1, 0x01 ); // marker
                }
                /* write i_dts */
                if( i_pts_dts & 0x01 )
                {
                    bits_write( &bits, 4, 0x01 ); // '0001'
                    bits_write( &bits, 3, i_dts >> 30 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_dts >> 15 );
                    bits_write( &bits, 1, 0x01 ); // marker
                    bits_write( &bits, 15, i_dts );
                    bits_write( &bits, 1, 0x01 ); // marker
                }
208
                if( !i_pts_dts )
209
210
211
212
                {
                    bits_write( &bits, 8, 0x0F );
                }

213
            }
214

215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
            /* now should be stuffing */
            /* and then pes data */

            bits_align( &bits );
            if( i_stream_id == PES_PRIVATE_STREAM_1 && i_private_id != -1 )
            {
                bits_write( &bits, 8, i_private_id );
                if( ( i_private_id&0xf0 ) == 0x80 )
                {
                    bits_write( &bits, 24, 0 ); // ac3
                }
            }
            bits_align( &bits );
            return( bits.i_data );
    }
}

232
233
int E_( EStoPES )( sout_instance_t *p_sout, block_t **pp_pes, block_t *p_es,
                   es_format_t *p_fmt, int i_stream_id,
234
235
                   int b_mpeg2, int b_data_alignment, int i_header_size,
                   int i_max_pes_size )
236
{
237
    block_t *p_pes;
238
    mtime_t i_pts, i_dts, i_length;
239

240
241
242
    uint8_t *p_data;
    int     i_size;

243
244
    int     i_private_id = -1;

245
    uint8_t header[50];     // PES header + extra < 50 (more like 17)
246
247
    int     i_pes_payload;
    int     i_pes_header;
248

249
250
    int     i_pes_count = 1;

251
    /* HACK for private stream 1 in ps */
252
    if( ( i_stream_id >> 8 ) == PES_PRIVATE_STREAM_1 )
253
254
255
256
257
    {
        i_private_id = i_stream_id & 0xff;
        i_stream_id  = PES_PRIVATE_STREAM_1;
    }

258
259
260
261
262
263
264
265
266
    if( p_fmt->i_codec == VLC_FOURCC( 'm', 'p','4', 'v' ) &&
        p_es->i_flags & BLOCK_FLAG_TYPE_I )
    {
        /* For MPEG4 video, add VOL before I-frames */
        p_es = block_Realloc( p_es, p_fmt->i_extra, p_es->i_buffer );

        memcpy( p_es->p_buffer, p_fmt->p_extra, p_fmt->i_extra );
    }

gbazin's avatar
   
gbazin committed
267
268
    i_pts = p_es->i_pts <= 0 ? 0 : p_es->i_pts * 9 / 100; // 90000 units clock
    i_dts = p_es->i_dts <= 0 ? 0 : p_es->i_dts * 9 / 100; // 90000 units clock
269

270
    i_size = p_es->i_buffer;
271
272
273
274
275
276
    p_data = p_es->p_buffer;

    *pp_pes = p_pes = NULL;

    do
    {
277
278
        i_pes_payload = __MIN( i_size, (i_max_pes_size ?
                               i_max_pes_size : PES_PAYLOAD_SIZE_MAX) );
gbazin's avatar
   
gbazin committed
279
        i_pes_header  = PESHeader( header, i_pts, i_dts, i_pes_payload,
280
                                   p_fmt, i_stream_id, i_private_id, b_mpeg2,
281
                                   b_data_alignment, i_header_size );
gbazin's avatar
   
gbazin committed
282
283
        i_dts = 0; // only first PES has a dts/pts
        i_pts = 0;
284

285
        if( p_es )
286
        {
287
            p_es = block_Realloc( p_es, i_pes_header, p_es->i_buffer );
288
289
290
291
            /* reuse p_es for first frame */
            *pp_pes = p_pes = p_es;
            /* don't touch i_dts, i_pts, i_length as are already set :) */
            p_es = NULL;
292
293
294
        }
        else
        {
295
            p_pes->p_next = block_New( p_sout, i_pes_header + i_pes_payload );
296
297
298
299
300
            p_pes = p_pes->p_next;

            p_pes->i_dts    = 0;
            p_pes->i_pts    = 0;
            p_pes->i_length = 0;
301
302
            if( i_pes_payload > 0 )
            {
gbazin's avatar
   
gbazin committed
303
304
                p_sout->p_vlc->pf_memcpy( p_pes->p_buffer + i_pes_header,
                                          p_data, i_pes_payload );
305
            }
306
            i_pes_count++;
307
308
        }

309
310
        /* copy header */
        memcpy( p_pes->p_buffer, header, i_pes_header );
311

312
313
        i_size -= i_pes_payload;
        p_data += i_pes_payload;
314
        p_pes->i_buffer =  i_pes_header + i_pes_payload;
315

316
    } while( i_size > 0 );
317

318
319
320
321
322
323
324
325
326
327
328
    /* Now redate all pes */
    i_dts    = (*pp_pes)->i_dts;
    i_length = (*pp_pes)->i_length / i_pes_count;
    for( p_pes = *pp_pes; p_pes != NULL; p_pes = p_pes->p_next )
    {
        p_pes->i_dts = i_dts;
        p_pes->i_length = i_length;

        i_dts += i_length;
    }

329
330
    return 0;
}