Uahpet/AC707N
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
bc195654bf0b01d4a17afe272a7d09a66e18cb7a
AC707N/SDK/apps/watch/video/avi/avilib.c
T
huxi bc195654bf 初版
2025-12-03 11:12:34 +08:00

2263 lines
63 KiB
C
Raw Blame History

/*
* avilib.c
*
* Copyright (C) Thomas streich - June 2001
* multiple audio track support Copyright (C) 2002 Thomas streich
*
* 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 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with GNU Make; see the file COPYING. If not, write to
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
*
*/
#include "avilib.h"
#include "app_config.h"
//#include <time.h>
#if TCFG_PSRAM_DEV_ENABLE
#define malloc(size) malloc_psram(size)
#define free(ptr) free_psram(ptr)
#else
#define malloc(size) malloc(size)
#define free(ptr) free(ptr)
#endif
#define INFO_LIST
/* The following variable indicates the kind of error */
long 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 size_t avi_read(void *fd, char *buf, size_t len)
{
size_t n = 0;
size_t r = 0;
while (r < len) {
n = fread(buf + r, 1, len - r, (FILE *)fd);
if ((ssize_t)n <= 0) {
return r;
}
r += n;
}
return r;
}
static size_t avi_write(void *fd, char *buf, size_t len)
{
size_t n = 0;
size_t r = 0;
while (r < len) {
n = fwrite(buf + r, 1, len - r, (FILE *)fd);
if ((ssize_t)n < 0) {
return n;
}
r += n;
}
/* printf("pos:%d fwrite:%lu \n",ftell((FILE *)fd),r); */
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 byte, little endian number.
Should also work on big endian machines */
static void long2str(unsigned char *dst, int n)
{
dst[0] = (n) & 0xff;
dst[1] = (n >> 8) & 0xff;
dst[2] = (n >> 16) & 0xff;
dst[3] = (n >> 24) & 0xff;
}
/* Convert a string of 4 or 2 bytes to a number,
also working on big endian machines */
static unsigned long str2ulong(unsigned char *str)
{
return (str[0] | (str[1] << 8) | (str[2] << 16) | (str[3] << 24));
}
static unsigned long str2ushort(unsigned char *str)
{
return (str[0] | (str[1] << 8));
}
/* 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 */
// TODO
/* 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, int length)
{
unsigned char c[8];
/* 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 */
length = PAD_EVEN(length);
if (avi_write(AVI->fdes, (char *)c, 8) != 8 ||
avi_write(AVI->fdes, (char *)data, length) != length) {
fseek(AVI->fdes, AVI->pos, SEEK_SET);
AVI_errno = AVI_ERR_WRITE;
return -1;
}
/* Update file position */
AVI->pos += 8 + length;
//fprintf(stderr, "pos=%lu %s\n", AVI->pos, tag);
return 0;
}
static int avi_add_index_entry(avi_t *AVI, unsigned char *tag, long flags, unsigned long pos, unsigned long len)
{
void *ptr;
if (AVI->n_idx >= AVI->max_idx) {
size_t old_count = AVI->max_idx; // 原有元素个数
size_t elem_size = 16; // 每个元素大小
size_t new_count = old_count + 4096; // 扩容后的元素个数
size_t new_size = new_count * elem_size; // 扩容后的字节数
// 1. 分配新的内存块
void *new_idx = malloc(new_size);
if (!new_idx) {
// 申请失败，保持原状或做错误处理
AVI_errno = AVI_ERR_NO_MEM;
return -1;
}
memcpy(new_idx, AVI->idx, old_count * elem_size);
free((void *)AVI->idx);
AVI->idx = new_idx;
AVI->max_idx = new_count;
/* ptr = 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 */
// fprintf(stderr, "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);
//TODO
//pos - movi_addr
long2str(AVI->idx[AVI->n_idx] + 8, pos - (HEADERBYTES - 4));
long2str(AVI->idx[AVI->n_idx] + 12, len);
/* Update counter */
AVI->n_idx++;
if (len > AVI->max_len) {
AVI->max_len = len;
}
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
/*
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
*/
avi_t *AVI_open_output_file(char *filename)
{
avi_t *AVI;
int i;
int mask;
unsigned char AVI_header[HEADERBYTES];
/* Allocate the avi_t struct and zero it */
AVI = (avi_t *) malloc(sizeof(avi_t));
if (AVI == 0) {
AVI_errno = AVI_ERR_NO_MEM;
return 0;
}
memset((void *)AVI, 0, sizeof(avi_t));
/* Since Linux needs a long time when deleting big files,
we do not truncate the file when we open it.
Instead it is truncated when the AVI file is closed */
/* mask = umask (0); */
/* umask (mask); */
AVI->fdes = fopen(filename, "w+");
if (AVI->fdes == NULL) {
AVI_errno = AVI_ERR_OPEN;
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) {
fclose(AVI->fdes);
AVI->fdes = NULL;
AVI_errno = AVI_ERR_WRITE;
free(AVI);
return 0;
}
AVI->pos = HEADERBYTES;
AVI->mode = AVI_MODE_WRITE; /* open for writing */
//init
AVI->anum = 0;
AVI->aptr = 0;
return AVI;
}
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);
}
void AVI_set_audio(avi_t *AVI, int channels, long rate, int bits, int format, long 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) {
/* fprintf(stderr, "error - only %d audio tracks supported\n", AVI_MAX_TRACKS); */
return;
}
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);
}
void AVI_set_video_suggestbuffersize(avi_t *AVI, int suggestbuffersize)
{
/* may only be called if file is open for writing */
if (AVI->mode == AVI_MODE_READ) {
return;
}
AVI->SuggestedBufferSize = suggestbuffersize;
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,n); nhb += 4
#define OUTSHRT(n) \
if(nhb<=HEADERBYTES-2) { \
AVI_header[nhb ] = (n )&0xff; \
AVI_header[nhb+1] = (n>>8)&0xff; \
} \
nhb += 2
//ThOe write preliminary AVI file header: 0 frames, max vid/aud size
int avi_update_header(avi_t *AVI)
{
int njunk, sampsize, hasIndex, ms_per_frame, frate, flag;
int movi_len, hdrl_start, strl_start, j;
unsigned char AVI_header[HEADERBYTES];
long nhb;
//assume max size
movi_len = AVI_MAX_LEN - HEADERBYTES + 4;
//assume index will be written
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(AVI->SuggestedBufferSize); /* 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(AVI->SuggestedBufferSize); /* SuggestedBufferSize */
OUTLONG(-1); /* Quality */
OUTLONG(0); /* SampleSize */
OUTLONG(0); /* Frame */
OUTLONG(0); /* Frame */
// OUTLONG(0); /* Frame */
//OUTLONG(0); /* Frame */
/* The video stream format */
OUT4CC("strf");
OUTLONG(40); /* # of bytes to follow */
OUTLONG(40); /* 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 */
/* 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) {
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
// TODO
OUTLONG(1); /* Scale */
OUTLONG(AVI->track[j].a_rate);
OUTLONG(0); /* Start */
OUTLONG(AVI->track[j].audio_bytes); /* Length */
OUTLONG(AVI->SuggestedBufferSize); /* SuggestedBufferSize */
OUTLONG(-1); /* Quality */
// ThOe /4
OUTLONG(sampsize); /* 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
// modify
OUTLONG(AVI->track[j].a_chans * AVI->track[j].a_rate * sampsize);
/* OUTLONG(1000 * AVI->track[j].mp3rate / 8); */
//ThOe (/4)
OUTSHRT(AVI->track[j].a_chans * sampsize); /* 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) {
/* fprintf(stderr,"AVI_close_output_file: # of header bytes too small\n"); */
return -1;
}
OUT4CC("JUNK");
OUTLONG(njunk);
memset(AVI_header + nhb, 0, njunk);
//2001-11-14 added id string
if (njunk > strlen(id_str) + 8) {
//sprintf(id_str, "%s-%s", PACKAGE, VERSION);
//memcpy(AVI_header+nhb, id_str, strlen(id_str));
}
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 (fseek(AVI->fdes, 0, SEEK_SET) < 0 ||
avi_write(AVI->fdes, (char *)AVI_header, HEADERBYTES) != HEADERBYTES ||
fseek(AVI->fdes, AVI->pos, SEEK_SET) < 0) {
AVI_errno = AVI_ERR_CLOSE;
return -1;
}
return 0;
}
/*
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 ret, njunk, sampsize, hasIndex, ms_per_frame, frate, idxerror, flag;
unsigned long movi_len;
int hdrl_start, strl_start, j;
unsigned char AVI_header[HEADERBYTES];
long nhb;
#ifdef INFO_LIST
long info_len;
// time_t calptr;
#endif
/* Calculate length of movi list */
movi_len = AVI->pos - HEADERBYTES + 4;
/* Try to ouput 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;
// fprintf(stderr, "pos=%lu, index_len=%ld \n", AVI->pos, AVI->n_idx*16);
ret = avi_add_chunk(AVI, (unsigned char *)"idx1", (unsigned char *)AVI->idx, AVI->n_idx * 16);
hasIndex = (ret == 0);
//fprintf(stderr, "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");
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(AVI->SuggestedBufferSize); /* 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->SuggestedBufferSize); /* SuggestedBufferSize */
OUTLONG(-1); /* Quality */
OUTLONG(0); /* SampleSize */
OUTLONG(0); /* Frame */
OUTLONG(0); /* Frame */
// OUTLONG(0); /* Frame */
//OUTLONG(0); /* Frame */
/* The video stream format */
OUT4CC("strf");
OUTLONG(40); /* # of bytes to follow */
OUTLONG(40); /* 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 */
/* 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)
{
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(1); /* Scale */
OUTLONG(AVI->track[j].a_rate);
OUTLONG(0); /* Start */
OUTLONG(AVI->track[j].audio_bytes); /* Length */
OUTLONG(AVI->SuggestedBufferSize); /* SuggestedBufferSize */
OUTLONG(-1); /* Quality */
// ThOe /4
OUTLONG(sampsize); /* 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(AVI->track[j].a_chans * AVI->track[j].a_rate * sampsize);
//ThOe (/4)
OUTSHRT(AVI->track[j].a_chans * sampsize); /* 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);
// add INFO list --- (0.6.0pre4)
#ifdef INFO_LIST
OUT4CC("LIST");
//FIXME
info_len = MAX_INFO_STRLEN + 12;
OUTLONG(info_len);
OUT4CC("INFO");
// OUT4CC ("INAM");
// OUTLONG(MAX_INFO_STRLEN);
// sprintf(id_str, "\t");
// memset(AVI_header+nhb, 0, MAX_INFO_STRLEN);
// memcpy(AVI_header+nhb, id_str, strlen(id_str));
// nhb += MAX_INFO_STRLEN;
OUT4CC("ISFT");
OUTLONG(MAX_INFO_STRLEN);
//sprintf(id_str, "%s-%s", PACKAGE, VERSION);
memset(AVI_header + nhb, 0, MAX_INFO_STRLEN);
//memcpy(AVI_header+nhb, id_str, strlen(id_str));
nhb += MAX_INFO_STRLEN;
// 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) {
/* fprintf(stderr,"AVI_close_output_file: # of header bytes too small\n"); */
return -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 (fseek(AVI->fdes, 0, SEEK_SET) < 0 ||
avi_write(AVI->fdes, (char *)AVI_header, HEADERBYTES) != HEADERBYTES ||
ftruncate(AVI->fdes, AVI->pos) < 0) {
AVI_errno = AVI_ERR_CLOSE;
return -1;
}
fseek(AVI->fdes, 0, SEEK_END);
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 long length, int audio, int keyframe)
{
int n;
unsigned char astr[5];
/* 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;
}
/* Add index entry */
//set tag for current audio track
sprintf((char *)astr, "0%1dwb", (int)(AVI->aptr + 1));
if (audio) {
n = avi_add_index_entry(AVI, astr, 0x00, AVI->pos, length);
} else {
//TODO
n = avi_add_index_entry(AVI, (unsigned char *)"00dc", ((keyframe) ? 0x10 : 0x0), AVI->pos, length);
}
if (n) {
return -1;
}
/* Output tag and data */
if (audio) {
n = avi_add_chunk(AVI, astr, (unsigned char *)data, length);
} else {
n = avi_add_chunk(AVI, (unsigned char *)"00dc", (unsigned char *)data, length);
}
if (n) {
return -1;
}
return 0;
}
int AVI_write_frame(avi_t *AVI, char *data, long bytes, int keyframe)
{
unsigned long 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;
}
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 *)"00dc", 0x10, AVI->last_pos, AVI->last_len)) {
return -1;
}
AVI->video_frames++;
AVI->must_use_index = 1;
return 0;
}
int AVI_write_audio(avi_t *AVI, char *data, long 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;
return 0;
}
int AVI_append_audio(avi_t *AVI, char *data, long bytes)
{
long 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
fseek(AVI->fdes, pos + 4, SEEK_SET);
long2str(c, length + bytes);
avi_write(AVI->fdes, (char *)c, 4);
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;
}
long 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));
}
long AVI_bytes_written(avi_t *AVI)
{
if (AVI->mode == AVI_MODE_READ) {
return 0;
}
return (AVI->pos + 8 + 16 * AVI->n_idx);
}
int AVI_set_audio_track(avi_t *AVI, int track)
{
if (track < 0 || 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);
}
/*******************************************************************
* *
* Utilities for reading video and audio from an AVI File *
* *
*******************************************************************/
int AVI_close(avi_t *AVI)
{
u32 rets;
__asm__ volatile("%0 = rets":"=r"(rets));
int ret, i;
/* 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 */
if (AVI->fdes) {
fclose(AVI->fdes);
AVI->fdes = NULL;
}
if (AVI->idx) {
free(AVI->idx);
}
if (AVI->video_index) {
free(AVI->video_index);
}
//FIXME
//if(AVI->audio_index) free(AVI->audio_index);
if (AVI->bitmap_info_header) {
free(AVI->bitmap_info_header);
}
for (i = 0; i < AVI->anum; i++) {
if (AVI->wave_format_ex[i]) {
free(AVI->wave_format_ex[i]);
}
if (AVI->track[i].audio_chunks) {
free(AVI->track[i].audio_index);
}
}
printf("%s rets:0x%x AVI:%X", __func__, rets, (u32)AVI);
free(AVI);
return ret;
}
#if 0
#define ERR_EXIT(x) \
{ \
printf("\n--func=%s,line=%d\n err = %d ", __FUNCTION__,__LINE__,x); \
AVI_close(AVI); \
AVI_errno = x; \
return 0; \
}
#else
#define ERR_EXIT(x) \
{ \
printf("\n--func=%s,line=%d\n err = %d ", __FUNCTION__,__LINE__,x); \
AVI_close(AVI); \
AVI_errno = x; \
return -AVI_errno; \
}
#endif
avi_t *AVI_open_input_file(const char *filename, int getIndex)
{
avi_t *AVI = NULL;
/* Create avi_t structure */
u32 usec1 = jiffies_usec();
AVI = (avi_t *) 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 = fopen(filename, "r");
u32 usec2 = jiffies_usec();
if (AVI->fdes == NULL) {
AVI_errno = AVI_ERR_OPEN;
free(AVI);
return 0;
}
if (avi_parse_input_file(AVI, getIndex)) {
//error
AVI = NULL;
} else {
AVI->aptr = 0; //reset
}
u32 usec3 = jiffies_usec();
printf("%s open:%d parse:%d ", __func__, usec2 - usec1, usec3 - usec2);
printf("%s AVI:0x%x", __func__, (u32)AVI);
return AVI;
}
avi_t *AVI_open_fd(int fd, int getIndex)
{
avi_t *AVI = NULL;
/* Create avi_t structure */
AVI = (avi_t *) 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 = (void *)fd;
avi_parse_input_file(AVI, getIndex);
AVI->aptr = 0; //reset
return AVI;
}
int avi_parse_input_file(avi_t *AVI, int getIndex)
{
u32 usec1 = jiffies_usec();
long i, rate, scale, idx_type;
off_t n;
unsigned char *hdrl_data;
long header_offset = 0, hdrl_len = 0;
long nvi, nai[AVI_MAX_TRACKS], ioff;
long tot[AVI_MAX_TRACKS];
int 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];
/* 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 (strncasecmp(data, "RIFF", 4) != 0 ||
strncasecmp(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 */
}
n = str2ulong((unsigned char *)data + 4);
n = PAD_EVEN(n);
if (strncasecmp(data, "LIST", 4) == 0) {
if (avi_read(AVI->fdes, data, 4) != 4) ERR_EXIT(AVI_ERR_READ)
n -= 4;
if (strncasecmp(data, "hdrl", 4) == 0) {
hdrl_len = n;
hdrl_data = (unsigned char *) malloc(n);
if (hdrl_data == 0) {
ERR_EXIT(AVI_ERR_NO_MEM);
}
// offset of header
fseek(AVI->fdes, 0, SEEK_CUR);
header_offset = ftell(AVI->fdes);
if (avi_read(AVI->fdes, (char *)hdrl_data, n) != n) ERR_EXIT(AVI_ERR_READ)
} else if (strncasecmp(data, "movi", 4) == 0) {
fseek(AVI->fdes, 0, SEEK_CUR);
AVI->movi_start = ftell(AVI->fdes);
fseek(AVI->fdes, n, SEEK_CUR);
} else {
fseek(AVI->fdes, n, SEEK_CUR);
}
} else if (strncasecmp(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 = n / 16;
AVI->idx = (unsigned char((*)[16])) malloc(n);
if (AVI->idx == 0) ERR_EXIT(AVI_ERR_NO_MEM)
if (avi_read(AVI->fdes, (char *) AVI->idx, n) != n) ERR_EXIT(AVI_ERR_READ)
} else {
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;) {
/* List tags are completly ignored */
if (strncasecmp((char *)hdrl_data + i, "LIST", 4) == 0) {
i += 12;
continue;
}
n = str2ulong(hdrl_data + i + 4);
n = PAD_EVEN(n);
/* Interpret the tag and its args */
if (strncasecmp((char *)hdrl_data + i, "strh", 4) == 0) {
i += 8;
if (strncasecmp((char *)hdrl_data + i, "vids", 4) == 0 && !vids_strh_seen) {
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 */
} else if (strncasecmp((char *)hdrl_data + i, "auds", 4) == 0 && ! auds_strh_seen) {
//inc audio tracks
AVI->aptr = AVI->anum;
++AVI->anum;
if (AVI->anum > AVI_MAX_TRACKS) {
/* fprintf(stderr, "error - only %d audio tracks supported\n", AVI_MAX_TRACKS); */
return (-1);
}
AVI->track[AVI->aptr].audio_bytes = str2ulong(hdrl_data + i + 32) * avi_sampsize(AVI, 0);
AVI->track[AVI->aptr].audio_strn = num_stream;
// auds_strh_seen = 1;
lasttag = 2; /* auds */
// ThOe
AVI->track[AVI->aptr].a_codech_off = header_offset + i;
} else {
lasttag = 0;
}
num_stream++;
} else if (strncasecmp((char *)hdrl_data + i, "strf", 4) == 0) {
i += 8;
if (lasttag == 1) {
BITMAPINFOHEADER_avilib bih;
memcpy(&bih, hdrl_data + i, sizeof(BITMAPINFOHEADER_avilib));
AVI->bitmap_info_header = (BITMAPINFOHEADER_avilib *)malloc(bih.bi_size);
if (AVI->bitmap_info_header != NULL) {
memcpy(AVI->bitmap_info_header, hdrl_data + i, bih.bi_size);
}
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;
} else if (lasttag == 2) {
WAVEFORMATEX_avilib *wfe;
char *nwfe;
int wfes;
if ((hdrl_len - i) < sizeof(WAVEFORMATEX_avilib)) {
wfes = hdrl_len - i;
} else {
wfes = sizeof(WAVEFORMATEX_avilib);
}
wfe = (WAVEFORMATEX_avilib *)malloc(sizeof(WAVEFORMATEX_avilib));
if (wfe != NULL) {
memset(wfe, 0, sizeof(WAVEFORMATEX_avilib));
memcpy(wfe, hdrl_data + i, wfes);
if (wfe->cb_size != 0) {
nwfe = malloc(sizeof(WAVEFORMATEX_avilib) + wfe->cb_size);
if (nwfe != 0) {
memcpy(nwfe, wfe, sizeof(WAVEFORMATEX_avilib));
free(wfe);
}
/* (char *)realloc(wfe, sizeof(WAVEFORMATEX_avilib) + */
/* wfe->cb_size); */
if (nwfe != 0) {
fseek(AVI->fdes, 0, SEEK_CUR);
off_t lpos = ftell(AVI->fdes);
fseek(AVI->fdes, header_offset + i + sizeof(WAVEFORMATEX_avilib),
SEEK_SET);
wfe = (WAVEFORMATEX_avilib *)nwfe;
nwfe = &nwfe[sizeof(WAVEFORMATEX_avilib)];
avi_read(AVI->fdes, nwfe, wfe->cb_size);
fseek(AVI->fdes, lpos, SEEK_SET);
}
}
AVI->wave_format_ex[AVI->aptr] = wfe;
}
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;
}
lasttag = 0;
} else {
i += 8;
lasttag = 0;
}
i += n;
}
free(hdrl_data);
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;
}
for (j = 0; j < AVI->anum; ++j) {
AVI->track[j].audio_tag[0] = (j + 1) / 10 + '0';
AVI->track[j].audio_tag[1] = (j + 1) % 10 + '0';
AVI->track[j].audio_tag[2] = 'w';
AVI->track[j].audio_tag[3] = 'b';
}
fseek(AVI->fdes, AVI->movi_start, SEEK_SET);
/* get index if wanted */
if (!getIndex) {
return (0);
}
u32 usec2 = jiffies_usec();
printf("%s %d", __func__, usec2 - usec1);
/* 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) {
off_t 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 (strncasecmp((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);
fseek(AVI->fdes, pos, SEEK_SET);
if (avi_read(AVI->fdes, data, 8) != 8) ERR_EXIT(AVI_ERR_READ)
if (strncasecmp(data, (char *)AVI->idx[i], 4) == 0 && str2ulong((unsigned char *)data + 4) == len) {
idx_type = 1; /* Index from start of file */
} else {
fseek(AVI->fdes, pos + AVI->movi_start - 4, SEEK_SET);
if (avi_read(AVI->fdes, data, 8) != 8) ERR_EXIT(AVI_ERR_READ)
if (strncasecmp(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) {
/* we must search through the file to get the index */
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 (strncasecmp(data, "LIST", 4) == 0) {
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'))) {
fseek(AVI->fdes, 0, SEEK_CUR);
avi_add_index_entry(AVI, (unsigned char *)data, 0, ftell(AVI->fdes) - 8, n);
}
fseek(AVI->fdes, PAD_EVEN(n), SEEK_CUR);
}
idx_type = 1;
}
/* 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 (strncasecmp((char *)AVI->idx[i], AVI->video_tag, 3) == 0) {
nvi++;
}
for (j = 0; j < AVI->anum; ++j) if (strncasecmp((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];
}
// fprintf(stderr, "chunks = %ld %d %s\n", AVI->track[0].audio_chunks, AVI->anum, AVI->track[0].audio_tag);
if (AVI->video_frames == 0) {
ERR_EXIT(AVI_ERR_NO_VIDS);
}
AVI->video_index = (video_index_entry *) 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) {
AVI->track[j].audio_index = (audio_index_entry *) 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] = tot[j] = 0;
}
ioff = idx_type == 1 ? 8 : AVI->movi_start + 4;
for (i = 0; i < AVI->n_idx; i++) {
//video
if (strncasecmp((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 (strncasecmp((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];
}
/* Reposition the file */
fseek(AVI->fdes, AVI->movi_start, SEEK_SET);
AVI->video_pos = 0;
return (0);
}
long 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;
}
long AVI_max_video_chunk(avi_t *AVI)
{
return AVI->max_len;
}
int AVI_audio_tracks(avi_t *AVI)
{
return (AVI->anum);
}
int AVI_audio_channels(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_chans;
}
long AVI_audio_mp3rate(avi_t *AVI)
{
return AVI->track[AVI->aptr].mp3rate;
}
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;
}
long AVI_audio_rate(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_rate;
}
long AVI_audio_bytes(avi_t *AVI)
{
return AVI->track[AVI->aptr].audio_bytes;
}
long AVI_audio_chunks(avi_t *AVI)
{
return AVI->track[AVI->aptr].audio_chunks;
}
long AVI_audio_codech_offset(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_codech_off;
}
long AVI_audio_codecf_offset(avi_t *AVI)
{
return AVI->track[AVI->aptr].a_codecf_off;
}
long AVI_video_codech_offset(avi_t *AVI)
{
return AVI->v_codech_off;
}
long AVI_video_codecf_offset(avi_t *AVI)
{
return AVI->v_codecf_off;
}
long AVI_frame_size(avi_t *AVI, long 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 (AVI->video_index[frame].len);
}
long AVI_audio_size(avi_t *AVI, long 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 0;
}
return (AVI->track[AVI->aptr].audio_index[frame].len);
}
long AVI_get_video_position(avi_t *AVI, long 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 (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;
}
fseek(AVI->fdes, AVI->movi_start, SEEK_SET);
AVI->video_pos = 0;
return 0;
}
int AVI_set_video_position(avi_t *AVI, long 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;
}
int AVI_set_audio_bitrate(avi_t *AVI, long bitrate)
{
if (AVI->mode == AVI_MODE_READ) {
AVI_errno = AVI_ERR_NOT_PERM;
return -1;
}
AVI->track[AVI->aptr].mp3rate = bitrate;
return 0;
}
long AVI_read_frame(avi_t *AVI, char *vidbuf, int *keyframe)
{
long 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 = AVI->video_index[AVI->video_pos].len;
*keyframe = (AVI->video_index[AVI->video_pos].key == 0x10) ? 1 : 0;
fseek(AVI->fdes, AVI->video_index[AVI->video_pos].pos, SEEK_SET);
if (avi_read(AVI->fdes, vidbuf, n) != n) {
AVI_errno = AVI_ERR_READ;
return -1;
}
AVI->video_pos++;
return n;
}
int get_first_video_frame_info(const char *filename, int *offset, int *size)
{
#define AVIIF_KEYFRAME 0x10
FILE *fp = fopen(filename, "r");
if (!fp) {
printf("%s %d", __func__, __LINE__);
return -1;
}
// 检查 RIFF 头
char riff[4];
fread(riff, 1, 4, fp);
if (memcmp(riff, "RIFF", 4) != 0) {
fclose(fp);
printf("%s %d", __func__, __LINE__);
return -1;
}
// 跳过文件大小
fseek(fp, 4, SEEK_CUR);
// 检查 AVI 标识
char avi[4];
fread(avi, 1, 4, fp);
if (memcmp(avi, "AVI ", 4) != 0) {
fclose(fp);
printf("%s %d", __func__, __LINE__);
return -1;
}
// 查找 movi 列表
long movi_offset = -1;
while (1) {
char chunk_id[4];
if (fread(chunk_id, 1, 4, fp) != 4) {
printf("%s %d", __func__, __LINE__);
break;
}
unsigned int chunk_size;
if (fread(&chunk_size, 4, 1, fp) != 4) {
printf("%s %d", __func__, __LINE__);
break;
}
if (memcmp(chunk_id, "LIST", 4) == 0) {
char list_type[4];
fread(list_type, 1, 4, fp);
if (memcmp(list_type, "movi", 4) == 0) {
movi_offset = ftell(fp) ; // 记录 movi 列表起始位置
printf("%s movi_offset:%d ftell:%d ", __func__, (u32)movi_offset, ftell(fp));
printf("%s %d", __func__, __LINE__);
break;
}
fseek(fp, chunk_size - 4, SEEK_CUR);
} else if (memcmp(chunk_id, "idx1", 4) == 0) {
// 如果找到索引，直接使用索引
int entry_count = chunk_size / 16;
for (int i = 0; i < entry_count; i++) {
char id[4];
int flags, data_offset, data_size;
fread(id, 1, 4, fp);
fread(&flags, 4, 1, fp);
fread(&data_offset, 4, 1, fp);
fread(&data_size, 4, 1, fp);
// 检查是否是视频关键帧（通常是第一帧）
if ((flags & AVIIF_KEYFRAME) && (id[0] == '0' && id[1] == '0' && id[2] == 'd' && id[3] == 'c')) {
*offset = data_offset + 8; // 跳过 '00dc' 和大小字段
*size = data_size;
fclose(fp);
return 0;
}
}
} else {
fseek(fp, chunk_size, SEEK_CUR);
}
}
// 如果没有索引，扫描 movi 列表查找第一个视频帧
if (movi_offset != -1) {
fseek(fp, movi_offset, SEEK_SET);
while (1) {
char chunk_id[4];
if (fread(chunk_id, 1, 4, fp) != 4) {
printf("%s %d", __func__, __LINE__);
break;
}
unsigned int chunk_size;
if (fread(&chunk_size, 4, 1, fp) != 4) {
printf("%s %d", __func__, __LINE__);
break;
}
// 检查是否是视频帧（'00dc' 表示第一个视频流）
if (chunk_id[0] == '0' && chunk_id[1] == '0' && chunk_id[2] == 'd' && chunk_id[3] == 'c') {
*offset = ftell(fp);
*size = chunk_size;
fclose(fp);
return 0;
}
// 对齐到 WORD 边界
long skip = (chunk_size + 1) & ~1;
fseek(fp, skip, SEEK_CUR);
}
}
fclose(fp);
printf("%s %d", __func__, __LINE__);
return -1;
}
int AVI_get_video_frame_info(avi_t *AVI, int frame, int *pos, int *len)
{
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 -1;
}
*len = AVI->video_index[frame].len;
*pos = AVI->video_index[frame].pos;
return 0;
}
int AVI_set_audio_position(avi_t *AVI, long byte)
{
long n0, n1, n;
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) {
n = (n0 + n1) / 2;
if (AVI->track[AVI->aptr].audio_index[n].tot > byte) {
n1 = n;
} else {
n0 = n;
}
}
AVI->track[AVI->aptr].audio_posc = n0;
AVI->track[AVI->aptr].audio_posb = byte - AVI->track[AVI->aptr].audio_index[n0].tot;
return 0;
}
long AVI_read_audio(avi_t *AVI, char *audbuf, long bytes)
{
long nr, pos, left, todo;
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 */
while (bytes > 0) {
left = 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;
continue;
}
if (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;
fseek(AVI->fdes, pos, SEEK_SET);
if (avi_read(AVI->fdes, audbuf + nr, todo) != todo) {
AVI_errno = AVI_ERR_READ;
return -1;
}
bytes -= todo;
nr += todo;
AVI->track[AVI->aptr].audio_posb += todo;
}
return nr;
}
long AVI_read_audio_chunk(avi_t *AVI, char *audbuf)
{
long pos, left;
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 (AVI->track[AVI->aptr].audio_index[AVI->track[AVI->aptr].audio_posc].len == 0) {
return 0;
}
left = AVI->track[AVI->aptr].audio_index[AVI->track[AVI->aptr].audio_posc].len - AVI->track[AVI->aptr].audio_posb;
if (audbuf == NULL) {
return left;
}
if (left == 0) {
return 0;
}
pos = AVI->track[AVI->aptr].audio_index[AVI->track[AVI->aptr].audio_posc].pos + AVI->track[AVI->aptr].audio_posb;
fseek(AVI->fdes, pos, SEEK_SET);
if (avi_read(AVI->fdes, audbuf, left) != left) {
AVI_errno = AVI_ERR_READ;
return -1;
}
AVI->track[AVI->aptr].audio_posc++;
AVI->track[AVI->aptr].audio_posb = 0;
return left;
}
/* 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, long max_vidbuf,
char *audbuf, long max_audbuf,
long *len)
{
/*
* Return codes:
*
* 1 = video data read
* 2 = audio data read
* 0 = reached EOF
* -1 = video buffer too small
* -2 = audio buffer too small
*/
off_t 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 (strncasecmp(data, "LIST", 4) == 0) {
fseek(AVI->fdes, 4, SEEK_CUR);
continue;
}
n = PAD_EVEN(str2ulong((unsigned char *)data + 4));
if (strncasecmp(data, AVI->video_tag, 3) == 0) {
*len = n;
AVI->video_pos++;
if (n > max_vidbuf) {
fseek(AVI->fdes, n, SEEK_CUR);
return -1;
}
if (avi_read(AVI->fdes, vidbuf, n) != n) {
return 0;
}
return 1;
} else if (strncasecmp(data, AVI->track[AVI->aptr].audio_tag, 4) == 0) {
*len = n;
if (n > max_audbuf) {
fseek(AVI->fdes, n, SEEK_CUR);
return -2;
}
if (avi_read(AVI->fdes, audbuf, n) != n) {
return 0;
}
return 2;
break;
} else if (fseek(AVI->fdes, n, SEEK_CUR) != 0) {
return 0;
}
}
}
/* AVI_print_error: Print most recent error (similar to perror) */
char *(avi_errors[]) = {
/* 0 */ "avilib - No Error",
/* 1 */ "avilib - AVI file size limit reached",
/* 2 */ "avilib - Error opening AVI file",
/* 3 */ "avilib - Error reading from AVI file",
/* 4 */ "avilib - Error writing to AVI file",
/* 5 */ "avilib - Error writing index (file may still be useable)",
/* 6 */ "avilib - Error closing AVI file",
/* 7 */ "avilib - Operation (read/write) not permitted",
/* 8 */ "avilib - Out of memory (malloc failed)",
/* 9 */ "avilib - Not an AVI file",
/* 10 */ "avilib - AVI file has no header list (corrupted?)",
/* 11 */ "avilib - AVI file has no MOVI list (corrupted?)",
/* 12 */ "avilib - AVI file has no video data",
/* 13 */ "avilib - operation needs an index",
/* 14 */ "avilib - Unkown Error"
};
static int num_avi_errors = sizeof(avi_errors) / sizeof(char *);
static char error_string[4096];
void AVI_print_error(char *str)
{
int aerrno;
aerrno = (AVI_errno >= 0 && AVI_errno < num_avi_errors) ? AVI_errno : num_avi_errors - 1;
/* fprintf(stderr,"%s: %s\n",str,avi_errors[aerrno]); */
/* for the following errors, perror should report a more detailed reason: */
if (AVI_errno == AVI_ERR_OPEN ||
AVI_errno == AVI_ERR_READ ||
AVI_errno == AVI_ERR_WRITE ||
AVI_errno == AVI_ERR_WRITE_INDEX ||
AVI_errno == AVI_ERR_CLOSE) {
printf("REASON \n");
}
}
char *AVI_strerror()
{
int aerrno;
aerrno = (AVI_errno >= 0 && AVI_errno < num_avi_errors) ? AVI_errno : num_avi_errors - 1;
if (AVI_errno == AVI_ERR_OPEN ||
AVI_errno == AVI_ERR_READ ||
AVI_errno == AVI_ERR_WRITE ||
AVI_errno == AVI_ERR_WRITE_INDEX ||
AVI_errno == AVI_ERR_CLOSE) {
sprintf(error_string, "%s ", avi_errors[aerrno]);
return error_string;
} else {
return avi_errors[aerrno];
}
}
uint64_t AVI_max_size()
{
return ((uint64_t) AVI_MAX_LEN);
}
Reference in New Issue View Git Blame Copy Permalink