]> git.sesse.net Git - ffmpeg/blob - libavformat/raw.c
Add CODEC_ID_DNXHD into the 25 fps case.
[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     enum CodecID 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 ret;
125
126     bps= av_get_bits_per_sample(s->streams[0]->codec->codec_id);
127     assert(bps); // if false there IS a bug elsewhere (NOT in this function)
128     pkt->dts=
129     pkt->pts= pkt->pos*8 / (bps * s->streams[0]->codec->channels);
130
131     return ret;
132 }
133
134 int ff_raw_read_partial_packet(AVFormatContext *s, AVPacket *pkt)
135 {
136     int ret, size;
137
138     size = RAW_PACKET_SIZE;
139
140     if (av_new_packet(pkt, size) < 0)
141         return AVERROR(ENOMEM);
142
143     pkt->pos= url_ftell(s->pb);
144     pkt->stream_index = 0;
145     ret = get_partial_buffer(s->pb, pkt->data, size);
146     if (ret < 0) {
147         av_free_packet(pkt);
148         return ret;
149     }
150     pkt->size = ret;
151     return ret;
152 }
153 #endif
154
155 #if CONFIG_RAWVIDEO_DEMUXER
156 static int rawvideo_read_packet(AVFormatContext *s, AVPacket *pkt)
157 {
158     int packet_size, ret, width, height;
159     AVStream *st = s->streams[0];
160
161     width = st->codec->width;
162     height = st->codec->height;
163
164     packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
165     if (packet_size < 0)
166         return -1;
167
168     ret= av_get_packet(s->pb, pkt, packet_size);
169     pkt->pts=
170     pkt->dts= pkt->pos / packet_size;
171
172     pkt->stream_index = 0;
173     if (ret < 0)
174         return ret;
175     return 0;
176 }
177 #endif
178
179 #if CONFIG_INGENIENT_DEMUXER
180 // http://www.artificis.hu/files/texts/ingenient.txt
181 static int ingenient_read_packet(AVFormatContext *s, AVPacket *pkt)
182 {
183     int ret, size, w, h, unk1, unk2;
184
185     if (get_le32(s->pb) != MKTAG('M', 'J', 'P', 'G'))
186         return AVERROR(EIO); // FIXME
187
188     size = get_le32(s->pb);
189
190     w = get_le16(s->pb);
191     h = get_le16(s->pb);
192
193     url_fskip(s->pb, 8); // zero + size (padded?)
194     url_fskip(s->pb, 2);
195     unk1 = get_le16(s->pb);
196     unk2 = get_le16(s->pb);
197     url_fskip(s->pb, 22); // ASCII timestamp
198
199     av_log(s, AV_LOG_DEBUG, "Ingenient packet: size=%d, width=%d, height=%d, unk1=%d unk2=%d\n",
200         size, w, h, unk1, unk2);
201
202     if (av_new_packet(pkt, size) < 0)
203         return AVERROR(ENOMEM);
204
205     pkt->pos = url_ftell(s->pb);
206     pkt->stream_index = 0;
207     ret = get_buffer(s->pb, pkt->data, size);
208     if (ret < 0) {
209         av_free_packet(pkt);
210         return ret;
211     }
212     pkt->size = ret;
213     return ret;
214 }
215 #endif
216
217 #if CONFIG_DEMUXERS
218 int pcm_read_seek(AVFormatContext *s,
219                   int stream_index, int64_t timestamp, int flags)
220 {
221     AVStream *st;
222     int block_align, byte_rate;
223     int64_t pos, ret;
224
225     st = s->streams[0];
226
227     block_align = st->codec->block_align ? st->codec->block_align :
228         (av_get_bits_per_sample(st->codec->codec_id) * st->codec->channels) >> 3;
229     byte_rate = st->codec->bit_rate ? st->codec->bit_rate >> 3 :
230         block_align * st->codec->sample_rate;
231
232     if (block_align <= 0 || byte_rate <= 0)
233         return -1;
234     if (timestamp < 0) timestamp = 0;
235
236     /* compute the position by aligning it to block_align */
237     pos = av_rescale_rnd(timestamp * byte_rate,
238                          st->time_base.num,
239                          st->time_base.den * (int64_t)block_align,
240                          (flags & AVSEEK_FLAG_BACKWARD) ? AV_ROUND_DOWN : AV_ROUND_UP);
241     pos *= block_align;
242
243     /* recompute exact position */
244     st->cur_dts = av_rescale(pos, st->time_base.den, byte_rate * (int64_t)st->time_base.num);
245     if ((ret = url_fseek(s->pb, pos + s->data_offset, SEEK_SET)) < 0)
246         return ret;
247     return 0;
248 }
249
250 static int audio_read_header(AVFormatContext *s,
251                              AVFormatParameters *ap)
252 {
253     AVStream *st = av_new_stream(s, 0);
254     if (!st)
255         return AVERROR(ENOMEM);
256     st->codec->codec_type = CODEC_TYPE_AUDIO;
257     st->codec->codec_id = s->iformat->value;
258     st->need_parsing = AVSTREAM_PARSE_FULL;
259     /* the parameters will be extracted from the compressed bitstream */
260
261     return 0;
262 }
263
264 /* MPEG-1/H.263 input */
265 static int video_read_header(AVFormatContext *s,
266                              AVFormatParameters *ap)
267 {
268     AVStream *st;
269
270     st = av_new_stream(s, 0);
271     if (!st)
272         return AVERROR(ENOMEM);
273
274     st->codec->codec_type = CODEC_TYPE_VIDEO;
275     st->codec->codec_id = s->iformat->value;
276     st->need_parsing = AVSTREAM_PARSE_FULL;
277
278     /* for MJPEG, specify frame rate */
279     /* for MPEG-4 specify it, too (most MPEG-4 streams do not have the fixed_vop_rate set ...)*/
280     if (ap->time_base.num) {
281         st->codec->time_base= ap->time_base;
282     } else if ( st->codec->codec_id == CODEC_ID_MJPEG ||
283                 st->codec->codec_id == CODEC_ID_MPEG4 ||
284                 st->codec->codec_id == CODEC_ID_DIRAC ||
285                 st->codec->codec_id == CODEC_ID_DNXHD ||
286                 st->codec->codec_id == CODEC_ID_H264) {
287         st->codec->time_base= (AVRational){1,25};
288     }
289     av_set_pts_info(st, 64, 1, 1200000);
290
291     return 0;
292 }
293 #endif
294
295 #if CONFIG_MPEGVIDEO_DEMUXER
296 #define SEQ_START_CODE          0x000001b3
297 #define GOP_START_CODE          0x000001b8
298 #define PICTURE_START_CODE      0x00000100
299 #define SLICE_START_CODE        0x00000101
300 #define PACK_START_CODE         0x000001ba
301 #define VIDEO_ID                0x000001e0
302 #define AUDIO_ID                0x000001c0
303
304 static int mpegvideo_probe(AVProbeData *p)
305 {
306     uint32_t code= -1;
307     int pic=0, seq=0, slice=0, pspack=0, pes=0;
308     int i;
309
310     for(i=0; i<p->buf_size; i++){
311         code = (code<<8) + p->buf[i];
312         if ((code & 0xffffff00) == 0x100) {
313             switch(code){
314             case     SEQ_START_CODE:   seq++; break;
315             case PICTURE_START_CODE:   pic++; break;
316             case   SLICE_START_CODE: slice++; break;
317             case    PACK_START_CODE: pspack++; break;
318             }
319             if     ((code & 0x1f0) == VIDEO_ID)   pes++;
320             else if((code & 0x1e0) == AUDIO_ID)   pes++;
321         }
322     }
323     if(seq && seq*9<=pic*10 && pic*9<=slice*10 && !pspack && !pes)
324         return pic>1 ? AVPROBE_SCORE_MAX/2+1 : AVPROBE_SCORE_MAX/4; // +1 for .mpg
325     return 0;
326 }
327 #endif
328
329 #if CONFIG_CAVSVIDEO_DEMUXER
330 #define CAVS_SEQ_START_CODE       0x000001b0
331 #define CAVS_PIC_I_START_CODE     0x000001b3
332 #define CAVS_UNDEF_START_CODE     0x000001b4
333 #define CAVS_PIC_PB_START_CODE    0x000001b6
334 #define CAVS_VIDEO_EDIT_CODE      0x000001b7
335 #define CAVS_PROFILE_JIZHUN       0x20
336
337 static int cavsvideo_probe(AVProbeData *p)
338 {
339     uint32_t code= -1;
340     int pic=0, seq=0, slice_pos = 0;
341     int i;
342
343     for(i=0; i<p->buf_size; i++){
344         code = (code<<8) + p->buf[i];
345         if ((code & 0xffffff00) == 0x100) {
346             if(code < CAVS_SEQ_START_CODE) {
347                 /* slices have to be consecutive */
348                 if(code < slice_pos)
349                     return 0;
350                 slice_pos = code;
351             } else {
352                 slice_pos = 0;
353             }
354             if (code == CAVS_SEQ_START_CODE) {
355                 seq++;
356                 /* check for the only currently supported profile */
357                 if(p->buf[i+1] != CAVS_PROFILE_JIZHUN)
358                     return 0;
359             } else if ((code == CAVS_PIC_I_START_CODE) ||
360                        (code == CAVS_PIC_PB_START_CODE)) {
361                 pic++;
362             } else if ((code == CAVS_UNDEF_START_CODE) ||
363                        (code >  CAVS_VIDEO_EDIT_CODE)) {
364                 return 0;
365             }
366         }
367     }
368     if(seq && seq*9<=pic*10)
369         return AVPROBE_SCORE_MAX/2;
370     return 0;
371 }
372 #endif
373
374 #if CONFIG_M4V_DEMUXER
375 #define VISUAL_OBJECT_START_CODE       0x000001b5
376 #define VOP_START_CODE                 0x000001b6
377
378 static int mpeg4video_probe(AVProbeData *probe_packet)
379 {
380     uint32_t temp_buffer= -1;
381     int VO=0, VOL=0, VOP = 0, VISO = 0, res=0;
382     int i;
383
384     for(i=0; i<probe_packet->buf_size; i++){
385         temp_buffer = (temp_buffer<<8) + probe_packet->buf[i];
386         if ((temp_buffer & 0xffffff00) != 0x100)
387             continue;
388
389         if (temp_buffer == VOP_START_CODE)                         VOP++;
390         else if (temp_buffer == VISUAL_OBJECT_START_CODE)          VISO++;
391         else if (temp_buffer < 0x120)                              VO++;
392         else if (temp_buffer < 0x130)                              VOL++;
393         else if (   !(0x1AF < temp_buffer && temp_buffer < 0x1B7)
394                  && !(0x1B9 < temp_buffer && temp_buffer < 0x1C4)) res++;
395     }
396
397     if ( VOP >= VISO && VOP >= VOL && VO >= VOL && VOL > 0 && res==0)
398         return AVPROBE_SCORE_MAX/2;
399     return 0;
400 }
401 #endif
402
403 #if CONFIG_H264_DEMUXER
404 static int h264_probe(AVProbeData *p)
405 {
406     uint32_t code= -1;
407     int sps=0, pps=0, idr=0, res=0, sli=0;
408     int i;
409
410     for(i=0; i<p->buf_size; i++){
411         code = (code<<8) + p->buf[i];
412         if ((code & 0xffffff00) == 0x100) {
413             int ref_idc= (code>>5)&3;
414             int type   = code & 0x1F;
415             static const int8_t ref_zero[32]={
416                 2, 0, 0, 0, 0,-1, 1,-1,
417                -1, 1, 1, 1, 1,-1, 2, 2,
418                 2, 2, 2, 0, 2, 2, 2, 2,
419                 2, 2, 2, 2, 2, 2, 2, 2
420             };
421
422             if(code & 0x80) //forbidden bit
423                 return 0;
424
425             if(ref_zero[type] == 1 && ref_idc)
426                 return 0;
427             if(ref_zero[type] ==-1 && !ref_idc)
428                 return 0;
429             if(ref_zero[type] == 2)
430                 res++;
431
432             switch(type){
433             case     1:   sli++; break;
434             case     5:   idr++; break;
435             case     7:
436                 if(p->buf[i+2]&0x0F)
437                     return 0;
438                 sps++;
439                 break;
440             case     8:   pps++; break;
441             }
442         }
443     }
444     if(sps && pps && (idr||sli>3) && res<(sps+pps+idr))
445         return AVPROBE_SCORE_MAX/2+1; // +1 for .mpg
446     return 0;
447 }
448 #endif
449
450 #if CONFIG_H263_DEMUXER
451 static int h263_probe(AVProbeData *p)
452 {
453     uint64_t code= -1;
454     int i;
455     int valid_psc=0;
456     int invalid_psc=0;
457     int res_change=0;
458     int src_fmt, last_src_fmt=-1;
459
460     for(i=0; i<p->buf_size; i++){
461         code = (code<<8) + p->buf[i];
462         if ((code & 0xfffffc0000) == 0x800000) {
463             src_fmt= (code>>2)&3;
464             if(   src_fmt != last_src_fmt
465                && last_src_fmt>0 && last_src_fmt<6
466                && src_fmt<6)
467                 res_change++;
468
469             if((code&0x300)==0x200 && src_fmt){
470                 valid_psc++;
471             }else
472                 invalid_psc++;
473             last_src_fmt= src_fmt;
474         }
475     }
476 //av_log(NULL, AV_LOG_ERROR, "h263_probe: psc:%d invalid:%d res_change:%d\n", valid_psc, invalid_psc, res_change);
477 //h263_probe: psc:3 invalid:0 res_change:0 (1588/recent_ffmpeg_parses_mpg_incorrectly.mpg)
478     if(valid_psc > 2*invalid_psc + 2*res_change + 3){
479         return 50;
480     }else if(valid_psc > 2*invalid_psc)
481         return 25;
482     return 0;
483 }
484 #endif
485
486 #if CONFIG_H261_DEMUXER
487 static int h261_probe(AVProbeData *p)
488 {
489     uint32_t code= -1;
490     int i;
491     int valid_psc=0;
492     int invalid_psc=0;
493     int next_gn=0;
494     int src_fmt=0;
495     GetBitContext gb;
496
497     init_get_bits(&gb, p->buf, p->buf_size*8);
498
499     for(i=0; i<p->buf_size*8; i++){
500         code = (code<<1) + get_bits1(&gb);
501         if ((code & 0xffff0000) == 0x10000) {
502             int gn= (code>>12)&0xf;
503             if(!gn)
504                 src_fmt= code&8;
505             if(gn != next_gn) invalid_psc++;
506             else              valid_psc++;
507
508             if(src_fmt){ // CIF
509                 next_gn= (gn+1     )%13;
510             }else{       //QCIF
511                 next_gn= (gn+1+!!gn)% 7;
512             }
513         }
514     }
515     if(valid_psc > 2*invalid_psc + 6){
516         return 50;
517     }else if(valid_psc > 2*invalid_psc + 2)
518         return 25;
519     return 0;
520 }
521 #endif
522
523 #if CONFIG_DTS_DEMUXER
524 #define DCA_MARKER_14B_BE 0x1FFFE800
525 #define DCA_MARKER_14B_LE 0xFF1F00E8
526 #define DCA_MARKER_RAW_BE 0x7FFE8001
527 #define DCA_MARKER_RAW_LE 0xFE7F0180
528 static int dts_probe(AVProbeData *p)
529 {
530     const uint8_t *buf, *bufp;
531     uint32_t state = -1;
532     int markers[3] = {0};
533     int sum, max;
534
535     buf = p->buf;
536
537     for(; buf < (p->buf+p->buf_size)-2; buf+=2) {
538         bufp = buf;
539         state = (state << 16) | bytestream_get_be16(&bufp);
540
541         /* regular bitstream */
542         if (state == DCA_MARKER_RAW_BE || state == DCA_MARKER_RAW_LE)
543             markers[0]++;
544
545         /* 14 bits big-endian bitstream */
546         if (state == DCA_MARKER_14B_BE)
547             if ((bytestream_get_be16(&bufp) & 0xFFF0) == 0x07F0)
548                 markers[1]++;
549
550         /* 14 bits little-endian bitstream */
551         if (state == DCA_MARKER_14B_LE)
552             if ((bytestream_get_be16(&bufp) & 0xF0FF) == 0xF007)
553                 markers[2]++;
554     }
555     sum = markers[0] + markers[1] + markers[2];
556     max = markers[1] > markers[0];
557     max = markers[2] > markers[max] ? 2 : max;
558     if (markers[max] > 3 && p->buf_size / markers[max] < 32*1024 &&
559         markers[max] * 4 > sum * 3)
560         return AVPROBE_SCORE_MAX/2+1;
561
562     return 0;
563 }
564 #endif
565
566 #if CONFIG_DIRAC_DEMUXER
567 static int dirac_probe(AVProbeData *p)
568 {
569     if (AV_RL32(p->buf) == MKTAG('B', 'B', 'C', 'D'))
570         return AVPROBE_SCORE_MAX;
571     else
572         return 0;
573 }
574 #endif
575
576 #if CONFIG_DNXHD_DEMUXER
577 static int dnxhd_probe(AVProbeData *p)
578 {
579     static const uint8_t header[] = {0x00,0x00,0x02,0x80,0x01};
580     int w, h, compression_id;
581     if (p->buf_size < 0x2c)
582         return 0;
583     if (memcmp(p->buf, header, 5))
584         return 0;
585     h = AV_RB16(p->buf + 0x18);
586     w = AV_RB16(p->buf + 0x1a);
587     if (!w || !h)
588         return 0;
589     compression_id = AV_RB32(p->buf + 0x28);
590     if (compression_id < 1237 || compression_id > 1253)
591         return 0;
592     return AVPROBE_SCORE_MAX;
593 }
594 #endif
595
596 #if CONFIG_AC3_DEMUXER || CONFIG_EAC3_DEMUXER
597 static int ac3_eac3_probe(AVProbeData *p, enum CodecID expected_codec_id)
598 {
599     int max_frames, first_frames = 0, frames;
600     uint8_t *buf, *buf2, *end;
601     AC3HeaderInfo hdr;
602     GetBitContext gbc;
603     enum CodecID codec_id = CODEC_ID_AC3;
604
605     max_frames = 0;
606     buf = p->buf;
607     end = buf + p->buf_size;
608
609     for(; buf < end; buf++) {
610         buf2 = buf;
611
612         for(frames = 0; buf2 < end; frames++) {
613             init_get_bits(&gbc, buf2, 54);
614             if(ff_ac3_parse_header(&gbc, &hdr) < 0)
615                 break;
616             if(buf2 + hdr.frame_size > end ||
617                av_crc(av_crc_get_table(AV_CRC_16_ANSI), 0, buf2 + 2, hdr.frame_size - 2))
618                 break;
619             if (hdr.bitstream_id > 10)
620                 codec_id = CODEC_ID_EAC3;
621             buf2 += hdr.frame_size;
622         }
623         max_frames = FFMAX(max_frames, frames);
624         if(buf == p->buf)
625             first_frames = frames;
626     }
627     if(codec_id != expected_codec_id) return 0;
628     // keep this in sync with mp3 probe, both need to avoid
629     // issues with MPEG-files!
630     if   (first_frames>=4) return AVPROBE_SCORE_MAX/2+1;
631     else if(max_frames>500)return AVPROBE_SCORE_MAX/2;
632     else if(max_frames>=4) return AVPROBE_SCORE_MAX/4;
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)