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