ref: 39bf5b987e8ab53a74adec5a15b30f4eda85b357
dir: /sys/src/cmd/audio/mp3enc/get_audio.c/
/* * Get Audio routines source file * * Copyright (c) 1999 Albert L Faber * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library 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 * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public * License along with this library; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. */ /* $Id: get_audio.c,v 1.61 2001/03/19 21:26:05 markt Exp $ */ #ifdef HAVE_CONFIG_H # include <config.h> #endif #include <assert.h> #ifdef HAVE_LIMITS_H # include <limits.h> #endif #include <stdio.h> #ifdef STDC_HEADERS # include <stdlib.h> # include <string.h> #else # ifndef HAVE_STRCHR # define strchr index # define strrchr rindex # endif char *strchr(), *strrchr(); # ifndef HAVE_MEMCPY # define memcpy(d, s, n) bcopy ((s), (d), (n)) # define memmove(d, s, n) bcopy ((s), (d), (n)) # endif #endif #include <math.h> #include <sys/types.h> #include <sys/stat.h> #include "lame.h" #include "main.h" #include "get_audio.h" #include "portableio.h" #include "timestatus.h" #include "lametime.h" #ifdef WITH_DMALLOC #include <dmalloc.h> #endif /* global data for get_audio.c. */ int count_samples_carefully; int pcmbitwidth; mp3data_struct mp3input_data; /* used by Ogg and MP3 */ unsigned int num_samples_read; FILE *musicin; #ifdef AMIGA_MPEGA int lame_decode_initfile(const char *fullname, mp3data_struct * const mp3data); #else int lame_decode_initfile(FILE * const fd, mp3data_struct * const mp3data); #endif /* read mp3 file until mpglib returns one frame of PCM data */ int lame_decode_fromfile(FILE * fd, short int pcm_l[], short int pcm_r[], mp3data_struct * mp3data); /* and for Vorbis: */ int lame_decode_ogg_initfile( lame_global_flags* gfp, FILE* fd, mp3data_struct* mp3data ); int lame_decode_ogg_fromfile( lame_global_flags* gfc, FILE* fd, short int pcm_l[], short int pcm_r[], mp3data_struct* mp3data ); static int read_samples_pcm(FILE * musicin, short sample_buffer[2304], int frame_size, int samples_to_read); static int read_samples_mp3(lame_global_flags * gfp, FILE * musicin, short int mpg123pcm[2][1152], int num_chan); static int read_samples_ogg(lame_global_flags * gfp, FILE * musicin, short int mpg123pcm[2][1152], int num_chan); void CloseSndFile(sound_file_format input, FILE * musicin); FILE *OpenSndFile(lame_global_flags * gfp, char *); /* Replacement for forward fseek(,,SEEK_CUR), because fseek() fails on pipes */ static int fskip(FILE * fp, long offset, int whence) { #ifndef PIPE_BUF char buffer[4096]; #else char buffer[PIPE_BUF]; #endif int read; if (0 == fseek(fp, offset, whence)) return 0; if (whence != SEEK_CUR || offset < 0) { fprintf(stderr, "fskip problem: Mostly the return status of functions is not evaluated so it is more secure to pollute <stderr>.\n"); return -1; } while (offset > 0) { read = offset > sizeof(buffer) ? sizeof(buffer) : offset; if ((read = fread(buffer, 1, read, fp)) <= 0) return -1; offset -= read; } return 0; } FILE * init_outfile(char *outPath, int decode) { FILE *outf; /* open the output file */ if (0 == strcmp(outPath, "-")) lame_set_stream_binary_mode(outf = stdout); else if ((outf = fopen(outPath, "wb+")) == NULL) return NULL; return outf; } void init_infile(lame_global_flags * gfp, char *inPath) { /* open the input file */ count_samples_carefully = 0; pcmbitwidth = 16; musicin = OpenSndFile(gfp, inPath); } void close_infile(void) { CloseSndFile(input_format, musicin); } void SwapBytesInWords(short *ptr, int short_words) { /* Some speedy code */ unsigned long val; unsigned long *p = (unsigned long *) ptr; #ifndef lint # if defined(CHAR_BIT) # if CHAR_BIT != 8 # error CHAR_BIT != 8 # endif # else # error can not determine number of bits in a char # endif #endif /* lint */ assert(sizeof(short) == 2); #if defined(SIZEOF_UNSIGNED_LONG) && SIZEOF_UNSIGNED_LONG == 4 for (; short_words >= 2; short_words -= 2, p++) { val = *p; *p = ((val << 8) & 0xFF00FF00) | ((val >> 8) & 0x00FF00FF); } ptr = (short *) p; for (; short_words >= 1; short_words -= 1, ptr++) { val = *ptr; *ptr = ((val << 8) & 0xFF00) | ((val >> 8) & 0x00FF); } #elif defined(SIZEOF_UNSIGNED_LONG) && SIZEOF_UNSIGNED_LONG == 8 for (; short_words >= 4; short_words -= 4, p++) { val = *p; *p = ((val << 8) & 0xFF00FF00FF00FF00) | ((val >> 8) & 0x00FF00FF00FF00FF); } ptr = (short *) p; for (; short_words >= 1; short_words -= 1, ptr++) { val = *ptr; *ptr = ((val << 8) & 0xFF00) | ((val >> 8) & 0x00FF); } #else # ifdef SIZEOF_UNSIGNED_LONG //# warning Using unoptimized SwapBytesInWords(). # endif for (; short_words >= 1; short_words -= 1, ptr++) { val = *ptr; *ptr = ((val << 8) & 0xFF00) | ((val >> 8) & 0x00FF); } #endif assert(short_words == 0); } /************************************************************************ * * get_audio() * * PURPOSE: reads a frame of audio data from a file to the buffer, * aligns the data for future processing, and separates the * left and right channels * ************************************************************************/ int get_audio(lame_global_flags * const gfp, short buffer[2][1152]) { int num_channels = lame_get_num_channels( gfp ); short insamp[2 * 1152]; int samples_read; int framesize; int samples_to_read; unsigned int remaining, tmp_num_samples; int j; short *p; /* * NOTE: LAME can now handle arbritray size input data packets, * so there is no reason to read the input data in chuncks of * size "gfp->framesize". EXCEPT: the LAME graphical frame analyzer * will get out of sync if we read more than framesize worth of data. */ samples_to_read = framesize = gfp->framesize; assert(framesize <= 1152); /* get num_samples */ tmp_num_samples = lame_get_num_samples( gfp ); /* if this flag has been set, then we are carefull to read * exactly num_samples and no more. This is useful for .wav and .aiff * files which have id3 or other tags at the end. Note that if you * are using LIBSNDFILE, this is not necessary */ if (count_samples_carefully) { remaining = tmp_num_samples - Min(tmp_num_samples, num_samples_read); if (remaining < framesize) samples_to_read = remaining; } switch (input_format) { case sf_mp1: case sf_mp2: case sf_mp3: samples_read = read_samples_mp3(gfp, musicin, buffer, num_channels); break; case sf_ogg: samples_read = read_samples_ogg(gfp, musicin, buffer, num_channels); break; default: samples_read = read_samples_pcm(musicin, insamp, num_channels * framesize, num_channels * samples_to_read); samples_read /= num_channels; p = insamp; switch (num_channels) { case 1: for (j = 0; j < framesize; j++) { buffer[0][j] = *p++; buffer[1][j] = 0; } break; case 2: for (j = 0; j < framesize; j++) { buffer[0][j] = *p++; buffer[1][j] = *p++; } break; default: assert(0); break; } } /* if num_samples = MAX_U_32_NUM, then it is considered infinitely long. Don't count the samples */ if ( tmp_num_samples != MAX_U_32_NUM ) num_samples_read += samples_read; return samples_read; } int read_samples_ogg(lame_global_flags * const gfp, FILE * const musicin, short int oggpcm[2][1152], const int stereo) { int out = 0; #ifdef HAVE_VORBIS static const char type_name[] = "Ogg Vorbis file"; out = lame_decode_ogg_fromfile( gfp, musicin, oggpcm[0], oggpcm[1], &mp3input_data ); /* * out < 0: error, probably EOF * out = 0: not possible with lame_decode_fromfile() ??? * out > 0: number of output samples */ if (out < 0) { memset(oggpcm, 0, sizeof(**oggpcm) * 2 * 1152); return 0; } if (lame_get_num_channels( gfp ) != mp3input_data.stereo) fprintf(stderr, "Error: number of channels has changed in %s - not supported\n", type_name); if ( lame_get_in_samplerate( gfp ) != mp3input_data.samplerate ) fprintf(stderr, "Error: sample frequency has changed in %s - not supported\n", type_name); #else out = -1; /* wanna read ogg without vorbis support? */ #endif return out; } int read_samples_mp3(lame_global_flags * const gfp, FILE * const musicin, short int mpg123pcm[2][1152], int stereo) { int out; #if defined(AMIGA_MPEGA) || defined(HAVE_MPGLIB) static const char type_name[] = "MP3 file"; out = lame_decode_fromfile(musicin, mpg123pcm[0], mpg123pcm[1], &mp3input_data); /* * out < 0: error, probably EOF * out = 0: not possible with lame_decode_fromfile() ??? * out > 0: number of output samples */ if (out < 0) { memset(mpg123pcm, 0, sizeof(**mpg123pcm) * 2 * 1152); return 0; } if ( lame_get_num_channels( gfp ) != mp3input_data.stereo ) fprintf(stderr, "Error: number of channels has changed in %s - not supported\n", type_name); if ( lame_get_in_samplerate( gfp ) != mp3input_data.samplerate ) fprintf(stderr, "Error: sample frequency has changed in %s - not supported\n", type_name); #else out = -1; #endif return out; } static int WriteWaveHeader(FILE * const fp, const int pcmbytes, const int freq, const int channels, const int bits) { int bytes = (bits + 7) / 8; /* quick and dirty, but documented */ fwrite("RIFF", 1, 4, fp); // label Write32BitsLowHigh(fp, pcmbytes + 44 - 8); // length in bytes without header fwrite("WAVEfmt ", 2, 4, fp); // 2 labels Write32BitsLowHigh(fp, 2 + 2 + 4 + 4 + 2 + 2); // length of PCM format declaration area Write16BitsLowHigh(fp, 1); // is PCM? Write16BitsLowHigh(fp, channels); // number of channels Write32BitsLowHigh(fp, freq); // sample frequency in [Hz] Write32BitsLowHigh(fp, freq * channels * bytes); // bytes per second Write16BitsLowHigh(fp, channels * bytes); // bytes per sample time Write16BitsLowHigh(fp, bits); // bits per sample fwrite("data", 1, 4, fp); // label Write32BitsLowHigh(fp, pcmbytes); // length in bytes of raw PCM data return ferror(fp) ? -1 : 0; } /* the simple lame decoder */ /* After calling lame_init(), lame_init_params() and * init_infile(), call this routine to read the input MP3 file * and output .wav data to the specified file pointer*/ /* lame_decoder will ignore the first 528 samples, since these samples * represent the mpglib delay (and are all 0). skip = number of additional * samples to skip, to (for example) compensate for the encoder delay */ int lame_decoder(lame_global_flags * gfp, FILE * outf, int skip, char *inPath, char *outPath) { short int Buffer[2][1152]; int iread; double wavsize; int i; void (*WriteFunction) (FILE * fp, char *p, int n); int tmp_num_channels = lame_get_num_channels( gfp ); fprintf(stderr, "\rinput: %s%s(%g kHz, %i channel%s, ", strcmp(inPath, "-") ? inPath : "<stdin>", strlen(inPath) > 26 ? "\n\t" : " ", lame_get_in_samplerate( gfp ) / 1.e3, tmp_num_channels, tmp_num_channels != 1 ? "s" : ""); switch (input_format) { case sf_mp3: skip += 528 + 1; /* mp3 decoder has a 528 sample delay, plus user supplied "skip" */ fprintf(stderr, "MPEG-%u%s Layer %s", 2 - gfp->version, lame_get_out_samplerate( gfp ) < 16000 ? ".5" : "", "III"); break; case sf_mp2: skip += 240 + 1; fprintf(stderr, "MPEG-%u%s Layer %s", 2 - gfp->version, lame_get_out_samplerate( gfp ) < 16000 ? ".5" : "", "II"); break; case sf_mp1: skip += 240 + 1; fprintf(stderr, "MPEG-%u%s Layer %s", 2 - gfp->version, lame_get_out_samplerate( gfp ) < 16000 ? ".5" : "", "I"); break; case sf_ogg: fprintf(stderr, "Ogg Vorbis"); skip = 0; /* other formats have no delay *//* is += 0 not better ??? */ break; case sf_raw: fprintf(stderr, "raw PCM data"); mp3input_data.nsamp = lame_get_num_samples( gfp ); mp3input_data.framesize = 1152; skip = 0; /* other formats have no delay *//* is += 0 not better ??? */ break; case sf_wave: fprintf(stderr, "Microsoft WAVE"); mp3input_data.nsamp = lame_get_num_samples( gfp ); mp3input_data.framesize = 1152; skip = 0; /* other formats have no delay *//* is += 0 not better ??? */ break; case sf_aiff: fprintf(stderr, "SGI/Apple AIFF"); mp3input_data.nsamp = lame_get_num_samples( gfp ); mp3input_data.framesize = 1152; skip = 0; /* other formats have no delay *//* is += 0 not better ??? */ break; default: fprintf(stderr, "unknown"); mp3input_data.nsamp = lame_get_num_samples( gfp ); mp3input_data.framesize = 1152; skip = 0; /* other formats have no delay *//* is += 0 not better ??? */ assert(0); break; } fprintf(stderr, ")\noutput: %s%s(16 bit, Microsoft WAVE)\n", strcmp(outPath, "-") ? outPath : "<stdout>", strlen(outPath) > 45 ? "\n\t" : " "); if (skip > 0) fprintf(stderr, "skipping initial %i samples (encoder+decoder delay)\n", skip); if ( 0 == lame_get_disable_waveheader( gfp ) ) WriteWaveHeader(outf, 0x7FFFFFFF, lame_get_in_samplerate( gfp ), tmp_num_channels, 16); /* unknown size, so write maximum 32 bit signed value */ wavsize = -skip; WriteFunction = swapbytes ? WriteBytesSwapped : WriteBytes; mp3input_data.totalframes = mp3input_data.nsamp / mp3input_data.framesize; assert(tmp_num_channels >= 1 && tmp_num_channels <= 2); do { iread = get_audio(gfp, Buffer); /* read in 'iread' samples */ mp3input_data.framenum += iread / mp3input_data.framesize; wavsize += iread; if (!silent) decoder_progress(gfp, &mp3input_data); skip -= (i = skip < iread ? skip : iread); /* 'i' samples are to skip in this frame */ for (; i < iread; i++) { if ( lame_get_disable_waveheader( gfp ) ) { WriteFunction(outf, (char *) Buffer[0] + i, sizeof(short)); if (tmp_num_channels == 2) WriteFunction(outf, (char *) Buffer[1] + i, sizeof(short)); } else { Write16BitsLowHigh(outf, Buffer[0][i]); if (tmp_num_channels == 2) Write16BitsLowHigh(outf, Buffer[1][i]); } } } while (iread); i = (16 / 8) * tmp_num_channels; assert(i > 0); if (wavsize <= 0) { fprintf(stderr, "WAVE file contains 0 PCM samples\n"); wavsize = 0; } else if (wavsize > 0xFFFFFFD0 / i) { fprintf(stderr, "Very huge WAVE file, can't set filesize accordingly\n"); wavsize = 0xFFFFFFD0; } else { wavsize *= i; } if ( 0 == lame_get_disable_waveheader( gfp ) ) if (!fseek(outf, 0l, SEEK_SET)) /* if outf is seekable, rewind and adjust length */ WriteWaveHeader(outf, wavsize, lame_get_in_samplerate( gfp ), tmp_num_channels, 16); fclose(outf); decoder_progress_finish(gfp); return 0; } #if defined(LIBSNDFILE) #if 0 /* currently disabled */ # include "sndfile.h" // prototype for sf_get_lib_version() void print_sndlib_version(FILE * fp) { char tmp[80]; sf_get_lib_version(tmp, sizeof(tmp)); fprintf(fp, "Input handled by %s (http://www.zip.com.au/~erikd/libsndfile/)\n", tmp); } #endif /* ** Copyright (C) 1999 Albert Faber ** * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library 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 * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public * License along with this library; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. */ void CloseSndFile(sound_file_format input, FILE * musicin) { SNDFILE *gs_pSndFileIn = (SNDFILE *) musicin; if (input == sf_mp1 || input == sf_mp2 || input == sf_mp3) { #ifndef AMIGA_MPEGA if (fclose(musicin) != 0) { fprintf(stderr, "Could not close audio input file\n"); exit(2); } #endif } else { if (gs_pSndFileIn) { if (sf_close(gs_pSndFileIn) != 0) { fprintf(stderr, "Could not close sound file \n"); exit(2); } } } } FILE * OpenSndFile(lame_global_flags * gfp, char *inPath) { char *lpszFileName = inPath; FILE *musicin; SNDFILE *gs_pSndFileIn; SF_INFO gs_wfInfo; if (input_format == sf_mp1 || input_format == sf_mp2 || input_format == sf_mp3) { #ifdef AMIGA_MPEGA if (-1 == lame_decode_initfile(lpszFileName, &mp3input_data)) { fprintf(stderr, "Error reading headers in mp3 input file %s.\n", lpszFileName); exit(1); } #endif #ifdef HAVE_MPGLIB if ((musicin = fopen(lpszFileName, "rb")) == NULL) { fprintf(stderr, "Could not find \"%s\".\n", lpszFileName); exit(1); } if (-1 == lame_decode_initfile(musicin, &mp3input_data)) { fprintf(stderr, "Error reading headers in mp3 input file %s.\n", lpszFileName); exit(1); } #endif if( -1 == lame_set_num_channels( gfp, mp3input_data.stereo ) ) { fprintf( stderr, "Unsupported number of channels: %ud\n", mp3input_data.stereo ); exit( 1 ); } (void) lame_set_in_samplerate( gfp, mp3input_data.samplerate ); (void) lame_set_num_samples( gfp, mp3input_data.nsamp ); } else if (input_format == sf_ogg) { #ifdef HAVE_VORBIS if ((musicin = fopen(lpszFileName, "rb")) == NULL) { fprintf(stderr, "Could not find \"%s\".\n", lpszFileName); exit(1); } if ( -1 == lame_decode_ogg_initfile( gfp, musicin, &mp3input_data ) ) { fprintf(stderr, "Error reading headers in mp3 input file %s.\n", lpszFileName); exit(1); } #else fprintf(stderr, "mp3enc not compiled with libvorbis support.\n"); exit(1); #endif } else { /* Try to open the sound file */ /* set some defaults incase input is raw PCM */ gs_wfInfo.seekable = (input_format != sf_raw); /* if user specified -r, set to not seekable */ gs_wfInfo.samplerate = lame_get_in_samplerate( gfp ); gs_wfInfo.pcmbitwidth = 16; gs_wfInfo.channels = lame_get_num_channels( gfp ); #ifndef WORDS_BIGENDIAN /* little endian */ if (swapbytes) gs_wfInfo.format = SF_FORMAT_RAW_BE; else gs_wfInfo.format = SF_FORMAT_RAW_LE; #else if (swapbytes) gs_wfInfo.format = SF_FORMAT_RAW_LE; else gs_wfInfo.format = SF_FORMAT_RAW_BE; #endif gs_pSndFileIn = sf_open_read(lpszFileName, &gs_wfInfo); musicin = (SNDFILE *) gs_pSndFileIn; /* Check result */ if (gs_pSndFileIn == NULL) { sf_perror(gs_pSndFileIn); fprintf(stderr, "Could not open sound file \"%s\".\n", lpszFileName); exit(1); } if ((gs_wfInfo.format == SF_FORMAT_RAW_LE) || (gs_wfInfo.format == SF_FORMAT_RAW_BE)) input_format = sf_raw; #ifdef _DEBUG_SND_FILE DEBUGF("\n\nSF_INFO structure\n"); DEBUGF("samplerate :%d\n", gs_wfInfo.samplerate); DEBUGF("samples :%d\n", gs_wfInfo.samples); DEBUGF("channels :%d\n", gs_wfInfo.channels); DEBUGF("pcmbitwidth :%d\n", gs_wfInfo.pcmbitwidth); DEBUGF("format :"); /* new formats from sbellon@sbellon.de 1/2000 */ switch (gs_wfInfo.format & SF_FORMAT_TYPEMASK) { case SF_FORMAT_WAV: DEBUGF("Microsoft WAV format (big endian). "); break; case SF_FORMAT_AIFF: DEBUGF("Apple/SGI AIFF format (little endian). "); break; case SF_FORMAT_AU: DEBUGF("Sun/NeXT AU format (big endian). "); break; case SF_FORMAT_AULE: DEBUGF("DEC AU format (little endian). "); break; case SF_FORMAT_RAW: DEBUGF("RAW PCM data. "); break; case SF_FORMAT_PAF: DEBUGF("Ensoniq PARIS file format. "); break; case SF_FORMAT_SVX: DEBUGF("Amiga IFF / SVX8 / SV16 format. "); break; case SF_FORMAT_NIST: DEBUGF("Sphere NIST format. "); break; default: assert(0); break; } switch (gs_wfInfo.format & SF_FORMAT_SUBMASK) { case SF_FORMAT_PCM: DEBUGF("PCM data in 8, 16, 24 or 32 bits."); break; case SF_FORMAT_FLOAT: DEBUGF("32 bit Intel x86 floats."); break; case SF_FORMAT_ULAW: DEBUGF("U-Law encoded."); break; case SF_FORMAT_ALAW: DEBUGF("A-Law encoded."); break; case SF_FORMAT_IMA_ADPCM: DEBUGF("IMA ADPCM."); break; case SF_FORMAT_MS_ADPCM: DEBUGF("Microsoft ADPCM."); break; case SF_FORMAT_PCM_BE: DEBUGF("Big endian PCM data."); break; case SF_FORMAT_PCM_LE: DEBUGF("Little endian PCM data."); break; case SF_FORMAT_PCM_S8: DEBUGF("Signed 8 bit PCM."); break; case SF_FORMAT_PCM_U8: DEBUGF("Unsigned 8 bit PCM."); break; case SF_FORMAT_SVX_FIB: DEBUGF("SVX Fibonacci Delta encoding."); break; case SF_FORMAT_SVX_EXP: DEBUGF("SVX Exponential Delta encoding."); break; default: assert(0); break; } DEBUGF("\n"); DEBUGF("pcmbitwidth :%d\n", gs_wfInfo.pcmbitwidth); DEBUGF("sections :%d\n", gs_wfInfo.sections); DEBUGF("seekable :\n", gs_wfInfo.seekable); #endif (void) lame_set_num_samples( gfp, gs_wfInfo.samples ); if( -1 == lame_set_num_channels( gfp, gs_wfInfo.channels ) ) { fprintf( stderr, "Unsupported number of channels: %ud\n", gs_wfInfo.channels ); exit( 1 ); } (void) lame_set_in_samplerate( gfp, gs_wfInfo.samplerate ); pcmbitwidth = gs_wfInfo.pcmbitwidth; } if (lame_get_num_samples( gfp ) == MAX_U_32_NUM) { /* try to figure out num_samples */ double flen = lame_get_file_size( lpszFileName ); if (flen >= 0) { /* try file size, assume 2 bytes per sample */ if (input_format == sf_mp1 || input_format == sf_mp2 || input_format == sf_mp3) { double totalseconds = (flen * 8.0 / (1000.0 * mp3input_data.bitrate)); unsigned long tmp_num_samples = totalseconds * lame_get_in_samplerate( gfp ); (void) lame_set_num_samples( gfp, tmp_num_samples ); mp3input_data.nsamp = tmp_num_samples; } else { lame_set_num_samples( gfp, flen / (2 * lame_get_num_channels( gfp )) ); } } } return musicin; } /************************************************************************ * * read_samples() * * PURPOSE: reads the PCM samples from a file to the buffer * * SEMANTICS: * Reads #samples_read# number of shorts from #musicin# filepointer * into #sample_buffer[]#. Returns the number of samples read. * ************************************************************************/ static int read_samples_pcm(FILE * const musicin, short sample_buffer[2304], int frame_size /* unused */ , int samples_to_read) { int i; int samples_read; samples_read = sf_read_short((SNDFILE *) musicin, sample_buffer, samples_to_read); switch (pcmbitwidth) { case 8: for (i = 0; i < samples_read; i++) sample_buffer[i] <<= 8; break; case 16: break; default: fprintf(stderr, "Only 8 and 16 bit input files supported \n"); exit(1); } return samples_read; } #else /* defined(LIBSNDFILE) */ /************************************************************************ ************************************************************************ ************************************************************************ ************************************************************************ ************************************************************************ ************************************************************************ * * OLD ISO/LAME routines follow. Used if you dont have LIBSNDFILE * or for stdin/stdout support * ************************************************************************ ************************************************************************ ************************************************************************ ************************************************************************ ************************************************************************ ************************************************************************/ /************************************************************************ * * read_samples() * * PURPOSE: reads the PCM samples from a file to the buffer * * SEMANTICS: * Reads #samples_read# number of shorts from #musicin# filepointer * into #sample_buffer[]#. Returns the number of samples read. * ************************************************************************/ int read_samples_pcm(FILE * musicin, short sample_buffer[2304], int frame_size, int samples_to_read) { int samples_read; int iswav = (input_format == sf_wave); if (16 == pcmbitwidth) { samples_read = fread(sample_buffer, 2, samples_to_read, musicin); } else if (8 == pcmbitwidth) { char temp[2304]; int i; samples_read = fread(temp, 1, samples_to_read, musicin); for (i = 0; i < samples_read; ++i) { /* note: 8bit .wav samples are unsigned */ /* map [0,255] -> [-32768,32767] */ sample_buffer[i] = ((short int)temp[i] - 128)*256 + 127; } } else { fprintf(stderr, "Only 8 and 16 bit input files supported \n"); exit(1); } if (ferror(musicin)) { fprintf(stderr, "Error reading input file\n"); exit(1); } if (16 == pcmbitwidth) { /* intel=littleEndian. wav files are always little endian */ #ifndef WORDS_BIGENDIAN /* little endian */ if (!iswav) SwapBytesInWords(sample_buffer, samples_read); #else /* big endian */ if (iswav) SwapBytesInWords(sample_buffer, samples_read); #endif if (swapbytes) SwapBytesInWords(sample_buffer, samples_read); } return samples_read; } /* AIFF Definitions */ #define IFF_ID_FORM 0x464f524d /* "FORM" */ #define IFF_ID_AIFF 0x41494646 /* "AIFF" */ #define IFF_ID_COMM 0x434f4d4d /* "COMM" */ #define IFF_ID_SSND 0x53534e44 /* "SSND" */ #define IFF_ID_MPEG 0x4d504547 /* "MPEG" */ #define WAV_ID_RIFF 0x52494646 /* "RIFF" */ #define WAV_ID_WAVE 0x57415645 /* "WAVE" */ #define WAV_ID_FMT 0x666d7420 /* "fmt " */ #define WAV_ID_DATA 0x64617461 /* "data" */ /***************************************************************************** * * Read Microsoft Wave headers * * By the time we get here the first 32-bits of the file have already been * read, and we're pretty sure that we're looking at a WAV file. * *****************************************************************************/ static int parse_wave_header(lame_global_flags * gfp, FILE * sf) { int format_tag = 0; int channels = 0; int block_align = 0; int bits_per_sample = 0; int samples_per_sec = 0; int avg_bytes_per_sec = 0; int is_wav = 0; long data_length = 0, file_length, subSize = 0; int loop_sanity = 0; file_length = Read32BitsHighLow(sf); if (Read32BitsHighLow(sf) != WAV_ID_WAVE) return 0; for (loop_sanity = 0; loop_sanity < 20; ++loop_sanity) { int type = Read32BitsHighLow(sf); if (type == WAV_ID_FMT) { subSize = Read32BitsLowHigh(sf); if (subSize < 16) { /*DEBUGF( "'fmt' chunk too short (only %ld bytes)!", subSize); */ return 0; } format_tag = Read16BitsLowHigh(sf); subSize -= 2; channels = Read16BitsLowHigh(sf); subSize -= 2; samples_per_sec = Read32BitsLowHigh(sf); subSize -= 4; avg_bytes_per_sec = Read32BitsLowHigh(sf); subSize -= 4; block_align = Read16BitsLowHigh(sf); subSize -= 2; bits_per_sample = Read16BitsLowHigh(sf); subSize -= 2; /* DEBUGF(" skipping %d bytes\n", subSize); */ if (subSize > 0) { if (fskip(sf, (long) subSize, SEEK_CUR) != 0) return 0; }; } else if (type == WAV_ID_DATA) { subSize = Read32BitsLowHigh(sf); data_length = subSize; is_wav = 1; /* We've found the audio data. Read no further! */ break; } else { subSize = Read32BitsLowHigh(sf); if (fskip(sf, (long) subSize, SEEK_CUR) != 0) return 0; } } if (format_tag != 1) { return 0; /* oh no! non-supported format */ } if (is_wav) { /* make sure the header is sane */ if( -1 == lame_set_num_channels( gfp, channels ) ) { fprintf( stderr, "Unsupported number of channels: %ud\n", channels ); exit( 1 ); } (void) lame_set_in_samplerate( gfp, samples_per_sec ); pcmbitwidth = bits_per_sample; (void) lame_set_num_samples( gfp, data_length / (channels * ((bits_per_sample+7) / 8)) ); } return is_wav; } /************************************************************************ * aiff_check2 * * PURPOSE: Checks AIFF header information to make sure it is valid. * returns 0 on success, 1 on errors ************************************************************************/ int aiff_check2(const char *file_name, IFF_AIFF * const pcm_aiff_data) { if (pcm_aiff_data->sampleType != IFF_ID_SSND) { fprintf(stderr, "Sound data is not PCM in '%s'\n", file_name); return 1; } if (pcm_aiff_data->sampleSize != sizeof(short) * CHAR_BIT) { fprintf(stderr, "Sound data is not %i bits in '%s'\n", sizeof(short) * CHAR_BIT, file_name); return 1; } if (pcm_aiff_data->numChannels != 1 && pcm_aiff_data->numChannels != 2) { fprintf(stderr, "Sound data is not mono or stereo in '%s'\n", file_name); return 1; } if (pcm_aiff_data->blkAlgn.blockSize != 0) { fprintf(stderr, "Block size is not 0 bytes in '%s'\n", file_name); return 1; } if (pcm_aiff_data->blkAlgn.offset != 0) { fprintf(stderr, "Block offset is not 0 bytes in '%s'\n", file_name); return 1; } return 0; } /***************************************************************************** * * Read Audio Interchange File Format (AIFF) headers. * * By the time we get here the first 32 bits of the file have already been * read, and we're pretty sure that we're looking at an AIFF file. * *****************************************************************************/ static int parse_aiff_header(lame_global_flags * gfp, FILE * sf) { int is_aiff = 0; long chunkSize = 0, subSize = 0; IFF_AIFF aiff_info; memset(&aiff_info, 0, sizeof(aiff_info)); chunkSize = Read32BitsHighLow(sf); if (Read32BitsHighLow(sf) != IFF_ID_AIFF) return 0; while (chunkSize > 0) { int type = Read32BitsHighLow(sf); chunkSize -= 4; /* DEBUGF( "found chunk type %08x '%4.4s'\n", type, (char*)&type); */ /* don't use a switch here to make it easier to use 'break' for SSND */ if (type == IFF_ID_COMM) { subSize = Read32BitsHighLow(sf); chunkSize -= subSize; aiff_info.numChannels = Read16BitsHighLow(sf); subSize -= 2; aiff_info.numSampleFrames = Read32BitsHighLow(sf); subSize -= 4; aiff_info.sampleSize = Read16BitsHighLow(sf); subSize -= 2; aiff_info.sampleRate = ReadIeeeExtendedHighLow(sf); subSize -= 10; if (fskip(sf, (long) subSize, SEEK_CUR) != 0) return 0; } else if (type == IFF_ID_SSND) { subSize = Read32BitsHighLow(sf); chunkSize -= subSize; aiff_info.blkAlgn.offset = Read32BitsHighLow(sf); subSize -= 4; aiff_info.blkAlgn.blockSize = Read32BitsHighLow(sf); subSize -= 4; if (fskip(sf, (long) aiff_info.blkAlgn.offset, SEEK_CUR) != 0) return 0; aiff_info.sampleType = IFF_ID_SSND; is_aiff = 1; /* We've found the audio data. Read no further! */ break; } else { subSize = Read32BitsHighLow(sf); chunkSize -= subSize; if (fskip(sf, (long) subSize, SEEK_CUR) != 0) return 0; } } /* DEBUGF("Parsed AIFF %d\n", is_aiff); */ if (is_aiff) { /* make sure the header is sane */ if (0 != aiff_check2("name" /*???????????? */ , &aiff_info)) return 0; if( -1 == lame_set_num_channels( gfp, aiff_info.numChannels ) ) { fprintf( stderr, "Unsupported number of channels: %ud\n", aiff_info.numChannels ); exit( 1 ); } (void) lame_set_in_samplerate( gfp, aiff_info.sampleRate ); pcmbitwidth = aiff_info.sampleSize; (void) lame_set_num_samples( gfp, aiff_info.numSampleFrames ); } return is_aiff; } /************************************************************************ * * parse_file_header * * PURPOSE: Read the header from a bytestream. Try to determine whether * it's a WAV file or AIFF without rewinding, since rewind * doesn't work on pipes and there's a good chance we're reading * from stdin (otherwise we'd probably be using libsndfile). * * When this function returns, the file offset will be positioned at the * beginning of the sound data. * ************************************************************************/ void parse_file_header(lame_global_flags * gfp, FILE * sf) { int type = Read32BitsHighLow(sf); /* DEBUGF( "First word of input stream: %08x '%4.4s'\n", type, (char*) &type); */ count_samples_carefully = 0; input_format = sf_raw; if (type == WAV_ID_RIFF) { /* It's probably a WAV file */ if (parse_wave_header(gfp, sf)) { input_format = sf_wave; count_samples_carefully = 1; } else { fprintf( stderr, "Warning: corrupt or unsupported WAVE format\n"); } } else if (type == IFF_ID_FORM) { /* It's probably an AIFF file */ if (parse_aiff_header(gfp, sf)) { input_format = sf_aiff; count_samples_carefully = 1; } } if (input_format == sf_raw) { /* ** Assume it's raw PCM. Since the audio data is assumed to begin ** at byte zero, this will unfortunately require seeking. */ if (fseek(sf, 0L, SEEK_SET) != 0) { /* ignore errors */ } input_format = sf_raw; } } void CloseSndFile(sound_file_format input, FILE * musicin) { if (fclose(musicin) != 0) { fprintf(stderr, "Could not close audio input file\n"); exit(2); } } FILE * OpenSndFile(lame_global_flags * gfp, char *inPath) { FILE *musicin; /* set the defaults from info incase we cannot determine them from file */ lame_set_num_samples( gfp, MAX_U_32_NUM ); if (!strcmp(inPath, "-")) { lame_set_stream_binary_mode(musicin = stdin); /* Read from standard input. */ } else { if ((musicin = fopen(inPath, "rb")) == NULL) { fprintf(stderr, "Could not find \"%s\".\n", inPath); exit(1); } } if (input_format == sf_mp1 || input_format == sf_mp2 || input_format == sf_mp3) { #ifdef AMIGA_MPEGA if (-1 == lame_decode_initfile(inPath, &mp3input_data)) { fprintf(stderr, "Error reading headers in mp3 input file %s.\n", inPath); exit(1); } #endif #ifdef HAVE_MPGLIB if (-1 == lame_decode_initfile(musicin, &mp3input_data)) { fprintf(stderr, "Error reading headers in mp3 input file %s.\n", inPath); exit(1); } #endif if( -1 == lame_set_num_channels( gfp, mp3input_data.stereo ) ) { fprintf( stderr, "Unsupported number of channels: %ud\n", mp3input_data.stereo ); exit( 1 ); } (void) lame_set_in_samplerate( gfp, mp3input_data.samplerate ); (void) lame_set_num_samples( gfp, mp3input_data.nsamp ); } else if (input_format == sf_ogg) { #ifdef HAVE_VORBIS if ( -1 == lame_decode_ogg_initfile( gfp, musicin, &mp3input_data ) ) { fprintf(stderr, "Error reading headers in ogg input file %s.\n", inPath); exit(1); } if( -1 == lame_set_num_channels( gfp, mp3input_data.stereo ) ) { fprintf( stderr, "Unsupported number of channels: %ud\n", mp3input_data.stereo ); exit( 1 ); } (void) lame_set_in_samplerate( gfp, mp3input_data.samplerate ); (void) lame_set_num_samples( gfp, mp3input_data.nsamp ); #else fprintf(stderr, "mp3enc not compiled with libvorbis support.\n"); exit(1); #endif } else { if (input_format != sf_raw) { parse_file_header(gfp, musicin); } if (0 && input_format == sf_raw) { fprintf(stderr, "Assuming raw pcm input file"); if (swapbytes) fprintf(stderr, " : Forcing byte-swapping\n"); else fprintf(stderr, "\n"); } } if (lame_get_num_samples( gfp ) == MAX_U_32_NUM && musicin != stdin) { double flen = lame_get_file_size(inPath); /* try to figure out num_samples */ if (flen >= 0) { /* try file size, assume 2 bytes per sample */ if (input_format == sf_mp1 || input_format == sf_mp2 || input_format == sf_mp3) { if (mp3input_data.bitrate > 0) { double totalseconds = (flen * 8.0 / (1000.0 * mp3input_data.bitrate)); unsigned long tmp_num_samples = totalseconds * lame_get_in_samplerate( gfp ); (void) lame_set_num_samples( gfp, tmp_num_samples ); mp3input_data.nsamp = tmp_num_samples; } } else { (void) lame_set_num_samples( gfp, flen / (2 * lame_get_num_channels( gfp )) ); } } } return musicin; } #endif /* defined(LIBSNDFILE) */ #if defined(HAVE_MPGLIB) static int check_aid(const unsigned char *header) { return 0 == strncmp(header, "AiD\1", 4); } /* * Please check this and don't kill me if there's a bug * This is a (nearly?) complete header analysis for a MPEG-1/2/2.5 Layer I, II or III * data stream */ static int is_syncword_mp123(const void *const headerptr) { const unsigned char *const p = headerptr; static const char abl2[16] = { 0, 7, 7, 7, 0, 7, 0, 0, 0, 0, 0, 8, 8, 8, 8, 8 }; if ((p[0] & 0xFF) != 0xFF) return 0; // first 8 bits must be '1' if ((p[1] & 0xE0) != 0xE0) return 0; // next 3 bits are also if ((p[1] & 0x18) == 0x08) return 0; // no MPEG-1, -2 or -2.5 if ((p[1] & 0x06) == 0x00) return 0; // no Layer I, II and III if ((p[2] & 0xF0) == 0xF0) return 0; // bad bitrate if ((p[2] & 0x0C) == 0x0C) return 0; // no sample frequency with (32,44.1,48)/(1,2,4) if ((p[1] & 0x06) == 0x04) // illegal Layer II bitrate/Channel Mode comb if (abl2[p[2] >> 4] & (1 << (p[3] >> 6))) return 0; return 1; } static int is_syncword_mp3(const void *const headerptr) { const unsigned char *const p = headerptr; if ((p[0] & 0xFF) != 0xFF) return 0; // first 8 bits must be '1' if ((p[1] & 0xE0) != 0xE0) return 0; // next 3 bits are also if ((p[1] & 0x18) == 0x08) return 0; // no MPEG-1, -2 or -2.5 if ((p[1] & 0x06) != 0x02) return 0; // no Layer III (can be merged with 'next 3 bits are also' test, but don't do this, this decreases readability) if ((p[2] & 0xF0) == 0xF0) return 0; // bad bitrate if ((p[2] & 0x0C) == 0x0C) return 0; // no sample frequency with (32,44.1,48)/(1,2,4) return 1; } int lame_decode_initfile(FILE * fd, mp3data_struct * mp3data) { // VBRTAGDATA pTagData; // int xing_header,len2,num_frames; unsigned char buf[100]; int ret; int len, aid_header; short int pcm_l[1152], pcm_r[1152]; memset(mp3data, 0, sizeof(mp3data_struct)); lame_decode_init(); len = 4; if (fread(&buf, 1, len, fd) != len) return -1; /* failed */ aid_header = check_aid(buf); if (aid_header) { if (fread(&buf, 1, 2, fd) != 2) return -1; /* failed */ aid_header = (unsigned char) buf[0] + 256 * (unsigned char) buf[1]; fprintf(stderr, "Album ID found. length=%i \n", aid_header); /* skip rest of AID, except for 6 bytes we have already read */ fskip(fd, aid_header - 6, SEEK_CUR); /* read 4 more bytes to set up buffer for MP3 header check */ len = fread(&buf, 1, 4, fd); } /* look for valid 4 byte MPEG header */ if (len < 4) return -1; while (!is_syncword_mp123(buf)) { int i; for (i = 0; i < len - 1; i++) buf[i] = buf[i + 1]; if (fread(buf + len - 1, 1, 1, fd) != 1) return -1; /* failed */ } #if 0 /* buffer 48 bytes so we can check for Xing header */ len2 = fread(&buf[len], 1, 48 - len, fd); if (len2 != 48 - len) return -1; len = 48; /* check first 48 bytes for Xing header */ xing_header = GetVbrTag(&pTagData, (unsigned char *) buf); if (xing_header && pTagData.headersize >= 48) { num_frames = pTagData.frames; fprintf(stderr, "\rXing VBR header dectected. MP3 file has %i frames\n", num_frames); // skip the rest of the Xing header. LAME decoder ignores TOC data fskip(fd, pTagData.headersize - 48, SEEK_CUR); // buffer a few more bytes for next header check: len = fread(buf, 1, 4, fd); } else { /* we have read 48 bytes, but did not find a Xing header */ /* lets try and rewind the stream: */ if (fseek(fd, -44, SEEK_CUR) != 0) { /* backwards fseek failed. input is probably a pipe */ /* keep 'len' unchanged */ } else { len -= 44; } } #endif // now parse the current buffer looking for MP3 headers // we dont want to feed too much data to lame_decode1_headers - // we dont want it to actually decode the first frame // (as of 11/00: mpglib modified so that for the first frame where // headers are parsed, no data will be decoded. So the above is // now a moot point. ret = lame_decode1_headers(buf, len, pcm_l, pcm_r, mp3data); if (-1 == ret) return -1; /* repeat until we decode a valid mp3 header */ while (!mp3data->header_parsed) { len = fread(buf, 1, sizeof(buf), fd); if (len != sizeof(buf)) return -1; ret = lame_decode1_headers(buf, len, pcm_l, pcm_r, mp3data); if (-1 == ret) return -1; } #if 1 if (mp3data->totalframes > 0) { /* mpglib found a Xing VBR header and computed nsamp & totalframes */ } else { mp3data->nsamp = MAX_U_32_NUM; } #else mp3data->nsamp = MAX_U_32_NUM; if (xing_header && num_frames) { mp3data->nsamp = mp3data->framesize * num_frames; } #endif /* fprintf(stderr,"ret = %i NEED_MORE=%i \n",ret,MP3_NEED_MORE); fprintf(stderr,"stereo = %i \n",mp.fr.stereo); fprintf(stderr,"samp = %i \n",freqs[mp.fr.sampling_frequency]); fprintf(stderr,"framesize = %i \n",framesize); fprintf(stderr,"bitrate = %i \n",mp3data->bitrate); fprintf(stderr,"num frames = %ui \n",num_frames); fprintf(stderr,"num samp = %ui \n",mp3data->nsamp); fprintf(stderr,"mode = %i \n",mp.fr.mode); */ return 0; } /* For lame_decode_fromfile: return code -1 error 0 ok, but need more data before outputing any samples n number of samples output. either 576 or 1152 depending on MP3 file. */ int lame_decode_fromfile(FILE * fd, short pcm_l[], short pcm_r[], mp3data_struct * mp3data) { int ret = 0, len; unsigned char buf[100]; /* read until we get a valid output frame */ while (0 == ret) { len = fread(buf, 1, 100, fd); if (len != 100) return -1; ret = lame_decode1_headers(buf, len, pcm_l, pcm_r, mp3data); if (ret == -1) return -1; } return ret; } #endif /* defined(HAVE_MPGLIB) */ /* end of get_audio.c */