gain.c

/*
 * This file is part of dsp.
 *
 * Copyright (c) 2013-2024 Michael Barbour <barbour.michael.0@gmail.com>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include "gain.h"
#include "util.h"

struct gain_state {
	int channel;
	sample_t v;
};

sample_t * gain_effect_run(struct effect *e, ssize_t *frames, sample_t *ibuf, sample_t *obuf)
{
	ssize_t i, k, samples = *frames * e->ostream.channels;
	struct gain_state *state = (struct gain_state *) e->data;
	if (state->channel == -1) {
		for (i = 0; i < samples; i += e->ostream.channels)
			for (k = 0; k < e->ostream.channels; ++k)
				if (GET_BIT(e->channel_selector, k))
					ibuf[i + k] *= state->v;
	}
	else {
		for (i = state->channel; i < samples; i += e->ostream.channels)
			ibuf[i] *= state->v;
	}
	return ibuf;
}

sample_t * add_effect_run(struct effect *e, ssize_t *frames, sample_t *ibuf, sample_t *obuf)
{
	ssize_t i, k, samples = *frames * e->ostream.channels;
	struct gain_state *state = (struct gain_state *) e->data;
	if (state->channel == -1) {
		for (i = 0; i < samples; i += e->ostream.channels)
			for (k = 0; k < e->ostream.channels; ++k)
				if (GET_BIT(e->channel_selector, k))
					ibuf[i + k] += state->v;
	}
	else {
		for (i = state->channel; i < samples; i += e->ostream.channels)
			ibuf[i] += state->v;
	}
	return ibuf;
}

void gain_effect_plot(struct effect *e, int i)
{
	struct gain_state *state = (struct gain_state *) e->data;
	for (int k = 0; k < e->ostream.channels; ++k) {
		if ((state->channel == -1 && GET_BIT(e->channel_selector, k)) || (state->channel != -1 && k == state->channel))
			printf("H%d_%d(w)=%.15e\n", k, i, state->v);
		else
			printf("H%d_%d(w)=1.0\n", k, i);
	}
}

void add_effect_plot(struct effect *e, int i)
{
	struct gain_state *state = (struct gain_state *) e->data;
	for (int k = 0; k < e->ostream.channels; ++k) {
		if ((state->channel == -1 && GET_BIT(e->channel_selector, k)) || (state->channel != -1 && k == state->channel))
			printf("H%d_%d(w)=(w==0.0)?1.0+%.15e:1.0\n", k, i, state->v);
		else
			printf("H%d_%d(w)=1.0\n", k, i);
	}
}

void gain_effect_destroy(struct effect *e)
{
	free(e->data);
	free(e->channel_selector);
}

struct effect * gain_effect_init(const struct effect_info *ei, const struct stream_info *istream, const char *channel_selector, const char *dir, int argc, const char *const *argv)
{
	struct effect *e;
	struct gain_state *state;
	double v;
	int channel = -1;
	char *endptr;

	if (argc != 2 && argc != 3) {
		LOG_FMT(LL_ERROR, "%s: usage: %s", argv[0], ei->usage);
		return NULL;
	}
	if (argc == 3) {
		channel = strtol(argv[1], &endptr, 10);
		CHECK_ENDPTR(argv[1], endptr, "channel", return NULL);
		CHECK_RANGE(channel >= 0 && channel < istream->channels, "channel", return NULL);
	}
	const char *arg = argv[argc - 1];

	switch (ei->effect_number) {
	case GAIN_EFFECT_NUMBER_GAIN:
		v = pow(10.0, strtod(arg, &endptr) / 20.0);
		CHECK_ENDPTR(arg, endptr, "gain", return NULL);
		break;
	case GAIN_EFFECT_NUMBER_MULT:
		v = strtod(arg, &endptr);
		CHECK_ENDPTR(arg, endptr, "multiplier", return NULL);
		break;
	case GAIN_EFFECT_NUMBER_ADD:
		v = strtod(arg, &endptr);
		CHECK_ENDPTR(arg, endptr, "value", return NULL);
		break;
	default:
		LOG_FMT(LL_ERROR, "%s: BUG: unknown effect: %s (%d)", __FILE__, argv[0], ei->effect_number);
		return NULL;
	}

	e = calloc(1, sizeof(struct effect));
	e->name = ei->name;
	e->istream.fs = e->ostream.fs = istream->fs;
	e->istream.channels = e->ostream.channels = istream->channels;
	e->channel_selector = NEW_SELECTOR(istream->channels);
	COPY_SELECTOR(e->channel_selector, channel_selector, istream->channels);
	e->run = (ei->effect_number == GAIN_EFFECT_NUMBER_ADD) ? add_effect_run : gain_effect_run;
	e->plot = (ei->effect_number == GAIN_EFFECT_NUMBER_ADD) ? add_effect_plot : gain_effect_plot;
	e->destroy = gain_effect_destroy;
	state = calloc(1, sizeof(struct gain_state));
	state->channel = channel;
	state->v = v;
	e->data = state;
	return e;
}