fixed sizeof detection
[swftools.git] / 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 #ifdef WIN32
32 #include "videoreader_vfw.hh"
33 #else
34 #include "videoreader_avifile.hh"
35 #endif
36
37 static char * filename = 0;
38 static char * outputfilename = "output.swf";
39 int verbose = 0;
40
41 static int quality = 80;
42 static double scale = 1.0;
43 static int flip = 0;
44 static int expensive = 0;
45 static int flashversion = 6;
46 static int keyframe_interval = -1;
47 static int skip = 0;
48 static float audio_adjust = 0;
49 static int mp3_bitrate = 32;
50 static int samplerate = 11025;
51 static int numframes = 0;
52 static char* skipframes = 0;
53
54 static struct options_t options[] = {
55 {"h", "help"},
56 {"o", "output"},
57 {"A", "adjust"},
58 {"n", "num"},
59 {"m", "mp3-bitrate"},
60 {"r", "mp3-samplerate"},
61 {"s", "scale"},
62 {"S", "skipframes"},
63 {"p", "flip"},
64 {"q", "quality"},
65 {"k", "keyframe"},
66 {"x", "extragood"},
67 {"T", "flashversion"},
68 {"V", "version"},
69 {0,0}
70 };
71
72 int args_callback_option(char*name,char*val)
73 {
74     if(!strcmp(name, "V")) {
75         printf("avi2swf-ng - part of %s %s\n", PACKAGE, VERSION);
76         exit(0);
77     } 
78     else if(!strcmp(name, "o")) {
79         outputfilename = val;
80         return 1;
81     }
82     else if(!strcmp(name, "n")) {
83         numframes = atoi(val);
84         return 1;
85     }
86     else if(!strcmp(name, "d")) {
87         scale = atof(val);
88         return 1;
89     }
90     else if(!strcmp(name, "q")) {
91         quality = atoi(val);
92         if(quality<0)
93             quality = 0;
94         if(quality>100)
95             quality = 100;
96         return 1;
97     }
98     else if(!strcmp(name, "p")) {
99         flip = 1;
100         return 0;
101     }
102     else if(!strcmp(name, "k")) {
103         keyframe_interval = atoi(val);
104         return 1;
105     }
106     else if(!strcmp(name, "A")) {
107         audio_adjust = atof(val);
108         return 1;
109     }
110     else if(!strcmp(name, "v")) {
111         verbose = 1;
112         return 0;
113     }
114     else if(!strcmp(name, "T")) {
115         flashversion = atoi(val);
116         return 1;
117     }
118     else if(!strcmp(name, "x")) {
119         expensive = 1;
120         return 0;
121     }
122     else if(!strcmp(name, "m")) {
123         mp3_bitrate = atoi(val);
124         return 1;
125     }
126     else if(!strcmp(name, "r")) {
127         samplerate = atoi(val);
128         if(samplerate >= 11000 && samplerate <= 12000)
129             samplerate = 11025;
130         else if(samplerate >= 22000 && samplerate <= 23000)
131             samplerate = 22050;
132         else if(samplerate >= 44000 && samplerate <= 45000)
133             samplerate = 44100;
134         else {
135             fprintf(stderr, "Invalid samplerate: %d\n", samplerate);
136             fprintf(stderr, "Allowed values: 11025, 22050, 44100\n", samplerate);
137             exit(1);
138         }
139         return 1;
140     }
141     else if(!strcmp(name, "S")) {
142         skip = atoi(val);
143         return 1;
144     }
145     else if(!strcmp(name, "C")) {
146         skipframes = strdup(val);
147         return 1;
148     }
149     else if(!strcmp(name, "s")) {
150         scale = atoi(val)/100.0;
151         if(scale>1.0 || scale<=0) {
152             fprintf(stderr, "Scale must be in the range 1-100!\n");
153             exit(1);
154         }
155         return 1;
156     }
157     fprintf(stderr, "Unknown option: -%s\n", name);
158     exit(1);
159 }
160 int args_callback_longoption(char*name,char*val)
161 {
162     return args_long2shortoption(options, name, val);
163 }
164 void args_callback_usage(char *name)
165 {
166     printf("\n");
167     printf("Usage: %s file.avi [-o output.swf]\n", name);
168     printf("\n");
169     printf("-h , --help                    Print help and exit\n");
170     printf("-o , --output filename         Specify output filename\n");
171     printf("-A , --adjust seconds          Audio adjust: Shift sound -seconds to the future or +seconds into the past.\n");
172     printf("-n , --num frames              Number of frames to encode\n");
173     printf("-m , --mp3-bitrate <kbps>      Set the mp3 bitrate to encode audio with\n");
174     printf("-r , --mp3-samplerate <hz>     Set the mp3 samplerate to encode audio with (default: 11025)\n");
175     printf("-s , --scale <val>             Scale down to factor <val>. (in %, e.g. 100 = original size)\n");
176     printf("-S , --skipframes <num>        Skip <num> frames before starting the conversion.\n");
177     printf("-p , --flip                    Turn movie upside down\n");
178     printf("-q , --quality <val>           Set the quality to <val>. (0-100, 0=worst, 100=best, default:80)\n");
179     printf("-k , --keyframe                Set the number of intermediate frames between keyframes.\n");
180     printf("-x , --extragood               Enable some *very* expensive compression strategies.\n");
181     printf("-T , --flashversion <n>        Set output flash version to <n>.\n");
182     printf("-V , --version                 Print program version and exit\n");
183     printf("\n");
184 }
185 int args_callback_command(char*name,char*val)
186 {
187     if(filename) {
188         fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
189                  filename, name);
190     }
191     filename = name;
192     return 0;
193 }
194
195 static char toabuf[128];
196 static char*ftoa(double a)
197 {
198     sprintf(toabuf, "%f", a);
199     return toabuf;
200 }
201 static char*itoa(int a)
202 {
203     sprintf(toabuf, "%d", a);
204     return toabuf;
205 }
206
207 #ifdef DO_SIGNALS
208 pthread_t main_thread;
209 static void sigterm(int sig)
210 {
211     if(pthread_equal (pthread_self(), main_thread))
212     {
213         if(frameno>0 && !shutdown_avi2swf) {
214             if(verbose)
215                 printf("Thread [%08x] got sigterm %d\n", pthread_self(), sig);
216             shutdown_avi2swf++;
217         } else {
218             exit(1);
219         }
220     }
221 }
222 #endif
223
224 int main (int argc,char ** argv)
225
226     videoreader_t video;
227     v2swf_t v2swf;
228     int ret;
229     FILE*fi;
230
231 #ifdef DO_SIGNALS
232     signal(SIGTERM, sigterm);
233     signal(SIGINT , sigterm);
234     signal(SIGQUIT, sigterm);
235     main_thread = pthread_self();
236 #endif
237
238     processargs(argc, argv);
239     if(!filename) {
240         fprintf(stderr, "You must supply a filename");
241         exit(0);
242     }
243     if(keyframe_interval<0) {
244         if(flashversion>=6)
245             keyframe_interval=20;
246         else
247             keyframe_interval=5;
248     }
249     
250     fi = fopen(outputfilename, "wb");
251     if(!fi) {
252         fflush(stdout); fflush(stderr);
253         fprintf(stderr, "Couldn't open %s\n", outputfilename);
254         exit(1);
255     }
256     
257 #ifdef WIN32
258     ret = videoreader_vfw_open(&video, filename);
259 #else
260     ret = videoreader_avifile_open(&video, filename);
261 #endif
262
263     if(ret<0) {
264         fprintf(stderr, "Error opening %s\n", filename);
265         exit(1);
266     }
267
268     if(verbose) {
269         printf("| video framerate: %f\n", video.fps);
270         printf("| video size: %dx%d\n", video.width, video.height);
271         printf("| audio rate: %d\n", video.samplerate);
272         printf("| audio channels: %d\n", video.channels);
273     }
274
275     ret = v2swf_init(&v2swf, &video);
276     if(verbose)
277         v2swf_setparameter(&v2swf, "verbose", "1");
278     if(numframes)
279         v2swf_setparameter(&v2swf, "numframes", itoa(numframes));
280     v2swf_setparameter(&v2swf, "quality", itoa(quality));
281     v2swf_setparameter(&v2swf, "blockdiff", "0");
282     v2swf_setparameter(&v2swf, "blockdiff_mode", "exact");
283     v2swf_setparameter(&v2swf, "mp3_bitrate", itoa(mp3_bitrate));
284     v2swf_setparameter(&v2swf, "samplerate", itoa(samplerate));
285     //v2swf_setparameter(&v2swf, "fixheader", "1");
286     //v2swf_setparameter(&v2swf, "framerate", "15");
287     v2swf_setparameter(&v2swf, "scale", ftoa(scale));
288     v2swf_setparameter(&v2swf, "prescale", "1");
289     v2swf_setparameter(&v2swf, "flash_version", itoa(flashversion));
290     v2swf_setparameter(&v2swf, "keyframe_interval", itoa(keyframe_interval));
291     if(skipframes)
292         v2swf_setparameter(&v2swf, "skipframes", skipframes);
293     if(expensive)
294         v2swf_setparameter(&v2swf, "motioncompensation", "1");
295     if(flip)
296         video.setparameter(&video, "flip", "1");
297     if(verbose)
298         video.setparameter(&video, "verbose", "1");
299
300     if(!verbose)
301         printf("\n");
302
303     if(audio_adjust>0) {
304         int num = ((int)(audio_adjust*video.samplerate))*video.channels*2;
305         void*buf = malloc(num);
306         video.getsamples(&video, buf, num);
307         free(buf);
308     } else if(audio_adjust<0) {
309         int num = (int)(-audio_adjust*video.fps);
310         void*buf = malloc(video.width*video.height*4);
311         int t;
312         for(t=0;t<num;t++) {
313             video.getimage(&video, buf);
314         }
315         free(buf);
316     }
317
318     if(skip) {
319         int t;
320         void*buf = malloc(video.width*video.height*4);
321         for(t=0;t<skip;t++) {
322             video.getimage(&video, buf);
323             video.getsamples(&video, buf, (int)((video.samplerate/video.fps)*video.channels*2));
324             if(!verbose) {
325                 printf("\rSkipping frame %d", video.frame);fflush(stdout);
326             }
327         }
328         free(buf);
329     }
330
331     char buffer[4096];
332     while(1) {
333         int l=v2swf_read(&v2swf, buffer, 4096);
334         fwrite(buffer, l, 1, fi);
335         if(!l)
336             break;
337         if(!verbose) {
338             printf("\rConverting frame %d", video.frame);fflush(stdout);
339         }
340     }
341     if(!verbose)
342         printf("\n");
343     fclose(fi);
344     v2swf_backpatch(&v2swf, outputfilename);
345     v2swf_close(&v2swf);
346 }
347