X-Git-Url: http://git.asbjorn.biz/?p=swftools.git;a=blobdiff_plain;f=avi2swf%2Favi2swf.cc;h=d0fd82379dd2ee7e991826cfce9222ec8f004127;hp=4d3d06da97802ed6c610051cd6b71974b39ef810;hb=133453cc112df7e7d5ce35f3dde5178303154da5;hpb=e2ac6fca142e8f7f81b971bd873d1c7c783f1142 diff --git a/avi2swf/avi2swf.cc b/avi2swf/avi2swf.cc index 4d3d06d..d0fd823 100644 --- a/avi2swf/avi2swf.cc +++ b/avi2swf/avi2swf.cc @@ -3,18 +3,46 @@ Part of the swftools package. - Copyright (c) 2001 Matthias Kramm + 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. - This file is distributed under the GPL, see file COPYING for 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 +#include +#include #include + +#include "../config.h" + +#ifdef HAVE_SIGNAL_H +#ifdef HAVE_PTHREAD_H +#include +#include +#define DO_SIGNALS +#endif +#endif + extern "C" { -#include "../lib/rfxswf.h" #include "../lib/args.h" } +#include "v2swf.h" +#include "../lib/q.h" + +#undef HAVE_CONFIG_H #include -#if (AVIFILE_MAJOR_VERSION == 0) && (AVIFILE_MINOR_VERSION==6) +#if (AVIFILE_MAJOR_VERSION == 0) && (AVIFILE_MINOR_VERSION>=6) #include #include #include @@ -31,40 +59,34 @@ extern "C" { #define Bpp bpp #endif -/* -statistics: (for now) - 37 bytes per shape (rectangle) - 8-12 bytes per placeobject - 4 bytes per removeobject2 - 696+ bytes per definejpeg2 (576 jpegtables) - 576 bytes per jpegtables - 122+ bytes per definejpeg - - blocks*140 = minimal bytes per frames - 5000/140 = maximal blocks with framerate 5000 - - 2 bytes per showframe -*/ - -int cache_size=38; //in frames - -char * filename = 0; -char * outputfilename = "output.swf"; -unsigned int firstframe = 0; -unsigned int lastframe = 0x7fffffff; - -int jpeg_quality = 20; - -#ifndef ST_DEFINEBITSJPEG -#define ST_DEFINEBITSJPEG 6 -#endif +static char * filename = 0; +static char * outputfilename = "output.swf"; +int verbose = 0; + +static int quality = 80; +static double scale = 1.0; +static int flip = 0; +static int expensive = 0; +static int flashversion = 6; +static int keyframe_interval = -1; +static int skip = 0; +static float audio_adjust = 0; +static int mp3_bitrate = 32; +static int samplerate = 11025; struct options_t options[] = { {"v","verbose"}, + {"A","adjust"}, {"o","output"}, - {"n","num"}, - {"s","start"}, + {"p","flip"}, + {"m","mp3-bitrate"}, + {"r","mp3-samplerate"}, + {"q","quality"}, + {"s","scale"}, + {"S","skip"}, + {"x","extragood"}, + {"T","flashversion"}, {"V","version"}, {0,0} }; @@ -72,21 +94,74 @@ struct options_t options[] = int args_callback_option(char*name,char*val) { if(!strcmp(name, "V")) { - printf("avi2swf - part of %s %s\n", PACKAGE, VERSION); + printf("avi2swf-ng - part of %s %s\n", PACKAGE, VERSION); exit(0); } else if(!strcmp(name, "o")) { outputfilename = val; return 1; } - else if(!strcmp(name, "n")) { - lastframe = atoi(val); + else if(!strcmp(name, "q")) { + quality = atoi(val); + if(quality<0) + quality = 0; + if(quality>100) + quality = 100; + return 1; + } + else if(!strcmp(name, "p")) { + flip = 1; + return 0; + } + else if(!strcmp(name, "A")) { + audio_adjust = atof(val); + return 1; + } + else if(!strcmp(name, "v")) { + verbose = 1; + return 0; + } + else if(!strcmp(name, "T")) { + flashversion = atoi(val); + return 1; + } + else if(!strcmp(name, "x")) { + expensive = 1; + return 0; + } + else if(!strcmp(name, "m")) { + mp3_bitrate = atoi(val); + return 0; + } + else if(!strcmp(name, "r")) { + samplerate = atoi(val); + if(samplerate >= 11000 && samplerate <= 12000) + samplerate = 11025; + else if(samplerate >= 22000 && samplerate <= 23000) + samplerate = 22050; + else if(samplerate >= 44000 && samplerate <= 45000) + samplerate = 44100; + else { + fprintf(stderr, "Invalid samplerate: %d\n", samplerate); + fprintf(stderr, "Allowed values: 11025, 22050, 44100\n", samplerate); + exit(1); + } + return 1; + } + else if(!strcmp(name, "S")) { + skip = atoi(val); return 1; } else if(!strcmp(name, "s")) { - firstframe = atoi(val); + scale = atoi(val)/100.0; + if(scale>1.0 || scale<=0) { + fprintf(stderr, "Scale must be in the range 1-100!\n"); + exit(1); + } return 1; } + fprintf(stderr, "Unknown option: -%s\n", name); + exit(1); } int args_callback_longoption(char*name,char*val) { @@ -94,11 +169,20 @@ int args_callback_longoption(char*name,char*val) } void args_callback_usage(char*name) { - printf("\nUsage: %s file.swf\n", name); + printf("\nUsage: %s file.avi\n", name); printf("\t-h , --help\t\t Print help and exit\n"); - printf("\t-o , --output=filename\t Specify output filename\n"); - printf("\t-n , --num=frames\t\t Number of frames to encode\n"); - printf("\t-s , --start=frame\t\t First frame to encode\n"); + printf("\t-o , --output filename\t Specify output filename\n"); + printf("\t-A , --adjust seconds\t Audio adjust: Shift sound -seconds to the future or +seconds into the past.\n"); + printf("\t-n , --num frames\t Number of frames to encode\n"); + printf("\t-m , --mp3-bitrate (kbps)\t Set the mp3 bitrate to encode audio with\n"); + printf("\t-r , --mp3-samplerate (Hz)\t Set the mp3 samplerate to encode audio with (default: 11025)\n"); + printf("\t-d , --scale \t Scale down to factor . (in %, e.g. 100 = original size)\n"); + printf("\t-p , --flip\t\t Turn movie upside down\n"); + printf("\t-q , --quality \t Set the quality to . (0-100, 0=worst, 100=best, default:80)\n"); + printf("\t-x , --extragood\t Enable some *very* expensive compression strategies. You may\n"); + printf("\t \t want to let this run overnight.\n"); + printf("\t-T , --flashversion \t Set output flash version to . Notice: H.263 compression will only be\n"); + printf("\t \t used for n >= 6.\n"); printf("\t-V , --version\t\t Print program version and exit\n"); exit(0); } @@ -112,584 +196,404 @@ int args_callback_command(char*name,char*val) return 0; } -/* id allocation/deallocation */ -char idtab[65536]; -unsigned int idtab_pos=1; -int get_free_id() +static char toabuf[128]; +static char*ftoa(double a) { - while(idtab[idtab_pos] || !idtab_pos) - idtab_pos++; - idtab[idtab_pos]=1; - return idtab_pos; + sprintf(toabuf, "%f", a); + return toabuf; } -void free_id(int id) +static char*itoa(int a) { - idtab[id] = 0; + sprintf(toabuf, "%d", a); + return toabuf; } -void makeshape(int file, int id, int gfxid, int width, int height) +typedef struct _videoreader_avifile_internal { - TAG*tag; - RGBA rgb; - MATRIX m; - SHAPE*s; - SRECT r; - int lines = 0; - int ls,fs; - tag = swf_InsertTag(NULL, ST_DEFINESHAPE); - swf_ShapeNew(&s); - rgb.b = rgb.g = rgb.r = 0xff; - if(lines) - ls = swf_ShapeAddLineStyle(s,20,&rgb); - swf_GetMatrix(NULL,&m); - m.sx = 20*65536; - m.sy = 20*65536; - - fs = swf_ShapeAddBitmapFillStyle(s,&m,gfxid,0); - swf_SetU16(tag,id); // ID - r.xmin = 0; - r.ymin = 0; - r.xmax = width*20; - r.ymax = height*20; - swf_SetRect(tag,&r); - - swf_SetShapeStyles(tag,s); - swf_ShapeCountBits(s,NULL,NULL); - swf_SetShapeBits(tag,s); - - swf_ShapeSetAll(tag,s,0,0,lines?ls:0,fs,0); - - swf_ShapeSetLine(tag,s,width*20,0); - swf_ShapeSetLine(tag,s,0,height*20); - swf_ShapeSetLine(tag,s,-width*20,0); - swf_ShapeSetLine(tag,s,0,-height*20); - swf_ShapeSetEnd(tag); - swf_WriteTag(file, tag); - swf_DeleteTag(tag); - swf_ShapeFree(s); + 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 int shutdown_avi2swf = 0; +static int frameno = 0; + +#ifdef DO_SIGNALS +pthread_t main_thread; +static void sigterm(int sig) +{ + if(pthread_equal (pthread_self(), main_thread)) + { + if(frameno>0 && !shutdown_avi2swf) { + if(verbose) + printf("Thread [%08x] got sigterm %d\n", pthread_self(), sig); + shutdown_avi2swf++; + } else { + exit(1); + } + } } +#endif -void setshape(int file,int id,int depth,int x,int y,CXFORM*cx) +static void readSamples(videoreader_avifile_internal*i, void*buffer, int buffer_size, int numsamples) { - TAG*tag; - MATRIX m; - m.sx = 0x10000; m.sy = 0x10000; - m.r0 = 0; m.r1 = 0; - m.tx = x*20; - m.ty = y*20; - if(cx && !((cx->a0!=256)||(cx->r0!=256)||(cx->g0!=256)||(cx->b0!=256) - ||(cx->a1|cx->r1|cx->g1|cx->b1))) cx = 0; - tag = swf_InsertTag(NULL,ST_PLACEOBJECT2); - swf_ObjectPlace(tag,id,depth,&m,cx,0); - swf_WriteTag(file, tag); - swf_DeleteTag(tag); + 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); } +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; +} +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; -int xblocksize; -int yblocksize; -struct GfxBlock { -// static int xblocksize; -// static int yblocksize; - U8*data; - int len; -}; - -int width=0; -int height=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; + } -int xblocks; -int yblocks; -U8* blockbuffer = 0; - -class GfxBlockCache { + 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; + } +} +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; +} +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); + } +} +void* videoreader_avifile_getinfo(videoreader_t* v, char* name) +{ + return 0; +} +void videoreader_avifile_setparameter(videoreader_t*v, char*name, char*value) +{ + if(verbose) { + printf("videoreader_setparameter(%s, %s)\n", name, value);fflush(stdout); + } +} - GfxBlock*list; - char*expire; //0=block's free - int*ids; - int size; - int pos; - int hits; - int misses; +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 - public: + if(!i->do_video && !i->do_audio) { + printf("File has neither audio nor video streams.(?)\n"); + return 0; + } - GfxBlockCache(int file) - { - list=0; - size = xblocks*yblocks*cache_size; - printf("initializing cache (%d entries)\n", size); - list = new GfxBlock[size]; - expire = new char[size]; - ids = new int[size]; - memset(expire,0,size); - memset(list,0,sizeof(GfxBlock)*size); - memset(ids,0,sizeof(int)*size); - pos = 0; - hits =0; - misses =0; - } - void insert(GfxBlock*block, int gfxid) +#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) { - int oldpos = pos; - while(++pos!=oldpos) - { - if(pos==size) pos=0; - if(!expire[pos]) - break; - } - if(pos==oldpos) { - // cache full- don't insert item - return; - } - if(list[pos].data) { - free(list[pos].data); - list[pos].data = 0; - //TODO: free this in the SWF, also - } - list[pos].data=(U8*)malloc(block->len); - memcpy(list[pos].data,block->data,block->len); - list[pos].len = block->len; - expire[pos] = cache_size; - ids[pos] = gfxid; + StreamInfo*videoinfo; + videoinfo = i->vstream->GetStreamInfo(); + v->width = videoinfo->GetVideoWidth(); + v->height = videoinfo->GetVideoHeight(); + v->fps = (double)(videoinfo->GetFps()); } - int find(GfxBlock*block, CXFORM*cxform) + if(i->do_audio) { - //TODO: do least square regression here to derive cxform - int s; - int bestsum=-1; - int bestid; - float best; - for(s=0;slen); - U8*ptr1 = block->data; - U8*ptr2 = list[s].data; - int sum2 = 0; - // notice: we treat r,g,b as equal here. - do { - int a = (*ptr1++)-(*ptr2++); - sum2 += a*a; - } while(--t); - if(bestsum < 0 || bestsum > sum2) { - bestid = s; - bestsum = sum2; - } - } - if(bestsum<0) { - misses++; - return -1; + 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(); } - best = bestsum/block->len; - - if(best > 96.0) { - misses++; - return -1; - } - expire[bestid]= cache_size; - hits++; - cxform->a0 = 256; - cxform->r0 = 256; - cxform->g0 = 256; - cxform->b0 = 256; - cxform->a1 = 0; - cxform->r1 = 0; - cxform->g1 = 0; - cxform->b1 = 0; - return ids[bestid]; - } - void newframe() - { - int t; - for(t=0;tbasedepth = depth; - this->posx = posx; - this->posy = posy; - this->sizex = sizex; - this->sizey = sizey; - this->depth[0] = this->depth[1] = this->depth[2] = -1; - } - void clear(int file) - { - /* clear everything in the block */ - int t; - for(t=0;t<3;t++) - if(depth[t]>=0) - { - TAG*tag; - tag = swf_InsertTag(NULL, ST_REMOVEOBJECT2); - swf_SetU16(tag, basedepth+t); //depth - swf_WriteTag(file, tag); - swf_DeleteTag(tag); - depth[t] = -1; + 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()); } - } - void writeiframe(int file, GfxBlock*block) - { - clear(file); - - int gfxid = get_free_id(); - int shapeid = get_free_id(); - - //memset(data,0,sizex*sizey*3); - TAG*tag = swf_InsertTag(NULL, ST_DEFINEBITS); - JPEGBITS * jb = swf_SetJPEGBitsStart(tag,sizex,sizey,jpeg_quality); - tag->len = 0; //bad hack - swf_SetU16(tag, gfxid); - int y; - for(y=0;ydata[y*sizex*3]); - swf_SetJPEGBitsFinish(jb); - swf_WriteTag(file, tag); - swf_DeleteTag(tag); - - cache->insert(block, shapeid); - - makeshape(file, shapeid, gfxid, sizex, sizey); - setshape(file, shapeid, basedepth+1, posx, posy, 0); - depth[1] = shapeid; - } - void writereference(int file, int shapeid, CXFORM*form) - { - if(depth[1]!=shapeid) - { - clear(file); - setshape(file, shapeid, basedepth+1, posx, posy, form); - depth[1] = shapeid; + 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; } } - void compress(int file, GfxBlock*block) - { - CXFORM form; - int id = cache->find(block, &form); - if(id<0) - writeiframe(file, block); - else { - writereference(file, id, &form); - } +#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 } -} *blocks = 0; - -void initdisplay(int file) -{ - if(blockbuffer) - free(blockbuffer); - if(blocks) { - int t; - for(t=0;tGetStream(0, AviStream::Audio); - vstream = player->GetStream(0, AviStream::Video); -#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", astream->GetEndPos(), - astream->GetEndTime()); - width = head.dwWidth; - height = head.dwHeight; - dwMicroSecPerFrame = head.dwMicroSecPerFrame; - samplesperframe = astream->GetEndPos()/astream->GetEndTime()*head.dwMicroSecPerFrame/1000000; - samplerate = (int)(astream->GetEndPos()/astream->GetEndTime()); - fps = 1000000.0/dwMicroSecPerFrame; -#else - StreamInfo*audioinfo; - StreamInfo*videoinfo; - audioinfo = astream->GetStreamInfo(); - videoinfo = vstream->GetStreamInfo(); - width = videoinfo->GetVideoWidth(); - height = videoinfo->GetVideoHeight(); - samplerate = audioinfo->GetAudioSamplesPerSec(); - samplesperframe = audioinfo->GetAudioSamplesPerSec()/videoinfo->GetFps(); - fps = (double)(videoinfo->GetFps()); - delete(audioinfo); - delete(videoinfo); + videoreader_t video; + v2swf_t v2swf; + int ret; + FILE*fi; + +#ifdef DO_SIGNALS + signal(SIGTERM, sigterm); + signal(SIGINT , sigterm); + signal(SIGQUIT, sigterm); + main_thread = pthread_self(); #endif - vstream -> StartStreaming(); - astream -> StartStreaming(); - - printf("%f samples/frame\n", samplesperframe); - printf("%d samplerate\n", samplerate); - samplefix = 44100/samplerate; - - if(!samplefix) { - printf("samplerate too high!\n"); - return 0; - } - printf("%d mp3 samples per movie sample\n", samplefix); - - file = open(outputfilename,O_WRONLY|O_CREAT|O_TRUNC, 0644); - - memset(&swf, 0, sizeof(swf)); - swf.frameRate = (int)(fps*256); - swf.fileVersion = 4; - swf.fileSize = 476549;//0x0fffffff; - swf.frameCount = lastframe - firstframe; - r.xmin = 0; - r.ymin = 0; - r.xmax = width*20; - r.ymax = height*20; - swf.movieSize = r; - - swf_WriteHeader(file, &swf); - - tag = swf_InsertTag(NULL, ST_SETBACKGROUNDCOLOR); - swf_SetU8(tag,0); //black - swf_SetU8(tag,0); - swf_SetU8(tag,0); - swf_WriteTag(file, tag); - swf_DeleteTag(tag); - - tag = swf_InsertTag(NULL, ST_SOUNDSTREAMHEAD2); - swf_SetSoundStreamHead(tag, 1152); - swf_WriteTag(file, tag); - swf_DeleteTag(tag); - - int frame = 0; - initdisplay(file); - - int mp3_block_size = 1152; - - int bufsize = mp3_block_size; - if(mp3_block_size < (int)(samplesperframe+1)) - bufsize = (int)(samplesperframe + 1); - unsigned char*buffer = (unsigned char*)malloc(bufsize); - short*block = (short*)malloc(bufsize*2*samplefix); - - unsigned samples_read, bytes_read; - - double movie_sound_pos = 0; - int mp3_sound_pos = 0; - - WAVEFORMATEX wave; - astream->GetAudioFormatInfo(&wave,0); - - printf("nChannels:%d\n", wave.nChannels); - printf("nSamplesPerSec:%d\n", wave.nChannels); - printf("nAvgBytesPerSec:%d\n", wave.nAvgBytesPerSec); - printf("nBlockAlign:%d\n", wave.nBlockAlign); - printf("wBitsPerSample:%d\n", wave.wBitsPerSample); - printf("cbSize:%d\n", wave.cbSize); - - while(1) { - if(vstream->ReadFrame()<0) { - printf("\n"); - break; + processargs(argc, argv); + if(!filename) + exit(0); + if(keyframe_interval<0) { + if(flashversion>=6) + keyframe_interval=200; + else + keyframe_interval=5; + } + + fi = fopen(outputfilename, "wb"); + if(!fi) { + fflush(stdout); fflush(stderr); + fprintf(stderr, "Couldn't open %s\n", outputfilename); + exit(1); } - if(frame < firstframe) - { - if(astream->ReadFrames(buffer, bufsize, - (int)samplesperframe, - samples_read, bytes_read)<0) { - printf("\n"); - break; - }; - printf("\rskipping frame %d",frame); - fflush(stdout); - frame++; - if(frame == firstframe) - printf("\n"); - continue; - } - - printf("\rconvert frame %d",frame); - fflush(stdout); - - // audio - movie_sound_pos += samplesperframe; - - int first=1; - while(mp3_sound_posReadFrames(buffer, bufsize, - mp3_block_size/samplefix, - samples_read, bytes_read)<0) { - printf("couldn't read %d samples\n", mp3_block_size); - break; - }; - int t=0; - int s; - int c=0; - for(s=0;s=movie_sound_pos) { // last run - swf_WriteTag(file, tag); - swf_DeleteTag(tag); - } - first = 0; + if(!ret) { + printf("Error opening %s\n", filename); + exit(1); } - - // video - CImage*img = vstream->GetFrame(); - img->ToRGB(); - U8*data = img->Data(); - int bpp = img->Bpp(); - int x,y; - int xx,yy; - int fs,ls; - SHAPE*s; - MATRIX m; - SRECT r; - RGBA rgb; - - /* some movies have changing dimensions */ - if(img->Width() != width || - img->Height() != height) { + if(verbose) { + printf("| video framerate: %f\n", video.fps); + printf("| video size: %dx%d\n", video.width, video.height); + printf("| audio rate: %d\n", video.rate); + printf("| audio channels: %d\n", video.channels); + } + + ret = v2swf_init(&v2swf, &video); + if(verbose) + v2swf_setparameter(&v2swf, "verbose", "1"); + v2swf_setparameter(&v2swf, "quality", itoa(quality)); + v2swf_setparameter(&v2swf, "blockdiff", "0"); + v2swf_setparameter(&v2swf, "blockdiff_mode", "exact"); + v2swf_setparameter(&v2swf, "mp3_bitrate", itoa(mp3_bitrate)); + v2swf_setparameter(&v2swf, "samplerate", itoa(samplerate)); + //v2swf_setparameter(&v2swf, "fixheader", "1"); + //v2swf_setparameter(&v2swf, "framerate", "15"); + v2swf_setparameter(&v2swf, "scale", ftoa(scale)); + v2swf_setparameter(&v2swf, "prescale", "1"); + v2swf_setparameter(&v2swf, "flash_version", itoa(flashversion)); + v2swf_setparameter(&v2swf, "keyframe_interval", itoa(keyframe_interval)); + if(expensive) + v2swf_setparameter(&v2swf, "motioncompensation", "1"); + + if(!verbose) printf("\n"); - width = img->Width(); - height = img->Height(); - initdisplay(file); + + if(audio_adjust>0) { + int num = ((int)(audio_adjust*video.rate))*video.channels*2; + void*buf = malloc(num); + video.getsamples(&video, buf, num); + free(buf); + } else if(audio_adjust<0) { + int num = (int)(-audio_adjust*video.fps); + void*buf = malloc(video.width*video.height*4); + int t; + for(t=0;tAt(yy*yblocksize+y); - for(x=0;xnewframe(); - - frame++; - if(frame == lastframe) - break; - } - printf("\n"); - destroydisplay(file); - - tag = swf_InsertTag(NULL, ST_END); - swf_WriteTag(file, tag); - swf_DeleteTag(tag); - - close(file); - - return 0; + char buffer[4096]; + while(1) { + int l=v2swf_read(&v2swf, buffer, 4096); + fwrite(buffer, l, 1, fi); + if(!l) + break; + if(!verbose) { + printf("\rConverting frame %d", frameno);fflush(stdout); + } + } + if(!verbose) + printf("\n"); + fclose(fi); + v2swf_backpatch(&v2swf, outputfilename); + v2swf_close(&v2swf); }