Mercurial > mplayer.hg
comparison libmpdemux/mpeg_packetizer.c @ 19141:6e6ca469073b
moved mpeg_packetizer helpers to libmpdemux
author | ben |
---|---|
date | Wed, 19 Jul 2006 22:13:50 +0000 |
parents | |
children | 3b839a8d297a |
comparison
equal
deleted
inserted
replaced
19140:c6c3ad8a5475 | 19141:6e6ca469073b |
---|---|
1 /* | |
2 * Copyright (C) 2006 Benjamin Zores | |
3 * Set of helper routines for building MPEG 1/2 PS/PES packets. | |
4 * | |
5 * Based on various code bororwed from vo_mpegpes/vo_dxr2 : | |
6 * (C) 2000 Ralph Metzler <ralph@convergence.de> | |
7 * Marcus Metzler <marcus@convergence.de> | |
8 * Gerard Lantau | |
9 * | |
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. | |
14 * | |
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. | |
19 * | |
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 Foundation, | |
22 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
23 */ | |
24 | |
25 #include <unistd.h> | |
26 #include <string.h> | |
27 #include <inttypes.h> | |
28 | |
29 #include "mp_msg.h" | |
30 #include "mpeg_packetizer.h" | |
31 | |
32 #define PES_MAX_SIZE 2048 | |
33 | |
34 static unsigned char pes_header[PES_MAX_SIZE]; | |
35 | |
36 static unsigned char ps_header[] = { | |
37 0x00, 0x00, 0x01, 0xba, 0x44, 0x00, 0x04, 0x00, | |
38 0x04, 0x01, 0x01, 0x86, 0xa3, 0xf8 | |
39 }; | |
40 | |
41 /* Send MPEG 1 PES packet */ | |
42 int | |
43 send_mpeg1_pes_packet (unsigned char *data, int len, int id, uint64_t pts, | |
44 int my_write (unsigned char *data, int len)) | |
45 { | |
46 int ptslen = pts ? 5 : 1; | |
47 int n = 0; | |
48 | |
49 mp_msg (MSGT_HEADER, MSGL_DBG2, | |
50 "MPEG1 PES packet: 0x%x => %lu \n", id, pts); | |
51 memset (pes_header, '\0', PES_MAX_SIZE); | |
52 | |
53 /* startcode */ | |
54 pes_header[0] = 0; | |
55 pes_header[1] = 0; | |
56 pes_header[2] = id >> 8; | |
57 pes_header[3] = id & 255; | |
58 | |
59 while (len > 0) | |
60 { | |
61 int payload_size = len; /* data + PTS */ | |
62 if (6 + ptslen + payload_size > PES_MAX_SIZE) | |
63 payload_size = PES_MAX_SIZE - (6 + ptslen); | |
64 | |
65 /* construct PES header: packetsize */ | |
66 pes_header[4] = (ptslen + payload_size) >> 8; | |
67 pes_header[5] = (ptslen + payload_size) & 255; | |
68 | |
69 if (ptslen == 5) | |
70 { | |
71 int x; | |
72 /* presentation time stamp */ | |
73 x = (0x02 << 4) | (((pts >> 30) & 0x07) << 1) | 1; | |
74 pes_header[6] = x; | |
75 | |
76 x = ((((pts >> 15) & 0x7fff) << 1) | 1); | |
77 pes_header[7] = x >> 8; | |
78 pes_header[8] = x & 255; | |
79 | |
80 x = (((pts & 0x7fff) << 1) | 1); | |
81 pes_header[9] = x >> 8; | |
82 pes_header[10] = x & 255; | |
83 } | |
84 else | |
85 { | |
86 /* stuffing and header bits */ | |
87 pes_header[6] = 0x0f; | |
88 } | |
89 | |
90 memcpy (&pes_header[6 + ptslen], data, payload_size); | |
91 n += my_write (pes_header, 6 + ptslen + payload_size); | |
92 | |
93 len -= payload_size; | |
94 data += payload_size; | |
95 ptslen = 1; /* store PTS only once, at first packet! */ | |
96 } | |
97 | |
98 return n; | |
99 } | |
100 | |
101 /* Send MPEG 1 PS packet */ | |
102 int | |
103 send_mpeg1_ps_packet (unsigned char *data, int len, int id, uint64_t pts, | |
104 int my_write (unsigned char *data, int len)) | |
105 { | |
106 my_write (ps_header, sizeof (ps_header)); | |
107 return send_mpeg1_pes_packet (data, len, id, pts, my_write); | |
108 } | |
109 | |
110 /* Send MPEG 2 PES packet */ | |
111 int | |
112 send_mpeg2_pes_packet (unsigned char *data, int len, int id, uint64_t pts, | |
113 int my_write (unsigned char *data, int len)) | |
114 { | |
115 int ptslen = 5; | |
116 int n = 0; | |
117 | |
118 mp_msg (MSGT_HEADER, MSGL_DBG2, | |
119 "MPEG2 PES packet: 0x%x => %lu \n", id, pts); | |
120 memset (pes_header, '\0', PES_MAX_SIZE); | |
121 | |
122 /* startcode */ | |
123 pes_header[0] = 0; | |
124 pes_header[1] = 0; | |
125 pes_header[2] = 0x01; | |
126 pes_header[3] = id; /* stream id */ | |
127 | |
128 while (len > 0) | |
129 { | |
130 int payload_size = len; /* data + PTS */ | |
131 if (9 + ptslen + payload_size > PES_MAX_SIZE) | |
132 payload_size = PES_MAX_SIZE - (6 + ptslen); | |
133 | |
134 /* construct PES header: packetize */ | |
135 pes_header[4] = (3 + ptslen + payload_size) >> 8; | |
136 pes_header[5] = (3 + ptslen + payload_size) & 255; | |
137 pes_header[6] = 0x81; | |
138 | |
139 if (ptslen) | |
140 { | |
141 int x; | |
142 pes_header[7] = 0x80; | |
143 pes_header[8] = ptslen; | |
144 | |
145 /* presentation time stamp */ | |
146 x = (0x02 << 4) | (((pts >> 30) & 0x07) << 1) | 1; | |
147 pes_header[9] = x; | |
148 | |
149 x = ((((pts >> 15) & 0x7fff) << 1) | 1); | |
150 pes_header[10] = x >>8; | |
151 pes_header[11] = x & 255; | |
152 | |
153 x = (((pts & 0x7fff) << 1) | 1); | |
154 pes_header[12] = x >> 8; | |
155 pes_header[13] = x & 255; | |
156 } | |
157 else | |
158 { | |
159 pes_header[7] = 0x00; | |
160 pes_header[8] = 0x00; | |
161 } | |
162 | |
163 my_write (pes_header, 9 + ptslen); | |
164 n += my_write (data, payload_size); | |
165 | |
166 len -= payload_size; | |
167 data += payload_size; | |
168 ptslen = 0; /* store PTS only once, at first packet! */ | |
169 } | |
170 | |
171 return n; | |
172 } | |
173 | |
174 /* Send MPEG 2 PS packet */ | |
175 int | |
176 send_mpeg2_ps_packet (unsigned char *data, int len, int id, uint64_t pts, | |
177 int my_write (unsigned char *data, int len)) | |
178 { | |
179 my_write (ps_header, sizeof (ps_header)); | |
180 return send_mpeg2_pes_packet (data, len, id, pts, my_write); | |
181 } | |
182 | |
183 /* Send MPEG LPCM packet */ | |
184 int | |
185 send_mpeg_lpcm_packet (unsigned char* data, int len, | |
186 int id, uint64_t pts, int freq_id, | |
187 int my_write (unsigned char *data, int len)) | |
188 { | |
189 int ptslen = pts ? 5 : 0; | |
190 int n = 0; | |
191 | |
192 mp_msg (MSGT_HEADER, MSGL_DBG2, | |
193 "MPEG LPCM packet: 0x%x => %lu \n", id, pts); | |
194 memset (pes_header, '\0', PES_MAX_SIZE); | |
195 | |
196 /* startcode */ | |
197 pes_header[0] = 0; | |
198 pes_header[1] = 0; | |
199 pes_header[2] = 1; | |
200 pes_header[3] = 0xBD; /* stream id */ | |
201 | |
202 while (len >= 4) | |
203 { | |
204 int payload_size = PES_MAX_SIZE - 6 - 20; /* max possible data len */ | |
205 if( payload_size > len) | |
206 payload_size = len; | |
207 | |
208 payload_size &= ~3; /* align! */ | |
209 | |
210 /* packetsize */ | |
211 pes_header[4] = (payload_size + 3 + ptslen + 7) >> 8; | |
212 pes_header[5] = (payload_size + 3 + ptslen + 7) & 255; | |
213 | |
214 /* stuffing */ | |
215 /* TTCCxxxx CC=css TT=type: 1=STD 0=mpeg1 2=vob */ | |
216 pes_header[6] = 0x81; | |
217 | |
218 /* FFxxxxxx FF=pts flags=2 vs 0 */ | |
219 pes_header[7] = ptslen ? 0x80 : 0; | |
220 | |
221 /* header length */ | |
222 pes_header[8] = ptslen; | |
223 | |
224 if (ptslen) | |
225 { | |
226 int x; | |
227 | |
228 /* presentation time stamp */ | |
229 x = (0x02 << 4) | (((pts >> 30) & 0x07) << 1) | 1; | |
230 pes_header[9] = x; | |
231 | |
232 x = ((((pts >> 15) & 0x7fff) << 1) | 1); | |
233 pes_header[10] = x >> 8; | |
234 pes_header[11] = x &255; | |
235 | |
236 x = (((pts & 0x7fff) << 1) | 1); | |
237 pes_header[12] = x >> 8; | |
238 pes_header[13] = x & 255; | |
239 } | |
240 | |
241 /* ============ LPCM header: (7 bytes) ================= */ | |
242 /* Info by mocm@convergence.de */ | |
243 | |
244 /* ID */ | |
245 pes_header[ptslen + 9] = id; | |
246 | |
247 /* number of frames */ | |
248 pes_header[ptslen + 10] = 0x07; | |
249 | |
250 /* first acces unit pointer, i.e. start of audio frame */ | |
251 pes_header[ptslen + 11] = 0x00; | |
252 pes_header[ptslen + 12] = 0x04; | |
253 | |
254 /* audio emphasis on-off 1 bit */ | |
255 /* audio mute on-off 1 bit */ | |
256 /* reserved 1 bit */ | |
257 /* audio frame number 5 bit */ | |
258 pes_header[ptslen + 13] = 0x0C; | |
259 | |
260 /* quantization word length 2 bit */ | |
261 /* audio sampling frequency (48khz = 0, 96khz = 1) 2 bit */ | |
262 /* reserved 1 bit */ | |
263 /* number of audio channels - 1 (e.g. stereo = 1) 3 bit */ | |
264 pes_header[ptslen + 14] = 1 | (freq_id << 4); | |
265 | |
266 /* dynamic range control (0x80 if off) */ | |
267 pes_header [ptslen + 15] = 0x80; | |
268 | |
269 memcpy (&pes_header[6 + 3 + ptslen + 7], data, payload_size); | |
270 n += my_write (pes_header, 6 + 3 + ptslen + 7 + payload_size); | |
271 | |
272 len -= payload_size; | |
273 data += payload_size; | |
274 ptslen = 0; /* store PTS only once, at first packet! */ | |
275 } | |
276 | |
277 return n; | |
278 } |