Mercurial > mplayer.hg
annotate libmpdemux/demux_lavf.c @ 21771:4f317822d1fc
first step toward amr-nb muxing support
author | michael |
---|---|
date | Fri, 29 Dec 2006 18:56:28 +0000 |
parents | 43c8f1aa82bb |
children | 2158c95b71f6 |
rev | line source |
---|---|
12164 | 1 /* |
2 Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at> | |
3 | |
4 This program is free software; you can redistribute it and/or modify | |
5 it under the terms of the GNU General Public License as published by | |
6 the Free Software Foundation; either version 2 of the License, or | |
7 (at your option) any later version. | |
8 | |
9 This program is distributed in the hope that it will be useful, | |
10 but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 GNU General Public License for more details. | |
13 | |
14 You should have received a copy of the GNU General Public License | |
15 along with this program; if not, write to the Free Software | |
17367
401b440a6d76
Update licensing information: The FSF changed postal address.
diego
parents:
17354
diff
changeset
|
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
12164 | 17 */ |
18 | |
19 // #include <stdio.h> | |
20 #include <stdlib.h> | |
21 // #include <unistd.h> | |
19611 | 22 #include <limits.h> |
12164 | 23 |
24 #include "config.h" | |
25 #include "mp_msg.h" | |
26 // #include "help_mp.h" | |
27 | |
28 #include "stream.h" | |
29 #include "demuxer.h" | |
30 #include "stheader.h" | |
19598 | 31 #include "m_option.h" |
12164 | 32 |
17354 | 33 #ifdef USE_LIBAVFORMAT_SO |
34 #include <ffmpeg/avformat.h> | |
19598 | 35 #include <ffmpeg/opt.h> |
20733
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
36 typedef struct CodecTag { |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
37 int id; |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
38 unsigned int tag; |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
39 unsigned int invalid_asf : 1; |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
40 } CodecTag; |
17354 | 41 #else |
12164 | 42 #include "avformat.h" |
20733
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
43 #include "riff.h" |
12164 | 44 #include "avi.h" |
19598 | 45 #include "opt.h" |
17354 | 46 #endif |
12164 | 47 |
48 #define PROBE_BUF_SIZE 2048 | |
49 | |
18775 | 50 extern char *audio_lang; |
19598 | 51 static unsigned int opt_probesize = 0; |
52 | |
53 m_option_t lavfdopts_conf[] = { | |
54 {"probesize", &(opt_probesize), CONF_TYPE_INT, CONF_RANGE, 32, INT_MAX, NULL}, | |
55 {NULL, NULL, 0, 0, 0, 0, NULL} | |
56 }; | |
57 | |
18775 | 58 |
12164 | 59 typedef struct lavf_priv_t{ |
60 AVInputFormat *avif; | |
61 AVFormatContext *avfc; | |
62 ByteIOContext pb; | |
63 int audio_streams; | |
64 int video_streams; | |
12168 | 65 int64_t last_pts; |
18762 | 66 int astreams[MAX_A_STREAMS]; |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
67 int vstreams[MAX_V_STREAMS]; |
12164 | 68 }lavf_priv_t; |
69 | |
17977
f70772d02eaa
Convert printfs in aviprint.c to mp_msg and give the information printing
diego
parents:
17932
diff
changeset
|
70 extern void print_wave_header(WAVEFORMATEX *h, int verbose_level); |
f70772d02eaa
Convert printfs in aviprint.c to mp_msg and give the information printing
diego
parents:
17932
diff
changeset
|
71 extern void print_video_header(BITMAPINFOHEADER *h, int verbose_level); |
12164 | 72 |
15011 | 73 int64_t ff_gcd(int64_t a, int64_t b); |
74 | |
20733
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
75 const CodecTag mp_wav_tags[] = { |
20944
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
76 { CODEC_ID_ADPCM_4XM, MKTAG('4', 'X', 'M', 'A')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
77 { CODEC_ID_ADPCM_EA, MKTAG('A', 'D', 'E', 'A')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
78 { CODEC_ID_ADPCM_IMA_WS, MKTAG('A', 'I', 'W', 'S')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
79 { CODEC_ID_DSICINAUDIO, MKTAG('D', 'C', 'I', 'A')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
80 { CODEC_ID_INTERPLAY_DPCM, MKTAG('I', 'N', 'P', 'A')}, |
21763
43c8f1aa82bb
add support for musepack native decoder from ffmpeg (require -demuxer lavf)
aurel
parents:
21568
diff
changeset
|
81 { CODEC_ID_MUSEPACK7, MKTAG('M', 'P', 'C', ' ')}, |
20944
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
82 { CODEC_ID_PCM_S24BE, MKTAG('i', 'n', '2', '4')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
83 { CODEC_ID_PCM_S8, MKTAG('t', 'w', 'o', 's')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
84 { CODEC_ID_ROQ_DPCM, MKTAG('R', 'o', 'Q', 'A')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
85 { CODEC_ID_SHORTEN, MKTAG('s', 'h', 'r', 'n')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
86 { CODEC_ID_TTA, MKTAG('T', 'T', 'A', '1')}, |
20910582789d
cosmetics: Restore alphabetical order, align both tables.
diego
parents:
20942
diff
changeset
|
87 { CODEC_ID_WAVPACK, MKTAG('W', 'V', 'P', 'K')}, |
21362 | 88 { CODEC_ID_WESTWOOD_SND1, MKTAG('S', 'N', 'D', '1')}, |
20990 | 89 { CODEC_ID_XAN_DPCM, MKTAG('A', 'x', 'a', 'n')}, |
21771 | 90 { CODEC_ID_AMR_NB, MKTAG('n', 'b', 0, 0)}, |
20733
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
91 { 0, 0 }, |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
92 }; |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
93 |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
94 const CodecTag mp_bmp_tags[] = { |
20940 | 95 { CODEC_ID_DSICINVIDEO, MKTAG('D', 'C', 'I', 'V')}, |
20963 | 96 { CODEC_ID_FLIC, MKTAG('F', 'L', 'I', 'C')}, |
20980
70ca50bcc4a8
support for some more fringe formats, still buggy ..
diego
parents:
20963
diff
changeset
|
97 { CODEC_ID_IDCIN, MKTAG('I', 'D', 'C', 'I')}, |
70ca50bcc4a8
support for some more fringe formats, still buggy ..
diego
parents:
20963
diff
changeset
|
98 { CODEC_ID_INTERPLAY_VIDEO, MKTAG('I', 'N', 'P', 'V')}, |
20942
2093d87ed14b
support for RoQ video and audio through libavformat
diego
parents:
20941
diff
changeset
|
99 { CODEC_ID_ROQ, MKTAG('R', 'o', 'Q', 'V')}, |
20933 | 100 { CODEC_ID_TIERTEXSEQVIDEO, MKTAG('T', 'S', 'E', 'Q')}, |
20935
4a8c67e44c7f
cosmetics: alphabetical order and prettyprinting for the CodecTag table
diego
parents:
20934
diff
changeset
|
101 { CODEC_ID_VMDVIDEO, MKTAG('V', 'M', 'D', 'V')}, |
20980
70ca50bcc4a8
support for some more fringe formats, still buggy ..
diego
parents:
20963
diff
changeset
|
102 { CODEC_ID_WS_VQA, MKTAG('V', 'Q', 'A', 'V')}, |
20935
4a8c67e44c7f
cosmetics: alphabetical order and prettyprinting for the CodecTag table
diego
parents:
20934
diff
changeset
|
103 { CODEC_ID_XAN_WC3, MKTAG('W', 'C', '3', 'V')}, |
20733
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
104 { 0, 0 }, |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
105 }; |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
106 |
12164 | 107 static int mp_open(URLContext *h, const char *filename, int flags){ |
108 return 0; | |
109 } | |
110 | |
111 static int mp_read(URLContext *h, unsigned char *buf, int size){ | |
112 stream_t *stream = (stream_t*)h->priv_data; | |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
113 int ret; |
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
114 |
12164 | 115 if(stream_eof(stream)) //needed? |
116 return -1; | |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
117 ret=stream_read(stream, buf, size); |
12166 | 118 |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
119 mp_msg(MSGT_HEADER,MSGL_DBG2,"%d=mp_read(%p, %p, %d), eof:%d\n", ret, h, buf, size, stream->eof); |
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
120 return ret; |
12164 | 121 } |
122 | |
123 static int mp_write(URLContext *h, unsigned char *buf, int size){ | |
124 return -1; | |
125 } | |
126 | |
127 static offset_t mp_seek(URLContext *h, offset_t pos, int whence){ | |
128 stream_t *stream = (stream_t*)h->priv_data; | |
12165
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
129 |
6ae21c78ed8d
libavformat really doesnt like it that the streams get stuck if the end is reached
michael
parents:
12164
diff
changeset
|
130 mp_msg(MSGT_HEADER,MSGL_DBG2,"mp_seek(%p, %d, %d)\n", h, (int)pos, whence); |
12164 | 131 if(whence == SEEK_CUR) |
132 pos +=stream_tell(stream); | |
133 else if(whence == SEEK_END) | |
134 pos += stream->end_pos; | |
135 else if(whence != SEEK_SET) | |
136 return -1; | |
137 | |
12167 | 138 if(pos<stream->end_pos && stream->eof) |
12166 | 139 stream_reset(stream); |
12164 | 140 if(stream_seek(stream, pos)==0) |
141 return -1; | |
12166 | 142 |
12164 | 143 return pos; |
144 } | |
145 | |
146 static int mp_close(URLContext *h){ | |
147 return 0; | |
148 } | |
149 | |
150 static URLProtocol mp_protocol = { | |
151 "mp", | |
152 mp_open, | |
153 mp_read, | |
154 mp_write, | |
155 mp_seek, | |
156 mp_close, | |
157 }; | |
158 | |
16175 | 159 static int lavf_check_file(demuxer_t *demuxer){ |
12164 | 160 AVProbeData avpd; |
161 uint8_t buf[PROBE_BUF_SIZE]; | |
162 lavf_priv_t *priv; | |
163 | |
164 if(!demuxer->priv) | |
165 demuxer->priv=calloc(sizeof(lavf_priv_t),1); | |
166 priv= demuxer->priv; | |
167 | |
168 av_register_all(); | |
169 | |
15819 | 170 if(stream_read(demuxer->stream, buf, PROBE_BUF_SIZE)!=PROBE_BUF_SIZE) |
171 return 0; | |
12164 | 172 avpd.filename= demuxer->stream->url; |
173 avpd.buf= buf; | |
174 avpd.buf_size= PROBE_BUF_SIZE; | |
175 | |
176 priv->avif= av_probe_input_format(&avpd, 1); | |
177 if(!priv->avif){ | |
178 mp_msg(MSGT_HEADER,MSGL_V,"LAVF_check: no clue about this gibberish!\n"); | |
179 return 0; | |
180 }else | |
181 mp_msg(MSGT_HEADER,MSGL_V,"LAVF_check: %s\n", priv->avif->long_name); | |
182 | |
16175 | 183 return DEMUXER_TYPE_LAVF; |
12164 | 184 } |
185 | |
16175 | 186 static demuxer_t* demux_open_lavf(demuxer_t *demuxer){ |
12164 | 187 AVFormatContext *avfc; |
188 AVFormatParameters ap; | |
19598 | 189 AVOption *opt; |
12164 | 190 lavf_priv_t *priv= demuxer->priv; |
15011 | 191 int i,g; |
12164 | 192 char mp_filename[256]="mp:"; |
193 | |
194 memset(&ap, 0, sizeof(AVFormatParameters)); | |
195 | |
196 stream_seek(demuxer->stream, 0); | |
197 | |
198 register_protocol(&mp_protocol); | |
199 | |
19598 | 200 avfc = av_alloc_format_context(); |
21548 | 201 |
202 if (correct_pts) | |
203 avfc->flags |= AVFMT_FLAG_GENPTS; | |
21568 | 204 if (index_mode == 0) |
205 avfc->flags |= AVFMT_FLAG_IGNIDX; | |
21548 | 206 |
19598 | 207 ap.prealloced_context = 1; |
208 if(opt_probesize) { | |
209 double d = (double) opt_probesize; | |
210 opt = av_set_double(avfc, "probesize", opt_probesize); | |
211 if(!opt) mp_msg(MSGT_HEADER,MSGL_ERR, "demux_lavf, couldn't set option probesize to %.3f\r\n", d); | |
212 } | |
213 | |
12463 | 214 if(demuxer->stream->url) |
215 strncpy(mp_filename + 3, demuxer->stream->url, sizeof(mp_filename)-3); | |
216 else | |
217 strncpy(mp_filename + 3, "foobar.dummy", sizeof(mp_filename)-3); | |
12164 | 218 |
219 url_fopen(&priv->pb, mp_filename, URL_RDONLY); | |
220 | |
221 ((URLContext*)(priv->pb.opaque))->priv_data= demuxer->stream; | |
222 | |
223 if(av_open_input_stream(&avfc, &priv->pb, mp_filename, priv->avif, &ap)<0){ | |
224 mp_msg(MSGT_HEADER,MSGL_ERR,"LAVF_header: av_open_input_stream() failed\n"); | |
16175 | 225 return NULL; |
12164 | 226 } |
227 | |
228 priv->avfc= avfc; | |
229 | |
230 if(av_find_stream_info(avfc) < 0){ | |
231 mp_msg(MSGT_HEADER,MSGL_ERR,"LAVF_header: av_find_stream_info() failed\n"); | |
16175 | 232 return NULL; |
12164 | 233 } |
234 | |
12167 | 235 if(avfc->title [0]) demux_info_add(demuxer, "name" , avfc->title ); |
236 if(avfc->author [0]) demux_info_add(demuxer, "author" , avfc->author ); | |
237 if(avfc->copyright[0]) demux_info_add(demuxer, "copyright", avfc->copyright); | |
238 if(avfc->comment [0]) demux_info_add(demuxer, "comments" , avfc->comment ); | |
239 if(avfc->album [0]) demux_info_add(demuxer, "album" , avfc->album ); | |
240 // if(avfc->year ) demux_info_add(demuxer, "year" , avfc->year ); | |
241 // if(avfc->track ) demux_info_add(demuxer, "track" , avfc->track ); | |
242 if(avfc->genre [0]) demux_info_add(demuxer, "genre" , avfc->genre ); | |
12164 | 243 |
244 for(i=0; i<avfc->nb_streams; i++){ | |
245 AVStream *st= avfc->streams[i]; | |
16000 | 246 AVCodecContext *codec= st->codec; |
19073
8b52dad54b1d
Remove #if LIBAVCODEC_BUILD >= XXX and #if LIBAVFORMAT_BUILD >= XXX jungle.
diego
parents:
19062
diff
changeset
|
247 |
12164 | 248 switch(codec->codec_type){ |
249 case CODEC_TYPE_AUDIO:{ | |
250 WAVEFORMATEX *wf= calloc(sizeof(WAVEFORMATEX) + codec->extradata_size, 1); | |
18985 | 251 sh_audio_t* sh_audio; |
18762 | 252 if(priv->audio_streams >= MAX_A_STREAMS) |
253 break; | |
18985 | 254 sh_audio=new_sh_audio(demuxer, i); |
18762 | 255 if(!sh_audio) |
256 break; | |
257 priv->astreams[priv->audio_streams] = i; | |
12164 | 258 priv->audio_streams++; |
259 if(!codec->codec_tag) | |
260 codec->codec_tag= codec_get_wav_tag(codec->codec_id); | |
20733
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
261 if(!codec->codec_tag) |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
262 codec->codec_tag= codec_get_tag(mp_wav_tags, codec->codec_id); |
12164 | 263 wf->wFormatTag= codec->codec_tag; |
264 wf->nChannels= codec->channels; | |
265 wf->nSamplesPerSec= codec->sample_rate; | |
266 wf->nAvgBytesPerSec= codec->bit_rate/8; | |
267 wf->nBlockAlign= codec->block_align; | |
268 wf->wBitsPerSample= codec->bits_per_sample; | |
269 wf->cbSize= codec->extradata_size; | |
270 if(codec->extradata_size){ | |
271 memcpy( | |
272 wf + 1, | |
273 codec->extradata, | |
274 codec->extradata_size); | |
275 } | |
276 sh_audio->wf= wf; | |
15011 | 277 sh_audio->audio.dwSampleSize= codec->block_align; |
278 if(codec->frame_size && codec->sample_rate){ | |
279 sh_audio->audio.dwScale=codec->frame_size; | |
280 sh_audio->audio.dwRate= codec->sample_rate; | |
281 }else{ | |
282 sh_audio->audio.dwScale= codec->block_align ? codec->block_align*8 : 8; | |
283 sh_audio->audio.dwRate = codec->bit_rate; | |
284 } | |
285 g= ff_gcd(sh_audio->audio.dwScale, sh_audio->audio.dwRate); | |
286 sh_audio->audio.dwScale /= g; | |
287 sh_audio->audio.dwRate /= g; | |
288 // printf("sca:%d rat:%d fs:%d sr:%d ba:%d\n", sh_audio->audio.dwScale, sh_audio->audio.dwRate, codec->frame_size, codec->sample_rate, codec->block_align); | |
12164 | 289 sh_audio->ds= demuxer->audio; |
290 sh_audio->format= codec->codec_tag; | |
291 sh_audio->channels= codec->channels; | |
292 sh_audio->samplerate= codec->sample_rate; | |
15007 | 293 sh_audio->i_bps= codec->bit_rate/8; |
16134
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
294 switch (codec->codec_id) { |
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
295 case CODEC_ID_PCM_S8: |
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
296 case CODEC_ID_PCM_U8: |
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
297 sh_audio->samplesize = 1; |
16135 | 298 break; |
16134
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
299 case CODEC_ID_PCM_S16LE: |
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
300 case CODEC_ID_PCM_S16BE: |
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
301 case CODEC_ID_PCM_U16LE: |
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
302 case CODEC_ID_PCM_U16BE: |
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
303 sh_audio->samplesize = 2; |
16135 | 304 break; |
305 case CODEC_ID_PCM_ALAW: | |
306 sh_audio->format = 0x6; | |
307 break; | |
308 case CODEC_ID_PCM_MULAW: | |
309 sh_audio->format = 0x7; | |
310 break; | |
16134
a1fd1a7eeb35
lavf demuxer with raw PCM fix (and a related hang)
reimar
parents:
16000
diff
changeset
|
311 } |
17977
f70772d02eaa
Convert printfs in aviprint.c to mp_msg and give the information printing
diego
parents:
17932
diff
changeset
|
312 if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_wave_header(sh_audio->wf, MSGL_V); |
18775 | 313 if((audio_lang && st->language[0] && !strncmp(audio_lang, st->language, 3)) |
314 || (demuxer->audio->id == i || demuxer->audio->id == -1) | |
315 ) { | |
316 demuxer->audio->id = i; | |
317 demuxer->audio->sh= demuxer->a_streams[i]; | |
318 } | |
319 else | |
15004 | 320 st->discard= AVDISCARD_ALL; |
12164 | 321 break;} |
322 case CODEC_TYPE_VIDEO:{ | |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
323 sh_video_t* sh_video; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
324 BITMAPINFOHEADER *bih; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
325 if(priv->video_streams >= MAX_V_STREAMS) |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
326 break; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
327 sh_video=new_sh_video(demuxer, i); |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
328 if(!sh_video) break; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
329 priv->vstreams[priv->video_streams] = i; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
330 priv->video_streams++; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
331 bih=calloc(sizeof(BITMAPINFOHEADER) + codec->extradata_size,1); |
12164 | 332 |
333 if(!codec->codec_tag) | |
334 codec->codec_tag= codec_get_bmp_tag(codec->codec_id); | |
20733
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
335 if(!codec->codec_tag) |
404fb439acba
Add our own CODEC_ID -> fourcc translation tables so we do not need
reimar
parents:
20070
diff
changeset
|
336 codec->codec_tag= codec_get_tag(mp_bmp_tags, codec->codec_id); |
12164 | 337 bih->biSize= sizeof(BITMAPINFOHEADER) + codec->extradata_size; |
338 bih->biWidth= codec->width; | |
339 bih->biHeight= codec->height; | |
340 bih->biBitCount= codec->bits_per_sample; | |
341 bih->biSizeImage = bih->biWidth * bih->biHeight * bih->biBitCount/8; | |
342 bih->biCompression= codec->codec_tag; | |
343 sh_video->bih= bih; | |
344 sh_video->disp_w= codec->width; | |
345 sh_video->disp_h= codec->height; | |
16718
044260623695
makes demux_lavf (-demuxer 35) use the framerate specified in the container
gpoirier
parents:
16346
diff
changeset
|
346 if (st->time_base.den) { /* if container has time_base, use that */ |
044260623695
makes demux_lavf (-demuxer 35) use the framerate specified in the container
gpoirier
parents:
16346
diff
changeset
|
347 sh_video->video.dwRate= st->time_base.den; |
044260623695
makes demux_lavf (-demuxer 35) use the framerate specified in the container
gpoirier
parents:
16346
diff
changeset
|
348 sh_video->video.dwScale= st->time_base.num; |
044260623695
makes demux_lavf (-demuxer 35) use the framerate specified in the container
gpoirier
parents:
16346
diff
changeset
|
349 } else { |
15308 | 350 sh_video->video.dwRate= codec->time_base.den; |
351 sh_video->video.dwScale= codec->time_base.num; | |
16718
044260623695
makes demux_lavf (-demuxer 35) use the framerate specified in the container
gpoirier
parents:
16346
diff
changeset
|
352 } |
17556 | 353 sh_video->fps=av_q2d(st->r_frame_rate); |
354 sh_video->frametime=1/av_q2d(st->r_frame_rate); | |
12164 | 355 sh_video->format = bih->biCompression; |
12167 | 356 sh_video->aspect= codec->width * codec->sample_aspect_ratio.num |
357 / (float)(codec->height * codec->sample_aspect_ratio.den); | |
15007 | 358 sh_video->i_bps= codec->bit_rate/8; |
12167 | 359 mp_msg(MSGT_DEMUX,MSGL_DBG2,"aspect= %d*%d/(%d*%d)\n", |
360 codec->width, codec->sample_aspect_ratio.num, | |
361 codec->height, codec->sample_aspect_ratio.den); | |
362 | |
12164 | 363 sh_video->ds= demuxer->video; |
364 if(codec->extradata_size) | |
365 memcpy(sh_video->bih + 1, codec->extradata, codec->extradata_size); | |
17977
f70772d02eaa
Convert printfs in aviprint.c to mp_msg and give the information printing
diego
parents:
17932
diff
changeset
|
366 if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_video_header(sh_video->bih, MSGL_V); |
12164 | 367 /* short biPlanes; |
368 int biXPelsPerMeter; | |
369 int biYPelsPerMeter; | |
370 int biClrUsed; | |
371 int biClrImportant;*/ | |
15004 | 372 if(demuxer->video->id != i && demuxer->video->id != -1) |
373 st->discard= AVDISCARD_ALL; | |
374 else{ | |
375 demuxer->video->id = i; | |
376 demuxer->video->sh= demuxer->v_streams[i]; | |
377 } | |
12164 | 378 break;} |
15004 | 379 default: |
380 st->discard= AVDISCARD_ALL; | |
12164 | 381 } |
382 } | |
383 | |
384 mp_msg(MSGT_HEADER,MSGL_V,"LAVF: %d audio and %d video streams found\n",priv->audio_streams,priv->video_streams); | |
13749 | 385 mp_msg(MSGT_HEADER,MSGL_V,"LAVF: build %d\n", LIBAVFORMAT_BUILD); |
12164 | 386 if(!priv->audio_streams) demuxer->audio->id=-2; // nosound |
387 // else if(best_audio > 0 && demuxer->audio->id == -1) demuxer->audio->id=best_audio; | |
388 if(!priv->video_streams){ | |
389 if(!priv->audio_streams){ | |
390 mp_msg(MSGT_HEADER,MSGL_ERR,"LAVF: no audio or video headers found - broken file?\n"); | |
16175 | 391 return NULL; |
12164 | 392 } |
393 demuxer->video->id=-2; // audio-only | |
394 } //else if (best_video > 0 && demuxer->video->id == -1) demuxer->video->id = best_video; | |
395 | |
16175 | 396 return demuxer; |
12164 | 397 } |
398 | |
16175 | 399 static int demux_lavf_fill_buffer(demuxer_t *demux, demux_stream_t *dsds){ |
12164 | 400 lavf_priv_t *priv= demux->priv; |
401 AVPacket pkt; | |
402 demux_packet_t *dp; | |
403 demux_stream_t *ds; | |
404 int id; | |
405 mp_msg(MSGT_DEMUX,MSGL_DBG2,"demux_lavf_fill_buffer()\n"); | |
406 | |
407 demux->filepos=stream_tell(demux->stream); | |
408 | |
409 if(av_read_frame(priv->avfc, &pkt) < 0) | |
410 return 0; | |
411 | |
412 id= pkt.stream_index; | |
413 | |
414 if(id==demux->audio->id){ | |
415 // audio | |
416 ds=demux->audio; | |
417 if(!ds->sh){ | |
418 ds->sh=demux->a_streams[id]; | |
419 mp_msg(MSGT_DEMUX,MSGL_V,"Auto-selected LAVF audio ID = %d\n",ds->id); | |
420 } | |
421 } else if(id==demux->video->id){ | |
422 // video | |
423 ds=demux->video; | |
424 if(!ds->sh){ | |
425 ds->sh=demux->v_streams[id]; | |
426 mp_msg(MSGT_DEMUX,MSGL_V,"Auto-selected LAVF video ID = %d\n",ds->id); | |
427 } | |
14611 | 428 } else { |
429 av_free_packet(&pkt); | |
430 return 1; | |
431 } | |
12164 | 432 |
433 if(0/*pkt.destruct == av_destruct_packet*/){ | |
434 //ok kids, dont try this at home :) | |
19062
83c3afeab35d
drops casts from void * on malloc/calloc from libmpdemux code
reynaldo
parents:
18985
diff
changeset
|
435 dp=malloc(sizeof(demux_packet_t)); |
12164 | 436 dp->len=pkt.size; |
437 dp->next=NULL; | |
438 dp->refcount=1; | |
439 dp->master=NULL; | |
440 dp->buffer=pkt.data; | |
441 pkt.destruct= NULL; | |
442 }else{ | |
443 dp=new_demux_packet(pkt.size); | |
444 memcpy(dp->buffer, pkt.data, pkt.size); | |
445 av_free_packet(&pkt); | |
446 } | |
447 | |
13747 | 448 if(pkt.pts != AV_NOPTS_VALUE){ |
15308 | 449 dp->pts=pkt.pts * av_q2d(priv->avfc->streams[id]->time_base); |
450 priv->last_pts= dp->pts * AV_TIME_BASE; | |
13747 | 451 } |
12164 | 452 dp->pos=demux->filepos; |
453 dp->flags= !!(pkt.flags&PKT_FLAG_KEY); | |
454 // append packet to DS stream: | |
455 ds_add_packet(ds,dp); | |
456 return 1; | |
457 } | |
458 | |
17636 | 459 static void demux_seek_lavf(demuxer_t *demuxer, float rel_seek_secs, float audio_delay, int flags){ |
12168 | 460 lavf_priv_t *priv = demuxer->priv; |
20070 | 461 int avsflags = 0; |
17636 | 462 mp_msg(MSGT_DEMUX,MSGL_DBG2,"demux_seek_lavf(%p, %f, %f, %d)\n", demuxer, rel_seek_secs, audio_delay, flags); |
19073
8b52dad54b1d
Remove #if LIBAVCODEC_BUILD >= XXX and #if LIBAVFORMAT_BUILD >= XXX jungle.
diego
parents:
19062
diff
changeset
|
463 |
20070 | 464 if (flags & 1) // absolute seek |
465 priv->last_pts = priv->avfc->start_time; | |
466 if (flags & 2) { // percent seek | |
467 if (priv->avfc->duration == 0 || priv->avfc->duration == AV_NOPTS_VALUE) | |
468 return; | |
469 priv->last_pts += rel_seek_secs * priv->avfc->duration; | |
470 } else { | |
471 priv->last_pts += rel_seek_secs * AV_TIME_BASE; | |
472 if (rel_seek_secs < 0) avsflags = AVSEEK_FLAG_BACKWARD; | |
473 } | |
474 av_seek_frame(priv->avfc, -1, priv->last_pts, avsflags); | |
12164 | 475 } |
476 | |
16175 | 477 static int demux_lavf_control(demuxer_t *demuxer, int cmd, void *arg) |
12164 | 478 { |
479 lavf_priv_t *priv = demuxer->priv; | |
480 | |
481 switch (cmd) { | |
12168 | 482 case DEMUXER_CTRL_GET_TIME_LENGTH: |
19207
c636a4e9565a
Do not treat AV_NOPTS_VALUE as a valid duration value.
reimar
parents:
19160
diff
changeset
|
483 if (priv->avfc->duration == 0 || priv->avfc->duration == AV_NOPTS_VALUE) |
12164 | 484 return DEMUXER_CTRL_DONTKNOW; |
485 | |
16346
6ff303d2876b
Make -identify's 'ID_LENGTH=' print a float and not an integer.. The
ods15
parents:
16175
diff
changeset
|
486 *((double *)arg) = (double)priv->avfc->duration / AV_TIME_BASE; |
12164 | 487 return DEMUXER_CTRL_OK; |
488 | |
489 case DEMUXER_CTRL_GET_PERCENT_POS: | |
19207
c636a4e9565a
Do not treat AV_NOPTS_VALUE as a valid duration value.
reimar
parents:
19160
diff
changeset
|
490 if (priv->avfc->duration == 0 || priv->avfc->duration == AV_NOPTS_VALUE) |
12164 | 491 return DEMUXER_CTRL_DONTKNOW; |
492 | |
19160
ccb42ce33c23
Take start time into consideration when calculation percentage position
reimar
parents:
19073
diff
changeset
|
493 *((int *)arg) = (int)((priv->last_pts - priv->avfc->start_time)*100 / priv->avfc->duration); |
12168 | 494 return DEMUXER_CTRL_OK; |
18762 | 495 case DEMUXER_CTRL_SWITCH_AUDIO: |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
496 case DEMUXER_CTRL_SWITCH_VIDEO: |
18762 | 497 { |
498 int id = *((int*)arg); | |
499 int newid = -2; | |
500 int i, curridx = -2; | |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
501 int nstreams, *pstreams; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
502 demux_stream_t *ds; |
18762 | 503 |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
504 if(cmd == DEMUXER_CTRL_SWITCH_VIDEO) |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
505 { |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
506 ds = demuxer->video; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
507 nstreams = priv->video_streams; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
508 pstreams = priv->vstreams; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
509 } |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
510 else |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
511 { |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
512 ds = demuxer->audio; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
513 nstreams = priv->audio_streams; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
514 pstreams = priv->astreams; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
515 } |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
516 if(ds->id == -2) |
18762 | 517 return DEMUXER_CTRL_NOTIMPL; |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
518 for(i = 0; i < nstreams; i++) |
18762 | 519 { |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
520 if(pstreams[i] == ds->id) //current stream id |
18762 | 521 { |
522 curridx = i; | |
523 break; | |
524 } | |
525 } | |
526 | |
527 if(id < 0) | |
528 { | |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
529 i = (curridx + 1) % nstreams; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
530 newid = pstreams[i]; |
18762 | 531 } |
532 else | |
533 { | |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
534 for(i = 0; i < nstreams; i++) |
18762 | 535 { |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
536 if(pstreams[i] == id) |
18762 | 537 { |
538 newid = id; | |
539 break; | |
540 } | |
541 } | |
542 } | |
543 if(newid == -2 || i == curridx) | |
544 return DEMUXER_CTRL_NOTIMPL; | |
545 else | |
546 { | |
21100
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
547 ds_free_packs(ds); |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
548 priv->avfc->streams[ds->id]->discard = AVDISCARD_ALL; |
6bc989360c8b
check we aren't short of sh_videos before allocating another one
nicodvb
parents:
20990
diff
changeset
|
549 *((int*)arg) = ds->id = newid; |
18762 | 550 priv->avfc->streams[newid]->discard = AVDISCARD_NONE; |
551 return DEMUXER_CTRL_OK; | |
552 } | |
553 } | |
12164 | 554 default: |
555 return DEMUXER_CTRL_NOTIMPL; | |
556 } | |
557 } | |
558 | |
16175 | 559 static void demux_close_lavf(demuxer_t *demuxer) |
12164 | 560 { |
561 lavf_priv_t* priv = demuxer->priv; | |
562 if (priv){ | |
12304
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
563 if(priv->avfc) |
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
564 { |
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
565 av_close_input_file(priv->avfc); priv->avfc= NULL; |
434242b0706c
fix possible segfault on lavf demuxer patch by (adland <adland123 at yahoo dot com>)
michael
parents:
12168
diff
changeset
|
566 } |
12164 | 567 free(priv); demuxer->priv= NULL; |
568 } | |
569 } | |
570 | |
16175 | 571 |
572 demuxer_desc_t demuxer_desc_lavf = { | |
573 "libavformat demuxer", | |
574 "lavf", | |
575 "libavformat", | |
576 "Michael Niedermayer", | |
577 "supports many formats, requires libavformat", | |
578 DEMUXER_TYPE_LAVF, | |
579 0, // Check after other demuxer | |
580 lavf_check_file, | |
581 demux_lavf_fill_buffer, | |
582 demux_open_lavf, | |
583 demux_close_lavf, | |
584 demux_seek_lavf, | |
585 demux_lavf_control | |
586 }; | |
587 |