]> git.sesse.net Git - ffmpeg/blob - libavformat/raw.c
3a53eee527015f95e3f542c645b18ff9bd81f655
[ffmpeg] / libavformat / raw.c
1 /*
2  * RAW muxer and demuxer
3  * Copyright (c) 2001 Fabrice Bellard
4  * Copyright (c) 2005 Alex Beregszaszi
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include "libavutil/crc.h"
24 #include "libavcodec/ac3_parser.h"
25 #include "libavcodec/get_bits.h"
26 #include "libavcodec/bytestream.h"
27 #include "avformat.h"
28 #include "raw.h"
29 #include "id3v2.h"
30 #include "id3v1.h"
31
32 /* simple formats */
33
34 #if CONFIG_ROQ_MUXER
35 static int roq_write_header(struct AVFormatContext *s)
36 {
37     static const uint8_t header[] = {
38         0x84, 0x10, 0xFF, 0xFF, 0xFF, 0xFF, 0x1E, 0x00
39     };
40
41     put_buffer(s->pb, header, 8);
42     put_flush_packet(s->pb);
43
44     return 0;
45 }
46 #endif
47
48 #if CONFIG_NULL_MUXER
49 static int null_write_packet(struct AVFormatContext *s, AVPacket *pkt)
50 {
51     return 0;
52 }
53 #endif
54
55 #if CONFIG_MUXERS
56 static int raw_write_packet(struct AVFormatContext *s, AVPacket *pkt)
57 {
58     put_buffer(s->pb, pkt->data, pkt->size);
59     put_flush_packet(s->pb);
60     return 0;
61 }
62 #endif
63
64 #if CONFIG_DEMUXERS
65 /* raw input */
66 static int raw_read_header(AVFormatContext *s, AVFormatParameters *ap)
67 {
68     AVStream *st;
69     int id;
70
71     st = av_new_stream(s, 0);
72     if (!st)
73         return AVERROR(ENOMEM);
74
75         id = s->iformat->value;
76         if (id == CODEC_ID_RAWVIDEO) {
77             st->codec->codec_type = CODEC_TYPE_VIDEO;
78         } else {
79             st->codec->codec_type = CODEC_TYPE_AUDIO;
80         }
81         st->codec->codec_id = id;
82
83         switch(st->codec->codec_type) {
84         case CODEC_TYPE_AUDIO:
85             st->codec->sample_rate = ap->sample_rate;
86             if(ap->channels) st->codec->channels = ap->channels;
87             else             st->codec->channels = 1;
88             st->codec->bits_per_coded_sample = av_get_bits_per_sample(st->codec->codec_id);
89             assert(st->codec->bits_per_coded_sample > 0);
90             st->codec->block_align = st->codec->bits_per_coded_sample*st->codec->channels/8;
91             av_set_pts_info(st, 64, 1, st->codec->sample_rate);
92             break;
93         case CODEC_TYPE_VIDEO:
94             if(ap->time_base.num)
95                 av_set_pts_info(st, 64, ap->time_base.num, ap->time_base.den);
96             else
97                 av_set_pts_info(st, 64, 1, 25);
98             st->codec->width = ap->width;
99             st->codec->height = ap->height;
100             st->codec->pix_fmt = ap->pix_fmt;
101             if(st->codec->pix_fmt == PIX_FMT_NONE)
102                 st->codec->pix_fmt= PIX_FMT_YUV420P;
103             break;
104         default:
105             return -1;
106         }
107     return 0;
108 }
109
110 #define RAW_PACKET_SIZE 1024
111 #define RAW_SAMPLES     1024
112
113 static int raw_read_packet(AVFormatContext *s, AVPacket *pkt)
114 {
115     int ret, size, bps;
116     //    AVStream *st = s->streams[0];
117
118     size= RAW_SAMPLES*s->streams[0]->codec->block_align;
119
120     ret= av_get_packet(s->pb, pkt, size);
121
122     pkt->stream_index = 0;
123     if (ret <= 0) {
124         return AVERROR(EIO);
125     }
126     /* note: we need to modify the packet size here to handle the last
127        packet */
128     pkt->size = ret;
129
130     bps= av_get_bits_per_sample(s->streams[0]->codec->codec_id);
131     assert(bps); // if false there IS a bug elsewhere (NOT in this function)
132     pkt->dts=
133     pkt->pts= pkt->pos*8 / (bps * s->streams[0]->codec->channels);
134
135     return ret;
136 }
137
138 int ff_raw_read_partial_packet(AVFormatContext *s, AVPacket *pkt)
139 {
140     int ret, size;
141
142     size = RAW_PACKET_SIZE;
143
144     if (av_new_packet(pkt, size) < 0)
145         return AVERROR(EIO);
146
147     pkt->pos= url_ftell(s->pb);
148     pkt->stream_index = 0;
149     ret = get_partial_buffer(s->pb, pkt->data, size);
150     if (ret <= 0) {
151         av_free_packet(pkt);
152         return AVERROR(EIO);
153     }
154     pkt->size = ret;
155     return ret;
156 }
157 #endif
158
159 #if CONFIG_RAWVIDEO_DEMUXER
160 static int rawvideo_read_packet(AVFormatContext *s, AVPacket *pkt)
161 {
162     int packet_size, ret, width, height;
163     AVStream *st = s->streams[0];
164
165     width = st->codec->width;
166     height = st->codec->height;
167
168     packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
169     if (packet_size < 0)
170         return -1;
171
172     ret= av_get_packet(s->pb, pkt, packet_size);
173     pkt->pts=
174     pkt->dts= pkt->pos / packet_size;
175
176     pkt->stream_index = 0;
177     if (ret != packet_size) {
178         return AVERROR(EIO);
179     } else {
180         return 0;
181     }
182 }
183 #endif
184
185 #if CONFIG_INGENIENT_DEMUXER
186 // http://www.artificis.hu/files/texts/ingenient.txt
187 static int ingenient_read_packet(AVFormatContext *s, AVPacket *pkt)
188 {
189     int ret, size, w, h, unk1, unk2;
190
191     if (get_le32(s->pb) != MKTAG('M', 'J', 'P', 'G'))
192         return AVERROR(EIO); // FIXME
193
194     size = get_le32(s->pb);
195
196     w = get_le16(s->pb);
197     h = get_le16(s->pb);
198
199     url_fskip(s->pb, 8); // zero + size (padded?)
200     url_fskip(s->pb, 2);
201     unk1 = get_le16(s->pb);
202     unk2 = get_le16(s->pb);
203     url_fskip(s->pb, 22); // ASCII timestamp
204
205     av_log(s, AV_LOG_DEBUG, "Ingenient packet: size=%d, width=%d, height=%d, unk1=%d unk2=%d\n",
206         size, w, h, unk1, unk2);
207
208     if (av_new_packet(pkt, size) < 0)
209         return AVERROR(EIO);
210
211     pkt->pos = url_ftell(s->pb);
212     pkt->stream_index = 0;
213     ret = get_buffer(s->pb, pkt->data, size);
214     if (ret <= 0) {
215         av_free_packet(pkt);
216         return AVERROR(EIO);
217     }
218     pkt->size = ret;
219     return ret;
220 }
221 #endif
222
223 #if CONFIG_DEMUXERS
224 int pcm_read_seek(AVFormatContext *s,
225                   int stream_index, int64_t timestamp, int flags)
226 {
227     AVStream *st;
228     int block_align, byte_rate, ret;
229     int64_t pos;
230
231     st = s->streams[0];
232
233     block_align = st->codec->block_align ? st->codec->block_align :
234         (av_get_bits_per_sample(st->codec->codec_id) * st->codec->channels) >> 3;
235     byte_rate = st->codec->bit_rate ? st->codec->bit_rate >> 3 :
236         block_align * st->codec->sample_rate;
237
238     if (block_align <= 0 || byte_rate <= 0)
239         return -1;
240     if (timestamp < 0) timestamp = 0;
241
242     /* compute the position by aligning it to block_align */
243     pos = av_rescale_rnd(timestamp * byte_rate,
244                          st->time_base.num,
245                          st->time_base.den * (int64_t)block_align,
246                          (flags & AVSEEK_FLAG_BACKWARD) ? AV_ROUND_DOWN : AV_ROUND_UP);
247     pos *= block_align;
248
249     /* recompute exact position */
250     st->cur_dts = av_rescale(pos, st->time_base.den, byte_rate * (int64_t)st->time_base.num);
251     if ((ret = url_fseek(s->pb, pos + s->data_offset, SEEK_SET)) < 0)
252         return ret;
253     return 0;
254 }
255
256 static int audio_read_header(AVFormatContext *s,
257                              AVFormatParameters *ap)
258 {
259     AVStream *st = av_new_stream(s, 0);
260     if (!st)
261         return AVERROR(ENOMEM);
262     st->codec->codec_type = CODEC_TYPE_AUDIO;
263     st->codec->codec_id = s->iformat->value;
264     st->need_parsing = AVSTREAM_PARSE_FULL;
265     /* the parameters will be extracted from the compressed bitstream */
266
267     return 0;
268 }
269
270 /* MPEG-1/H.263 input */
271 static int video_read_header(AVFormatContext *s,
272                              AVFormatParameters *ap)
273 {
274     AVStream *st;
275
276     st = av_new_stream(s, 0);
277     if (!st)
278         return AVERROR(ENOMEM);
279
280     st->codec->codec_type = CODEC_TYPE_VIDEO;
281     st->codec->codec_id = s->iformat->value;
282     st->need_parsing = AVSTREAM_PARSE_FULL;
283
284     /* for MJPEG, specify frame rate */
285     /* for MPEG-4 specify it, too (most MPEG-4 streams do not have the fixed_vop_rate set ...)*/
286     if (ap->time_base.num) {
287         st->codec->time_base= ap->time_base;
288     } else if ( st->codec->codec_id == CODEC_ID_MJPEG ||
289                 st->codec->codec_id == CODEC_ID_MPEG4 ||
290                 st->codec->codec_id == CODEC_ID_DIRAC ||
291                 st->codec->codec_id == CODEC_ID_H264) {
292         st->codec->time_base= (AVRational){1,25};
293     }
294     av_set_pts_info(st, 64, 1, 1200000);
295
296     return 0;
297 }
298 #endif
299
300 #if CONFIG_MPEGVIDEO_DEMUXER
301 #define SEQ_START_CODE          0x000001b3
302 #define GOP_START_CODE          0x000001b8
303 #define PICTURE_START_CODE      0x00000100
304 #define SLICE_START_CODE        0x00000101
305 #define PACK_START_CODE         0x000001ba
306 #define VIDEO_ID                0x000001e0
307 #define AUDIO_ID                0x000001c0
308
309 static int mpegvideo_probe(AVProbeData *p)
310 {
311     uint32_t code= -1;
312     int pic=0, seq=0, slice=0, pspack=0, pes=0;
313     int i;
314
315     for(i=0; i<p->buf_size; i++){
316         code = (code<<8) + p->buf[i];
317         if ((code & 0xffffff00) == 0x100) {
318             switch(code){
319             case     SEQ_START_CODE:   seq++; break;
320             case PICTURE_START_CODE:   pic++; break;
321             case   SLICE_START_CODE: slice++; break;
322             case    PACK_START_CODE: pspack++; break;
323             }
324             if     ((code & 0x1f0) == VIDEO_ID)   pes++;
325             else if((code & 0x1e0) == AUDIO_ID)   pes++;
326         }
327     }
328     if(seq && seq*9<=pic*10 && pic*9<=slice*10 && !pspack && !pes)
329         return AVPROBE_SCORE_MAX/2+1; // +1 for .mpg
330     return 0;
331 }
332 #endif
333
334 #if CONFIG_CAVSVIDEO_DEMUXER
335 #define CAVS_SEQ_START_CODE       0x000001b0
336 #define CAVS_PIC_I_START_CODE     0x000001b3
337 #define CAVS_UNDEF_START_CODE     0x000001b4
338 #define CAVS_PIC_PB_START_CODE    0x000001b6
339 #define CAVS_VIDEO_EDIT_CODE      0x000001b7
340 #define CAVS_PROFILE_JIZHUN       0x20
341
342 static int cavsvideo_probe(AVProbeData *p)
343 {
344     uint32_t code= -1;
345     int pic=0, seq=0, slice_pos = 0;
346     int i;
347
348     for(i=0; i<p->buf_size; i++){
349         code = (code<<8) + p->buf[i];
350         if ((code & 0xffffff00) == 0x100) {
351             if(code < CAVS_SEQ_START_CODE) {
352                 /* slices have to be consecutive */
353                 if(code < slice_pos)
354                     return 0;
355                 slice_pos = code;
356             } else {
357                 slice_pos = 0;
358             }
359             if (code == CAVS_SEQ_START_CODE) {
360                 seq++;
361                 /* check for the only currently supported profile */
362                 if(p->buf[i+1] != CAVS_PROFILE_JIZHUN)
363                     return 0;
364             } else if ((code == CAVS_PIC_I_START_CODE) ||
365                        (code == CAVS_PIC_PB_START_CODE)) {
366                 pic++;
367             } else if ((code == CAVS_UNDEF_START_CODE) ||
368                        (code >  CAVS_VIDEO_EDIT_CODE)) {
369                 return 0;
370             }
371         }
372     }
373     if(seq && seq*9<=pic*10)
374         return AVPROBE_SCORE_MAX/2;
375     return 0;
376 }
377 #endif
378
379 #if CONFIG_M4V_DEMUXER
380 #define VISUAL_OBJECT_START_CODE       0x000001b5
381 #define VOP_START_CODE                 0x000001b6
382
383 static int mpeg4video_probe(AVProbeData *probe_packet)
384 {
385     uint32_t temp_buffer= -1;
386     int VO=0, VOL=0, VOP = 0, VISO = 0, res=0;
387     int i;
388
389     for(i=0; i<probe_packet->buf_size; i++){
390         temp_buffer = (temp_buffer<<8) + probe_packet->buf[i];
391         if ((temp_buffer & 0xffffff00) != 0x100)
392             continue;
393
394         if (temp_buffer == VOP_START_CODE)                         VOP++;
395         else if (temp_buffer == VISUAL_OBJECT_START_CODE)          VISO++;
396         else if (temp_buffer < 0x120)                              VO++;
397         else if (temp_buffer < 0x130)                              VOL++;
398         else if (   !(0x1AF < temp_buffer && temp_buffer < 0x1B7)
399                  && !(0x1B9 < temp_buffer && temp_buffer < 0x1C4)) res++;
400     }
401
402     if ( VOP >= VISO && VOP >= VOL && VO >= VOL && VOL > 0 && res==0)
403         return AVPROBE_SCORE_MAX/2;
404     return 0;
405 }
406 #endif
407
408 #if CONFIG_H264_DEMUXER
409 static int h264_probe(AVProbeData *p)
410 {
411     uint32_t code= -1;
412     int sps=0, pps=0, idr=0, res=0, sli=0;
413     int i;
414
415     for(i=0; i<p->buf_size; i++){
416         code = (code<<8) + p->buf[i];
417         if ((code & 0xffffff00) == 0x100) {
418             int ref_idc= (code>>5)&3;
419             int type   = code & 0x1F;
420             static const int8_t ref_zero[32]={
421                 2, 0, 0, 0, 0,-1, 1,-1,
422                -1, 1, 1, 1, 1,-1, 2, 2,
423                 2, 2, 2, 0, 2, 2, 2, 2,
424                 2, 2, 2, 2, 2, 2, 2, 2
425             };
426
427             if(code & 0x80) //forbidden bit
428                 return 0;
429
430             if(ref_zero[type] == 1 && ref_idc)
431                 return 0;
432             if(ref_zero[type] ==-1 && !ref_idc)
433                 return 0;
434             if(ref_zero[type] == 2)
435                 res++;
436
437             switch(type){
438             case     1:   sli++; break;
439             case     5:   idr++; break;
440             case     7:
441                 if(p->buf[i+2]&0x0F)
442                     return 0;
443                 sps++;
444                 break;
445             case     8:   pps++; break;
446             }
447         }
448     }
449     if(sps && pps && (idr||sli>3) && res<(sps+pps+idr))
450         return AVPROBE_SCORE_MAX/2+1; // +1 for .mpg
451     return 0;
452 }
453 #endif
454
455 #if CONFIG_H263_DEMUXER
456 static int h263_probe(AVProbeData *p)
457 {
458     uint64_t code= -1;
459     int i;
460     int valid_psc=0;
461     int invalid_psc=0;
462     int res_change=0;
463     int src_fmt, last_src_fmt=-1;
464
465     for(i=0; i<p->buf_size; i++){
466         code = (code<<8) + p->buf[i];
467         if ((code & 0xfffffc0000) == 0x800000) {
468             src_fmt= (code>>2)&3;
469             if(   src_fmt != last_src_fmt
470                && last_src_fmt>0 && last_src_fmt<6
471                && src_fmt<6)
472                 res_change++;
473
474             if((code&0x300)==0x200 && src_fmt){
475                 valid_psc++;
476             }else
477                 invalid_psc++;
478             last_src_fmt= src_fmt;
479         }
480     }
481     if(valid_psc > 2*invalid_psc + 2*res_change + 2){
482         return 50;
483     }else if(valid_psc > 2*invalid_psc)
484         return 25;
485     return 0;
486 }
487 #endif
488
489 #if CONFIG_H261_DEMUXER
490 static int h261_probe(AVProbeData *p)
491 {
492     uint32_t code= -1;
493     int i;
494     int valid_psc=0;
495     int invalid_psc=0;
496     int next_gn=0;
497     int src_fmt=0;
498     GetBitContext gb;
499
500     init_get_bits(&gb, p->buf, p->buf_size*8);
501
502     for(i=0; i<p->buf_size*8; i++){
503         code = (code<<1) + get_bits1(&gb);
504         if ((code & 0xffff0000) == 0x10000) {
505             int gn= (code>>12)&0xf;
506             if(!gn)
507                 src_fmt= code&8;
508             if(gn != next_gn) invalid_psc++;
509             else              valid_psc++;
510
511             if(src_fmt){ // CIF
512                 next_gn= (gn+1     )%13;
513             }else{       //QCIF
514                 next_gn= (gn+1+!!gn)% 7;
515             }
516         }
517     }
518     if(valid_psc > 2*invalid_psc + 4){
519         return 50;
520     }else if(valid_psc > 2*invalid_psc + 2)
521         return 25;
522     return 0;
523 }
524 #endif
525
526 #if CONFIG_DTS_DEMUXER
527 #define DCA_MARKER_14B_BE 0x1FFFE800
528 #define DCA_MARKER_14B_LE 0xFF1F00E8
529 #define DCA_MARKER_RAW_BE 0x7FFE8001
530 #define DCA_MARKER_RAW_LE 0xFE7F0180
531 static int dts_probe(AVProbeData *p)
532 {
533     const uint8_t *buf, *bufp;
534     uint32_t state = -1;
535     int markers[3] = {0};
536     int sum, max;
537
538     buf = p->buf;
539
540     for(; buf < (p->buf+p->buf_size)-2; buf+=2) {
541         bufp = buf;
542         state = (state << 16) | bytestream_get_be16(&bufp);
543
544         /* regular bitstream */
545         if (state == DCA_MARKER_RAW_BE || state == DCA_MARKER_RAW_LE)
546             markers[0]++;
547
548         /* 14 bits big-endian bitstream */
549         if (state == DCA_MARKER_14B_BE)
550             if ((bytestream_get_be16(&bufp) & 0xFFF0) == 0x07F0)
551                 markers[1]++;
552
553         /* 14 bits little-endian bitstream */
554         if (state == DCA_MARKER_14B_LE)
555             if ((bytestream_get_be16(&bufp) & 0xF0FF) == 0xF007)
556                 markers[2]++;
557     }
558     sum = markers[0] + markers[1] + markers[2];
559     max = markers[1] > markers[0];
560     max = markers[2] > markers[max] ? 2 : max;
561     if (markers[max] > 3 && p->buf_size / markers[max] < 32*1024 &&
562         markers[max] * 4 > sum * 3)
563         return AVPROBE_SCORE_MAX/2+1;
564
565     return 0;
566 }
567 #endif
568
569 #if CONFIG_DIRAC_DEMUXER
570 static int dirac_probe(AVProbeData *p)
571 {
572     if (AV_RL32(p->buf) == MKTAG('B', 'B', 'C', 'D'))
573         return AVPROBE_SCORE_MAX;
574     else
575         return 0;
576 }
577 #endif
578
579 #if CONFIG_DNXHD_DEMUXER
580 static int dnxhd_probe(AVProbeData *p)
581 {
582     static const uint8_t header[] = {0x00,0x00,0x02,0x80,0x01};
583     int w, h, compression_id;
584     if (p->buf_size < 0x2c)
585         return 0;
586     if (memcmp(p->buf, header, 5))
587         return 0;
588     h = AV_RB16(p->buf + 0x18);
589     w = AV_RB16(p->buf + 0x1a);
590     if (!w || !h)
591         return 0;
592     compression_id = AV_RB32(p->buf + 0x28);
593     if (compression_id < 1237 || compression_id > 1253)
594         return 0;
595     return AVPROBE_SCORE_MAX;
596 }
597 #endif
598
599 #if CONFIG_AC3_DEMUXER || CONFIG_EAC3_DEMUXER
600 static int ac3_eac3_probe(AVProbeData *p, enum CodecID expected_codec_id)
601 {
602     int max_frames, first_frames = 0, frames;
603     uint8_t *buf, *buf2, *end;
604     AC3HeaderInfo hdr;
605     GetBitContext gbc;
606     enum CodecID codec_id = CODEC_ID_AC3;
607
608     max_frames = 0;
609     buf = p->buf;
610     end = buf + p->buf_size;
611
612     for(; buf < end; buf++) {
613         buf2 = buf;
614
615         for(frames = 0; buf2 < end; frames++) {
616             init_get_bits(&gbc, buf2, 54);
617             if(ff_ac3_parse_header(&gbc, &hdr) < 0)
618                 break;
619             if(buf2 + hdr.frame_size > end ||
620                av_crc(av_crc_get_table(AV_CRC_16_ANSI), 0, buf2 + 2, hdr.frame_size - 2))
621                 break;
622             if (hdr.bitstream_id > 10)
623                 codec_id = CODEC_ID_EAC3;
624             buf2 += hdr.frame_size;
625         }
626         max_frames = FFMAX(max_frames, frames);
627         if(buf == p->buf)
628             first_frames = frames;
629     }
630     if(codec_id != expected_codec_id) return 0;
631     if   (first_frames>=3) return AVPROBE_SCORE_MAX * 3 / 4;
632     else if(max_frames>=3) return AVPROBE_SCORE_MAX / 2;
633     else if(max_frames>=1) return 1;
634     else                   return 0;
635 }
636 #endif
637
638 #if CONFIG_AC3_DEMUXER
639 static int ac3_probe(AVProbeData *p)
640 {
641     return ac3_eac3_probe(p, CODEC_ID_AC3);
642 }
643 #endif
644
645 #if CONFIG_EAC3_DEMUXER
646 static int eac3_probe(AVProbeData *p)
647 {
648     return ac3_eac3_probe(p, CODEC_ID_EAC3);
649 }
650 #endif
651
652 #if CONFIG_AAC_DEMUXER
653 static int adts_aac_probe(AVProbeData *p)
654 {
655     int max_frames = 0, first_frames = 0;
656     int fsize, frames;
657     uint8_t *buf0 = p->buf;
658     uint8_t *buf2;
659     uint8_t *buf;
660     uint8_t *end = buf0 + p->buf_size - 7;
661
662     if (ff_id3v2_match(buf0)) {
663         buf0 += ff_id3v2_tag_len(buf0);
664     }
665     buf = buf0;
666
667     for(; buf < end; buf= buf2+1) {
668         buf2 = buf;
669
670         for(frames = 0; buf2 < end; frames++) {
671             uint32_t header = AV_RB16(buf2);
672             if((header&0xFFF6) != 0xFFF0)
673                 break;
674             fsize = (AV_RB32(buf2+3)>>13) & 0x8FFF;
675             if(fsize < 7)
676                 break;
677             buf2 += fsize;
678         }
679         max_frames = FFMAX(max_frames, frames);
680         if(buf == buf0)
681             first_frames= frames;
682     }
683     if   (first_frames>=3) return AVPROBE_SCORE_MAX/2+1;
684     else if(max_frames>500)return AVPROBE_SCORE_MAX/2;
685     else if(max_frames>=3) return AVPROBE_SCORE_MAX/4;
686     else if(max_frames>=1) return 1;
687     else                   return 0;
688 }
689
690 static int adts_aac_read_header(AVFormatContext *s,
691                                 AVFormatParameters *ap)
692 {
693     AVStream *st;
694
695     st = av_new_stream(s, 0);
696     if (!st)
697         return AVERROR(ENOMEM);
698
699     st->codec->codec_type = CODEC_TYPE_AUDIO;
700     st->codec->codec_id = s->iformat->value;
701     st->need_parsing = AVSTREAM_PARSE_FULL;
702
703     ff_id3v1_read(s);
704     ff_id3v2_read(s);
705
706     return 0;
707 }
708
709 #endif
710
711 /* Note: Do not forget to add new entries to the Makefile as well. */
712
713 #if CONFIG_AAC_DEMUXER
714 AVInputFormat aac_demuxer = {
715     "aac",
716     NULL_IF_CONFIG_SMALL("raw ADTS AAC"),
717     0,
718     adts_aac_probe,
719     adts_aac_read_header,
720     ff_raw_read_partial_packet,
721     .flags= AVFMT_GENERIC_INDEX,
722     .extensions = "aac",
723     .value = CODEC_ID_AAC,
724 };
725 #endif
726
727 #if CONFIG_AC3_DEMUXER
728 AVInputFormat ac3_demuxer = {
729     "ac3",
730     NULL_IF_CONFIG_SMALL("raw AC-3"),
731     0,
732     ac3_probe,
733     audio_read_header,
734     ff_raw_read_partial_packet,
735     .flags= AVFMT_GENERIC_INDEX,
736     .extensions = "ac3",
737     .value = CODEC_ID_AC3,
738 };
739 #endif
740
741 #if CONFIG_AC3_MUXER
742 AVOutputFormat ac3_muxer = {
743     "ac3",
744     NULL_IF_CONFIG_SMALL("raw AC-3"),
745     "audio/x-ac3",
746     "ac3",
747     0,
748     CODEC_ID_AC3,
749     CODEC_ID_NONE,
750     NULL,
751     raw_write_packet,
752     .flags= AVFMT_NOTIMESTAMPS,
753 };
754 #endif
755
756 #if CONFIG_DIRAC_DEMUXER
757 AVInputFormat dirac_demuxer = {
758     "dirac",
759     NULL_IF_CONFIG_SMALL("raw Dirac"),
760     0,
761     dirac_probe,
762     video_read_header,
763     ff_raw_read_partial_packet,
764     .flags= AVFMT_GENERIC_INDEX,
765     .value = CODEC_ID_DIRAC,
766 };
767 #endif
768
769 #if CONFIG_DIRAC_MUXER
770 AVOutputFormat dirac_muxer = {
771     "dirac",
772     NULL_IF_CONFIG_SMALL("raw Dirac"),
773     NULL,
774     "drc",
775     0,
776     CODEC_ID_NONE,
777     CODEC_ID_DIRAC,
778     NULL,
779     raw_write_packet,
780     .flags= AVFMT_NOTIMESTAMPS,
781 };
782 #endif
783
784 #if CONFIG_DNXHD_DEMUXER
785 AVInputFormat dnxhd_demuxer = {
786     "dnxhd",
787     NULL_IF_CONFIG_SMALL("raw DNxHD (SMPTE VC-3)"),
788     0,
789     dnxhd_probe,
790     video_read_header,
791     ff_raw_read_partial_packet,
792     .flags= AVFMT_GENERIC_INDEX,
793     .value = CODEC_ID_DNXHD,
794 };
795 #endif
796
797 #if CONFIG_DNXHD_MUXER
798 AVOutputFormat dnxhd_muxer = {
799     "dnxhd",
800     NULL_IF_CONFIG_SMALL("raw DNxHD (SMPTE VC-3)"),
801     NULL,
802     "dnxhd",
803     0,
804     CODEC_ID_NONE,
805     CODEC_ID_DNXHD,
806     NULL,
807     raw_write_packet,
808     .flags= AVFMT_NOTIMESTAMPS,
809 };
810 #endif
811
812 #if CONFIG_DTS_DEMUXER
813 AVInputFormat dts_demuxer = {
814     "dts",
815     NULL_IF_CONFIG_SMALL("raw DTS"),
816     0,
817     dts_probe,
818     audio_read_header,
819     ff_raw_read_partial_packet,
820     .flags= AVFMT_GENERIC_INDEX,
821     .extensions = "dts",
822     .value = CODEC_ID_DTS,
823 };
824 #endif
825
826 #if CONFIG_DTS_MUXER
827 AVOutputFormat dts_muxer = {
828     "dts",
829     NULL_IF_CONFIG_SMALL("raw DTS"),
830     "audio/x-dca",
831     "dts",
832     0,
833     CODEC_ID_DTS,
834     CODEC_ID_NONE,
835     NULL,
836     raw_write_packet,
837     .flags= AVFMT_NOTIMESTAMPS,
838 };
839 #endif
840
841 #if CONFIG_EAC3_DEMUXER
842 AVInputFormat eac3_demuxer = {
843     "eac3",
844     NULL_IF_CONFIG_SMALL("raw E-AC-3"),
845     0,
846     eac3_probe,
847     audio_read_header,
848     ff_raw_read_partial_packet,
849     .flags= AVFMT_GENERIC_INDEX,
850     .extensions = "eac3",
851     .value = CODEC_ID_EAC3,
852 };
853 #endif
854
855 #if CONFIG_EAC3_MUXER
856 AVOutputFormat eac3_muxer = {
857     "eac3",
858     NULL_IF_CONFIG_SMALL("raw E-AC-3"),
859     "audio/x-eac3",
860     "eac3",
861     0,
862     CODEC_ID_EAC3,
863     CODEC_ID_NONE,
864     NULL,
865     raw_write_packet,
866     .flags= AVFMT_NOTIMESTAMPS,
867 };
868 #endif
869
870 #if CONFIG_GSM_DEMUXER
871 AVInputFormat gsm_demuxer = {
872     "gsm",
873     NULL_IF_CONFIG_SMALL("raw GSM"),
874     0,
875     NULL,
876     audio_read_header,
877     ff_raw_read_partial_packet,
878     .flags= AVFMT_GENERIC_INDEX,
879     .extensions = "gsm",
880     .value = CODEC_ID_GSM,
881 };
882 #endif
883
884 #if CONFIG_H261_DEMUXER
885 AVInputFormat h261_demuxer = {
886     "h261",
887     NULL_IF_CONFIG_SMALL("raw H.261"),
888     0,
889     h261_probe,
890     video_read_header,
891     ff_raw_read_partial_packet,
892     .flags= AVFMT_GENERIC_INDEX,
893     .extensions = "h261",
894     .value = CODEC_ID_H261,
895 };
896 #endif
897
898 #if CONFIG_H261_MUXER
899 AVOutputFormat h261_muxer = {
900     "h261",
901     NULL_IF_CONFIG_SMALL("raw H.261"),
902     "video/x-h261",
903     "h261",
904     0,
905     CODEC_ID_NONE,
906     CODEC_ID_H261,
907     NULL,
908     raw_write_packet,
909     .flags= AVFMT_NOTIMESTAMPS,
910 };
911 #endif
912
913 #if CONFIG_H263_DEMUXER
914 AVInputFormat h263_demuxer = {
915     "h263",
916     NULL_IF_CONFIG_SMALL("raw H.263"),
917     0,
918     h263_probe,
919     video_read_header,
920     ff_raw_read_partial_packet,
921     .flags= AVFMT_GENERIC_INDEX,
922 //    .extensions = "h263", //FIXME remove after writing mpeg4_probe
923     .value = CODEC_ID_H263,
924 };
925 #endif
926
927 #if CONFIG_H263_MUXER
928 AVOutputFormat h263_muxer = {
929     "h263",
930     NULL_IF_CONFIG_SMALL("raw H.263"),
931     "video/x-h263",
932     "h263",
933     0,
934     CODEC_ID_NONE,
935     CODEC_ID_H263,
936     NULL,
937     raw_write_packet,
938     .flags= AVFMT_NOTIMESTAMPS,
939 };
940 #endif
941
942 #if CONFIG_H264_DEMUXER
943 AVInputFormat h264_demuxer = {
944     "h264",
945     NULL_IF_CONFIG_SMALL("raw H.264 video format"),
946     0,
947     h264_probe,
948     video_read_header,
949     ff_raw_read_partial_packet,
950     .flags= AVFMT_GENERIC_INDEX,
951     .extensions = "h26l,h264,264", //FIXME remove after writing mpeg4_probe
952     .value = CODEC_ID_H264,
953 };
954 #endif
955
956 #if CONFIG_H264_MUXER
957 AVOutputFormat h264_muxer = {
958     "h264",
959     NULL_IF_CONFIG_SMALL("raw H.264 video format"),
960     NULL,
961     "h264",
962     0,
963     CODEC_ID_NONE,
964     CODEC_ID_H264,
965     NULL,
966     raw_write_packet,
967     .flags= AVFMT_NOTIMESTAMPS,
968 };
969 #endif
970
971 #if CONFIG_INGENIENT_DEMUXER
972 AVInputFormat ingenient_demuxer = {
973     "ingenient",
974     NULL_IF_CONFIG_SMALL("raw Ingenient MJPEG"),
975     0,
976     NULL,
977     video_read_header,
978     ingenient_read_packet,
979     .flags= AVFMT_GENERIC_INDEX,
980     .extensions = "cgi", // FIXME
981     .value = CODEC_ID_MJPEG,
982 };
983 #endif
984
985 #if CONFIG_M4V_DEMUXER
986 AVInputFormat m4v_demuxer = {
987     "m4v",
988     NULL_IF_CONFIG_SMALL("raw MPEG-4 video format"),
989     0,
990     mpeg4video_probe, /** probing for MPEG-4 data */
991     video_read_header,
992     ff_raw_read_partial_packet,
993     .flags= AVFMT_GENERIC_INDEX,
994     .extensions = "m4v",
995     .value = CODEC_ID_MPEG4,
996 };
997 #endif
998
999 #if CONFIG_M4V_MUXER
1000 AVOutputFormat m4v_muxer = {
1001     "m4v",
1002     NULL_IF_CONFIG_SMALL("raw MPEG-4 video format"),
1003     NULL,
1004     "m4v",
1005     0,
1006     CODEC_ID_NONE,
1007     CODEC_ID_MPEG4,
1008     NULL,
1009     raw_write_packet,
1010     .flags= AVFMT_NOTIMESTAMPS,
1011 };
1012 #endif
1013
1014 #if CONFIG_MJPEG_DEMUXER
1015 AVInputFormat mjpeg_demuxer = {
1016     "mjpeg",
1017     NULL_IF_CONFIG_SMALL("raw MJPEG video"),
1018     0,
1019     NULL,
1020     video_read_header,
1021     ff_raw_read_partial_packet,
1022     .flags= AVFMT_GENERIC_INDEX,
1023     .extensions = "mjpg,mjpeg",
1024     .value = CODEC_ID_MJPEG,
1025 };
1026 #endif
1027
1028 #if CONFIG_MJPEG_MUXER
1029 AVOutputFormat mjpeg_muxer = {
1030     "mjpeg",
1031     NULL_IF_CONFIG_SMALL("raw MJPEG video"),
1032     "video/x-mjpeg",
1033     "mjpg,mjpeg",
1034     0,
1035     CODEC_ID_NONE,
1036     CODEC_ID_MJPEG,
1037     NULL,
1038     raw_write_packet,
1039     .flags= AVFMT_NOTIMESTAMPS,
1040 };
1041 #endif
1042
1043 #if CONFIG_MLP_DEMUXER
1044 AVInputFormat mlp_demuxer = {
1045     "mlp",
1046     NULL_IF_CONFIG_SMALL("raw MLP"),
1047     0,
1048     NULL,
1049     audio_read_header,
1050     ff_raw_read_partial_packet,
1051     .flags= AVFMT_GENERIC_INDEX,
1052     .extensions = "mlp",
1053     .value = CODEC_ID_MLP,
1054 };
1055 #endif
1056
1057 #if CONFIG_MLP_MUXER
1058 AVOutputFormat mlp_muxer = {
1059     "mlp",
1060     NULL_IF_CONFIG_SMALL("raw MLP"),
1061     NULL,
1062     "mlp",
1063     0,
1064     CODEC_ID_MLP,
1065     CODEC_ID_NONE,
1066     NULL,
1067     raw_write_packet,
1068     .flags= AVFMT_NOTIMESTAMPS,
1069 };
1070 #endif
1071
1072 #if CONFIG_TRUEHD_DEMUXER
1073 AVInputFormat truehd_demuxer = {
1074     "truehd",
1075     NULL_IF_CONFIG_SMALL("raw TrueHD"),
1076     0,
1077     NULL,
1078     audio_read_header,
1079     ff_raw_read_partial_packet,
1080     .flags= AVFMT_GENERIC_INDEX,
1081     .extensions = "thd",
1082     .value = CODEC_ID_TRUEHD,
1083 };
1084 #endif
1085
1086 #if CONFIG_TRUEHD_MUXER
1087 AVOutputFormat truehd_muxer = {
1088     "truehd",
1089     NULL_IF_CONFIG_SMALL("raw TrueHD"),
1090     NULL,
1091     "thd",
1092     0,
1093     CODEC_ID_TRUEHD,
1094     CODEC_ID_NONE,
1095     NULL,
1096     raw_write_packet,
1097     .flags= AVFMT_NOTIMESTAMPS,
1098 };
1099 #endif
1100
1101 #if CONFIG_MPEG1VIDEO_MUXER
1102 AVOutputFormat mpeg1video_muxer = {
1103     "mpeg1video",
1104     NULL_IF_CONFIG_SMALL("raw MPEG-1 video"),
1105     "video/x-mpeg",
1106     "mpg,mpeg,m1v",
1107     0,
1108     CODEC_ID_NONE,
1109     CODEC_ID_MPEG1VIDEO,
1110     NULL,
1111     raw_write_packet,
1112     .flags= AVFMT_NOTIMESTAMPS,
1113 };
1114 #endif
1115
1116 #if CONFIG_MPEG2VIDEO_MUXER
1117 AVOutputFormat mpeg2video_muxer = {
1118     "mpeg2video",
1119     NULL_IF_CONFIG_SMALL("raw MPEG-2 video"),
1120     NULL,
1121     "m2v",
1122     0,
1123     CODEC_ID_NONE,
1124     CODEC_ID_MPEG2VIDEO,
1125     NULL,
1126     raw_write_packet,
1127     .flags= AVFMT_NOTIMESTAMPS,
1128 };
1129 #endif
1130
1131 #if CONFIG_MPEGVIDEO_DEMUXER
1132 AVInputFormat mpegvideo_demuxer = {
1133     "mpegvideo",
1134     NULL_IF_CONFIG_SMALL("raw MPEG video"),
1135     0,
1136     mpegvideo_probe,
1137     video_read_header,
1138     ff_raw_read_partial_packet,
1139     .flags= AVFMT_GENERIC_INDEX,
1140     .value = CODEC_ID_MPEG1VIDEO,
1141 };
1142 #endif
1143
1144 #if CONFIG_CAVSVIDEO_DEMUXER
1145 AVInputFormat cavsvideo_demuxer = {
1146     "cavsvideo",
1147     NULL_IF_CONFIG_SMALL("raw Chinese AVS video"),
1148     0,
1149     cavsvideo_probe,
1150     video_read_header,
1151     ff_raw_read_partial_packet,
1152     .flags= AVFMT_GENERIC_INDEX,
1153     .value = CODEC_ID_CAVS,
1154 };
1155 #endif
1156
1157 #if CONFIG_NULL_MUXER
1158 AVOutputFormat null_muxer = {
1159     "null",
1160     NULL_IF_CONFIG_SMALL("raw null video format"),
1161     NULL,
1162     NULL,
1163     0,
1164 #if HAVE_BIGENDIAN
1165     CODEC_ID_PCM_S16BE,
1166 #else
1167     CODEC_ID_PCM_S16LE,
1168 #endif
1169     CODEC_ID_RAWVIDEO,
1170     NULL,
1171     null_write_packet,
1172     .flags = AVFMT_NOFILE | AVFMT_RAWPICTURE | AVFMT_NOTIMESTAMPS,
1173 };
1174 #endif
1175
1176 #if CONFIG_RAWVIDEO_DEMUXER
1177 AVInputFormat rawvideo_demuxer = {
1178     "rawvideo",
1179     NULL_IF_CONFIG_SMALL("raw video format"),
1180     0,
1181     NULL,
1182     raw_read_header,
1183     rawvideo_read_packet,
1184     .flags= AVFMT_GENERIC_INDEX,
1185     .extensions = "yuv,cif,qcif,rgb",
1186     .value = CODEC_ID_RAWVIDEO,
1187 };
1188 #endif
1189
1190 #if CONFIG_RAWVIDEO_MUXER
1191 AVOutputFormat rawvideo_muxer = {
1192     "rawvideo",
1193     NULL_IF_CONFIG_SMALL("raw video format"),
1194     NULL,
1195     "yuv,rgb",
1196     0,
1197     CODEC_ID_NONE,
1198     CODEC_ID_RAWVIDEO,
1199     NULL,
1200     raw_write_packet,
1201     .flags= AVFMT_NOTIMESTAMPS,
1202 };
1203 #endif
1204
1205 #if CONFIG_ROQ_MUXER
1206 AVOutputFormat roq_muxer =
1207 {
1208     "RoQ",
1209     NULL_IF_CONFIG_SMALL("raw id RoQ format"),
1210     NULL,
1211     "roq",
1212     0,
1213     CODEC_ID_ROQ_DPCM,
1214     CODEC_ID_ROQ,
1215     roq_write_header,
1216     raw_write_packet,
1217 };
1218 #endif
1219
1220 #if CONFIG_SHORTEN_DEMUXER
1221 AVInputFormat shorten_demuxer = {
1222     "shn",
1223     NULL_IF_CONFIG_SMALL("raw Shorten"),
1224     0,
1225     NULL,
1226     audio_read_header,
1227     ff_raw_read_partial_packet,
1228     .flags= AVFMT_GENERIC_INDEX,
1229     .extensions = "shn",
1230     .value = CODEC_ID_SHORTEN,
1231 };
1232 #endif
1233
1234 #if CONFIG_VC1_DEMUXER
1235 AVInputFormat vc1_demuxer = {
1236     "vc1",
1237     NULL_IF_CONFIG_SMALL("raw VC-1"),
1238     0,
1239     NULL /* vc1_probe */,
1240     video_read_header,
1241     ff_raw_read_partial_packet,
1242     .extensions = "vc1",
1243     .value = CODEC_ID_VC1,
1244 };
1245 #endif
1246
1247 /* PCM formats */
1248
1249 #define PCMINPUTDEF(name, long_name, ext, codec) \
1250 AVInputFormat pcm_ ## name ## _demuxer = {\
1251     #name,\
1252     NULL_IF_CONFIG_SMALL(long_name),\
1253     0,\
1254     NULL,\
1255     raw_read_header,\
1256     raw_read_packet,\
1257     NULL,\
1258     pcm_read_seek,\
1259     .flags= AVFMT_GENERIC_INDEX,\
1260     .extensions = ext,\
1261     .value = codec,\
1262 };
1263
1264 #define PCMOUTPUTDEF(name, long_name, ext, codec) \
1265 AVOutputFormat pcm_ ## name ## _muxer = {\
1266     #name,\
1267     NULL_IF_CONFIG_SMALL(long_name),\
1268     NULL,\
1269     ext,\
1270     0,\
1271     codec,\
1272     CODEC_ID_NONE,\
1273     NULL,\
1274     raw_write_packet,\
1275     .flags= AVFMT_NOTIMESTAMPS,\
1276 };
1277
1278
1279 #if  !CONFIG_MUXERS && CONFIG_DEMUXERS
1280 #define PCMDEF(name, long_name, ext, codec) \
1281         PCMINPUTDEF(name, long_name, ext, codec)
1282 #elif CONFIG_MUXERS && !CONFIG_DEMUXERS
1283 #define PCMDEF(name, long_name, ext, codec) \
1284         PCMOUTPUTDEF(name, long_name, ext, codec)
1285 #elif CONFIG_MUXERS && CONFIG_DEMUXERS
1286 #define PCMDEF(name, long_name, ext, codec) \
1287         PCMINPUTDEF(name, long_name, ext, codec)\
1288         PCMOUTPUTDEF(name, long_name, ext, codec)
1289 #else
1290 #define PCMDEF(name, long_name, ext, codec)
1291 #endif
1292
1293 #if HAVE_BIGENDIAN
1294 #define BE_DEF(s) s
1295 #define LE_DEF(s) NULL
1296 #else
1297 #define BE_DEF(s) NULL
1298 #define LE_DEF(s) s
1299 #endif
1300
1301 PCMDEF(f64be, "PCM 64 bit floating-point big-endian format",
1302        NULL, CODEC_ID_PCM_F64BE)
1303
1304 PCMDEF(f64le, "PCM 64 bit floating-point little-endian format",
1305        NULL, CODEC_ID_PCM_F64LE)
1306
1307 PCMDEF(f32be, "PCM 32 bit floating-point big-endian format",
1308        NULL, CODEC_ID_PCM_F32BE)
1309
1310 PCMDEF(f32le, "PCM 32 bit floating-point little-endian format",
1311        NULL, CODEC_ID_PCM_F32LE)
1312
1313 PCMDEF(s32be, "PCM signed 32 bit big-endian format",
1314        NULL, CODEC_ID_PCM_S32BE)
1315
1316 PCMDEF(s32le, "PCM signed 32 bit little-endian format",
1317        NULL, CODEC_ID_PCM_S32LE)
1318
1319 PCMDEF(s24be, "PCM signed 24 bit big-endian format",
1320        NULL, CODEC_ID_PCM_S24BE)
1321
1322 PCMDEF(s24le, "PCM signed 24 bit little-endian format",
1323        NULL, CODEC_ID_PCM_S24LE)
1324
1325 PCMDEF(s16be, "PCM signed 16 bit big-endian format",
1326        BE_DEF("sw"), CODEC_ID_PCM_S16BE)
1327
1328 PCMDEF(s16le, "PCM signed 16 bit little-endian format",
1329        LE_DEF("sw"), CODEC_ID_PCM_S16LE)
1330
1331 PCMDEF(s8, "PCM signed 8 bit format",
1332        "sb", CODEC_ID_PCM_S8)
1333
1334 PCMDEF(u32be, "PCM unsigned 32 bit big-endian format",
1335        NULL, CODEC_ID_PCM_U32BE)
1336
1337 PCMDEF(u32le, "PCM unsigned 32 bit little-endian format",
1338        NULL, CODEC_ID_PCM_U32LE)
1339
1340 PCMDEF(u24be, "PCM unsigned 24 bit big-endian format",
1341        NULL, CODEC_ID_PCM_U24BE)
1342
1343 PCMDEF(u24le, "PCM unsigned 24 bit little-endian format",
1344        NULL, CODEC_ID_PCM_U24LE)
1345
1346 PCMDEF(u16be, "PCM unsigned 16 bit big-endian format",
1347        BE_DEF("uw"), CODEC_ID_PCM_U16BE)
1348
1349 PCMDEF(u16le, "PCM unsigned 16 bit little-endian format",
1350        LE_DEF("uw"), CODEC_ID_PCM_U16LE)
1351
1352 PCMDEF(u8, "PCM unsigned 8 bit format",
1353        "ub", CODEC_ID_PCM_U8)
1354
1355 PCMDEF(alaw, "PCM A-law format",
1356        "al", CODEC_ID_PCM_ALAW)
1357
1358 PCMDEF(mulaw, "PCM mu-law format",
1359        "ul", CODEC_ID_PCM_MULAW)