e60d71dc601fea572b2e7acfded73d5816fc3c5c
[vlc.git] / modules / demux / avi / libavi.c
1 /*****************************************************************************
2  * libavi.c : 
3  *****************************************************************************
4  * Copyright (C) 2001 VideoLAN
5  * $Id: libavi.c,v 1.9 2002/12/06 16:34:06 sam Exp $
6  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
7  * 
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  * 
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
21  *****************************************************************************/
22
23 #include <stdlib.h>                                      /* malloc(), free() */
24 #include <string.h>                                              /* strdup() */
25 #include <errno.h>
26 #include <sys/types.h>
27
28 #include <vlc/vlc.h>
29 #include <vlc/input.h>
30
31 #include "libavi.h"
32
33 #define AVI_DEBUG 1
34 #define AVIFOURCC_PRINT( x ) \
35     (x)&0xff,           \
36     ( (x) >>  8 )&0xff, \
37     ( (x) >> 16 )&0xff, \
38     ( (x) >> 24 )&0xff
39
40 #define FREE( p ) \
41     if( p ) {free( p ); p = NULL; }
42
43 #define __EVEN( x ) ( (x)&0x01 ? (x)+1 : (x) )
44   
45 /* Some functions to manipulate memory */
46 static uint16_t GetWLE( uint8_t *p_buff )
47 {
48     return (uint16_t)p_buff[0] | ( ((uint16_t)p_buff[1]) << 8 );
49 }
50
51 static uint32_t GetDWLE( uint8_t *p_buff )
52 {
53     return (uint32_t)p_buff[0] | ( ((uint32_t)p_buff[1]) << 8 ) |
54             ( ((uint32_t)p_buff[2]) << 16 ) | ( ((uint32_t)p_buff[3]) << 24 );
55 }
56 static vlc_fourcc_t GetFOURCC( byte_t *p_buff )
57 {
58     return VLC_FOURCC( p_buff[0], p_buff[1], p_buff[2], p_buff[3] );
59 }
60 /*****************************************************************************
61  * Some basic functions to manipulate stream more easily in vlc
62  *
63  * AVI_TellAbsolute get file position
64  * 
65  * AVI_SeekAbsolute seek in the file
66  *
67  * AVI_ReadData read data from the file in a buffer
68  *
69  * AVI_SkipBytes skip bytes
70  *
71  *****************************************************************************/
72 off_t AVI_TellAbsolute( input_thread_t *p_input )
73 {
74     off_t i_pos;
75     
76     vlc_mutex_lock( &p_input->stream.stream_lock );
77     
78     i_pos= p_input->stream.p_selected_area->i_tell;
79 //            - ( p_input->p_last_data - p_input->p_current_data  );
80
81     vlc_mutex_unlock( &p_input->stream.stream_lock );
82
83     return i_pos;
84 }
85  
86 int AVI_SeekAbsolute( input_thread_t *p_input,
87                       off_t i_pos)
88 {
89     off_t i_filepos;
90
91     if( i_pos >= p_input->stream.p_selected_area->i_size )
92     {
93         return VLC_EGENERIC;
94     }
95     
96     i_filepos = AVI_TellAbsolute( p_input );
97
98     if( i_filepos == i_pos )
99     {
100         return VLC_SUCCESS;
101     }
102
103     if( p_input->stream.b_seekable &&
104         p_input->stream.i_method == INPUT_METHOD_FILE )
105     {
106         p_input->pf_seek( p_input, i_pos );
107         input_AccessReinit( p_input );
108         return VLC_SUCCESS;
109     }
110     else
111     {
112         data_packet_t   *p_data;
113         int             i_skip = i_pos - i_filepos;
114         
115         msg_Warn( p_input, "will skip %d bytes, slow", i_skip );
116         if( i_skip < 0 )
117         {
118             return VLC_EGENERIC; // failed
119         }
120         while (i_skip > 0 )
121         {
122             int i_read;
123             
124             i_read = input_SplitBuffer( p_input, &p_data, 
125                                         __MIN( 4096, i_skip ) );
126             if( i_read < 0 )
127             {
128                 return VLC_EGENERIC;
129             }
130             i_skip -= i_read;
131             
132             input_DeletePacket( p_input->p_method_data, p_data );
133             if( i_read == 0 && i_skip > 0 )
134             {
135                 return VLC_EGENERIC;
136             }
137         }
138 #if 0
139         while( i_skip > 0 )
140         {
141             i_peek = input_Peek( p_input, &p_peek, i_skip+1 );
142             i_peek--;
143             i_skip -= i_peek;
144             vlc_mutex_lock( &p_input->stream.stream_lock );
145             p_input->p_current_data += i_peek;  // skip them
146             vlc_mutex_unlock( &p_input->stream.stream_lock );
147             if( i_peek <= 0 )
148             {
149                 return VLC_EGENERIC;
150             }
151         }
152 #endif
153         return VLC_SUCCESS;
154     }
155 }
156
157 /* return amount read if success, 0 if failed */
158 int AVI_ReadData( input_thread_t *p_input, uint8_t *p_buff, int i_size )
159 {
160     data_packet_t *p_data;
161
162     int i_count;
163     int i_read = 0;
164
165     
166     if( !i_size )
167     {
168         return 0;
169     }
170
171     do
172     {
173         i_count = input_SplitBuffer(p_input, &p_data, __MIN( i_size, 1024 ) );
174         if( i_count <= 0 )
175         {
176             return i_read;
177         }
178         memcpy( p_buff, p_data->p_payload_start, i_count );
179         input_DeletePacket( p_input->p_method_data, p_data );
180         
181         p_buff += i_count;
182         i_size -= i_count;
183         i_read += i_count;
184                 
185     } while( i_size );
186     
187     return i_read;
188 }
189
190 int  AVI_SkipBytes( input_thread_t *p_input, int i_count )
191 {
192     /* broken with new use of i_tell */
193 #if 0
194     int i_buff_size;
195     vlc_mutex_lock( &p_input->stream.stream_lock );
196     i_buff_size = p_input->p_last_data - p_input->p_current_data;
197     vlc_mutex_unlock( &p_input->stream.stream_lock );
198             
199     if( i_count > 0 && i_count + 1 < i_buff_size )
200     {
201         uint8_t *p_peek;
202         
203         input_Peek( p_input, &p_peek, i_count + 1 );
204
205         vlc_mutex_lock( &p_input->stream.stream_lock );
206         p_input->p_current_data += i_count;  // skip them
207         vlc_mutex_unlock( &p_input->stream.stream_lock );
208         return VLC_SUCCESS;
209     }
210     else
211 #endif
212     {
213         return AVI_SeekAbsolute( p_input, 
214                                  AVI_TellAbsolute( p_input ) + i_count );
215     }
216 }
217
218 /*****************************************************************************
219  *
220  * AVI_TestFile: look at first bytes to see if it's a valid avi file
221  * 
222  * unseekable: ok
223  *
224  *****************************************************************************/
225 int AVI_TestFile( input_thread_t *p_input )
226 {
227     uint8_t  *p_peek;
228     
229     if( input_Peek( p_input, &p_peek, 8 ) < 8 )
230     {
231         msg_Err( p_input, "cannot peek()" );
232         return VLC_EGENERIC;
233     }
234
235     if( GetFOURCC( p_peek ) == AVIFOURCC_RIFF && 
236         GetFOURCC( p_peek + 8 ) == AVIFOURCC_AVI )
237     {
238         return VLC_SUCCESS;
239     }
240     else
241     {
242         return VLC_EGENERIC;
243     }
244 }
245 /****************************************************************************
246  *
247  * Basics functions to manipulates chunks
248  *
249  ****************************************************************************/
250 static int AVI_ChunkReadCommon( input_thread_t *p_input,
251                                 avi_chunk_t *p_chk )
252 {
253     uint8_t  *p_peek;
254     int i_peek;
255
256     memset( p_chk, 0, sizeof( avi_chunk_t ) );
257
258     if( ( i_peek = input_Peek( p_input, &p_peek, 8 ) ) < 8 )
259     {
260         return VLC_EGENERIC;
261     }
262     
263     p_chk->common.i_chunk_fourcc = GetFOURCC( p_peek );
264     p_chk->common.i_chunk_size   = GetDWLE( p_peek + 4 );
265     p_chk->common.i_chunk_pos    = AVI_TellAbsolute( p_input );
266
267     p_chk->common.p_father = NULL;
268     p_chk->common.p_next = NULL;
269     p_chk->common.p_first = NULL;
270     p_chk->common.p_next = NULL;
271 #ifdef AVI_DEBUG
272     msg_Dbg( p_input, 
273              "Found Chunk fourcc:%c%c%c%c size:"I64Fd" pos:"I64Fd,
274              AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ),
275              p_chk->common.i_chunk_size,
276              p_chk->common.i_chunk_pos );
277 #endif
278     return VLC_SUCCESS;
279 }
280
281 static int AVI_NextChunk( input_thread_t *p_input,
282                           avi_chunk_t *p_chk )
283 {
284     avi_chunk_t chk;
285     
286     if( !p_chk )
287     {
288         if( AVI_ChunkReadCommon( p_input, &chk ) )
289         {
290             return VLC_EGENERIC;
291         }
292         p_chk = &chk;
293     }
294
295     if( p_chk->common.p_father )
296     {
297         if( p_chk->common.p_father->common.i_chunk_pos + 
298                 __EVEN( p_chk->common.p_father->common.i_chunk_size ) + 8 <
299             p_chk->common.i_chunk_pos + 
300                 __EVEN( p_chk->common.i_chunk_size ) + 8 )
301         {
302             return VLC_EGENERIC;
303         }
304     }
305     return AVI_SeekAbsolute( p_input,
306                              p_chk->common.i_chunk_pos + 
307                                  __EVEN( p_chk->common.i_chunk_size ) + 8 );
308 }
309
310 int _AVI_ChunkGoto( input_thread_t *p_input,
311                    avi_chunk_t *p_chk )
312 {
313     if( !p_chk )
314     {
315         return VLC_EGENERIC;
316     }
317     return AVI_SeekAbsolute( p_input, p_chk->common.i_chunk_pos );
318 }
319
320 /****************************************************************************
321  *
322  * Functions to read chunks 
323  *
324  ****************************************************************************/
325 static int AVI_ChunkRead_list( input_thread_t *p_input,
326                                avi_chunk_t *p_container,
327                                vlc_bool_t b_seekable )
328 {
329     avi_chunk_t *p_chk;
330     uint8_t *p_peek;
331     
332     if( p_container->common.i_chunk_size < 8 )
333     {
334         /* empty box */
335         msg_Warn( p_input, "empty list chunk" );
336         return VLC_EGENERIC;
337     }
338     if( input_Peek( p_input, &p_peek, 12 ) < 12 )
339     {
340         msg_Warn( p_input, "cannot peek while reading list chunk" );
341         return VLC_EGENERIC;
342     }
343     p_container->list.i_type = GetFOURCC( p_peek + 8 );
344
345     if( p_container->common.i_chunk_fourcc == AVIFOURCC_LIST &&
346         p_container->list.i_type == AVIFOURCC_movi )
347     {
348         msg_Dbg( p_input, "Skipping movi chunk" );
349         if( b_seekable )
350         {
351             return AVI_NextChunk( p_input, p_container );
352         }
353         else
354         {
355             return VLC_SUCCESS; // point at begining of LIST-movi 
356         }
357     }
358
359     AVI_SkipBytes( p_input, 12 );
360 #ifdef AVI_DEBUG
361     msg_Dbg( p_input, 
362              "found LIST chunk: \'%c%c%c%c\'",
363              AVIFOURCC_PRINT( p_container->list.i_type ) );
364 #endif
365     for( ; ; )
366     {
367         p_chk = malloc( sizeof( avi_chunk_t ) );
368         memset( p_chk, 0, sizeof( avi_chunk_t ) );
369         if( !p_container->common.p_first )
370         {
371             p_container->common.p_first = p_chk;
372         }
373         else
374         {
375             p_container->common.p_last->common.p_next = p_chk;
376         }
377         p_container->common.p_last = p_chk;
378
379         if( AVI_ChunkRead( p_input, p_chk, p_container, b_seekable ) ||
380            ( AVI_TellAbsolute( p_input ) >=
381                 (off_t)p_chk->common.p_father->common.i_chunk_pos + 
382                     __EVEN( p_chk->common.p_father->common.i_chunk_size ) ) )
383         {
384             break;
385         }
386         /* If we can't seek then stop when we 've found LIST-movi */
387         if( p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST &&
388             p_chk->list.i_type == AVIFOURCC_movi && !b_seekable )
389         {
390             break;
391         }
392
393     } 
394     
395     return VLC_SUCCESS;
396 }
397
398 #define AVI_READCHUNK_ENTER \
399     int64_t i_read = __EVEN(p_chk->common.i_chunk_size ) + 8; \
400     uint8_t  *p_read, *p_buff;    \
401     if( !( p_read = p_buff = malloc(i_read ) ) ) \
402     { \
403         return 0; \
404     } \
405     i_read = AVI_ReadData( p_input, p_read, i_read ); \
406     p_read += 8; \
407     i_read -= 8
408     
409 #define AVI_READCHUNK_EXIT( code ) \
410     free( p_buff ); \
411     if( i_read < 0 ) \
412     { \
413         msg_Warn( p_input, "not enough data" ); \
414     } \
415     return code
416 #define AVI_READ2BYTES( i_word ) \
417     i_word = GetWLE( p_read ); \
418     p_read += 2; \
419     i_read -= 2
420
421 #define AVI_READ4BYTES( i_dword ) \
422     i_dword = GetDWLE( p_read ); \
423     p_read += 4; \
424     i_read -= 4
425     
426 #define AVI_READFOURCC( i_dword ) \
427     i_dword = GetFOURCC( p_read ); \
428     p_read += 4; \
429     i_read -= 4
430
431 static int AVI_ChunkRead_avih( input_thread_t *p_input,
432                                avi_chunk_t *p_chk,
433                                vlc_bool_t b_seekable )
434 {
435     AVI_READCHUNK_ENTER;
436
437     AVI_READ4BYTES( p_chk->avih.i_microsecperframe);
438     AVI_READ4BYTES( p_chk->avih.i_maxbytespersec );
439     AVI_READ4BYTES( p_chk->avih.i_reserved1 );
440     AVI_READ4BYTES( p_chk->avih.i_flags );
441     AVI_READ4BYTES( p_chk->avih.i_totalframes );
442     AVI_READ4BYTES( p_chk->avih.i_initialframes );
443     AVI_READ4BYTES( p_chk->avih.i_streams );
444     AVI_READ4BYTES( p_chk->avih.i_suggestedbuffersize );
445     AVI_READ4BYTES( p_chk->avih.i_width );
446     AVI_READ4BYTES( p_chk->avih.i_height );
447     AVI_READ4BYTES( p_chk->avih.i_scale );
448     AVI_READ4BYTES( p_chk->avih.i_rate );
449     AVI_READ4BYTES( p_chk->avih.i_start );
450     AVI_READ4BYTES( p_chk->avih.i_length );
451 #ifdef AVI_DEBUG
452     msg_Dbg( p_input, 
453              "avih: streams:%d flags:%s%s%s%s %dx%d", 
454              p_chk->avih.i_streams,
455              p_chk->avih.i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
456              p_chk->avih.i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
457              p_chk->avih.i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
458              p_chk->avih.i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"",
459              p_chk->avih.i_width, p_chk->avih.i_height );
460 #endif 
461     AVI_READCHUNK_EXIT( VLC_SUCCESS );
462 }
463
464 static int AVI_ChunkRead_strh( input_thread_t *p_input,
465                                avi_chunk_t *p_chk,
466                                vlc_bool_t b_seekable )
467 {
468     AVI_READCHUNK_ENTER;
469
470     AVI_READFOURCC( p_chk->strh.i_type );
471     AVI_READFOURCC( p_chk->strh.i_handler );
472     AVI_READ4BYTES( p_chk->strh.i_flags );
473     AVI_READ4BYTES( p_chk->strh.i_reserved1 );
474     AVI_READ4BYTES( p_chk->strh.i_initialframes );
475     AVI_READ4BYTES( p_chk->strh.i_scale );
476     AVI_READ4BYTES( p_chk->strh.i_rate );
477     AVI_READ4BYTES( p_chk->strh.i_start );
478     AVI_READ4BYTES( p_chk->strh.i_length );
479     AVI_READ4BYTES( p_chk->strh.i_suggestedbuffersize );
480     AVI_READ4BYTES( p_chk->strh.i_quality );
481     AVI_READ4BYTES( p_chk->strh.i_samplesize );
482 #ifdef AVI_DEBUG
483     msg_Dbg( p_input, 
484              "strh: type:%c%c%c%c handler:0x%8.8x samplesize:%d %.2ffps",
485              AVIFOURCC_PRINT( p_chk->strh.i_type ),
486              p_chk->strh.i_handler,
487              p_chk->strh.i_samplesize,
488              ( p_chk->strh.i_scale ? 
489                 (float)p_chk->strh.i_rate / (float)p_chk->strh.i_scale : -1) );
490 #endif
491     
492     AVI_READCHUNK_EXIT( VLC_SUCCESS );
493 }
494
495 static int AVI_ChunkRead_strf( input_thread_t *p_input,
496                                avi_chunk_t *p_chk,
497                                vlc_bool_t b_seekable )
498 {
499     avi_chunk_t *p_strh;
500
501     AVI_READCHUNK_ENTER;
502     if( p_chk->common.p_father == NULL )
503     {
504         msg_Err( p_input, "malformed avi file" );
505         AVI_READCHUNK_EXIT( VLC_EGENERIC );
506     }
507     if( !( p_strh = AVI_ChunkFind( p_chk->common.p_father, AVIFOURCC_strh, 0 ) ) )
508     {
509         msg_Err( p_input, "malformed avi file" );
510         AVI_READCHUNK_EXIT( VLC_EGENERIC );
511     }
512     
513     switch( p_strh->strh.i_type )
514     {
515         case( AVIFOURCC_auds ):
516             p_chk->strf.auds.p_wf = malloc( p_chk->common.i_chunk_size );
517             AVI_READ2BYTES( p_chk->strf.auds.p_wf->wFormatTag );
518             AVI_READ2BYTES( p_chk->strf.auds.p_wf->nChannels );
519             AVI_READ4BYTES( p_chk->strf.auds.p_wf->nSamplesPerSec );
520             AVI_READ4BYTES( p_chk->strf.auds.p_wf->nAvgBytesPerSec );
521             AVI_READ2BYTES( p_chk->strf.auds.p_wf->nBlockAlign );
522             AVI_READ2BYTES( p_chk->strf.auds.p_wf->wBitsPerSample );
523             if( p_chk->strf.auds.p_wf->wFormatTag != WAVE_FORMAT_PCM )
524             {
525                 AVI_READ2BYTES( p_chk->strf.auds.p_wf->cbSize );
526             }
527             else
528             {
529                 p_chk->strf.auds.p_wf->cbSize = 0;
530             }
531             if( p_chk->strf.auds.p_wf->cbSize > 0 )
532             {
533                 memcpy( &p_chk->strf.auds.p_wf[1] , 
534                         p_buff + sizeof( WAVEFORMATEX ), 
535                         p_chk->common.i_chunk_size - sizeof( WAVEFORMATEX ));
536             }
537 #ifdef AVI_DEBUG
538             msg_Dbg( p_input, 
539                      "strf: audio:0x%4.4x channels:%d %dHz %dbits/sample %dkb/s",
540                      p_chk->strf.auds.p_wf->wFormatTag,
541                      p_chk->strf.auds.p_wf->nChannels,
542                      p_chk->strf.auds.p_wf->nSamplesPerSec,
543                      p_chk->strf.auds.p_wf->wBitsPerSample,
544                      p_chk->strf.auds.p_wf->nAvgBytesPerSec * 8 / 1024 );
545 #endif
546             break;
547         case( AVIFOURCC_vids ):
548             p_strh->strh.i_samplesize = 0; // XXX for ffmpeg avi file
549             p_chk->strf.vids.p_bih = malloc( p_chk->common.i_chunk_size );
550             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biSize );
551             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biWidth );
552             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biHeight );
553             AVI_READ2BYTES( p_chk->strf.vids.p_bih->biPlanes );
554             AVI_READ2BYTES( p_chk->strf.vids.p_bih->biBitCount );
555             AVI_READFOURCC( p_chk->strf.vids.p_bih->biCompression );
556             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biSizeImage );
557             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biXPelsPerMeter );
558             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biYPelsPerMeter );
559             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrUsed );
560             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrImportant );
561             memcpy( &p_chk->strf.vids.p_bih[1],
562                     p_buff + sizeof(BITMAPINFOHEADER),
563                     p_chk->common.i_chunk_size - sizeof(BITMAPINFOHEADER) );
564 #ifdef AVI_DEBUG
565             msg_Dbg( p_input,
566                      "strf: video:%c%c%c%c %dx%d planes:%d %dbpp",
567                      AVIFOURCC_PRINT( p_chk->strf.vids.p_bih->biCompression ),
568                      p_chk->strf.vids.p_bih->biWidth,
569                      p_chk->strf.vids.p_bih->biHeight,
570                      p_chk->strf.vids.p_bih->biPlanes,
571                      p_chk->strf.vids.p_bih->biBitCount );
572 #endif
573             break;
574         default:
575             msg_Warn( p_input, "unknown stream type" );
576             break;
577     }
578     AVI_READCHUNK_EXIT( VLC_SUCCESS );
579 }
580 static void AVI_ChunkFree_strf( input_thread_t *p_input,
581                                avi_chunk_t *p_chk )
582 {
583
584 }
585
586 static int AVI_ChunkRead_strd( input_thread_t *p_input,
587                                avi_chunk_t *p_chk,
588                                vlc_bool_t b_seekable )
589 {
590     AVI_READCHUNK_ENTER;
591     p_chk->strd.p_data = malloc( p_chk->common.i_chunk_size );
592     memcpy( p_chk->strd.p_data,
593             p_buff,
594             p_chk->common.i_chunk_size );
595     AVI_READCHUNK_EXIT( VLC_SUCCESS );
596 }
597
598 static int AVI_ChunkRead_idx1( input_thread_t *p_input,
599                                avi_chunk_t *p_chk,
600                                vlc_bool_t b_seekable )
601 {
602     unsigned int i_count, i_index;
603
604     AVI_READCHUNK_ENTER;
605
606     i_count = __MIN( p_chk->common.i_chunk_size, i_read ) / 16;
607
608     p_chk->idx1.i_entry_count = i_count;
609     p_chk->idx1.i_entry_max   = i_count;
610     if( i_count > 0 )
611     {
612         p_chk->idx1.entry = calloc( i_count, sizeof( idx1_entry_t ) );
613
614         for( i_index = 0; i_index < i_count ; i_index++ )
615         {
616             AVI_READFOURCC( p_chk->idx1.entry[i_index].i_fourcc );
617             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_flags );
618             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_pos );
619             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_length );
620         }
621     }
622     else
623     {
624         p_chk->idx1.entry = NULL;
625     }
626 #ifdef AVI_DEBUG
627     msg_Dbg( p_input, "idx1: index entry:%d", i_count );
628 #endif
629     AVI_READCHUNK_EXIT( VLC_SUCCESS );
630 }
631
632 static void AVI_ChunkFree_idx1( input_thread_t *p_input,
633                                avi_chunk_t *p_chk )
634 {
635     p_chk->idx1.i_entry_count = 0;
636     p_chk->idx1.i_entry_max   = 0;
637     FREE( p_chk->idx1.entry )
638 }
639
640 static struct 
641 {
642     vlc_fourcc_t i_fourcc;
643     char *psz_type;
644 } AVI_strz_type[] =
645 {
646     { AVIFOURCC_IARL, "archive location" },
647     { AVIFOURCC_IART, "artist" },
648     { AVIFOURCC_ICMS, "commisioned" },
649     { AVIFOURCC_ICMT, "comments" },
650     { AVIFOURCC_ICOP, "copyright" },
651     { AVIFOURCC_ICRD, "creation date" },
652     { AVIFOURCC_ICRP, "cropped" },
653     { AVIFOURCC_IDIM, "dimensions" },
654     { AVIFOURCC_IDPI, "dots per inch" },
655     { AVIFOURCC_IENG, "enginner" },
656     { AVIFOURCC_IGNR, "genre" },
657     { AVIFOURCC_IKEY, "keywords" },
658     { AVIFOURCC_ILGT, "lightness" },
659     { AVIFOURCC_IMED, "medium" },
660     { AVIFOURCC_INAM, "name" },
661     { AVIFOURCC_IPLT, "palette setting" },
662     { AVIFOURCC_IPRD, "product" },
663     { AVIFOURCC_ISBJ, "subject" },
664     { AVIFOURCC_ISFT, "software" },
665     { AVIFOURCC_ISHP, "sharpness" },
666     { AVIFOURCC_ISRC, "source" },
667     { AVIFOURCC_ISRF, "source form" },
668     { AVIFOURCC_ITCH, "technician" },
669     { AVIFOURCC_ISMP, "time code" },
670     { AVIFOURCC_IDIT, "digitalization time" },
671     { 0,              "???" }
672 };
673 static int AVI_ChunkRead_strz( input_thread_t *p_input,
674                                avi_chunk_t *p_chk,
675                                vlc_bool_t b_seekable )
676 {
677     int i_index;
678     avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk;
679     AVI_READCHUNK_ENTER;
680
681     for( i_index = 0;; i_index++)
682     {
683         if( !AVI_strz_type[i_index].i_fourcc ||
684             AVI_strz_type[i_index].i_fourcc == p_strz->i_chunk_fourcc )
685         {
686             break;
687         }
688     }
689     p_strz->p_type = strdup( AVI_strz_type[i_index].psz_type );
690     p_strz->p_str = malloc( i_read + 1);
691
692     if( p_strz->i_chunk_size )
693     {
694         memcpy( p_strz->p_str, p_read, i_read );
695     }
696     p_strz->p_str[i_read] = 0;
697     
698 #ifdef AVI_DEBUG
699     msg_Dbg( p_input, "%c%c%c%c: %s : %s", 
700              AVIFOURCC_PRINT( p_strz->i_chunk_fourcc), p_strz->p_type, p_strz->p_str);
701 #endif
702     AVI_READCHUNK_EXIT( VLC_SUCCESS );
703 }
704 static void AVI_ChunkFree_strz( input_thread_t *p_input,
705                                 avi_chunk_t *p_chk )
706 {
707     avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk;
708     FREE( p_strz->p_type );
709     FREE( p_strz->p_str );
710 }
711
712 static int AVI_ChunkRead_nothing( input_thread_t *p_input,
713                                avi_chunk_t *p_chk,
714                                vlc_bool_t b_seekable )
715 {
716     return AVI_NextChunk( p_input, p_chk );
717 }
718 static void AVI_ChunkFree_nothing( input_thread_t *p_input,
719                                avi_chunk_t *p_chk )
720 {
721
722 }
723
724 static struct
725 {
726     vlc_fourcc_t i_fourcc;
727     int   (*AVI_ChunkRead_function)( input_thread_t *p_input, 
728                                      avi_chunk_t *p_chk,
729                                      vlc_bool_t b_seekable );
730     void  (*AVI_ChunkFree_function)( input_thread_t *p_input,
731                                      avi_chunk_t *p_chk );
732 } AVI_Chunk_Function [] =
733 {
734     { AVIFOURCC_RIFF, AVI_ChunkRead_list, AVI_ChunkFree_nothing },
735     { AVIFOURCC_LIST, AVI_ChunkRead_list, AVI_ChunkFree_nothing },
736     { AVIFOURCC_avih, AVI_ChunkRead_avih, AVI_ChunkFree_nothing },
737     { AVIFOURCC_strh, AVI_ChunkRead_strh, AVI_ChunkFree_nothing },
738     { AVIFOURCC_strf, AVI_ChunkRead_strf, AVI_ChunkFree_strf },
739     { AVIFOURCC_strd, AVI_ChunkRead_strd, AVI_ChunkFree_nothing },
740     { AVIFOURCC_idx1, AVI_ChunkRead_idx1, AVI_ChunkFree_idx1 },
741     { AVIFOURCC_JUNK, AVI_ChunkRead_nothing, AVI_ChunkFree_nothing },
742
743     { AVIFOURCC_IARL, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
744     { AVIFOURCC_IARL, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
745     { AVIFOURCC_IART, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
746     { AVIFOURCC_ICMS, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
747     { AVIFOURCC_ICMT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
748     { AVIFOURCC_ICOP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
749     { AVIFOURCC_ICRD, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
750     { AVIFOURCC_ICRP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
751     { AVIFOURCC_IDIM, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
752     { AVIFOURCC_IDPI, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
753     { AVIFOURCC_IENG, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
754     { AVIFOURCC_IGNR, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
755     { AVIFOURCC_IKEY, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
756     { AVIFOURCC_ILGT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
757     { AVIFOURCC_IMED, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
758     { AVIFOURCC_INAM, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
759     { AVIFOURCC_IPLT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
760     { AVIFOURCC_IPRD, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
761     { AVIFOURCC_ISBJ, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
762     { AVIFOURCC_ISFT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
763     { AVIFOURCC_ISHP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
764     { AVIFOURCC_ISRC, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
765     { AVIFOURCC_ISRF, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
766     { AVIFOURCC_ITCH, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
767     { AVIFOURCC_ISMP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
768     { AVIFOURCC_IDIT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
769     { 0,           NULL,               NULL }
770 };
771
772 static int AVI_ChunkFunctionFind( vlc_fourcc_t i_fourcc )
773 {
774     unsigned int i_index;
775     for( i_index = 0; ; i_index++ )
776     {
777         if( ( AVI_Chunk_Function[i_index].i_fourcc == i_fourcc )||
778             ( AVI_Chunk_Function[i_index].i_fourcc == 0 ) )
779         {
780             return i_index;
781         }
782     }
783 }
784
785 int  _AVI_ChunkRead( input_thread_t *p_input,
786                      avi_chunk_t *p_chk,
787                      avi_chunk_t *p_father,
788                      vlc_bool_t b_seekable )
789 {
790     int i_index;
791
792     if( !p_chk )
793     {
794         return VLC_EGENERIC;
795     }
796
797     if( AVI_ChunkReadCommon( p_input, p_chk ) )
798     {
799         msg_Warn( p_input, "cannot read one chunk" );
800         return VLC_EGENERIC;
801     }
802     p_chk->common.p_father = p_father;
803
804     i_index = AVI_ChunkFunctionFind( p_chk->common.i_chunk_fourcc );
805     if( AVI_Chunk_Function[i_index].AVI_ChunkRead_function )
806     {
807         return AVI_Chunk_Function[i_index].
808                         AVI_ChunkRead_function( p_input, p_chk, b_seekable );
809     }
810
811     msg_Warn( p_input, "unknown chunk (not loaded)" );
812     return AVI_NextChunk( p_input, p_chk );
813 }
814
815 void _AVI_ChunkFree( input_thread_t *p_input,
816                      avi_chunk_t *p_chk )
817 {
818     int i_index;
819     avi_chunk_t *p_child, *p_next;
820
821     if( !p_chk )
822     {
823         return;
824     }
825     
826     /* Free all child chunk */
827     p_child = p_chk->common.p_first;
828     while( p_child )
829     {
830         p_next = p_child->common.p_next;
831         AVI_ChunkFree( p_input, p_child );
832         free( p_child );
833         p_child = p_next;
834     }
835
836     i_index = AVI_ChunkFunctionFind( p_chk->common.i_chunk_fourcc );
837     if( AVI_Chunk_Function[i_index].AVI_ChunkFree_function )
838     {
839 #ifdef AVI_DEBUG
840         msg_Dbg( p_input, "free chunk %c%c%c%c", 
841                  AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ) );
842 #endif
843         AVI_Chunk_Function[i_index].AVI_ChunkFree_function( p_input, p_chk);
844     }
845     else
846     {
847         msg_Warn( p_input, "unknown chunk (not unloaded)" );
848     }
849     p_chk->common.p_first = NULL;
850     p_chk->common.p_last  = NULL;
851     
852     return;
853 }
854
855 int AVI_ChunkReadRoot( input_thread_t *p_input,
856                        avi_chunk_t *p_root,
857                        vlc_bool_t b_seekable )
858 {
859     avi_chunk_list_t *p_list = (avi_chunk_list_t*)p_root;
860     avi_chunk_t      *p_chk;
861     
862     p_list->i_chunk_pos  = 0;
863     p_list->i_chunk_size = p_input->stream.p_selected_area->i_size;
864     p_list->i_chunk_fourcc = AVIFOURCC_LIST;
865     p_list->p_father = NULL;
866     p_list->p_next  = NULL;
867     p_list->p_first = NULL;
868     p_list->p_last  = NULL;
869
870     p_list->i_type = VLC_FOURCC( 'r', 'o', 'o', 't' );
871     
872     for( ; ; )
873     {
874         p_chk = malloc( sizeof( avi_chunk_t ) );
875         memset( p_chk, 0, sizeof( avi_chunk_t ) );
876         if( !p_root->common.p_first )
877         {
878             p_root->common.p_first = p_chk;
879         }
880         else
881         {
882             p_root->common.p_last->common.p_next = p_chk;
883         }
884         p_root->common.p_last = p_chk;
885
886         if( AVI_ChunkRead( p_input, p_chk, p_root, b_seekable ) ||
887            ( AVI_TellAbsolute( p_input ) >=
888                 (off_t)p_chk->common.p_father->common.i_chunk_pos + 
889                     __EVEN( p_chk->common.p_father->common.i_chunk_size ) ) )
890         {
891             break;
892         }
893         /* If we can't seek then stop when we 've found first RIFF-AVI */
894         if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF &&
895             p_chk->list.i_type == AVIFOURCC_AVI && !b_seekable )
896         {
897             break;
898         }
899     } 
900     
901     return VLC_SUCCESS;
902 }
903
904 void AVI_ChunkFreeRoot( input_thread_t *p_input,
905                         avi_chunk_t  *p_chk )
906 {
907     AVI_ChunkFree( p_input, p_chk );
908 }
909
910
911 int  _AVI_ChunkCount( avi_chunk_t *p_chk, vlc_fourcc_t i_fourcc )
912 {
913     int i_count;
914     avi_chunk_t *p_child;
915
916     if( !p_chk )
917     {
918         return 0;
919     }
920
921     i_count = 0;
922     p_child = p_chk->common.p_first;
923     while( p_child )
924     {
925         if( p_child->common.i_chunk_fourcc == i_fourcc ||
926             ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && 
927               p_child->list.i_type == i_fourcc ) )
928         {
929             i_count++;
930         }
931         p_child = p_child->common.p_next;
932     }
933     return i_count;
934 }
935
936 avi_chunk_t *_AVI_ChunkFind( avi_chunk_t *p_chk,
937                              vlc_fourcc_t i_fourcc, int i_number )
938 {
939     avi_chunk_t *p_child;
940     if( !p_chk )
941     {
942         return NULL;
943     }
944     p_child = p_chk->common.p_first;
945
946     while( p_child )
947     {
948         if( p_child->common.i_chunk_fourcc == i_fourcc ||
949             ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && 
950               p_child->list.i_type == i_fourcc ) )
951         {
952             if( i_number == 0 )
953             {
954                 /* We found it */
955                 return p_child;
956             }
957
958             i_number--;
959         }
960         p_child = p_child->common.p_next;
961     }
962     return NULL;
963 }
964
965 static void AVI_ChunkDumpDebug_level( input_thread_t *p_input,
966                                       avi_chunk_t  *p_chk, int i_level )
967 {
968     char str[1024];
969     int i;
970     avi_chunk_t *p_child;
971     
972     memset( str, ' ', sizeof( str ) );
973     for( i = 1; i < i_level; i++ )
974     {
975         str[i * 5] = '|';
976     }
977     if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF||
978         p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST )
979     {
980         sprintf( str + i_level * 5, 
981                  "%c %c%c%c%c-%c%c%c%c size:"I64Fu" pos:"I64Fu,
982                  i_level ? '+' : '*',
983                  AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ),
984                  AVIFOURCC_PRINT( p_chk->list.i_type ),
985                  p_chk->common.i_chunk_size,
986                  p_chk->common.i_chunk_pos );
987     }
988     else
989     {
990         sprintf( str + i_level * 5, 
991                  "+ %c%c%c%c size:"I64Fu" pos:"I64Fu,
992                  AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ),
993                  p_chk->common.i_chunk_size,
994                  p_chk->common.i_chunk_pos );
995     }
996     msg_Dbg( p_input, "%s", str );
997
998     p_child = p_chk->common.p_first;
999     while( p_child )
1000     {
1001         AVI_ChunkDumpDebug_level( p_input, p_child, i_level + 1 );
1002         p_child = p_child->common.p_next;
1003     }
1004 }
1005 void _AVI_ChunkDumpDebug( input_thread_t *p_input,
1006                          avi_chunk_t  *p_chk )
1007 {
1008     AVI_ChunkDumpDebug_level( p_input, p_chk, 0 );
1009 }
1010