Mercurial > libavformat.hg
comparison rmdec.c @ 2103:95139f2053a9 libavformat
split rm muxer and demuxer in their own files
author | aurel |
---|---|
date | Wed, 30 May 2007 09:44:16 +0000 |
parents | rm.c@8b400be65542 |
children | ebf61cad0b61 |
comparison
equal
deleted
inserted
replaced
2102:ea6094e10af8 | 2103:95139f2053a9 |
---|---|
1 /* | |
2 * "Real" compatible demuxer. | |
3 * Copyright (c) 2000, 2001 Fabrice Bellard. | |
4 * | |
5 * This file is part of FFmpeg. | |
6 * | |
7 * FFmpeg is free software; you can redistribute it and/or | |
8 * modify it under the terms of the GNU Lesser General Public | |
9 * License as published by the Free Software Foundation; either | |
10 * version 2.1 of the License, or (at your option) any later version. | |
11 * | |
12 * FFmpeg is distributed in the hope that it will be useful, | |
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
15 * Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public | |
18 * License along with FFmpeg; if not, write to the Free Software | |
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
20 */ | |
21 #include "avformat.h" | |
22 #include "rm.h" | |
23 | |
24 | |
25 static void get_str(ByteIOContext *pb, char *buf, int buf_size) | |
26 { | |
27 int len, i; | |
28 char *q; | |
29 | |
30 len = get_be16(pb); | |
31 q = buf; | |
32 for(i=0;i<len;i++) { | |
33 if (i < buf_size - 1) | |
34 *q++ = get_byte(pb); | |
35 } | |
36 *q = '\0'; | |
37 } | |
38 | |
39 static void get_str8(ByteIOContext *pb, char *buf, int buf_size) | |
40 { | |
41 int len, i; | |
42 char *q; | |
43 | |
44 len = get_byte(pb); | |
45 q = buf; | |
46 for(i=0;i<len;i++) { | |
47 if (i < buf_size - 1) | |
48 *q++ = get_byte(pb); | |
49 } | |
50 *q = '\0'; | |
51 } | |
52 | |
53 static int rm_read_audio_stream_info(AVFormatContext *s, AVStream *st, | |
54 int read_all) | |
55 { | |
56 RMContext *rm = s->priv_data; | |
57 ByteIOContext *pb = &s->pb; | |
58 char buf[256]; | |
59 uint32_t version; | |
60 int i; | |
61 | |
62 /* ra type header */ | |
63 version = get_be32(pb); /* version */ | |
64 if (((version >> 16) & 0xff) == 3) { | |
65 int64_t startpos = url_ftell(pb); | |
66 /* very old version */ | |
67 for(i = 0; i < 14; i++) | |
68 get_byte(pb); | |
69 get_str8(pb, s->title, sizeof(s->title)); | |
70 get_str8(pb, s->author, sizeof(s->author)); | |
71 get_str8(pb, s->copyright, sizeof(s->copyright)); | |
72 get_str8(pb, s->comment, sizeof(s->comment)); | |
73 if ((startpos + (version & 0xffff)) >= url_ftell(pb) + 2) { | |
74 // fourcc (should always be "lpcJ") | |
75 get_byte(pb); | |
76 get_str8(pb, buf, sizeof(buf)); | |
77 } | |
78 // Skip extra header crap (this should never happen) | |
79 if ((startpos + (version & 0xffff)) > url_ftell(pb)) | |
80 url_fskip(pb, (version & 0xffff) + startpos - url_ftell(pb)); | |
81 st->codec->sample_rate = 8000; | |
82 st->codec->channels = 1; | |
83 st->codec->codec_type = CODEC_TYPE_AUDIO; | |
84 st->codec->codec_id = CODEC_ID_RA_144; | |
85 } else { | |
86 int flavor, sub_packet_h, coded_framesize, sub_packet_size; | |
87 /* old version (4) */ | |
88 get_be32(pb); /* .ra4 */ | |
89 get_be32(pb); /* data size */ | |
90 get_be16(pb); /* version2 */ | |
91 get_be32(pb); /* header size */ | |
92 flavor= get_be16(pb); /* add codec info / flavor */ | |
93 rm->coded_framesize = coded_framesize = get_be32(pb); /* coded frame size */ | |
94 get_be32(pb); /* ??? */ | |
95 get_be32(pb); /* ??? */ | |
96 get_be32(pb); /* ??? */ | |
97 rm->sub_packet_h = sub_packet_h = get_be16(pb); /* 1 */ | |
98 st->codec->block_align= get_be16(pb); /* frame size */ | |
99 rm->sub_packet_size = sub_packet_size = get_be16(pb); /* sub packet size */ | |
100 get_be16(pb); /* ??? */ | |
101 if (((version >> 16) & 0xff) == 5) { | |
102 get_be16(pb); get_be16(pb); get_be16(pb); } | |
103 st->codec->sample_rate = get_be16(pb); | |
104 get_be32(pb); | |
105 st->codec->channels = get_be16(pb); | |
106 if (((version >> 16) & 0xff) == 5) { | |
107 get_be32(pb); | |
108 buf[0] = get_byte(pb); | |
109 buf[1] = get_byte(pb); | |
110 buf[2] = get_byte(pb); | |
111 buf[3] = get_byte(pb); | |
112 buf[4] = 0; | |
113 } else { | |
114 get_str8(pb, buf, sizeof(buf)); /* desc */ | |
115 get_str8(pb, buf, sizeof(buf)); /* desc */ | |
116 } | |
117 st->codec->codec_type = CODEC_TYPE_AUDIO; | |
118 if (!strcmp(buf, "dnet")) { | |
119 st->codec->codec_id = CODEC_ID_AC3; | |
120 } else if (!strcmp(buf, "28_8")) { | |
121 st->codec->codec_id = CODEC_ID_RA_288; | |
122 st->codec->extradata_size= 0; | |
123 rm->audio_framesize = st->codec->block_align; | |
124 st->codec->block_align = coded_framesize; | |
125 | |
126 if(rm->audio_framesize >= UINT_MAX / sub_packet_h){ | |
127 av_log(s, AV_LOG_ERROR, "rm->audio_framesize * sub_packet_h too large\n"); | |
128 return -1; | |
129 } | |
130 | |
131 rm->audiobuf = av_malloc(rm->audio_framesize * sub_packet_h); | |
132 } else if ((!strcmp(buf, "cook")) || (!strcmp(buf, "atrc"))) { | |
133 int codecdata_length, i; | |
134 get_be16(pb); get_byte(pb); | |
135 if (((version >> 16) & 0xff) == 5) | |
136 get_byte(pb); | |
137 codecdata_length = get_be32(pb); | |
138 if(codecdata_length + FF_INPUT_BUFFER_PADDING_SIZE <= (unsigned)codecdata_length){ | |
139 av_log(s, AV_LOG_ERROR, "codecdata_length too large\n"); | |
140 return -1; | |
141 } | |
142 | |
143 if (!strcmp(buf, "cook")) st->codec->codec_id = CODEC_ID_COOK; | |
144 else st->codec->codec_id = CODEC_ID_ATRAC3; | |
145 st->codec->extradata_size= codecdata_length; | |
146 st->codec->extradata= av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); | |
147 for(i = 0; i < codecdata_length; i++) | |
148 ((uint8_t*)st->codec->extradata)[i] = get_byte(pb); | |
149 rm->audio_framesize = st->codec->block_align; | |
150 st->codec->block_align = rm->sub_packet_size; | |
151 | |
152 if(rm->audio_framesize >= UINT_MAX / sub_packet_h){ | |
153 av_log(s, AV_LOG_ERROR, "rm->audio_framesize * sub_packet_h too large\n"); | |
154 return -1; | |
155 } | |
156 | |
157 rm->audiobuf = av_malloc(rm->audio_framesize * sub_packet_h); | |
158 } else if (!strcmp(buf, "raac") || !strcmp(buf, "racp")) { | |
159 int codecdata_length, i; | |
160 get_be16(pb); get_byte(pb); | |
161 if (((version >> 16) & 0xff) == 5) | |
162 get_byte(pb); | |
163 st->codec->codec_id = CODEC_ID_AAC; | |
164 codecdata_length = get_be32(pb); | |
165 if(codecdata_length + FF_INPUT_BUFFER_PADDING_SIZE <= (unsigned)codecdata_length){ | |
166 av_log(s, AV_LOG_ERROR, "codecdata_length too large\n"); | |
167 return -1; | |
168 } | |
169 if (codecdata_length >= 1) { | |
170 st->codec->extradata_size = codecdata_length - 1; | |
171 st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); | |
172 get_byte(pb); | |
173 for(i = 0; i < st->codec->extradata_size; i++) | |
174 ((uint8_t*)st->codec->extradata)[i] = get_byte(pb); | |
175 } | |
176 } else { | |
177 st->codec->codec_id = CODEC_ID_NONE; | |
178 pstrcpy(st->codec->codec_name, sizeof(st->codec->codec_name), | |
179 buf); | |
180 } | |
181 if (read_all) { | |
182 get_byte(pb); | |
183 get_byte(pb); | |
184 get_byte(pb); | |
185 | |
186 get_str8(pb, s->title, sizeof(s->title)); | |
187 get_str8(pb, s->author, sizeof(s->author)); | |
188 get_str8(pb, s->copyright, sizeof(s->copyright)); | |
189 get_str8(pb, s->comment, sizeof(s->comment)); | |
190 } | |
191 } | |
192 return 0; | |
193 } | |
194 | |
195 static int rm_read_header_old(AVFormatContext *s, AVFormatParameters *ap) | |
196 { | |
197 RMContext *rm = s->priv_data; | |
198 AVStream *st; | |
199 | |
200 rm->old_format = 1; | |
201 st = av_new_stream(s, 0); | |
202 if (!st) | |
203 return -1; | |
204 return rm_read_audio_stream_info(s, st, 1); | |
205 } | |
206 | |
207 static int rm_read_header(AVFormatContext *s, AVFormatParameters *ap) | |
208 { | |
209 RMContext *rm = s->priv_data; | |
210 AVStream *st; | |
211 ByteIOContext *pb = &s->pb; | |
212 unsigned int tag, v; | |
213 int tag_size, size, codec_data_size, i; | |
214 int64_t codec_pos; | |
215 unsigned int start_time, duration; | |
216 char buf[128]; | |
217 int flags = 0; | |
218 | |
219 tag = get_le32(pb); | |
220 if (tag == MKTAG('.', 'r', 'a', 0xfd)) { | |
221 /* very old .ra format */ | |
222 return rm_read_header_old(s, ap); | |
223 } else if (tag != MKTAG('.', 'R', 'M', 'F')) { | |
224 return AVERROR_IO; | |
225 } | |
226 | |
227 get_be32(pb); /* header size */ | |
228 get_be16(pb); | |
229 get_be32(pb); | |
230 get_be32(pb); /* number of headers */ | |
231 | |
232 for(;;) { | |
233 if (url_feof(pb)) | |
234 goto fail; | |
235 tag = get_le32(pb); | |
236 tag_size = get_be32(pb); | |
237 get_be16(pb); | |
238 #if 0 | |
239 printf("tag=%c%c%c%c (%08x) size=%d\n", | |
240 (tag) & 0xff, | |
241 (tag >> 8) & 0xff, | |
242 (tag >> 16) & 0xff, | |
243 (tag >> 24) & 0xff, | |
244 tag, | |
245 tag_size); | |
246 #endif | |
247 if (tag_size < 10 && tag != MKTAG('D', 'A', 'T', 'A')) | |
248 goto fail; | |
249 switch(tag) { | |
250 case MKTAG('P', 'R', 'O', 'P'): | |
251 /* file header */ | |
252 get_be32(pb); /* max bit rate */ | |
253 get_be32(pb); /* avg bit rate */ | |
254 get_be32(pb); /* max packet size */ | |
255 get_be32(pb); /* avg packet size */ | |
256 get_be32(pb); /* nb packets */ | |
257 get_be32(pb); /* duration */ | |
258 get_be32(pb); /* preroll */ | |
259 get_be32(pb); /* index offset */ | |
260 get_be32(pb); /* data offset */ | |
261 get_be16(pb); /* nb streams */ | |
262 flags = get_be16(pb); /* flags */ | |
263 break; | |
264 case MKTAG('C', 'O', 'N', 'T'): | |
265 get_str(pb, s->title, sizeof(s->title)); | |
266 get_str(pb, s->author, sizeof(s->author)); | |
267 get_str(pb, s->copyright, sizeof(s->copyright)); | |
268 get_str(pb, s->comment, sizeof(s->comment)); | |
269 break; | |
270 case MKTAG('M', 'D', 'P', 'R'): | |
271 st = av_new_stream(s, 0); | |
272 if (!st) | |
273 goto fail; | |
274 st->id = get_be16(pb); | |
275 get_be32(pb); /* max bit rate */ | |
276 st->codec->bit_rate = get_be32(pb); /* bit rate */ | |
277 get_be32(pb); /* max packet size */ | |
278 get_be32(pb); /* avg packet size */ | |
279 start_time = get_be32(pb); /* start time */ | |
280 get_be32(pb); /* preroll */ | |
281 duration = get_be32(pb); /* duration */ | |
282 st->start_time = start_time; | |
283 st->duration = duration; | |
284 get_str8(pb, buf, sizeof(buf)); /* desc */ | |
285 get_str8(pb, buf, sizeof(buf)); /* mimetype */ | |
286 codec_data_size = get_be32(pb); | |
287 codec_pos = url_ftell(pb); | |
288 st->codec->codec_type = CODEC_TYPE_DATA; | |
289 av_set_pts_info(st, 64, 1, 1000); | |
290 | |
291 v = get_be32(pb); | |
292 if (v == MKTAG(0xfd, 'a', 'r', '.')) { | |
293 /* ra type header */ | |
294 if (rm_read_audio_stream_info(s, st, 0)) | |
295 return -1; | |
296 } else { | |
297 int fps, fps2; | |
298 if (get_le32(pb) != MKTAG('V', 'I', 'D', 'O')) { | |
299 fail1: | |
300 av_log(st->codec, AV_LOG_ERROR, "Unsupported video codec\n"); | |
301 goto skip; | |
302 } | |
303 st->codec->codec_tag = get_le32(pb); | |
304 // av_log(NULL, AV_LOG_DEBUG, "%X %X\n", st->codec->codec_tag, MKTAG('R', 'V', '2', '0')); | |
305 if ( st->codec->codec_tag != MKTAG('R', 'V', '1', '0') | |
306 && st->codec->codec_tag != MKTAG('R', 'V', '2', '0') | |
307 && st->codec->codec_tag != MKTAG('R', 'V', '3', '0') | |
308 && st->codec->codec_tag != MKTAG('R', 'V', '4', '0')) | |
309 goto fail1; | |
310 st->codec->width = get_be16(pb); | |
311 st->codec->height = get_be16(pb); | |
312 st->codec->time_base.num= 1; | |
313 fps= get_be16(pb); | |
314 st->codec->codec_type = CODEC_TYPE_VIDEO; | |
315 get_be32(pb); | |
316 fps2= get_be16(pb); | |
317 get_be16(pb); | |
318 | |
319 st->codec->extradata_size= codec_data_size - (url_ftell(pb) - codec_pos); | |
320 | |
321 if(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE <= (unsigned)st->codec->extradata_size){ | |
322 //check is redundant as get_buffer() will catch this | |
323 av_log(s, AV_LOG_ERROR, "st->codec->extradata_size too large\n"); | |
324 return -1; | |
325 } | |
326 st->codec->extradata= av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); | |
327 get_buffer(pb, st->codec->extradata, st->codec->extradata_size); | |
328 | |
329 // av_log(NULL, AV_LOG_DEBUG, "fps= %d fps2= %d\n", fps, fps2); | |
330 st->codec->time_base.den = fps * st->codec->time_base.num; | |
331 switch(((uint8_t*)st->codec->extradata)[4]>>4){ | |
332 case 1: st->codec->codec_id = CODEC_ID_RV10; break; | |
333 case 2: st->codec->codec_id = CODEC_ID_RV20; break; | |
334 case 3: st->codec->codec_id = CODEC_ID_RV30; break; | |
335 case 4: st->codec->codec_id = CODEC_ID_RV40; break; | |
336 default: goto fail1; | |
337 } | |
338 } | |
339 skip: | |
340 /* skip codec info */ | |
341 size = url_ftell(pb) - codec_pos; | |
342 url_fskip(pb, codec_data_size - size); | |
343 break; | |
344 case MKTAG('D', 'A', 'T', 'A'): | |
345 goto header_end; | |
346 default: | |
347 /* unknown tag: skip it */ | |
348 url_fskip(pb, tag_size - 10); | |
349 break; | |
350 } | |
351 } | |
352 header_end: | |
353 rm->nb_packets = get_be32(pb); /* number of packets */ | |
354 if (!rm->nb_packets && (flags & 4)) | |
355 rm->nb_packets = 3600 * 25; | |
356 get_be32(pb); /* next data header */ | |
357 return 0; | |
358 | |
359 fail: | |
360 for(i=0;i<s->nb_streams;i++) { | |
361 av_free(s->streams[i]); | |
362 } | |
363 return AVERROR_IO; | |
364 } | |
365 | |
366 static int get_num(ByteIOContext *pb, int *len) | |
367 { | |
368 int n, n1; | |
369 | |
370 n = get_be16(pb); | |
371 (*len)-=2; | |
372 if (n >= 0x4000) { | |
373 return n - 0x4000; | |
374 } else { | |
375 n1 = get_be16(pb); | |
376 (*len)-=2; | |
377 return (n << 16) | n1; | |
378 } | |
379 } | |
380 | |
381 /* multiple of 20 bytes for ra144 (ugly) */ | |
382 #define RAW_PACKET_SIZE 1000 | |
383 | |
384 static int sync(AVFormatContext *s, int64_t *timestamp, int *flags, int *stream_index, int64_t *pos){ | |
385 RMContext *rm = s->priv_data; | |
386 ByteIOContext *pb = &s->pb; | |
387 int len, num, res, i; | |
388 AVStream *st; | |
389 uint32_t state=0xFFFFFFFF; | |
390 | |
391 while(!url_feof(pb)){ | |
392 *pos= url_ftell(pb); | |
393 if(rm->remaining_len > 0){ | |
394 num= rm->current_stream; | |
395 len= rm->remaining_len; | |
396 *timestamp = AV_NOPTS_VALUE; | |
397 *flags= 0; | |
398 }else{ | |
399 state= (state<<8) + get_byte(pb); | |
400 | |
401 if(state == MKBETAG('I', 'N', 'D', 'X')){ | |
402 len = get_be16(pb) - 6; | |
403 if(len<0) | |
404 continue; | |
405 goto skip; | |
406 } | |
407 | |
408 if(state > (unsigned)0xFFFF || state < 12) | |
409 continue; | |
410 len=state; | |
411 state= 0xFFFFFFFF; | |
412 | |
413 num = get_be16(pb); | |
414 *timestamp = get_be32(pb); | |
415 res= get_byte(pb); /* reserved */ | |
416 *flags = get_byte(pb); /* flags */ | |
417 | |
418 | |
419 len -= 12; | |
420 } | |
421 for(i=0;i<s->nb_streams;i++) { | |
422 st = s->streams[i]; | |
423 if (num == st->id) | |
424 break; | |
425 } | |
426 if (i == s->nb_streams) { | |
427 skip: | |
428 /* skip packet if unknown number */ | |
429 url_fskip(pb, len); | |
430 rm->remaining_len -= len; | |
431 continue; | |
432 } | |
433 *stream_index= i; | |
434 | |
435 return len; | |
436 } | |
437 return -1; | |
438 } | |
439 | |
440 static int rm_read_packet(AVFormatContext *s, AVPacket *pkt) | |
441 { | |
442 RMContext *rm = s->priv_data; | |
443 ByteIOContext *pb = &s->pb; | |
444 AVStream *st; | |
445 int i, len, tmp, j; | |
446 int64_t timestamp, pos; | |
447 uint8_t *ptr; | |
448 int flags; | |
449 | |
450 if (rm->audio_pkt_cnt) { | |
451 // If there are queued audio packet return them first | |
452 st = s->streams[rm->audio_stream_num]; | |
453 if (st->codec->codec_id == CODEC_ID_AAC) | |
454 av_get_packet(pb, pkt, rm->sub_packet_lengths[rm->sub_packet_cnt - rm->audio_pkt_cnt]); | |
455 else { | |
456 av_new_packet(pkt, st->codec->block_align); | |
457 memcpy(pkt->data, rm->audiobuf + st->codec->block_align * | |
458 (rm->sub_packet_h * rm->audio_framesize / st->codec->block_align - rm->audio_pkt_cnt), | |
459 st->codec->block_align); | |
460 } | |
461 rm->audio_pkt_cnt--; | |
462 pkt->flags = 0; | |
463 pkt->stream_index = rm->audio_stream_num; | |
464 } else if (rm->old_format) { | |
465 st = s->streams[0]; | |
466 if (st->codec->codec_id == CODEC_ID_RA_288) { | |
467 int x, y; | |
468 | |
469 for (y = 0; y < rm->sub_packet_h; y++) | |
470 for (x = 0; x < rm->sub_packet_h/2; x++) | |
471 if (get_buffer(pb, rm->audiobuf+x*2*rm->audio_framesize+y*rm->coded_framesize, rm->coded_framesize) <= 0) | |
472 return AVERROR_IO; | |
473 rm->audio_stream_num = 0; | |
474 rm->audio_pkt_cnt = rm->sub_packet_h * rm->audio_framesize / st->codec->block_align - 1; | |
475 // Release first audio packet | |
476 av_new_packet(pkt, st->codec->block_align); | |
477 memcpy(pkt->data, rm->audiobuf, st->codec->block_align); | |
478 pkt->flags |= PKT_FLAG_KEY; // Mark first packet as keyframe | |
479 pkt->stream_index = 0; | |
480 } else { | |
481 /* just read raw bytes */ | |
482 len = RAW_PACKET_SIZE; | |
483 len= av_get_packet(pb, pkt, len); | |
484 pkt->stream_index = 0; | |
485 if (len <= 0) { | |
486 return AVERROR_IO; | |
487 } | |
488 pkt->size = len; | |
489 } | |
490 } else { | |
491 int seq=1; | |
492 resync: | |
493 len=sync(s, ×tamp, &flags, &i, &pos); | |
494 if(len<0) | |
495 return AVERROR_IO; | |
496 st = s->streams[i]; | |
497 | |
498 if (st->codec->codec_type == CODEC_TYPE_VIDEO) { | |
499 int h, pic_num, len2, pos; | |
500 | |
501 h= get_byte(pb); len--; | |
502 if(!(h & 0x40)){ | |
503 seq = get_byte(pb); len--; | |
504 } | |
505 | |
506 if((h & 0xc0) == 0x40){ | |
507 len2= pos= 0; | |
508 }else{ | |
509 len2 = get_num(pb, &len); | |
510 pos = get_num(pb, &len); | |
511 } | |
512 /* picture number */ | |
513 pic_num= get_byte(pb); len--; | |
514 rm->remaining_len= len; | |
515 rm->current_stream= st->id; | |
516 | |
517 // av_log(NULL, AV_LOG_DEBUG, "%X len:%d pos:%d len2:%d pic_num:%d\n",h, len, pos, len2, pic_num); | |
518 if(len2 && len2<len) | |
519 len=len2; | |
520 rm->remaining_len-= len; | |
521 av_get_packet(pb, pkt, len); | |
522 | |
523 } else if (st->codec->codec_type == CODEC_TYPE_AUDIO) { | |
524 if ((st->codec->codec_id == CODEC_ID_RA_288) || | |
525 (st->codec->codec_id == CODEC_ID_COOK) || | |
526 (st->codec->codec_id == CODEC_ID_ATRAC3)) { | |
527 int x; | |
528 int sps = rm->sub_packet_size; | |
529 int cfs = rm->coded_framesize; | |
530 int h = rm->sub_packet_h; | |
531 int y = rm->sub_packet_cnt; | |
532 int w = rm->audio_framesize; | |
533 | |
534 if (flags & 2) | |
535 y = rm->sub_packet_cnt = 0; | |
536 if (!y) | |
537 rm->audiotimestamp = timestamp; | |
538 | |
539 switch(st->codec->codec_id) { | |
540 case CODEC_ID_RA_288: | |
541 for (x = 0; x < h/2; x++) | |
542 get_buffer(pb, rm->audiobuf+x*2*w+y*cfs, cfs); | |
543 break; | |
544 case CODEC_ID_ATRAC3: | |
545 case CODEC_ID_COOK: | |
546 for (x = 0; x < w/sps; x++) | |
547 get_buffer(pb, rm->audiobuf+sps*(h*x+((h+1)/2)*(y&1)+(y>>1)), sps); | |
548 break; | |
549 } | |
550 | |
551 if (++(rm->sub_packet_cnt) < h) | |
552 goto resync; | |
553 else { | |
554 rm->sub_packet_cnt = 0; | |
555 rm->audio_stream_num = i; | |
556 rm->audio_pkt_cnt = h * w / st->codec->block_align - 1; | |
557 // Release first audio packet | |
558 av_new_packet(pkt, st->codec->block_align); | |
559 memcpy(pkt->data, rm->audiobuf, st->codec->block_align); | |
560 timestamp = rm->audiotimestamp; | |
561 flags = 2; // Mark first packet as keyframe | |
562 } | |
563 } else if (st->codec->codec_id == CODEC_ID_AAC) { | |
564 int x; | |
565 rm->audio_stream_num = i; | |
566 rm->sub_packet_cnt = (get_be16(pb) & 0xf0) >> 4; | |
567 if (rm->sub_packet_cnt) { | |
568 for (x = 0; x < rm->sub_packet_cnt; x++) | |
569 rm->sub_packet_lengths[x] = get_be16(pb); | |
570 // Release first audio packet | |
571 rm->audio_pkt_cnt = rm->sub_packet_cnt - 1; | |
572 av_get_packet(pb, pkt, rm->sub_packet_lengths[0]); | |
573 flags = 2; // Mark first packet as keyframe | |
574 } | |
575 } else | |
576 av_get_packet(pb, pkt, len); | |
577 | |
578 } else | |
579 av_get_packet(pb, pkt, len); | |
580 | |
581 if( (st->discard >= AVDISCARD_NONKEY && !(flags&2)) | |
582 || st->discard >= AVDISCARD_ALL){ | |
583 av_free_packet(pkt); | |
584 goto resync; | |
585 } | |
586 | |
587 pkt->stream_index = i; | |
588 | |
589 #if 0 | |
590 if (st->codec->codec_type == CODEC_TYPE_VIDEO) { | |
591 if(st->codec->codec_id == CODEC_ID_RV20){ | |
592 int seq= 128*(pkt->data[2]&0x7F) + (pkt->data[3]>>1); | |
593 av_log(NULL, AV_LOG_DEBUG, "%d %"PRId64" %d\n", timestamp, timestamp*512LL/25, seq); | |
594 | |
595 seq |= (timestamp&~0x3FFF); | |
596 if(seq - timestamp > 0x2000) seq -= 0x4000; | |
597 if(seq - timestamp < -0x2000) seq += 0x4000; | |
598 } | |
599 } | |
600 #endif | |
601 pkt->pts= timestamp; | |
602 if(flags&2){ | |
603 pkt->flags |= PKT_FLAG_KEY; | |
604 if((seq&0x7F) == 1) | |
605 av_add_index_entry(st, pos, timestamp, 0, 0, AVINDEX_KEYFRAME); | |
606 } | |
607 } | |
608 | |
609 /* for AC3, needs to swap bytes */ | |
610 if (st->codec->codec_id == CODEC_ID_AC3) { | |
611 ptr = pkt->data; | |
612 for(j=0;j<len;j+=2) { | |
613 tmp = ptr[0]; | |
614 ptr[0] = ptr[1]; | |
615 ptr[1] = tmp; | |
616 ptr += 2; | |
617 } | |
618 } | |
619 return 0; | |
620 } | |
621 | |
622 static int rm_read_close(AVFormatContext *s) | |
623 { | |
624 RMContext *rm = s->priv_data; | |
625 | |
626 av_free(rm->audiobuf); | |
627 return 0; | |
628 } | |
629 | |
630 static int rm_probe(AVProbeData *p) | |
631 { | |
632 /* check file header */ | |
633 if ((p->buf[0] == '.' && p->buf[1] == 'R' && | |
634 p->buf[2] == 'M' && p->buf[3] == 'F' && | |
635 p->buf[4] == 0 && p->buf[5] == 0) || | |
636 (p->buf[0] == '.' && p->buf[1] == 'r' && | |
637 p->buf[2] == 'a' && p->buf[3] == 0xfd)) | |
638 return AVPROBE_SCORE_MAX; | |
639 else | |
640 return 0; | |
641 } | |
642 | |
643 static int64_t rm_read_dts(AVFormatContext *s, int stream_index, | |
644 int64_t *ppos, int64_t pos_limit) | |
645 { | |
646 RMContext *rm = s->priv_data; | |
647 int64_t pos, dts; | |
648 int stream_index2, flags, len, h; | |
649 | |
650 pos = *ppos; | |
651 | |
652 if(rm->old_format) | |
653 return AV_NOPTS_VALUE; | |
654 | |
655 url_fseek(&s->pb, pos, SEEK_SET); | |
656 rm->remaining_len=0; | |
657 for(;;){ | |
658 int seq=1; | |
659 AVStream *st; | |
660 | |
661 len=sync(s, &dts, &flags, &stream_index2, &pos); | |
662 if(len<0) | |
663 return AV_NOPTS_VALUE; | |
664 | |
665 st = s->streams[stream_index2]; | |
666 if (st->codec->codec_type == CODEC_TYPE_VIDEO) { | |
667 h= get_byte(&s->pb); len--; | |
668 if(!(h & 0x40)){ | |
669 seq = get_byte(&s->pb); len--; | |
670 } | |
671 } | |
672 | |
673 if((flags&2) && (seq&0x7F) == 1){ | |
674 // av_log(s, AV_LOG_DEBUG, "%d %d-%d %"PRId64" %d\n", flags, stream_index2, stream_index, dts, seq); | |
675 av_add_index_entry(st, pos, dts, 0, 0, AVINDEX_KEYFRAME); | |
676 if(stream_index2 == stream_index) | |
677 break; | |
678 } | |
679 | |
680 url_fskip(&s->pb, len); | |
681 } | |
682 *ppos = pos; | |
683 return dts; | |
684 } | |
685 | |
686 AVInputFormat rm_demuxer = { | |
687 "rm", | |
688 "rm format", | |
689 sizeof(RMContext), | |
690 rm_probe, | |
691 rm_read_header, | |
692 rm_read_packet, | |
693 rm_read_close, | |
694 NULL, | |
695 rm_read_dts, | |
696 }; |