shithub: sox

ref: 678e9696e73fcc7502fa443364ce1d8f88cb0944
dir: /src/util.c/

View raw version
/*
 * util.c.
 * Utility functions mostly related to logging.
 *
 * July 5, 1991
 * Copyright 1991 Lance Norskog And Sundry Contributors
 * This source code is freely redistributable and may be used for
 * any purpose.  This copyright notice must be maintained.
 * Lance Norskog And Sundry Contributors are not responsible for
 * the consequences of using this software.
 */

#include "sox_i.h"
#include <assert.h>
#include <stddef.h>
#include <string.h>
#include <ctype.h>
#include <stdarg.h>

sox_size_t sox_bufsiz = 8192;
sox_output_message_handler_t sox_output_message_handler = NULL;
unsigned sox_output_verbosity_level = 2;

sox_global_info_t sox_global_info;

sox_effects_global_info_t effects_global_info =
    {sox_false, 1, &sox_global_info};

void sox_output_message(FILE *file, const char *filename, const char *fmt, va_list ap)
{
  char buffer[10];
  char const * handler_name;
  char const * dot_pos;
 
  handler_name = strrchr(filename, '/');
  if (handler_name != NULL) {
    ++handler_name;
  } else {
    handler_name = strrchr(filename, '\\');
    if (handler_name != NULL)
      ++handler_name;
    else
      handler_name = filename;
  }

  dot_pos = strrchr(handler_name, '.');
  if (dot_pos != NULL && dot_pos - handler_name <= (ptrdiff_t)(sizeof(buffer) - 1)) {
    strncpy(buffer, handler_name, (size_t)(dot_pos - handler_name));
    buffer[dot_pos - handler_name] = '\0';
    handler_name = buffer;
  }

  fprintf(file, "%s: ", handler_name);
  vfprintf(file, fmt, ap);
}



/* This is a bit of a hack.  It's useful to have libSoX
 * report which format or effect handler is outputing
 * the message.  Using the filename for this purpose is only an
 * approximation, but it saves a lot of work. ;)
 */
char const * sox_message_filename = 0;



static void sox_emit_message(unsigned level, char const *fmt, va_list ap)
{
  if (sox_output_message_handler != NULL)
    (*sox_output_message_handler)(level, sox_message_filename, fmt, ap);
}



#undef sox_fail
#undef sox_warn
#undef sox_report
#undef sox_debug
#undef sox_debug_more
#undef sox_debug_most

#define SOX_MESSAGE_FUNCTION(name,level) \
void name(char const * fmt, ...) \
{ \
  va_list args; \
\
  va_start(args, fmt); \
  sox_emit_message(level, fmt, args); \
  va_end(args); \
}

SOX_MESSAGE_FUNCTION(sox_fail  , 1)
SOX_MESSAGE_FUNCTION(sox_warn  , 2)
SOX_MESSAGE_FUNCTION(sox_report, 3)
SOX_MESSAGE_FUNCTION(sox_debug , 4)
SOX_MESSAGE_FUNCTION(sox_debug_more , 5)
SOX_MESSAGE_FUNCTION(sox_debug_most , 6)

#undef SOX_MESSAGE_FUNCTION

void sox_fail_errno(ft_t ft, int sox_errno, const char *fmt, ...)
{
        va_list args;

        ft->sox_errno = sox_errno;

        va_start(args, fmt);
#ifdef HAVE_VSNPRINTF
        vsnprintf(ft->sox_errstr, sizeof(ft->sox_errstr), fmt, args);
#else
        vsprintf(ft->sox_errstr, fmt, args);
#endif
        va_end(args);
        ft->sox_errstr[255] = '\0';
}

/*
 * Check that we have a known format suffix string.
 */
int sox_gettype(ft_t formp, sox_bool is_file_extension)
{
    const char * const *list;
    unsigned i;

    if (!formp->filetype) {
        sox_fail_errno(formp, SOX_EFMT, "Filetype was not specified");
        return SOX_EFMT;
    }
    for (i = 0; i < sox_formats; i++) {
      const sox_format_t *f = sox_format_fns[i].fn();
      if (is_file_extension && (f->flags & SOX_FILE_DEVICE))
        continue; /* don't match device name in file name extensions */
      for (list = f->names; *list; list++) {
        const char *s1 = *list, *s2 = formp->filetype;
        if (!strcasecmp(s1, s2))
          break;  /* not a match */
      }
      if (!*list)
        continue;
      /* Found it! */
      formp->h = f;
      return SOX_SUCCESS;
    }
    sox_fail_errno(formp, SOX_EFMT, "File type `%s' is not known",
                  formp->filetype);
    return SOX_EFMT;
}

sox_effect_handler_t const * sox_find_effect(char const * name)
{
  int i;

  for (i = 0; sox_effect_fns[i]; ++i) {
    const sox_effect_handler_t *e = sox_effect_fns[i] ();
    if (e && e->name && strcasecmp(e->name, name) == 0)
      return e;                 /* Found it. */
  }
  return NULL;
}

/* dummy effect routine for do-nothing functions */
static int effect_nothing(sox_effect_t * effp UNUSED)
{
  return SOX_SUCCESS;
}

static int effect_nothing_flow(sox_effect_t * effp UNUSED, const sox_ssample_t *ibuf UNUSED, sox_ssample_t *obuf UNUSED, sox_size_t *isamp, sox_size_t *osamp)
{
  /* Pass through samples verbatim */
  *isamp = *osamp = min(*isamp, *osamp);
  memcpy(obuf, ibuf, *isamp * sizeof(sox_ssample_t));
  return SOX_SUCCESS;
}

