Mercurial > mplayer.hg
view aviheader.c @ 644:88eb1a3f7bfb
Changed code, should be faster on Athlon/K6 but slower on PIII with SSE, more portable.
author | atmosfear |
---|---|
date | Thu, 26 Apr 2001 20:35:58 +0000 |
parents | 2d3a4339bb5a |
children | 14a2f35921a0 |
line wrap: on
line source
#include <stdio.h> #include <stdlib.h> extern int verbose; // defined in mplayer.c #include "stream.h" #include "demuxer.h" #include "wine/mmreg.h" #include "wine/avifmt.h" #include "wine/vfw.h" #include "codec-cfg.h" #include "stheader.h" #define MIN(a,b) (((a)<(b))?(a):(b)) static MainAVIHeader avih; extern void print_avih(MainAVIHeader *h); extern void print_strh(AVIStreamHeader *h); extern void print_wave_header(WAVEFORMATEX *h); extern void print_index(AVIINDEXENTRY *idx,int idx_size); void read_avi_header(demuxer_t *demuxer,int index_mode){ sh_audio_t *sh_audio=NULL; sh_video_t *sh_video=NULL; int stream_id=-1; int idxfix_videostream=0; int idxfix_divx=0; //---- AVI header: demuxer->idx_size=0; while(1){ int id=stream_read_dword_le(demuxer->stream); int chunksize,size2; static int last_fccType=0; // if(stream_eof(demuxer->stream)) break; // if(id==mmioFOURCC('L','I','S','T')){ int len=stream_read_dword_le(demuxer->stream)-4; // list size id=stream_read_dword_le(demuxer->stream); // list type if(verbose>=2) printf("LIST %.4s len=%d\n",(char *) &id,len); if(id==listtypeAVIMOVIE){ // found MOVI header demuxer->movi_start=stream_tell(demuxer->stream); demuxer->movi_end=demuxer->movi_start+len; if(verbose>=1) printf("Found movie at 0x%X - 0x%X\n",demuxer->movi_start,demuxer->movi_end); len=(len+1)&(~1); stream_skip(demuxer->stream,len); } continue; } size2=stream_read_dword_le(demuxer->stream); if(verbose>=2) printf("CHUNK %.4s len=%d\n",(char *) &id,size2); chunksize=(size2+1)&(~1); switch(id){ case ckidAVIMAINHDR: // read 'avih' stream_read(demuxer->stream,(char*) &avih,MIN(size2,sizeof(avih))); chunksize-=MIN(size2,sizeof(avih)); if(verbose) print_avih(&avih); break; case ckidSTREAMHEADER: { // read 'strh' AVIStreamHeader h; stream_read(demuxer->stream,(char*) &h,MIN(size2,sizeof(h))); chunksize-=MIN(size2,sizeof(h)); ++stream_id; if(h.fccType==streamtypeVIDEO){ sh_video=new_sh_video(stream_id); memcpy(&sh_video->video,&h,sizeof(h)); } else if(h.fccType==streamtypeAUDIO){ sh_audio=new_sh_audio(stream_id); memcpy(&sh_audio->audio,&h,sizeof(h)); } last_fccType=h.fccType; if(verbose>=1) print_strh(&h); break; } case ckidSTREAMFORMAT: { // read 'strf' if(last_fccType==streamtypeVIDEO){ sh_video->bih=calloc((chunksize<sizeof(BITMAPINFOHEADER))?sizeof(BITMAPINFOHEADER):chunksize,1); // sh_video->bih=malloc(chunksize); memset(sh_video->bih,0,chunksize); if(verbose>=1) printf("found 'bih', %d bytes of %d\n",chunksize,sizeof(BITMAPINFOHEADER)); stream_read(demuxer->stream,(char*) sh_video->bih,chunksize); chunksize=0; // sh_video->fps=(float)sh_video->video.dwRate/(float)sh_video->video.dwScale; // sh_video->frametime=(float)sh_video->video.dwScale/(float)sh_video->video.dwRate; // if(demuxer->video->id==-1) demuxer->video->id=stream_id; // IdxFix: idxfix_videostream=stream_id; switch(sh_video->bih->biCompression){ case mmioFOURCC('D', 'I', 'V', '3'): case mmioFOURCC('d', 'i', 'v', '3'): case mmioFOURCC('D', 'I', 'V', '4'): case mmioFOURCC('d', 'i', 'v', '4'): case mmioFOURCC('D', 'I', 'V', '5'): case mmioFOURCC('d', 'i', 'v', '5'): case mmioFOURCC('D', 'I', 'V', '6'): case mmioFOURCC('d', 'i', 'v', '6'): case mmioFOURCC('M', 'P', '4', '3'): case mmioFOURCC('m', 'p', '4', '3'): case mmioFOURCC('M', 'P', '4', '2'): case mmioFOURCC('m', 'p', '4', '2'): case mmioFOURCC('A', 'P', '4', '1'): idxfix_divx=1; // we can fix keyframes only for divx coded files! } } else if(last_fccType==streamtypeAUDIO){ sh_audio->wf=calloc((chunksize<sizeof(WAVEFORMATEX))?sizeof(WAVEFORMATEX):chunksize,1); // sh_audio->wf=malloc(chunksize); memset(sh_audio->wf,0,chunksize); if(verbose>=1) printf("found 'wf', %d bytes of %d\n",chunksize,sizeof(WAVEFORMATEX)); stream_read(demuxer->stream,(char*) sh_audio->wf,chunksize); chunksize=0; if(verbose>=1) print_wave_header(sh_audio->wf); // if(demuxer->audio->id==-1) demuxer->audio->id=stream_id; } break; } case ckidAVINEWINDEX: if(index_mode){ demuxer->idx_size=size2>>4; if(verbose>=1) printf("Reading INDEX block, %d chunks for %ld frames\n", demuxer->idx_size,avih.dwTotalFrames); demuxer->idx=malloc(demuxer->idx_size<<4); stream_read(demuxer->stream,(char*)demuxer->idx,demuxer->idx_size<<4); chunksize-=demuxer->idx_size<<4; if(verbose>=2) print_index(demuxer->idx,demuxer->idx_size); break; } } if(chunksize>0) stream_skip(demuxer->stream,chunksize); else if(chunksize<0) printf("WARNING!!! chunksize=%d (id=%.4s)\n",chunksize,(char *) &id); } if(index_mode>=2 || (demuxer->idx_size==0 && index_mode==1)){ // build index for file: stream_reset(demuxer->stream); stream_seek(demuxer->stream,demuxer->movi_start); demuxer->idx_pos=0; demuxer->idx=NULL; while(1){ int id,len,skip; AVIINDEXENTRY* idx; unsigned char c; demuxer->filepos=stream_tell(demuxer->stream); if(demuxer->filepos>=demuxer->movi_end) break; id=stream_read_dword_le(demuxer->stream); len=stream_read_dword_le(demuxer->stream); if(id==mmioFOURCC('L','I','S','T')){ id=stream_read_dword_le(demuxer->stream); // list type continue; } if(stream_eof(demuxer->stream)) break; if(demuxer->idx_pos<=demuxer->idx_size){ demuxer->idx_size+=32; demuxer->idx=realloc(demuxer->idx,demuxer->idx_size*sizeof(AVIINDEXENTRY)); if(!demuxer->idx){demuxer->idx_pos=0; break;} // error! } idx=&((AVIINDEXENTRY *)demuxer->idx)[demuxer->idx_pos++]; idx->ckid=id; idx->dwFlags=AVIIF_KEYFRAME; // FIXME idx->dwChunkOffset=demuxer->filepos; idx->dwChunkLength=len; c=stream_read_char(demuxer->stream); // Fix keyframes for DivX files: if(idxfix_divx) if(avi_stream_id(id)==idxfix_videostream){ if(c&0x40) idx->dwFlags=0; } if(verbose>=2) printf("%08X %08X %.4s %02X %X\n",demuxer->filepos,id,(char *) &id,c,(unsigned int) idx->dwFlags); #if 0 { unsigned char tmp[64]; int i; stream_read(demuxer->stream,tmp,64); printf("%.4s",&id); for(i=0;i<64;i++) printf(" %02X",tmp[i]); printf("\n"); } #endif skip=(len+1)&(~1); // total bytes in this chunk stream_seek(demuxer->stream,8+demuxer->filepos+skip); } demuxer->idx_size=demuxer->idx_pos; printf("AVI: Generated index table for %d chunks!\n",demuxer->idx_size); } } #undef MIN