added LAME mp3 routines.
[swftools.git] / lib / modules / swfsound.c
1 /* swfaction.c
2
3    SWF Sound handling routines
4    
5    Extension module for the rfxswf library.
6    Part of the swftools package.
7
8    Copyright (c) 2001, 2002 Matthias Kramm <kramm@quiss.org>
9  
10    This file is distributed under the GPL, see file COPYING for details 
11
12 */
13
14 #ifndef RFXSWF_DISABLESOUND
15
16 #include "../rfxswf.h"
17
18 #ifdef BLADEENC
19 fjokjklj
20 CodecInitOut * init = 0;
21 void swf_SetSoundStreamHead(TAG*tag, U16 avgnumsamples)
22 {
23     U8 playbackrate = 3; // 0 = 5.5 Khz, 1 = 11 Khz, 2 = 22 Khz, 3 = 44 Khz
24     U8 playbacksize = 1; // 0 = 8 bit, 1 = 16 bit
25     U8 playbacktype = 0; // 0 = mono, 1 = stereo
26     U8 compression = 2; // 0 = raw, 1 = ADPCM, 2 = mp3
27     U8 rate = 3; // 0 = 5.5 Khz, 1 = 11 Khz, 2 = 22 Khz, 3 = 44 Khz
28     U8 size = 1; // 0 = 8 bit, 1 = 16 bit
29     U8 type = 0; // 0 = mono, 1 = stereo
30
31     CodecInitIn params;
32     memset(&params, 0, sizeof(params));
33     params.frequency = 44100;  //48000, 44100 or 32000
34     params.mode = 3;      //0 = Stereo, 2 = Dual Channel, 3 = Mono
35     params.emphasis = 0;  //0 = None, 1 = 50/15 microsec, 3 = CCITT J.17
36     params.bitrate = 128;         //default is 128 (64 for mono)
37     init = codecInit(&params);
38
39     swf_SetU8(tag,(playbackrate<<2)|(playbacksize<<1)|playbacktype);
40     swf_SetU8(tag,(compression<<4)|(rate<<2)|(size<<1)|type);
41     swf_SetU16(tag,avgnumsamples);
42
43     printf("numSamples:%d\n",init->nSamples);
44     printf("bufferSize:%d\n",init->bufferSize);
45 }
46
47 void swf_SetSoundStreamBlock(TAG*tag, S16*samples, int numsamples, char first)
48 {
49     char*buf;
50     int len = 0;
51
52     buf = malloc(init->bufferSize);
53     if(!buf)
54         return;
55     
56     len = codecEncodeChunk (numsamples, samples, buf);
57     len += codecFlush (&buf[len]);
58     len += codecExit (&buf[len]);
59
60     if(first) {
61         swf_SetU16(tag, numsamples); // number of samples
62         swf_SetU16(tag, 0); // seek
63     }
64     swf_SetBlock(tag, buf, len);
65     free(buf);
66 }
67 #endif
68
69
70 #ifdef LAME
71
72 #include "../lame/lame.h"
73     
74 static lame_global_flags*lame_flags;
75
76 void swf_SetSoundStreamHead(TAG*tag, U16 avgnumsamples)
77 {
78     unsigned char buf[4096];
79     int bufsize = 4096;
80     int len;
81     short int samples[1152];
82
83     U8 playbackrate = 1; // 0 = 5.5 Khz, 1 = 11 Khz, 2 = 22 Khz, 3 = 44 Khz
84     U8 playbacksize = 1; // 0 = 8 bit, 1 = 16 bit
85     U8 playbacktype = 0; // 0 = mono, 1 = stereo
86     U8 compression = 2; // 0 = raw, 1 = ADPCM, 2 = mp3
87     U8 rate = 1; // 0 = 5.5 Khz, 1 = 11 Khz, 2 = 22 Khz, 3 = 44 Khz
88     U8 size = 1; // 0 = 8 bit, 1 = 16 bit
89     U8 type = 0; // 0 = mono, 1 = stereo
90     
91     memset(samples,0,sizeof(samples));
92
93     lame_flags = lame_init();
94
95     lame_set_in_samplerate(lame_flags, 44100);
96     lame_set_num_channels(lame_flags, 1);
97     lame_set_scale(lame_flags, 0);
98
99     // MPEG1    32, 44.1,   48khz
100     // MPEG2    16, 22.05,  24
101     // MPEG2.5   8, 11.025, 12
102     // (not used by decoding routines)
103     lame_set_out_samplerate(lame_flags, 11025);
104
105     lame_set_quality(lame_flags, 0);
106     lame_set_mode(lame_flags, MONO/*3*/);
107     //lame_set_compression_ratio(lame_flags, 11.025);
108     lame_set_bWriteVbrTag(lame_flags, 0);
109
110     lame_init_params(lame_flags);
111     lame_init_bitstream(lame_flags);
112
113     swf_SetU8(tag,(playbackrate<<2)|(playbacksize<<1)|playbacktype);
114     swf_SetU8(tag,(compression<<4)|(rate<<2)|(size<<1)|type);
115     swf_SetU16(tag,avgnumsamples);
116    
117     /* The first two flush calls to lame always fail, for
118        some reason. Do them here where they cause no damage. */
119     len = lame_encode_flush_nogap(lame_flags, buf, bufsize);
120     //printf("init:flush_nogap():%d\n", len);
121     len = lame_encode_flush(lame_flags, buf, bufsize);
122     //printf("init:flush():%d\n", len);
123 }
124
125 void swf_SetSoundStreamBlock(TAG*tag, S16*samples, int numsamples, char first)
126 {
127     char*buf;
128     int oldlen=0,len = 0;
129     int bufsize = 16384;
130
131     buf = malloc(bufsize);
132     if(!buf)
133         return;
134
135     if(first) {
136         int fs = lame_get_framesize(lame_flags);
137         //printf("framesize:%d\n", fs);
138         swf_SetU16(tag, fs); // samples per mp3 frame
139         swf_SetU16(tag, 0); // seek
140     }
141
142     len += lame_encode_buffer(lame_flags, samples, samples, numsamples, &buf[len], bufsize-len);
143     //printf("block: %d (+%d)\n", len, len-oldlen);
144     oldlen = len;
145
146     len += lame_encode_flush_nogap(lame_flags, &buf[len], bufsize-len);
147     //printf("flush: %d (+%d)\n", len, len-oldlen);
148     oldlen = len;
149     
150     swf_SetBlock(tag, buf, len);
151
152    /* len += lame_encode_flush(lame_flags, &buf[len], bufsize-len);
153     printf("flush! %d (+%d)\n", len, len-oldlen);*/
154
155     free(buf);
156 }
157
158 void swf_SetSoundStreamEnd(TAG*tag)
159 {
160     lame_close (lame_flags);
161 }
162 #endif
163
164 #endif // RFXSWF_DISABLESOUND