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.
28 #include <string.h> /* strtok */
35 //#include "stream/stream.h"
41 y4m_stream_info_t
* si
;
45 static int y4m_check_file(demuxer_t
* demuxer
){
46 int orig_pos
= stream_tell(demuxer
->stream
);
50 mp_msg(MSGT_DEMUX
, MSGL_V
, "Checking for YUV4MPEG2\n");
52 if(stream_read(demuxer
->stream
, buf
, 9)!=9)
57 if (strncmp("YUV4MPEG2", buf
, 9) && strncmp("YUV4MPEG ", buf
, 9)) {
61 demuxer
->priv
= malloc(sizeof(y4m_priv_t
));
66 if (!strncmp("YUV4MPEG ", buf
, 9))
68 mp_msg(MSGT_DEMUX
, MSGL_V
, "Found older YUV4MPEG format (used by xawtv)\n");
72 mp_msg(MSGT_DEMUX
,MSGL_DBG2
,"Success: YUV4MPEG2\n");
74 stream_seek(demuxer
->stream
, orig_pos
);
76 return DEMUXER_TYPE_Y4M
;
79 static void read_streaminfo(demuxer_t
*demuxer
);
82 // 0 = EOF or no stream found
83 // 1 = successfully read a packet
84 static int demux_y4m_fill_buffer(demuxer_t
*demux
, demux_stream_t
*dsds
) {
85 demux_stream_t
*ds
=demux
->video
;
87 y4m_priv_t
*priv
=demux
->priv
;
89 unsigned char *buf
[3];
93 nextc
= stream_read_char(demux
->stream
);
94 stream_skip(demux
->stream
, -1);
96 read_streaminfo(demux
);
100 y4m_init_frame_info(&fi
);
102 demux
->filepos
=stream_tell(demux
->stream
);
104 size
= ((sh_video_t
*)ds
->sh
)->disp_w
*((sh_video_t
*)ds
->sh
)->disp_h
;
106 dp
= new_demux_packet(3*size
/2);
108 /* swap U and V components */
110 buf
[1] = dp
->buffer
+ 5*size
/4;
111 buf
[2] = dp
->buffer
+ size
;
117 c
= stream_read_char(demux
->stream
); /* F */
122 mp_msg(MSGT_DEMUX
, MSGL_V
, "Bad frame at %d\n", (int)stream_tell(demux
->stream
)-1);
125 stream_skip(demux
->stream
, 5); /* RAME\n */
126 stream_read(demux
->stream
, buf
[0], size
);
127 stream_read(demux
->stream
, buf
[1], size
/4);
128 stream_read(demux
->stream
, buf
[2], size
/4);
132 if ((err
=y4m_read_frame(demux
->stream
, priv
->si
, &fi
, buf
)) != Y4M_OK
) {
133 mp_msg(MSGT_DEMUX
, MSGL_V
, "error reading frame %s\n", y4m_strerr(err
));
138 /* This seems to be the right way to calculate the presentation time stamp */
139 dp
->pts
=(float)priv
->framenum
/((sh_video_t
*)ds
->sh
)->fps
;
141 dp
->pos
=demux
->filepos
;
143 ds_add_packet(ds
, dp
);
148 static void read_streaminfo(demuxer_t
*demuxer
)
150 y4m_priv_t
*priv
= demuxer
->priv
;
151 sh_video_t
*sh
= demuxer
->video
->sh
;
160 stream_skip(demuxer
->stream
, 8); /* YUV4MPEG */
161 stream_skip(demuxer
->stream
, 1); /* space */
162 stream_read(demuxer
->stream
, (char *)&buf
[0], 3);
164 sh
->disp_w
= atoi(buf
);
165 stream_skip(demuxer
->stream
, 1); /* space */
166 stream_read(demuxer
->stream
, (char *)&buf
[0], 3);
168 sh
->disp_h
= atoi(buf
);
169 stream_skip(demuxer
->stream
, 1); /* space */
170 stream_read(demuxer
->stream
, (char *)&buf
[0], 1);
172 frame_rate_code
= atoi(buf
);
173 stream_skip(demuxer
->stream
, 1); /* new-line */
177 /* values from xawtv */
178 switch(frame_rate_code
)
208 sh
->frametime
= 1.0f
/sh
->fps
;
212 y4m_init_stream_info(priv
->si
);
213 if ((err
=y4m_read_stream_header(demuxer
->stream
, priv
->si
)) != Y4M_OK
)
214 mp_msg(MSGT_DEMUXER
, MSGL_FATAL
, "error parsing YUV4MPEG header: %s\n", y4m_strerr(err
));
217 ratio
= y4m_si_get_framerate(priv
->si
);
219 sh
->fps
=(float)ratio
.n
/(float)ratio
.d
;
223 sh
->frametime
=1.0f
/sh
->fps
;
225 ratio
= y4m_si_get_sampleaspect(priv
->si
);
227 sh
->disp_w
= y4m_si_get_width(priv
->si
);
228 sh
->disp_h
= y4m_si_get_height(priv
->si
);
230 if (ratio
.d
!= 0 && ratio
.n
!= 0)
231 sh
->aspect
= (float)(sh
->disp_w
*ratio
.n
)/(float)(sh
->disp_h
*ratio
.d
);
233 demuxer
->seekable
= 0;
236 sh
->format
= mmioFOURCC('Y', 'V', '1', '2');
239 sh
->bih
->biWidth
= sh
->disp_w
;
240 sh
->bih
->biHeight
= sh
->disp_h
;
242 sh
->bih
->biBitCount
=12;
243 sh
->bih
->biCompression
=sh
->format
;
244 sh
->bih
->biSizeImage
=sh
->bih
->biWidth
*sh
->bih
->biHeight
*3/2; /* YV12 */
246 mp_msg(MSGT_DEMUX
, MSGL_INFO
, "YUV4MPEG2 Video stream %d size: display: %dx%d, codec: %ux%u\n",
247 demuxer
->video
->id
, sh
->disp_w
, sh
->disp_h
, sh
->bih
->biWidth
,
251 static demuxer_t
* demux_open_y4m(demuxer_t
* demuxer
){
252 y4m_priv_t
* priv
= demuxer
->priv
;
253 sh_video_t
* sh
=new_sh_video(demuxer
,0);
256 priv
->si
= malloc(sizeof(y4m_stream_info_t
));
258 sh
->bih
=calloc(1, sizeof(BITMAPINFOHEADER
));
260 demuxer
->video
->sh
=sh
;
261 sh
->ds
=demuxer
->video
;
262 demuxer
->video
->id
=0;
264 read_streaminfo(demuxer
);
269 static void demux_seek_y4m(demuxer_t
*demuxer
, float rel_seek_secs
, float audio_delay
, int flags
) {
270 sh_video_t
* sh
= demuxer
->video
->sh
;
271 y4m_priv_t
* priv
= demuxer
->priv
;
272 int rel_seek_frames
= sh
->fps
*rel_seek_secs
;
273 int size
= 3*sh
->disp_w
*sh
->disp_h
/2;
274 off_t curr_pos
= stream_tell(demuxer
->stream
);
276 if (priv
->framenum
+ rel_seek_frames
< 0) rel_seek_frames
= -priv
->framenum
;
278 //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));
279 //printf("framenum=%d, curr_pos=%ld, currpos/(size+6)=%f\n", priv->framenum, curr_pos, (float)curr_pos/(float)(size+6));
280 priv
->framenum
+= rel_seek_frames
;
282 if (priv
->is_older
) {
283 /* Well this is easy: every frame takes up size+6 bytes
284 * in the stream and we may assume that the stream pointer
285 * is always at the beginning of a frame.
286 * framenum is the number of the frame that is about to be
287 * demuxed (counting from ONE (see demux_open_y4m)) */
288 stream_seek(demuxer
->stream
, curr_pos
+ rel_seek_frames
*(size
+6));
290 /* should never come here, because seeking for YUV4MPEG2
292 mp_msg(MSGT_DEMUX
, MSGL_WARN
, "Seeking for YUV4MPEG2 not yet implemented!\n");
296 static void demux_close_y4m(demuxer_t
*demuxer
)
298 y4m_priv_t
* priv
= demuxer
->priv
;
303 y4m_fini_stream_info(((y4m_priv_t
*)demuxer
->priv
)->si
);
304 free(((y4m_priv_t
*)demuxer
->priv
)->si
);
310 const demuxer_desc_t demuxer_desc_y4m
= {
317 1, // safe autodetect
319 demux_y4m_fill_buffer
,