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