annotate yuv4mpeg.c @ 1801:9cd35d039830 libavformat

spit an error message in case of invalid chunk
author alex
date Sun, 18 Feb 2007 00:47:13 +0000
parents 0899bfe4105c
children 1a3c9056982a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
1 /*
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
2 * YUV4MPEG format
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
3 * Copyright (c) 2001, 2002, 2003 Fabrice Bellard.
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
4 *
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
5 * This file is part of FFmpeg.
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
6 *
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
7 * FFmpeg is free software; you can redistribute it and/or
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
9 * License as published by the Free Software Foundation; either
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
10 * version 2.1 of the License, or (at your option) any later version.
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
11 *
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
12 * FFmpeg is distributed in the hope that it will be useful,
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
15 * Lesser General Public License for more details.
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
16 *
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
17 * You should have received a copy of the GNU Lesser General Public
1358
0899bfe4105c Change license headers to say 'FFmpeg' instead of 'this program/this library'
diego
parents: 1169
diff changeset
18 * License along with FFmpeg; if not, write to the Free Software
896
edbe5c3717f9 Update licensing information: The FSF changed postal address.
diego
parents: 887
diff changeset
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
20 */
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
21 #include "avformat.h"
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
22
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
23 #define Y4M_MAGIC "YUV4MPEG2"
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
24 #define Y4M_FRAME_MAGIC "FRAME"
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
25 #define Y4M_LINE_MAX 256
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
26
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
27 struct frame_attributes {
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
28 int interlaced_frame;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
29 int top_field_first;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
30 };
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
31
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
32 static int yuv4_generate_header(AVFormatContext *s, char* buf)
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
33 {
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
34 AVStream *st;
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
35 int width, height;
288
981b0b3c95dd AVRational
michael
parents: 287
diff changeset
36 int raten, rated, aspectn, aspectd, n;
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
37 char inter;
1123
6992dd78ff68 Add (mostly) const to variable and parameter declaration, where a char* was
diego
parents: 935
diff changeset
38 const char *colorspace = "";
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
39
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
40 st = s->streams[0];
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
41 width = st->codec->width;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
42 height = st->codec->height;
85
25062c9b1f86 per context frame_rate_base, this should finally fix frame_rate related av sync issues
michaelni
parents: 65
diff changeset
43
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
44 av_reduce(&raten, &rated, st->codec->time_base.den, st->codec->time_base.num, (1UL<<31)-1);
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
45
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
46 aspectn = st->codec->sample_aspect_ratio.num;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
47 aspectd = st->codec->sample_aspect_ratio.den;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
48
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
49 if ( aspectn == 0 && aspectd == 1 ) aspectd = 0; // 0:0 means unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
50
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
51 inter = 'p'; /* progressive is the default */
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
52 if (st->codec->coded_frame && st->codec->coded_frame->interlaced_frame) {
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
53 inter = st->codec->coded_frame->top_field_first ? 't' : 'b';
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
54 }
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
55
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
56 switch(st->codec->pix_fmt) {
648
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
57 case PIX_FMT_GRAY8:
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
58 colorspace = " Cmono";
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
59 break;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
60 case PIX_FMT_YUV411P:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
61 colorspace = " C411 XYSCSS=411";
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
62 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
63 case PIX_FMT_YUV420P:
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
64 colorspace = (st->codec->codec_id == CODEC_ID_DVVIDEO)?" C420paldv XYSCSS=420PALDV":" C420mpeg2 XYSCSS=420MPEG2";
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
65 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
66 case PIX_FMT_YUV422P:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
67 colorspace = " C422 XYSCSS=422";
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
68 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
69 case PIX_FMT_YUV444P:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
70 colorspace = " C444 XYSCSS=444";
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
71 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
72 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
73
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
74 /* construct stream header, if this is the first frame */
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
75 n = snprintf(buf, Y4M_LINE_MAX, "%s W%d H%d F%d:%d I%c A%d:%d%s\n",
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
76 Y4M_MAGIC,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
77 width,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
78 height,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
79 raten, rated,
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
80 inter,
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
81 aspectn, aspectd,
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
82 colorspace);
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
83
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
84 return n;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
85 }
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
86
468
60f897e8dd2d pass AVPacket into av_write_frame()
michael
parents: 402
diff changeset
87 static int yuv4_write_packet(AVFormatContext *s, AVPacket *pkt)
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
88 {
468
60f897e8dd2d pass AVPacket into av_write_frame()
michael
parents: 402
diff changeset
89 AVStream *st = s->streams[pkt->stream_index];
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
90 ByteIOContext *pb = &s->pb;
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
91 AVPicture *picture;
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
92 int* first_pkt = s->priv_data;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
93 int width, height, h_chroma_shift, v_chroma_shift;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
94 int i, m;
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
95 char buf2[Y4M_LINE_MAX+1];
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
96 char buf1[20];
65
a58a8a53eb46 * UINTX -> uintx_t INTX -> intx_t
kabi
parents: 18
diff changeset
97 uint8_t *ptr, *ptr1, *ptr2;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
98
468
60f897e8dd2d pass AVPacket into av_write_frame()
michael
parents: 402
diff changeset
99 picture = (AVPicture *)pkt->data;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
100
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
101 /* for the first packet we have to output the header as well */
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
102 if (*first_pkt) {
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
103 *first_pkt = 0;
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
104 if (yuv4_generate_header(s, buf2) < 0) {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
105 av_log(s, AV_LOG_ERROR, "Error. YUV4MPEG stream header write failed.\n");
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
106 return AVERROR_IO;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
107 } else {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
108 put_buffer(pb, buf2, strlen(buf2));
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
109 }
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
110 }
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
111
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
112 /* construct frame header */
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
113
256
2efd6fe95fc6 yuv4mpeg.c extra space patch by ("Steven M. Schultz" <sms at 2BSD dot COM>)
michaelni
parents: 241
diff changeset
114 m = snprintf(buf1, sizeof(buf1), "%s\n", Y4M_FRAME_MAGIC);
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
115 put_buffer(pb, buf1, strlen(buf1));
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
116
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
117 width = st->codec->width;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
118 height = st->codec->height;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
119
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
120 ptr = picture->data[0];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
121 for(i=0;i<height;i++) {
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
122 put_buffer(pb, ptr, width);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
123 ptr += picture->linesize[0];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
124 }
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
125
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
126 if (st->codec->pix_fmt != PIX_FMT_GRAY8){
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
127 // Adjust for smaller Cb and Cr planes
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
128 avcodec_get_chroma_sub_sample(st->codec->pix_fmt, &h_chroma_shift, &v_chroma_shift);
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
129 width >>= h_chroma_shift;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
130 height >>= v_chroma_shift;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
131
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
132 ptr1 = picture->data[1];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
133 ptr2 = picture->data[2];
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
134 for(i=0;i<height;i++) { /* Cb */
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
135 put_buffer(pb, ptr1, width);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
136 ptr1 += picture->linesize[1];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
137 }
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
138 for(i=0;i<height;i++) { /* Cr */
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
139 put_buffer(pb, ptr2, width);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
140 ptr2 += picture->linesize[2];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
141 }
648
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
142 }
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
143 put_flush_packet(pb);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
144 return 0;
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
145 }
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
146
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
147 static int yuv4_write_header(AVFormatContext *s)
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
148 {
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
149 int* first_pkt = s->priv_data;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
150
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
151 if (s->nb_streams != 1)
482
0fdc96c2f2fe sweeping change from -EIO -> AVERROR_IO
melanson
parents: 468
diff changeset
152 return AVERROR_IO;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
153
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
154 if (s->streams[0]->codec->pix_fmt == PIX_FMT_YUV411P) {
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
155 av_log(s, AV_LOG_ERROR, "Warning: generating rarely used 4:1:1 YUV stream, some mjpegtools might not work.\n");
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
156 }
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
157 else if ((s->streams[0]->codec->pix_fmt != PIX_FMT_YUV420P) &&
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
158 (s->streams[0]->codec->pix_fmt != PIX_FMT_YUV422P) &&
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
159 (s->streams[0]->codec->pix_fmt != PIX_FMT_GRAY8) &&
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
160 (s->streams[0]->codec->pix_fmt != PIX_FMT_YUV444P)) {
648
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
161 av_log(s, AV_LOG_ERROR, "ERROR: yuv4mpeg only handles yuv444p, yuv422p, yuv420p, yuv411p and gray pixel formats. Use -pix_fmt to select one.\n");
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
162 return AVERROR_IO;
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
163 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
164
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
165 *first_pkt = 1;
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
166 return 0;
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
167 }
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
168
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
169 static int yuv4_write_trailer(AVFormatContext *s)
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
170 {
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
171 return 0;
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
172 }
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
173
1169
d18cc9a1fd02 allow individual selection of muxers and demuxers
mru
parents: 1167
diff changeset
174 #ifdef CONFIG_YUV4MPEGPIPE_MUXER
1167
d89d7ef290da give AVInput/OutputFormat structs consistent names
mru
parents: 1123
diff changeset
175 AVOutputFormat yuv4mpegpipe_muxer = {
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
176 "yuv4mpegpipe",
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
177 "YUV4MPEG pipe format",
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
178 "",
739
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
179 "y4m",
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
180 sizeof(int),
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
181 CODEC_ID_NONE,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
182 CODEC_ID_RAWVIDEO,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
183 yuv4_write_header,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
184 yuv4_write_packet,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
185 yuv4_write_trailer,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
186 .flags = AVFMT_RAWPICTURE,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
187 };
1169
d18cc9a1fd02 allow individual selection of muxers and demuxers
mru
parents: 1167
diff changeset
188 #endif
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
189
207
7865656658dc stdin patch by (Charles Yates <charles dot yates at pandora dot be>)
michaelni
parents: 184
diff changeset
190 /* Header size increased to allow room for optional flags */
7865656658dc stdin patch by (Charles Yates <charles dot yates at pandora dot be>)
michaelni
parents: 184
diff changeset
191 #define MAX_YUV4_HEADER 80
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
192 #define MAX_FRAME_HEADER 80
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
193
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
194 static int yuv4_read_header(AVFormatContext *s, AVFormatParameters *ap)
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
195 {
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
196 char header[MAX_YUV4_HEADER+10]; // Include headroom for the longest option
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
197 char *tokstart,*tokend,*header_end;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
198 int i;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
199 ByteIOContext *pb = &s->pb;
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
200 int width=-1, height=-1, raten=0, rated=0, aspectn=0, aspectd=0;
738
dad78387544a Update yuv4mpeg to use PIX_FMT_NONE patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 648
diff changeset
201 enum PixelFormat pix_fmt=PIX_FMT_NONE,alt_pix_fmt=PIX_FMT_NONE;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
202 AVStream *st;
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
203 struct frame_attributes *s1 = s->priv_data;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
204
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
205 for (i=0; i<MAX_YUV4_HEADER; i++) {
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
206 header[i] = get_byte(pb);
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
207 if (header[i] == '\n') {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
208 header[i+1] = 0x20; // Add a space after last option. Makes parsing "444" vs "444alpha" easier.
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
209 header[i+2] = 0;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
210 break;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
211 }
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
212 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
213 if (i == MAX_YUV4_HEADER) return -1;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
214 if (strncmp(header, Y4M_MAGIC, strlen(Y4M_MAGIC))) return -1;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
215
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
216 s1->interlaced_frame = 0;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
217 s1->top_field_first = 0;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
218 header_end = &header[i+1]; // Include space
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
219 for(tokstart = &header[strlen(Y4M_MAGIC) + 1]; tokstart < header_end; tokstart++) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
220 if (*tokstart==0x20) continue;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
221 switch (*tokstart++) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
222 case 'W': // Width. Required.
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
223 width = strtol(tokstart, &tokend, 10);
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
224 tokstart=tokend;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
225 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
226 case 'H': // Height. Required.
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
227 height = strtol(tokstart, &tokend, 10);
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
228 tokstart=tokend;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
229 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
230 case 'C': // Color space
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
231 if (strncmp("420jpeg",tokstart,7)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
232 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
233 else if (strncmp("420mpeg2",tokstart,8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
234 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
235 else if (strncmp("420paldv", tokstart, 8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
236 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
237 else if (strncmp("411", tokstart, 3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
238 pix_fmt = PIX_FMT_YUV411P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
239 else if (strncmp("422", tokstart, 3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
240 pix_fmt = PIX_FMT_YUV422P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
241 else if (strncmp("444alpha", tokstart, 8)==0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
242 av_log(s, AV_LOG_ERROR, "Cannot handle 4:4:4:4 YUV4MPEG stream.\n");
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
243 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
244 } else if (strncmp("444", tokstart, 3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
245 pix_fmt = PIX_FMT_YUV444P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
246 else if (strncmp("mono",tokstart, 4)==0) {
648
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
247 pix_fmt = PIX_FMT_GRAY8;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
248 } else {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
249 av_log(s, AV_LOG_ERROR, "YUV4MPEG stream contains an unknown pixel format.\n");
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
250 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
251 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
252 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
253 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
254 case 'I': // Interlace type
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
255 switch (*tokstart++){
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
256 case '?':
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
257 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
258 case 'p':
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
259 s1->interlaced_frame=0;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
260 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
261 case 't':
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
262 s1->interlaced_frame=1;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
263 s1->top_field_first=1;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
264 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
265 case 'b':
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
266 s1->interlaced_frame=1;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
267 s1->top_field_first=0;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
268 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
269 case 'm':
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
270 av_log(s, AV_LOG_ERROR, "YUV4MPEG stream contains mixed interlaced and non-interlaced frames.\n");
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
271 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
272 default:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
273 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
274 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
275 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
276 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
277 case 'F': // Frame rate
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
278 sscanf(tokstart,"%d:%d",&raten,&rated); // 0:0 if unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
279 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
280 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
281 case 'A': // Pixel aspect
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
282 sscanf(tokstart,"%d:%d",&aspectn,&aspectd); // 0:0 if unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
283 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
284 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
285 case 'X': // Vendor extensions
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
286 if (strncmp("YSCSS=",tokstart,6)==0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
287 // Older nonstandard pixel format representation
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
288 tokstart+=6;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
289 if (strncmp("420JPEG",tokstart,7)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
290 alt_pix_fmt=PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
291 else if (strncmp("420MPEG2",tokstart,8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
292 alt_pix_fmt=PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
293 else if (strncmp("420PALDV",tokstart,8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
294 alt_pix_fmt=PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
295 else if (strncmp("411",tokstart,3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
296 alt_pix_fmt=PIX_FMT_YUV411P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
297 else if (strncmp("422",tokstart,3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
298 alt_pix_fmt=PIX_FMT_YUV422P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
299 else if (strncmp("444",tokstart,3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
300 alt_pix_fmt=PIX_FMT_YUV444P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
301 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
302 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
303 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
304 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
305 }
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
306
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
307 if ((width == -1) || (height == -1)) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
308 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
309 return -1;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
310 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
311
738
dad78387544a Update yuv4mpeg to use PIX_FMT_NONE patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 648
diff changeset
312 if (pix_fmt == PIX_FMT_NONE) {
dad78387544a Update yuv4mpeg to use PIX_FMT_NONE patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 648
diff changeset
313 if (alt_pix_fmt == PIX_FMT_NONE)
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
314 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
315 else
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
316 pix_fmt = alt_pix_fmt;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
317 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
318
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
319 if (raten == 0 && rated == 0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
320 // Frame rate unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
321 raten = 25;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
322 rated = 1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
323 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
324
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
325 if (aspectn == 0 && aspectd == 0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
326 // Pixel aspect unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
327 aspectd = 1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
328 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
329
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
330 st = av_new_stream(s, 0);
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
331 st = s->streams[0];
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
332 st->codec->width = width;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
333 st->codec->height = height;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
334 av_reduce(&raten, &rated, raten, rated, (1UL<<31)-1);
743
af4e24d6310c switch to native time bases
michael
parents: 739
diff changeset
335 av_set_pts_info(st, 64, rated, raten);
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
336 st->codec->pix_fmt = pix_fmt;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
337 st->codec->codec_type = CODEC_TYPE_VIDEO;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
338 st->codec->codec_id = CODEC_ID_RAWVIDEO;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
339 st->codec->sample_aspect_ratio= (AVRational){aspectn, aspectd};
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
340
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
341 return 0;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
342 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
343
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
344 static int yuv4_read_packet(AVFormatContext *s, AVPacket *pkt)
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
345 {
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
346 int i;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
347 char header[MAX_FRAME_HEADER+1];
838
128838bc4bb6 kill a bunch of compiler warnings
mru
parents: 820
diff changeset
348 int packet_size, width, height;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
349 AVStream *st = s->streams[0];
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
350 struct frame_attributes *s1 = s->priv_data;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
351
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
352 for (i=0; i<MAX_FRAME_HEADER; i++) {
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
353 header[i] = get_byte(&s->pb);
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
354 if (header[i] == '\n') {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
355 header[i+1] = 0;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
356 break;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
357 }
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
358 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
359 if (i == MAX_FRAME_HEADER) return -1;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
360 if (strncmp(header, Y4M_FRAME_MAGIC, strlen(Y4M_FRAME_MAGIC))) return -1;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
361
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
362 width = st->codec->width;
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
363 height = st->codec->height;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
364
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
365 packet_size = avpicture_get_size(st->codec->pix_fmt, width, height);
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
366 if (packet_size < 0)
537
558a093b04db do not call (av_)abort()
michael
parents: 482
diff changeset
367 return -1;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
368
775
c5077fdab490 AVPacket.pos
michael
parents: 743
diff changeset
369 if (av_get_packet(&s->pb, pkt, packet_size) != packet_size)
482
0fdc96c2f2fe sweeping change from -EIO -> AVERROR_IO
melanson
parents: 468
diff changeset
370 return AVERROR_IO;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
371
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
372 if (s->streams[0]->codec->coded_frame) {
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
373 s->streams[0]->codec->coded_frame->interlaced_frame = s1->interlaced_frame;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
374 s->streams[0]->codec->coded_frame->top_field_first = s1->top_field_first;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
375 }
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
376
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
377 pkt->stream_index = 0;
775
c5077fdab490 AVPacket.pos
michael
parents: 743
diff changeset
378 return 0;
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
379 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
380
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
381 static int yuv4_read_close(AVFormatContext *s)
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
382 {
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
383 return 0;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
384 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
385
739
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
386 static int yuv4_probe(AVProbeData *pd)
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
387 {
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
388 /* check file header */
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
389 if (pd->buf_size <= sizeof(Y4M_MAGIC))
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
390 return 0;
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
391 if (strncmp(pd->buf, Y4M_MAGIC, sizeof(Y4M_MAGIC)-1)==0)
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
392 return AVPROBE_SCORE_MAX;
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
393 else
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
394 return 0;
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
395 }
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
396
1169
d18cc9a1fd02 allow individual selection of muxers and demuxers
mru
parents: 1167
diff changeset
397 #ifdef CONFIG_YUV4MPEGPIPE_DEMUXER
1167
d89d7ef290da give AVInput/OutputFormat structs consistent names
mru
parents: 1123
diff changeset
398 AVInputFormat yuv4mpegpipe_demuxer = {
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
399 "yuv4mpegpipe",
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
400 "YUV4MPEG pipe format",
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
401 sizeof(struct frame_attributes),
739
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
402 yuv4_probe,
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
403 yuv4_read_header,
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
404 yuv4_read_packet,
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
405 yuv4_read_close,
739
db0a5e0f4db5 Adds read probe to y4m, and changes the extension to .y4m patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 738
diff changeset
406 .extensions = "y4m"
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
407 };
1169
d18cc9a1fd02 allow individual selection of muxers and demuxers
mru
parents: 1167
diff changeset
408 #endif