support for raw sound.
[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 file is distributed under the GPL, see file COPYING for details */
9
10 #include <stdlib.h>
11 #include <stdio.h>
12 #include "../lib/rfxswf.h"
13 #include "../lib/log.h"
14 #include "../lib/args.h"
15 #include "wav.h"
16
17 char * filename = 0;
18 char * outputname = "output.swf";
19 int verbose = 2;
20
21 #define DEFINESOUND_MP3 1 //define sound uses mp3?- undefine for raw sound.
22
23 struct options_t options[] =
24 {
25  {"o","output"},
26  {"v","verbose"},
27  {"d","definesound"},
28  {"l","loop"},
29  {"r","framerate"},
30  {"b","bitrate"},
31  {"V","version"},
32  {0,0}
33 };
34
35 static int loop = 0;
36 static int definesound = 0;
37 static int framerate = 0;
38 static int bitrate = 32;
39
40 int args_callback_option(char*name,char*val)
41 {
42     if(!strcmp(name, "V")) {
43         printf("wav2swf - part of %s %s\n", PACKAGE, VERSION);
44         exit(0);
45     }
46     else if(!strcmp(name, "o")) {
47         outputname = val;
48         return 1;
49     }
50     else if(!strcmp(name, "d")) {
51         definesound = 1;
52         return 0;
53     }
54     else if(!strcmp(name, "l")) {
55         loop = atoi(val);
56         definesound = 1;
57         return 1;
58     }
59     else if(!strcmp(name, "v")) {
60         verbose ++;
61         return 0;
62     }
63     else if(!strcmp(name, "r")) {
64         float f;
65         sscanf(val, "%f", &f);
66         framerate = f*256;
67         return 1;
68     }
69     else if(!strcmp(name, "b")) {
70         bitrate = atoi(val);
71         if(bitrate<=0) {
72             fprintf(stderr, "Not a valid bitrate: %s\n", val);
73             exit(1);
74         }
75         if(bitrate>144) {
76             fprintf(stderr, "Bitrate must be <144. (%s)\n", val);
77             exit(1);
78         }
79         return 1;
80     }
81     else {
82         printf("Unknown option: -%s\n", name);
83         exit(1);
84     }
85     return 0;
86 }
87 int args_callback_longoption(char*name,char*val)
88 {
89     return args_long2shortoption(options, name, val);
90 }
91 void args_callback_usage(char*name)
92 {
93     printf("Usage: %s [-o filename] file.wav\n", name);
94     printf("\t-v , --verbose\t\t\t Be more verbose\n");
95     printf("\t-d , --definesound\t\t Generate a DefineSound tag instead of streaming sound\n");
96     printf("\t-l , --loop n\t\t\t Loop sound n times (implies -d)\n");
97     printf("\t-r , --framerate fps\t\t Set framerate to fps frames per seond\n");
98     printf("\t-b , --bitrate bps\t\t Set mp3 bitrate (default: 32)\n");
99     printf("\t-o , --output filename\t\t set output filename (default: output.swf)\n");
100     printf("\t-V , --version\t\t\t Print program version and exit\n");
101 }
102 int args_callback_command(char*name,char*val)
103 {
104     if(filename) {
105         fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
106                  filename, name);
107     }
108     filename = name;
109     return 0;
110 }
111
112 extern int swf_mp3_bitrate;
113
114 int main (int argc,char ** argv)
115
116     SWF swf;
117     RGBA rgb;
118     SRECT r;
119     S32 width=300,height = 300;
120     TAG * tag;
121
122     int f,i,ls1,fs1;
123     int count;
124     int t;
125     struct WAV wav,wav2;
126     int blocksize;
127     float blockspersecond;
128     float framespersecond;
129     float samplesperframe;
130     float framesperblock;
131     float samplesperblock;
132     U16* samples;
133     int numsamples;
134
135     processargs(argc, argv);
136
137     blocksize = 576;
138     blockspersecond = 11025.0/blocksize;
139
140     framespersecond = blockspersecond;
141     if(framerate)
142         framespersecond = framerate/256.0;
143
144     framesperblock = framespersecond / blockspersecond;
145     samplesperframe = (blocksize * blockspersecond) / framespersecond;
146     samplesperblock = samplesperframe * framesperblock;
147     
148     initLog(0,-1,0,0,-1,verbose);
149
150     if(!readWAV(filename, &wav))
151     {
152         logf("<fatal> Error reading %s", filename);
153         exit(1);
154     }
155     convertWAV2mono(&wav,&wav2, 44100);
156     //printWAVInfo(&wav);
157     //printWAVInfo(&wav2);
158     samples = (U16*)wav2.data;
159     numsamples = wav2.size/2;
160
161     memset(&swf,0x00,sizeof(SWF));
162
163     swf.fileVersion    = 5;
164     swf.frameRate      = (int)(framespersecond*256);
165
166     swf.movieSize.xmax = 20*width;
167     swf.movieSize.ymax = 20*height;
168
169     swf.firstTag = swf_InsertTag(NULL,ST_SETBACKGROUNDCOLOR);
170     tag = swf.firstTag;
171     rgb.r = 0xff;
172     rgb.g = 0xff;
173     rgb.b = 0xff;
174     swf_SetRGB(tag,&rgb);
175         
176     swf_mp3_bitrate = bitrate;
177
178     if(!definesound)
179     {
180         int oldframepos=-1, newframepos=0;
181         float framesamplepos = 0;
182         float framepos = 0;
183         float samplepos = 0;
184         U16 v1=0,v2=0;
185         tag = swf_InsertTag(tag, ST_SOUNDSTREAMHEAD);
186         swf_SetSoundStreamHead(tag, samplesperframe);
187
188         logf("<notice> %d blocks", numsamples/(blocksize*4));
189         for(t=0;t<numsamples/(blocksize*4);t++) {
190             int s;
191             U16*block1;
192             int seek = blocksize - ((int)samplepos - (int)framesamplepos);
193
194             if(newframepos!=oldframepos) {
195                 tag = swf_InsertTag(tag, ST_SOUNDSTREAMBLOCK);
196                 logf("<notice> Starting block %d %d+%d", t, (int)samplepos, (int)blocksize);
197                 block1 = &samples[t*blocksize*4];
198                 swf_SetSoundStreamBlock(tag, block1, seek, 1);
199                 v1 = v2 = GET16(tag->data);
200             } else {
201                 logf("<notice> Adding data...", t);
202                 block1 = &samples[t*blocksize*4];
203                 swf_SetSoundStreamBlock(tag, block1, seek, 0);
204                 v1+=v2;
205                 PUT16(tag->data, v1);
206             }
207             samplepos += blocksize;
208
209             oldframepos = (int)framepos;
210             framepos += framesperblock;
211             newframepos = (int)framepos;
212
213             for(s=oldframepos;s<newframepos;s++) {
214                 tag = swf_InsertTag(tag, ST_SHOWFRAME);
215                 framesamplepos += samplesperframe;
216             }
217         }
218         tag = swf_InsertTag(tag, ST_END);
219     } else {
220         SOUNDINFO info;
221         tag = swf_InsertTag(tag, ST_DEFINESOUND);
222         swf_SetU16(tag, 24); //id
223 #ifdef DEFINESOUND_MP3
224         swf_SetSoundDefine(tag, samples, numsamples);
225 #else
226         swf_SetU8(tag,(/*compression*/0<<4)|(/*rate*/3<<2)|(/*size*/1<<1)|/*mono*/0);
227         swf_SetU32(tag, numsamples); // 44100 -> 11025
228         swf_SetBlock(tag, wav2.data, numsamples*2);
229 #endif
230
231         tag = swf_InsertTag(tag, ST_STARTSOUND);
232         swf_SetU16(tag, 24); //id
233         memset(&info, 0, sizeof(info));
234         info.loops = loop;
235         swf_SetSoundInfo(tag, &info);
236         tag = swf_InsertTag(tag, ST_SHOWFRAME);
237         tag = swf_InsertTag(tag, ST_END);
238     }
239
240     f = open(outputname,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0644);
241     if FAILED(swf_WriteSWF(f,&swf)) fprintf(stderr,"WriteSWF() failed.\n");
242     close(f);
243
244     swf_FreeTags(&swf);
245     return 0;
246 }
247
248