You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
MicroDexed/sgtl5000_graphic_eq.hpp

329 lines
8.2 KiB

#ifdef SGTL5000_AUDIO_ENHANCE
#include <Arduino.h>
#include <Audio.h>
#define EQ_LOWPASS 0
#define EQ_HIGHPASS 1
#define EQ_BANDPASS 2
#define EQ_NOTCH 3
#define EQ_PEAK 4
#define EQ_LOWSHELF 5
#define EQ_HIGHSHELF 6
#define GRAPHIC_EQ_TYPE_0 EQ_HIGHPASS
#define GRAPHIC_EQ_CENTER_FRQ_0 115.0
#define GRAPHIC_EQ_Q_0 2.0
#define GRAPHIC_EQ_TYPE_1 EQ_BANDPASS
#define GRAPHIC_EQ_CENTER_FRQ_1 330.0
#define GRAPHIC_EQ_Q_1 2.0
#define GRAPHIC_EQ_TYPE_2 EQ_BANDPASS
#define GRAPHIC_EQ_CENTER_FRQ_2 990.0
#define GRAPHIC_EQ_Q_2 2.0
#define GRAPHIC_EQ_TYPE_3 EQ_BANDPASS
#define GRAPHIC_EQ_CENTER_FRQ_3 2000.0
#define GRAPHIC_EQ_Q_3 2.0
#define GRAPHIC_EQ_TYPE_4 EQ_BANDPASS
#define GRAPHIC_EQ_CENTER_FRQ_4 4000.0
#define GRAPHIC_EQ_Q_4 2.0
#define GRAPHIC_EQ_TYPE_5 EQ_BANDPASS
#define GRAPHIC_EQ_CENTER_FRQ_5 9900.0
#define GRAPHIC_EQ_Q_5 2.0
#define GRAPHIC_EQ_TYPE_6 EQ_LOWPASS
#define GRAPHIC_EQ_CENTER_FRQ_6 11000.0
#define GRAPHIC_EQ_Q_6 2.0
extern AudioControlSGTL5000 sgtl5000_1;
class BiquadCoef
{
public:
BiquadCoef(uint8_t num_bands);
~BiquadCoef();
void set_eq_type(uint8_t band, uint8_t ft);
void set_eq_Fc(uint8_t band, float32_t frq);
void set_eq_Q(uint8_t band, float32_t q);
void set_gain(uint8_t band, float32_t gain);
void get_coef(uint8_t band, int* c);
private:
void calcBiquadCoefficients(uint8_t band);
uint8_t num_bands;
uint8_t *filter_type;
float32_t *Fc;
float32_t *Q;
float32_t *peakGainDB;
float32_t *a0;
float32_t *a1;
float32_t *a2;
float32_t *b1;
float32_t *b2;
};
BiquadCoef::BiquadCoef(uint8_t num_bands)
{
num_bands = constrain(num_bands, 1, 7);
sgtl5000_1.eqFilterCount(num_bands);
filter_type = new uint8_t[num_bands];
Fc = new float32_t[num_bands];
Q = new float32_t[num_bands];
peakGainDB = new float32_t[num_bands];
a0 = new float32_t[num_bands];
a1 = new float32_t[num_bands];
a2 = new float32_t[num_bands];
b1 = new float32_t[num_bands];
b2 = new float32_t[num_bands];
set_eq_type(0, GRAPHIC_EQ_TYPE_0);
set_eq_Fc(0, GRAPHIC_EQ_CENTER_FRQ_0);
set_eq_Q(0, GRAPHIC_EQ_Q_0);
set_gain(0, 0.0);
if (num_bands > 1)
{
set_eq_type(1, GRAPHIC_EQ_TYPE_1);
set_eq_Fc(1, GRAPHIC_EQ_CENTER_FRQ_1);
set_eq_Q(1, GRAPHIC_EQ_Q_1);
set_gain(1, 0.0);
}
if (num_bands > 2)
{
set_eq_type(2, GRAPHIC_EQ_TYPE_2);
set_eq_Fc(2, GRAPHIC_EQ_CENTER_FRQ_2);
set_eq_Q(2, GRAPHIC_EQ_Q_2);
set_gain(2, 0.0);
}
if (num_bands > 3)
{
set_eq_type(3, GRAPHIC_EQ_TYPE_3);
set_eq_Fc(3, GRAPHIC_EQ_CENTER_FRQ_3);
set_eq_Q(3, GRAPHIC_EQ_Q_3);
set_gain(3, 0.0);
}
if (num_bands > 4)
{
set_eq_type(4, GRAPHIC_EQ_TYPE_4);
set_eq_Fc(4, GRAPHIC_EQ_CENTER_FRQ_4);
set_eq_Q(4, GRAPHIC_EQ_Q_4);
set_gain(4, 0.0);
}
if (num_bands > 5)
{
set_eq_type(5, GRAPHIC_EQ_TYPE_5);
set_eq_Fc(5, GRAPHIC_EQ_CENTER_FRQ_5);
set_eq_Q(5, GRAPHIC_EQ_Q_5);
set_gain(5, 0.0);
}
if (num_bands > 6)
{
set_eq_type(6, GRAPHIC_EQ_TYPE_6);
set_eq_Fc(6, GRAPHIC_EQ_CENTER_FRQ_6);
set_eq_Q(6, GRAPHIC_EQ_Q_6);
set_gain(6, 0.0);
}
for (uint8_t i = 0; i < num_bands; i++)
{
int tmp[num_bands];
calcBiquadCoefficients(i);
get_coef(i, tmp);
sgtl5000_1.eqFilter(i, tmp);
}
}
BiquadCoef::~BiquadCoef()
{
;
}
void BiquadCoef::set_eq_type(uint8_t band, uint8_t ft)
{
int tmp[num_bands];
filter_type[band] = ft;
calcBiquadCoefficients(band);
get_coef(band, tmp);
sgtl5000_1.eqFilter(band, tmp);
}
void BiquadCoef::set_eq_Fc(uint8_t band, float32_t frq)
{
int tmp[num_bands];
Fc[band] = frq;
calcBiquadCoefficients(band);
get_coef(band, tmp);
sgtl5000_1.eqFilter(band, tmp);
}
void BiquadCoef::set_eq_Q(uint8_t band, float32_t q)
{
int tmp[num_bands];
Q[band] = q;
calcBiquadCoefficients(band);
get_coef(band, tmp);
sgtl5000_1.eqFilter(band, tmp);
}
void BiquadCoef::set_gain(uint8_t band, float32_t gain)
{
int tmp[num_bands];
peakGainDB[band] = gain;
calcBiquadCoefficients(band);
get_coef(band, tmp);
sgtl5000_1.eqFilter(band, tmp);
}
void BiquadCoef::get_coef(uint8_t band, int* c)
{
if (c != NULL)
{
c[0] = a0[band] * 0x8000;
c[1] = a1[band] * 0x8000;
c[2] = a2[band] * 0x8000;
c[3] = b1[band] * 0x8000;
c[4] = b2[band] * 0x8000;
}
}
// Taken from https://www.earlevel.com/main/2012/11/26/biquad-c-source-code/
//
// Biquad.h
//
// Created by Nigel Redmon on 11/24/12
// EarLevel Engineering: earlevel.com
// Copyright 2012 Nigel Redmon
//
// For a complete explanation of the Biquad code:
// http://www.earlevel.com/main/2012/11/25/biquad-c-source-code/
//
// License:
//
// This source code is provided as is, without warranty.
// You may copy and distribute verbatim copies of this document.
// You may modify and use this source code to create binary code
// for your own purposes, free or commercial.
//
void BiquadCoef::calcBiquadCoefficients(uint8_t band)
{
if (band > num_bands)
band = num_bands;
float32_t norm;
float32_t V = pow(10, fabs(peakGainDB[band]) / 20.0);
float32_t K = tan(M_PI * Fc[band]);
switch (filter_type[band]) {
case EQ_LOWPASS:
norm = 1 / (1 + K / Q[band] + K * K);
a0[band] = K * K * norm;
a1[band] = 2 * a0[band];
a2[band] = a0[band];
b1[band] = 2 * (K * K - 1) * norm;
b2[band] = (1 - K / Q[band] + K * K) * norm;
break;
case EQ_HIGHPASS:
norm = 1 / (1 + K / Q[band] + K * K);
a0[band] = 1 * norm;
a1[band] = -2 * a0[band];
a2[band] = a0[band];
b1[band] = 2 * (K * K - 1) * norm;
b2[band] = (1 - K / Q[band] + K * K) * norm;
break;
case EQ_BANDPASS:
norm = 1 / (1 + K / Q[band] + K * K);
a0[band] = K / Q[band] * norm;
a1[band] = 0;
a2[band] = -a0[band];
b1[band] = 2 * (K * K - 1) * norm;
b2[band] = (1 - K / Q[band] + K * K) * norm;
break;
case EQ_NOTCH:
norm = 1 / (1 + K / Q[band] + K * K);
a0[band] = (1 + K * K) * norm;
a1[band] = 2 * (K * K - 1) * norm;
a2[band] = a0[band];
b1[band] = a1[band];
b2[band] = (1 - K / Q[band] + K * K) * norm;
break;
case EQ_PEAK:
if (peakGainDB[band] >= 0) { // boost
norm = 1 / (1 + 1 / Q[band] * K + K * K);
a0[band] = (1 + V / Q[band] * K + K * K) * norm;
a1[band] = 2 * (K * K - 1) * norm;
a2[band] = (1 - V / Q[band] * K + K * K) * norm;
b1[band] = a1[band];
b2[band] = (1 - 1 / Q[band] * K + K * K) * norm;
}
else { // cut
norm = 1 / (1 + V / Q[band] * K + K * K);
a0[band] = (1 + 1 / Q[band] * K + K * K) * norm;
a1[band] = 2 * (K * K - 1) * norm;
a2[band] = (1 - 1 / Q[band] * K + K * K) * norm;
b1[band] = a1[band];
b2[band] = (1 - V / Q[band] * K + K * K) * norm;
}
break;
case EQ_LOWSHELF:
if (peakGainDB[band] >= 0) { // boost
norm = 1 / (1 + sqrt(2) * K + K * K);
a0[band] = (1 + sqrt(2 * V) * K + V * K * K) * norm;
a1[band] = 2 * (V * K * K - 1) * norm;
a2[band] = (1 - sqrt(2 * V) * K + V * K * K) * norm;
b1[band] = 2 * (K * K - 1) * norm;
b2[band] = (1 - sqrt(2) * K + K * K) * norm;
}
else { // cut
norm = 1 / (1 + sqrt(2 * V) * K + V * K * K);
a0[band] = (1 + sqrt(2) * K + K * K) * norm;
a1[band] = 2 * (K * K - 1) * norm;
a2[band] = (1 - sqrt(2) * K + K * K) * norm;
b1[band] = 2 * (V * K * K - 1) * norm;
b2[band] = (1 - sqrt(2 * V) * K + V * K * K) * norm;
}
break;
case EQ_HIGHSHELF:
if (peakGainDB[band] >= 0) { // boost
norm = 1 / (1 + sqrt(2) * K + K * K);
a0[band] = (V + sqrt(2 * V) * K + K * K) * norm;
a1[band] = 2 * (K * K - V) * norm;
a2[band] = (V - sqrt(2 * V) * K + K * K) * norm;
b1[band] = 2 * (K * K - 1) * norm;
b2[band] = (1 - sqrt(2) * K + K * K) * norm;
}
else { // cut
norm = 1 / (V + sqrt(2 * V) * K + K * K);
a0[band] = (1 + sqrt(2) * K + K * K) * norm;
a1[band] = 2 * (K * K - 1) * norm;
a2[band] = (1 - sqrt(2) * K + K * K) * norm;
b1[band] = 2 * (K * K - V) * norm;
b2[band] = (V - sqrt(2 * V) * K + K * K) * norm;
}
break;
}
return;
}
#endif