2 * Copyright (c) 2008-2009 Ariff Abdullah <ariff@FreeBSD.org>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 * feeder_eq: Parametric (compile time) Software Equalizer. Though accidental,
29 * it proves good enough for educational and general consumption.
31 * "Cookbook formulae for audio EQ biquad filter coefficients"
32 * by Robert Bristow-Johnson <rbj@audioimagination.com>
33 * - http://www.musicdsp.org/files/Audio-EQ-Cookbook.txt
37 #ifdef HAVE_KERNEL_OPTION_HEADERS
40 #include <dev/sound/pcm/sound.h>
41 #include <dev/sound/pcm/pcm.h>
42 #include "feeder_if.h"
45 #include "snd_fxdiv_gen.h"
47 SND_DECLARE_FILE("$FreeBSD$");
50 #include "feeder_eq_gen.h"
52 #define FEEDEQ_LEVELS \
53 (((FEEDEQ_GAIN_MAX - FEEDEQ_GAIN_MIN) * \
54 (FEEDEQ_GAIN_DIV / FEEDEQ_GAIN_STEP)) + 1)
56 #define FEEDEQ_L2GAIN(v) \
57 ((int)min(((v) * FEEDEQ_LEVELS) / 100, FEEDEQ_LEVELS - 1))
59 #define FEEDEQ_PREAMP_IPART(x) (abs(x) >> FEEDEQ_GAIN_SHIFT)
60 #define FEEDEQ_PREAMP_FPART(x) (abs(x) & FEEDEQ_GAIN_FMASK)
61 #define FEEDEQ_PREAMP_SIGNVAL(x) ((x) < 0 ? -1 : 1)
62 #define FEEDEQ_PREAMP_SIGNMARK(x) (((x) < 0) ? '-' : '+')
64 #define FEEDEQ_PREAMP_IMIN -192
65 #define FEEDEQ_PREAMP_IMAX 192
66 #define FEEDEQ_PREAMP_FMIN 0
67 #define FEEDEQ_PREAMP_FMAX 9
69 #define FEEDEQ_PREAMP_INVALID INT_MAX
71 #define FEEDEQ_IF2PREAMP(i, f) \
72 ((abs(i) << FEEDEQ_GAIN_SHIFT) | \
73 (((abs(f) / FEEDEQ_GAIN_STEP) * FEEDEQ_GAIN_STEP) & \
76 #define FEEDEQ_PREAMP_MIN \
77 (FEEDEQ_PREAMP_SIGNVAL(FEEDEQ_GAIN_MIN) * \
78 FEEDEQ_IF2PREAMP(FEEDEQ_GAIN_MIN, 0))
80 #define FEEDEQ_PREAMP_MAX \
81 (FEEDEQ_PREAMP_SIGNVAL(FEEDEQ_GAIN_MAX) * \
82 FEEDEQ_IF2PREAMP(FEEDEQ_GAIN_MAX, 0))
84 #define FEEDEQ_PREAMP_DEFAULT FEEDEQ_IF2PREAMP(0, 0)
86 #define FEEDEQ_PREAMP2IDX(v) \
87 ((int32_t)((FEEDEQ_GAIN_MAX * (FEEDEQ_GAIN_DIV / \
88 FEEDEQ_GAIN_STEP)) + (FEEDEQ_PREAMP_SIGNVAL(v) * \
89 FEEDEQ_PREAMP_IPART(v) * (FEEDEQ_GAIN_DIV / \
90 FEEDEQ_GAIN_STEP)) + (FEEDEQ_PREAMP_SIGNVAL(v) * \
91 (FEEDEQ_PREAMP_FPART(v) / FEEDEQ_GAIN_STEP))))
93 static int feeder_eq_exact_rate = 0;
96 static char feeder_eq_presets[] = FEEDER_EQ_PRESETS;
97 SYSCTL_STRING(_hw_snd, OID_AUTO, feeder_eq_presets, CTLFLAG_RD,
98 &feeder_eq_presets, 0, "compile-time eq presets");
100 SYSCTL_INT(_hw_snd, OID_AUTO, feeder_eq_exact_rate, CTLFLAG_RWTUN,
101 &feeder_eq_exact_rate, 0, "force exact rate validation");
106 typedef void (*feed_eq_t)(struct feed_eq_info *, uint8_t *, uint32_t);
108 struct feed_eq_tone {
109 intpcm_t o1[SND_CHN_MAX];
110 intpcm_t o2[SND_CHN_MAX];
111 intpcm_t i1[SND_CHN_MAX];
112 intpcm_t i2[SND_CHN_MAX];
116 struct feed_eq_info {
117 struct feed_eq_tone treble;
118 struct feed_eq_tone bass;
119 struct feed_eq_coeff *coeff;
128 #if !defined(_KERNEL) && defined(FEEDEQ_ERR_CLIP)
129 #define FEEDEQ_ERR_CLIP_CHECK(t, v) do { \
130 if ((v) < PCM_S32_MIN || (v) > PCM_S32_MAX) \
131 errx(1, "\n\n%s(): ["#t"] Sample clipping: %jd\n", \
132 __func__, (intmax_t)(v)); \
135 #define FEEDEQ_ERR_CLIP_CHECK(...)
138 #define FEEDEQ_CLAMP(v) (((v) > PCM_S32_MAX) ? PCM_S32_MAX : \
139 (((v) < PCM_S32_MIN) ? PCM_S32_MIN : \
142 #define FEEDEQ_DECLARE(SIGN, BIT, ENDIAN) \
144 feed_eq_biquad_##SIGN##BIT##ENDIAN(struct feed_eq_info *info, \
145 uint8_t *dst, uint32_t count) \
147 struct feed_eq_coeff_tone *treble, *bass; \
151 int32_t pmul, pshift; \
153 pmul = feed_eq_preamp[info->preamp].mul; \
154 pshift = feed_eq_preamp[info->preamp].shift; \
156 if (info->state == FEEDEQ_DISABLE) { \
157 j = count * info->channels; \
158 dst += j * PCM_##BIT##_BPS; \
160 dst -= PCM_##BIT##_BPS; \
161 v = _PCM_READ_##SIGN##BIT##_##ENDIAN(dst); \
162 v = ((intpcm64_t)pmul * v) >> pshift; \
163 _PCM_WRITE_##SIGN##BIT##_##ENDIAN(dst, v); \
164 } while (--j != 0); \
169 treble = &(info->coeff[info->treble.gain].treble); \
170 bass = &(info->coeff[info->bass.gain].bass); \
174 j = info->channels; \
176 v = _PCM_READ_##SIGN##BIT##_##ENDIAN(dst); \
178 v = ((intpcm64_t)pmul * v) >> pshift; \
180 w = (intpcm64_t)v * treble->b0; \
181 w += (intpcm64_t)info->treble.i1[i] * treble->b1; \
182 w += (intpcm64_t)info->treble.i2[i] * treble->b2; \
183 w -= (intpcm64_t)info->treble.o1[i] * treble->a1; \
184 w -= (intpcm64_t)info->treble.o2[i] * treble->a2; \
185 info->treble.i2[i] = info->treble.i1[i]; \
186 info->treble.i1[i] = v; \
187 info->treble.o2[i] = info->treble.o1[i]; \
188 w >>= FEEDEQ_COEFF_SHIFT; \
189 FEEDEQ_ERR_CLIP_CHECK(treble, w); \
190 v = FEEDEQ_CLAMP(w); \
191 info->treble.o1[i] = v; \
193 w = (intpcm64_t)v * bass->b0; \
194 w += (intpcm64_t)info->bass.i1[i] * bass->b1; \
195 w += (intpcm64_t)info->bass.i2[i] * bass->b2; \
196 w -= (intpcm64_t)info->bass.o1[i] * bass->a1; \
197 w -= (intpcm64_t)info->bass.o2[i] * bass->a2; \
198 info->bass.i2[i] = info->bass.i1[i]; \
199 info->bass.i1[i] = v; \
200 info->bass.o2[i] = info->bass.o1[i]; \
201 w >>= FEEDEQ_COEFF_SHIFT; \
202 FEEDEQ_ERR_CLIP_CHECK(bass, w); \
203 v = FEEDEQ_CLAMP(w); \
204 info->bass.o1[i] = v; \
207 _PCM_WRITE_##SIGN##BIT##_##ENDIAN(dst, v); \
208 dst += PCM_##BIT##_BPS; \
210 } while (--j != 0); \
211 } while (--count != 0); \
214 #if BYTE_ORDER == LITTLE_ENDIAN || defined(SND_FEEDER_MULTIFORMAT)
215 FEEDEQ_DECLARE(S, 16, LE)
216 FEEDEQ_DECLARE(S, 32, LE)
218 #if BYTE_ORDER == BIG_ENDIAN || defined(SND_FEEDER_MULTIFORMAT)
219 FEEDEQ_DECLARE(S, 16, BE)
220 FEEDEQ_DECLARE(S, 32, BE)
222 #ifdef SND_FEEDER_MULTIFORMAT
223 FEEDEQ_DECLARE(S, 8, NE)
224 FEEDEQ_DECLARE(S, 24, LE)
225 FEEDEQ_DECLARE(S, 24, BE)
226 FEEDEQ_DECLARE(U, 8, NE)
227 FEEDEQ_DECLARE(U, 16, LE)
228 FEEDEQ_DECLARE(U, 24, LE)
229 FEEDEQ_DECLARE(U, 32, LE)
230 FEEDEQ_DECLARE(U, 16, BE)
231 FEEDEQ_DECLARE(U, 24, BE)
232 FEEDEQ_DECLARE(U, 32, BE)
235 #define FEEDEQ_ENTRY(SIGN, BIT, ENDIAN) \
237 AFMT_##SIGN##BIT##_##ENDIAN, \
238 feed_eq_biquad_##SIGN##BIT##ENDIAN \
242 static const struct {
245 } feed_eq_biquad_tab[] = {
246 #if BYTE_ORDER == LITTLE_ENDIAN || defined(SND_FEEDER_MULTIFORMAT)
247 FEEDEQ_ENTRY(S, 16, LE),
248 FEEDEQ_ENTRY(S, 32, LE),
250 #if BYTE_ORDER == BIG_ENDIAN || defined(SND_FEEDER_MULTIFORMAT)
251 FEEDEQ_ENTRY(S, 16, BE),
252 FEEDEQ_ENTRY(S, 32, BE),
254 #ifdef SND_FEEDER_MULTIFORMAT
255 FEEDEQ_ENTRY(S, 8, NE),
256 FEEDEQ_ENTRY(S, 24, LE),
257 FEEDEQ_ENTRY(S, 24, BE),
258 FEEDEQ_ENTRY(U, 8, NE),
259 FEEDEQ_ENTRY(U, 16, LE),
260 FEEDEQ_ENTRY(U, 24, LE),
261 FEEDEQ_ENTRY(U, 32, LE),
262 FEEDEQ_ENTRY(U, 16, BE),
263 FEEDEQ_ENTRY(U, 24, BE),
264 FEEDEQ_ENTRY(U, 32, BE)
268 #define FEEDEQ_BIQUAD_TAB_SIZE \
269 ((int32_t)(sizeof(feed_eq_biquad_tab) / sizeof(feed_eq_biquad_tab[0])))
271 static struct feed_eq_coeff *
272 feed_eq_coeff_rate(uint32_t rate)
274 uint32_t spd, threshold;
277 if (rate < FEEDEQ_RATE_MIN || rate > FEEDEQ_RATE_MAX)
281 * Not all rates are supported. Choose the best rate that we can to
282 * allow 'sloppy' conversion. Good enough for naive listeners.
284 for (i = 0; i < FEEDEQ_TAB_SIZE; i++) {
285 spd = feed_eq_tab[i].rate;
286 threshold = spd + ((i < (FEEDEQ_TAB_SIZE - 1) &&
287 feed_eq_tab[i + 1].rate > spd) ?
288 ((feed_eq_tab[i + 1].rate - spd) >> 1) : 0);
290 (feeder_eq_exact_rate == 0 && rate <= threshold))
291 return (feed_eq_tab[i].coeff);
298 feeder_eq_validrate(uint32_t rate)
301 if (feed_eq_coeff_rate(rate) != NULL)
308 feed_eq_reset(struct feed_eq_info *info)
312 for (i = 0; i < info->channels; i++) {
313 info->treble.i1[i] = 0;
314 info->treble.i2[i] = 0;
315 info->treble.o1[i] = 0;
316 info->treble.o2[i] = 0;
317 info->bass.i1[i] = 0;
318 info->bass.i2[i] = 0;
319 info->bass.o1[i] = 0;
320 info->bass.o2[i] = 0;
325 feed_eq_setup(struct feed_eq_info *info)
328 info->coeff = feed_eq_coeff_rate(info->rate);
329 if (info->coeff == NULL)
338 feed_eq_init(struct pcm_feeder *f)
340 struct feed_eq_info *info;
344 if (f->desc->in != f->desc->out)
349 for (i = 0; i < FEEDEQ_BIQUAD_TAB_SIZE && biquad_op == NULL; i++) {
350 if (AFMT_ENCODING(f->desc->in) == feed_eq_biquad_tab[i].format)
351 biquad_op = feed_eq_biquad_tab[i].biquad;
354 if (biquad_op == NULL)
357 info = malloc(sizeof(*info), M_DEVBUF, M_NOWAIT | M_ZERO);
361 info->channels = AFMT_CHANNEL(f->desc->in);
362 info->align = info->channels * AFMT_BPS(f->desc->in);
364 info->rate = FEEDEQ_RATE_MIN;
365 info->treble.gain = FEEDEQ_L2GAIN(50);
366 info->bass.gain = FEEDEQ_L2GAIN(50);
367 info->preamp = FEEDEQ_PREAMP2IDX(FEEDEQ_PREAMP_DEFAULT);
368 info->state = FEEDEQ_UNKNOWN;
370 info->biquad = biquad_op;
374 return (feed_eq_setup(info));
378 feed_eq_set(struct pcm_feeder *f, int what, int value)
380 struct feed_eq_info *info;
385 case FEEDEQ_CHANNELS:
386 if (value < SND_CHN_MIN || value > SND_CHN_MAX)
388 info->channels = (uint32_t)value;
389 info->align = info->channels * AFMT_BPS(f->desc->in);
393 if (feeder_eq_validrate(value) == 0)
395 info->rate = (uint32_t)value;
396 if (info->state == FEEDEQ_UNKNOWN)
397 info->state = FEEDEQ_ENABLE;
398 return (feed_eq_setup(info));
402 if (value < 0 || value > 100)
404 if (what == FEEDEQ_TREBLE)
405 info->treble.gain = FEEDEQ_L2GAIN(value);
407 info->bass.gain = FEEDEQ_L2GAIN(value);
410 if (value < FEEDEQ_PREAMP_MIN || value > FEEDEQ_PREAMP_MAX)
412 info->preamp = FEEDEQ_PREAMP2IDX(value);
415 if (!(value == FEEDEQ_BYPASS || value == FEEDEQ_ENABLE ||
416 value == FEEDEQ_DISABLE))
430 feed_eq_free(struct pcm_feeder *f)
432 struct feed_eq_info *info;
436 free(info, M_DEVBUF);
444 feed_eq_feed(struct pcm_feeder *f, struct pcm_channel *c, uint8_t *b,
445 uint32_t count, void *source)
447 struct feed_eq_info *info;
455 * FEEDEQ_BYPASS - Bypass entirely, nothing happened.
456 * FEEDEQ_ENABLE - Preamp+biquad filtering.
457 * FEEDEQ_DISABLE - Preamp only.
459 if (info->state == FEEDEQ_BYPASS)
460 return (FEEDER_FEED(f->source, c, b, count, source));
463 count = SND_FXROUND(count, info->align);
466 if (count < info->align)
469 j = SND_FXDIV(FEEDER_FEED(f->source, c, dst, count, source),
474 info->biquad(info, dst, j);
480 } while (count != 0);
485 static struct pcm_feederdesc feeder_eq_desc[] = {
486 { FEEDER_EQ, 0, 0, 0, 0 },
490 static kobj_method_t feeder_eq_methods[] = {
491 KOBJMETHOD(feeder_init, feed_eq_init),
492 KOBJMETHOD(feeder_free, feed_eq_free),
493 KOBJMETHOD(feeder_set, feed_eq_set),
494 KOBJMETHOD(feeder_feed, feed_eq_feed),
498 FEEDER_DECLARE(feeder_eq, NULL);
501 feed_eq_scan_preamp_arg(const char *s)
507 bzero(buf, sizeof(buf));
509 /* XXX kind of ugly, but works for now.. */
511 r = sscanf(s, "%d.%d", &i, &f);
513 if (r == 1 && !(i < FEEDEQ_PREAMP_IMIN || i > FEEDEQ_PREAMP_IMAX)) {
514 snprintf(buf, sizeof(buf), "%c%d",
515 FEEDEQ_PREAMP_SIGNMARK(i), abs(i));
518 !(i < FEEDEQ_PREAMP_IMIN || i > FEEDEQ_PREAMP_IMAX ||
519 f < FEEDEQ_PREAMP_FMIN || f > FEEDEQ_PREAMP_FMAX))
520 snprintf(buf, sizeof(buf), "%c%d.%d",
521 FEEDEQ_PREAMP_SIGNMARK(i), abs(i), f);
523 return (FEEDEQ_PREAMP_INVALID);
526 if (len > 2 && strcasecmp(s + len - 2, "dB") == 0)
527 strlcat(buf, "dB", sizeof(buf));
529 if (i == 0 && *s == '-')
532 if (strcasecmp(buf + ((*s >= '0' && *s <= '9') ? 1 : 0), s) != 0)
533 return (FEEDEQ_PREAMP_INVALID);
535 while ((f / FEEDEQ_GAIN_DIV) > 0)
536 f /= FEEDEQ_GAIN_DIV;
538 return (((i < 0 || *buf == '-') ? -1 : 1) * FEEDEQ_IF2PREAMP(i, f));
543 sysctl_dev_pcm_eq(SYSCTL_HANDLER_ARGS)
545 struct snddev_info *d;
546 struct pcm_channel *c;
547 struct pcm_feeder *f;
551 if (!PCM_REGISTERED(d))
556 if (d->flags & SD_F_EQ_BYPASSED)
558 else if (d->flags & SD_F_EQ_ENABLED)
566 err = sysctl_handle_int(oidp, &val, 0, req);
568 if (err == 0 && req->newptr != NULL && val != oval) {
569 if (!(val == 0 || val == 1 || val == 2)) {
570 PCM_RELEASE_QUICK(d);
576 d->flags &= ~(SD_F_EQ_ENABLED | SD_F_EQ_BYPASSED);
579 d->flags |= SD_F_EQ_BYPASSED;
580 } else if (val == 1) {
582 d->flags |= SD_F_EQ_ENABLED;
584 val = FEEDEQ_DISABLE;
586 CHN_FOREACH(c, d, channels.pcm.busy) {
588 f = chn_findfeeder(c, FEEDER_EQ);
590 (void)FEEDER_SET(f, FEEDEQ_STATE, val);
597 PCM_RELEASE_QUICK(d);
603 sysctl_dev_pcm_eq_preamp(SYSCTL_HANDLER_ARGS)
605 struct snddev_info *d;
606 struct pcm_channel *c;
607 struct pcm_feeder *f;
612 if (!PCM_REGISTERED(d))
618 bzero(buf, sizeof(buf));
619 (void)snprintf(buf, sizeof(buf), "%c%d.%ddB",
620 FEEDEQ_PREAMP_SIGNMARK(val), FEEDEQ_PREAMP_IPART(val),
621 FEEDEQ_PREAMP_FPART(val));
626 err = sysctl_handle_string(oidp, buf, sizeof(buf), req);
628 if (err == 0 && req->newptr != NULL) {
629 val = feed_eq_scan_preamp_arg(buf);
630 if (val == FEEDEQ_PREAMP_INVALID) {
631 PCM_RELEASE_QUICK(d);
638 if (val < FEEDEQ_PREAMP_MIN)
639 val = FEEDEQ_PREAMP_MIN;
640 else if (val > FEEDEQ_PREAMP_MAX)
641 val = FEEDEQ_PREAMP_MAX;
645 CHN_FOREACH(c, d, channels.pcm.busy) {
647 f = chn_findfeeder(c, FEEDER_EQ);
649 (void)FEEDER_SET(f, FEEDEQ_PREAMP, val);
658 PCM_RELEASE_QUICK(d);
664 feeder_eq_initsys(device_t dev)
666 struct snddev_info *d;
670 d = device_get_softc(dev);
672 if (!(resource_string_value(device_get_name(dev), device_get_unit(dev),
673 "eq_preamp", &preamp) == 0 &&
674 (d->eqpreamp = feed_eq_scan_preamp_arg(preamp)) !=
675 FEEDEQ_PREAMP_INVALID))
676 d->eqpreamp = FEEDEQ_PREAMP_DEFAULT;
678 if (d->eqpreamp < FEEDEQ_PREAMP_MIN)
679 d->eqpreamp = FEEDEQ_PREAMP_MIN;
680 else if (d->eqpreamp > FEEDEQ_PREAMP_MAX)
681 d->eqpreamp = FEEDEQ_PREAMP_MAX;
683 SYSCTL_ADD_PROC(device_get_sysctl_ctx(dev),
684 SYSCTL_CHILDREN(device_get_sysctl_tree(dev)), OID_AUTO,
685 "eq", CTLTYPE_INT | CTLFLAG_RWTUN, d, sizeof(d),
686 sysctl_dev_pcm_eq, "I",
687 "Bass/Treble Equalizer (0=disable, 1=enable, 2=bypass)");
689 (void)snprintf(buf, sizeof(buf), "Bass/Treble Equalizer Preamp "
690 "(-/+ %d.0dB , %d.%ddB step)",
691 FEEDEQ_GAIN_MAX, FEEDEQ_GAIN_STEP / FEEDEQ_GAIN_DIV,
692 FEEDEQ_GAIN_STEP - ((FEEDEQ_GAIN_STEP / FEEDEQ_GAIN_DIV) *
695 SYSCTL_ADD_PROC(device_get_sysctl_ctx(dev),
696 SYSCTL_CHILDREN(device_get_sysctl_tree(dev)), OID_AUTO,
697 "eq_preamp", CTLTYPE_STRING | CTLFLAG_RWTUN, d, sizeof(d),
698 sysctl_dev_pcm_eq_preamp, "A", buf);