a-EQ: Revert one of the previous changes
[ardour.git] / libs / plugins / a-eq.lv2 / a-eq.c
index 8083d4742d20259965ec1fd9225027d0005e90ca..b0851f2120f73bc3d47829414c19973f3f310ab4 100644 (file)
 #include <stdbool.h>
 #include <stdio.h>
 
+#ifdef COMPILER_MSVC
+#include <float.h>
+#define isfinite_local(val) (bool)_finite((double)val)
+#else
+#define isfinite_local isfinite
+#endif
+
 #include "lv2/lv2plug.in/ns/lv2core/lv2.h"
 
 #ifdef LV2_EXTENDED
 
 #define AEQ_URI        "urn:ardour:a-eq"
 #define BANDS  6
-#define SMALL  0.0001f
-
 #ifndef MIN
 #define MIN(A,B) ((A) < (B)) ? (A) : (B)
 #endif
 
 typedef enum {
-       AEQ_SHELFTOGL = 0,
-       AEQ_FREQL,
+       AEQ_FREQL = 0,
        AEQ_GAINL,
        AEQ_FREQ1,
        AEQ_GAIN1,
@@ -54,7 +58,6 @@ typedef enum {
        AEQ_FREQ4,
        AEQ_GAIN4,
        AEQ_BW4,
-       AEQ_SHELFTOGH,
        AEQ_FREQH,
        AEQ_GAINH,
        AEQ_MASTER,
@@ -64,6 +67,7 @@ typedef enum {
        AEQ_FILTOG3,
        AEQ_FILTOG4,
        AEQ_FILTOGH,
+       AEQ_ENABLE,
        AEQ_INPUT,
        AEQ_OUTPUT,
 } PortIndex;
@@ -79,8 +83,8 @@ from_dB(double gdb) {
 }
 
 static inline bool
-is_eq(float a, float b) {
-       return (fabsf(a - b) < SMALL);
+is_eq(float a, float b, float small) {
+       return (fabsf(a - b) < small);
 }
 
 struct linear_svf {
@@ -95,16 +99,23 @@ static void linear_svf_reset(struct linear_svf *self)
        self->s[0] = self->s[1] = 0.0;
 }
 
+static void linear_svf_protect(struct linear_svf *self)
+{
+       if (!isfinite_local (self->s[0]) || !isfinite_local (self->s[1])) {
+               linear_svf_reset (self);
+       }
+}
+
 typedef struct {
-       float* shelftogl;
-       float* shelftogh;
        float* f0[BANDS];
        float* g[BANDS];
        float* bw[BANDS];
        float* filtog[BANDS];
        float* master;
+       float* enable;
 
        float srate;
+       float tau;
 
        float* input;
        float* output;
@@ -113,9 +124,6 @@ typedef struct {
        float v_g[BANDS];
        float v_bw[BANDS];
        float v_f0[BANDS];
-       float v_filtog[BANDS];
-       float v_shelftogl;
-       float v_shelftogh;
        float v_master;
 
        bool need_expose;
@@ -136,6 +144,7 @@ instantiate(const LV2_Descriptor* descriptor,
 {
        Aeq* aeq = (Aeq*)calloc(1, sizeof(Aeq));
        aeq->srate = rate;
+       aeq->tau = 1.0 - expf (-2.f * M_PI * 64.f * 25.f / aeq->srate); // 25Hz time constant @ 64fpp
 
 #ifdef LV2_EXTENDED
        for (int i=0; features[i]; ++i) {
@@ -164,8 +173,8 @@ connect_port(LV2_Handle instance,
        Aeq* aeq = (Aeq*)instance;
 
        switch ((PortIndex)port) {
-       case AEQ_SHELFTOGL:
-               aeq->shelftogl = (float*)data;
+       case AEQ_ENABLE:
+               aeq->enable = (float*)data;
                break;
        case AEQ_FREQL:
                aeq->f0[0] = (float*)data;
@@ -209,9 +218,6 @@ connect_port(LV2_Handle instance,
        case AEQ_BW4:
                aeq->bw[4] = (float*)data;
                break;
-       case AEQ_SHELFTOGH:
-               aeq->shelftogh = (float*)data;
-               break;
        case AEQ_FREQH:
                aeq->f0[5] = (float*)data;
                break;
@@ -261,46 +267,10 @@ activate(LV2_Handle instance)
 // SVF filters
 // http://www.cytomic.com/files/dsp/SvfLinearTrapOptimised2.pdf
 
-static void linear_svf_set_hp(struct linear_svf *self, float sample_rate, float cutoff, float resonance)
-{
-       double f0 = (double)cutoff;
-       double q = (double)resonance;
-       double sr = (double)sample_rate;
-
-       self->g = tan(M_PI * (f0 / sr));
-       self->k = 1.0 / q;
-
-       self->a[0] = 1.0 / (1.0 + self->g * (self->g + self->k));
-       self->a[1] = self->g * self->a[0];
-       self->a[2] = self->g * self->a[1];
-
-       self->m[0] = 1.0;
-       self->m[1] = -self->k;
-       self->m[2] = -1.0;
-}
-
-static void linear_svf_set_lp(struct linear_svf *self, float sample_rate, float cutoff, float resonance)
-{
-       double f0 = (double)cutoff;
-       double q = (double)resonance;
-       double sr = (double)sample_rate;
-
-       self->g = tan(M_PI * (f0 / sr));
-       self->k = 1.0 / q;
-
-       self->a[0] = 1.0 / (1.0 + self->g * (self->g + self->k));
-       self->a[1] = self->g * self->a[0];
-       self->a[2] = self->g * self->a[1];
-
-       self->m[0] = 0.0;
-       self->m[1] = 0.0;
-       self->m[2] = 1.0;
-}
-
 static void linear_svf_set_peq(struct linear_svf *self, float gdb, float sample_rate, float cutoff, float bandwidth)
 {
        double f0 = (double)cutoff;
-       double q = (double)pow(2.0, 1.0 / bandwidth) / (pow(2.0, bandwidth) - 1.0);
+       double q = (double)pow(2.0, 0.5 * bandwidth) / (pow(2.0, bandwidth) - 1.0);
        double sr = (double)sample_rate;
        double A = pow(10.0, gdb/40.0);
 
@@ -379,11 +349,7 @@ static void set_params(LV2_Handle instance, int band) {
 
        switch (band) {
        case 0:
-               if (aeq->v_shelftogl > 0.5) {
-                       linear_svf_set_lowshelf(&aeq->v_filter[0], aeq->v_g[0], aeq->srate, aeq->v_f0[0], 0.7071068);
-               } else {
-                       linear_svf_set_hp(&aeq->v_filter[0], aeq->srate, aeq->v_f0[0], 0.7071068);
-               }
+               linear_svf_set_lowshelf(&aeq->v_filter[0], aeq->v_g[0], aeq->srate, aeq->v_f0[0], 0.7071068);
                break;
        case 1:
        case 2:
@@ -392,11 +358,7 @@ static void set_params(LV2_Handle instance, int band) {
                linear_svf_set_peq(&aeq->v_filter[band], aeq->v_g[band], aeq->srate, aeq->v_f0[band], aeq->v_bw[band]);
                break;
        case 5:
-               if (aeq->v_shelftogh > 0.5) {
-                       linear_svf_set_highshelf(&aeq->v_filter[5], aeq->v_g[5], aeq->srate, aeq->v_f0[5], 0.7071068);
-               } else {
-                       linear_svf_set_lp(&aeq->v_filter[5], aeq->srate, aeq->v_f0[5], 0.7071068);
-               }
+               linear_svf_set_highshelf(&aeq->v_filter[5], aeq->v_g[5], aeq->srate, aeq->v_f0[5], 0.7071068);
                break;
        }
 }
@@ -409,59 +371,75 @@ run(LV2_Handle instance, uint32_t n_samples)
        const float* const input = aeq->input;
        float* const output = aeq->output;
 
-       float in0, out;
-       uint32_t i, j;
+       const float tau = aeq->tau;
+       uint32_t offset = 0;
 
-       // 15Hz time constant
-       const float tau = (1.0 - exp(-2.0 * M_PI * n_samples * 15. / aeq->srate));
+       const float target_gain = *aeq->enable <= 0 ? 0 : *aeq->master; // dB
 
-       for (i = 0; i < n_samples; i++) {
-               in0 = input[i];
-               out = in0;
-               for (j = 0; j < BANDS; j++) {
-                       out = run_linear_svf(&aeq->v_filter[j], out);
-               }
-               output[i] = out * from_dB(*(aeq->master));
-       }
+       while (n_samples > 0) {
+               uint32_t block = n_samples;
+               bool any_changed = false;
 
-       for (i = 0; i < BANDS; i++) {
-               if (!is_eq(aeq->v_filtog[i], *aeq->filtog[i])) {
-                       aeq->v_filtog[i] = *(aeq->filtog[i]);
-               }
-               if (!is_eq(aeq->v_f0[i], *aeq->f0[i])) {
-                       aeq->v_f0[i] += tau * (*aeq->f0[i] - aeq->v_f0[i]);
-                       aeq->need_expose = true;
+               if (!is_eq(aeq->v_master, target_gain, 0.1)) {
+                       aeq->v_master += tau * (target_gain - aeq->v_master);
+                       any_changed = true;
+               } else {
+                       aeq->v_master = target_gain;
                }
-               if (aeq->v_filtog[i] < 0.5) {
-                       if (!is_eq(aeq->v_g[i], 0.f)) {
-                               aeq->v_g[i] += tau * (0.0 - aeq->v_g[i]);
-                               aeq->need_expose = true;
+
+               for (int i = 0; i < BANDS; ++i) {
+                       bool changed = false;
+
+                       if (!is_eq(aeq->v_f0[i], *aeq->f0[i], 0.1)) {
+                               aeq->v_f0[i] += tau * (*aeq->f0[i] - aeq->v_f0[i]);
+                               changed = true;
                        }
-               } else if (aeq->v_filtog[i] >= 0.5) {
-                       if (!is_eq(aeq->v_g[i], *aeq->g[i])) {
-                               aeq->v_g[i] += tau * (*aeq->g[i] - aeq->v_g[i]);
-                               aeq->need_expose = true;
+
+                       if (*aeq->filtog[i] <= 0 || *aeq->enable <= 0) {
+                               if (!is_eq(aeq->v_g[i], 0.f, 0.05)) {
+                                       aeq->v_g[i] += tau * (0.0 - aeq->v_g[i]);
+                                       changed = true;
+                               }
+                       } else {
+                               if (!is_eq(aeq->v_g[i], *aeq->g[i], 0.05)) {
+                                       aeq->v_g[i] += tau * (*aeq->g[i] - aeq->v_g[i]);
+                                       changed = true;
+                               }
+                       }
+
+                       if (i != 0 && i != 5) {
+                               if (!is_eq(aeq->v_bw[i], *aeq->bw[i], 0.001)) {
+                                       aeq->v_bw[i] += tau * (*aeq->bw[i] - aeq->v_bw[i]);
+                                       changed = true;
+                               }
+                       }
+
+                       if (changed) {
+                               set_params(aeq, i);
+                               any_changed = true;
                        }
                }
-               if (i != 0 && i != 5 && !is_eq(aeq->v_bw[i], *aeq->bw[i])) {
-                       aeq->v_bw[i] += tau * (*aeq->bw[i] - aeq->v_bw[i]);
-                       aeq->need_expose = true;
-               }
-               if (!is_eq(aeq->v_shelftogl, *aeq->shelftogl)) {
-                       aeq->v_shelftogl = *(aeq->shelftogl);
-                       aeq->need_expose = true;
-               }
-               if (!is_eq(aeq->v_shelftogh, *aeq->shelftogh)) {
-                       aeq->v_shelftogh = *(aeq->shelftogh);
-                       aeq->need_expose = true;
-               }
-               if (!is_eq(aeq->v_master, *aeq->master)) {
-                       aeq->v_master = *(aeq->master);
+
+               if (any_changed) {
                        aeq->need_expose = true;
+                       block = MIN (64, n_samples);
                }
-               if (aeq->need_expose == true) {
-                       set_params(aeq, i);
+
+               for (uint32_t i = 0; i < block; ++i) {
+                       float in0, out;
+                       in0 = input[i + offset];
+                       out = in0;
+                       for (uint32_t j = 0; j < BANDS; j++) {
+                               out = run_linear_svf(&aeq->v_filter[j], out);
+                       }
+                       output[i + offset] = out * from_dB(aeq->v_master);
                }
+               n_samples -= block;
+               offset += block;
+       }
+
+       for (uint32_t j = 0; j < BANDS; j++) {
+               linear_svf_protect(&aeq->v_filter[j]);
        }
 
 #ifdef LV2_EXTENDED
@@ -490,38 +468,6 @@ calc_peq(Aeq* self, int i, double omega) {
        return cabs(H);
 }
 
-static double
-calc_lowpass(Aeq* self, double omega) {
-       double complex H = 0.0;
-       double complex z = cexp(I * omega);
-       double complex zz = cexp(2. * I * omega);
-       double complex zm = z - 1.0;
-       double complex zp = z + 1.0;
-       double complex zzm = zz - 1.0;
-
-       double g = self->v_filter[5].g;
-       double k = self->v_filter[5].k;
-
-       H = (g*g*zp*zp) / (zm*zm + g*g*zp*zp + g*k*zzm);
-       return cabs(H);
-}
-
-static double
-calc_highpass(Aeq* self, double omega) {
-       double complex H = 0.0;
-       double complex z = cexp(I * omega);
-       double complex zz = cexp(2. * I * omega);
-       double complex zm = z - 1.0;
-       double complex zp = z + 1.0;
-       double complex zzm = zz - 1.0;
-
-       double g = self->v_filter[0].g;
-       double k = self->v_filter[0].k;
-
-       H = zm*zm / (zm*zm + g*g*zp*zp + g*k*zzm);
-       return cabs(H);
-}
-
 static double
 calc_lowshelf(Aeq* self, double omega) {
        double complex H = 0.0;
@@ -569,35 +515,17 @@ eq_curve (Aeq* self, float f) {
        double SR = (double)self->srate;
        double omega = f * 2. * M_PI / SR;
 
-       // low
-       if (self->v_shelftogl) {
-               // lowshelf
-               response *= calc_lowshelf(self, omega);
-       } else {
-               // hp:
-               response *= calc_highpass(self, omega);
-       }
+       // lowshelf
+       response *= calc_lowshelf(self, omega);
 
-       // peq1:
+       // peq 1 - 4:
        response *= calc_peq(self, 1, omega);
-
-       // peq2:
        response *= calc_peq(self, 2, omega);
-
-       // peq3:
        response *= calc_peq(self, 3, omega);
-
-       // peq4:
        response *= calc_peq(self, 4, omega);
 
-       // high
-       if (self->v_shelftogh) {
-               // highshelf:
-               response *= calc_highshelf(self, omega);
-       } else {
-               // lp:
-               response *= calc_lowpass(self, omega);
-       }
+       // highshelf:
+       response *= calc_highshelf(self, omega);
 
        return (float)response;
 }
@@ -606,7 +534,7 @@ static LV2_Inline_Display_Image_Surface *
 render_inline (LV2_Handle instance, uint32_t w, uint32_t max_h)
 {
        Aeq* self = (Aeq*)instance;
-       uint32_t h = MIN (w * 9 / 16, max_h);
+       uint32_t h = MIN (1 | (uint32_t)ceilf (w * 9.f / 16.f), max_h);
 
        if (!self->display || self->w != w || self->h != h) {
                if (self->display) cairo_surface_destroy(self->display);
@@ -624,19 +552,30 @@ render_inline (LV2_Handle instance, uint32_t w, uint32_t max_h)
 
        cairo_set_line_width(cr, 1.0);
 
-       // draw grid 5dB steps
-       const double dash2[] = {1, 3};
+       // prepare grid drawing
        cairo_save (cr);
-       cairo_set_line_cap(cr, CAIRO_LINE_CAP_ROUND);
+       const double dash2[] = {1, 3};
+       //cairo_set_line_cap(cr, CAIRO_LINE_CAP_ROUND);
        cairo_set_dash(cr, dash2, 2, 2);
        cairo_set_source_rgba (cr, 0.5, 0.5, 0.5, 0.5);
 
-       for (uint32_t d = 1; d < 8; ++d) {
-               const float y = -.5 + floorf (h * (d * 5.f / 40.f));
+       // draw x-grid 6dB steps
+       for (int32_t d = -18; d <= 18; d+=6) {
+               float y = (float)h * (d / 40.0 + 0.5);
+               y = rint (y) - .5;
                cairo_move_to (cr, 0, y);
                cairo_line_to (cr, w, y);
                cairo_stroke (cr);
        }
+       // draw y-axis grid 100, 1k, 10K
+       for (int32_t f = 100; f <= 10000; f *= 10) {
+               float x = w * log10 (f / 20.0) / log10 (1000.0);
+               x = rint (x) - .5;
+               cairo_move_to (cr, x, 0);
+               cairo_line_to (cr, x, h);
+               cairo_stroke (cr);
+       }
+
        cairo_restore (cr);