changed .swf to .flash.
[swftools.git] / src / wav2swf.c
1 /* wav2swf.c
2    Converts WAV/WAVE files to SWF.
3
4    Part of the swftools package.
5
6    Copyright (c) 2001 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 <stdlib.h>
23 #include <stdio.h>
24 #include "../lib/rfxswf.h"
25 #include "../lib/log.h"
26 #include "../lib/args.h"
27 #include "wav.h"
28
29 char * filename = 0;
30 char * outputname = "output.swf";
31 int verbose = 2;
32
33 #define DEFINESOUND_MP3 1 //define sound uses mp3?- undefine for raw sound.
34
35 struct options_t options[] =
36 {
37  {"o","output"},
38  {"v","verbose"},
39  {"d","definesound"},
40  {"l","loop"},
41  {"r","framerate"},
42  {"s","samplerate"},
43  {"b","bitrate"},
44  {"C","cgi"},
45  {"V","version"},
46  {0,0}
47 };
48
49 static int loop = 0;
50 static int definesound = 0;
51 static int framerate = 0;
52 static int samplerate = 11025;
53 static int bitrate = 32;
54 static int do_cgi = 0;
55
56 static int mp3_bitrates[] =
57 { 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160, 0};
58
59 int args_callback_option(char*name,char*val)
60 {
61     if(!strcmp(name, "V")) {
62         printf("wav2swf - part of %s %s\n", PACKAGE, VERSION);
63         exit(0);
64     }
65     else if(!strcmp(name, "o")) {
66         outputname = val;
67         return 1;
68     }
69     else if(!strcmp(name, "d")) {
70         definesound = 1;
71         return 0;
72     }
73     else if(!strcmp(name, "l")) {
74         loop = atoi(val);
75         definesound = 1;
76         return 1;
77     }
78     else if(!strcmp(name, "v")) {
79         verbose ++;
80         return 0;
81     }
82     else if(!strcmp(name, "C")) {
83         do_cgi = 1;
84         return 0;
85     }
86     else if(!strcmp(name, "r")) {
87         float f;
88         sscanf(val, "%f", &f);
89         framerate = f*256;
90         return 1;
91     }
92     else if(!strcmp(name, "s")) {
93         samplerate = atoi(val);
94         if(samplerate > 5000 && samplerate < 6000)
95             samplerate = 5512;
96         else if(samplerate > 11000 && samplerate < 12000)
97             samplerate = 11025;
98         else if(samplerate > 22000 && samplerate < 23000)
99             samplerate = 22050;
100         else if(samplerate > 44000 && samplerate < 45000)
101             samplerate = 44100;
102         else {
103             fprintf(stderr, "Invalid samplerate: %d\n", samplerate);
104             fprintf(stderr, "Allowed values: 11025, 22050\n", samplerate);
105             exit(1);
106         }
107         return 1;
108     }
109     else if(!strcmp(name, "b")) {
110         int t;
111         int b = atoi(val);
112         if(b<=0) {
113             fprintf(stderr, "Not a valid bitrate: %s\n", val);
114             exit(1);
115         }
116         if(b>160) {
117             fprintf(stderr, "Bitrate must be <144. (%s)\n", val);
118             exit(1);
119         }
120         for(t=0;mp3_bitrates[t];t++) {
121             if(b== mp3_bitrates[t]) {
122                 bitrate = b;
123                 return 1;
124             }
125         }
126         fprintf(stderr, "Invalid bitrate. Allowed bitrates are:\n");
127         for(t=0;mp3_bitrates[t];t++) {
128             printf("%d ", mp3_bitrates[t]);
129         }
130         printf("\n");
131         exit(1);
132     }
133     else {
134         printf("Unknown option: -%s\n", name);
135         exit(1);
136     }
137     return 0;
138 }
139 int args_callback_longoption(char*name,char*val)
140 {
141     return args_long2shortoption(options, name, val);
142 }
143 void args_callback_usage(char*name)
144 {
145     printf("Usage: %s [-o filename] file.wav\n", name);
146     printf("\t-v , --verbose\t\t\t Be more verbose\n");
147     printf("\t-d , --definesound\t\t Generate a DefineSound tag instead of streaming sound\n");
148     printf("\t-l , --loop n\t\t\t Loop sound n times (implies -d)\n");
149     printf("\t-r , --framerate fps\t\t Set framerate to fps frames per second\n");
150     printf("\t-s , --samplerate sps\t\t Set samplerate to sps frames per second (default: 11025)\n");
151     printf("\t-b , --bitrate bps\t\t Set mp3 bitrate (default: 32)\n");
152     printf("\t-o , --output filename\t\t set output filename (default: output.swf)\n");
153     printf("\t-C , --cgi\t\t\t For use as CGI- prepend http header, write to stdout\n");
154     printf("\t-V , --version\t\t\t Print program version and exit\n");
155 }
156 int args_callback_command(char*name,char*val)
157 {
158     if(filename) {
159         fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
160                  filename, name);
161     }
162     filename = name;
163     return 0;
164 }
165
166 extern int swf_mp3_bitrate;
167 extern int swf_mp3_out_samplerate;
168 extern int swf_mp3_in_samplerate;
169
170 int main (int argc,char ** argv)
171
172     SWF swf;
173     RGBA rgb;
174     SRECT r;
175     S32 width=300,height = 300;
176     TAG * tag;
177
178     int f,i,ls1,fs1;
179     int count;
180     int t;
181     struct WAV wav,wav2;
182     int blocksize;
183     float blockspersecond;
184     float framespersecond;
185     float samplesperframe;
186     float framesperblock;
187     float samplesperblock;
188     U16* samples;
189     int numsamples;
190
191     processargs(argc, argv);
192
193     blocksize = (samplerate > 22050) ? 1152 : 576;
194
195     blockspersecond = (float)samplerate/blocksize;
196
197     framespersecond = blockspersecond;
198     if(framerate)
199         framespersecond = framerate/256.0;
200
201     framesperblock = framespersecond / blockspersecond;
202     samplesperframe = (blocksize * blockspersecond) / framespersecond;
203     samplesperblock = samplesperframe * framesperblock;
204     
205     initLog(0,-1,0,0,-1,verbose);
206
207     if(!filename) {
208         msg("<fatal> You must supply a filename");
209         exit(1);
210     }
211
212     if(!readWAV(filename, &wav))
213     {
214         msg("<fatal> Error reading %s", filename);
215         exit(1);
216     }
217     convertWAV2mono(&wav,&wav2, samplerate);
218     //printWAVInfo(&wav);
219     //printWAVInfo(&wav2);
220     samples = (U16*)wav2.data;
221     numsamples = wav2.size/2;
222
223     if(numsamples%blocksize != 0)
224     {
225         // apply padding, so that block is a multiple of blocksize
226         int numblocks = (numsamples+blocksize-1)/blocksize;
227         int numsamples2;
228         U16* samples2;
229         numsamples2 = numblocks * blocksize;
230         samples2 = malloc(sizeof(U16)*numsamples2);
231         memcpy(samples2, samples, numsamples*sizeof(U16));
232         memset(&samples2[numsamples], 0, sizeof(U16)*(numsamples2 - numsamples));
233         numsamples = numsamples2;
234         samples = samples2;
235     }
236
237     memset(&swf,0x00,sizeof(SWF));
238
239     swf.fileVersion    = 5;
240     swf.frameRate      = (int)(framespersecond*256);
241
242     swf.movieSize.xmax = 20*width;
243     swf.movieSize.ymax = 20*height;
244
245     swf.firstTag = swf_InsertTag(NULL,ST_SETBACKGROUNDCOLOR);
246     tag = swf.firstTag;
247     rgb.r = 0xff;
248     rgb.g = 0xff;
249     rgb.b = 0xff;
250     swf_SetRGB(tag,&rgb);
251         
252     swf_mp3_bitrate = bitrate;
253     swf_mp3_out_samplerate = samplerate;
254     swf_mp3_in_samplerate = samplerate;
255
256     if(!definesound)
257     {
258         int oldframepos=-1, newframepos=0;
259         float framesamplepos = 0;
260         float framepos = 0;
261         float samplepos = 0;
262         ActionTAG* a = 0;
263         U16 v1=0,v2=0;
264         tag = swf_InsertTag(tag, ST_SOUNDSTREAMHEAD);
265         swf_SetSoundStreamHead(tag, samplesperframe);
266         msg("<notice> %d blocks", numsamples/blocksize);
267         for(t=0;t<numsamples/blocksize;t++) {
268             int s;
269             U16*block1;
270             int seek = blocksize - ((int)samplepos - (int)framesamplepos);
271
272             if(newframepos!=oldframepos) {
273                 tag = swf_InsertTag(tag, ST_SOUNDSTREAMBLOCK);
274                 msg("<notice> Starting block %d %d+%d", t, (int)samplepos, (int)blocksize);
275                 block1 = &samples[t*blocksize];
276                 swf_SetSoundStreamBlock(tag, block1, seek, 1);
277                 v1 = v2 = GET16(tag->data);
278             } else {
279                 msg("<notice> Adding data...", t);
280                 block1 = &samples[t*blocksize];
281                 swf_SetSoundStreamBlock(tag, block1, seek, 0);
282                 v1+=v2;
283                 PUT16(tag->data, v1);
284             }
285             samplepos += blocksize;
286
287             oldframepos = (int)framepos;
288             framepos += framesperblock;
289             newframepos = (int)framepos;
290
291             for(s=oldframepos;s<newframepos;s++) {
292                 tag = swf_InsertTag(tag, ST_SHOWFRAME);
293                 framesamplepos += samplesperframe;
294             }
295         }
296         tag = swf_InsertTag(tag, ST_END);
297     } else {
298         SOUNDINFO info;
299         tag = swf_InsertTag(tag, ST_DEFINESOUND);
300         swf_SetU16(tag, 24); //id
301 #ifdef DEFINESOUND_MP3
302         swf_SetSoundDefine(tag, samples, numsamples);
303 #else
304         swf_SetU8(tag,(/*compression*/0<<4)|(/*rate*/3<<2)|(/*size*/1<<1)|/*mono*/0);
305         swf_SetU32(tag, numsamples); // 44100 -> 11025
306         swf_SetBlock(tag, samples, numsamples*2);
307 #endif
308
309
310         tag = swf_InsertTag(tag, ST_STARTSOUND);
311         swf_SetU16(tag, 24); //id
312         memset(&info, 0, sizeof(info));
313         info.loops = loop;
314         swf_SetSoundInfo(tag, &info);
315         tag = swf_InsertTag(tag, ST_SHOWFRAME);
316         tag = swf_InsertTag(tag, ST_END);
317     }
318
319     if(do_cgi) {
320         if FAILED(swf_WriteCGI(&swf)) fprintf(stderr,"WriteCGI() failed.\n");
321     } else {
322         f = open(outputname,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0644);
323         if FAILED(swf_WriteSWF(f,&swf)) fprintf(stderr,"WriteSWF() failed.\n");
324         close(f);
325     }
326
327     swf_FreeTags(&swf);
328     return 0;
329 }
330
331