avi2swf/avi2swf.cc

   1 /* avi2swf.cc
   2    Convert avi movie files into swf.
   3
   4    Part of the swftools package.
   5
   6    Copyright (c) 2001,2002,2003 Matthias Kramm <kramm@quiss.org>
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  21
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <fcntl.h>
  26
  27 #include "../config.h"
  28
  29 #include "../lib/args.h"
  30 #include "v2swf.h"
  31 #include "videoreader_avifile.hh"
  32
  33 static char * filename = 0;
  34 static char * outputfilename = "output.swf";
  35 int verbose = 0;
  36
  37 static int quality = 80;
  38 static double scale = 1.0;
  39 static int flip = 0;
  40 static int expensive = 0;
  41 static int flashversion = 6;
  42 static int keyframe_interval = -1;
  43 static int skip = 0;
  44 static float audio_adjust = 0;
  45 static int mp3_bitrate = 32;
  46 static int samplerate = 11025;
  47
  48 static struct options_t options[] = {
  49 {"h", "help"},
  50 {"o", "output"},
  51 {"A", "adjust"},
  52 {"n", "num"},
  53 {"m", "mp3-bitrate"},
  54 {"r", "mp3-samplerate"},
  55 {"d", "scale"},
  56 {"p", "flip"},
  57 {"q", "quality"},
  58 {"x", "extragood"},
  59 {"T", "flashversion"},
  60 {"V", "version"},
  61 {0,0}
  62 };
  63
  64 int args_callback_option(char*name,char*val)
  65 {
  66     if(!strcmp(name, "V")) {
  67         printf("avi2swf-ng - part of %s %s\n", PACKAGE, VERSION);
  68         exit(0);
  69     }
  70     else if(!strcmp(name, "o")) {
  71         outputfilename = val;
  72         return 1;
  73     }
  74     else if(!strcmp(name, "q")) {
  75         quality = atoi(val);
  76         if(quality<0)
  77             quality = 0;
  78         if(quality>100)
  79             quality = 100;
  80         return 1;
  81     }
  82     else if(!strcmp(name, "p")) {
  83         flip = 1;
  84         return 0;
  85     }
  86     else if(!strcmp(name, "A")) {
  87         audio_adjust = atof(val);
  88         return 1;
  89     }
  90     else if(!strcmp(name, "v")) {
  91         verbose = 1;
  92         return 0;
  93     }
  94     else if(!strcmp(name, "T")) {
  95         flashversion = atoi(val);
  96         return 1;
  97     }
  98     else if(!strcmp(name, "x")) {
  99         expensive = 1;
 100         return 0;
 101     }
 102     else if(!strcmp(name, "m")) {
 103         mp3_bitrate = atoi(val);
 104         return 0;
 105     }
 106     else if(!strcmp(name, "r")) {
 107         samplerate = atoi(val);
 108         if(samplerate >= 11000 && samplerate <= 12000)
 109             samplerate = 11025;
 110         else if(samplerate >= 22000 && samplerate <= 23000)
 111             samplerate = 22050;
 112         else if(samplerate >= 44000 && samplerate <= 45000)
 113             samplerate = 44100;
 114         else {
 115             fprintf(stderr, "Invalid samplerate: %d\n", samplerate);
 116             fprintf(stderr, "Allowed values: 11025, 22050, 44100\n", samplerate);
 117             exit(1);
 118         }
 119         return 1;
 120     }
 121     else if(!strcmp(name, "S")) {
 122         skip = atoi(val);
 123         return 1;
 124     }
 125     else if(!strcmp(name, "s")) {
 126         scale = atoi(val)/100.0;
 127         if(scale>1.0 || scale<=0) {
 128             fprintf(stderr, "Scale must be in the range 1-100!\n");
 129             exit(1);
 130         }
 131         return 1;
 132     }
 133     fprintf(stderr, "Unknown option: -%s\n", name);
 134     exit(1);
 135 }
 136 int args_callback_longoption(char*name,char*val)
 137 {
 138     return args_long2shortoption(options, name, val);
 139 }
 140 void args_callback_usage(char *name)
 141 {
 142     printf("\n");
 143     printf("Usage: %s file.avi [-o output.swf]\n", name);
 144     printf("\n");
 145     printf("-h , --help                    Print help and exit\n");
 146     printf("-o , --output filename         Specify output filename\n");
 147     printf("-A , --adjust seconds          Audio adjust: Shift sound -seconds to the future or +seconds into the past.\n");
 148     printf("-n , --num frames              Number of frames to encode\n");
 149     printf("-m , --mp3-bitrate <rate> (kbps)    Set the mp3 bitrate to encode audio with\n");
 150     printf("-r , --mp3-samplerate <rate> (Hz)    Set the mp3 samplerate to encode audio with (default: 11025)\n");
 151     printf("-d , --scale <val>             Scale down to factor <val>. (in %, e.g. 100 = original size)\n");
 152     printf("-p , --flip                    Turn movie upside down\n");
 153     printf("-q , --quality <val>           Set the quality to <val>. (0-100, 0=worst, 100=best, default:80)\n");
 154     printf("-x , --extragood               Enable some *very* expensive compression strategies.\n");
 155     printf("-T , --flashversion <n>        Set output flash version to <n>.\n");
 156     printf("-V , --version                 Print program version and exit\n");
 157     printf("\n");
 158 }
 159 int args_callback_command(char*name,char*val)
 160 {
 161     if(filename) {
 162         fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
 163                  filename, name);
 164     }
 165     filename = name;
 166     return 0;
 167 }
 168
 169 static char toabuf[128];
 170 static char*ftoa(double a)
 171 {
 172     sprintf(toabuf, "%f", a);
 173     return toabuf;
 174 }
 175 static char*itoa(int a)
 176 {
 177     sprintf(toabuf, "%d", a);
 178     return toabuf;
 179 }
 180
 181 #ifdef DO_SIGNALS
 182 pthread_t main_thread;
 183 static void sigterm(int sig)
 184 {
 185     if(pthread_equal (pthread_self(), main_thread))
 186     {
 187         if(frameno>0 && !shutdown_avi2swf) {
 188             if(verbose)
 189                 printf("Thread [%08x] got sigterm %d\n", pthread_self(), sig);
 190             shutdown_avi2swf++;
 191         } else {
 192             exit(1);
 193         }
 194     }
 195 }
 196 #endif
 197
 198 int main (int argc,char ** argv)
 199 {
 200     videoreader_t video;
 201     v2swf_t v2swf;
 202     int ret;
 203     FILE*fi;
 204
 205 #ifdef DO_SIGNALS
 206     signal(SIGTERM, sigterm);
 207     signal(SIGINT , sigterm);
 208     signal(SIGQUIT, sigterm);
 209     main_thread = pthread_self();
 210 #endif
 211
 212     processargs(argc, argv);
 213     if(!filename)
 214         exit(0);
 215     if(keyframe_interval<0) {
 216         if(flashversion>=6)
 217             keyframe_interval=200;
 218         else
 219             keyframe_interval=5;
 220     }
 221
 222     fi = fopen(outputfilename, "wb");
 223     if(!fi) {
 224         fflush(stdout); fflush(stderr);
 225         fprintf(stderr, "Couldn't open %s\n", outputfilename);
 226         exit(1);
 227     }
 228
 229     ret = videoreader_avifile_open(&video, filename);
 230
 231     if(!ret) {
 232         printf("Error opening %s\n", filename);
 233         exit(1);
 234     }
 235
 236     if(verbose) {
 237         printf("| video framerate: %f\n", video.fps);
 238         printf("| video size: %dx%d\n", video.width, video.height);
 239         printf("| audio rate: %d\n", video.rate);
 240         printf("| audio channels: %d\n", video.channels);
 241     }
 242
 243     ret = v2swf_init(&v2swf, &video);
 244     if(verbose)
 245         v2swf_setparameter(&v2swf, "verbose", "1");
 246     v2swf_setparameter(&v2swf, "quality", itoa(quality));
 247     v2swf_setparameter(&v2swf, "blockdiff", "0");
 248     v2swf_setparameter(&v2swf, "blockdiff_mode", "exact");
 249     v2swf_setparameter(&v2swf, "mp3_bitrate", itoa(mp3_bitrate));
 250     v2swf_setparameter(&v2swf, "samplerate", itoa(samplerate));
 251     //v2swf_setparameter(&v2swf, "fixheader", "1");
 252     //v2swf_setparameter(&v2swf, "framerate", "15");
 253     v2swf_setparameter(&v2swf, "scale", ftoa(scale));
 254     v2swf_setparameter(&v2swf, "prescale", "1");
 255     v2swf_setparameter(&v2swf, "flash_version", itoa(flashversion));
 256     v2swf_setparameter(&v2swf, "keyframe_interval", itoa(keyframe_interval));
 257     if(expensive)
 258         v2swf_setparameter(&v2swf, "motioncompensation", "1");
 259     if(flip)
 260         video.setparameter(&video, "flip", "1");
 261     if(verbose)
 262         video.setparameter(&video, "verbose", "1");
 263
 264     if(!verbose)
 265         printf("\n");
 266
 267     if(audio_adjust>0) {
 268         int num = ((int)(audio_adjust*video.rate))*video.channels*2;
 269         void*buf = malloc(num);
 270         video.getsamples(&video, buf, num);
 271         free(buf);
 272     } else if(audio_adjust<0) {
 273         int num = (int)(-audio_adjust*video.fps);
 274         void*buf = malloc(video.width*video.height*4);
 275         int t;
 276         for(t=0;t<num;t++) {
 277             video.getimage(&video, buf);
 278         }
 279         free(buf);
 280     }
 281
 282     if(skip) {
 283         int t;
 284         void*buf = malloc(video.width*video.height*4);
 285         for(t=0;t<skip;t++) {
 286             video.getimage(&video, buf);
 287             video.getsamples(&video, buf, (int)((video.rate/video.fps)*video.channels*2));
 288             if(!verbose) {
 289                 printf("\rSkipping frame %d", video.frame);fflush(stdout);
 290             }
 291         }
 292         free(buf);
 293     }
 294
 295     char buffer[4096];
 296     while(1) {
 297         int l=v2swf_read(&v2swf, buffer, 4096);
 298         fwrite(buffer, l, 1, fi);
 299         if(!l)
 300             break;
 301         if(!verbose) {
 302             printf("\rConverting frame %d", video.frame);fflush(stdout);
 303         }
 304     }
 305     if(!verbose)
 306         printf("\n");
 307     fclose(fi);
 308     v2swf_backpatch(&v2swf, outputfilename);
 309     v2swf_close(&v2swf);
 310 }
 311