mirror of https://github.com/mpv-player/mpv
292 lines
11 KiB
C
292 lines
11 KiB
C
|
|
#include "config.h"
|
|
#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 "bswap.h"
|
|
#include "stheader.h"
|
|
|
|
#define MIN(a,b) (((a)<(b))?(a):(b))
|
|
|
|
/*
|
|
* Some macros to swap little endian structures read from an AVI file
|
|
* into machine endian format
|
|
*/
|
|
#ifdef WORDS_BIGENDIAN
|
|
#define le2me_MainAVIHeader(h) { \
|
|
(h)->dwMicroSecPerFrame = le2me_32((h)->dwMicroSecPerFrame); \
|
|
(h)->dwMaxBytesPerSec = le2me_32((h)->dwMaxBytesPerSec); \
|
|
(h)->dwPaddingGranularity = le2me_32((h)->dwPaddingGranularity); \
|
|
(h)->dwFlags = le2me_32((h)->dwFlags); \
|
|
(h)->dwTotalFrames = le2me_32((h)->dwTotalFrames); \
|
|
(h)->dwInitialFrames = le2me_32((h)->dwInitialFrames); \
|
|
(h)->dwStreams = le2me_32((h)->dwStreams); \
|
|
(h)->dwSuggestedBufferSize = le2me_32((h)->dwSuggestedBufferSize); \
|
|
(h)->dwWidth = le2me_32((h)->dwWidth); \
|
|
(h)->dwHeight = le2me_32((h)->dwHeight); \
|
|
}
|
|
|
|
#define le2me_AVIStreamHeader(h) { \
|
|
(h)->fccType = le2me_32((h)->fccType); \
|
|
(h)->fccHandler = le2me_32((h)->fccHandler); \
|
|
(h)->dwFlags = le2me_32((h)->dwFlags); \
|
|
(h)->wPriority = le2me_16((h)->wPriority); \
|
|
(h)->wLanguage = le2me_16((h)->wLanguage); \
|
|
(h)->dwInitialFrames = le2me_32((h)->dwInitialFrames); \
|
|
(h)->dwScale = le2me_32((h)->dwScale); \
|
|
(h)->dwRate = le2me_32((h)->dwRate); \
|
|
(h)->dwStart = le2me_32((h)->dwStart); \
|
|
(h)->dwLength = le2me_32((h)->dwLength); \
|
|
(h)->dwSuggestedBufferSize = le2me_32((h)->dwSuggestedBufferSize); \
|
|
(h)->dwQuality = le2me_32((h)->dwQuality); \
|
|
(h)->dwSampleSize = le2me_32((h)->dwSampleSize); \
|
|
le2me_RECT(&(h)->rcFrame); \
|
|
}
|
|
#define le2me_RECT(h) { \
|
|
(h)->left = le2me_16((h)->left); \
|
|
(h)->top = le2me_16((h)->top); \
|
|
(h)->right = le2me_16((h)->right); \
|
|
(h)->bottom = le2me_16((h)->bottom); \
|
|
}
|
|
#define le2me_BITMAPINFOHEADER(h) { \
|
|
(h)->biSize = le2me_32((h)->biSize); \
|
|
(h)->biWidth = le2me_32((h)->biWidth); \
|
|
(h)->biHeight = le2me_32((h)->biHeight); \
|
|
(h)->biPlanes = le2me_16((h)->biPlanes); \
|
|
(h)->biBitCount = le2me_16((h)->biBitCount); \
|
|
(h)->biCompression = le2me_32((h)->biCompression); \
|
|
(h)->biSizeImage = le2me_32((h)->biSizeImage); \
|
|
(h)->biXPelsPerMeter = le2me_32((h)->biXPelsPerMeter); \
|
|
(h)->biYPelsPerMeter = le2me_32((h)->biYPelsPerMeter); \
|
|
(h)->biClrUsed = le2me_32((h)->biClrUsed); \
|
|
(h)->biClrImportant = le2me_32((h)->biClrImportant); \
|
|
}
|
|
#define le2me_WAVEFORMATEX(h) { \
|
|
(h)->wFormatTag = le2me_16((h)->wFormatTag); \
|
|
(h)->nChannels = le2me_16((h)->nChannels); \
|
|
(h)->nSamplesPerSec = le2me_32((h)->nSamplesPerSec); \
|
|
(h)->nAvgBytesPerSec = le2me_32((h)->nAvgBytesPerSec); \
|
|
(h)->nBlockAlign = le2me_16((h)->nBlockAlign); \
|
|
(h)->wBitsPerSample = le2me_16((h)->wBitsPerSample); \
|
|
(h)->cbSize = le2me_16((h)->cbSize); \
|
|
}
|
|
#define le2me_AVIINDEXENTRY(h) { \
|
|
(h)->ckid = le2me_32((h)->ckid); \
|
|
(h)->dwFlags = le2me_32((h)->dwFlags); \
|
|
(h)->dwChunkOffset = le2me_32((h)->dwChunkOffset); \
|
|
(h)->dwChunkLength = le2me_32((h)->dwChunkLength); \
|
|
}
|
|
#else
|
|
#define le2me_MainAVIHeader(h) /**/
|
|
#define le2me_AVIStreamHeader(h) /**/
|
|
#define le2me_RECT(h) /**/
|
|
#define le2me_BITMAPINFOHEADER(h) /**/
|
|
#define le2me_WAVEFORMATEX(h) /**/
|
|
#define le2me_AVIINDEXENTRY(h) /**/
|
|
#endif
|
|
|
|
|
|
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);
|
|
if(index_mode==-2) break; // reading from non-seekable source (stdin)
|
|
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)));
|
|
le2me_MainAVIHeader(&avih); // swap to machine endian
|
|
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)));
|
|
le2me_AVIStreamHeader(&h); // swap to machine endian
|
|
chunksize-=MIN(size2,sizeof(h));
|
|
++stream_id;
|
|
if(h.fccType==streamtypeVIDEO){
|
|
sh_video=new_sh_video(demuxer,stream_id);
|
|
memcpy(&sh_video->video,&h,sizeof(h));
|
|
} else
|
|
if(h.fccType==streamtypeAUDIO){
|
|
sh_audio=new_sh_audio(demuxer,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);
|
|
le2me_BITMAPINFOHEADER(sh_video->bih); // swap to machine endian
|
|
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('D', 'I', 'V', '2'):
|
|
case mmioFOURCC('A', 'P', '4', '1'):
|
|
idxfix_divx=1; // we can fix keyframes only for divx coded files!
|
|
}
|
|
} else
|
|
if(last_fccType==streamtypeAUDIO){
|
|
int wf_size = chunksize<sizeof(WAVEFORMATEX)?sizeof(WAVEFORMATEX):chunksize;
|
|
sh_audio->wf=calloc(wf_size,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);
|
|
le2me_WAVEFORMATEX(sh_audio->wf);
|
|
if (sh_audio->wf->cbSize != 0 &&
|
|
wf_size < sizeof(WAVEFORMATEX)+sh_audio->wf->cbSize) {
|
|
sh_audio->wf=realloc(sh_audio->wf, sizeof(WAVEFORMATEX)+sh_audio->wf->cbSize);
|
|
}
|
|
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){
|
|
int i;
|
|
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);
|
|
for (i = 0; i < demuxer->idx_size; i++) // swap index to machine endian
|
|
le2me_AVIINDEXENTRY((AVIINDEXENTRY*)demuxer->idx + i);
|
|
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
|
|
|