libsamplerate/examples/sndfile-resample.c

   1 /*
   2 ** Copyright (c) 2002-2016, Erik de Castro Lopo <erikd@mega-nerd.com>
   3 ** All rights reserved.
   4 **
   5 ** This code is released under 2-clause BSD license. Please see the
   6 ** file at : https://github.com/erikd/libsamplerate/blob/master/COPYING
   7 */
   8
   9 #include "config.h"
  10
  11 #include <stdio.h>
  12 #include <stdlib.h>
  13 #include <unistd.h>
  14 #include <string.h>
  15 #include <math.h>
  16
  17 #if (HAVE_SNDFILE)
  18
  19 #include <samplerate.h>
  20 #include <sndfile.h>
  21
  22 #define DEFAULT_CONVERTER SRC_SINC_MEDIUM_QUALITY
  23
  24 #define BUFFER_LEN              4096    /*-(1<<16)-*/
  25
  26 static void usage_exit (const char *progname) ;
  27 static sf_count_t sample_rate_convert (SNDFILE *infile, SNDFILE *outfile, int converter, double src_ratio, int channels, double * gain, int normalize) ;
  28 static double apply_gain (float * data, long frames, int channels, double max, double gain) ;
  29
  30 int
  31 main (int argc, char *argv [])
  32 {       SNDFILE *infile, *outfile = NULL ;
  33         SF_INFO sfinfo ;
  34
  35         int normalize = 1 ;
  36         sf_count_t      count ;
  37         double          src_ratio = -1.0, gain = 1.0 ;
  38         int                     new_sample_rate = -1, k, converter, max_speed = SF_FALSE ;
  39
  40         if (argc == 2 && strcmp (argv [1], "--version") == 0)
  41         {       char buffer [64], *cptr ;
  42
  43                 if ((cptr = strrchr (argv [0], '/')) != NULL)
  44                         argv [0] = cptr + 1 ;
  45                 if ((cptr = strrchr (argv [0], '\\')) != NULL)
  46                         argv [0] = cptr + 1 ;
  47
  48                 sf_command (NULL, SFC_GET_LIB_VERSION, buffer, sizeof (buffer)) ;
  49
  50                 printf ("%s (%s,%s)\n", argv [0], src_get_version (), buffer) ;
  51                 exit (0) ;
  52                 } ;
  53
  54         if (argc != 5 && argc != 7 && argc != 8)
  55                 usage_exit (argv [0]) ;
  56
  57         /* Set default converter. */
  58         converter = DEFAULT_CONVERTER ;
  59
  60         for (k = 1 ; k < argc - 2 ; k++)
  61         {       if (strcmp (argv [k], "--max-speed") == 0)
  62                         max_speed = SF_TRUE ;
  63                 else if (strcmp (argv [k], "--no-normalize") == 0)
  64                         normalize = 0 ;
  65                 else if (strcmp (argv [k], "-to") == 0)
  66                 {       k ++ ;
  67                         new_sample_rate = atoi (argv [k]) ;
  68                         }
  69                 else if (strcmp (argv [k], "-by") == 0)
  70                 {       k ++ ;
  71                         src_ratio = atof (argv [k]) ;
  72                         }
  73                 else if (strcmp (argv [k], "-c") == 0)
  74                 {       k ++ ;
  75                         converter = atoi (argv [k]) ;
  76                         }
  77                 else
  78                         usage_exit (argv [0]) ;
  79                 } ;
  80
  81         if (new_sample_rate <= 0 && src_ratio <= 0.0)
  82                 usage_exit (argv [0]) ;
  83
  84         if (src_get_name (converter) == NULL)
  85         {       printf ("Error : bad converter number.\n") ;
  86                 usage_exit (argv [0]) ;
  87                 } ;
  88
  89         if (strcmp (argv [argc - 2], argv [argc - 1]) == 0)
  90         {       printf ("Error : input and output file names are the same.\n") ;
  91                 exit (1) ;
  92                 } ;
  93
  94         if ((infile = sf_open (argv [argc - 2], SFM_READ, &sfinfo)) == NULL)
  95         {       printf ("Error : Not able to open input file '%s'\n", argv [argc - 2]) ;
  96                 exit (1) ;
  97                 } ;
  98
  99         printf ("Input File    : %s\n", argv [argc - 2]) ;
 100         printf ("Sample Rate   : %d\n", sfinfo.samplerate) ;
 101         printf ("Input Frames  : %ld\n\n", (long) sfinfo.frames) ;
 102
 103         if (new_sample_rate > 0)
 104         {       src_ratio = (1.0 * new_sample_rate) / sfinfo.samplerate ;
 105                 sfinfo.samplerate = new_sample_rate ;
 106                 }
 107         else if (src_is_valid_ratio (src_ratio))
 108                 sfinfo.samplerate = (int) floor (sfinfo.samplerate * src_ratio) ;
 109         else
 110         {       printf ("Not able to determine new sample rate. Exiting.\n") ;
 111                 sf_close (infile) ;
 112                 exit (1) ;
 113                 } ;
 114
 115         if (fabs (src_ratio - 1.0) < 1e-20)
 116         {       printf ("Target samplerate and input samplerate are the same. Exiting.\n") ;
 117                 sf_close (infile) ;
 118                 exit (0) ;
 119                 } ;
 120
 121         printf ("SRC Ratio     : %f\n", src_ratio) ;
 122         printf ("Converter     : %s\n\n", src_get_name (converter)) ;
 123
 124         if (src_is_valid_ratio (src_ratio) == 0)
 125         {       printf ("Error : Sample rate change out of valid range.\n") ;
 126                 sf_close (infile) ;
 127                 exit (1) ;
 128                 } ;
 129
 130         /* Delete the output file length to zero if already exists. */
 131         remove (argv [argc - 1]) ;
 132
 133         printf ("Output file   : %s\n", argv [argc - 1]) ;
 134         printf ("Sample Rate   : %d\n", sfinfo.samplerate) ;
 135
 136         do
 137         {       sf_close (outfile) ;
 138
 139                 if ((outfile = sf_open (argv [argc - 1], SFM_WRITE, &sfinfo)) == NULL)
 140                 {       printf ("Error : Not able to open output file '%s'\n", argv [argc - 1]) ;
 141                         sf_close (infile) ;
 142                         exit (1) ;
 143                         } ;
 144
 145                 if (max_speed)
 146                 {       /* This is mainly for the comparison program tests/src-evaluate.c */
 147                         sf_command (outfile, SFC_SET_ADD_PEAK_CHUNK, NULL, SF_FALSE) ;
 148                         }
 149                 else
 150                 {       /* Update the file header after every write. */
 151                         sf_command (outfile, SFC_SET_UPDATE_HEADER_AUTO, NULL, SF_TRUE) ;
 152                         } ;
 153
 154                 sf_command (outfile, SFC_SET_CLIPPING, NULL, SF_TRUE) ;
 155
 156                 count = sample_rate_convert (infile, outfile, converter, src_ratio, sfinfo.channels, &gain, normalize) ;
 157                 }
 158         while (count < 0) ;
 159
 160         printf ("Output Frames : %ld\n\n", (long) count) ;
 161
 162         sf_close (infile) ;
 163         sf_close (outfile) ;
 164
 165         return 0 ;
 166 } /* main */
 167
 168 /*==============================================================================
 169 */
 170
 171 static sf_count_t
 172 sample_rate_convert (SNDFILE *infile, SNDFILE *outfile, int converter, double src_ratio, int channels, double * gain, int normalize)
 173 {       static float input [BUFFER_LEN] ;
 174         static float output [BUFFER_LEN] ;
 175
 176         SRC_STATE       *src_state ;
 177         SRC_DATA        src_data ;
 178         int                     error ;
 179         double          max = 0.0 ;
 180         sf_count_t      output_count = 0 ;
 181
 182         sf_seek (infile, 0, SEEK_SET) ;
 183         sf_seek (outfile, 0, SEEK_SET) ;
 184
 185         /* Initialize the sample rate converter. */
 186         if ((src_state = src_new (converter, channels, &error)) == NULL)
 187         {       printf ("\n\nError : src_new() failed : %s.\n\n", src_strerror (error)) ;
 188                 exit (1) ;
 189                 } ;
 190
 191         src_data.end_of_input = 0 ; /* Set this later. */
 192
 193         /* Start with zero to force load in while loop. */
 194         src_data.input_frames = 0 ;
 195         src_data.data_in = input ;
 196
 197         src_data.src_ratio = src_ratio ;
 198
 199         src_data.data_out = output ;
 200         src_data.output_frames = BUFFER_LEN /channels ;
 201
 202         while (1)
 203         {
 204                 /* If the input buffer is empty, refill it. */
 205                 if (src_data.input_frames == 0)
 206                 {       src_data.input_frames = sf_readf_float (infile, input, BUFFER_LEN / channels) ;
 207                         src_data.data_in = input ;
 208
 209                         /* The last read will not be a full buffer, so snd_of_input. */
 210                         if (src_data.input_frames < BUFFER_LEN / channels)
 211                                 src_data.end_of_input = SF_TRUE ;
 212                         } ;
 213
 214                 if ((error = src_process (src_state, &src_data)))
 215                 {       printf ("\nError : %s\n", src_strerror (error)) ;
 216                         exit (1) ;
 217                         } ;
 218
 219                 /* Terminate if done. */
 220                 if (src_data.end_of_input && src_data.output_frames_gen == 0)
 221                         break ;
 222
 223                 max = apply_gain (src_data.data_out, src_data.output_frames_gen, channels, max, *gain) ;
 224
 225                 /* Write output. */
 226                 sf_writef_float (outfile, output, src_data.output_frames_gen) ;
 227                 output_count += src_data.output_frames_gen ;
 228
 229                 src_data.data_in += src_data.input_frames_used * channels ;
 230                 src_data.input_frames -= src_data.input_frames_used ;
 231                 } ;
 232
 233         src_delete (src_state) ;
 234
 235         if (normalize && max > 1.0)
 236         {       *gain = 1.0 / max ;
 237                 printf ("\nOutput has clipped. Restarting conversion to prevent clipping.\n\n") ;
 238                 return -1 ;
 239                 } ;
 240
 241         return output_count ;
 242 } /* sample_rate_convert */
 243
 244 static double
 245 apply_gain (float * data, long frames, int channels, double max, double gain)
 246 {
 247         long k ;
 248
 249         for (k = 0 ; k < frames * channels ; k++)
 250         {       data [k] *= gain ;
 251
 252                 if (fabs (data [k]) > max)
 253                         max = fabs (data [k]) ;
 254                 } ;
 255
 256         return max ;
 257 } /* apply_gain */
 258
 259 static void
 260 usage_exit (const char *progname)
 261 {       char lsf_ver [128] ;
 262         const char      *cptr ;
 263         int             k ;
 264
 265         if ((cptr = strrchr (progname, '/')) != NULL)
 266                 progname = cptr + 1 ;
 267
 268         if ((cptr = strrchr (progname, '\\')) != NULL)
 269                 progname = cptr + 1 ;
 270
 271
 272         sf_command (NULL, SFC_GET_LIB_VERSION, lsf_ver, sizeof (lsf_ver)) ;
 273
 274         printf ("\n"
 275                 "  A Sample Rate Converter using libsndfile for file I/O and Secret \n"
 276                 "  Rabbit Code (aka libsamplerate) for performing the conversion.\n"
 277                 "  It works on any file format supported by libsndfile with any \n"
 278                 "  number of channels (limited only by host memory).\n"
 279                 "\n"
 280                 "       %s\n"
 281                 "       %s\n"
 282                 "\n"
 283                 "  Usage : \n"
 284                 "       %s -to <new sample rate> [-c <number>] <input file> <output file>\n"
 285                 "       %s -by <amount> [-c <number>] <input file> <output file>\n"
 286                 "\n", src_get_version (), lsf_ver, progname, progname) ;
 287
 288         puts (
 289                 "  The optional -c argument allows the converter type to be chosen from\n"
 290                 "  the following list :"
 291                 "\n"
 292                 ) ;
 293
 294         for (k = 0 ; (cptr = src_get_name (k)) != NULL ; k++)
 295                 printf ("       %d : %s%s\n", k, cptr, k == DEFAULT_CONVERTER ? " (default)" : "") ;
 296
 297         puts ("\n"
 298                 "  The --no-normalize option disables clipping check and normalization.") ;
 299
 300         puts ("") ;
 301
 302         exit (1) ;
 303 } /* usage_exit */
 304
 305 /*==============================================================================
 306 */
 307
 308 #else /* (HAVE_SNFILE == 0) */
 309
 310 /* Alternative main function when libsndfile is not available. */
 311
 312 int
 313 main (void)
 314 {       puts (
 315                 "\n"
 316                 "****************************************************************\n"
 317                 "  This example program was compiled without libsndfile \n"
 318                 "  (http://www.mega-nerd.com/libsndfile/).\n"
 319                 "  It is therefore completely broken and non-functional.\n"
 320                 "****************************************************************\n"
 321                 "\n"
 322                 ) ;
 323
 324         return 0 ;
 325 } /* main */
 326
 327 #endif
 328