af_scaletempo: reset latency info when reconfiguring
[mplayer.git] / libmpdemux / aviheader.c
blob1b0683339e587d3e68ea83dec69d51cf079ce2be
1 /*
2 * This file is part of MPlayer.
4 * MPlayer 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.
9 * MPlayer 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.
14 * You should have received a copy of the GNU General Public License along
15 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <unistd.h>
22 #include <errno.h>
24 #include <libavutil/common.h>
25 #include <libavutil/intreadwrite.h>
27 #include "config.h"
28 #include "mp_msg.h"
30 #include "stream/stream.h"
31 #include "demuxer.h"
32 #include "stheader.h"
33 #include "aviprint.h"
34 #include "aviheader.h"
36 static MainAVIHeader avih;
38 static int odml_get_vstream_id(int id, unsigned char res[])
40 if ((char)(id >> 16) == 'd') {
41 if (res) {
42 res[0] = id;
43 res[1] = id >> 8;
45 return 1;
47 return 0;
50 static int avi_idx_cmp(const void *elem1, const void *elem2)
52 register off_t a = AVI_IDX_OFFSET((AVIINDEXENTRY *)elem1);
53 register off_t b = AVI_IDX_OFFSET((AVIINDEXENTRY *)elem2);
54 return (a > b) - (b > a);
57 void read_avi_header(demuxer_t *demuxer,int index_mode){
58 sh_audio_t *sh_audio=NULL;
59 sh_video_t *sh_video=NULL;
60 int stream_id=-1;
61 int idxfix_videostream=0;
62 int idxfix_divx=0;
63 avi_priv_t* priv=demuxer->priv;
64 off_t list_end=0;
66 //---- AVI header:
67 priv->idx_size=0;
68 priv->audio_streams=0;
69 while(1){
70 int id=stream_read_dword_le(demuxer->stream);
71 unsigned chunksize,size2;
72 static int last_fccType=0;
73 static int last_fccHandler=0;
74 char* hdr=NULL;
76 if(stream_eof(demuxer->stream)) break;
77 // Imply -forceidx if -saveidx is specified
78 if (index_file_save)
79 index_mode = 2;
81 if(id==mmioFOURCC('L','I','S','T')){
82 unsigned len=stream_read_dword_le(demuxer->stream); // list size
83 id=stream_read_dword_le(demuxer->stream); // list type
84 mp_msg(MSGT_HEADER,MSGL_DBG2,"LIST %.4s len=%u\n",(char *) &id,len);
85 if(len >= 4) {
86 len -= 4;
87 list_end=stream_tell(demuxer->stream)+((len+1)&(~1));
88 } else {
89 mp_tmsg(MSGT_HEADER,MSGL_WARN,"** empty list?!\n");
90 list_end = 0;
92 mp_msg(MSGT_HEADER,MSGL_V,"list_end=0x%X\n",(int)list_end);
93 if(id==listtypeAVIMOVIE){
94 // found MOVI header
95 if(!demuxer->movi_start) demuxer->movi_start=stream_tell(demuxer->stream);
96 demuxer->movi_end=stream_tell(demuxer->stream)+len;
97 mp_tmsg(MSGT_HEADER,MSGL_V,"Found movie at 0x%X - 0x%X\n",(int)demuxer->movi_start,(int)demuxer->movi_end);
98 if(demuxer->stream->end_pos>demuxer->movi_end) demuxer->movi_end=demuxer->stream->end_pos;
99 if(index_mode==-2 || index_mode==2 || index_mode==0)
100 break; // reading from non-seekable source (stdin) or forced index or no index forced
101 if(list_end>0) stream_seek(demuxer->stream,list_end); // skip movi
102 list_end=0;
104 continue;
106 size2=stream_read_dword_le(demuxer->stream);
107 mp_msg(MSGT_HEADER,MSGL_DBG2,"CHUNK %.4s len=%u\n",(char *) &id,size2);
108 chunksize=(size2+1)&(~1);
109 switch(id){
111 // Indicates where the subject of the file is archived
112 case mmioFOURCC('I','A','R','L'): hdr="Archival Location";break;
113 // Lists the artist of the original subject of the file;
114 // for example, "Michaelangelo."
115 case mmioFOURCC('I','A','R','T'): hdr="Artist";break;
116 // Lists the name of the person or organization that commissioned
117 // the subject of the file; for example "Pope Julian II."
118 case mmioFOURCC('I','C','M','S'): hdr="Commissioned";break;
119 // Provides general comments about the file or the subject
120 // of the file. If the comment is several sentences long, end each
121 // sentence with a period. Do not include new-line characters.
122 case mmioFOURCC('I','C','M','T'): hdr="Comments";break;
123 // Records the copyright information for the file; for example,
124 // "Copyright Encyclopedia International 1991." If there are multiple
125 // copyrights, separate them by semicolon followed by a space.
126 case mmioFOURCC('I','C','O','P'): hdr="Copyright";break;
127 // Describes whether an image has been cropped and, if so, how it
128 // was cropped; for example, "lower-right corner."
129 case mmioFOURCC('I','C','R','D'): hdr="Creation Date";break;
130 // Describes whether an image has been cropped and, if so, how it
131 // was cropped; for example, "lower-right corner."
132 case mmioFOURCC('I','C','R','P'): hdr="Cropped";break;
133 // Specifies the size of the original subject of the file; for
134 // example, "8.5 in h, 11 in w."
135 case mmioFOURCC('I','D','I','M'): hdr="Dimensions";break;
136 // Stores dots per inch setting of the digitizer used to
137 // produce the file, such as "300."
138 case mmioFOURCC('I','D','P','I'): hdr="Dots Per Inch";break;
139 // Stores the of the engineer who worked on the file. If there are
140 // multiple engineers, separate the names by a semicolon and a blank;
141 // for example, "Smith, John; Adams, Joe."
142 case mmioFOURCC('I','E','N','G'): hdr="Engineer";break;
143 // Describes the original work, such as "landscape,", "portrait,"
144 // "still liefe," etc.
145 case mmioFOURCC('I','G','N','R'): hdr="Genre";break;
146 // Provides a list of keywords that refer to the file or subject of the
147 // file. Separate multiple keywords with a semicolon and a blank;
148 // for example, "Seattle, aerial view; scenery."
149 case mmioFOURCC('I','K','E','Y'): hdr="Keywords";break;
150 // ILGT - Describes the changes in the lightness settings on the digitizer
151 // required to produce the file. Note that the format of this information
152 // depends on the hardware used.
153 case mmioFOURCC('I','L','G','T'): hdr="Lightness";break;
154 // IMED - Decribes the original subject of the file, such as
155 // "computer image," "drawing," "lithograph," and so on.
156 case mmioFOURCC('I','M','E','D'): hdr="Medium";break;
157 // INAM - Stores the title of the subject of the file, such as
158 // "Seattle from Above."
159 case mmioFOURCC('I','N','A','M'): hdr="Title";break;
160 // IPLT - Specifies the number of colors requested when digitizing
161 // an image, such as "256."
162 case mmioFOURCC('I','P','L','T'): hdr="Palette Setting";break;
163 // IPRD - Specifies the name of title the file was originally intended
164 // for, such as "Encyclopedia of Pacific Northwest Geography."
165 case mmioFOURCC('I','P','R','D'): hdr="Product";break;
166 // ISBJ - Decsribes the contents of the file, such as
167 // "Aerial view of Seattle."
168 case mmioFOURCC('I','S','B','J'): hdr="Subject";break;
169 // ISFT - Identifies the name of the software packages used to create the
170 // file, such as "Microsoft WaveEdit"
171 case mmioFOURCC('I','S','F','T'): hdr="Software";break;
172 // ISHP - Identifies the change in sharpness for the digitizer
173 // required to produce the file (the format depends on the hardware used).
174 case mmioFOURCC('I','S','H','P'): hdr="Sharpness";break;
175 // ISRC - Identifies the name of the person or organization who
176 // supplied the original subject of the file; for example, "Try Research."
177 case mmioFOURCC('I','S','R','C'): hdr="Source";break;
178 // ISRF - Identifies the original form of the material that was digitized,
179 // such as "slide," "paper," "map," and so on. This is not necessarily
180 // the same as IMED
181 case mmioFOURCC('I','S','R','F'): hdr="Source Form";break;
182 // ITCH - Identifies the technician who digitized the subject file;
183 // for example, "Smith, John."
184 case mmioFOURCC('I','T','C','H'): hdr="Technician";break;
185 case mmioFOURCC('I','S','M','P'): hdr="Time Code";break;
186 case mmioFOURCC('I','D','I','T'): hdr="Digitization Time";break;
188 case ckidAVIMAINHDR: // read 'avih'
189 stream_read(demuxer->stream,(char*) &avih,FFMIN(size2,sizeof(avih)));
190 le2me_MainAVIHeader(&avih); // swap to machine endian
191 chunksize-=FFMIN(size2,sizeof(avih));
192 if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_avih(&avih,MSGL_V); // else print_avih_flags(&avih,MSGL_V);
193 break;
194 case ckidSTREAMHEADER: { // read 'strh'
195 AVIStreamHeader h;
196 stream_read(demuxer->stream,(char*) &h,FFMIN(size2,sizeof(h)));
197 le2me_AVIStreamHeader(&h); // swap to machine endian
198 chunksize-=FFMIN(size2,sizeof(h));
199 ++stream_id;
200 if(h.fccType==streamtypeVIDEO){
201 sh_video=new_sh_video(demuxer,stream_id);
202 mp_tmsg(MSGT_DEMUX, MSGL_INFO, "[%s] Video stream found, -vid %d\n", "aviheader", stream_id);
203 memcpy(&sh_video->video,&h,sizeof(h));
204 sh_video->stream_delay = (float)sh_video->video.dwStart * sh_video->video.dwScale/sh_video->video.dwRate;
205 } else
206 if(h.fccType==streamtypeAUDIO){
207 sh_audio=new_sh_audio(demuxer,stream_id);
208 mp_tmsg(MSGT_DEMUX, MSGL_INFO, "[%s] Audio stream found, -aid %d\n", "aviheader", stream_id);
209 memcpy(&sh_audio->audio,&h,sizeof(h));
210 sh_audio->stream_delay = (float)sh_audio->audio.dwStart * sh_audio->audio.dwScale/sh_audio->audio.dwRate;
211 sh_audio->needs_parsing = 1;
213 last_fccType=h.fccType;
214 last_fccHandler=h.fccHandler;
215 if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_strh(&h,MSGL_V);
216 break; }
217 case mmioFOURCC('i', 'n', 'd', 'x'): {
218 uint32_t i;
219 avisuperindex_chunk *s;
221 if(!index_mode) break;
223 if(chunksize<=24){
224 break;
226 priv->suidx_size++;
227 priv->suidx = realloc_struct(priv->suidx, priv->suidx_size, sizeof (avisuperindex_chunk));
228 s = &priv->suidx[priv->suidx_size-1];
230 chunksize-=24;
231 memcpy(s->fcc, "indx", 4);
232 s->dwSize = size2;
233 s->wLongsPerEntry = stream_read_word_le(demuxer->stream);
234 s->bIndexSubType = stream_read_char(demuxer->stream);
235 s->bIndexType = stream_read_char(demuxer->stream);
236 s->nEntriesInUse = stream_read_dword_le(demuxer->stream);
237 AV_WN32(s->dwChunkId, stream_read_dword_le(demuxer->stream));
238 stream_read(demuxer->stream, (char *)s->dwReserved, 3*4);
239 memset(s->dwReserved, 0, 3*4);
241 print_avisuperindex_chunk(s,MSGL_V);
243 // Check and fix this useless crap
244 if(s->wLongsPerEntry != sizeof (avisuperindex_entry)/4) {
245 mp_msg (MSGT_HEADER, MSGL_WARN, "Broken super index chunk size: %u\n",s->wLongsPerEntry);
246 s->wLongsPerEntry = sizeof(avisuperindex_entry)/4;
248 if( ((chunksize/4)/s->wLongsPerEntry) < s->nEntriesInUse){
249 mp_msg (MSGT_HEADER, MSGL_WARN, "Broken super index chunk\n");
250 s->nEntriesInUse = (chunksize/4)/s->wLongsPerEntry;
253 s->aIndex = calloc(s->nEntriesInUse, sizeof (avisuperindex_entry));
254 s->stdidx = calloc(s->nEntriesInUse, sizeof (avistdindex_chunk));
256 // now the real index of indices
257 for (i=0; i<s->nEntriesInUse; i++) {
258 chunksize-=16;
259 s->aIndex[i].qwOffset = stream_read_qword_le(demuxer->stream);
260 s->aIndex[i].dwSize = stream_read_dword_le(demuxer->stream);
261 s->aIndex[i].dwDuration = stream_read_dword_le(demuxer->stream);
262 mp_msg (MSGT_HEADER, MSGL_V, "ODML (%.4s): [%d] 0x%016"PRIx64" 0x%04x %u\n",
263 s->dwChunkId, i,
264 (uint64_t)s->aIndex[i].qwOffset, s->aIndex[i].dwSize, s->aIndex[i].dwDuration);
267 break; }
268 case ckidSTREAMFORMAT: { // read 'strf'
269 if(last_fccType==streamtypeVIDEO){
270 sh_video->bih=calloc(FFMAX(chunksize, sizeof(*sh_video->bih)), 1);
271 // sh_video->bih=malloc(chunksize); memset(sh_video->bih,0,chunksize);
272 mp_tmsg(MSGT_HEADER,MSGL_V,"Found 'bih', %u bytes of %zu\n",chunksize,sizeof(*sh_video->bih));
273 stream_read(demuxer->stream,(char*) sh_video->bih,chunksize);
274 le2me_BITMAPINFOHEADER(sh_video->bih); // swap to machine endian
275 if (sh_video->bih->biSize > chunksize && sh_video->bih->biSize > sizeof(*sh_video->bih))
276 sh_video->bih->biSize = chunksize;
277 // fixup MS-RLE header (seems to be broken for <256 color files)
278 if(sh_video->bih->biCompression<=1 && sh_video->bih->biSize==40)
279 sh_video->bih->biSize=chunksize;
280 if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_video_header(sh_video->bih,MSGL_V);
281 chunksize=0;
282 sh_video->fps=(float)sh_video->video.dwRate/(float)sh_video->video.dwScale;
283 sh_video->frametime=(float)sh_video->video.dwScale/(float)sh_video->video.dwRate;
284 sh_video->format = sh_video->bih->biCompression;
285 // if(demuxer->video->id==-1) demuxer->video->id=stream_id;
286 // IdxFix:
287 idxfix_videostream=stream_id;
288 switch(sh_video->bih->biCompression){
289 case mmioFOURCC('M', 'P', 'G', '4'):
290 case mmioFOURCC('m', 'p', 'g', '4'):
291 case mmioFOURCC('D', 'I', 'V', '1'):
292 idxfix_divx=3; // set index recovery mpeg4 flavour: msmpeg4v1
293 mp_tmsg(MSGT_HEADER,MSGL_V,"Regenerating keyframe table for M$ mpg4v1 video.\n");
294 break;
295 case mmioFOURCC('D', 'I', 'V', '3'):
296 case mmioFOURCC('d', 'i', 'v', '3'):
297 case mmioFOURCC('D', 'I', 'V', '4'):
298 case mmioFOURCC('d', 'i', 'v', '4'):
299 case mmioFOURCC('D', 'I', 'V', '5'):
300 case mmioFOURCC('d', 'i', 'v', '5'):
301 case mmioFOURCC('D', 'I', 'V', '6'):
302 case mmioFOURCC('d', 'i', 'v', '6'):
303 case mmioFOURCC('M', 'P', '4', '3'):
304 case mmioFOURCC('m', 'p', '4', '3'):
305 case mmioFOURCC('M', 'P', '4', '2'):
306 case mmioFOURCC('m', 'p', '4', '2'):
307 case mmioFOURCC('D', 'I', 'V', '2'):
308 case mmioFOURCC('A', 'P', '4', '1'):
309 idxfix_divx=1; // set index recovery mpeg4 flavour: msmpeg4v3
310 mp_tmsg(MSGT_HEADER,MSGL_V,"Regenerating keyframe table for DIVX3 video.\n");
311 break;
312 case mmioFOURCC('D', 'I', 'V', 'X'):
313 case mmioFOURCC('d', 'i', 'v', 'x'):
314 case mmioFOURCC('D', 'X', '5', '0'):
315 case mmioFOURCC('X', 'V', 'I', 'D'):
316 case mmioFOURCC('x', 'v', 'i', 'd'):
317 case mmioFOURCC('F', 'M', 'P', '4'):
318 case mmioFOURCC('f', 'm', 'p', '4'):
319 idxfix_divx=2; // set index recovery mpeg4 flavour: generic mpeg4
320 mp_tmsg(MSGT_HEADER,MSGL_V,"Regenerating keyframe table for MPEG-4 video.\n");
321 break;
323 } else
324 if(last_fccType==streamtypeAUDIO){
325 unsigned wf_size = chunksize<sizeof(*sh_audio->wf)?sizeof(*sh_audio->wf):chunksize;
326 sh_audio->wf=calloc(wf_size,1);
327 // sh_audio->wf=malloc(chunksize); memset(sh_audio->wf,0,chunksize);
328 mp_tmsg(MSGT_HEADER,MSGL_V,"Found 'wf', %d bytes of %zu\n",chunksize,sizeof(*sh_audio->wf));
329 stream_read(demuxer->stream,(char*) sh_audio->wf,chunksize);
330 le2me_WAVEFORMATEX(sh_audio->wf);
331 if (sh_audio->wf->cbSize != 0 &&
332 wf_size < sizeof(*sh_audio->wf)+sh_audio->wf->cbSize) {
333 sh_audio->wf=realloc(sh_audio->wf, sizeof(*sh_audio->wf)+sh_audio->wf->cbSize);
335 sh_audio->format=sh_audio->wf->wFormatTag;
336 if (sh_audio->wf->wFormatTag == 0xfffe && sh_audio->wf->cbSize >= 22)
337 sh_audio->format = le2me_16(((WAVEFORMATEXTENSIBLE *)sh_audio->wf)->SubFormat);
338 if (sh_audio->format == 1 &&
339 last_fccHandler == mmioFOURCC('A', 'x', 'a', 'n'))
340 sh_audio->format = last_fccHandler;
341 sh_audio->i_bps=sh_audio->wf->nAvgBytesPerSec;
342 chunksize=0;
343 if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_wave_header(sh_audio->wf,MSGL_V);
344 ++priv->audio_streams;
345 // if(demuxer->audio->id==-1) demuxer->audio->id=stream_id;
347 break;
349 case mmioFOURCC('v', 'p', 'r', 'p'): {
350 VideoPropHeader *vprp = malloc(chunksize);
351 unsigned int i;
352 stream_read(demuxer->stream, (void*)vprp, chunksize);
353 le2me_VideoPropHeader(vprp);
354 chunksize -= sizeof(*vprp)-sizeof(vprp->FieldInfo);
355 chunksize /= sizeof(VIDEO_FIELD_DESC);
356 if (vprp->nbFieldPerFrame > chunksize) {
357 vprp->nbFieldPerFrame = chunksize;
359 chunksize = 0;
360 for (i=0; i<vprp->nbFieldPerFrame; i++) {
361 le2me_VIDEO_FIELD_DESC(&vprp->FieldInfo[i]);
363 if (sh_video) {
364 sh_video->aspect = GET_AVI_ASPECT(vprp->dwFrameAspectRatio);
366 if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_vprp(vprp,MSGL_V);
367 free(vprp);
368 break;
370 case mmioFOURCC('d', 'm', 'l', 'h'): {
371 // dmlh 00 00 00 04 frms
372 unsigned int total_frames = stream_read_dword_le(demuxer->stream);
373 mp_tmsg(MSGT_HEADER,MSGL_V,"AVI: dmlh found (size=%d) (total_frames=%d)\n", chunksize, total_frames);
374 stream_skip(demuxer->stream, chunksize-4);
375 chunksize = 0;
377 break;
378 case ckidAVINEWINDEX:
379 if(demuxer->movi_end>stream_tell(demuxer->stream))
380 demuxer->movi_end=stream_tell(demuxer->stream); // fixup movi-end
381 if(index_mode && !priv->isodml){
382 int read;
383 int i;
384 priv->idx_size=size2>>4;
385 mp_tmsg(MSGT_HEADER,MSGL_V,"Reading INDEX block, %d chunks for %d frames (fpos=%"PRId64").\n",
386 priv->idx_size,avih.dwTotalFrames, (int64_t)stream_tell(demuxer->stream));
387 priv->idx=malloc(priv->idx_size<<4);
388 // printf("\nindex to %p !!!!! (priv=%p)\n",priv->idx,priv);
389 read = stream_read(demuxer->stream,(char*)priv->idx,priv->idx_size<<4);
390 priv->idx_size = FFMAX(read, 0) >> 4;
391 for (i = 0; i < priv->idx_size; i++) { // swap index to machine endian
392 AVIINDEXENTRY *entry=(AVIINDEXENTRY*)priv->idx + i;
393 le2me_AVIINDEXENTRY(entry);
395 * We (ab)use the upper word for bits 32-47 of the offset, so
396 * we'll clear them here.
397 * FIXME: AFAIK no codec uses them, but if one does it will break
399 entry->dwFlags&=0xffff;
401 chunksize-=priv->idx_size<<4;
402 if( mp_msg_test(MSGT_HEADER,MSGL_DBG2) ) print_index(priv->idx,priv->idx_size,MSGL_DBG2);
404 break;
405 /* added May 2002 */
406 case mmioFOURCC('R','I','F','F'): {
407 char riff_type[4];
409 mp_tmsg(MSGT_HEADER, MSGL_V, "Additional RIFF header...\n");
410 stream_read(demuxer->stream, riff_type, sizeof riff_type);
411 if (strncmp(riff_type, "AVIX", sizeof riff_type))
412 mp_tmsg(MSGT_HEADER, MSGL_WARN, "** Warning: this is no extended AVI header..\n");
413 else {
415 * We got an extended AVI header, so we need to switch to
416 * ODML to get seeking to work, provided we got indx chunks
417 * in the header (suidx_size > 0).
419 if (priv->suidx_size > 0)
420 priv->isodml = 1;
422 chunksize = 0;
423 list_end = 0; /* a new list will follow */
424 break; }
425 case ckidAVIPADDING:
426 stream_skip(demuxer->stream, chunksize);
427 chunksize = 0;
428 break;
430 if(hdr){
431 mp_msg(MSGT_HEADER,MSGL_V,"hdr=%s size=%u\n",hdr,size2);
432 if(size2==3)
433 chunksize=1; // empty
434 else {
435 char buf[256];
436 int len=(size2<250)?size2:250;
437 stream_read(demuxer->stream,buf,len);
438 chunksize-=len;
439 buf[len]=0;
440 mp_msg(MSGT_HEADER,MSGL_V,"%-10s: %s\n",hdr,buf);
441 demux_info_add(demuxer, hdr, buf);
444 mp_msg(MSGT_HEADER,MSGL_DBG2,"list_end=0x%"PRIX64" pos=0x%"PRIX64" chunksize=0x%"PRIX64" next=0x%"PRIX64"\n",
445 (int64_t)list_end, (int64_t)stream_tell(demuxer->stream),
446 (int64_t)chunksize, (int64_t)chunksize+(int64_t)stream_tell(demuxer->stream));
447 if(list_end>0 &&
448 chunksize+stream_tell(demuxer->stream) == list_end) list_end=0;
449 if(list_end>0 && chunksize+stream_tell(demuxer->stream)>list_end){
450 mp_tmsg(MSGT_HEADER,MSGL_V,"Broken chunk? chunksize=%d (id=%.4s)\n",chunksize,(char *) &id);
451 stream_seek(demuxer->stream,list_end);
452 list_end=0;
453 } else
454 if(chunksize>0) stream_skip(demuxer->stream,chunksize); else
455 if((int)chunksize<0) mp_msg(MSGT_HEADER,MSGL_WARN,"chunksize=%u (id=%.4s)\n",chunksize,(char *) &id);
459 if (priv->suidx_size > 0 && priv->idx_size == 0) {
461 * No NEWAVIINDEX, but we got an OpenDML index.
463 priv->isodml = 1;
466 if (priv->isodml && (index_mode==-1 || index_mode==0 || index_mode==1)) {
467 int i, j, k;
469 avisuperindex_chunk *cx;
470 AVIINDEXENTRY *idx;
473 if (priv->idx_size) free(priv->idx);
474 priv->idx_size = 0;
475 priv->idx_offset = 0;
476 priv->idx = NULL;
478 mp_tmsg(MSGT_HEADER, MSGL_INFO, "AVI: ODML: Building ODML index (%d superindexchunks).\n", priv->suidx_size);
480 // read the standard indices
481 for (cx = &priv->suidx[0], i=0; i<priv->suidx_size; cx++, i++) {
482 stream_reset(demuxer->stream);
483 for (j=0; j<cx->nEntriesInUse; j++) {
484 int ret1, ret2;
485 memset(&cx->stdidx[j], 0, 32);
486 ret1 = stream_seek(demuxer->stream, (off_t)cx->aIndex[j].qwOffset);
487 ret2 = stream_read(demuxer->stream, (char *)&cx->stdidx[j], 32);
488 if (ret1 != 1 || ret2 != 32 || cx->stdidx[j].nEntriesInUse==0) {
489 // this is a broken file (probably incomplete) let the standard
490 // gen_index routine handle this
491 priv->isodml = 0;
492 priv->idx_size = 0;
493 mp_tmsg(MSGT_HEADER, MSGL_WARN, "AVI: ODML: Broken (incomplete?) file detected. Will use traditional index.\n");
494 goto freeout;
497 le2me_AVISTDIDXCHUNK(&cx->stdidx[j]);
498 print_avistdindex_chunk(&cx->stdidx[j],MSGL_V);
499 priv->idx_size += cx->stdidx[j].nEntriesInUse;
500 cx->stdidx[j].aIndex = malloc(cx->stdidx[j].nEntriesInUse*sizeof(avistdindex_entry));
501 stream_read(demuxer->stream, (char *)cx->stdidx[j].aIndex,
502 cx->stdidx[j].nEntriesInUse*sizeof(avistdindex_entry));
503 for (k=0;k<cx->stdidx[j].nEntriesInUse; k++)
504 le2me_AVISTDIDXENTRY(&cx->stdidx[j].aIndex[k]);
506 cx->stdidx[j].dwReserved3 = 0;
512 * We convert the index by translating all entries into AVIINDEXENTRYs
513 * and sorting them by offset. The result should be the same index
514 * we would get with -forceidx.
517 idx = priv->idx = malloc(priv->idx_size * sizeof (AVIINDEXENTRY));
519 for (cx = priv->suidx; cx != &priv->suidx[priv->suidx_size]; cx++) {
520 avistdindex_chunk *sic;
521 for (sic = cx->stdidx; sic != &cx->stdidx[cx->nEntriesInUse]; sic++) {
522 avistdindex_entry *sie;
523 for (sie = sic->aIndex; sie != &sic->aIndex[sic->nEntriesInUse]; sie++) {
524 uint64_t off = sic->qwBaseOffset + sie->dwOffset - 8;
525 memcpy(&idx->ckid, sic->dwChunkId, 4);
526 idx->dwChunkOffset = off;
527 idx->dwFlags = (off >> 32) << 16;
528 idx->dwChunkLength = sie->dwSize & 0x7fffffff;
529 idx->dwFlags |= (sie->dwSize&0x80000000)?0x0:AVIIF_KEYFRAME; // bit 31 denotes !keyframe
530 idx++;
534 qsort(priv->idx, priv->idx_size, sizeof(AVIINDEXENTRY), avi_idx_cmp);
537 Hack to work around a "wrong" index in some divx odml files
538 (processor_burning.avi as an example)
539 They have ##dc on non keyframes but the ix00 tells us they are ##db.
540 Read the fcc of a non-keyframe vid frame and check it.
544 uint32_t id;
545 uint32_t db = 0;
546 stream_reset (demuxer->stream);
548 // find out the video stream id. I have seen files with 01db.
549 for (idx = &((AVIINDEXENTRY *)priv->idx)[0], i=0; i<priv->idx_size; i++, idx++){
550 unsigned char res[2];
551 if (odml_get_vstream_id(idx->ckid, res)) {
552 db = mmioFOURCC(res[0], res[1], 'd', 'b');
553 break;
557 // find first non keyframe
558 for (idx = &((AVIINDEXENTRY *)priv->idx)[0], i=0; i<priv->idx_size; i++, idx++){
559 if (!(idx->dwFlags & AVIIF_KEYFRAME) && idx->ckid == db) break;
561 if (i<priv->idx_size && db) {
562 stream_seek(demuxer->stream, AVI_IDX_OFFSET(idx));
563 id = stream_read_dword_le(demuxer->stream);
564 if (id && id != db) // index fcc and real fcc differ? fix it.
565 for (idx = &((AVIINDEXENTRY *)priv->idx)[0], i=0; i<priv->idx_size; i++, idx++){
566 if (!(idx->dwFlags & AVIIF_KEYFRAME) && idx->ckid == db)
567 idx->ckid = id;
572 if ( mp_msg_test(MSGT_HEADER,MSGL_DBG2) ) print_index(priv->idx, priv->idx_size,MSGL_DBG2);
574 demuxer->movi_end=demuxer->stream->end_pos;
576 freeout:
578 // free unneeded stuff
579 cx = &priv->suidx[0];
580 do {
581 for (j=0;j<cx->nEntriesInUse;j++)
582 if (cx->stdidx[j].nEntriesInUse) free(cx->stdidx[j].aIndex);
583 free(cx->stdidx);
585 } while (cx++ != &priv->suidx[priv->suidx_size-1]);
586 free(priv->suidx);
590 /* Read a saved index file */
591 if (index_file_load) {
592 FILE *fp;
593 char magic[7];
594 unsigned int i;
596 if ((fp = fopen(index_file_load, "r")) == NULL) {
597 mp_tmsg(MSGT_HEADER,MSGL_ERR, "Can't read index file %s: %s\n", index_file_load, strerror(errno));
598 goto gen_index;
600 fread(&magic, 6, 1, fp);
601 if (strncmp(magic, "MPIDX1", 6)) {
602 mp_tmsg(MSGT_HEADER,MSGL_ERR, "%s is not a valid MPlayer index file.\n", index_file_load);
603 goto gen_index;
605 fread(&priv->idx_size, sizeof(priv->idx_size), 1, fp);
606 priv->idx=malloc(priv->idx_size*sizeof(AVIINDEXENTRY));
607 if (!priv->idx) {
608 mp_tmsg(MSGT_HEADER,MSGL_ERR, "Could not allocate memory for index data from %s.\n", index_file_load);
609 priv->idx_size = 0;
610 goto gen_index;
613 for (i=0; i<priv->idx_size;i++) {
614 AVIINDEXENTRY *idx;
615 idx=&((AVIINDEXENTRY *)priv->idx)[i];
616 fread(idx, sizeof(AVIINDEXENTRY), 1, fp);
617 if (feof(fp)) {
618 mp_tmsg(MSGT_HEADER,MSGL_ERR, "premature end of index file %s\n", index_file_load);
619 free(priv->idx);
620 priv->idx_size = 0;
621 goto gen_index;
624 fclose(fp);
625 mp_tmsg(MSGT_HEADER,MSGL_INFO, "Loaded index file: %s\n", index_file_load);
627 gen_index:
628 if(index_mode>=2 || (priv->idx_size==0 && index_mode==1)){
629 int idx_pos = 0;
630 // build index for file:
631 stream_reset(demuxer->stream);
632 stream_seek(demuxer->stream,demuxer->movi_start);
634 priv->idx_size=0;
635 priv->idx=NULL;
637 while(1){
638 int id;
639 unsigned len;
640 off_t skip;
641 AVIINDEXENTRY* idx;
642 unsigned int c;
643 demuxer->filepos=stream_tell(demuxer->stream);
644 if(demuxer->filepos>=demuxer->movi_end && demuxer->movi_start<demuxer->movi_end) break;
645 id=stream_read_dword_le(demuxer->stream);
646 len=stream_read_dword_le(demuxer->stream);
647 if(id==mmioFOURCC('L','I','S','T') || id==mmioFOURCC('R', 'I', 'F', 'F')){
648 id=stream_read_dword_le(demuxer->stream); // list or RIFF type
649 continue;
651 if(stream_eof(demuxer->stream)) break;
652 if(!id || avi_stream_id(id)==100) goto skip_chunk; // bad ID (or padding?)
654 if(idx_pos>=priv->idx_size){
655 // priv->idx_size+=32;
656 priv->idx_size+=1024; // +16kB
657 priv->idx=realloc(priv->idx,priv->idx_size*sizeof(AVIINDEXENTRY));
658 if(!priv->idx){idx_pos=0; break;} // error!
660 idx=&((AVIINDEXENTRY *)priv->idx)[idx_pos++];
661 idx->ckid=id;
662 idx->dwFlags=AVIIF_KEYFRAME; // FIXME
663 idx->dwFlags|=(demuxer->filepos>>16)&0xffff0000U;
664 idx->dwChunkOffset=(unsigned long)demuxer->filepos;
665 idx->dwChunkLength=len;
667 c=stream_read_dword(demuxer->stream);
669 if(!len) idx->dwFlags&=~AVIIF_KEYFRAME;
671 // Fix keyframes for DivX files:
672 if(idxfix_divx)
673 if(avi_stream_id(id)==idxfix_videostream){
674 switch(idxfix_divx){
675 case 3: c=stream_read_dword(demuxer->stream)<<5; //skip 32+5 bits for m$mpeg4v1
676 case 1: if(c&0x40000000) idx->dwFlags&=~AVIIF_KEYFRAME;break; // divx 3
677 case 2: if(c==0x1B6) idx->dwFlags&=~AVIIF_KEYFRAME;break; // divx 4
681 // update status line:
682 { static off_t lastpos;
683 off_t pos;
684 off_t len=demuxer->movi_end-demuxer->movi_start;
685 if(len){
686 pos=100*(demuxer->filepos-demuxer->movi_start)/len; // %
687 } else {
688 pos=(demuxer->filepos-demuxer->movi_start)>>20; // MB
690 if(pos!=lastpos){
691 lastpos=pos;
692 mp_tmsg(MSGT_HEADER,MSGL_STATUS, "Generating Index: %3lu %s \r",
693 (unsigned long)pos, len?"%":"MB");
696 mp_dbg(MSGT_HEADER,MSGL_DBG2,"%08X %08X %.4s %08X %X\n",(unsigned int)demuxer->filepos,id,(char *) &id,(int)c,(unsigned int) idx->dwFlags);
697 #if 0
698 { unsigned char tmp[64];
699 int i;
700 stream_read(demuxer->stream,tmp,64);
701 printf("%.4s",&id);
702 for(i=0;i<64;i++) printf(" %02X",tmp[i]);
703 printf("\n");
705 #endif
706 skip_chunk:
707 skip=(len+1)&(~1UL); // total bytes in this chunk
708 stream_seek(demuxer->stream,8+demuxer->filepos+skip);
710 priv->idx_size=idx_pos;
711 mp_tmsg(MSGT_HEADER,MSGL_INFO,"AVI: Generated index table for %d chunks!\n",priv->idx_size);
712 if( mp_msg_test(MSGT_HEADER,MSGL_DBG2) ) print_index(priv->idx,priv->idx_size,MSGL_DBG2);
714 /* Write generated index to a file */
715 if (index_file_save) {
716 FILE *fp;
717 unsigned int i;
719 if ((fp=fopen(index_file_save, "w")) == NULL) {
720 mp_tmsg(MSGT_HEADER,MSGL_ERR, "Couldn't write index file %s: %s\n", index_file_save, strerror(errno));
721 return;
723 fwrite("MPIDX1", 6, 1, fp);
724 fwrite(&priv->idx_size, sizeof(priv->idx_size), 1, fp);
725 for (i=0; i<priv->idx_size; i++) {
726 AVIINDEXENTRY *idx = &((AVIINDEXENTRY *)priv->idx)[i];
727 fwrite(idx, sizeof(AVIINDEXENTRY), 1, fp);
729 fclose(fp);
730 mp_tmsg(MSGT_HEADER,MSGL_INFO, "Saved index file: %s\n", index_file_save);