OSDN Git Service

additional tweaks to AC3 probe function. give a higher score to a single frame
[coroid/ffmpeg_saccubus.git] / 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 #include "avformat.h"
23 #include "ac3_parser.h"
24
25 #ifdef CONFIG_MUXERS
26 /* simple formats */
27 static int raw_write_header(struct AVFormatContext *s)
28 {
29     return 0;
30 }
31
32 static int flac_write_header(struct AVFormatContext *s)
33 {
34     static const uint8_t header[8] = {
35         0x66, 0x4C, 0x61, 0x43, 0x80, 0x00, 0x00, 0x22
36     };
37     uint8_t *streaminfo = s->streams[0]->codec->extradata;
38     int len = s->streams[0]->codec->extradata_size;
39     if(streaminfo != NULL && len > 0) {
40         put_buffer(&s->pb, header, 8);
41         put_buffer(&s->pb, streaminfo, len);
42     }
43     return 0;
44 }
45
46
47 static int roq_write_header(struct AVFormatContext *s)
48 {
49     static const uint8_t header[] = {
50         0x84, 0x10, 0xFF, 0xFF, 0xFF, 0xFF, 0x1E, 0x00
51     };
52
53     put_buffer(&s->pb, header, 8);
54     put_flush_packet(&s->pb);
55
56     return 0;
57 }
58
59 static int raw_write_packet(struct AVFormatContext *s, AVPacket *pkt)
60 {
61     put_buffer(&s->pb, pkt->data, pkt->size);
62     put_flush_packet(&s->pb);
63     return 0;
64 }
65 #endif //CONFIG_MUXERS
66
67 /* raw input */
68 static int raw_read_header(AVFormatContext *s, AVFormatParameters *ap)
69 {
70     AVStream *st;
71     int id;
72
73     st = av_new_stream(s, 0);
74     if (!st)
75         return AVERROR(ENOMEM);
76
77         id = s->iformat->value;
78         if (id == CODEC_ID_RAWVIDEO) {
79             st->codec->codec_type = CODEC_TYPE_VIDEO;
80         } else {
81             st->codec->codec_type = CODEC_TYPE_AUDIO;
82         }
83         st->codec->codec_id = id;
84
85         switch(st->codec->codec_type) {
86         case CODEC_TYPE_AUDIO:
87             st->codec->sample_rate = ap->sample_rate;
88             st->codec->channels = ap->channels;
89             av_set_pts_info(st, 64, 1, st->codec->sample_rate);
90             break;
91         case CODEC_TYPE_VIDEO:
92             av_set_pts_info(st, 64, ap->time_base.num, ap->time_base.den);
93             st->codec->width = ap->width;
94             st->codec->height = ap->height;
95             st->codec->pix_fmt = ap->pix_fmt;
96             if(st->codec->pix_fmt == PIX_FMT_NONE)
97                 st->codec->pix_fmt= PIX_FMT_YUV420P;
98             break;
99         default:
100             return -1;
101         }
102     return 0;
103 }
104
105 #define RAW_PACKET_SIZE 1024
106
107 static int raw_read_packet(AVFormatContext *s, AVPacket *pkt)
108 {
109     int ret, size;
110     //    AVStream *st = s->streams[0];
111
112     size= RAW_PACKET_SIZE;
113
114     ret= av_get_packet(&s->pb, pkt, size);
115
116     pkt->stream_index = 0;
117     if (ret <= 0) {
118         return AVERROR(EIO);
119     }
120     /* note: we need to modify the packet size here to handle the last
121        packet */
122     pkt->size = ret;
123     return ret;
124 }
125
126 static int raw_read_partial_packet(AVFormatContext *s, AVPacket *pkt)
127 {
128     int ret, size;
129
130     size = RAW_PACKET_SIZE;
131
132     if (av_new_packet(pkt, size) < 0)
133         return AVERROR(EIO);
134
135     pkt->pos= url_ftell(&s->pb);
136     pkt->stream_index = 0;
137     ret = get_partial_buffer(&s->pb, pkt->data, size);
138     if (ret <= 0) {
139         av_free_packet(pkt);
140         return AVERROR(EIO);
141     }
142     pkt->size = ret;
143     return ret;
144 }
145
146 // http://www.artificis.hu/files/texts/ingenient.txt
147 static int ingenient_read_packet(AVFormatContext *s, AVPacket *pkt)
148 {
149     int ret, size, w, h, unk1, unk2;
150
151     if (get_le32(&s->pb) != MKTAG('M', 'J', 'P', 'G'))
152         return AVERROR(EIO); // FIXME
153
154     size = get_le32(&s->pb);
155
156     w = get_le16(&s->pb);
157     h = get_le16(&s->pb);
158
159     url_fskip(&s->pb, 8); // zero + size (padded?)
160     url_fskip(&s->pb, 2);
161     unk1 = get_le16(&s->pb);
162     unk2 = get_le16(&s->pb);
163     url_fskip(&s->pb, 22); // ascii timestamp
164
165     av_log(NULL, AV_LOG_DEBUG, "Ingenient packet: size=%d, width=%d, height=%d, unk1=%d unk2=%d\n",
166         size, w, h, unk1, unk2);
167
168     if (av_new_packet(pkt, size) < 0)
169         return AVERROR(EIO);
170
171     pkt->pos = url_ftell(&s->pb);
172     pkt->stream_index = 0;
173     ret = get_buffer(&s->pb, pkt->data, size);
174     if (ret <= 0) {
175         av_free_packet(pkt);
176         return AVERROR(EIO);
177     }
178     pkt->size = ret;
179     return ret;
180 }
181
182 static int raw_read_close(AVFormatContext *s)
183 {
184     return 0;
185 }
186
187 int pcm_read_seek(AVFormatContext *s,
188                   int stream_index, int64_t timestamp, int flags)
189 {
190     AVStream *st;
191     int block_align, byte_rate;
192     int64_t pos;
193
194     st = s->streams[0];
195
196     block_align = st->codec->block_align ? st->codec->block_align :
197         (av_get_bits_per_sample(st->codec->codec_id) * st->codec->channels) >> 3;
198     byte_rate = st->codec->bit_rate ? st->codec->bit_rate >> 3 :
199         block_align * st->codec->sample_rate;
200
201     if (block_align <= 0 || byte_rate <= 0)
202         return -1;
203
204     /* compute the position by aligning it to block_align */
205     pos = av_rescale_rnd(timestamp * byte_rate,
206                          st->time_base.num,
207                          st->time_base.den * (int64_t)block_align,
208                          (flags & AVSEEK_FLAG_BACKWARD) ? AV_ROUND_DOWN : AV_ROUND_UP);
209     pos *= block_align;
210
211     /* recompute exact position */
212     st->cur_dts = av_rescale(pos, st->time_base.den, byte_rate * (int64_t)st->time_base.num);
213     url_fseek(&s->pb, pos + s->data_offset, SEEK_SET);
214     return 0;
215 }
216
217 /* ac3 read */
218 static int ac3_read_header(AVFormatContext *s,
219                            AVFormatParameters *ap)
220 {
221     AVStream *st;
222
223     st = av_new_stream(s, 0);
224     if (!st)
225         return AVERROR(ENOMEM);
226
227     st->codec->codec_type = CODEC_TYPE_AUDIO;
228     st->codec->codec_id = CODEC_ID_AC3;
229     st->need_parsing = AVSTREAM_PARSE_FULL;
230     /* the parameters will be extracted from the compressed bitstream */
231     return 0;
232 }
233
234 static int shorten_read_header(AVFormatContext *s,
235                                AVFormatParameters *ap)
236 {
237     AVStream *st;
238
239     st = av_new_stream(s, 0);
240     if (!st)
241         return AVERROR(ENOMEM);
242     st->codec->codec_type = CODEC_TYPE_AUDIO;
243     st->codec->codec_id = CODEC_ID_SHORTEN;
244     st->need_parsing = AVSTREAM_PARSE_FULL;
245     /* the parameters will be extracted from the compressed bitstream */
246     return 0;
247 }
248
249 /* flac read */
250 static int flac_read_header(AVFormatContext *s,
251                             AVFormatParameters *ap)
252 {
253     AVStream *st;
254
255     st = av_new_stream(s, 0);
256     if (!st)
257         return AVERROR(ENOMEM);
258     st->codec->codec_type = CODEC_TYPE_AUDIO;
259     st->codec->codec_id = CODEC_ID_FLAC;
260     st->need_parsing = AVSTREAM_PARSE_FULL;
261     /* the parameters will be extracted from the compressed bitstream */
262     return 0;
263 }
264
265 /* dts read */
266 static int dts_read_header(AVFormatContext *s,
267                            AVFormatParameters *ap)
268 {
269     AVStream *st;
270
271     st = av_new_stream(s, 0);
272     if (!st)
273         return AVERROR(ENOMEM);
274
275     st->codec->codec_type = CODEC_TYPE_AUDIO;
276     st->codec->codec_id = CODEC_ID_DTS;
277     st->need_parsing = AVSTREAM_PARSE_FULL;
278     /* the parameters will be extracted from the compressed bitstream */
279     return 0;
280 }
281
282 /* aac read */
283 static int aac_read_header(AVFormatContext *s,
284                            AVFormatParameters *ap)
285 {
286     AVStream *st;
287
288     st = av_new_stream(s, 0);
289     if (!st)
290         return AVERROR(ENOMEM);
291
292     st->codec->codec_type = CODEC_TYPE_AUDIO;
293     st->codec->codec_id = CODEC_ID_AAC;
294     st->need_parsing = AVSTREAM_PARSE_FULL;
295     /* the parameters will be extracted from the compressed bitstream */
296     return 0;
297 }
298
299 /* mpeg1/h263 input */
300 static int video_read_header(AVFormatContext *s,
301                              AVFormatParameters *ap)
302 {
303     AVStream *st;
304
305     st = av_new_stream(s, 0);
306     if (!st)
307         return AVERROR(ENOMEM);
308
309     st->codec->codec_type = CODEC_TYPE_VIDEO;
310     st->codec->codec_id = s->iformat->value;
311     st->need_parsing = AVSTREAM_PARSE_FULL;
312
313     /* for mjpeg, specify frame rate */
314     /* for mpeg4 specify it too (most mpeg4 streams do not have the fixed_vop_rate set ...)*/
315     if (ap->time_base.num) {
316         av_set_pts_info(st, 64, ap->time_base.num, ap->time_base.den);
317     } else if ( st->codec->codec_id == CODEC_ID_MJPEG ||
318                 st->codec->codec_id == CODEC_ID_MPEG4 ||
319                 st->codec->codec_id == CODEC_ID_H264) {
320         av_set_pts_info(st, 64, 1, 25);
321     }
322
323     return 0;
324 }
325
326 #define SEQ_START_CODE          0x000001b3
327 #define GOP_START_CODE          0x000001b8
328 #define PICTURE_START_CODE      0x00000100
329 #define SLICE_START_CODE        0x00000101
330 #define PACK_START_CODE         0x000001ba
331 #define VIDEO_ID                0x000001e0
332 #define AUDIO_ID                0x000001c0
333
334 static int mpegvideo_probe(AVProbeData *p)
335 {
336     uint32_t code= -1;
337     int pic=0, seq=0, slice=0, pspack=0, pes=0;
338     int i;
339
340     for(i=0; i<p->buf_size; i++){
341         code = (code<<8) + p->buf[i];
342         if ((code & 0xffffff00) == 0x100) {
343             switch(code){
344             case     SEQ_START_CODE:   seq++; break;
345             case PICTURE_START_CODE:   pic++; break;
346             case   SLICE_START_CODE: slice++; break;
347             case    PACK_START_CODE: pspack++; break;
348             }
349             if     ((code & 0x1f0) == VIDEO_ID)   pes++;
350             else if((code & 0x1e0) == AUDIO_ID)   pes++;
351         }
352     }
353     if(seq && seq*9<=pic*10 && pic*9<=slice*10 && !pspack && !pes)
354         return AVPROBE_SCORE_MAX/2+1; // +1 for .mpg
355     return 0;
356 }
357
358 #define VISUAL_OBJECT_START_CODE       0x000001b5
359 #define VOP_START_CODE                 0x000001b6
360
361 static int mpeg4video_probe(AVProbeData *probe_packet)
362 {
363     uint32_t temp_buffer= -1;
364     int VO=0, VOL=0, VOP = 0, VISO = 0, res=0;
365     int i;
366
367     for(i=0; i<probe_packet->buf_size; i++){
368         temp_buffer = (temp_buffer<<8) + probe_packet->buf[i];
369         if ((temp_buffer & 0xffffff00) != 0x100)
370             continue;
371
372         if (temp_buffer == VOP_START_CODE)                         VOP++;
373         else if (temp_buffer == VISUAL_OBJECT_START_CODE)          VISO++;
374         else if (temp_buffer < 0x120)                              VO++;
375         else if (temp_buffer < 0x130)                              VOL++;
376         else if (   !(0x1AF < temp_buffer && temp_buffer < 0x1B7)
377                  && !(0x1B9 < temp_buffer && temp_buffer < 0x1C4)) res++;
378     }
379
380     if ( VOP >= VISO && VOP >= VOL && VO >= VOL && VOL > 0 && res==0)
381         return AVPROBE_SCORE_MAX/2;
382     return 0;
383 }
384
385 static int h263_probe(AVProbeData *p)
386 {
387     int code;
388     const uint8_t *d;
389
390     d = p->buf;
391     code = (d[0] << 14) | (d[1] << 6) | (d[2] >> 2);
392     if (code == 0x20) {
393         return 50;
394     }
395     return 0;
396 }
397
398 static int h261_probe(AVProbeData *p)
399 {
400     int code;
401     const uint8_t *d;
402
403     d = p->buf;
404     code = (d[0] << 12) | (d[1] << 4) | (d[2] >> 4);
405     if (code == 0x10) {
406         return 50;
407     }
408     return 0;
409 }
410
411 static int ac3_probe(AVProbeData *p)
412 {
413     int max_frames, first_frames = 0, frames;
414     uint8_t *buf, *buf2, *end;
415     AC3HeaderInfo hdr;
416
417     if(p->buf_size < 7)
418         return 0;
419
420     max_frames = 0;
421     buf = p->buf;
422     end = buf + p->buf_size;
423
424     for(; buf < end; buf++) {
425         buf2 = buf;
426
427         for(frames = 0; buf2 < end; frames++) {
428             if(ff_ac3_parse_header(buf2, &hdr) < 0)
429                 break;
430             buf2 += hdr.frame_size;
431         }
432         max_frames = FFMAX(max_frames, frames);
433         if(buf == p->buf)
434             first_frames = frames;
435     }
436     if   (first_frames>=3) return AVPROBE_SCORE_MAX * 3 / 4;
437     else if(max_frames>=2 || first_frames>=1) return AVPROBE_SCORE_MAX / 2;
438     else if(max_frames>=1) return 1;
439     else                   return 0;
440 }
441
442 AVInputFormat shorten_demuxer = {
443     "shn",
444     "raw shorten",
445     0,
446     NULL,
447     shorten_read_header,
448     raw_read_partial_packet,
449     raw_read_close,
450     .flags= AVFMT_GENERIC_INDEX,
451     .extensions = "shn",
452 };
453
454 AVInputFormat flac_demuxer = {
455     "flac",
456     "raw flac",
457     0,
458     NULL,
459     flac_read_header,
460     raw_read_partial_packet,
461     raw_read_close,
462     .flags= AVFMT_GENERIC_INDEX,
463     .extensions = "flac",
464 };
465
466 #ifdef CONFIG_MUXERS
467 AVOutputFormat flac_muxer = {
468     "flac",
469     "raw flac",
470     "audio/x-flac",
471     "flac",
472     0,
473     CODEC_ID_FLAC,
474     0,
475     flac_write_header,
476     raw_write_packet,
477     .flags= AVFMT_NOTIMESTAMPS,
478 };
479 #endif //CONFIG_MUXERS
480
481 #ifdef CONFIG_AC3_DEMUXER
482 AVInputFormat ac3_demuxer = {
483     "ac3",
484     "raw ac3",
485     0,
486     ac3_probe,
487     ac3_read_header,
488     raw_read_partial_packet,
489     raw_read_close,
490     .flags= AVFMT_GENERIC_INDEX,
491     .extensions = "ac3",
492 };
493 #endif
494
495 #ifdef CONFIG_MUXERS
496 AVOutputFormat ac3_muxer = {
497     "ac3",
498     "raw ac3",
499     "audio/x-ac3",
500     "ac3",
501     0,
502     CODEC_ID_AC3,
503     0,
504     raw_write_header,
505     raw_write_packet,
506     .flags= AVFMT_NOTIMESTAMPS,
507 };
508 #endif //CONFIG_MUXERS
509
510 AVInputFormat dts_demuxer = {
511     "dts",
512     "raw dts",
513     0,
514     NULL,
515     dts_read_header,
516     raw_read_partial_packet,
517     raw_read_close,
518     .flags= AVFMT_GENERIC_INDEX,
519     .extensions = "dts",
520 };
521
522 AVInputFormat aac_demuxer = {
523     "aac",
524     "ADTS AAC",
525     0,
526     NULL,
527     aac_read_header,
528     raw_read_partial_packet,
529     raw_read_close,
530     .flags= AVFMT_GENERIC_INDEX,
531     .extensions = "aac",
532 };
533
534 #ifdef CONFIG_ROQ_MUXER
535 AVOutputFormat roq_muxer =
536 {
537     "RoQ",
538     "Id RoQ format",
539     NULL,
540     "roq",
541     0,
542     CODEC_ID_ROQ_DPCM,
543     CODEC_ID_ROQ,
544     roq_write_header,
545     raw_write_packet,
546 };
547 #endif //CONFIG_ROQ_MUXER
548
549 AVInputFormat h261_demuxer = {
550     "h261",
551     "raw h261",
552     0,
553     h261_probe,
554     video_read_header,
555     raw_read_partial_packet,
556     raw_read_close,
557     .flags= AVFMT_GENERIC_INDEX,
558     .extensions = "h261",
559     .value = CODEC_ID_H261,
560 };
561
562 #ifdef CONFIG_MUXERS
563 AVOutputFormat h261_muxer = {
564     "h261",
565     "raw h261",
566     "video/x-h261",
567     "h261",
568     0,
569     0,
570     CODEC_ID_H261,
571     raw_write_header,
572     raw_write_packet,
573     .flags= AVFMT_NOTIMESTAMPS,
574 };
575 #endif //CONFIG_MUXERS
576
577 AVInputFormat h263_demuxer = {
578     "h263",
579     "raw h263",
580     0,
581     h263_probe,
582     video_read_header,
583     raw_read_partial_packet,
584     raw_read_close,
585     .flags= AVFMT_GENERIC_INDEX,
586 //    .extensions = "h263", //FIXME remove after writing mpeg4_probe
587     .value = CODEC_ID_H263,
588 };
589
590 #ifdef CONFIG_MUXERS
591 AVOutputFormat h263_muxer = {
592     "h263",
593     "raw h263",
594     "video/x-h263",
595     "h263",
596     0,
597     0,
598     CODEC_ID_H263,
599     raw_write_header,
600     raw_write_packet,
601     .flags= AVFMT_NOTIMESTAMPS,
602 };
603 #endif //CONFIG_MUXERS
604
605 AVInputFormat m4v_demuxer = {
606     "m4v",
607     "raw MPEG4 video format",
608     0,
609     mpeg4video_probe, /** probing for mpeg4 data */
610     video_read_header,
611     raw_read_partial_packet,
612     raw_read_close,
613     .flags= AVFMT_GENERIC_INDEX,
614     .extensions = "m4v", //FIXME remove after writing mpeg4_probe
615     .value = CODEC_ID_MPEG4,
616 };
617
618 #ifdef CONFIG_MUXERS
619 AVOutputFormat m4v_muxer = {
620     "m4v",
621     "raw MPEG4 video format",
622     NULL,
623     "m4v",
624     0,
625     CODEC_ID_NONE,
626     CODEC_ID_MPEG4,
627     raw_write_header,
628     raw_write_packet,
629     .flags= AVFMT_NOTIMESTAMPS,
630 };
631 #endif //CONFIG_MUXERS
632
633 AVInputFormat h264_demuxer = {
634     "h264",
635     "raw H264 video format",
636     0,
637     NULL /*mpegvideo_probe*/,
638     video_read_header,
639     raw_read_partial_packet,
640     raw_read_close,
641     .flags= AVFMT_GENERIC_INDEX,
642     .extensions = "h26l,h264,264", //FIXME remove after writing mpeg4_probe
643     .value = CODEC_ID_H264,
644 };
645
646 #ifdef CONFIG_MUXERS
647 AVOutputFormat h264_muxer = {
648     "h264",
649     "raw H264 video format",
650     NULL,
651     "h264",
652     0,
653     CODEC_ID_NONE,
654     CODEC_ID_H264,
655     raw_write_header,
656     raw_write_packet,
657     .flags= AVFMT_NOTIMESTAMPS,
658 };
659 #endif //CONFIG_MUXERS
660
661 AVInputFormat mpegvideo_demuxer = {
662     "mpegvideo",
663     "MPEG video",
664     0,
665     mpegvideo_probe,
666     video_read_header,
667     raw_read_partial_packet,
668     raw_read_close,
669     .flags= AVFMT_GENERIC_INDEX,
670     .value = CODEC_ID_MPEG1VIDEO,
671 };
672
673 #ifdef CONFIG_MUXERS
674 AVOutputFormat mpeg1video_muxer = {
675     "mpeg1video",
676     "MPEG video",
677     "video/x-mpeg",
678     "mpg,mpeg,m1v",
679     0,
680     0,
681     CODEC_ID_MPEG1VIDEO,
682     raw_write_header,
683     raw_write_packet,
684     .flags= AVFMT_NOTIMESTAMPS,
685 };
686 #endif //CONFIG_MUXERS
687
688 #ifdef CONFIG_MUXERS
689 AVOutputFormat mpeg2video_muxer = {
690     "mpeg2video",
691     "MPEG2 video",
692     NULL,
693     "m2v",
694     0,
695     0,
696     CODEC_ID_MPEG2VIDEO,
697     raw_write_header,
698     raw_write_packet,
699     .flags= AVFMT_NOTIMESTAMPS,
700 };
701 #endif //CONFIG_MUXERS
702
703 AVInputFormat mjpeg_demuxer = {
704     "mjpeg",
705     "MJPEG video",
706     0,
707     NULL,
708     video_read_header,
709     raw_read_partial_packet,
710     raw_read_close,
711     .flags= AVFMT_GENERIC_INDEX,
712     .extensions = "mjpg,mjpeg",
713     .value = CODEC_ID_MJPEG,
714 };
715
716 AVInputFormat ingenient_demuxer = {
717     "ingenient",
718     "Ingenient MJPEG",
719     0,
720     NULL,
721     video_read_header,
722     ingenient_read_packet,
723     raw_read_close,
724     .flags= AVFMT_GENERIC_INDEX,
725     .extensions = "cgi", // FIXME
726     .value = CODEC_ID_MJPEG,
727 };
728
729 #ifdef CONFIG_MUXERS
730 AVOutputFormat mjpeg_muxer = {
731     "mjpeg",
732     "MJPEG video",
733     "video/x-mjpeg",
734     "mjpg,mjpeg",
735     0,
736     0,
737     CODEC_ID_MJPEG,
738     raw_write_header,
739     raw_write_packet,
740     .flags= AVFMT_NOTIMESTAMPS,
741 };
742 #endif //CONFIG_MUXERS
743
744 AVInputFormat vc1_demuxer = {
745     "vc1",
746     "raw vc1",
747     0,
748     NULL /* vc1_probe */,
749     video_read_header,
750     raw_read_partial_packet,
751     raw_read_close,
752     .extensions = "vc1",
753     .value = CODEC_ID_VC1,
754 };
755
756 /* pcm formats */
757
758 #define PCMINPUTDEF(name, long_name, ext, codec) \
759 AVInputFormat pcm_ ## name ## _demuxer = {\
760     #name,\
761     long_name,\
762     0,\
763     NULL,\
764     raw_read_header,\
765     raw_read_packet,\
766     raw_read_close,\
767     pcm_read_seek,\
768     .flags= AVFMT_GENERIC_INDEX,\
769     .extensions = ext,\
770     .value = codec,\
771 };
772
773 #define PCMOUTPUTDEF(name, long_name, ext, codec) \
774 AVOutputFormat pcm_ ## name ## _muxer = {\
775     #name,\
776     long_name,\
777     NULL,\
778     ext,\
779     0,\
780     codec,\
781     0,\
782     raw_write_header,\
783     raw_write_packet,\
784     .flags= AVFMT_NOTIMESTAMPS,\
785 };
786
787
788 #if !defined(CONFIG_MUXERS) && defined(CONFIG_DEMUXERS)
789 #define PCMDEF(name, long_name, ext, codec) \
790         PCMINPUTDEF(name, long_name, ext, codec)
791 #elif defined(CONFIG_MUXERS) && !defined(CONFIG_DEMUXERS)
792 #define PCMDEF(name, long_name, ext, codec) \
793         PCMOUTPUTDEF(name, long_name, ext, codec)
794 #elif defined(CONFIG_MUXERS) && defined(CONFIG_DEMUXERS)
795 #define PCMDEF(name, long_name, ext, codec) \
796         PCMINPUTDEF(name, long_name, ext, codec)\
797         PCMOUTPUTDEF(name, long_name, ext, codec)
798 #else
799 #define PCMDEF(name, long_name, ext, codec)
800 #endif
801
802 #ifdef WORDS_BIGENDIAN
803 #define BE_DEF(s) s
804 #define LE_DEF(s) NULL
805 #else
806 #define BE_DEF(s) NULL
807 #define LE_DEF(s) s
808 #endif
809
810
811 PCMDEF(s16le, "pcm signed 16 bit little endian format",
812        LE_DEF("sw"), CODEC_ID_PCM_S16LE)
813
814 PCMDEF(s16be, "pcm signed 16 bit big endian format",
815        BE_DEF("sw"), CODEC_ID_PCM_S16BE)
816
817 PCMDEF(u16le, "pcm unsigned 16 bit little endian format",
818        LE_DEF("uw"), CODEC_ID_PCM_U16LE)
819
820 PCMDEF(u16be, "pcm unsigned 16 bit big endian format",
821        BE_DEF("uw"), CODEC_ID_PCM_U16BE)
822
823 PCMDEF(s8, "pcm signed 8 bit format",
824        "sb", CODEC_ID_PCM_S8)
825
826 PCMDEF(u8, "pcm unsigned 8 bit format",
827        "ub", CODEC_ID_PCM_U8)
828
829 PCMDEF(mulaw, "pcm mu law format",
830        "ul", CODEC_ID_PCM_MULAW)
831
832 PCMDEF(alaw, "pcm A law format",
833        "al", CODEC_ID_PCM_ALAW)
834
835 static int rawvideo_read_packet(AVFormatContext *s, AVPacket *pkt)
836 {
837     int packet_size, ret, width, height;
838     AVStream *st = s->streams[0];
839
840     width = st->codec->width;
841     height = st->codec->height;
842
843     packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
844     if (packet_size < 0)
845         return -1;
846
847     ret= av_get_packet(&s->pb, pkt, packet_size);
848
849     pkt->stream_index = 0;
850     if (ret != packet_size) {
851         return AVERROR(EIO);
852     } else {
853         return 0;
854     }
855 }
856
857 AVInputFormat rawvideo_demuxer = {
858     "rawvideo",
859     "raw video format",
860     0,
861     NULL,
862     raw_read_header,
863     rawvideo_read_packet,
864     raw_read_close,
865     .flags= AVFMT_GENERIC_INDEX,
866     .extensions = "yuv,cif,qcif,rgb",
867     .value = CODEC_ID_RAWVIDEO,
868 };
869
870 #ifdef CONFIG_MUXERS
871 AVOutputFormat rawvideo_muxer = {
872     "rawvideo",
873     "raw video format",
874     NULL,
875     "yuv,rgb",
876     0,
877     CODEC_ID_NONE,
878     CODEC_ID_RAWVIDEO,
879     raw_write_header,
880     raw_write_packet,
881     .flags= AVFMT_NOTIMESTAMPS,
882 };
883 #endif //CONFIG_MUXERS
884
885 #ifdef CONFIG_MUXERS
886 static int null_write_packet(struct AVFormatContext *s, AVPacket *pkt)
887 {
888     return 0;
889 }
890
891 AVOutputFormat null_muxer = {
892     "null",
893     "null video format",
894     NULL,
895     NULL,
896     0,
897 #ifdef WORDS_BIGENDIAN
898     CODEC_ID_PCM_S16BE,
899 #else
900     CODEC_ID_PCM_S16LE,
901 #endif
902     CODEC_ID_RAWVIDEO,
903     raw_write_header,
904     null_write_packet,
905     .flags = AVFMT_NOFILE | AVFMT_RAWPICTURE | AVFMT_NOTIMESTAMPS,
906 };
907 #endif //CONFIG_MUXERS