Google Git Google Git
Sign in
third-party-mirror/gpac/refs/heads/main/./src/media_tools/avilib.c
blob: b2158b0c7a4f22a248ebd03b55e249d875c3b013 [file] [edit]
/*
* avilib.c
*
* Copyright (C) Thomas ostreich - June 2001
* multiple audio track support Copyright (C) 2002 Thomas ostreich
*
* Original code:
* Copyright (C) 1999 Rainer Johanni <Rainer@Johanni.de>
*
* This file is part of transcode, a linux video stream processing tool
*
* transcode is free software; you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation; either version 2, or (at your option)
* any later version.
*
* transcode is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License aint with this library; see the file COPYING. If not, write to
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
*
*/
#include <gpac/setup.h>
#ifndef GPAC_DISABLE_AVILIB
#if defined(__GNUC__) && __GNUC__ >= 11
#pragma GCC diagnostic ignored "-Wstringop-overflow"
#pragma GCC diagnostic ignored "-Warray-bounds"
#endif
#include <gpac/internal/avilib.h>
#define INFO_LIST
// add a new riff chunk after XX MB
#define NEW_RIFF_THRES (1900*1024*1024)
// Maximum number of indices per stream
#define NR_IXNN_CHUNKS 96
#define DEBUG_ODML
#undef DEBUG_ODML
/* The following variable indicates the kind of error */
int AVI_errno = 0;
#define MAX_INFO_STRLEN 64
static char id_str[MAX_INFO_STRLEN];
#define FRAME_RATE_SCALE 1000000
/*******************************************************************
* *
* Utilities for writing an AVI File *
* *
*******************************************************************/
static u32 avi_read(FILE *fd, char *buf, u32 len)
{
u32 r = 0;
while (r < len) {
s32 n = (s32) gf_fread(buf + r, len - r, fd);
if (n == 0) break;
if (n < 0) return r;
r += n;
}
return r;
}
static u32 avi_write (FILE *fd, char *buf, u32 len)
{
u32 r = 0;
while (r < len) {
s32 n = (u32) gf_fwrite (buf + r, len - r, fd);
if (n < 0)
return n;
r += n;
}
return r;
}
/* HEADERBYTES: The number of bytes to reserve for the header */
#define HEADERBYTES 2048
/* AVI_MAX_LEN: The maximum length of an AVI file, we stay a bit below
the 2GB limit (Remember: 2*10^9 is smaller than 2 GB) */
#define AVI_MAX_LEN (UINT_MAX-(1<<20)*16-HEADERBYTES)
#define PAD_EVEN(x) ( ((x)+1) & ~1 )
/* Copy n into dst as a 4 or 2 byte, little endian number.
Should also work on big endian machines */
static void long2str(unsigned char *dst, s32 n)
{
dst[0] = (n )&0xff;
dst[1] = (n>> 8)&0xff;
dst[2] = (n>>16)&0xff;
dst[3] = (n>>24)&0xff;
}
#ifdef WORDS_BIGENDIAN
static void short2str(unsigned char *dst, s32 n)
{
dst[0] = (n )&0xff;
dst[1] = (n>> 8)&0xff;
}
#endif
/* Convert a string of 4 or 2 bytes to a number,
also working on big endian machines */
static u64 str2ullong(unsigned char *str)
{
u64 r = ((u32)str[0] | ((u32)str[1]<<8) | ((u32)str[2]<<16) | ((u32)str[3]<<24));
u64 s = ((u32)str[4] | ((u32)str[5]<<8) | ((u32)str[6]<<16) | ((u32)str[7]<<24));
#ifdef __GNUC__
return ((s<<32)&0xffffffff00000000ULL)|(r&0xffffffff);
#else
return ((s<<32)&0xffffffff00000000)|(r&0xffffffff);
#endif
}
static u32 str2ulong(unsigned char *str)
{
return ( (u32)str[0] | ((u32)str[1]<<8) | ((u32)str[2]<<16) | ((u32)str[3]<<24) );
}
static u32 str2ushort(unsigned char *str)
{
return ( (u32)str[0] | ((u32)str[1]<<8) );
}
// bit 31 denotes a keyframe
static u32 str2ulong_len (unsigned char *str)
{
return str2ulong(str) & 0x7fffffff;
}
// if bit 31 is 0, its a keyframe
static u32 str2ulong_key (unsigned char *str)
{
u32 c = str2ulong(str);
c &= 0x80000000;
if (c == 0) return 0x10;
else return 0;
}
/* Calculate audio sample size from number of bits and number of channels.
This may have to be adjusted for eg. 12 bits and stereo */
static int avi_sampsize(avi_t *AVI, int j)
{
int s;
s = ((AVI->track[j].a_bits+7)/8)*AVI->track[j].a_chans;
// if(s==0) s=1; /* avoid possible zero divisions */
if(s<4) s=4; /* avoid possible zero divisions */
return s;
}
/* Add a chunk (=tag and data) to the AVI file,
returns -1 on write error, 0 on success */
static int avi_add_chunk(avi_t *AVI, unsigned char *tag, unsigned char *data, u32 length)
{
unsigned char c[8];
char p=0;
/* Copy tag and length int c, so that we need only 1 write system call
for these two values */
memcpy(c,tag,4);
long2str(c+4,length);
/* Output tag, length and data, restore previous position
if the write fails */
if( avi_write(AVI->fdes,(char *)c,8) != 8 ||
avi_write(AVI->fdes,(char *)data,length) != length ||
avi_write(AVI->fdes,&p,length&1) != (length&1)) // if len is uneven, write a pad byte
{
gf_fseek(AVI->fdes,AVI->pos,SEEK_SET);
AVI_errno = AVI_ERR_WRITE;
return -1;
}
/* Update file position */
AVI->pos += 8 + PAD_EVEN(length);
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] pos=%lu %s\n", AVI->pos, tag));
return 0;
}
#define OUTD(n) long2str((unsigned char*) (ix00+bl),(s32)n); bl+=4
#define OUTW(n) ix00[bl] = (n)&0xff; ix00[bl+1] = (n>>8)&0xff; bl+=2
#define OUTC(n) ix00[bl] = (n)&0xff; bl+=1
#define OUTS(s) memcpy(ix00+bl,s,4); bl+=4
// this does the physical writeout of the ix## structure
static int avi_ixnn_entry(avi_t *AVI, avistdindex_chunk *ch, avisuperindex_entry *en)
{
int bl;
u32 k;
unsigned int max = ch->nEntriesInUse * sizeof (u32) * ch->wLongsPerEntry + 24; // header
char *ix00 = (char *)gf_malloc (max);
char dfcc[5];
memcpy (dfcc, ch->fcc, 4);
dfcc[4] = 0;
bl = 0;
if (en) {
en->qwOffset = AVI->pos;
en->dwSize = max;
//en->dwDuration = ch->nEntriesInUse -1; // NUMBER OF stream ticks == frames for video/samples for audio
}
#ifdef DEBUG_ODML
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] ODML Write %s: Entries %ld size %d \n", dfcc, ch->nEntriesInUse, max));
#endif
//OUTS(ch->fcc);
//OUTD(max);
OUTW(ch->wLongsPerEntry);
OUTC(ch->bIndexSubType);
OUTC(ch->bIndexType);
OUTD(ch->nEntriesInUse);
OUTS(ch->dwChunkId);
OUTD(ch->qwBaseOffset&0xffffffff);
OUTD((ch->qwBaseOffset>>32)&0xffffffff);
OUTD(ch->dwReserved3);
for (k = 0; k < ch->nEntriesInUse; k++) {
OUTD(ch->aIndex[k].dwOffset);
OUTD(ch->aIndex[k].dwSize);
}
avi_add_chunk (AVI, (unsigned char*)ch->fcc, (unsigned char*)ix00, max);
gf_free(ix00);
return 0;
}
#undef OUTS
#undef OUTW
#undef OUTD
#undef OUTC
// inits a super index structure including its enclosed stdindex
static int avi_init_super_index(avi_t *AVI, unsigned char *idxtag, avisuperindex_chunk **si)
{
int k;
avisuperindex_chunk *sil = NULL;
if ((sil = (avisuperindex_chunk *) gf_malloc (sizeof (avisuperindex_chunk))) == NULL) {
AVI_errno = AVI_ERR_NO_MEM;
return -1;
}
memcpy (sil->fcc, "indx", 4);
sil->dwSize = 0; // size of this chunk
sil->wLongsPerEntry = 4;
sil->bIndexSubType = 0;
sil->bIndexType = AVI_INDEX_OF_INDEXES;
sil->nEntriesInUse = 0; // none are in use
memcpy (sil->dwChunkId, idxtag, 4);
memset (sil->dwReserved, 0, sizeof (sil->dwReserved));
// NR_IXNN_CHUNKS == allow 32 indices which means 32 GB files -- arbitrary
sil->aIndex = (avisuperindex_entry *) gf_malloc (sil->wLongsPerEntry * NR_IXNN_CHUNKS * sizeof (void*));
if (!sil->aIndex) {
AVI_errno = AVI_ERR_NO_MEM;
return -1;
}
memset (sil->aIndex, 0, sil->wLongsPerEntry * NR_IXNN_CHUNKS * sizeof (u32));
sil->stdindex = (avistdindex_chunk **)gf_malloc (NR_IXNN_CHUNKS * sizeof (avistdindex_chunk *));
if (!sil->stdindex) {
AVI_errno = AVI_ERR_NO_MEM;
return -1;
}
for (k = 0; k < NR_IXNN_CHUNKS; k++) {
sil->stdindex[k] = (avistdindex_chunk *) gf_malloc (sizeof (avistdindex_chunk));
// gets rewritten later
sil->stdindex[k]->qwBaseOffset = (u64)k * AVI->new_riff_threshold;
sil->stdindex[k]->aIndex = NULL;
}
*si = sil;
return 0;
}
// fills an alloc'ed stdindex structure and mallocs some entries for the actual chunks
static int avi_add_std_index(avi_t *AVI, unsigned char *idxtag, unsigned char *strtag,
avistdindex_chunk *stdil)
{
memcpy (stdil->fcc, idxtag, 4);
stdil->dwSize = 4096;
stdil->wLongsPerEntry = 2; //sizeof(avistdindex_entry)/sizeof(u32);
stdil->bIndexSubType = 0;
stdil->bIndexType = AVI_INDEX_OF_CHUNKS;
stdil->nEntriesInUse = 0;
// cp 00db ChunkId
memcpy(stdil->dwChunkId, strtag, 4);
//stdil->qwBaseOffset = AVI->video_superindex->aIndex[ cur_std_idx ]->qwOffset;
stdil->aIndex = (avistdindex_entry *)gf_malloc(stdil->dwSize * sizeof (u32) * stdil->wLongsPerEntry);
if (!stdil->aIndex) {
AVI_errno = AVI_ERR_NO_MEM;
return -1;
}
return 0;
}
static int avi_add_odml_index_entry_core(avi_t *AVI, int flags, u64 pos, unsigned int len, avistdindex_chunk *si)
{
u32 cur_chunk_idx;
// put new chunk into index
si->nEntriesInUse++;
cur_chunk_idx = si->nEntriesInUse-1;
// need to fetch more memory
if (cur_chunk_idx >= si->dwSize) {
si->dwSize += 4096;
si->aIndex = (avistdindex_entry *)gf_realloc ( si->aIndex, si->dwSize * sizeof (u32) * si->wLongsPerEntry);
}
if(len>AVI->max_len) AVI->max_len=len;
// if bit 31 is set, it is NOT a keyframe
if (flags != 0x10) {
len |= 0x80000000;
}
si->aIndex [ cur_chunk_idx ].dwSize = len;
si->aIndex [ cur_chunk_idx ].dwOffset = (u32) (pos - si->qwBaseOffset + 8);
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] ODML: POS: 0x%lX\n", si->aIndex [ cur_chunk_idx ].dwOffset));
return 0;
}
static int avi_add_odml_index_entry(avi_t *AVI, unsigned char *tag, int flags, u64 pos, unsigned int len)
{
char fcc[5];
int audio = (strchr ((char*)tag, 'w')?1:0);
int video = !audio;
unsigned int cur_std_idx;
u32 audtr;
s64 towrite = 0;
if (video) {
if (!AVI->video_superindex) {
if (avi_init_super_index(AVI, (unsigned char *)"ix00", &AVI->video_superindex) < 0) return -1;
AVI->video_superindex->nEntriesInUse++;
cur_std_idx = AVI->video_superindex->nEntriesInUse-1;
if (avi_add_std_index (AVI, (unsigned char *)"ix00", (unsigned char *)"00db", AVI->video_superindex->stdindex[ cur_std_idx ]) < 0)
return -1;
} // init
} // video
if (audio) {
fcc[0] = 'i';
fcc[1] = 'x';
fcc[2] = tag[0];
fcc[3] = tag[1];
fcc[4] = '\0';
if (!AVI->track[AVI->aptr].audio_superindex) {
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] ODML: fcc = %s\n", fcc));
#endif
if (avi_init_super_index(AVI, (unsigned char *)fcc, &AVI->track[AVI->aptr].audio_superindex) < 0) return -1;
AVI->track[AVI->aptr].audio_superindex->nEntriesInUse++;
sprintf(fcc, "ix%02d", AVI->aptr+1);
if (avi_add_std_index (AVI, (unsigned char *)fcc, tag, AVI->track[AVI->aptr].audio_superindex->stdindex[
AVI->track[AVI->aptr].audio_superindex->nEntriesInUse - 1 ]) < 0
) return -1;
} // init
}
towrite = 0;
if (AVI->video_superindex) {
cur_std_idx = AVI->video_superindex->nEntriesInUse-1;
towrite += AVI->video_superindex->stdindex[cur_std_idx]->nEntriesInUse*8
+ 4+4+2+1+1+4+4+8+4;
if (cur_std_idx == 0) {
towrite += AVI->n_idx*16 + 8;
towrite += HEADERBYTES;
}
}
for (audtr=0; audtr<AVI->anum; audtr++) {
if (AVI->track[audtr].audio_superindex) {
cur_std_idx = AVI->track[audtr].audio_superindex->nEntriesInUse-1;
towrite += AVI->track[audtr].audio_superindex->stdindex[cur_std_idx]->nEntriesInUse*8
+ 4+4+2+1+1+4+4+8+4;
}
}
towrite += len + (len&1) + 8;
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] ODML: towrite = 0x%llX = %"LLD"\n", towrite, towrite));
if (AVI->video_superindex &&
(s64)(AVI->pos+towrite) > (s64)((s64) AVI->new_riff_threshold*AVI->video_superindex->nEntriesInUse)) {
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] Adding a new RIFF chunk: %d\n", AVI->video_superindex->nEntriesInUse));
// rotate ALL indices
AVI->video_superindex->nEntriesInUse++;
cur_std_idx = AVI->video_superindex->nEntriesInUse-1;
if (AVI->video_superindex->nEntriesInUse > NR_IXNN_CHUNKS) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] Internal error in avilib - redefine NR_IXNN_CHUNKS\n"));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] cur_std_idx=%d NR_IXNN_CHUNKS=%d"
"POS=%"LLD" towrite=%"LLD"\n",
cur_std_idx,NR_IXNN_CHUNKS, AVI->pos, towrite));
return -1;
}
if (avi_add_std_index (AVI, (unsigned char *)"ix00", (unsigned char *)"00db", AVI->video_superindex->stdindex[ cur_std_idx ]) < 0)
return -1;
for (audtr = 0; audtr < AVI->anum; audtr++) {
char aud[5];
if (!AVI->track[audtr].audio_superindex) {
// not initialized -> no index
continue;
}
AVI->track[audtr].audio_superindex->nEntriesInUse++;
sprintf(fcc, "ix%02d", audtr+1);
sprintf(aud, "0%01dwb", audtr+1);
if (avi_add_std_index (AVI, (unsigned char *)fcc, (unsigned char *)aud, AVI->track[audtr].audio_superindex->stdindex[
AVI->track[audtr].audio_superindex->nEntriesInUse - 1 ]) < 0
) return -1;
}
// write the new riff;
if (cur_std_idx > 0) {
// dump the _previous_ == already finished index
avi_ixnn_entry (AVI, AVI->video_superindex->stdindex[cur_std_idx - 1],
&AVI->video_superindex->aIndex[cur_std_idx - 1]);
AVI->video_superindex->aIndex[cur_std_idx - 1].dwDuration =
AVI->video_superindex->stdindex[cur_std_idx - 1]->nEntriesInUse - 1;
for (audtr = 0; audtr < AVI->anum; audtr++) {
if (!AVI->track[audtr].audio_superindex) {
// not initialized -> no index
continue;
}
avi_ixnn_entry (AVI, AVI->track[audtr].audio_superindex->stdindex[cur_std_idx - 1],
&AVI->track[audtr].audio_superindex->aIndex[cur_std_idx - 1]);
AVI->track[audtr].audio_superindex->aIndex[cur_std_idx - 1].dwDuration =
AVI->track[audtr].audio_superindex->stdindex[cur_std_idx - 1]->nEntriesInUse - 1;
if (AVI->track[audtr].a_fmt == 0x1) {
AVI->track[audtr].audio_superindex->aIndex[cur_std_idx - 1].dwDuration *=
AVI->track[audtr].a_bits*AVI->track[audtr].a_rate*AVI->track[audtr].a_chans/800;
}
}
// XXX: dump idx1 structure
if (cur_std_idx == 1) {
avi_add_chunk(AVI, (unsigned char *)"idx1", (unsigned char *)AVI->idx, AVI->n_idx*16);
// qwBaseOffset will contain the start of the second riff chunk
}
// Fix the Offsets later at closing time
avi_add_chunk(AVI, (unsigned char *)"RIFF", (unsigned char *)"AVIXLIST\0\0\0\0movi", 16);
AVI->video_superindex->stdindex[ cur_std_idx ]->qwBaseOffset = AVI->pos -16 -8;
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] ODML: RIFF No.%02d at Offset 0x%llX\n", cur_std_idx, AVI->pos -16 -8));
#endif
for (audtr = 0; audtr < AVI->anum; audtr++) {
if (AVI->track[audtr].audio_superindex)
AVI->track[audtr].audio_superindex->stdindex[ cur_std_idx ]->qwBaseOffset =
AVI->pos -16 -8;
}
// now we can be sure
AVI->is_opendml++;
}
}
if (video) {
avi_add_odml_index_entry_core(AVI, flags, AVI->pos, len,
AVI->video_superindex->stdindex[ AVI->video_superindex->nEntriesInUse-1 ]);
AVI->total_frames++;
} // video
if (audio) {
avi_add_odml_index_entry_core(AVI, flags, AVI->pos, len,
AVI->track[AVI->aptr].audio_superindex->stdindex[
AVI->track[AVI->aptr].audio_superindex->nEntriesInUse-1 ]);
}
return 0;
}
// #undef NR_IXNN_CHUNKS
static int avi_add_index_entry(avi_t *AVI, unsigned char *tag, int flags, u64 pos, u64 len)
{
if(AVI->n_idx>=AVI->max_idx) {
void *ptr = gf_realloc((void *)AVI->idx,(AVI->max_idx+4096)*16);
if(ptr == 0) {
AVI_errno = AVI_ERR_NO_MEM;
return -1;
}
AVI->max_idx += 4096;
AVI->idx = (unsigned char((*)[16]) ) ptr;
}
/* Add index entry */
// GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] INDEX %s %ld %lu %lu\n", tag, flags, pos, len));
memcpy(AVI->idx[AVI->n_idx],tag,4);
long2str(AVI->idx[AVI->n_idx]+ 4,flags);
long2str(AVI->idx[AVI->n_idx]+ 8, (s32) pos);
long2str(AVI->idx[AVI->n_idx]+12, (s32) len);
/* Update counter */
AVI->n_idx++;
if(len>AVI->max_len) AVI->max_len=(u32) len;
return 0;
}
#if 0
/* Returns 1 if more audio is in that video junk */
int AVI_can_read_audio(avi_t *AVI)
{
if(AVI->mode==AVI_MODE_WRITE) {
return -1;
}
if(!AVI->video_index) {
return -1;
}
if(!AVI->track[AVI->aptr].audio_index) {
return -1;
}
// is it -1? the last ones got left out --tibit
//if (AVI->track[AVI->aptr].audio_posc>=AVI->track[AVI->aptr].audio_chunks-1) {
if (AVI->track[AVI->aptr].audio_posc>=AVI->track[AVI->aptr].audio_chunks) {
return 0;
}
if (AVI->video_pos >= AVI->video_frames) return 1;
if (AVI->track[AVI->aptr].audio_index[AVI->track[AVI->aptr].audio_posc].pos < AVI->video_index[AVI->video_pos].pos) return 1;
else return 0;
}
#endif
/*
AVI_open_output_file: Open an AVI File and write a bunch
of zero bytes as space for the header.
returns a pointer to avi_t on success, a zero pointer on error
*/
GF_EXPORT
avi_t* AVI_open_output_file(char * filename, u64 opendml_threshold)
{
avi_t *AVI;
int i;
unsigned char AVI_header[HEADERBYTES];
/* Allocate the avi_t struct and zero it */
AVI = (avi_t *) gf_malloc(sizeof(avi_t));
if(AVI==0)
{
AVI_errno = AVI_ERR_NO_MEM;
return 0;
}
memset((void *)AVI,0,sizeof(avi_t));
AVI->fdes = gf_fopen(filename, "w+b");
if (!AVI->fdes )
{
AVI_errno = AVI_ERR_OPEN;
gf_free(AVI);
return 0;
}
/* Write out HEADERBYTES bytes, the header will go here
when we are finished with writing */
for (i=0; i<HEADERBYTES; i++) AVI_header[i] = 0;
i = avi_write(AVI->fdes,(char *)AVI_header,HEADERBYTES);
if (i != HEADERBYTES)
{
gf_fclose(AVI->fdes);
AVI_errno = AVI_ERR_WRITE;
gf_free(AVI);
return 0;
}
AVI->pos = HEADERBYTES;
AVI->mode = AVI_MODE_WRITE; /* open for writing */
if (opendml_threshold)
AVI->new_riff_threshold = opendml_threshold;
else
AVI->new_riff_threshold = (1900*1024*1024);
//init
AVI->anum = 0;
AVI->aptr = 0;
return AVI;
}
GF_EXPORT
void AVI_set_video(avi_t *AVI, int width, int height, double fps, char *compressor)
{
/* may only be called if file is open for writing */
if(AVI->mode==AVI_MODE_READ) return;
AVI->width = width;
AVI->height = height;
AVI->fps = fps;
if(strncmp(compressor, "RGB", 3)==0) {
memset(AVI->compressor, 0, 4);
} else {
memcpy(AVI->compressor,compressor,4);
}
AVI->compressor[4] = 0;
avi_update_header(AVI);
}
GF_EXPORT
void AVI_set_audio(avi_t *AVI, int channels, int rate, int bits, int format, int mp3rate)
{
/* may only be called if file is open for writing */
if(AVI->mode==AVI_MODE_READ) return;
//inc audio tracks
AVI->aptr=AVI->anum;
++AVI->anum;
if(AVI->anum > AVI_MAX_TRACKS) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] error - only %d audio tracks supported\n", AVI_MAX_TRACKS));
exit(1);
}
AVI->track[AVI->aptr].a_chans = channels;
AVI->track[AVI->aptr].a_rate = rate;
AVI->track[AVI->aptr].a_bits = bits;
AVI->track[AVI->aptr].a_fmt = format;
AVI->track[AVI->aptr].mp3rate = mp3rate;
avi_update_header(AVI);
}
#define OUT4CC(s) \
if(nhb<=HEADERBYTES-4) memcpy(AVI_header+nhb,s,4); \
nhb += 4
#define OUTLONG(n) \
if(nhb<=HEADERBYTES-4) long2str(AVI_header+nhb, (s32)(n)); \
nhb += 4
#define OUTSHRT(n) \
if(nhb<=HEADERBYTES-2) { \
AVI_header[nhb ] = (u8) ((n )&0xff); \
AVI_header[nhb+1] = (u8) ((n>>8)&0xff); \
} \
nhb += 2
#define OUTCHR(n) \
if(nhb<=HEADERBYTES-1) { \
AVI_header[nhb ] = (n )&0xff; \
} \
nhb += 1
#define OUTMEM(d, s) \
{ \
u32 s_ = (u32) (s); \
if(nhb + s_ <= HEADERBYTES) \
memcpy(AVI_header+nhb, (d), s_); \
nhb += s_; \
}
//ThOe write preliminary AVI file header: 0 frames, max vid/aud size
int avi_update_header(avi_t *AVI)
{
int njunk, ms_per_frame, frate, flag;
int movi_len, hdrl_start, strl_start;
u32 j;
unsigned char AVI_header[HEADERBYTES];
u32 nhb;
unsigned int xd_size, xd_size_align2;
//assume max size
movi_len = AVI_MAX_LEN - HEADERBYTES + 4;
//assume index will be written
// int hasIndex=1;
if(AVI->fps < 0.001) {
frate=0;
ms_per_frame=0;
} else {
frate = (int) (FRAME_RATE_SCALE*AVI->fps + 0.5);
ms_per_frame=(int) (1000000/AVI->fps + 0.5);
}
/* Prepare the file header */
nhb = 0;
/* The RIFF header */
OUT4CC ("RIFF");
OUTLONG(movi_len); // assume max size
OUT4CC ("AVI ");
/* Start the header list */
OUT4CC ("LIST");
OUTLONG(0); /* Length of list in bytes, don't know yet */
hdrl_start = nhb; /* Store start position */
OUT4CC ("hdrl");
/* The main AVI header */
/* The Flags in AVI File header */
#define AVIF_HASINDEX 0x00000010 /* Index at end of file */
#define AVIF_MUSTUSEINDEX 0x00000020
#define AVIF_ISINTERLEAVED 0x00000100
#define AVIF_TRUSTCKTYPE 0x00000800 /* Use CKType to find key frames */
#define AVIF_WASCAPTUREFILE 0x00010000
#define AVIF_COPYRIGHTED 0x00020000
OUT4CC ("avih");
OUTLONG(56); /* # of bytes to follow */
OUTLONG(ms_per_frame); /* Microseconds per frame */
//ThOe ->0
// OUTLONG(10000000); /* MaxBytesPerSec, I hope this will never be used */
OUTLONG(0);
OUTLONG(0); /* PaddingGranularity (whatever that might be) */
/* Other sources call it 'reserved' */
flag = AVIF_ISINTERLEAVED;
//if (hasIndex)
flag |= AVIF_HASINDEX;
if (/*hasIndex && */AVI->must_use_index)
flag |= AVIF_MUSTUSEINDEX;
OUTLONG(flag); /* Flags */
OUTLONG(0); // no frames yet
OUTLONG(0); /* InitialFrames */
OUTLONG(AVI->anum+1);
OUTLONG(0); /* SuggestedBufferSize */
OUTLONG(AVI->width); /* Width */
OUTLONG(AVI->height); /* Height */
/* MS calls the following 'reserved': */
OUTLONG(0); /* TimeScale: Unit used to measure time */
OUTLONG(0); /* DataRate: Data rate of playback */
OUTLONG(0); /* StartTime: Starting time of AVI data */
OUTLONG(0); /* DataLength: Size of AVI data chunk */
/* Start the video stream list ---------------------------------- */
OUT4CC ("LIST");
OUTLONG(0); /* Length of list in bytes, don't know yet */
strl_start = nhb; /* Store start position */
OUT4CC ("strl");
/* The video stream header */
OUT4CC ("strh");
OUTLONG(56); /* # of bytes to follow */
OUT4CC ("vids"); /* Type */
OUT4CC (AVI->compressor); /* Handler */
OUTLONG(0); /* Flags */
OUTLONG(0); /* Reserved, MS says: wPriority, wLanguage */
OUTLONG(0); /* InitialFrames */
OUTLONG(FRAME_RATE_SCALE); /* Scale */
OUTLONG(frate); /* Rate: Rate/Scale == samples/second */
OUTLONG(0); /* Start */
OUTLONG(0); // no frames yet
OUTLONG(0); /* SuggestedBufferSize */
OUTLONG(-1); /* Quality */
OUTLONG(0); /* SampleSize */
OUTLONG(0); /* Frame */
OUTLONG(0); /* Frame */
// OUTLONG(0); /* Frame */
//OUTLONG(0); /* Frame */
/* The video stream format */
xd_size = AVI->extradata_size;
xd_size_align2 = (AVI->extradata_size+1) & ~1;
OUT4CC ("strf");
OUTLONG(40 + xd_size_align2);/* # of bytes to follow */
OUTLONG(40 + xd_size); /* Size */
OUTLONG(AVI->width); /* Width */
OUTLONG(AVI->height); /* Height */
OUTSHRT(1);
OUTSHRT(24); /* Planes, Count */
OUT4CC (AVI->compressor); /* Compression */
// ThOe (*3)
OUTLONG(AVI->width*AVI->height*3); /* SizeImage (in bytes?) */
OUTLONG(0); /* XPelsPerMeter */
OUTLONG(0); /* YPelsPerMeter */
OUTLONG(0); /* ClrUsed: Number of colors used */
OUTLONG(0); /* ClrImportant: Number of colors important */
// write extradata
if (xd_size > 0 && AVI->extradata) {
OUTMEM(AVI->extradata, xd_size);
if (xd_size != xd_size_align2) {
OUTCHR(0);
}
}
/* Finish stream list, i.e. put number of bytes in the list to proper pos */
long2str(AVI_header+strl_start-4,nhb-strl_start);
/* Start the audio stream list ---------------------------------- */
for(j=0; j<AVI->anum; ++j) {
int sampsize = avi_sampsize(AVI, j);
OUT4CC ("LIST");
OUTLONG(0); /* Length of list in bytes, don't know yet */
strl_start = nhb; /* Store start position */
OUT4CC ("strl");
/* The audio stream header */
OUT4CC ("strh");
OUTLONG(56); /* # of bytes to follow */
OUT4CC ("auds");
// -----------
// ThOe
OUTLONG(0); /* Format (Optionally) */
// -----------
OUTLONG(0); /* Flags */
OUTLONG(0); /* Reserved, MS says: wPriority, wLanguage */
OUTLONG(0); /* InitialFrames */
// ThOe /4
OUTLONG(sampsize/4); /* Scale */
OUTLONG(1000*AVI->track[j].mp3rate/8);
OUTLONG(0); /* Start */
OUTLONG(4*AVI->track[j].audio_bytes/sampsize); /* Length */
OUTLONG(0); /* SuggestedBufferSize */
OUTLONG(-1); /* Quality */
// ThOe /4
OUTLONG(sampsize/4); /* SampleSize */
OUTLONG(0); /* Frame */
OUTLONG(0); /* Frame */
// OUTLONG(0); /* Frame */
//OUTLONG(0); /* Frame */
/* The audio stream format */
OUT4CC ("strf");
OUTLONG(16); /* # of bytes to follow */
OUTSHRT(AVI->track[j].a_fmt); /* Format */
OUTSHRT(AVI->track[j].a_chans); /* Number of channels */
OUTLONG(AVI->track[j].a_rate); /* SamplesPerSec */
// ThOe
OUTLONG(1000*AVI->track[j].mp3rate/8);
//ThOe (/4)
OUTSHRT(sampsize/4); /* BlockAlign */
OUTSHRT(AVI->track[j].a_bits); /* BitsPerSample */
/* Finish stream list, i.e. put number of bytes in the list to proper pos */
long2str(AVI_header+strl_start-4,nhb-strl_start);
}
/* Finish header list */
long2str(AVI_header+hdrl_start-4,nhb-hdrl_start);
/* Calculate the needed amount of junk bytes, output junk */
njunk = HEADERBYTES - nhb - 8 - 12;
/* Safety first: if njunk <= 0, somebody has played with
HEADERBYTES without knowing what (s)he did.
This is a fatal error */
if(njunk<=0)
{
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] AVI_close_output_file: # of header bytes too small\n"));
exit(1);
}
OUT4CC ("JUNK");
OUTLONG(njunk);
memset(AVI_header+nhb,0,njunk);
nhb += njunk;
/* Start the movi list */
OUT4CC ("LIST");
OUTLONG(movi_len); /* Length of list in bytes */
OUT4CC ("movi");
/* Output the header, truncate the file to the number of bytes
actually written, report an error if someting goes wrong */
if ( (gf_fseek(AVI->fdes, 0, SEEK_SET) ==(u64)-1) ||
avi_write(AVI->fdes,(char *)AVI_header,HEADERBYTES)!=HEADERBYTES ||
(gf_fseek(AVI->fdes,AVI->pos,SEEK_SET)==(u64)-1)
) {
AVI_errno = AVI_ERR_CLOSE;
return -1;
}
return 0;
}
//SLM
#ifndef S_IRUSR
#define S_IRWXU 00700 /* read, write, execute: owner */
#define S_IRUSR 00400 /* read permission: owner */
#define S_IWUSR 00200 /* write permission: owner */
#define S_IXUSR 00100 /* execute permission: owner */
#define S_IRWXG 00070 /* read, write, execute: group */
#define S_IRGRP 00040 /* read permission: group */
#define S_IWGRP 00020 /* write permission: group */
#define S_IXGRP 00010 /* execute permission: group */
#define S_IRWXO 00007 /* read, write, execute: other */
#define S_IROTH 00004 /* read permission: other */
#define S_IWOTH 00002 /* write permission: other */
#define S_IXOTH 00001 /* execute permission: other */
#endif
/*
Write the header of an AVI file and close it.
returns 0 on success, -1 on write error.
*/
static int avi_close_output_file(avi_t *AVI)
{
int njunk, hasIndex, ms_per_frame, frate, idxerror, flag;
u64 movi_len;
int hdrl_start, strl_start;
u32 j;
unsigned char AVI_header[HEADERBYTES];
int nhb;
unsigned int xd_size, xd_size_align2;
#ifdef INFO_LIST
int info_len;
int id_len, real_id_len;
int info_start_pos;
// time_t calptr;
#endif
/* Calculate length of movi list */
// dump the rest of the index
if (AVI->is_opendml) {
int cur_std_idx = AVI->video_superindex->nEntriesInUse-1;
u32 audtr;
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] ODML dump the rest indices\n"));
#endif
avi_ixnn_entry (AVI, AVI->video_superindex->stdindex[cur_std_idx],
&AVI->video_superindex->aIndex[cur_std_idx]);
AVI->video_superindex->aIndex[cur_std_idx].dwDuration =
AVI->video_superindex->stdindex[cur_std_idx]->nEntriesInUse - 1;
for (audtr = 0; audtr < AVI->anum; audtr++) {
if (!AVI->track[audtr].audio_superindex) {
// not initialized -> no index
continue;
}
avi_ixnn_entry (AVI, AVI->track[audtr].audio_superindex->stdindex[cur_std_idx],
&AVI->track[audtr].audio_superindex->aIndex[cur_std_idx]);
AVI->track[audtr].audio_superindex->aIndex[cur_std_idx].dwDuration =
AVI->track[audtr].audio_superindex->stdindex[cur_std_idx]->nEntriesInUse - 1;
if (AVI->track[audtr].a_fmt == 0x1) {
AVI->track[audtr].audio_superindex->aIndex[cur_std_idx].dwDuration *=
AVI->track[audtr].a_bits*AVI->track[audtr].a_rate*AVI->track[audtr].a_chans/800;
}
}
// The AVI->video_superindex->nEntriesInUse contains the offset
AVI->video_superindex->stdindex[ cur_std_idx+1 ]->qwBaseOffset = AVI->pos;
// Correct!
movi_len = AVI->video_superindex->stdindex[ 1 ]->qwBaseOffset - HEADERBYTES+4 - AVI->n_idx*16 - 8;
} else {
movi_len = AVI->pos - HEADERBYTES + 4;
}
/* Try to output the index entries. This may fail e.g. if no space
is left on device. We will report this as an error, but we still
try to write the header correctly (so that the file still may be
readable in the most cases */
idxerror = 0;
hasIndex = 1;
if (!AVI->is_opendml) {
// GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] pos=%lu, index_len=%ld \n", AVI->pos, AVI->n_idx*16));
int ret = avi_add_chunk(AVI, (unsigned char *)"idx1", (unsigned char *)AVI->idx, AVI->n_idx*16);
hasIndex = (ret==0);
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] pos=%lu, index_len=%d\n", AVI->pos, hasIndex));
if(ret) {
idxerror = 1;
AVI_errno = AVI_ERR_WRITE_INDEX;
}
}
/* Calculate Microseconds per frame */
if(AVI->fps < 0.001) {
frate=0;
ms_per_frame=0;
} else {
frate = (int) (FRAME_RATE_SCALE*AVI->fps + 0.5);
ms_per_frame=(int) (1000000/AVI->fps + 0.5);
}
/* Prepare the file header */
nhb = 0;
/* The RIFF header */
OUT4CC ("RIFF");
if (AVI->is_opendml) {
OUTLONG(AVI->video_superindex->stdindex[ 1 ]->qwBaseOffset - 8); /* # of bytes to follow */
} else {
OUTLONG(AVI->pos - 8); /* # of bytes to follow */
}
OUT4CC ("AVI ");
/* Start the header list */
OUT4CC ("LIST");
OUTLONG(0); /* Length of list in bytes, don't know yet */
hdrl_start = nhb; /* Store start position */
OUT4CC ("hdrl");
/* The main AVI header */
/* The Flags in AVI File header */
#define AVIF_HASINDEX 0x00000010 /* Index at end of file */
#define AVIF_MUSTUSEINDEX 0x00000020
#define AVIF_ISINTERLEAVED 0x00000100
#define AVIF_TRUSTCKTYPE 0x00000800 /* Use CKType to find key frames */
#define AVIF_WASCAPTUREFILE 0x00010000
#define AVIF_COPYRIGHTED 0x00020000
OUT4CC ("avih");
OUTLONG(56); /* # of bytes to follow */
OUTLONG(ms_per_frame); /* Microseconds per frame */
//ThOe ->0
// OUTLONG(10000000); /* MaxBytesPerSec, I hope this will never be used */
OUTLONG(0);
OUTLONG(0); /* PaddingGranularity (whatever that might be) */
/* Other sources call it 'reserved' */
flag = AVIF_ISINTERLEAVED;
if(hasIndex) flag |= AVIF_HASINDEX;
if(hasIndex && AVI->must_use_index) flag |= AVIF_MUSTUSEINDEX;
OUTLONG(flag); /* Flags */
OUTLONG(AVI->video_frames); /* TotalFrames */
OUTLONG(0); /* InitialFrames */
OUTLONG(AVI->anum+1);
// if (AVI->track[0].audio_bytes)
// { OUTLONG(2); } /* Streams */
// else
// { OUTLONG(1); } /* Streams */
OUTLONG(0); /* SuggestedBufferSize */
OUTLONG(AVI->width); /* Width */
OUTLONG(AVI->height); /* Height */
/* MS calls the following 'reserved': */
OUTLONG(0); /* TimeScale: Unit used to measure time */
OUTLONG(0); /* DataRate: Data rate of playback */
OUTLONG(0); /* StartTime: Starting time of AVI data */
OUTLONG(0); /* DataLength: Size of AVI data chunk */
/* Start the video stream list ---------------------------------- */
OUT4CC ("LIST");
OUTLONG(0); /* Length of list in bytes, don't know yet */
strl_start = nhb; /* Store start position */
OUT4CC ("strl");
/* The video stream header */
OUT4CC ("strh");
OUTLONG(56); /* # of bytes to follow */
OUT4CC ("vids"); /* Type */
OUT4CC (AVI->compressor); /* Handler */
OUTLONG(0); /* Flags */
OUTLONG(0); /* Reserved, MS says: wPriority, wLanguage */
OUTLONG(0); /* InitialFrames */
OUTLONG(FRAME_RATE_SCALE); /* Scale */
OUTLONG(frate); /* Rate: Rate/Scale == samples/second */
OUTLONG(0); /* Start */
OUTLONG(AVI->video_frames); /* Length */
OUTLONG(AVI->max_len); /* SuggestedBufferSize */
OUTLONG(0); /* Quality */
OUTLONG(0); /* SampleSize */
OUTLONG(0); /* Frame */
OUTLONG(0); /* Frame */
//OUTLONG(0); /* Frame */
//OUTLONG(0); /* Frame */
/* The video stream format */
xd_size = AVI->extradata_size;
xd_size_align2 = (AVI->extradata_size+1) & ~1;
OUT4CC ("strf");
OUTLONG(40 + xd_size_align2);/* # of bytes to follow */
OUTLONG(40 + xd_size); /* Size */
OUTLONG(AVI->width); /* Width */
OUTLONG(AVI->height); /* Height */
OUTSHRT(1);
OUTSHRT(24); /* Planes, Count */
OUT4CC (AVI->compressor); /* Compression */
// ThOe (*3)
OUTLONG(AVI->width*AVI->height*3); /* SizeImage (in bytes?) */
OUTLONG(0); /* XPelsPerMeter */
OUTLONG(0); /* YPelsPerMeter */
OUTLONG(0); /* ClrUsed: Number of colors used */
OUTLONG(0); /* ClrImportant: Number of colors important */
// write extradata if present
if (xd_size > 0 && AVI->extradata) {
OUTMEM(AVI->extradata, xd_size);
if (xd_size != xd_size_align2) {
OUTCHR(0);
}
}
// dump index of indices for audio
if (AVI->is_opendml) {
u32 k;
OUT4CC(AVI->video_superindex->fcc);
OUTLONG(2+1+1+4+4+3*4 + AVI->video_superindex->nEntriesInUse * (8+4+4));
OUTSHRT(AVI->video_superindex->wLongsPerEntry);
OUTCHR(AVI->video_superindex->bIndexSubType);
OUTCHR(AVI->video_superindex->bIndexType);
OUTLONG(AVI->video_superindex->nEntriesInUse);
OUT4CC(AVI->video_superindex->dwChunkId);
OUTLONG(0);
OUTLONG(0);
OUTLONG(0);
for (k = 0; k < AVI->video_superindex->nEntriesInUse; k++) {
u32 r = (u32) ((AVI->video_superindex->aIndex[k].qwOffset >> 32) & 0xffffffff);
u32 s = (u32) ((AVI->video_superindex->aIndex[k].qwOffset) & 0xffffffff);
OUTLONG(s);
OUTLONG(r);
OUTLONG(AVI->video_superindex->aIndex[k].dwSize);
OUTLONG(AVI->video_superindex->aIndex[k].dwDuration);
}
}
/* Finish stream list, i.e. put number of bytes in the list to proper pos */
long2str(AVI_header+strl_start-4,nhb-strl_start);
/* Start the audio stream list ---------------------------------- */
for(j=0; j<AVI->anum; ++j) {
//if (AVI->track[j].a_chans && AVI->track[j].audio_bytes)
{
unsigned int nBlockAlign = 0;
unsigned int avgbsec = 0;
unsigned int scalerate = 0;
int sampsize = avi_sampsize(AVI, j);
sampsize = AVI->track[j].a_fmt==0x1?sampsize*4:sampsize;
nBlockAlign = (AVI->track[j].a_rate<32000)?576:1152;
/*
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] XXX sampsize (%d) block (%ld) rate (%ld) audio_bytes (%ld) mp3rate(%ld,%ld)\n",
sampsize, nBlockAlign, AVI->track[j].a_rate,
(int int)AVI->track[j].audio_bytes,
1000*AVI->track[j].mp3rate/8, AVI->track[j].mp3rate));
*/
if (AVI->track[j].a_fmt==0x1) {
sampsize = (AVI->track[j].a_chans<2)?sampsize/2:sampsize;
avgbsec = AVI->track[j].a_rate*sampsize/4;
scalerate = AVI->track[j].a_rate*sampsize/4;
} else {
avgbsec = 1000*AVI->track[j].mp3rate/8;
scalerate = 1000*AVI->track[j].mp3rate/8;
}
OUT4CC ("LIST");
OUTLONG(0); /* Length of list in bytes, don't know yet */
strl_start = nhb; /* Store start position */
OUT4CC ("strl");
/* The audio stream header */
OUT4CC ("strh");
OUTLONG(56); /* # of bytes to follow */
OUT4CC ("auds");
// -----------
// ThOe
OUTLONG(0); /* Format (Optionally) */
// -----------
OUTLONG(0); /* Flags */
OUTLONG(0); /* Reserved, MS says: wPriority, wLanguage */
OUTLONG(0); /* InitialFrames */
// VBR
if (AVI->track[j].a_fmt == 0x55 && AVI->track[j].a_vbr) {
OUTLONG(nBlockAlign); /* Scale */
OUTLONG(AVI->track[j].a_rate); /* Rate */
OUTLONG(0); /* Start */
OUTLONG(AVI->track[j].audio_chunks); /* Length */
OUTLONG(0); /* SuggestedBufferSize */
OUTLONG(0); /* Quality */
OUTLONG(0); /* SampleSize */
OUTLONG(0); /* Frame */
OUTLONG(0); /* Frame */
} else {
OUTLONG(sampsize/4); /* Scale */
OUTLONG(scalerate); /* Rate */
OUTLONG(0); /* Start */
OUTLONG(4*AVI->track[j].audio_bytes/sampsize); /* Length */
OUTLONG(0); /* SuggestedBufferSize */
OUTLONG(0xffffffff); /* Quality */
OUTLONG(sampsize/4); /* SampleSize */
OUTLONG(0); /* Frame */
OUTLONG(0); /* Frame */
}
/* The audio stream format */
OUT4CC ("strf");
if (AVI->track[j].a_fmt == 0x55 && AVI->track[j].a_vbr) {
OUTLONG(30); /* # of bytes to follow */ // mplayer writes 28
OUTSHRT(AVI->track[j].a_fmt); /* Format */ // 2
OUTSHRT(AVI->track[j].a_chans); /* Number of channels */ // 2
OUTLONG(AVI->track[j].a_rate); /* SamplesPerSec */ // 4
//ThOe/tibit
OUTLONG(1000*AVI->track[j].mp3rate/8); /* maybe we should write an avg. */ // 4
OUTSHRT(nBlockAlign); /* BlockAlign */ // 2
OUTSHRT(AVI->track[j].a_bits); /* BitsPerSample */ // 2
OUTSHRT(12); /* cbSize */ // 2
OUTSHRT(1); /* wID */ // 2
OUTLONG(2); /* fdwFlags */ // 4
OUTSHRT(nBlockAlign); /* nBlockSize */ // 2
OUTSHRT(1); /* nFramesPerBlock */ // 2
OUTSHRT(0); /* nCodecDelay */ // 2
} else if (AVI->track[j].a_fmt == 0x55 && !AVI->track[j].a_vbr) {
OUTLONG(30); /* # of bytes to follow */
OUTSHRT(AVI->track[j].a_fmt); /* Format */
OUTSHRT(AVI->track[j].a_chans); /* Number of channels */
OUTLONG(AVI->track[j].a_rate); /* SamplesPerSec */
//ThOe/tibit
OUTLONG(1000*AVI->track[j].mp3rate/8);
OUTSHRT(sampsize/4); /* BlockAlign */
OUTSHRT(AVI->track[j].a_bits); /* BitsPerSample */
OUTSHRT(12); /* cbSize */
OUTSHRT(1); /* wID */
OUTLONG(2); /* fdwFlags */
OUTSHRT(nBlockAlign); /* nBlockSize */
OUTSHRT(1); /* nFramesPerBlock */
OUTSHRT(0); /* nCodecDelay */
} else {
OUTLONG(18); /* # of bytes to follow */
OUTSHRT(AVI->track[j].a_fmt); /* Format */
OUTSHRT(AVI->track[j].a_chans); /* Number of channels */
OUTLONG(AVI->track[j].a_rate); /* SamplesPerSec */
//ThOe/tibit
OUTLONG(avgbsec); /* Avg bytes/sec */
OUTSHRT(sampsize/4); /* BlockAlign */
OUTSHRT(AVI->track[j].a_bits); /* BitsPerSample */
OUTSHRT(0); /* cbSize */
}
}
if (AVI->is_opendml) {
u32 k;
if (!AVI->track[j].audio_superindex) {
// not initialized -> no index
continue;
}
OUT4CC(AVI->track[j].audio_superindex->fcc); /* "indx" */
OUTLONG(2+1+1+4+4+3*4 + AVI->track[j].audio_superindex->nEntriesInUse * (8+4+4));
OUTSHRT(AVI->track[j].audio_superindex->wLongsPerEntry);
OUTCHR(AVI->track[j].audio_superindex->bIndexSubType);
OUTCHR(AVI->track[j].audio_superindex->bIndexType);
OUTLONG(AVI->track[j].audio_superindex->nEntriesInUse);
OUT4CC(AVI->track[j].audio_superindex->dwChunkId);
OUTLONG(0);
OUTLONG(0);
OUTLONG(0);
for (k = 0; k < AVI->track[j].audio_superindex->nEntriesInUse; k++) {
u32 r = (u32) ((AVI->track[j].audio_superindex->aIndex[k].qwOffset >> 32) & 0xffffffff);
u32 s = (u32) ((AVI->track[j].audio_superindex->aIndex[k].qwOffset) & 0xffffffff);
/*
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] AUD[%d] NrEntries %d/%ld (%c%c%c%c) |0x%llX|%ld|%ld| \n", j, k,
AVI->track[j].audio_superindex->nEntriesInUse,
AVI->track[j].audio_superindex->dwChunkId[0], AVI->track[j].audio_superindex->dwChunkId[1],
AVI->track[j].audio_superindex->dwChunkId[2], AVI->track[j].audio_superindex->dwChunkId[3],
AVI->track[j].audio_superindex->aIndex[k].qwOffset,
AVI->track[j].audio_superindex->aIndex[k].dwSize,
AVI->track[j].audio_superindex->aIndex[k].dwDuration
));
*/
OUTLONG(s);
OUTLONG(r);
OUTLONG(AVI->track[j].audio_superindex->aIndex[k].dwSize);
OUTLONG(AVI->track[j].audio_superindex->aIndex[k].dwDuration);
}
}
/* Finish stream list, i.e. put number of bytes in the list to proper pos */
long2str(AVI_header+strl_start-4,nhb-strl_start);
}
if (AVI->is_opendml) {
OUT4CC("LIST");
OUTLONG(16);
OUT4CC("odml");
OUT4CC("dmlh");
OUTLONG(4);
OUTLONG(AVI->total_frames);
}
/* Finish header list */
long2str(AVI_header+hdrl_start-4,nhb-hdrl_start);
// add INFO list --- (0.6.0pre4)
#ifdef INFO_LIST
OUT4CC ("LIST");
info_start_pos = nhb;
info_len = MAX_INFO_STRLEN + 12;
OUTLONG(info_len); // rewritten later
OUT4CC ("INFO");
OUT4CC ("ISFT");
//OUTLONG(MAX_INFO_STRLEN);
memset(id_str, 0, MAX_INFO_STRLEN);
if (gf_sys_is_test_mode()) {
snprintf(id_str, MAX_INFO_STRLEN, "GPAC/avilib");
} else {
snprintf(id_str, MAX_INFO_STRLEN, "GPAC/avilib-%s", gf_gpac_version());
}
real_id_len = id_len = (u32) strlen(id_str)+1;
if (id_len&1) id_len++;
OUTLONG(real_id_len);
memset(AVI_header+nhb, 0, id_len);
memcpy(AVI_header+nhb, id_str, id_len);
nhb += id_len;
info_len = 0;
// write correct len
long2str(AVI_header+info_start_pos, info_len + id_len + 4+4+4);
nhb += info_len;
// OUT4CC ("ICMT");
// OUTLONG(MAX_INFO_STRLEN);
// calptr=time(NULL);
// sprintf(id_str, "\t%s %s", ctime(&calptr), "");
// memset(AVI_header+nhb, 0, MAX_INFO_STRLEN);
// memcpy(AVI_header+nhb, id_str, 25);
// nhb += MAX_INFO_STRLEN;
#endif
// ----------------------------
/* Calculate the needed amount of junk bytes, output junk */
njunk = HEADERBYTES - nhb - 8 - 12;
/* Safety first: if njunk <= 0, somebody has played with
HEADERBYTES without knowing what (s)he did.
This is a fatal error */
if(njunk<=0)
{
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] AVI_close_output_file: # of header bytes too small\n"));
exit(1);
}
OUT4CC ("JUNK");
OUTLONG(njunk);
memset(AVI_header+nhb,0,njunk);
nhb += njunk;
/* Start the movi list */
OUT4CC ("LIST");
OUTLONG(movi_len); /* Length of list in bytes */
OUT4CC ("movi");
/* Output the header, truncate the file to the number of bytes
actually written, report an error if someting goes wrong */
if ( (gf_fseek(AVI->fdes,0,SEEK_SET)==(u64)-1) ||
avi_write(AVI->fdes,(char *)AVI_header,HEADERBYTES)!=HEADERBYTES
// || ftruncate(AVI->fdes,AVI->pos)<0
)
{
AVI_errno = AVI_ERR_CLOSE;
return -1;
}
// Fix up the empty additional RIFF and LIST chunks
if (AVI->is_opendml) {
u32 k;
char f[4];
u32 len;
for (k=1; k<AVI->video_superindex->nEntriesInUse; k++) {
// the len of the RIFF Chunk
gf_fseek(AVI->fdes, AVI->video_superindex->stdindex[k]->qwBaseOffset+4, SEEK_SET);
len = (u32) (AVI->video_superindex->stdindex[k+1]->qwBaseOffset - AVI->video_superindex->stdindex[k]->qwBaseOffset - 8);
long2str((unsigned char *)f, len);
avi_write(AVI->fdes, f, 4);
// len of the LIST/movi chunk
gf_fseek(AVI->fdes, 8, SEEK_CUR);
len -= 12;
long2str((unsigned char *)f, len);
avi_write(AVI->fdes, f, 4);
}
}
if(idxerror) return -1;
return 0;
}
/*
AVI_write_data:
Add video or audio data to the file;
Return values:
0 No error;
-1 Error, AVI_errno is set appropriatly;
*/
static int avi_write_data(avi_t *AVI, char *data, unsigned int length, int audio, int keyframe)
{
int n = 0;
unsigned char astr[5];
// transcode core itself checks for the size -- unneeded and
// does harm to xvid 2pass encodes where the first pass can get
// _very_ large -- tibit.
#if 0
/* Check for maximum file length */
if ( (AVI->pos + 8 + length + 8 + (AVI->n_idx+1)*16) > AVI_MAX_LEN ) {
AVI_errno = AVI_ERR_SIZELIM;
return -1;
}
#endif
/* Add index entry */
//set tag for current audio track
sprintf((char *)astr, "0%1dwb", (int)(AVI->aptr+1));
if(audio) {
if (!AVI->is_opendml) n = avi_add_index_entry(AVI,astr,0x10,AVI->pos,length);
n += avi_add_odml_index_entry(AVI,astr,0x10,AVI->pos,length);
} else {
if (!AVI->is_opendml) n = avi_add_index_entry(AVI,(unsigned char *)"00db",((keyframe)?0x10:0x0),AVI->pos,length);
n += avi_add_odml_index_entry(AVI,(unsigned char *)"00db",((keyframe)?0x10:0x0),AVI->pos,length);
}
if(n) return -1;
/* Output tag and data */
if(audio)
n = avi_add_chunk(AVI,(unsigned char *)astr, (unsigned char *)data, length);
else
n = avi_add_chunk(AVI,(unsigned char *)"00db", (unsigned char *)data, length);
if (n) return -1;
return 0;
}
GF_EXPORT
int AVI_write_frame(avi_t *AVI, u8 *data, int bytes, int keyframe)
{
s64 pos;
if(AVI->mode==AVI_MODE_READ) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
pos = AVI->pos;
if(avi_write_data(AVI,data,bytes,0,keyframe)) return -1;
AVI->last_pos = pos;
AVI->last_len = bytes;
AVI->video_frames++;
return 0;
}
#if 0 //unused
int AVI_dup_frame(avi_t *AVI)
{
if(AVI->mode==AVI_MODE_READ) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(AVI->last_pos==0) return 0; /* No previous real frame */
if(avi_add_index_entry(AVI,(unsigned char *)"00db",0x10,AVI->last_pos,AVI->last_len)) return -1;
AVI->video_frames++;
AVI->must_use_index = 1;
return 0;
}
#endif
GF_EXPORT
int AVI_write_audio(avi_t *AVI, u8 *data, int bytes)
{
if(AVI->mode==AVI_MODE_READ) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if( avi_write_data(AVI,data,bytes,1,0) ) return -1;
AVI->track[AVI->aptr].audio_bytes += bytes;
AVI->track[AVI->aptr].audio_chunks++;
return 0;
}
#if 0 //unused
int AVI_append_audio(avi_t *AVI, u8 *data, int bytes)
{
// won't work for >2gb
int i, length, pos;
unsigned char c[4];
if(AVI->mode==AVI_MODE_READ) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
// update last index entry:
--AVI->n_idx;
length = str2ulong(AVI->idx[AVI->n_idx]+12);
pos = str2ulong(AVI->idx[AVI->n_idx]+8);
//update;
long2str(AVI->idx[AVI->n_idx]+12,length+bytes);
++AVI->n_idx;
AVI->track[AVI->aptr].audio_bytes += bytes;
//update chunk header
gf_fseek(AVI->fdes, pos+4, SEEK_SET);
long2str(c, length+bytes);
avi_write(AVI->fdes, (char *)c, 4);
gf_fseek(AVI->fdes, pos+8+length, SEEK_SET);
i=PAD_EVEN(length + bytes);
bytes = i - length;
avi_write(AVI->fdes, data, bytes);
AVI->pos = pos + 8 + i;
return 0;
}
u64 AVI_bytes_remain(avi_t *AVI)
{
if(AVI->mode==AVI_MODE_READ) return 0;
return ( AVI_MAX_LEN - (AVI->pos + 8 + 16*AVI->n_idx));
}
u64 AVI_bytes_written(avi_t *AVI)
{
if(AVI->mode==AVI_MODE_READ) return 0;
return (AVI->pos + 8 + 16*AVI->n_idx);
}
#endif
int AVI_set_audio_track(avi_t *AVI, u32 track)
{
if (track + 1 > AVI->anum) return(-1);
//this info is not written to file anyway
AVI->aptr=track;
return 0;
}
int AVI_get_audio_track(avi_t *AVI)
{
return(AVI->aptr);
}
#if 0 //unused
void AVI_set_audio_vbr(avi_t *AVI, int is_vbr)
{
AVI->track[AVI->aptr].a_vbr = is_vbr;
}
int AVI_get_audio_vbr(avi_t *AVI)
{
return(AVI->track[AVI->aptr].a_vbr);
}
#endif
/*******************************************************************
* *
* Utilities for reading video and audio from an AVI File *
* *
*******************************************************************/
GF_EXPORT
int AVI_close(avi_t *AVI)
{
int ret;
u32 j;
/* If the file was open for writing, the header and index still have
to be written */
if(AVI->mode == AVI_MODE_WRITE)
ret = avi_close_output_file(AVI);
else
ret = 0;
/* Even if there happened an error, we first clean up */
gf_fclose(AVI->fdes);
if(AVI->idx) gf_free(AVI->idx);
if(AVI->video_index) gf_free(AVI->video_index);
if(AVI->video_superindex) {
if(AVI->video_superindex->aIndex) gf_free(AVI->video_superindex->aIndex);
if (AVI->video_superindex->stdindex) {
for (j=0; j < NR_IXNN_CHUNKS; j++) {
if (AVI->video_superindex->stdindex[j]->aIndex)
gf_free(AVI->video_superindex->stdindex[j]->aIndex);
gf_free(AVI->video_superindex->stdindex[j]);
}
gf_free(AVI->video_superindex->stdindex);
}
gf_free(AVI->video_superindex);
}
for (j=0; j<AVI->anum; j++)
{
u32 k;
if(AVI->track[j].audio_index) gf_free(AVI->track[j].audio_index);
if(AVI->track[j].audio_superindex) {
avisuperindex_chunk *asi = AVI->track[j].audio_superindex;
if (asi->aIndex) gf_free(asi->aIndex);
if (asi->stdindex) {
for (k=0; k < NR_IXNN_CHUNKS; k++) {
if (!asi->stdindex[k]) continue;
if (asi->stdindex[k]->aIndex)
gf_free(asi->stdindex[k]->aIndex);
gf_free(asi->stdindex[k]);
}
gf_free(asi->stdindex);
}
gf_free(asi);
}
}
if (AVI->bitmap_info_header)
gf_free(AVI->bitmap_info_header);
for (j = 0; j < AVI->anum; j++)
if (AVI->wave_format_ex[j])
gf_free(AVI->wave_format_ex[j]);
if (AVI->extradata)
gf_free(AVI->extradata);
gf_free(AVI);
return ret;
}
#define ERR_EXIT(x) \
{ \
if (hdrl_data) { gf_free(hdrl_data); hdrl_data=NULL; }\
AVI_close(AVI); \
AVI_errno = x; \
return 0; \
}
avi_t *AVI_open_input_file(char *filename, int getIndex)
{
avi_t *AVI;
/* Create avi_t structure */
AVI = (avi_t *) gf_malloc(sizeof(avi_t));
if(AVI==NULL)
{
AVI_errno = AVI_ERR_NO_MEM;
return 0;
}
memset((void *)AVI,0,sizeof(avi_t));
AVI->mode = AVI_MODE_READ; /* open for reading */
/* Open the file */
AVI->fdes = gf_fopen(filename,"rb");
if(!AVI->fdes )
{
AVI_errno = AVI_ERR_OPEN;
gf_free(AVI);
return 0;
}
AVI_errno = 0;
avi_parse_input_file(AVI, getIndex);
if (AVI != NULL && !AVI_errno) {
AVI->aptr=0; //reset
}
if (AVI_errno) return NULL;
return AVI;
}
#if 0
avi_t *AVI_open_fd(FILE *fd, int getIndex)
{
avi_t *AVI=NULL;
/* Create avi_t structure */
AVI = (avi_t *) gf_malloc(sizeof(avi_t));
if(AVI==NULL)
{
AVI_errno = AVI_ERR_NO_MEM;
return 0;
}
memset((void *)AVI,0,sizeof(avi_t));
AVI->mode = AVI_MODE_READ; /* open for reading */
// file alread open
AVI->fdes = fd;
AVI_errno = 0;
avi_parse_input_file(AVI, getIndex);
if (AVI != NULL && !AVI_errno) {
AVI->aptr=0; //reset
}
if (AVI_errno)
return AVI=NULL;
else
return AVI;
}
#endif
int avi_parse_input_file(avi_t *AVI, int getIndex)
{
int rate, scale, idx_type;
s64 n, i;
unsigned char *hdrl_data = NULL;
u64 header_offset=0;
int hdrl_len=0;
int nvi, nai[AVI_MAX_TRACKS], ioff;
u64 tot[AVI_MAX_TRACKS];
u32 j;
int lasttag = 0;
int vids_strh_seen = 0;
int vids_strf_seen = 0;
int auds_strh_seen = 0;
// int auds_strf_seen = 0;
int num_stream = 0;
char data[256];
s64 oldpos=-1, newpos=-1;
int aud_chunks = 0;
if (!AVI) {
AVI_errno = AVI_ERR_OPEN;
return 0;
}
/* Read first 12 bytes and check that this is an AVI file */
if (avi_read(AVI->fdes,data,12) != 12 )
ERR_EXIT(AVI_ERR_READ)
if (strnicmp(data ,"RIFF",4) !=0 || strnicmp(data+8,"AVI ",4) !=0 )
ERR_EXIT(AVI_ERR_NO_AVI)
/* Go through the AVI file and extract the header list,
the start position of the 'movi' list and an optionally
present idx1 tag */
hdrl_data = 0;
while(1)
{
if( avi_read(AVI->fdes,data,8) != 8 ) break; /* We assume it's EOF */
newpos = gf_ftell(AVI->fdes);
if(oldpos==newpos) {
/* This is a broken AVI stream... */
return -1;
}
oldpos=newpos;
n = str2ulong((unsigned char *)data+4);
n = PAD_EVEN(n);
if(strnicmp(data,"LIST",4) == 0)
{
if( avi_read(AVI->fdes,data,4) != 4 ) ERR_EXIT(AVI_ERR_READ)
n -= 4;
if(strnicmp(data,"hdrl",4) == 0)
{
if (n>0xFFFFFFFF) ERR_EXIT(AVI_ERR_READ)
hdrl_len = (u32) n;
if (hdrl_data)
gf_free(hdrl_data);
hdrl_data = (unsigned char *) gf_malloc((u32)n);
if(hdrl_data==0) ERR_EXIT(AVI_ERR_NO_MEM);
// offset of header
header_offset = gf_ftell(AVI->fdes);
if( avi_read(AVI->fdes,(char *)hdrl_data, (u32) n) != n ) {
ERR_EXIT(AVI_ERR_READ)
}
}
else if(strnicmp(data,"movi",4) == 0)
{
AVI->movi_start = gf_ftell(AVI->fdes);
if (gf_fseek(AVI->fdes,n,SEEK_CUR)==(u64)-1) break;
}
else if (gf_fseek(AVI->fdes,n,SEEK_CUR)==(u64)-1) break;
}
else if(strnicmp(data,"idx1",4) == 0)
{
/* n must be a multiple of 16, but the reading does not
break if this is not the case */
AVI->n_idx = AVI->max_idx = (u32) (n/16);
if (AVI->idx) gf_free(AVI->idx);
AVI->idx = (unsigned char((*)[16]) ) gf_malloc((u32)n);
if(AVI->idx==0) {
ERR_EXIT(AVI_ERR_NO_MEM)
}
if(avi_read(AVI->fdes, (char *) AVI->idx, (u32) n) != n ) {
gf_free( AVI->idx);
AVI->idx=NULL;
AVI->n_idx = 0;
ERR_EXIT(AVI_ERR_READ)
}
}
else
gf_fseek(AVI->fdes,n,SEEK_CUR);
}
if(!hdrl_data) ERR_EXIT(AVI_ERR_NO_HDRL)
if(!AVI->movi_start) ERR_EXIT(AVI_ERR_NO_MOVI)
/* Interpret the header list */
for(i=0; i<hdrl_len;)
{
if (i+4>hdrl_len) ERR_EXIT(AVI_ERR_READ)
/* List tags are completly ignored */
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] TAG %c%c%c%c\n", (hdrl_data+i)[0], (hdrl_data+i)[1], (hdrl_data+i)[2], (hdrl_data+i)[3]));
#endif
if(strnicmp((char *)hdrl_data+i,"LIST",4)==0) {
i+= 12;
continue;
}
if (i+8>hdrl_len) ERR_EXIT(AVI_ERR_READ)
n = str2ulong(hdrl_data+i+4);
n = PAD_EVEN(n);
if (i+n>hdrl_len) ERR_EXIT(AVI_ERR_READ)
/* Interpret the tag and its args */
if(strnicmp((char *)hdrl_data+i,"strh",4)==0)
{
i += 8;
if (i+4>hdrl_len) ERR_EXIT(AVI_ERR_READ)
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] TAG %c%c%c%c\n", (hdrl_data+i)[0], (hdrl_data+i)[1], (hdrl_data+i)[2], (hdrl_data+i)[3]));
#endif
if(strnicmp((char *)hdrl_data+i,"vids",4) == 0 && !vids_strh_seen)
{
if ( (i+32+4>hdrl_len) || (i+sizeof(alAVISTREAMHEADER))>hdrl_len ) ERR_EXIT(AVI_ERR_READ)
memcpy(AVI->compressor,hdrl_data+i+4,4);
AVI->compressor[4] = 0;
// ThOe
AVI->v_codech_off = header_offset + i+4;
scale = str2ulong(hdrl_data+i+20);
rate = str2ulong(hdrl_data+i+24);
if(scale!=0) AVI->fps = (double)rate/(double)scale;
AVI->video_frames = str2ulong(hdrl_data+i+32);
AVI->video_strn = num_stream;
AVI->max_len = 0;
vids_strh_seen = 1;
lasttag = 1; /* vids */
memcpy(&AVI->video_stream_header, hdrl_data + i,
sizeof(alAVISTREAMHEADER));
}
else if (strnicmp ((char *)hdrl_data+i,"auds",4) ==0 && ! auds_strh_seen)
{
//inc audio tracks
if(AVI->anum >= AVI_MAX_TRACKS) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] error - only %d audio tracks supported\n", AVI_MAX_TRACKS));
ERR_EXIT(AVI_ERR_READ)
}
AVI->aptr=AVI->anum;
++AVI->anum;
if ( (i+44+4>hdrl_len) || (i+sizeof(alAVISTREAMHEADER))>hdrl_len ) ERR_EXIT(AVI_ERR_READ)
AVI->track[AVI->aptr].audio_bytes = str2ulong(hdrl_data+i+32)*avi_sampsize(AVI, 0);
AVI->track[AVI->aptr].audio_strn = num_stream;
// if samplesize==0 -> vbr
AVI->track[AVI->aptr].a_vbr = !str2ulong(hdrl_data+i+44);
AVI->track[AVI->aptr].padrate = str2ulong(hdrl_data+i+24);
memcpy(&AVI->stream_headers[AVI->aptr], hdrl_data + i, sizeof(alAVISTREAMHEADER));
// auds_strh_seen = 1;
lasttag = 2; /* auds */
// ThOe
AVI->track[AVI->aptr].a_codech_off = header_offset + i;
}
else if (strnicmp ((char*)hdrl_data+i,"iavs",4) ==0 && ! auds_strh_seen) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] AVILIB: error - DV AVI Type 1 no supported\n"));
ERR_EXIT(AVI_ERR_READ)
}
else
lasttag = 0;
num_stream++;
}
else if(strnicmp((char*)hdrl_data+i,"dmlh",4) == 0) {
if (i+8+4>hdrl_len) ERR_EXIT(AVI_ERR_READ)
AVI->total_frames = str2ulong(hdrl_data+i+8);
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] real number of frames %d\n", AVI->total_frames));
#endif
i += 8;
}
else if(strnicmp((char *)hdrl_data+i,"strf",4)==0)
{
i += 8;
if(lasttag == 1)
{
if (i+sizeof(alBITMAPINFOHEADER)>hdrl_len) ERR_EXIT(AVI_ERR_READ)
alBITMAPINFOHEADER bih;
memcpy(&bih, hdrl_data + i, sizeof(alBITMAPINFOHEADER));
if (bih.bi_size < 4) ERR_EXIT(AVI_ERR_READ)
bih.bi_size = str2ulong((unsigned char *)&bih.bi_size);
if (i + bih.bi_size > hdrl_len) ERR_EXIT(AVI_ERR_READ)
if (AVI->bitmap_info_header)
gf_free(AVI->bitmap_info_header);
AVI->bitmap_info_header = (alBITMAPINFOHEADER *) gf_malloc(bih.bi_size);
if (AVI->bitmap_info_header != NULL)
memcpy(AVI->bitmap_info_header, hdrl_data + i, bih.bi_size);
if (i+16+4>hdrl_len) ERR_EXIT(AVI_ERR_READ)
AVI->width = str2ulong(hdrl_data+i+4);
AVI->height = str2ulong(hdrl_data+i+8);
vids_strf_seen = 1;
//ThOe
AVI->v_codecf_off = header_offset + i+16;
memcpy(AVI->compressor2, hdrl_data+i+16, 4);
AVI->compressor2[4] = 0;
if ((n>40) && (hdrl_len > (i+40))) {
if (n>0xFFFFFFFF) ERR_EXIT(AVI_ERR_READ)
AVI->extradata_size = (u32) MIN(n - 40, hdrl_len-i-40);
AVI->extradata = gf_malloc(sizeof(u8)* AVI->extradata_size);
if (!AVI->extradata) ERR_EXIT(AVI_ERR_NO_MEM)
memcpy(AVI->extradata, hdrl_data + i + 40, AVI->extradata_size);
}
}
else if(lasttag == 2)
{
if (i>=hdrl_len) ERR_EXIT(AVI_ERR_READ)
alWAVEFORMATEX *wfe;
char *nwfe;
int wfes;
if ((u32) (hdrl_len - i) < sizeof(alWAVEFORMATEX))
wfes = (int) (hdrl_len - i);
else
wfes = sizeof(alWAVEFORMATEX);
wfe = (alWAVEFORMATEX *)gf_malloc(sizeof(alWAVEFORMATEX));
if (wfe != NULL) {
memset(wfe, 0, sizeof(alWAVEFORMATEX));
memcpy(wfe, hdrl_data + i, wfes);
if (str2ushort((unsigned char *)&wfe->cb_size) != 0) {
nwfe = (char *)
gf_realloc(wfe, sizeof(alWAVEFORMATEX) +
str2ushort((unsigned char *)&wfe->cb_size));
if (nwfe != 0) {
s64 lpos = gf_ftell(AVI->fdes);
gf_fseek(AVI->fdes, header_offset + i + sizeof(alWAVEFORMATEX),
SEEK_SET);
wfe = (alWAVEFORMATEX *)nwfe;
nwfe = &nwfe[sizeof(alWAVEFORMATEX)];
avi_read(AVI->fdes, nwfe,
str2ushort((unsigned char *)&wfe->cb_size));
gf_fseek(AVI->fdes, lpos, SEEK_SET);
}
}
if (AVI->wave_format_ex[AVI->aptr])
gf_free(AVI->wave_format_ex[AVI->aptr]);
AVI->wave_format_ex[AVI->aptr] = wfe;
}
if (i+14+4>hdrl_len) ERR_EXIT(AVI_ERR_READ)
AVI->track[AVI->aptr].a_fmt = str2ushort(hdrl_data+i );
//ThOe
AVI->track[AVI->aptr].a_codecf_off = header_offset + i;
AVI->track[AVI->aptr].a_chans = str2ushort(hdrl_data+i+2);
AVI->track[AVI->aptr].a_rate = str2ulong (hdrl_data+i+4);
//ThOe: read mp3bitrate
AVI->track[AVI->aptr].mp3rate = 8*str2ulong(hdrl_data+i+8)/1000;
//:ThOe
AVI->track[AVI->aptr].a_bits = str2ushort(hdrl_data+i+14);
// auds_strf_seen = 1;
}
}
else if(strnicmp((char*)hdrl_data+i,"indx",4) == 0) {
char *a;
if(lasttag == 1) // V I D E O
{
a = (char*)hdrl_data+i;
int avail =(int) (hdrl_len-i);
if (avail<32) ERR_EXIT(AVI_ERR_READ)
AVI->video_superindex = (avisuperindex_chunk *) gf_malloc (sizeof (avisuperindex_chunk));
memset(AVI->video_superindex, 0, sizeof (avisuperindex_chunk));
memcpy (AVI->video_superindex->fcc, a, 4);
a += 4;
AVI->video_superindex->dwSize = str2ulong((unsigned char *)a);
a += 4;
AVI->video_superindex->wLongsPerEntry = str2ushort((unsigned char *)a);
a += 2;
AVI->video_superindex->bIndexSubType = *a;
a += 1;
AVI->video_superindex->bIndexType = *a;
a += 1;
AVI->video_superindex->nEntriesInUse = str2ulong((unsigned char *)a);
a += 4;
memcpy (AVI->video_superindex->dwChunkId, a, 4);
a += 4;
// 3 * reserved
a += 4;
a += 4;
a += 4;
if (AVI->video_superindex->bIndexSubType != 0) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] Invalid Header, bIndexSubType != 0\n"));
}
avail -= 32;
if (AVI->video_superindex->nEntriesInUse >= GF_INT_MAX/16 ||
avail < (int) AVI->video_superindex->nEntriesInUse*16) {
ERR_EXIT(AVI_ERR_READ)
}
// must be 4 so that wLongsPerEntry*sizeof(u32) == sizeof(avisuperindex_entry)
if (AVI->video_superindex->wLongsPerEntry != 4) {
ERR_EXIT(AVI_ERR_READ)
}
AVI->video_superindex->aIndex = (avisuperindex_entry*)
gf_malloc (AVI->video_superindex->wLongsPerEntry * AVI->video_superindex->nEntriesInUse * sizeof (u32));
// position of ix## chunks
for (j=0; j<AVI->video_superindex->nEntriesInUse; ++j) {
AVI->video_superindex->aIndex[j].qwOffset = str2ullong ((unsigned char*)a);
a += 8;
AVI->video_superindex->aIndex[j].dwSize = str2ulong ((unsigned char*)a);
a += 4;
AVI->video_superindex->aIndex[j].dwDuration = str2ulong ((unsigned char*)a);
a += 4;
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] [%d] 0x%llx 0x%lx %lu\n", j,
(unsigned int long)AVI->video_superindex->aIndex[j].qwOffset,
(unsigned long)AVI->video_superindex->aIndex[j].dwSize,
(unsigned long)AVI->video_superindex->aIndex[j].dwDuration));
#endif
}
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] FOURCC \"%c%c%c%c\"\n", AVI->video_superindex->fcc[0], AVI->video_superindex->fcc[1],
AVI->video_superindex->fcc[2], AVI->video_superindex->fcc[3]));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] LEN \"%ld\"\n", (long)AVI->video_superindex->dwSize));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] wLongsPerEntry \"%d\"\n", AVI->video_superindex->wLongsPerEntry));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] bIndexSubType \"%d\"\n", AVI->video_superindex->bIndexSubType));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] bIndexType \"%d\"\n", AVI->video_superindex->bIndexType));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] nEntriesInUse \"%ld\"\n", (long)AVI->video_superindex->nEntriesInUse));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] dwChunkId \"%c%c%c%c\"\n", AVI->video_superindex->dwChunkId[0], AVI->video_superindex->dwChunkId[1],
AVI->video_superindex->dwChunkId[2], AVI->video_superindex->dwChunkId[3]));
#endif
AVI->is_opendml = 1;
}
else if(lasttag == 2) // A U D I O
{
a = (char*) hdrl_data+i;
int avail = (int) (hdrl_len-i);
if (avail<32) ERR_EXIT(AVI_ERR_READ)
GF_SAFEALLOC(AVI->track[AVI->aptr].audio_superindex, avisuperindex_chunk);
memcpy (AVI->track[AVI->aptr].audio_superindex->fcc, a, 4);
a += 4;
AVI->track[AVI->aptr].audio_superindex->dwSize = str2ulong((unsigned char*)a);
a += 4;
AVI->track[AVI->aptr].audio_superindex->wLongsPerEntry = str2ushort((unsigned char*)a);
a += 2;
AVI->track[AVI->aptr].audio_superindex->bIndexSubType = *a;
a += 1;
AVI->track[AVI->aptr].audio_superindex->bIndexType = *a;
a += 1;
AVI->track[AVI->aptr].audio_superindex->nEntriesInUse = str2ulong((unsigned char*)a);
a += 4;
memcpy (AVI->track[AVI->aptr].audio_superindex->dwChunkId, a, 4);
a += 4;
// 3 * reserved
a += 4;
a += 4;
a += 4;
if (AVI->track[AVI->aptr].audio_superindex->bIndexSubType != 0) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] Invalid Header, bIndexSubType != 0\n"));
}
avail -= 32;
if (AVI->track[AVI->aptr].audio_superindex->nEntriesInUse >= GF_INT_MAX/16
|| avail < (int) AVI->track[AVI->aptr].audio_superindex->nEntriesInUse*16) {
ERR_EXIT(AVI_ERR_READ)
}
if (AVI->track[AVI->aptr].audio_superindex->nEntriesInUse) {
// must be 4 so that wLongsPerEntry*sizeof(u32) == sizeof(avisuperindex_entry)
if (AVI->track[AVI->aptr].audio_superindex->wLongsPerEntry != 4) {
ERR_EXIT(AVI_ERR_READ)
}
AVI->track[AVI->aptr].audio_superindex->aIndex = (avisuperindex_entry*)
gf_malloc (AVI->track[AVI->aptr].audio_superindex->wLongsPerEntry *
AVI->track[AVI->aptr].audio_superindex->nEntriesInUse * sizeof (u32));
}
else {
AVI->track[AVI->aptr].audio_superindex->aIndex = NULL;
}
// position of ix## chunks
for (j=0; j<AVI->track[AVI->aptr].audio_superindex->nEntriesInUse; ++j) {
AVI->track[AVI->aptr].audio_superindex->aIndex[j].qwOffset = str2ullong ((unsigned char*)a);
a += 8;
AVI->track[AVI->aptr].audio_superindex->aIndex[j].dwSize = str2ulong ((unsigned char*)a);
a += 4;
AVI->track[AVI->aptr].audio_superindex->aIndex[j].dwDuration = str2ulong ((unsigned char*)a);
a += 4;
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] [%d] 0x%llx 0x%lx %lu\n", j,
(unsigned int long)AVI->track[AVI->aptr].audio_superindex->aIndex[j].qwOffset,
(unsigned long)AVI->track[AVI->aptr].audio_superindex->aIndex[j].dwSize,
(unsigned long)AVI->track[AVI->aptr].audio_superindex->aIndex[j].dwDuration));
#endif
}
AVI->track[AVI->aptr].audio_superindex->stdindex = NULL;
#ifdef DEBUG_ODML
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] FOURCC \"%.4s\"\n", AVI->track[AVI->aptr].audio_superindex->fcc));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] LEN \"%ld\"\n", (long)AVI->track[AVI->aptr].audio_superindex->dwSize));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] wLongsPerEntry \"%d\"\n", AVI->track[AVI->aptr].audio_superindex->wLongsPerEntry));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] bIndexSubType \"%d\"\n", AVI->track[AVI->aptr].audio_superindex->bIndexSubType));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] bIndexType \"%d\"\n", AVI->track[AVI->aptr].audio_superindex->bIndexType));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] nEntriesInUse \"%ld\"\n", (long)AVI->track[AVI->aptr].audio_superindex->nEntriesInUse));
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] dwChunkId \"%.4s\"\n", AVI->track[AVI->aptr].audio_superindex->dwChunkId[0]));
#endif
}
i += 8;
}
else if((strnicmp((char*)hdrl_data+i,"JUNK",4) == 0) ||
(strnicmp((char*)hdrl_data+i,"strn",4) == 0) ||
(strnicmp((char*)hdrl_data+i,"vprp",4) == 0)) {
i += 8;
// do not reset lasttag
} else
{
i += 8;
lasttag = 0;
}
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] adding %ld bytes\n", (int int)n));
i += (u32) n;
}
gf_free(hdrl_data);
hdrl_data = NULL;
if(!vids_strh_seen || !vids_strf_seen) ERR_EXIT(AVI_ERR_NO_VIDS)
AVI->video_tag[0] = AVI->video_strn/10 + '0';
AVI->video_tag[1] = AVI->video_strn%10 + '0';
AVI->video_tag[2] = 'd';
AVI->video_tag[3] = 'b';
/* Audio tag is set to "99wb" if no audio present */
if(!AVI->track[0].a_chans) AVI->track[0].audio_strn = 99;
{
int tk=0;
for(j=0; j<AVI->anum+1 && tk<AVI_MAX_TRACKS; ++j) {
if (j == AVI->video_strn) continue;
AVI->track[tk].audio_tag[0] = j/10 + '0';
AVI->track[tk].audio_tag[1] = j%10 + '0';
AVI->track[tk].audio_tag[2] = 'w';
AVI->track[tk].audio_tag[3] = 'b';
++tk;
}
}
gf_fseek(AVI->fdes,AVI->movi_start,SEEK_SET);
if(!getIndex) return(0);
/* if the file has an idx1, check if this is relative
to the start of the file or to the start of the movi list */
idx_type = 0;
if(AVI->idx)
{
s64 pos, len;
/* Search the first videoframe in the idx1 and look where
it is in the file */
for(i=0; i<AVI->n_idx; i++)
if( strnicmp((char *)AVI->idx[i],(char *)AVI->video_tag,3)==0 ) break;
if(i>=AVI->n_idx) ERR_EXIT(AVI_ERR_NO_VIDS)
pos = str2ulong(AVI->idx[i]+ 8);
len = str2ulong(AVI->idx[i]+12);
gf_fseek(AVI->fdes,pos,SEEK_SET);
if(avi_read(AVI->fdes,data,8)!=8) ERR_EXIT(AVI_ERR_READ)
if( strnicmp(data,(char *)AVI->idx[i],4)==0 && str2ulong((unsigned char *)data+4)==len )
{
idx_type = 1; /* Index from start of file */
}
else
{
gf_fseek(AVI->fdes,pos+AVI->movi_start-4,SEEK_SET);
if(avi_read(AVI->fdes,data,8)!=8) ERR_EXIT(AVI_ERR_READ)
if( strnicmp(data,(char *)AVI->idx[i],4)==0 && str2ulong((unsigned char *)data+4)==len )
{
idx_type = 2; /* Index from start of movi list */
}
}
/* idx_type remains 0 if neither of the two tests above succeeds */
}
if(idx_type == 0 && !AVI->is_opendml && !AVI->total_frames)
{
/* we must search through the file to get the index */
gf_fseek(AVI->fdes, AVI->movi_start, SEEK_SET);
AVI->n_idx = 0;
while(1)
{
if( avi_read(AVI->fdes,data,8) != 8 ) break;
n = str2ulong((unsigned char *)data+4);
/* The movi list may contain sub-lists, ignore them */
if(strnicmp(data,"LIST",4)==0)
{
gf_fseek(AVI->fdes,4,SEEK_CUR);
continue;
}
/* Check if we got a tag ##db, ##dc or ##wb */
if( ( (data[2]=='d' || data[2]=='D') &&
(data[3]=='b' || data[3]=='B' || data[3]=='c' || data[3]=='C') )
|| ( (data[2]=='w' || data[2]=='W') &&
(data[3]=='b' || data[3]=='B') ) )
{
u64 __pos = gf_ftell(AVI->fdes) - 8;
avi_add_index_entry(AVI,(unsigned char *)data,0,__pos,n);
}
gf_fseek(AVI->fdes,PAD_EVEN(n),SEEK_CUR);
}
idx_type = 1;
}
// ************************
// OPENDML
// ************************
// read extended index chunks
if (AVI->is_opendml) {
u64 offset = 0;
hdrl_len = 4+4+2+1+1+4+4+8+4;
char *en, *chunk_start;
int k = 0;
u32 audtr = 0;
u32 nrEntries = 0;
AVI->video_index = NULL;
nvi = 0;
for(audtr=0; audtr<AVI->anum; ++audtr) {
nai[audtr] = 0;
tot[audtr] = 0;
}
// ************************
// VIDEO
// ************************
for (j=0; j<AVI->video_superindex->nEntriesInUse; j++) {
// read from file
u32 chunk_size = (u32) (AVI->video_superindex->aIndex[j].dwSize+hdrl_len);
if (!chunk_size || chunk_size < 28)
continue;
chunk_start = en = (char*) gf_malloc(chunk_size);
if (gf_fseek(AVI->fdes, AVI->video_superindex->aIndex[j].qwOffset, SEEK_SET) == (u64)-1) {
gf_free(chunk_start);
continue;
}
if (avi_read(AVI->fdes, en, (u32) (AVI->video_superindex->aIndex[j].dwSize+hdrl_len) ) <= 0) {
gf_free(chunk_start);
continue;
}
nrEntries = str2ulong((unsigned char*)en + 12);
#ifdef DEBUG_ODML
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] [%d:0] Video nrEntries %ld\n", j, nrEntries));
#endif
offset = str2ullong((unsigned char*)en + 20);
// skip header
en += hdrl_len;
nvi += nrEntries;
if (nvi <= 0 || nvi >= GF_INT_MAX/sizeof(video_index_entry)) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] invalid nvi value %d\n", nvi));
gf_free(chunk_start);
ERR_EXIT(AVI_ERR_READ);
}
AVI->video_index = (video_index_entry *) gf_realloc (AVI->video_index, nvi * sizeof (video_index_entry));
if (!AVI->video_index) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] out of mem (size = %ld)\n", nvi * sizeof (video_index_entry)));
gf_free(chunk_start);
ERR_EXIT(AVI_ERR_NO_MEM);
}
while (k < nvi) {
if (en-chunk_start+8 > chunk_size)
break;
AVI->video_index[k].pos = offset + str2ulong((unsigned char*)en);
en += 4;
AVI->video_index[k].len = str2ulong_len((unsigned char*)en);
AVI->video_index[k].key = str2ulong_key((unsigned char*)en);
en += 4;
// completely empty chunk
if (AVI->video_index[k].pos-offset == 0 && AVI->video_index[k].len == 0) {
k--;
nvi--;
}
#ifdef DEBUG_ODML
/*
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] [%d] POS 0x%llX len=%d key=%s offset (%llx) (%ld)\n", k,
AVI->video_index[k].pos,
(int)AVI->video_index[k].len,
AVI->video_index[k].key?"yes":"no ", offset,
AVI->video_superindex->aIndex[j].dwSize));
*/
#endif
k++;
}
gf_free(chunk_start);
}
AVI->video_frames = nvi;
// this should deal with broken 'rec ' odml files.
if (AVI->video_frames == 0) {
AVI->is_opendml=0;
goto multiple_riff;
}
// ************************
// AUDIO
// ************************
for(audtr=0; audtr<AVI->anum; ++audtr) {
k = 0;
if (!AVI->track[audtr].audio_superindex) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] (%s) cannot read audio index for track %d\n", __FILE__, audtr));
continue;
}
for (j=0; j<AVI->track[audtr].audio_superindex->nEntriesInUse; j++) {
// read from file
u32 chunk_size = (u32) (AVI->track[audtr].audio_superindex->aIndex[j].dwSize+hdrl_len);
if (!chunk_size || chunk_size < 28)
continue;
chunk_start = en = (char*)gf_malloc(chunk_size);
if (gf_fseek(AVI->fdes, AVI->track[audtr].audio_superindex->aIndex[j].qwOffset, SEEK_SET) == (u64)-1) {
gf_free(chunk_start);
continue;
}
if (avi_read(AVI->fdes, en, (u32) (AVI->track[audtr].audio_superindex->aIndex[j].dwSize+hdrl_len)) <= 0) {
gf_free(chunk_start);
continue;
}
nrEntries = str2ulong((unsigned char*)en + 12);
//if (nrEntries > 50) nrEntries = 2; // XXX
#ifdef DEBUG_ODML
//GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] [%d:%d] Audio nrEntries %ld\n", j, audtr, nrEntries));
#endif
offset = str2ullong((unsigned char*)en + 20);
// skip header
en += hdrl_len;
nai[audtr] += nrEntries;
if (nai[audtr] <= 0 || nai[audtr] >= GF_INT_MAX/sizeof(audio_index_entry)) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] invalid nai value %d\n", nai[audtr]));
gf_free(chunk_start);
ERR_EXIT(AVI_ERR_READ);
}
AVI->track[audtr].audio_index = (audio_index_entry *) gf_realloc (AVI->track[audtr].audio_index, nai[audtr] * sizeof (audio_index_entry));
if (!AVI->track[audtr].audio_index) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] out of mem (size = %ld)\n", nai[audtr] * sizeof (audio_index_entry)));
gf_free(chunk_start);
ERR_EXIT(AVI_ERR_NO_MEM);
}
while (k < nai[audtr]) {
if (en - chunk_start + 8 > chunk_size)
break;
AVI->track[audtr].audio_index[k].pos = offset + str2ulong((unsigned char*)en);
en += 4;
AVI->track[audtr].audio_index[k].len = str2ulong_len((unsigned char*)en);
en += 4;
AVI->track[audtr].audio_index[k].tot = tot[audtr];
tot[audtr] += AVI->track[audtr].audio_index[k].len;
#ifdef DEBUG_ODML
/*
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] [%d:%d] POS 0x%llX len=%d offset (%llx) (%ld)\n", k, audtr,
AVI->track[audtr].audio_index[k].pos,
(int)AVI->track[audtr].audio_index[k].len,
offset, AVI->track[audtr].audio_superindex->aIndex[j].dwSize));
*/
#endif
++k;
}
gf_free(chunk_start);
}
AVI->track[audtr].audio_chunks = nai[audtr];
AVI->track[audtr].audio_bytes = tot[audtr];
}
} // is opendml
else if (AVI->total_frames && !AVI->is_opendml && idx_type==0) {
// *********************
// MULTIPLE RIFF CHUNKS (and no index)
// *********************
multiple_riff:
gf_fseek(AVI->fdes, AVI->movi_start, SEEK_SET);
AVI->n_idx = 0;
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] Reconstructing index..."));
// Number of frames; only one audio track supported
nvi = AVI->video_frames = AVI->total_frames;
if (nvi <= 0 || nvi >= GF_INT_MAX/sizeof(video_index_entry)) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] invalid nvi value %d\n", nvi));
ERR_EXIT(AVI_ERR_READ);
}
nai[0] = AVI->track[0].audio_chunks = AVI->total_frames;
for(j=1; j<AVI->anum; ++j) AVI->track[j].audio_chunks = 0;
AVI->video_index = (video_index_entry *) gf_malloc(nvi*sizeof(video_index_entry));
if(AVI->video_index==0) ERR_EXIT(AVI_ERR_NO_MEM);
for(j=0; j<AVI->anum; ++j) {
if(AVI->track[j].audio_chunks) {
if (AVI->track[j].audio_index) gf_free(AVI->track[j].audio_index);
AVI->track[j].audio_index = (audio_index_entry *) gf_malloc((nai[j]+1)*sizeof(audio_index_entry));
memset(AVI->track[j].audio_index, 0, (nai[j]+1)*(sizeof(audio_index_entry)));
if(AVI->track[j].audio_index==0) ERR_EXIT(AVI_ERR_NO_MEM);
}
}
nvi = 0;
for(j=0; j<AVI->anum; ++j) {
nai[j] = 0;
tot[j] = 0;
}
aud_chunks = AVI->total_frames;
while(1)
{
if (nvi >= AVI->total_frames) break;
if( avi_read(AVI->fdes,data,8) != 8 ) break;
n = str2ulong((unsigned char *)data+4);
j=0;
if (aud_chunks - nai[j] -1 <= 0) {
aud_chunks += AVI->total_frames;
AVI->track[j].audio_index = (audio_index_entry *)
gf_realloc( AVI->track[j].audio_index, (aud_chunks+1)*sizeof(audio_index_entry));
if (!AVI->track[j].audio_index) {
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] Internal error in avilib -- no mem\n"));
AVI_errno = AVI_ERR_NO_MEM;
return -1;
}
if (AVI->anum <= j)
AVI->anum = j+1;
}
/* Check if we got a tag ##db, ##dc or ##wb */
// VIDEO
if(
(data[0]=='0' || data[1]=='0') &&
(data[2]=='d' || data[2]=='D') &&
(data[3]=='b' || data[3]=='B' || data[3]=='c' || data[3]=='C') ) {
AVI->video_index[nvi].key = 0x0;
AVI->video_index[nvi].pos = gf_ftell(AVI->fdes);
AVI->video_index[nvi].len = (u32) n;
/*
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] Frame %ld pos %"LLD" len %"LLD" key %ld\n",
nvi, AVI->video_index[nvi].pos, AVI->video_index[nvi].len, (long)AVI->video_index[nvi].key));
*/
nvi++;
gf_fseek(AVI->fdes,PAD_EVEN(n),SEEK_CUR);
}
//AUDIO
else if(
(data[0]=='0' || data[1]=='1') &&
(data[2]=='w' || data[2]=='W') &&
(data[3]=='b' || data[3]=='B') ) {
AVI->track[j].audio_index[nai[j]].pos = gf_ftell(AVI->fdes);
AVI->track[j].audio_index[nai[j]].len = (u32) n;
AVI->track[j].audio_index[nai[j]].tot = tot[j];
tot[j] += AVI->track[j].audio_index[nai[j]].len;
nai[j]++;
gf_fseek(AVI->fdes,PAD_EVEN(n),SEEK_CUR);
}
else {
gf_fseek(AVI->fdes,-4,SEEK_CUR);
}
}
if (nvi < AVI->total_frames) {
GF_LOG(GF_LOG_WARNING, GF_LOG_CONTAINER, ("[avilib] Uh? Some frames seems missing (%ld/%d)\n",
nvi, AVI->total_frames));
}
AVI->video_frames = nvi;
AVI->track[0].audio_chunks = nai[0];
for(j=0; j<AVI->anum; ++j) AVI->track[j].audio_bytes = tot[j];
GF_LOG(GF_LOG_DEBUG, GF_LOG_CONTAINER, ("[avilib] done. nvi=%ld nai=%ld tot=%ld\n", nvi, nai[0], tot[0]));
} // total_frames but no indx chunk (xawtv does this)
else
{
// ******************
// NO OPENDML
// ******************
/* Now generate the video index and audio index arrays */
nvi = 0;
for(j=0; j<AVI->anum; ++j) nai[j] = 0;
for(i=0; i<AVI->n_idx; i++) {
if(strnicmp((char *)AVI->idx[i],AVI->video_tag,3) == 0) nvi++;
for(j=0; j<AVI->anum; ++j) if(strnicmp((char *)AVI->idx[i], AVI->track[j].audio_tag,4) == 0) nai[j]++;
}
AVI->video_frames = nvi;
for(j=0; j<AVI->anum; ++j) AVI->track[j].audio_chunks = nai[j];
if(AVI->video_frames==0) ERR_EXIT(AVI_ERR_NO_VIDS);
AVI->video_index = (video_index_entry *) gf_malloc(nvi*sizeof(video_index_entry));
if(AVI->video_index==0) ERR_EXIT(AVI_ERR_NO_MEM);
for(j=0; j<AVI->anum; ++j) {
if(AVI->track[j].audio_chunks) {
if (AVI->track[j].audio_index) gf_free(AVI->track[j].audio_index);
AVI->track[j].audio_index = (audio_index_entry *) gf_malloc((nai[j]+1)*sizeof(audio_index_entry));
memset(AVI->track[j].audio_index, 0, (nai[j]+1)*(sizeof(audio_index_entry)));
if(AVI->track[j].audio_index==0) ERR_EXIT(AVI_ERR_NO_MEM);
}
}
nvi = 0;
for(j=0; j<AVI->anum; ++j) {
nai[j] = 0;
tot[j] = 0;
}
ioff = idx_type == 1 ? 8 : (u32)AVI->movi_start+4;
for(i=0; i<AVI->n_idx; i++) {
//video
if(strnicmp((char *)AVI->idx[i],AVI->video_tag,3) == 0) {
AVI->video_index[nvi].key = str2ulong(AVI->idx[i]+ 4);
AVI->video_index[nvi].pos = str2ulong(AVI->idx[i]+ 8)+ioff;
AVI->video_index[nvi].len = str2ulong(AVI->idx[i]+12);
nvi++;
}
//audio
for(j=0; j<AVI->anum; ++j) {
if(strnicmp((char *)AVI->idx[i],AVI->track[j].audio_tag,4) == 0) {
AVI->track[j].audio_index[nai[j]].pos = str2ulong(AVI->idx[i]+ 8)+ioff;
AVI->track[j].audio_index[nai[j]].len = str2ulong(AVI->idx[i]+12);
AVI->track[j].audio_index[nai[j]].tot = tot[j];
tot[j] += AVI->track[j].audio_index[nai[j]].len;
nai[j]++;
}
}
}
for(j=0; j<AVI->anum; ++j) AVI->track[j].audio_bytes = tot[j];
} // is no opendml
/* Reposition the file */
gf_fseek(AVI->fdes,AVI->movi_start,SEEK_SET);
AVI->video_pos = 0;
return(0);
}
int AVI_video_frames(avi_t *AVI)
{
return AVI->video_frames;
}
int AVI_video_width(avi_t *AVI)
{
return AVI->width;
}
int AVI_video_height(avi_t *AVI)
{
return AVI->height;
}
double AVI_frame_rate(avi_t *AVI)
{
return AVI->fps;
}
char* AVI_video_compressor(avi_t *AVI)
{
return AVI->compressor2;
}
#if 0
int AVI_max_video_chunk(avi_t *AVI)
{
return AVI->max_len;
}
#endif
int AVI_audio_tracks(avi_t *AVI)
{
return(AVI->anum);
}
int AVI_audio_channels(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_chans;
}
int AVI_audio_mp3rate(avi_t *AVI)
{
return AVI->track[AVI->aptr].mp3rate;
}
#if 0 //unused
int AVI_audio_padrate(avi_t *AVI)
{
return AVI->track[AVI->aptr].padrate;
}
#endif
int AVI_audio_bits(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_bits;
}
int AVI_audio_format(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_fmt;
}
int AVI_audio_rate(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_rate;
}
int AVI_frame_size(avi_t *AVI, int frame)
{
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->video_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
if(frame < 0 || frame >= AVI->video_frames) return 0;
return (u32) (AVI->video_index[frame].len);
}
int AVI_audio_size(avi_t *AVI, int frame)
{
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->track[AVI->aptr].audio_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
if(frame < 0 || frame >= AVI->track[AVI->aptr].audio_chunks) return -1;
return (u32) (AVI->track[AVI->aptr].audio_index[frame].len);
}
u64 AVI_get_video_position(avi_t *AVI, int frame)
{
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return (u64) -1;
}
if(!AVI->video_index) {
AVI_errno = AVI_ERR_NO_IDX;
return (u64) -1;
}
if(frame < 0 || frame >= AVI->video_frames) return 0;
return(AVI->video_index[frame].pos);
}
int AVI_seek_start(avi_t *AVI)
{
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
gf_fseek(AVI->fdes,AVI->movi_start,SEEK_SET);
AVI->video_pos = 0;
return 0;
}
int AVI_set_video_position(avi_t *AVI, int frame)
{
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->video_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
if (frame < 0 ) frame = 0;
AVI->video_pos = frame;
return 0;
}
#if 0 //unused
int AVI_set_audio_bitrate(avi_t *AVI, int bitrate)
{
if(AVI->mode==AVI_MODE_READ) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
AVI->track[AVI->aptr].mp3rate = bitrate;
return 0;
}
#endif
int AVI_read_frame(avi_t *AVI, u8 *vidbuf, int *keyframe)
{
int n;
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->video_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
if(AVI->video_pos < 0 || AVI->video_pos >= AVI->video_frames) return -1;
n = (u32) AVI->video_index[AVI->video_pos].len;
*keyframe = (AVI->video_index[AVI->video_pos].key==0x10) ? 1:0;
if (vidbuf == NULL) {
AVI->video_pos++;
return n;
}
gf_fseek(AVI->fdes, AVI->video_index[AVI->video_pos].pos, SEEK_SET);
AVI->video_pos++;
if (avi_read(AVI->fdes,vidbuf,n) != (u32) n)
{
AVI_errno = AVI_ERR_READ;
return -1;
}
return n;
}
#if 0 //unused
int AVI_get_audio_position_index(avi_t *AVI)
{
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->track[AVI->aptr].audio_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
return (AVI->track[AVI->aptr].audio_posc);
}
int AVI_set_audio_position_index(avi_t *AVI, int indexpos)
{
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->track[AVI->aptr].audio_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
if(indexpos > AVI->track[AVI->aptr].audio_chunks) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
AVI->track[AVI->aptr].audio_posc = indexpos;
AVI->track[AVI->aptr].audio_posb = 0;
return 0;
}
#endif
int AVI_set_audio_position(avi_t *AVI, int byte)
{
int n0, n1;
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->track[AVI->aptr].audio_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
if(byte < 0) byte = 0;
/* Binary search in the audio chunks */
n0 = 0;
n1 = AVI->track[AVI->aptr].audio_chunks;
while(n0<n1-1)
{
int n = (n0+n1)/2;
if(AVI->track[AVI->aptr].audio_index[n].tot>(u32) byte)
n1 = n;
else
n0 = n;
}
AVI->track[AVI->aptr].audio_posc = n0;
AVI->track[AVI->aptr].audio_posb = (u32) (byte - AVI->track[AVI->aptr].audio_index[n0].tot);
return 0;
}
int AVI_read_audio(avi_t *AVI, u8 *audbuf, int bytes, int *continuous)
{
int nr;
u32 todo;
s64 pos;
if(AVI->mode==AVI_MODE_WRITE) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
if(!AVI->track[AVI->aptr].audio_index) {
AVI_errno = AVI_ERR_NO_IDX;
return -1;
}
nr = 0; /* total number of bytes read */
if (bytes==0) {
AVI->track[AVI->aptr].audio_posc++;
AVI->track[AVI->aptr].audio_posb = 0;
}
*continuous = 1;
while(bytes>0)
{
s64 ret;
u32 left = (u32) (AVI->track[AVI->aptr].audio_index[AVI->track[AVI->aptr].audio_posc].len - AVI->track[AVI->aptr].audio_posb);
if(left==0)
{
if(AVI->track[AVI->aptr].audio_posc>=AVI->track[AVI->aptr].audio_chunks-1) return nr;
AVI->track[AVI->aptr].audio_posc++;
AVI->track[AVI->aptr].audio_posb = 0;
*continuous = 0;
continue;
}
if((u32)bytes<left)
todo = bytes;
else
todo = left;
pos = AVI->track[AVI->aptr].audio_index[AVI->track[AVI->aptr].audio_posc].pos + AVI->track[AVI->aptr].audio_posb;
gf_fseek(AVI->fdes, pos, SEEK_SET);
AVI->track[AVI->aptr].audio_posb += (int)todo;
if ( (ret = avi_read(AVI->fdes,audbuf+nr,todo)) != (s64)todo)
{
GF_LOG(GF_LOG_ERROR, GF_LOG_CONTAINER, ("[avilib] avi_read returned before being done :: pos = "LLD", ret = "LLD", todo = %ld\n", pos, ret, todo));
AVI_errno = AVI_ERR_READ;
return -1;
}
bytes -= todo;
nr += todo;
}
return nr;
}
#if 0 //unused
/* AVI_read_data: Special routine for reading the next audio or video chunk
without having an index of the file. */
int AVI_read_data(avi_t *AVI, char *vidbuf, int max_vidbuf,
char *audbuf, int max_audbuf,
int *len)
{
/*
* Return codes:
*
* 1 = video data read
* 2 = audio data read
* 0 = reached EOF
* -1 = video buffer too small
* -2 = audio buffer too small
*/
s64 n;
char data[8];
if(AVI->mode==AVI_MODE_WRITE) return 0;
while(1)
{
/* Read tag and length */
if( avi_read(AVI->fdes,data,8) != 8 ) return 0;
/* if we got a list tag, ignore it */
if(strnicmp(data,"LIST",4) == 0)
{
gf_fseek(AVI->fdes,4,SEEK_CUR);
continue;
}
n = PAD_EVEN(str2ulong((unsigned char *)data+4));
if(strnicmp(data,AVI->video_tag,3) == 0)
{
*len = (u32) n;
AVI->video_pos++;
if(n>max_vidbuf)
{
gf_fseek(AVI->fdes,n,SEEK_CUR);
return -1;
}
if(avi_read(AVI->fdes,vidbuf, (u32) n) != n ) return 0;
return 1;
}
else if(strnicmp(data,AVI->track[AVI->aptr].audio_tag,4) == 0)
{
*len = (u32) n;
if(n>max_audbuf)
{
gf_fseek(AVI->fdes,n,SEEK_CUR);
return -2;
}
if(avi_read(AVI->fdes,audbuf, (u32) n) != n ) return 0;
return 2;
break;
}
else if(gf_fseek(AVI->fdes,n,SEEK_CUR) == (u64) -1) return 0;
}
}
u64 AVI_max_size(void)
{
return((u64) AVI_MAX_LEN);
}
#endif
#endif /*GPAC_DISABLE_AVILIB*/