cannam@85: /* cannam@85: ** Copyright (C) 2002-2011 Erik de Castro Lopo cannam@85: ** cannam@85: ** This program is free software; you can redistribute it and/or modify cannam@85: ** it under the terms of the GNU General Public License as published by cannam@85: ** the Free Software Foundation; either version 2 of the License, or cannam@85: ** (at your option) any later version. cannam@85: ** cannam@85: ** This program is distributed in the hope that it will be useful, cannam@85: ** but WITHOUT ANY WARRANTY; without even the implied warranty of cannam@85: ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the cannam@85: ** GNU General Public License for more details. cannam@85: ** cannam@85: ** You should have received a copy of the GNU General Public License cannam@85: ** along with this program; if not, write to the Free Software cannam@85: ** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA. cannam@85: */ cannam@85: cannam@85: #include "config.h" cannam@85: cannam@85: #include cannam@85: #include cannam@85: #include cannam@85: #include cannam@85: #include cannam@85: cannam@85: #if (HAVE_SNDFILE) cannam@85: cannam@85: #include cannam@85: #include cannam@85: cannam@85: #define DEFAULT_CONVERTER SRC_SINC_MEDIUM_QUALITY cannam@85: cannam@85: #define BUFFER_LEN 4096 /*-(1<<16)-*/ cannam@85: cannam@85: static void usage_exit (const char *progname) ; cannam@85: static sf_count_t sample_rate_convert (SNDFILE *infile, SNDFILE *outfile, int converter, double src_ratio, int channels, double * gain) ; cannam@85: static double apply_gain (float * data, long frames, int channels, double max, double gain) ; cannam@85: cannam@85: int cannam@85: main (int argc, char *argv []) cannam@85: { SNDFILE *infile, *outfile = NULL ; cannam@85: SF_INFO sfinfo ; cannam@85: cannam@85: sf_count_t count ; cannam@85: double src_ratio = -1.0, gain = 1.0 ; cannam@85: int new_sample_rate = -1, k, converter, max_speed = SF_FALSE ; cannam@85: cannam@85: if (argc == 2 && strcmp (argv [1], "--version") == 0) cannam@85: { char buffer [64], *cptr ; cannam@85: cannam@85: if ((cptr = strrchr (argv [0], '/')) != NULL) cannam@85: argv [0] = cptr + 1 ; cannam@85: if ((cptr = strrchr (argv [0], '\\')) != NULL) cannam@85: argv [0] = cptr + 1 ; cannam@85: cannam@85: sf_command (NULL, SFC_GET_LIB_VERSION, buffer, sizeof (buffer)) ; cannam@85: cannam@85: printf ("%s (%s,%s)\n", argv [0], src_get_version (), buffer) ; cannam@85: exit (0) ; cannam@85: } ; cannam@85: cannam@85: if (argc != 5 && argc != 7 && argc != 8) cannam@85: usage_exit (argv [0]) ; cannam@85: cannam@85: /* Set default converter. */ cannam@85: converter = DEFAULT_CONVERTER ; cannam@85: cannam@85: for (k = 1 ; k < argc - 2 ; k++) cannam@85: { if (strcmp (argv [k], "--max-speed") == 0) cannam@85: max_speed = SF_TRUE ; cannam@85: else if (strcmp (argv [k], "-to") == 0) cannam@85: { k ++ ; cannam@85: new_sample_rate = atoi (argv [k]) ; cannam@85: } cannam@85: else if (strcmp (argv [k], "-by") == 0) cannam@85: { k ++ ; cannam@85: src_ratio = atof (argv [k]) ; cannam@85: } cannam@85: else if (strcmp (argv [k], "-c") == 0) cannam@85: { k ++ ; cannam@85: converter = atoi (argv [k]) ; cannam@85: } cannam@85: else cannam@85: usage_exit (argv [0]) ; cannam@85: } ; cannam@85: cannam@85: if (new_sample_rate <= 0 && src_ratio <= 0.0) cannam@85: usage_exit (argv [0]) ; cannam@85: cannam@85: if (src_get_name (converter) == NULL) cannam@85: { printf ("Error : bad converter number.\n") ; cannam@85: usage_exit (argv [0]) ; cannam@85: } ; cannam@85: cannam@85: if (strcmp (argv [argc - 2], argv [argc - 1]) == 0) cannam@85: { printf ("Error : input and output file names are the same.\n") ; cannam@85: exit (1) ; cannam@85: } ; cannam@85: cannam@85: if ((infile = sf_open (argv [argc - 2], SFM_READ, &sfinfo)) == NULL) cannam@85: { printf ("Error : Not able to open input file '%s'\n", argv [argc - 2]) ; cannam@85: exit (1) ; cannam@85: } ; cannam@85: cannam@85: printf ("Input File : %s\n", argv [argc - 2]) ; cannam@85: printf ("Sample Rate : %d\n", sfinfo.samplerate) ; cannam@85: printf ("Input Frames : %ld\n\n", (long) sfinfo.frames) ; cannam@85: cannam@85: if (new_sample_rate > 0) cannam@85: { src_ratio = (1.0 * new_sample_rate) / sfinfo.samplerate ; cannam@85: sfinfo.samplerate = new_sample_rate ; cannam@85: } cannam@85: else if (src_is_valid_ratio (src_ratio)) cannam@85: sfinfo.samplerate = (int) floor (sfinfo.samplerate * src_ratio) ; cannam@85: else cannam@85: { printf ("Not able to determine new sample rate. Exiting.\n") ; cannam@85: sf_close (infile) ; cannam@85: exit (1) ; cannam@85: } ; cannam@85: cannam@85: if (fabs (src_ratio - 1.0) < 1e-20) cannam@85: { printf ("Target samplerate and input samplerate are the same. Exiting.\n") ; cannam@85: sf_close (infile) ; cannam@85: exit (0) ; cannam@85: } ; cannam@85: cannam@85: printf ("SRC Ratio : %f\n", src_ratio) ; cannam@85: printf ("Converter : %s\n\n", src_get_name (converter)) ; cannam@85: cannam@85: if (src_is_valid_ratio (src_ratio) == 0) cannam@85: { printf ("Error : Sample rate change out of valid range.\n") ; cannam@85: sf_close (infile) ; cannam@85: exit (1) ; cannam@85: } ; cannam@85: cannam@85: /* Delete the output file length to zero if already exists. */ cannam@85: remove (argv [argc - 1]) ; cannam@85: cannam@85: printf ("Output file : %s\n", argv [argc - 1]) ; cannam@85: printf ("Sample Rate : %d\n", sfinfo.samplerate) ; cannam@85: cannam@85: do cannam@85: { sf_close (outfile) ; cannam@85: cannam@85: if ((outfile = sf_open (argv [argc - 1], SFM_WRITE, &sfinfo)) == NULL) cannam@85: { printf ("Error : Not able to open output file '%s'\n", argv [argc - 1]) ; cannam@85: sf_close (infile) ; cannam@85: exit (1) ; cannam@85: } ; cannam@85: cannam@85: if (max_speed) cannam@85: { /* This is mainly for the comparison program tests/src-evaluate.c */ cannam@85: sf_command (outfile, SFC_SET_ADD_PEAK_CHUNK, NULL, SF_FALSE) ; cannam@85: } cannam@85: else cannam@85: { /* Update the file header after every write. */ cannam@85: sf_command (outfile, SFC_SET_UPDATE_HEADER_AUTO, NULL, SF_TRUE) ; cannam@85: } ; cannam@85: cannam@85: sf_command (outfile, SFC_SET_CLIPPING, NULL, SF_TRUE) ; cannam@85: cannam@85: count = sample_rate_convert (infile, outfile, converter, src_ratio, sfinfo.channels, &gain) ; cannam@85: } cannam@85: while (count < 0) ; cannam@85: cannam@85: printf ("Output Frames : %ld\n\n", (long) count) ; cannam@85: cannam@85: sf_close (infile) ; cannam@85: sf_close (outfile) ; cannam@85: cannam@85: return 0 ; cannam@85: } /* main */ cannam@85: cannam@85: /*============================================================================== cannam@85: */ cannam@85: cannam@85: static sf_count_t cannam@85: sample_rate_convert (SNDFILE *infile, SNDFILE *outfile, int converter, double src_ratio, int channels, double * gain) cannam@85: { static float input [BUFFER_LEN] ; cannam@85: static float output [BUFFER_LEN] ; cannam@85: cannam@85: SRC_STATE *src_state ; cannam@85: SRC_DATA src_data ; cannam@85: int error ; cannam@85: double max = 0.0 ; cannam@85: sf_count_t output_count = 0 ; cannam@85: cannam@85: sf_seek (infile, 0, SEEK_SET) ; cannam@85: sf_seek (outfile, 0, SEEK_SET) ; cannam@85: cannam@85: /* Initialize the sample rate converter. */ cannam@85: if ((src_state = src_new (converter, channels, &error)) == NULL) cannam@85: { printf ("\n\nError : src_new() failed : %s.\n\n", src_strerror (error)) ; cannam@85: exit (1) ; cannam@85: } ; cannam@85: cannam@85: src_data.end_of_input = 0 ; /* Set this later. */ cannam@85: cannam@85: /* Start with zero to force load in while loop. */ cannam@85: src_data.input_frames = 0 ; cannam@85: src_data.data_in = input ; cannam@85: cannam@85: src_data.src_ratio = src_ratio ; cannam@85: cannam@85: src_data.data_out = output ; cannam@85: src_data.output_frames = BUFFER_LEN /channels ; cannam@85: cannam@85: while (1) cannam@85: { cannam@85: /* If the input buffer is empty, refill it. */ cannam@85: if (src_data.input_frames == 0) cannam@85: { src_data.input_frames = sf_readf_float (infile, input, BUFFER_LEN / channels) ; cannam@85: src_data.data_in = input ; cannam@85: cannam@85: /* The last read will not be a full buffer, so snd_of_input. */ cannam@85: if (src_data.input_frames < BUFFER_LEN / channels) cannam@85: src_data.end_of_input = SF_TRUE ; cannam@85: } ; cannam@85: cannam@85: if ((error = src_process (src_state, &src_data))) cannam@85: { printf ("\nError : %s\n", src_strerror (error)) ; cannam@85: exit (1) ; cannam@85: } ; cannam@85: cannam@85: /* Terminate if done. */ cannam@85: if (src_data.end_of_input && src_data.output_frames_gen == 0) cannam@85: break ; cannam@85: cannam@85: max = apply_gain (src_data.data_out, src_data.output_frames_gen, channels, max, *gain) ; cannam@85: cannam@85: /* Write output. */ cannam@85: sf_writef_float (outfile, output, src_data.output_frames_gen) ; cannam@85: output_count += src_data.output_frames_gen ; cannam@85: cannam@85: src_data.data_in += src_data.input_frames_used * channels ; cannam@85: src_data.input_frames -= src_data.input_frames_used ; cannam@85: } ; cannam@85: cannam@85: src_state = src_delete (src_state) ; cannam@85: cannam@85: if (max > 1.0) cannam@85: { *gain = 1.0 / max ; cannam@85: printf ("\nOutput has clipped. Restarting conversion to prevent clipping.\n\n") ; cannam@85: return -1 ; cannam@85: } ; cannam@85: cannam@85: return output_count ; cannam@85: } /* sample_rate_convert */ cannam@85: cannam@85: static double cannam@85: apply_gain (float * data, long frames, int channels, double max, double gain) cannam@85: { cannam@85: long k ; cannam@85: cannam@85: for (k = 0 ; k < frames * channels ; k++) cannam@85: { data [k] *= gain ; cannam@85: cannam@85: if (fabs (data [k]) > max) cannam@85: max = fabs (data [k]) ; cannam@85: } ; cannam@85: cannam@85: return max ; cannam@85: } /* apply_gain */ cannam@85: cannam@85: static void cannam@85: usage_exit (const char *progname) cannam@85: { char lsf_ver [128] ; cannam@85: const char *cptr ; cannam@85: int k ; cannam@85: cannam@85: if ((cptr = strrchr (progname, '/')) != NULL) cannam@85: progname = cptr + 1 ; cannam@85: cannam@85: if ((cptr = strrchr (progname, '\\')) != NULL) cannam@85: progname = cptr + 1 ; cannam@85: cannam@85: cannam@85: sf_command (NULL, SFC_GET_LIB_VERSION, lsf_ver, sizeof (lsf_ver)) ; cannam@85: cannam@85: printf ("\n" cannam@85: " A Sample Rate Converter using libsndfile for file I/O and Secret \n" cannam@85: " Rabbit Code (aka libsamplerate) for performing the conversion.\n" cannam@85: " It works on any file format supported by libsndfile with any \n" cannam@85: " number of channels (limited only by host memory).\n" cannam@85: "\n" cannam@85: " %s\n" cannam@85: " %s\n" cannam@85: "\n" cannam@85: " Usage : \n" cannam@85: " %s -to [-c ] \n" cannam@85: " %s -by [-c ] \n" cannam@85: "\n", src_get_version (), lsf_ver, progname, progname) ; cannam@85: cannam@85: puts ( cannam@85: " The optional -c argument allows the converter type to be chosen from\n" cannam@85: " the following list :" cannam@85: "\n" cannam@85: ) ; cannam@85: cannam@85: for (k = 0 ; (cptr = src_get_name (k)) != NULL ; k++) cannam@85: printf (" %d : %s%s\n", k, cptr, k == DEFAULT_CONVERTER ? " (default)" : "") ; cannam@85: cannam@85: puts ("") ; cannam@85: cannam@85: exit (1) ; cannam@85: } /* usage_exit */ cannam@85: cannam@85: /*============================================================================== cannam@85: */ cannam@85: cannam@85: #else /* (HAVE_SNFILE == 0) */ cannam@85: cannam@85: /* Alternative main function when libsndfile is not available. */ cannam@85: cannam@85: int cannam@85: main (void) cannam@85: { puts ( cannam@85: "\n" cannam@85: "****************************************************************\n" cannam@85: " This example program was compiled without libsndfile \n" cannam@85: " (http://www.mega-nerd.com/libsndfile/).\n" cannam@85: " It is therefore completely broken and non-functional.\n" cannam@85: "****************************************************************\n" cannam@85: "\n" cannam@85: ) ; cannam@85: cannam@85: return 0 ; cannam@85: } /* main */ cannam@85: cannam@85: #endif cannam@85: