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