From d145fa6f792d6c44da2feec90507adb94e40323e Mon Sep 17 00:00:00 2001 From: Hans-Christoph Steiner Date: Sun, 31 Jul 2011 03:59:16 +0000 Subject: got everything building using the template Makefile svn path=/trunk/externals/unauthorized/; revision=15173 --- mp3write~.c | 608 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 608 insertions(+) create mode 100644 mp3write~.c (limited to 'mp3write~.c') diff --git a/mp3write~.c b/mp3write~.c new file mode 100644 index 0000000..d16c427 --- /dev/null +++ b/mp3write~.c @@ -0,0 +1,608 @@ +/* ------------------------ mp3write~ ----------------------------------------- */ +/* */ +/* Tilde object to record audio in mp3 format */ +/* Note that it is, in no way, related to mp3play~ */ +/* Written by Yves Degoyon (ydegoyon@free.fr). */ +/* Tarballs and updates at http://ydegoyon.free.fr */ +/* */ +/* This program is free software; you can redistribute it and/or */ +/* modify it under the terms of the GNU General Public License */ +/* as published by the Free Software Foundation; either version 2 */ +/* of the License, or (at your option) any later version. */ +/* */ +/* See file LICENSE for further informations on licensing terms. */ +/* */ +/* This program is distributed in the hope that it will be useful, */ +/* but WITHOUT ANY WARRANTY; without even the implied warranty of */ +/* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the */ +/* GNU General Public License for more details. */ +/* */ +/* You should have received a copy of the GNU General Public License */ +/* along with this program; if not, write to the Free Software */ +/* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ +/* */ +/* Based on PureData by Miller Puckette and others. */ +/* Uses the LAME MPEG 1 Layer 3 version 3.92 */ +/* Get it via http://www.freshmeat.net */ +/* */ +/* "All this talk of blood and iron is the cause of all my kicking" */ +/* Gang Of Four - "Guns Before Butter" */ +/* ---------------------------------------------------------------------------- */ + +#include +#include +#include +#include +#include +#include +#ifndef __APPLE__ +#include +#endif +#include +#ifdef _WIN32 +#include +#include +#include +#include +#else +#include +#include +#include +#include +#include +#include +#define SOCKET_ERROR -1 +#endif /* _WIN32 */ +#include +#include /* lame encoder stuff */ +#include "m_pd.h" /* standard pd stuff */ + +#define MY_MP3_MALLOC_IN_SIZE 65536 + /* max size taken from lame readme */ +#define MY_MP3_MALLOC_OUT_SIZE 1.25*MY_MP3_MALLOC_IN_SIZE+7200 + +#define MAXDATARATE 320 /* maximum mp3 data rate is 320kbit/s */ +#define STRBUF_SIZE 32 + +static char *mp3write_version = "mp3write~: mp3 file recorder version 0.4, written by Yves Degoyon"; +static int sockfd; + +static t_class *mp3write_class; + +typedef struct _mp3write +{ + t_object x_obj; + + /* LAME stuff */ + int x_lame; /* info about encoder status */ + int x_lamechunk; /* chunk size for LAME encoder */ + + /* buffer stuff */ + unsigned short x_inp; /* in position for buffer */ + unsigned short x_outp; /* out position for buffer*/ + short *x_mp3inbuf; /* data to be sent to LAME */ + char *x_mp3outbuf; /* data returned by LAME -> our mp3 stream */ + int x_mp3size; /* number of returned mp3 samples */ + short *x_buffer; /* data to be buffered */ + int x_bytesbuffered; /* number of unprocessed bytes in buffer */ + int x_start; + + /* mp3 format stuff */ + int x_samplerate; + int x_bitrate; /* bitrate of mp3 stream */ + int x_mp3mode; /* mode (mono, joint stereo, stereo, dual mono) */ + int x_mp3quality; /* quality of encoding */ + + /* recording stuff */ + int x_fd; /* file descriptor of the mp3 output */ + int x_file_open_mode; /* file opening mode */ + int x_byteswritten; /* number of bytes written */ + int x_recflag; /* recording flag toggled by messages "start" and "stop" */ + + t_float x_f; /* float needed for signal input */ + char *x_title; /* title of the mp3 */ + + lame_global_flags *lgfp; /* lame encoder configuration */ +} t_mp3write; + + + /* encode PCM data to mp3 stream */ +static void mp3write_encode(t_mp3write *x) +{ + unsigned short i, wp; + int err = -1; + int n = x->x_lamechunk; + +#ifdef UNIX + if(x->x_lamechunk < (int)sizeof(x->x_mp3inbuf)) +#else + if(x->x_lamechunk < sizeof(x->x_mp3inbuf)) +#endif + { + error("not enough memory!"); + return; + } + + /* on start/reconnect set outpoint so that it won't interfere with inpoint */ + if(x->x_start == -1) + { + post("mp3write~: reseting buffer positions"); + /* we try to keep 2.5 times the data the encoder needs in the buffer */ + if(x->x_inp > (2 * x->x_lamechunk)) + { + x->x_outp = (short) x->x_inp - (2.5 * x->x_lamechunk); + } + else if(x->x_inp < (2 * x->x_lamechunk)) + { + x->x_outp = (short) MY_MP3_MALLOC_IN_SIZE - (2.5 * x->x_lamechunk); + } + x->x_start = 1; + } + if((unsigned short)(x->x_outp - x->x_inp) < x->x_lamechunk)error("mp3write~: buffers overlap!"); + + i = MY_MP3_MALLOC_IN_SIZE - x->x_outp; + + /* read from buffer */ + if(x->x_lamechunk <= i) + { + /* enough data until end of buffer */ + for(n = 0; n < x->x_lamechunk; n++) /* fill encode buffer */ + { + x->x_mp3inbuf[n] = x->x_buffer[n + x->x_outp]; + } + x->x_outp += x->x_lamechunk; + } + else /* split data */ + { + for(wp = 0; wp < i; wp++) /* data at end of buffer */ + { + x->x_mp3inbuf[wp] = x->x_buffer[wp + x->x_outp]; + } + + for(wp = i; wp < x->x_lamechunk; wp++) /* write rest of data at beginning of buffer */ + { + x->x_mp3inbuf[wp] = x->x_buffer[wp - i]; + } + x->x_outp = x->x_lamechunk - i; + } + + /* encode mp3 data */ + + x->x_mp3size = lame_encode_buffer_interleaved(x->lgfp, x->x_mp3inbuf, + x->x_lamechunk/lame_get_num_channels(x->lgfp), + x->x_mp3outbuf, MY_MP3_MALLOC_OUT_SIZE); + // post( "mp3write~ : encoding returned %d frames", x->x_mp3size ); + + /* check result */ + if(x->x_mp3size<0) + { + lame_close( x->lgfp ); + error("mp3write~: lame_encode_buffer_interleaved failed (%d)", x->x_mp3size); + x->x_lame = -1; + } +} + + /* store mp3 frames in the file */ +static void mp3write_writeframes(t_mp3write *x) +{ + int err = -1; /* error return code */ + + if ( x->x_fd < 0 ) { + post( "mp3write~ : error : trying to write frames but no valid file is opened" ); + return; + } + +#ifdef _WIN32 + err = _write(x->x_fd, x->x_mp3outbuf, x->x_mp3size); +#else + err = write(x->x_fd, x->x_mp3outbuf, x->x_mp3size); +#endif + + if(err < 0) + { + error("mp3write~: could not write encoded data to file (%d)", err); + lame_close( x->lgfp ); + x->x_lame = -1; +#ifdef _WIN32 + error("mp3write~: writing data"); + _close(x->x_fd); +#else + perror("mp3write~: writing data"); + close(x->x_fd); +#endif + x->x_fd = -1; + } + else + { + x->x_byteswritten += err; + outlet_float( x->x_obj.ob_outlet, x->x_byteswritten ); + } + if((err > 0)&&(err != x->x_mp3size))error("mp3write~: %d bytes skipped", x->x_mp3size - err); +} + + + /* buffer data as channel interleaved PCM */ +static t_int *mp3write_perform(t_int *w) +{ + t_float *in1 = (t_float *)(w[1]); /* left audio inlet */ + t_float *in2 = (t_float *)(w[2]); /* right audio inlet */ + t_mp3write *x = (t_mp3write *)(w[3]); + int n = (int)(w[4]); /* number of samples */ + unsigned short i,wp; + float in; + + /* copy the data into the buffer */ + i = MY_MP3_MALLOC_IN_SIZE - x->x_inp; /* space left at the end of buffer */ + + n *= 2; /* two channels go into one buffer */ + + if( n <= i ) + { + /* the place between inp and MY_MP3_MALLOC_IN_SIZE */ + /* is big enough to hold the data */ + + for(wp = 0; wp < n; wp++) + { + if(wp%2) + { + in = *(in2++); /* right channel / inlet */ + } + else + { + in = *(in1++); /* left channel / inlet */ + } + if (in > 1.0) { in = 1.0; } + if (in < -1.0) { in = -1.0; } + x->x_buffer[wp + x->x_inp] = (short) (32767.0 * in); + } + x->x_inp += n; /* n more samples written to buffer */ + } + else + { + /* the place between inp and MY_MP3_MALLOC_IN_SIZE is not */ + /* big enough to hold the data */ + /* writing will take place in two turns, one from */ + /* x->x_inp -> MY_MP3_MALLOC_IN_SIZE, then from 0 on */ + + for(wp = 0; wp < i; wp++) /* fill up to end of buffer */ + { + if(wp%2) + { + in = *(in2++); + } + else + { + in = *(in1++); + } + if (in > 1.0) { in = 1.0; } + if (in < -1.0) { in = -1.0; } + x->x_buffer[wp + x->x_inp] = (short) (32767.0 * in); + } + for(wp = i; wp < n; wp++) /* write rest at start of buffer */ + { + if(wp%2) + { + in = *(in2++); + } + else + { + in = *(in1++); + } + if (in > 1.0) { in = 1.0; } + if (in < -1.0) { in = -1.0; } + x->x_buffer[wp - i] = (short) (32767.0 * in); + } + x->x_inp = n - i; /* new writeposition in buffer */ + } + + if((x->x_fd >= 0)&&(x->x_lame >= 0)&&(x->x_recflag)) + { + /* count buffered samples when things are running */ + x->x_bytesbuffered += n; + + /* encode and send to server */ + if(x->x_bytesbuffered > x->x_lamechunk) + { + mp3write_encode(x); /* encode to mp3 */ + mp3write_writeframes(x); /* write mp3 to file */ + x->x_bytesbuffered -= x->x_lamechunk; + } + } + else + { + x->x_start = -1; + } + return (w+5); +} + +static void mp3write_dsp(t_mp3write *x, t_signal **sp) +{ + dsp_add(mp3write_perform, 4, sp[0]->s_vec, sp[1]->s_vec, x, sp[0]->s_n); +} + + /* initialize the lame library */ +static int mp3write_tilde_lame_init(t_mp3write *x) +{ + time_t now; + + int ret; + x->lgfp = lame_init(); /* set default parameters for now */ + +#ifdef _WIN32 + /* load lame_enc.dll library */ + HINSTANCE dll; + dll=LoadLibrary("lame_enc.dll"); + if(dll==NULL) + { + error("mp3write~: error loading lame_enc.dll"); + closesocket(x->x_fd); + x->x_fd = -1; + post("mp3write~: connection closed"); + return -1; + } +#endif + { + const char *lameVersion = get_lame_version(); + post( "mp3write~ : using lame version : %s", lameVersion ); + } + + /* setting lame parameters */ + lame_set_num_channels( x->lgfp, 2); + lame_set_in_samplerate( x->lgfp, sys_getsr() ); + lame_set_out_samplerate( x->lgfp, x->x_samplerate ); + lame_set_brate( x->lgfp, x->x_bitrate ); + lame_set_mode( x->lgfp, x->x_mp3mode ); + lame_set_quality( x->lgfp, x->x_mp3quality ); + lame_set_emphasis( x->lgfp, 1 ); + lame_set_original( x->lgfp, 1 ); + lame_set_copyright( x->lgfp, 1 ); /* viva free music societies !!! */ + lame_set_disable_reservoir( x->lgfp, 0 ); + lame_set_padding_type( x->lgfp, PAD_NO ); + ret = lame_init_params( x->lgfp ); + if ( ret<0 ) { + post( "mp3write~ : error : lame params initialization returned : %d", ret ); + return -1; + } else { + x->x_lame=1; + /* magic formula copied from windows dll for MPEG-I */ + x->x_lamechunk = 2*1152; + + post( "mp3write~ : lame initialization done. (%d)", x->x_lame ); + } + lame_init_bitstream( x->lgfp ); + + /* setting tag information */ + id3tag_init(x->lgfp); + id3tag_v1_only(x->lgfp); + id3tag_space_v1(x->lgfp); + id3tag_set_artist(x->lgfp, "Pd Session"); + now=time(NULL); + sprintf( x->x_title, "Started at %s", ctime(&now) ); + id3tag_set_title(x->lgfp, x->x_title ); + + return 0; + +} + + /* open file and initialize lame */ +static void mp3write_open(t_mp3write *x, t_symbol *sfile) +{ + if ( mp3write_tilde_lame_init(x) < 0 ) { + error( "mp3write~ : lame initialization failed ... check parameters."); + return; + } + + /* closing previous file descriptor */ + if ( x->x_fd > 0 ) { +#ifdef _WIN32 + if(_close(x->x_fd) < 0 ) +#else + if(close(x->x_fd) < 0) +#endif + { + perror( "mp3write~ : closing file" ); + } + } + + if ( x->x_recflag ) { + x->x_recflag = 0; + } + +#ifdef _WIN32 + if ( ( x->x_fd = _open( sfile->s_name, x->x_file_open_mode, _S_IREAD|_S_IWRITE) ) < 0 ) +#else + if ( ( x->x_fd = open( sfile->s_name, x->x_file_open_mode, S_IRWXU|S_IRWXG|S_IRWXO ) ) < 0 ) +#endif + { + error( "mp3write~ : cannot open >%s<", sfile->s_name); + x->x_fd=-1; + return; + } + x->x_byteswritten = 0; + post( "mp3write~ : opened >%s< fd=%d", sfile->s_name, x->x_fd); +} + + /* setting file write mode to append */ +static void mp3write_append(t_mp3write *x) +{ +#ifdef _WIN32 + x->x_file_open_mode = _O_CREAT|_O_WRONLY|_O_APPEND|_O_BINARY; +#else + x->x_file_open_mode = O_CREAT|O_WRONLY|O_APPEND|O_NONBLOCK; +#endif + if(x->x_fd>=0)post("mp3write~ : mode set to append : open a new file to make changes take effect! "); +} + + /* setting file write mode to truncate */ +static void mp3write_truncate(t_mp3write *x) +{ +#ifdef _WIN32 + x->x_file_open_mode = _O_CREAT|_O_WRONLY|_O_TRUNC|_O_BINARY; +#else + x->x_file_open_mode = O_CREAT|O_WRONLY|O_TRUNC|O_NONBLOCK; +#endif + if(x->x_fd>=0)post("mp3write~ : mode set to truncate : open a new file to make changes take effect! "); +} + + /* settings for mp3 encoding */ +static void mp3write_mpeg(t_mp3write *x, t_floatarg fsamplerate, t_floatarg fbitrate, + t_floatarg fmode, t_floatarg fquality) +{ + x->x_samplerate = fsamplerate; + if(fbitrate > MAXDATARATE) + { + fbitrate = MAXDATARATE; + } + x->x_bitrate = fbitrate; + x->x_mp3mode = fmode; + x->x_mp3quality = fquality; + post("mp3write~: setting mp3 stream to %dHz, %dkbit/s, mode %d, quality %d", + x->x_samplerate, x->x_bitrate, x->x_mp3mode, x->x_mp3quality); + if(x->x_fd>=0)post("mp3write~ : restart recording to make changes take effect! "); +} + + /* print settings */ +static void mp3write_print(t_mp3write *x) +{ + const char * buf = 0; + post(mp3write_version); + post(" LAME mp3 settings:\n" + " output sample rate: %d Hz\n" + " bitrate: %d kbit/s", x->x_samplerate, x->x_bitrate); + switch(x->x_mp3mode) + { + case 0 : + buf = "stereo"; + break; + case 1 : + buf = "joint stereo"; + break; + case 2 : + buf = "dual channel"; + break; + case 3 : + buf = "mono"; + break; + } + post(" mode: %s\n" + " quality: %d", buf, x->x_mp3quality); +#ifdef _WIN32 + if(x->x_lamechunk!=0)post(" calculated mp3 chunk size: %d", x->x_lamechunk); +#else + post(" mp3 chunk size: %d", x->x_lamechunk); +#endif + if(x->x_samplerate!=sys_getsr()) + { + post(" resampling from %d to %d Hz!", (int)sys_getsr(), x->x_samplerate); + } +} + + /* start recording */ +static void mp3write_start(t_mp3write *x) +{ + if ( x->x_fd < 0 ) { + post("mp3write~: start received but no file has been set ... ignored."); + return; + } + + if ( x->x_recflag == 1 ) { + post("mp3write~: start received but recording is started ... ignored."); + return; + } + + x->x_recflag = 1; + post("mp3write~: start recording"); +} + + /* stop recording */ +static void mp3write_stop(t_mp3write *x) +{ + int err = -1; + + if ( x->x_fd < 0 ) { + post("mp3write~: stop received but no file has been set ... ignored."); + return; + } + + if ( x->x_recflag == 0 ) { + post("mp3write~: stop received but recording is stopped ... ignored."); + return; + } + /* first stop recording / buffering and so on, than do the rest */ + x->x_recflag = 0; + + /* flushing remaining frames and tag */ + x->x_mp3size = lame_encode_flush( x->lgfp, x->x_mp3outbuf, MY_MP3_MALLOC_OUT_SIZE ); + + mp3write_writeframes(x); /* write mp3 to file */ + + x->x_recflag = 0; + post("mp3write~: stop recording, flushed %d bytes", x->x_mp3size); +} + + /* clean up */ +static void mp3write_free(t_mp3write *x) +{ + if(x->x_lame >= 0) + lame_close( x->lgfp ); + if(x->x_fd >= 0) +#ifdef _WIN32 + _close(x->x_fd); +#else + close(x->x_fd); +#endif + freebytes(x->x_mp3inbuf, MY_MP3_MALLOC_IN_SIZE*sizeof(short)); + freebytes(x->x_mp3outbuf, MY_MP3_MALLOC_OUT_SIZE); + freebytes(x->x_buffer, MY_MP3_MALLOC_IN_SIZE*sizeof(short)); + freebytes( x->x_title, 255 ); +} + +static void *mp3write_new(void) +{ + t_mp3write *x = (t_mp3write *)pd_new(mp3write_class); + inlet_new (&x->x_obj, &x->x_obj.ob_pd, gensym ("signal"), gensym ("signal")); + outlet_new (&x->x_obj, &s_float); + x->x_fd = -1; +#ifdef _WIN32 + x->x_file_open_mode = _O_CREAT|_O_WRONLY|_O_APPEND|_O_BINARY; +#else + x->x_file_open_mode = O_CREAT|O_WRONLY|O_APPEND|O_NONBLOCK; +#endif + x->x_lame = -1; + x->x_samplerate = sys_getsr(); + x->x_bitrate = 128; + x->x_byteswritten = 0; + x->x_mp3mode = 1; + x->lgfp = NULL; + x->x_title = getbytes( 255 ); + x->x_mp3quality = 5; + x->x_mp3inbuf = getbytes(MY_MP3_MALLOC_IN_SIZE*sizeof(short)); /* buffer for encoder input */ + x->x_mp3outbuf = getbytes(MY_MP3_MALLOC_OUT_SIZE*sizeof(char)); /* our mp3 stream */ + x->x_buffer = getbytes(MY_MP3_MALLOC_IN_SIZE*sizeof(short)); /* what we get from pd, converted to PCM */ + if ((!x->x_buffer)||(!x->x_mp3inbuf)||(!x->x_mp3outbuf)) /* check buffers... */ + { + error("mp3write~ : cannot allocate internal buffers"); + return NULL; + } + x->x_bytesbuffered = 0; + x->x_inp = 0; + x->x_outp = 0; + x->x_start = -1; + return(x); +} + +void mp3write_tilde_setup(void) +{ + post(mp3write_version); + mp3write_class = class_new(gensym("mp3write~"), (t_newmethod)mp3write_new, (t_method)mp3write_free, + sizeof(t_mp3write), 0, 0); + CLASS_MAINSIGNALIN(mp3write_class, t_mp3write, x_f ); + class_addmethod(mp3write_class, (t_method)mp3write_dsp, gensym("dsp"), 0); + class_addmethod(mp3write_class, (t_method)mp3write_open, gensym("open"), A_SYMBOL, 0); + class_addmethod(mp3write_class, (t_method)mp3write_mpeg, gensym("mpeg"), A_FLOAT, A_FLOAT, A_FLOAT, A_FLOAT, 0); + class_addmethod(mp3write_class, (t_method)mp3write_start, gensym("start"), 0); + class_addmethod(mp3write_class, (t_method)mp3write_stop, gensym("stop"), 0); + class_addmethod(mp3write_class, (t_method)mp3write_print, gensym("print"), 0); + class_addmethod(mp3write_class, (t_method)mp3write_append, gensym("append"), 0); + class_addmethod(mp3write_class, (t_method)mp3write_truncate, gensym("truncate"), 0); +} -- cgit v1.2.1