annotate yuv4mpeg.c @ 4772:fb0b1e68356c libavformat

Allow (and parse) incoming server messages (notices) interleaved with TCP data packets or in addition to UDP data packets, over the RTSP/TCP connection. See discussion in [PATCH] rtsp.c: read TCP server notifications/messages" thread on mailinglist.
author rbultje
date Sat, 21 Mar 2009 20:46:36 +0000
parents 77e0c7511d41
children a88b31642727
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
4251
77e0c7511d41 cosmetics: Remove pointless period after copyright statement non-sentences.
diego
parents: 4206
diff changeset
3 * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
18
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
4206
c3102b189cb6 Change semantic of CONFIG_*, HAVE_* and ARCH_*.
aurel
parents: 3759
diff changeset
32 #if CONFIG_YUV4MPEGPIPE_MUXER
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
33 static int yuv4_generate_header(AVFormatContext *s, char* buf)
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
34 {
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
35 AVStream *st;
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
36 int width, height;
288
981b0b3c95dd AVRational
michael
parents: 287
diff changeset
37 int raten, rated, aspectn, aspectd, n;
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
38 char inter;
1123
6992dd78ff68 Add (mostly) const to variable and parameter declaration, where a char* was
diego
parents: 935
diff changeset
39 const char *colorspace = "";
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
40
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
41 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
42 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
43 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
44
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
45 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
46
3759
27537074f2a9 convert every muxer/demuxer to write/read sample_aspect_ratio from/to
aurel
parents: 3559
diff changeset
47 aspectn = st->sample_aspect_ratio.num;
27537074f2a9 convert every muxer/demuxer to write/read sample_aspect_ratio from/to
aurel
parents: 3559
diff changeset
48 aspectd = st->sample_aspect_ratio.den;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
49
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
50 if ( aspectn == 0 && aspectd == 1 ) aspectd = 0; // 0:0 means unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
51
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
52 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
53 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
54 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
55 }
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
56
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
57 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
58 case PIX_FMT_GRAY8:
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
59 colorspace = " Cmono";
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
60 break;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
61 case PIX_FMT_YUV411P:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
62 colorspace = " C411 XYSCSS=411";
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
63 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
64 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
65 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
66 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
67 case PIX_FMT_YUV422P:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
68 colorspace = " C422 XYSCSS=422";
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
69 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
70 case PIX_FMT_YUV444P:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
71 colorspace = " C444 XYSCSS=444";
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
72 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
73 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
74
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
75 /* 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
76 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
77 Y4M_MAGIC,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
78 width,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
79 height,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
80 raten, rated,
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
81 inter,
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
82 aspectn, aspectd,
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
83 colorspace);
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
84
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
85 return n;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
86 }
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
87
468
60f897e8dd2d pass AVPacket into av_write_frame()
michael
parents: 402
diff changeset
88 static int yuv4_write_packet(AVFormatContext *s, AVPacket *pkt)
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
89 {
468
60f897e8dd2d pass AVPacket into av_write_frame()
michael
parents: 402
diff changeset
90 AVStream *st = s->streams[pkt->stream_index];
2771
d52c718e83f9 Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents: 2274
diff changeset
91 ByteIOContext *pb = s->pb;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
92 AVPicture *picture;
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
93 int* first_pkt = s->priv_data;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
94 int width, height, h_chroma_shift, v_chroma_shift;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
95 int i, m;
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
96 char buf2[Y4M_LINE_MAX+1];
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
97 char buf1[20];
65
a58a8a53eb46 * UINTX -> uintx_t INTX -> intx_t
kabi
parents: 18
diff changeset
98 uint8_t *ptr, *ptr1, *ptr2;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
99
468
60f897e8dd2d pass AVPacket into av_write_frame()
michael
parents: 402
diff changeset
100 picture = (AVPicture *)pkt->data;
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
101
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
102 /* 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
103 if (*first_pkt) {
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
104 *first_pkt = 0;
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
105 if (yuv4_generate_header(s, buf2) < 0) {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
106 av_log(s, AV_LOG_ERROR, "Error. YUV4MPEG stream header write failed.\n");
2274
b21c2af60bc9 Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents: 2210
diff changeset
107 return AVERROR(EIO);
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
108 } else {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
109 put_buffer(pb, buf2, strlen(buf2));
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
110 }
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
111 }
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
112
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
113 /* construct frame header */
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
114
256
2efd6fe95fc6 yuv4mpeg.c extra space patch by ("Steven M. Schultz" <sms at 2BSD dot COM>)
michaelni
parents: 241
diff changeset
115 m = snprintf(buf1, sizeof(buf1), "%s\n", Y4M_FRAME_MAGIC);
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
116 put_buffer(pb, buf1, strlen(buf1));
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
117
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
118 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
119 height = st->codec->height;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
120
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
121 ptr = picture->data[0];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
122 for(i=0;i<height;i++) {
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
123 put_buffer(pb, ptr, width);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
124 ptr += picture->linesize[0];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
125 }
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
126
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
127 if (st->codec->pix_fmt != PIX_FMT_GRAY8){
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
128 // 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
129 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
130 width >>= h_chroma_shift;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
131 height >>= v_chroma_shift;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
132
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
133 ptr1 = picture->data[1];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
134 ptr2 = picture->data[2];
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
135 for(i=0;i<height;i++) { /* Cb */
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
136 put_buffer(pb, ptr1, width);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
137 ptr1 += picture->linesize[1];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
138 }
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
139 for(i=0;i<height;i++) { /* Cr */
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
140 put_buffer(pb, ptr2, width);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
141 ptr2 += picture->linesize[2];
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
142 }
648
340f1911cd54 add luma only support to yuv4mpeg patch by (Roine Gustafsson <roine users.sourceforge net>)
michael
parents: 634
diff changeset
143 }
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
144 put_flush_packet(pb);
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
145 return 0;
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
146 }
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
147
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
148 static int yuv4_write_header(AVFormatContext *s)
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
149 {
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
150 int* first_pkt = s->priv_data;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
151
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
152 if (s->nb_streams != 1)
2274
b21c2af60bc9 Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents: 2210
diff changeset
153 return AVERROR(EIO);
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
154
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
155 if (s->streams[0]->codec->pix_fmt == PIX_FMT_YUV411P) {
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
156 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
157 }
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
158 else if ((s->streams[0]->codec->pix_fmt != PIX_FMT_YUV420P) &&
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
159 (s->streams[0]->codec->pix_fmt != PIX_FMT_YUV422P) &&
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
160 (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
161 (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
162 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");
2274
b21c2af60bc9 Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents: 2210
diff changeset
163 return AVERROR(EIO);
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
164 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
165
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
166 *first_pkt = 1;
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
167 return 0;
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
168 }
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
169
1167
d89d7ef290da give AVInput/OutputFormat structs consistent names
mru
parents: 1123
diff changeset
170 AVOutputFormat yuv4mpegpipe_muxer = {
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
171 "yuv4mpegpipe",
3424
7a0230981402 Make long_names in lavf/lavdev optional depending on CONFIG_SMALL.
diego
parents: 3402
diff changeset
172 NULL_IF_CONFIG_SMALL("YUV4MPEG pipe format"),
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
173 "",
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
174 "y4m",
284
c77ce17451a1 * providing MPEG codecs with a generic fields in AVFrame to use.
romansh
parents: 277
diff changeset
175 sizeof(int),
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
176 CODEC_ID_NONE,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
177 CODEC_ID_RAWVIDEO,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
178 yuv4_write_header,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
179 yuv4_write_packet,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
180 .flags = AVFMT_RAWPICTURE,
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
181 };
1169
d18cc9a1fd02 allow individual selection of muxers and demuxers
mru
parents: 1167
diff changeset
182 #endif
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
183
207
7865656658dc stdin patch by (Charles Yates <charles dot yates at pandora dot be>)
michaelni
parents: 184
diff changeset
184 /* 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
185 #define MAX_YUV4_HEADER 80
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
186 #define MAX_FRAME_HEADER 80
18
01e35bb2689a extracted yuv4mpeg from img.c (untested)
bellard
parents:
diff changeset
187
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
188 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
189 {
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
190 char header[MAX_YUV4_HEADER+10]; // Include headroom for the longest option
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
191 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
192 int i;
2771
d52c718e83f9 Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents: 2274
diff changeset
193 ByteIOContext *pb = s->pb;
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
194 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
195 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
196 AVStream *st;
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
197 struct frame_attributes *s1 = s->priv_data;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
198
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
199 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
200 header[i] = get_byte(pb);
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
201 if (header[i] == '\n') {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
202 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
203 header[i+2] = 0;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
204 break;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
205 }
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
206 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
207 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
208 if (strncmp(header, Y4M_MAGIC, strlen(Y4M_MAGIC))) return -1;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
209
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
210 s1->interlaced_frame = 0;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
211 s1->top_field_first = 0;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
212 header_end = &header[i+1]; // Include space
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
213 for(tokstart = &header[strlen(Y4M_MAGIC) + 1]; tokstart < header_end; tokstart++) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
214 if (*tokstart==0x20) continue;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
215 switch (*tokstart++) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
216 case 'W': // Width. Required.
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
217 width = strtol(tokstart, &tokend, 10);
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
218 tokstart=tokend;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
219 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
220 case 'H': // Height. Required.
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
221 height = strtol(tokstart, &tokend, 10);
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
222 tokstart=tokend;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
223 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
224 case 'C': // Color space
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
225 if (strncmp("420jpeg",tokstart,7)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
226 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
227 else if (strncmp("420mpeg2",tokstart,8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
228 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
229 else if (strncmp("420paldv", tokstart, 8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
230 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
231 else if (strncmp("411", tokstart, 3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
232 pix_fmt = PIX_FMT_YUV411P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
233 else if (strncmp("422", tokstart, 3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
234 pix_fmt = PIX_FMT_YUV422P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
235 else if (strncmp("444alpha", tokstart, 8)==0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
236 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
237 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
238 } else if (strncmp("444", tokstart, 3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
239 pix_fmt = PIX_FMT_YUV444P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
240 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
241 pix_fmt = PIX_FMT_GRAY8;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
242 } else {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
243 av_log(s, AV_LOG_ERROR, "YUV4MPEG stream contains an unknown pixel format.\n");
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
244 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
245 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
246 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
247 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
248 case 'I': // Interlace type
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
249 switch (*tokstart++){
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
250 case '?':
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
251 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
252 case 'p':
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
253 s1->interlaced_frame=0;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
254 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
255 case 't':
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
256 s1->interlaced_frame=1;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
257 s1->top_field_first=1;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
258 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
259 case 'b':
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
260 s1->interlaced_frame=1;
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
261 s1->top_field_first=0;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
262 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
263 case 'm':
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
264 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
265 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
266 default:
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
267 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
268 return -1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
269 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
270 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
271 case 'F': // Frame rate
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
272 sscanf(tokstart,"%d:%d",&raten,&rated); // 0:0 if unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
273 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
274 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
275 case 'A': // Pixel aspect
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
276 sscanf(tokstart,"%d:%d",&aspectn,&aspectd); // 0:0 if unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
277 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
278 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
279 case 'X': // Vendor extensions
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
280 if (strncmp("YSCSS=",tokstart,6)==0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
281 // Older nonstandard pixel format representation
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
282 tokstart+=6;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
283 if (strncmp("420JPEG",tokstart,7)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
284 alt_pix_fmt=PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
285 else if (strncmp("420MPEG2",tokstart,8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
286 alt_pix_fmt=PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
287 else if (strncmp("420PALDV",tokstart,8)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
288 alt_pix_fmt=PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
289 else if (strncmp("411",tokstart,3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
290 alt_pix_fmt=PIX_FMT_YUV411P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
291 else if (strncmp("422",tokstart,3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
292 alt_pix_fmt=PIX_FMT_YUV422P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
293 else if (strncmp("444",tokstart,3)==0)
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
294 alt_pix_fmt=PIX_FMT_YUV444P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
295 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
296 while(tokstart<header_end&&*tokstart!=0x20) tokstart++;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
297 break;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
298 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
299 }
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
300
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
301 if ((width == -1) || (height == -1)) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
302 av_log(s, AV_LOG_ERROR, "YUV4MPEG has invalid header.\n");
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
303 return -1;
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
304 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
305
738
dad78387544a Update yuv4mpeg to use PIX_FMT_NONE patch by (Roine Gustafsson <roine users sourceforge net)
michael
parents: 648
diff changeset
306 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
307 if (alt_pix_fmt == PIX_FMT_NONE)
634
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
308 pix_fmt = PIX_FMT_YUV420P;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
309 else
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
310 pix_fmt = alt_pix_fmt;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
311 }
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
312
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
313 if (raten == 0 && rated == 0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
314 // Frame rate unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
315 raten = 25;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
316 rated = 1;
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 (aspectn == 0 && aspectd == 0) {
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
320 // Pixel aspect unknown
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
321 aspectd = 1;
6c5c8feed381 - Writes correct unknown aspect
michael
parents: 537
diff changeset
322 }
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
323
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
324 st = av_new_stream(s, 0);
3386
9513f0d1ce0b Null pointer check / CID26.
michael
parents: 3385
diff changeset
325 if(!st)
3402
c30e9e11baf9 ENOMEM is better, yes ...
michael
parents: 3386
diff changeset
326 return AVERROR(ENOMEM);
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
327 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
328 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
329 av_reduce(&raten, &rated, raten, rated, (1UL<<31)-1);
743
af4e24d6310c switch to native time bases
michael
parents: 739
diff changeset
330 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
331 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
332 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
333 st->codec->codec_id = CODEC_ID_RAWVIDEO;
3759
27537074f2a9 convert every muxer/demuxer to write/read sample_aspect_ratio from/to
aurel
parents: 3559
diff changeset
334 st->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
335
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
336 return 0;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
337 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
338
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
339 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
340 {
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
341 int i;
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
342 char header[MAX_FRAME_HEADER+1];
838
128838bc4bb6 kill a bunch of compiler warnings
mru
parents: 820
diff changeset
343 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
344 AVStream *st = s->streams[0];
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
345 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
346
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
347 for (i=0; i<MAX_FRAME_HEADER; i++) {
2771
d52c718e83f9 Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents: 2274
diff changeset
348 header[i] = get_byte(s->pb);
887
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
349 if (header[i] == '\n') {
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
350 header[i+1] = 0;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
351 break;
d70e50f1495f COSMETICS: tabs --> spaces, some prettyprinting
diego
parents: 885
diff changeset
352 }
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
353 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
354 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
355 if (strncmp(header, Y4M_FRAME_MAGIC, strlen(Y4M_FRAME_MAGIC))) return -1;
885
da1d5db0ce5c COSMETICS: Remove all trailing whitespace.
diego
parents: 858
diff changeset
356
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
357 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
358 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
359
820
feca73904e67 changing AVCodecContext codec -> *codec in AVStream so additions to AVCodecContext dont randomize AVStream and break binary compatibility
michael
parents: 775
diff changeset
360 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
361 if (packet_size < 0)
537
558a093b04db do not call (av_)abort()
michael
parents: 482
diff changeset
362 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
363
2771
d52c718e83f9 Use dynamically allocated ByteIOContext in AVFormatContext
andoma
parents: 2274
diff changeset
364 if (av_get_packet(s->pb, pkt, packet_size) != packet_size)
2274
b21c2af60bc9 Replace all occurrences of AVERROR_IO with AVERROR(EIO).
takis
parents: 2210
diff changeset
365 return AVERROR(EIO);
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
366
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
367 if (s->streams[0]->codec->coded_frame) {
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
368 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
369 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
370 }
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
371
184
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
372 pkt->stream_index = 0;
775
c5077fdab490 AVPacket.pos
michael
parents: 743
diff changeset
373 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
374 }
2438e76dde67 yuv4mpeg pipe reader for libavformat patch by (D Richard Felker III <dalias at aerifal dot cx>)
michaelni
parents: 178
diff changeset
375
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
376 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
377 {
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
378 /* 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
379 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
380 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
381 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
382 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
383 }
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
384
4206
c3102b189cb6 Change semantic of CONFIG_*, HAVE_* and ARCH_*.
aurel
parents: 3759
diff changeset
385 #if CONFIG_YUV4MPEGPIPE_DEMUXER
1167
d89d7ef290da give AVInput/OutputFormat structs consistent names
mru
parents: 1123
diff changeset
386 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
387 "yuv4mpegpipe",
3424
7a0230981402 Make long_names in lavf/lavdev optional depending on CONFIG_SMALL.
diego
parents: 3402
diff changeset
388 NULL_IF_CONFIG_SMALL("YUV4MPEG pipe format"),
934
8973dbae81e8 Correctly set the interlaced_frame and top_field_first fields.
diego
parents: 896
diff changeset
389 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
390 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
391 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
392 yuv4_read_packet,
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
393 .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
394 };
1169
d18cc9a1fd02 allow individual selection of muxers and demuxers
mru
parents: 1167
diff changeset
395 #endif