From d40beabc5e8751ea4d43b12e1f1b338e719de2b4 Mon Sep 17 00:00:00 2001 From: kramm Date: Wed, 31 Mar 2004 13:20:49 +0000 Subject: [PATCH] moved out of avi2swf.cc --- avi2swf/videoreader_avifile.cc | 312 ++++++++++++++++++++++++++++++++++++++++ avi2swf/videoreader_avifile.hh | 29 ++++ 2 files changed, 341 insertions(+) create mode 100644 avi2swf/videoreader_avifile.cc create mode 100644 avi2swf/videoreader_avifile.hh diff --git a/avi2swf/videoreader_avifile.cc b/avi2swf/videoreader_avifile.cc new file mode 100644 index 0000000..825d12a --- /dev/null +++ b/avi2swf/videoreader_avifile.cc @@ -0,0 +1,312 @@ +/* videoreader_avifile.cc + Read avi files using the avifile library. + + Part of the swftools package. + + Copyright (c) 2001,2002,2003 Matthias Kramm + + This program 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 of the License, or + (at your option) any later version. + + This program 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 this program; if not, write to the Free Software + Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ + +#include "../config.h" + +extern "C" { +#include "../lib/args.h" +} +#include "v2swf.h" +#include "../lib/q.h" + +#undef HAVE_CONFIG_H + +#ifdef HAVE_VERSION_H + #include +#endif +#ifdef HAVE_AVIFILE_VERSION_H + #include +#endif + +#if (AVIFILE_MAJOR_VERSION == 0) && (AVIFILE_MINOR_VERSION>=6) + #include + #include + #include + #include + #include + #define VERSION6 +#else + #include + #include + #include + #define Width width + #define Height height + #define Data data + #define Bpp bpp +#endif + +#include "videoreader.h" + +typedef struct _videoreader_avifile_internal +{ + IAviReadFile* player; + IAviReadStream* astream; + IAviReadStream* vstream; + int do_audio; + int do_video; + int eof; + int frame; + int soundbits; + ringbuffer_t audio_buffer; +} videoreader_avifile_internal; + +static void readSamples(videoreader_avifile_internal*i, void*buffer, int buffer_size, int numsamples) +{ + int ret; + while(i->audio_buffer.available < buffer_size) { + unsigned int samples_read = 0, bytes_read = 0; + ret = i->astream->ReadFrames(buffer, buffer_size, numsamples, samples_read, bytes_read); + if(samples_read<=0) + return; + ringbuffer_put(&i->audio_buffer, buffer, bytes_read); + } + ringbuffer_read(&i->audio_buffer, buffer, buffer_size); +} +static int videoreader_avifile_getsamples(videoreader_t* v, void*buffer, int num) +{ + if(verbose) { + printf("videoreader_getsamples(%d)\n", num);fflush(stdout); + } + videoreader_avifile_internal*i = (videoreader_avifile_internal*)v->internal; + if(i->soundbits == 8) { + readSamples(i, buffer, num/2, num/(v->channels*2)); + unsigned char*b = (unsigned char*)buffer; + int t; + for(t=num-2;t>=0;t-=2) { + unsigned char x = b[t/2]; + b[t] = 0; + b[t+1] = x-128; + } + return num; + } + if(i->soundbits == 16) { + readSamples(i, buffer, num, num/(v->channels*2)); + return num; + } + return 0; +} +static int videoreader_avifile_getimage(videoreader_t* v, void*buffer) +{ + videoreader_avifile_internal*i = (videoreader_avifile_internal*)v->internal; + if(verbose) { + printf("videoreader_getimage()\n");fflush(stdout); + } + + if(shutdown_avi2swf) + i->eof = 1; + + if(i->eof) + return 0; + + if(i->vstream->ReadFrame() < 0) { + if(verbose) printf("vstream->ReadFrame() returned value < 0, shutting down...\n"); + i->eof = 1; + return 0; + } + CImage*img2 = 0; + CImage*img = i->vstream->GetFrame(); + if(!img) { + if(verbose) printf("vstream->GetFrame() returned NULL, shutting down...\n"); + i->eof = 1; + return 0; + } + /* we convert the image to YUV first, because we can convert to RGB from YUV only */ + img->ToYUV(); + img->ToRGB(); + if(img->Bpp() != 3) { + if(verbose) printf("Warning: converthing from bpp %d to bpp 3, this fails on older avifile versions...\n", img->Bpp()); + BitmapInfo tmp(v->width, v->height, 24); + img2 = new CImage(img, &tmp); + img = img2; + } + + + frameno++; + i->frame++; + unsigned char*data = img->Data(); + int bpp = img->Bpp(); + if(bpp == 3) { + int x,y; + for(y=0;yheight;y++) { + unsigned char*from,*to; + to = &((unsigned char*)buffer)[y*v->width*4]; + if(flip) + from = img->At(v->height-y-1); + else + from = img->At(y); + for(x=0;xwidth;x++) { + to[x*4+0] = 0; + to[x*4+1] = from[x*3+2]; + to[x*4+2] = from[x*3+1]; + to[x*4+3] = from[x*3+0]; + } + } + if(img2) delete img2; + return v->width*v->height*4; + } else { + if(img2) delete img2; + if(verbose) printf("Can't handle bpp %d, shutting down...\n", bpp); + return 0; + } +} +static bool videoreader_avifile_eof(videoreader_t* v) +{ + videoreader_avifile_internal*i = (videoreader_avifile_internal*)v->internal; + if(verbose) { + printf("videoreader_eof()\n");fflush(stdout); + } + return i->eof; +} +static void videoreader_avifile_close(videoreader_t* v) +{ + videoreader_avifile_internal*i = (videoreader_avifile_internal*)v->internal; + if(verbose) { + printf("videoreader_close()\n");fflush(stdout); + } + if(i->do_audio) { + ringbuffer_clear(&i->audio_buffer); + } +} +static void videoreader_avifile_setparameter(videoreader_t*v, char*name, char*value) +{ + if(verbose) { + printf("videoreader_setparameter(%s, %s)\n", name, value);fflush(stdout); + } +} + +int videoreader_avifile_open(videoreader_t* v, char* filename) +{ + videoreader_avifile_internal* i; + i = (videoreader_avifile_internal*)malloc(sizeof(videoreader_avifile_internal)); + memset(i, 0, sizeof(videoreader_avifile_internal)); + memset(v, 0, sizeof(videoreader_t)); + v->getsamples = videoreader_avifile_getsamples; + v->getinfo = videoreader_avifile_getinfo; + v->close = videoreader_avifile_close; + v->eof = videoreader_avifile_eof; + v->getimage = videoreader_avifile_getimage; + v->getsamples = videoreader_avifile_getsamples; + v->setparameter = videoreader_avifile_setparameter; + v->internal = i; + + i->do_video = 1; + i->do_audio = 1; + + i->player = CreateIAviReadFile(filename); + if(verbose) { + printf("%d streams (%d video, %d audio)\n", + i->player->StreamCount(), + i->player->VideoStreamCount(), + i->player->AudioStreamCount()); + } + i->astream = i->player->GetStream(0, AviStream::Audio); + i->vstream = i->player->GetStream(0, AviStream::Video); + if(!i->vstream) { + printf("Couldn't open video stream\n"); + i->do_video = 0; + } + if(!i->astream) { + printf("Couldn't open video stream\n"); + i->do_audio = 0; + } +#ifdef NO_MP3 + if(i->do_audio) { + printf(stderr, "MP3 support has been disabled at compile time, not converting soundtrack"); + i->do_audio = 0; + } +#endif + + if(!i->do_video && !i->do_audio) { + printf("File has neither audio nor video streams.(?)\n"); + return 0; + } + +#ifndef VERSION6 + MainAVIHeader head; + int dwMicroSecPerFrame = 0; + player->GetFileHeader(&head); + printf("fps: %d\n", 1000000/head.dwMicroSecPerFrame); + printf("frames: %d\n", head.dwTotalFrames); + printf("streams: %d\n", head.dwStreams); + printf("width: %d\n", head.dwWidth); + printf("height: %d\n", head.dwHeight); + printf("sound: %u samples (%f seconds)\n", i->astream->GetEndPos(), i->astream->GetEndTime()); + v->width = head.dwWidth; + v->height = head.dwHeight; + dwMicroSecPerFrame = head.dwMicroSecPerFrame; + samplesperframe = astream->GetEndPos()/astream->GetEndTime()*head.dwMicroSecPerFrame/1000000; + v->rate = (int)(astream->GetEndPos()/astream->GetEndTime()); + v->fps = 1000000.0/dwMicroSecPerFrame; + i->soundbits = 16; +#else + if(i->do_video) + { + StreamInfo*videoinfo; + videoinfo = i->vstream->GetStreamInfo(); + v->width = videoinfo->GetVideoWidth(); + v->height = videoinfo->GetVideoHeight(); + v->fps = (double)(videoinfo->GetFps()); + } + if(i->do_audio) + { + WAVEFORMATEX wave; + StreamInfo*audioinfo; + + i->astream->GetAudioFormatInfo(&wave,0); + audioinfo = i->astream->GetStreamInfo(); + + v->channels = wave.nChannels; + v->rate = wave.nSamplesPerSec; + i->soundbits = wave.wBitsPerSample; + + if(v->channels==0 || v->rate==0 || i->soundbits==0 || wave.wFormatTag!=1) { + v->rate = audioinfo->GetAudioSamplesPerSec(); + v->channels = audioinfo->GetAudioChannels(); + i->soundbits = audioinfo->GetAudioBitsPerSample(); + } + + if(verbose) { + printf("formatinfo: format %d, %d channels, %d bits/sample, rate %d, blockalign %d\n", wave.wFormatTag, wave.nChannels, wave.wBitsPerSample, wave.nSamplesPerSec, wave.nBlockAlign); + printf("audioinfo: %d channels, %d bits/sample, rate %d\n", audioinfo->GetAudioChannels(), audioinfo->GetAudioBitsPerSample(), audioinfo->GetAudioSamplesPerSec()); + } + if(i->soundbits != 8 && i->soundbits != 16) { + printf("Can't handle %d bit audio, disabling sound\n", wave.wBitsPerSample); + i->do_audio = 0; + i->soundbits = 0; + v->channels = 0; + v->rate = 0; + } + } +#endif + i->vstream -> StartStreaming(); + if(i->do_audio) { + i->astream -> StartStreaming(); + ringbuffer_init(&i->audio_buffer); +#ifdef VERSION6 + WAVEFORMATEX wave; + i->astream -> GetOutputFormat(&wave, sizeof(wave)); + printf("formatinfo: format %d, %d channels, %d bits/sample, rate %d, blockalign %d\n", wave.wFormatTag, wave.nChannels, wave.wBitsPerSample, wave.nSamplesPerSec, wave.nBlockAlign); +#endif + } + + return 1; +} + diff --git a/avi2swf/videoreader_avifile.hh b/avi2swf/videoreader_avifile.hh new file mode 100644 index 0000000..4486536 --- /dev/null +++ b/avi2swf/videoreader_avifile.hh @@ -0,0 +1,29 @@ +/* videoreader_avifile.hh + Read avi files using the avifile library. + + Part of the swftools package. + + Copyright (c) 2001,2002,2003 Matthias Kramm + + This program 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 of the License, or + (at your option) any later version. + + This program 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 this program; if not, write to the Free Software + Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ + +#ifndef __videoreader_avifile_h__ +#define __videoreader_avifile_h__ + +#include "videoreader.h" + +int videoreader_avifile_open(videoreader_t* v, char* filename); + +#endif -- 1.7.10.4