static int effect_nothing_drain(sox_effect_t * effp UNUSED, sox_ssample_t *obuf UNUSED, sox_size_t *osamp)
{
  /* Inform no more samples to drain */
  *osamp = 0;
  return SOX_EOF;
}

static int effect_nothing_getopts(sox_effect_t * effp, int n, char **argv UNUSED)
{
#undef sox_fail
#define sox_fail sox_message_filename=effp->handler.name,sox_fail
  if (n) {
    sox_fail(effp->handler.usage);
    return (SOX_EOF);
  }
  return (SOX_SUCCESS);
}

void sox_create_effect(sox_effect_t * effp, sox_effect_handler_t const * e)
{
  assert(e);
  memset(effp, 0, sizeof(*effp));
  effp->global_info = &effects_global_info;
  effp->handler = *e;
  if (!effp->handler.getopts) effp->handler.getopts = effect_nothing_getopts;
  if (!effp->handler.start) effp->handler.start = effect_nothing;
  if (!effp->handler.flow) effp->handler.flow = effect_nothing_flow;
  if (!effp->handler.drain) effp->handler.drain = effect_nothing_drain;
  if (!effp->handler.stop) effp->handler.stop = effect_nothing;
  if (!effp->handler.kill) effp->handler.kill = effect_nothing;
}

/*
 * Copy input and output signal info into effect structures.
 * Must pass in a bitmask containing info on whether SOX_EFF_CHAN
 * or SOX_EFF_RATE has been used previously on this effect stream.
 * If not running multiple effects then just pass in a value of 0.
 *
 * Return value is the same mask plus addition of SOX_EFF_CHAN or
 * SOX_EFF_RATE if it was used in this effect.  That make this
 * return value can be passed back into this function in future
 * calls.
 */

int sox_update_effect(sox_effect_t * effp, const sox_signalinfo_t *in, const sox_signalinfo_t *out, 
                    int effect_mask)
{
    effp->ininfo = *in;
    effp->outinfo = *out;

    if (in->channels != out->channels) {
      /* Only effects with SOX_EFF_CHAN flag can actually handle
       * outputing a different number of channels then the input.
       */
      if (!(effp->handler.flags & SOX_EFF_CHAN)) {
        /* If this effect is being run before a SOX_EFF_CHAN effect
         * then its output is the same as the input file; otherwise,
         * its input contains the same number of channels as the
         * output file. */
        if (effect_mask & SOX_EFF_CHAN)
          effp->ininfo.channels = out->channels;
        else
          effp->outinfo.channels = in->channels;
      }
    }

    if (in->rate != out->rate)
    {
        /* Only SOX_EFF_RATE effects can handle an input that
         * has a different sample rate from the output. */
        if (!(effp->handler.flags & SOX_EFF_RATE))
        {
            if (effect_mask & SOX_EFF_RATE)
                effp->ininfo.rate = out->rate;
            else
                effp->outinfo.rate = in->rate;
        }
    }

    if (effp->handler.flags & SOX_EFF_CHAN)
        effect_mask |= SOX_EFF_CHAN;
    if (effp->handler.flags & SOX_EFF_RATE)
        effect_mask |= SOX_EFF_RATE;

    return effect_mask;
}

/*
 * sox_parsesamples
 *
 * Parse a string for # of samples.  If string ends with a 's'
 * then the string is interpreted as a user calculated # of samples.
 * If string contains ':' or '.' or if it ends with a 't' then its
 * treated as an amount of time.  This is converted into seconds and
 * fraction of seconds and then use the sample rate to calculate
 * # of samples.
 * Returns NULL on error, pointer to next char to parse otherwise.
 */
char const * sox_parsesamples(sox_rate_t rate, const char *str, sox_size_t *samples, int def)
{
    int found_samples = 0, found_time = 0;
    int time = 0;
    long long_samples;
    float frac = 0;
    char const * end;
    char const * pos;
    sox_bool found_colon, found_dot;

    for (end = str; *end && strchr("0123456789:.ts", *end); ++end);
    if (end == str)
      return NULL;

    pos = strchr(str, ':');
    found_colon = pos && pos < end;
    
    pos = strchr(str, '.');
    found_dot = pos && pos < end;

    if (found_colon || found_dot || *(end-1) == 't')
        found_time = 1;
    else if (*(end-1) == 's')
        found_samples = 1;

    if (found_time || (def == 't' && !found_samples))
    {
        *samples = 0;

        while(1)
        {
            if (str[0] != '.' && sscanf(str, "%d", &time) != 1)
                return NULL;
            *samples += time;

            while (*str != ':' && *str != '.' && *str != 0)
                str++;

            if (*str == '.' || *str == 0)
                break;

            /* Skip past ':' */
            str++;
            *samples *= 60;
        }

        if (*str == '.')
        {
            if (sscanf(str, "%f", &frac) != 1)
                return NULL;
        }

        *samples *= rate;
        *samples += (rate * frac) + 0.5;
        return end;
    }
    if (found_samples || (def == 's' && !found_time))
    {
        if (sscanf(str, "%ld", &long_samples) != 1)
            return NULL;
        *samples = long_samples;
        return end;
    }
    return NULL;
}