ref: 6e459618b12eddfe4791b7f43b2f58605352229b
dir: /src/effects.c/
/*
* SoX Effects chain
* (c) 2007 robs@users.sourceforge.net
*
* This library is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser 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 Lesser
* General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this library. If not, write to the Free Software Foundation,
* Fifth Floor, 51 Franklin Street, Boston, MA 02111-1301, USA.
*/
#include "sox_i.h"
#include <assert.h>
#include <string.h>
#ifdef HAVE_STRINGS_H
#include <strings.h>
#endif
#undef sox_fail
#undef sox_report
#define sox_fail sox_globals.subsystem=effp->handler.name,sox_fail
#define sox_report sox_globals.subsystem=effp->handler.name,sox_report
#define DEBUG_EFFECTS_CHAIN 0
sox_effects_globals_t sox_effects_globals =
{sox_plot_off, 1, &sox_globals};
int sox_usage(sox_effect_t * effp)
{
if (effp->handler.usage)
sox_fail("usage: %s", effp->handler.usage);
else
sox_fail("this effect takes no parameters");
return SOX_EOF;
}
/* Default effect handler functions for do-nothing situations: */
static int default_function(sox_effect_t * effp UNUSED)
{
return SOX_SUCCESS;
}
/* Pass through samples verbatim */
static int default_flow(sox_effect_t * effp UNUSED, const sox_sample_t *ibuf UNUSED, sox_sample_t *obuf UNUSED, sox_size_t *isamp, sox_size_t *osamp)
{
*isamp = *osamp = min(*isamp, *osamp);
memcpy(obuf, ibuf, *isamp * sizeof(*obuf));
return SOX_SUCCESS;
}
/* Inform no more samples to drain */
static int default_drain(sox_effect_t * effp UNUSED, sox_sample_t *obuf UNUSED, sox_size_t *osamp)
{
*osamp = 0;
return SOX_EOF;
}
/* Check that no parameters have been given */
static int default_getopts(sox_effect_t * effp, int argc, char **argv UNUSED)
{
return argc? sox_usage(effp) : SOX_SUCCESS;
}
/* Partially initialise the effect structure; signal info will come later */
void sox_create_effect(sox_effect_t * effp, sox_effect_handler_t const * eh)
{
assert(eh);
memset(effp, 0, sizeof(*effp));
effp->global_info = &sox_effects_globals;
effp->handler = *eh;
if (!effp->handler.getopts) effp->handler.getopts = default_getopts;
if (!effp->handler.start ) effp->handler.start = default_function;
if (!effp->handler.flow ) effp->handler.flow = default_flow;
if (!effp->handler.drain ) effp->handler.drain = default_drain;
if (!effp->handler.stop ) effp->handler.stop = default_function;
if (!effp->handler.kill ) effp->handler.kill = default_function;
}
/* Effects chain: */
sox_effects_chain_t * sox_create_effects_chain(void)
{
sox_effects_chain_t * result = xcalloc(1, sizeof(sox_effects_chain_t));
result->global_info = sox_effects_globals;
return result;
}
/* Effect can call in start() or flow() to set minimum input size to flow() */
int sox_effect_set_imin(sox_effect_t * effp, sox_size_t imin)
{
if (imin > sox_globals.bufsiz / effp->flows) {
sox_fail("sox_bufsiz not big enough");
return SOX_EOF;
}
effp->imin = imin;
return SOX_SUCCESS;
}
/* Add an effect to the chain. *in is the input signal for this effect. *out is
* a suggestion as to what the output signal should be, but depending on its
* given options and *in, the effect can choose to do differently. Whatever
* output rate and channels the effect does produce are written back to *in,
* ready for the next effect in the chain.
*/
int sox_add_effect(sox_effects_chain_t * chain, sox_effect_t * effp, sox_signalinfo_t * in, sox_signalinfo_t
const * out)
{
int ret, (*start)(sox_effect_t * effp) = effp->handler.start;
unsigned f;
sox_effect_t eff0; /* Copy of effect for flow 0 before calling start */
if (effp->handler.flags & SOX_EFF_NULL) {
sox_report("has no effect (is a proxy effect)");
return SOX_SUCCESS;
}
effp->global_info = &chain->global_info;
effp->ininfo = *in;
effp->outinfo = *out;
if (!(effp->handler.flags & SOX_EFF_CHAN))
effp->outinfo.channels = in->channels;
if (!(effp->handler.flags & SOX_EFF_RATE))
effp->outinfo.rate = in->rate;
if (!(effp->handler.flags & SOX_EFF_PREC))
effp->outinfo.size = in->size;
effp->flows =
(effp->handler.flags & SOX_EFF_MCHAN)? 1 : effp->ininfo.channels;
effp->clips = 0;
effp->imin = 0;
eff0 = *effp;
ret = start(effp);
if (ret == SOX_EFF_NULL) {
sox_report("has no effect in this configuration");
return SOX_SUCCESS;
}
if (ret != SOX_SUCCESS)
return SOX_EOF;
*in = effp->outinfo;
if (chain->length == SOX_MAX_EFFECTS) {
sox_fail("Too many effects!");
return SOX_EOF;
}
chain->effects[chain->length] =
xcalloc(effp->flows, sizeof(chain->effects[chain->length][0]));
chain->effects[chain->length][0] = *effp;
for (f = 1; f < effp->flows; ++f) {
chain->effects[chain->length][f] = eff0;
chain->effects[chain->length][f].flow = f;
if (start(&chain->effects[chain->length][f]) != SOX_SUCCESS)
return SOX_EOF;
}
++chain->length;
return SOX_SUCCESS;
}
static int flow_effect(sox_effects_chain_t * chain, unsigned n)
{
sox_effect_t * effp1 = &chain->effects[n - 1][0];
sox_effect_t * effp = &chain->effects[n][0];
int effstatus = SOX_SUCCESS;
sox_size_t i, f;
const sox_sample_t *ibuf;
sox_size_t idone = effp1->oend - effp1->obeg;
sox_size_t obeg = sox_globals.bufsiz - effp->oend;
#if DEBUG_EFFECTS_CHAIN
sox_size_t pre_idone = idone;
sox_size_t pre_odone = obeg;
#endif
if (effp->flows == 1) /* Run effect on all channels at once */
effstatus = effp->handler.flow(effp, &effp1->obuf[effp1->obeg],
&effp->obuf[effp->oend], &idone, &obeg);
else { /* Run effect on each channel individually */
sox_sample_t *obuf = &effp->obuf[effp->oend];
sox_size_t idone_last = 0, odone_last = 0; /* Initialised to prevent warning */
ibuf = &effp1->obuf[effp1->obeg];
for (i = 0; i < idone; i += effp->flows)
for (f = 0; f < effp->flows; ++f)
chain->ibufc[f][i / effp->flows] = *ibuf++;
for (f = 0; f < effp->flows; ++f) {
sox_size_t idonec = idone / effp->flows;
sox_size_t odonec = obeg / effp->flows;
int eff_status_c = effp->handler.flow(&chain->effects[n][f],
chain->ibufc[f], chain->obufc[f], &idonec, &odonec);
if (f && (idonec != idone_last || odonec != odone_last)) {
sox_fail("flowed asymmetrically!");
effstatus = SOX_EOF;
}
idone_last = idonec;
odone_last = odonec;
if (eff_status_c != SOX_SUCCESS)
effstatus = SOX_EOF;
}
for (i = 0; i < odone_last; ++i)
for (f = 0; f < effp->flows; ++f)
*obuf++ = chain->obufc[f][i];
idone = f * idone_last;
obeg = f * odone_last;
}
#if DEBUG_EFFECTS_CHAIN
sox_report("flow: %5u%5u%5u%5u", pre_idone, pre_odone, idone, obeg);
#endif
effp1->obeg += idone;
if (effp1->obeg == effp1->oend)
effp1->obeg = effp1->oend = 0;
else if (effp1->oend - effp1->obeg < effp->imin ) { /* Need to refill? */
memmove(effp1->obuf, &effp1->obuf[effp1->obeg], (effp1->oend - effp1->obeg) * sizeof(*effp1->obuf));
effp1->oend -= effp1->obeg;
effp1->obeg = 0;
}
effp->oend += obeg;
return effstatus == SOX_SUCCESS? SOX_SUCCESS : SOX_EOF;
}
/* The same as flow_effect but with no input */
static int drain_effect(sox_effects_chain_t * chain, unsigned n)
{
sox_effect_t * effp = &chain->effects[n][0];
int effstatus = SOX_SUCCESS;
sox_size_t i, f;
sox_size_t obeg = sox_globals.bufsiz - effp->oend;
#if DEBUG_EFFECTS_CHAIN
sox_size_t pre_odone = obeg;
#endif
if (effp->flows == 1) /* Run effect on all channels at once */
effstatus = effp->handler.drain(effp, &effp->obuf[effp->oend], &obeg);
else { /* Run effect on each channel individually */
sox_sample_t *obuf = &effp->obuf[effp->oend];
sox_size_t odone_last = 0; /* Initialised to prevent warning */
for (f = 0; f < effp->flows; ++f) {
sox_size_t odonec = obeg / effp->flows;
int eff_status_c = effp->handler.drain(&chain->effects[n][f], chain->obufc[f], &odonec);
if (f && (odonec != odone_last)) {
sox_fail("drained asymmetrically!");
effstatus = SOX_EOF;
}
odone_last = odonec;
if (eff_status_c != SOX_SUCCESS)
effstatus = SOX_EOF;
}
for (i = 0; i < odone_last; ++i)
for (f = 0; f < effp->flows; ++f)
*obuf++ = chain->obufc[f][i];
obeg = f * odone_last;
}
#if DEBUG_EFFECTS_CHAIN
sox_report("drain: %5u%5u%5u%5u", 0, pre_odone, 0, obeg);
#endif
if (!obeg) /* This is the only thing that drain has and flow hasn't */
effstatus = SOX_EOF;
effp->oend += obeg;
return effstatus == SOX_SUCCESS? SOX_SUCCESS : SOX_EOF;
}
/* Flow data through the effects chain until an effect or callback gives EOF */
int sox_flow_effects(sox_effects_chain_t * chain, int (* callback)(sox_bool all_done))
{
int flow_status = SOX_SUCCESS;
sox_size_t e, source_e = 0; /* effect indices */
sox_size_t f, max_flows = 0;
sox_bool draining = sox_true;
for (e = 0; e < chain->length; ++e) {
chain->effects[e][0].obuf = xmalloc(sox_globals.bufsiz * sizeof(chain->effects[e][0].obuf[0]));
chain->effects[e][0].obeg = chain->effects[e][0].oend = 0;
max_flows = max(max_flows, chain->effects[e][0].flows);
}
chain->ibufc = xcalloc(max_flows, sizeof(*chain->ibufc));
chain->obufc = xcalloc(max_flows, sizeof(*chain->obufc));
for (f = 0; f < max_flows; ++f) {
chain->ibufc[f] = xcalloc(sox_globals.bufsiz / 2, sizeof(chain->ibufc[f][0]));
chain->obufc[f] = xcalloc(sox_globals.bufsiz / 2, sizeof(chain->obufc[f][0]));
}
e = chain->length - 1;
while (source_e < chain->length) {
#define have_imin (e > 0 && e < chain->length && chain->effects[e - 1][0].oend - chain->effects[e - 1][0].obeg >= chain->effects[e][0].imin)
size_t osize = chain->effects[e][0].oend - chain->effects[e][0].obeg;
if (e == source_e && (draining || !have_imin)) {
if (drain_effect(chain, e) == SOX_EOF) {
++source_e;
draining = sox_false;
}
} else if (have_imin && flow_effect(chain, e) == SOX_EOF) {
flow_status = SOX_EOF;
source_e = e;
draining = sox_true;
}
if (e < chain->length && chain->effects[e][0].oend - chain->effects[e][0].obeg > osize) /* False for output */
++e;
else if (e == source_e)
draining = sox_true;
else if ((int)--e < (int)source_e)
e = source_e;
if (callback && callback(source_e == chain->length) != SOX_SUCCESS) {
flow_status = SOX_EOF; /* Client has requested to stop the flow. */
break;
}
}
for (f = 0; f < max_flows; ++f) {
free(chain->ibufc[f]);
free(chain->obufc[f]);
}
free(chain->obufc);
free(chain->ibufc);
for (e = 0; e < chain->length; ++e)
free(chain->effects[e][0].obuf);
return flow_status;
}
sox_size_t sox_effects_clips(sox_effects_chain_t * chain)
{
unsigned i, f;
sox_size_t clips = 0;
for (i = 1; i < chain->length - 1; ++i)
for (f = 0; f < chain->effects[i][0].flows; ++f)
clips += chain->effects[i][f].clips;
return clips;
}
sox_size_t sox_stop_effect(sox_effects_chain_t * chain, sox_size_t e)
{
unsigned f;
sox_effect_t * effp = &chain->effects[e][0];
sox_size_t clips = 0;
for (f = 0; f < effp->flows; ++f) {
effp->handler.stop(&chain->effects[e][f]);
clips += chain->effects[e][f].clips;
}
return clips;
}
/* Remove all effects from the chain */
void sox_delete_effects(sox_effects_chain_t * chain)
{
sox_size_t e, clips;
for (e = 0; e < chain->length; ++e) {
sox_effect_t * effp = chain->effects[e];
if ((clips = sox_stop_effect(chain, e)) != 0)
sox_warn("%s clipped %u samples; decrease volume?",
chain->effects[e][0].handler.name, clips);
effp->handler.kill(effp);
free(chain->effects[e]);
}
chain->length = 0;
}
/* Effects library: */
sox_effect_fn_t sox_effect_fns[] = {
#define EFFECT(f) sox_##f##_effect_fn,
#include "effects.h"
#undef EFFECT
NULL
};
/* Find a named effect in the effects library */
sox_effect_handler_t const * sox_find_effect(char const * name)
{
int e;
for (e = 0; sox_effect_fns[e]; ++e) {
const sox_effect_handler_t *eh = sox_effect_fns[e] ();
if (eh && eh->name && strcasecmp(eh->name, name) == 0)
return eh; /* Found it. */
}
return NULL;
}