audio: add af_lavrresample, remove old resampling filters
[mplayer.git] / libmpdemux / demux_y4m.c
blobcf0f7cf7e61600a73acb3cfdb0c25891daf60b6e
1 /*
2 * Y4M file parser
3 * copyright (c) 2001 Rik Snel
4 * (using yuv4mpeg*.[ch] from mjpeg.sourceforge.net)
5 * (derived from demux_viv.c)
6 * older YUV4MPEG (used by xawtv) support by Alex Beregszaszi
8 * This file is part of MPlayer.
10 * MPlayer is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * MPlayer is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License along
21 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
22 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <unistd.h>
28 #include <string.h> /* strtok */
30 #include "config.h"
31 #include "mp_msg.h"
32 #include "yuv4mpeg.h"
34 #include "stream/stream.h"
35 #include "demuxer.h"
36 #include "stheader.h"
38 typedef struct {
39 int framenum;
40 y4m_stream_info_t* si;
41 int is_older;
42 } y4m_priv_t;
44 static int y4m_check_file(demuxer_t* demuxer){
45 int orig_pos = stream_tell(demuxer->stream);
46 char buf[10];
47 y4m_priv_t* priv;
49 mp_msg(MSGT_DEMUX, MSGL_V, "Checking for YUV4MPEG2\n");
51 if(stream_read(demuxer->stream, buf, 9)!=9)
52 return 0;
54 buf[9] = 0;
56 if (strncmp("YUV4MPEG2", buf, 9) && strncmp("YUV4MPEG ", buf, 9)) {
57 return 0;
60 demuxer->priv = malloc(sizeof(y4m_priv_t));
61 priv = demuxer->priv;
63 priv->is_older = 0;
65 if (!strncmp("YUV4MPEG ", buf, 9))
67 mp_msg(MSGT_DEMUX, MSGL_V, "Found older YUV4MPEG format (used by xawtv)\n");
68 priv->is_older = 1;
71 mp_msg(MSGT_DEMUX,MSGL_DBG2,"Success: YUV4MPEG2\n");
73 stream_seek(demuxer->stream, orig_pos);
75 return DEMUXER_TYPE_Y4M;
78 static void read_streaminfo(demuxer_t *demuxer);
80 // return value:
81 // 0 = EOF or no stream found
82 // 1 = successfully read a packet
83 static int demux_y4m_fill_buffer(demuxer_t *demux, demux_stream_t *dsds) {
84 demux_stream_t *ds=demux->video;
85 demux_packet_t *dp;
86 y4m_priv_t *priv=demux->priv;
87 y4m_frame_info_t fi;
88 unsigned char *buf[3];
89 int err, size;
90 int nextc;
92 // Concatenated stream check; only done if seekable so skip(-1) works
93 if (demux->stream->flags & MP_STREAM_SEEK_BW) {
94 nextc = stream_read_char(demux->stream);
95 stream_skip(demux->stream, -1);
96 if (nextc == 'Y') {
97 read_streaminfo(demux);
98 demux->seekable = 0;
102 y4m_init_frame_info(&fi);
104 demux->filepos=stream_tell(demux->stream);
106 size = ((sh_video_t*)ds->sh)->disp_w*((sh_video_t*)ds->sh)->disp_h;
108 dp = new_demux_packet(3*size/2);
110 /* swap U and V components */
111 buf[0] = dp->buffer;
112 buf[1] = dp->buffer + 5*size/4;
113 buf[2] = dp->buffer + size;
115 if (priv->is_older)
117 int c;
119 c = stream_read_char(demux->stream); /* F */
120 if (c == -256)
121 return 0; /* EOF */
122 if (c != 'F')
124 mp_msg(MSGT_DEMUX, MSGL_V, "Bad frame at %d\n", (int)stream_tell(demux->stream)-1);
125 return 0;
127 stream_skip(demux->stream, 5); /* RAME\n */
128 stream_read(demux->stream, buf[0], size);
129 stream_read(demux->stream, buf[1], size/4);
130 stream_read(demux->stream, buf[2], size/4);
132 else
134 if ((err=y4m_read_frame(demux->stream, priv->si, &fi, buf)) != Y4M_OK) {
135 mp_msg(MSGT_DEMUX, MSGL_ERR, "error reading frame %s\n", y4m_strerr(err));
136 return 0;
140 /* This seems to be the right way to calculate the presentation time stamp */
141 dp->pts=(float)priv->framenum/((sh_video_t*)ds->sh)->fps;
142 priv->framenum++;
143 dp->pos=demux->filepos;
144 ds_add_packet(ds, dp);
146 return 1;
149 static void read_streaminfo(demuxer_t *demuxer)
151 y4m_priv_t *priv = demuxer->priv;
152 sh_video_t *sh = demuxer->video->sh;
153 y4m_ratio_t ratio;
154 int err;
156 if (priv->is_older)
158 char buf[4];
159 int frame_rate_code;
161 stream_skip(demuxer->stream, 8); /* YUV4MPEG */
162 stream_skip(demuxer->stream, 1); /* space */
163 stream_read(demuxer->stream, (char *)&buf[0], 3);
164 buf[3] = 0;
165 sh->disp_w = atoi(buf);
166 stream_skip(demuxer->stream, 1); /* space */
167 stream_read(demuxer->stream, (char *)&buf[0], 3);
168 buf[3] = 0;
169 sh->disp_h = atoi(buf);
170 stream_skip(demuxer->stream, 1); /* space */
171 stream_read(demuxer->stream, (char *)&buf[0], 1);
172 buf[1] = 0;
173 frame_rate_code = atoi(buf);
174 stream_skip(demuxer->stream, 1); /* new-line */
176 if (!sh->fps)
178 /* values from xawtv */
179 switch(frame_rate_code)
181 case 1:
182 sh->fps = 23.976f;
183 break;
184 case 2:
185 sh->fps = 24.0f;
186 break;
187 case 3:
188 sh->fps = 25.0f;
189 break;
190 case 4:
191 sh->fps = 29.97f;
192 break;
193 case 5:
194 sh->fps = 30.0f;
195 break;
196 case 6:
197 sh->fps = 50.0f;
198 break;
199 case 7:
200 sh->fps = 59.94f;
201 break;
202 case 8:
203 sh->fps = 60.0f;
204 break;
205 default:
206 sh->fps = 25.0f;
209 sh->frametime = 1.0f/sh->fps;
211 else
213 y4m_init_stream_info(priv->si);
214 if ((err=y4m_read_stream_header(demuxer->stream, priv->si)) != Y4M_OK)
215 mp_msg(MSGT_DEMUXER, MSGL_FATAL, "error parsing YUV4MPEG header: %s\n", y4m_strerr(err));
217 if(!sh->fps) {
218 ratio = y4m_si_get_framerate(priv->si);
219 if (ratio.d != 0)
220 sh->fps=(float)ratio.n/(float)ratio.d;
221 else
222 sh->fps=15.0f;
224 sh->frametime=1.0f/sh->fps;
226 ratio = y4m_si_get_sampleaspect(priv->si);
228 sh->disp_w = y4m_si_get_width(priv->si);
229 sh->disp_h = y4m_si_get_height(priv->si);
231 if (ratio.d != 0 && ratio.n != 0)
232 sh->aspect = (float)(sh->disp_w*ratio.n)/(float)(sh->disp_h*ratio.d);
234 demuxer->seekable = 0;
237 sh->format = mmioFOURCC('Y', 'V', '1', '2');
239 sh->bih->biSize=40;
240 sh->bih->biWidth = sh->disp_w;
241 sh->bih->biHeight = sh->disp_h;
242 sh->bih->biPlanes=3;
243 sh->bih->biBitCount=12;
244 sh->bih->biCompression=sh->format;
245 sh->bih->biSizeImage=sh->bih->biWidth*sh->bih->biHeight*3/2; /* YV12 */
247 mp_msg(MSGT_DEMUX, MSGL_INFO, "YUV4MPEG2 Video stream %d size: display: %dx%d, codec: %ux%u\n",
248 demuxer->video->id, sh->disp_w, sh->disp_h, sh->bih->biWidth,
249 sh->bih->biHeight);
252 static demuxer_t* demux_open_y4m(demuxer_t* demuxer){
253 y4m_priv_t* priv = demuxer->priv;
254 sh_video_t* sh=new_sh_video(demuxer,0);
256 priv->framenum = 0;
257 priv->si = malloc(sizeof(y4m_stream_info_t));
259 sh->bih=calloc(1, sizeof(*sh->bih));
261 demuxer->video->sh=sh;
262 sh->ds=demuxer->video;
263 demuxer->video->id=0;
265 read_streaminfo(demuxer);
267 return demuxer;
270 static void demux_seek_y4m(demuxer_t *demuxer, float rel_seek_secs, float audio_delay, int flags) {
271 sh_video_t* sh = demuxer->video->sh;
272 y4m_priv_t* priv = demuxer->priv;
273 int rel_seek_frames = sh->fps*rel_seek_secs;
274 int size = 3*sh->disp_w*sh->disp_h/2;
275 off_t curr_pos = stream_tell(demuxer->stream);
277 if (priv->framenum + rel_seek_frames < 0) rel_seek_frames = -priv->framenum;
279 //printf("seektoframe=%d rel_seek_secs=%f seektooffset=%ld\n", priv->framenum + rel_seek_frames, rel_seek_secs, curr_pos + rel_seek_frames*(size+6));
280 //printf("framenum=%d, curr_pos=%ld, currpos/(size+6)=%f\n", priv->framenum, curr_pos, (float)curr_pos/(float)(size+6));
281 priv->framenum += rel_seek_frames;
283 if (priv->is_older) {
284 /* Well this is easy: every frame takes up size+6 bytes
285 * in the stream and we may assume that the stream pointer
286 * is always at the beginning of a frame.
287 * framenum is the number of the frame that is about to be
288 * demuxed (counting from ONE (see demux_open_y4m)) */
289 stream_seek(demuxer->stream, curr_pos + rel_seek_frames*(size+6));
290 } else {
291 /* should never come here, because seeking for YUV4MPEG2
292 * is disabled. */
293 mp_msg(MSGT_DEMUX, MSGL_WARN, "Seeking for YUV4MPEG2 not yet implemented!\n");
297 static void demux_close_y4m(demuxer_t *demuxer)
299 y4m_priv_t* priv = demuxer->priv;
301 if(!priv)
302 return;
303 if (!priv->is_older)
304 y4m_fini_stream_info(((y4m_priv_t*)demuxer->priv)->si);
305 free(((y4m_priv_t*)demuxer->priv)->si);
306 free(demuxer->priv);
307 return;
311 const demuxer_desc_t demuxer_desc_y4m = {
312 "YUV4MPEG2 demuxer",
313 "y4m",
314 "YUV4MPEG2",
315 "Rik snel",
317 DEMUXER_TYPE_Y4M,
318 1, // safe autodetect
319 y4m_check_file,
320 demux_y4m_fill_buffer,
321 demux_open_y4m,
322 demux_close_y4m,
323 demux_seek_y4m,
324 NULL