--- /dev/null
+
+/*
+ * CINELERRA
+ * Copyright (C) 1997-2011 Adam Williams <broadcast at earthling dot net>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ *
+ */
+
+#include "bcdisplayinfo.h"
+#include "clip.h"
+#include "cursors.h"
+#include "bchash.h"
+#include "filexml.h"
+#include "language.h"
+#include "cicolors.h"
+#include "samples.h"
+#include "echocancel.h"
+#include "theme.h"
+#include "transportque.inc"
+#include "units.h"
+#include "vframe.h"
+
+
+#include <string.h>
+#define TMPDIR "/tmp/echocancel/"
+
+
+REGISTER_PLUGIN(EchoCancel)
+
+
+EchoCancelConfig::EchoCancelConfig()
+{
+ level = 0.0;
+ normalize = 0;
+ xzoom = MIN_XZOOM;
+ peaks = MIN_PEAKS;
+ damp = MIN_DAMP;
+ cutoff = (MIN_CUTOFF+MAX_CUTOFF)/2;
+ gain = 0;
+ offset = 0;
+ window_size = 0;
+ mode = CANCEL_OFF;
+ history_size = 4;
+}
+
+int EchoCancelConfig::equivalent(EchoCancelConfig &that)
+{
+ return EQUIV(level, that.level) &&
+ normalize == that.normalize &&
+ xzoom == that.xzoom &&
+ damp == that.damp &&
+ cutoff == that.cutoff &&
+ window_size == that.window_size &&
+ mode == that.mode &&
+ history_size == that.history_size;
+}
+
+void EchoCancelConfig::copy_from(EchoCancelConfig &that)
+{
+ level = that.level;
+ normalize = that.normalize;
+ xzoom = that.xzoom;
+ peaks = that.peaks;
+ damp = that.damp;
+ cutoff = that.cutoff;
+ gain = that.gain;
+ offset = that.offset;
+ window_size = that.window_size;
+ mode = that.mode;
+ history_size = that.history_size;
+
+ if( window_size ) CLAMP(window_size, MIN_WINDOW, MAX_WINDOW);
+ CLAMP(history_size, MIN_HISTORY, MAX_HISTORY);
+ CLAMP(xzoom, MIN_XZOOM, MAX_XZOOM);
+ CLAMP(peaks, MIN_PEAKS, MAX_PEAKS);
+ CLAMP(damp, MIN_DAMP, MAX_DAMP);
+ CLAMP(cutoff, MIN_CUTOFF, MAX_CUTOFF);
+}
+
+void EchoCancelConfig::interpolate(EchoCancelConfig &prev,
+ EchoCancelConfig &next,
+ int64_t prev_frame,
+ int64_t next_frame,
+ int64_t current_frame)
+{
+ copy_from(prev);
+}
+
+
+
+EchoCancelFrame::EchoCancelFrame(int n)
+{
+ this->len = n;
+ data = new float[n + 1];
+ data[0] = 1;
+}
+
+EchoCancelFrame::~EchoCancelFrame()
+{
+ delete [] data;
+}
+
+
+
+EchoCancelLevel::EchoCancelLevel(EchoCancel *plugin, int x, int y)
+ : BC_FPot(x, y, plugin->config.level, INFINITYGAIN, MAXGAIN)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelLevel::handle_event()
+{
+ plugin->config.level = get_value();
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+
+
+EchoCancelMode::EchoCancelMode(EchoCancel *plugin, int x, int y)
+ : BC_PopupMenu(x, y, 120, to_text(plugin->config.mode))
+{
+ this->plugin = plugin;
+}
+
+void EchoCancelMode::create_objects()
+{
+ add_item(new BC_MenuItem(to_text(CANCEL_OFF)));
+ add_item(new BC_MenuItem(to_text(CANCEL_ON)));
+ add_item(new BC_MenuItem(to_text(CANCEL_MAN)));
+}
+
+int EchoCancelMode::handle_event()
+{
+ int new_mode = to_mode(get_text());
+ if( plugin->config.mode != new_mode ) {
+ plugin->config.mode = new_mode;
+ plugin->send_configure_change();
+ }
+ return 1;
+}
+
+const char* EchoCancelMode::to_text(int mode)
+{
+ switch(mode) {
+ case CANCEL_ON: return _("ON");
+ case CANCEL_MAN: return _("MAN");
+ }
+ return _("OFF");
+}
+
+int EchoCancelMode::to_mode(const char *text)
+{
+ if(!strcmp(to_text(CANCEL_ON), text)) return CANCEL_ON;
+ if(!strcmp(to_text(CANCEL_MAN), text)) return CANCEL_MAN;
+ return CANCEL_OFF;
+}
+
+
+
+
+EchoCancelHistory::EchoCancelHistory(EchoCancel *plugin,
+ int x,
+ int y)
+ : BC_IPot(x, y, plugin->config.history_size, MIN_HISTORY, MAX_HISTORY)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelHistory::handle_event()
+{
+ plugin->config.history_size = get_value();
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+
+
+
+EchoCancelWindowSize::EchoCancelWindowSize(EchoCancel *plugin, int x, int y, const char *text)
+ : BC_PopupMenu(x, y, 80, text)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelWindowSize::handle_event()
+{
+ plugin->config.window_size = atoi(get_text());
+ plugin->send_configure_change();
+ return 1;
+}
+
+const char *EchoCancelWindowSize::to_text(int size)
+{
+ if( !size ) return _("default");
+ static char string[BCSTRLEN];
+ sprintf(string, "%d", size);
+ return string;
+}
+
+int EchoCancelWindowSize::to_size(const char *text)
+{
+ return !strcmp(to_text(0),text) ? 0 : strtol(text,0,0);
+}
+
+
+
+EchoCancelWindowSizeTumbler::EchoCancelWindowSizeTumbler(EchoCancel *plugin, int x, int y)
+ : BC_Tumbler(x, y)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelWindowSizeTumbler::handle_up_event()
+{
+ if( !plugin->config.window_size )
+ plugin->config.window_size = MIN_WINDOW;
+ else if( (plugin->config.window_size *= 2) > MAX_WINDOW )
+ plugin->config.window_size = MAX_WINDOW;
+ EchoCancelWindowSize *window_size =
+ ((EchoCancelWindow *)plugin->get_thread()->get_window())->window_size;
+ const char *wsp = EchoCancelWindowSize::to_text(plugin->config.window_size);
+ window_size->set_text(wsp);
+ plugin->send_configure_change();
+ return 0;
+}
+
+int EchoCancelWindowSizeTumbler::handle_down_event()
+{
+ plugin->config.window_size /= 2;
+ if(plugin->config.window_size < MIN_WINDOW)
+ plugin->config.window_size = 0;
+ EchoCancelWindowSize *window_size =
+ ((EchoCancelWindow *)plugin->get_thread()->get_window())->window_size;
+ const char *wsp = EchoCancelWindowSize::to_text(plugin->config.window_size);
+ window_size->set_text(wsp);
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+
+
+EchoCancelNormalize::EchoCancelNormalize(EchoCancel *plugin, int x, int y)
+ : BC_CheckBox(x, y, plugin->config.normalize, _("Normalize"))
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelNormalize::handle_event()
+{
+ plugin->config.normalize = get_value();
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+
+EchoCancelXZoom::EchoCancelXZoom(EchoCancel *plugin, int x, int y)
+ : BC_IPot(x, y, plugin->config.xzoom, MIN_XZOOM, MAX_XZOOM)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelXZoom::handle_event()
+{
+ plugin->config.xzoom = get_value();
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+EchoCancelPeaks::EchoCancelPeaks(EchoCancel *plugin, int x, int y)
+ : BC_IPot(x, y, plugin->config.peaks, MIN_PEAKS, MAX_PEAKS)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelPeaks::handle_event()
+{
+ plugin->config.peaks = get_value();
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+EchoCancelDamp::EchoCancelDamp(EchoCancel *plugin, int x, int y)
+ : BC_IPot(x, y, plugin->config.damp, MIN_DAMP, MAX_DAMP)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelDamp::handle_event()
+{
+ plugin->config.damp = get_value();
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+EchoCancelCutoff::EchoCancelCutoff(EchoCancel *plugin, int x, int y)
+ : BC_IPot(x, y, plugin->config.cutoff, MIN_CUTOFF, MAX_CUTOFF)
+{
+ this->plugin = plugin;
+}
+
+int EchoCancelCutoff::handle_event()
+{
+ plugin->config.cutoff = get_value();
+ plugin->send_configure_change();
+ return 1;
+}
+
+
+
+EchoCancelCanvas::EchoCancelCanvas(EchoCancel *plugin, int x, int y, int w, int h)
+ : BC_SubWindow(x, y, w, h, BLACK)
+{
+ this->plugin = plugin;
+ current_operation = NONE;
+}
+
+int EchoCancelCanvas::button_press_event()
+{
+ if(is_event_win() && cursor_inside())
+ {
+ calculate_point(1);
+ current_operation = DRAG;
+ plugin->send_configure_change();
+ return 1;
+ }
+ return 0;
+}
+
+int EchoCancelCanvas::button_release_event()
+{
+ if(current_operation == DRAG)
+ {
+ calculate_point(2);
+ current_operation = NONE;
+ return 1;
+ }
+ return 0;
+}
+
+int EchoCancelCanvas::cursor_motion_event()
+{
+ if(current_operation == DRAG)
+ {
+ calculate_point(3);
+ }
+ return 0;
+}
+
+
+void EchoCancelCanvas::calculate_point(int do_overlay)
+{
+ int x = get_cursor_x();
+ int y = get_cursor_y();
+ CLAMP(x, 0, get_w() - 1);
+ CLAMP(y, 0, get_h() - 1);
+
+ EchoCancelWindow *window = (EchoCancelWindow *)plugin->thread->window;
+ window->calculate_frequency(x, y, do_overlay);
+}
+
+void EchoCancelCanvas::draw_overlay()
+{
+ EchoCancelWindow *window = (EchoCancelWindow*)plugin->thread->window;
+ if(window->probe_x >= 0 || window->probe_y >= 0)
+ {
+ set_color(GREEN);
+ set_inverse();
+ char string[BCTEXTLEN];
+ sprintf(string, "%d, %f", plugin->config.offset, plugin->config.gain);
+ draw_text(window->probe_x, window->probe_y, string);
+ draw_line(0, window->probe_y, get_w(), window->probe_y);
+ draw_line(window->probe_x, 0, window->probe_x, get_h());
+ set_opaque();
+ }
+}
+
+
+
+
+
+
+
+
+EchoCancelWindow::EchoCancelWindow(EchoCancel *plugin)
+ : PluginClientWindow(plugin, plugin->w, plugin->h, 320, 320, 1)
+{
+ this->plugin = plugin;
+ probe_x = probe_y = -1;
+}
+
+EchoCancelWindow::~EchoCancelWindow()
+{
+}
+
+void EchoCancelWindow::create_objects()
+{
+ int pad = plugin->get_theme()->widget_border;
+ add_subwindow(canvas = new EchoCancelCanvas(plugin, 0, 0,
+ get_w(), get_h() - 2*BC_Pot::calculate_h() - 3*pad));
+ canvas->set_cursor(CROSS_CURSOR, 0, 0);
+
+ int x = pad, y = canvas->get_y() + canvas->get_h() + pad;
+ int x1 = x, y1 = y;
+
+ add_subwindow(level_title = new BC_Title(x, y, _("Level:")));
+ x += level_title->get_w() + pad;
+ add_subwindow(level = new EchoCancelLevel(plugin, x, y));
+ x += level->get_w() + pad;
+ y += level->get_h() + pad;
+
+ x = x1;
+ add_subwindow(normalize = new EchoCancelNormalize(plugin, x, y));
+ x += normalize->get_w() + 3*pad;
+ y += normalize->get_h() - BC_Title::calculate_h(this,"Gain: ");
+ add_subwindow(gain_title = new EchoCancelTitle(x, y, _("Gain: "), 0.));
+ x += gain_title->get_w() + 2*pad;
+ add_subwindow(offset_title = new EchoCancelTitle(x, y, _("Offset: "), 0));
+
+ x = x1 + level_title->get_w() + level->get_w() + 2*pad;
+ x1 = x;
+ y = y1;
+
+ add_subwindow(mode_title = new BC_Title(x, y, _("Mode:")));
+ x += mode_title->get_w() + pad;
+ add_subwindow(mode = new EchoCancelMode(plugin, x, y));
+ mode->create_objects();
+
+ x = x1;
+ y += mode->get_h() + pad;
+
+ add_subwindow(window_size_title = new BC_Title(x, y, _("Window size:")));
+ x += window_size_title->get_w() + pad;
+ const char *wsp = EchoCancelWindowSize::to_text(plugin->config.window_size);
+ add_subwindow(window_size = new EchoCancelWindowSize(plugin, x, y, wsp));
+ x += window_size->get_w();
+ add_subwindow(window_size_tumbler = new EchoCancelWindowSizeTumbler(plugin, x, y));
+
+ window_size->add_item(new BC_MenuItem(EchoCancelWindowSize::to_text(0)));
+ for( int i=MIN_WINDOW; i<=MAX_WINDOW; i*=2 ) {
+ window_size->add_item(new BC_MenuItem(EchoCancelWindowSize::to_text(i)));
+ }
+
+ x = x1;
+ y += window_size->get_h() + pad;
+
+ x = x1 = window_size_tumbler->get_x() + window_size_tumbler->get_w() + pad;
+ y = y1;
+ add_subwindow(history_title = new BC_Title(x, y, _("History:")));
+ x += history_title->get_w() + pad;
+ add_subwindow(history = new EchoCancelHistory(plugin, x, y));
+
+ x = x1;
+ int y2 = y;
+ y += history->get_h() + pad;
+ add_subwindow(xzoom_title = new BC_Title(x, y, _("X Zoom:")));
+ x += xzoom_title->get_w() + pad;
+ add_subwindow(xzoom = new EchoCancelXZoom(plugin, x, y));
+ x += xzoom->get_w() + pad;
+ x1 = x;
+ add_subwindow(damp_title = new BC_Title(x, y, _("Damp:")));
+ x += damp_title->get_w() + pad;
+ add_subwindow(damp = new EchoCancelDamp(plugin, x, y));
+ x += damp->get_w() + pad;
+ add_subwindow(cutoff_title = new BC_Title(x, y, _("Cutoff Hz:")));
+ x += cutoff_title->get_w() + pad;
+ add_subwindow(cutoff = new EchoCancelCutoff(plugin, x, y));
+ int x2 = x - BC_Title::calculate_w(this, _("Peaks:")) - pad;
+ add_subwindow(peaks_title = new BC_Title(x2, y2, _("Peaks:")));
+ add_subwindow(peaks = new EchoCancelPeaks(plugin, x, y2));
+
+ x = x1 + 3*pad;
+ y = y1;
+ add_subwindow(freq_title = new BC_Title(x, y, _("0 Hz")));
+ y += freq_title->get_h() + pad;
+ add_subwindow(amplitude_title = new BC_Title(x, y, "Amplitude: 0 dB"));
+
+ show_window();
+}
+
+int EchoCancelWindow::resize_event(int w, int h)
+{
+ int canvas_h = canvas->get_h();
+ int canvas_difference = get_h() - canvas_h;
+
+ canvas->reposition_window(0, 0, w, h - canvas_difference);
+ canvas->clear_box(0, 0, canvas->get_w(), canvas->get_h());
+ probe_x = probe_y = -1;
+
+ int dx = 0, dy = canvas->get_h() - canvas_h;
+
+// Remove all columns which may be a different size.
+ plugin->frame_buffer.remove_all_objects();
+ plugin->frame_history.remove_all_objects();
+
+ level_title->reposition_window_relative(dx, dy);
+ level->reposition_window_relative(dx, dy);
+ mode_title->reposition_window_relative(dx, dy);
+ mode->reposition_window_relative(dx, dy);
+ window_size_title->reposition_window_relative(dx, dy);
+ window_size->reposition_window_relative(dx, dy);
+ window_size_tumbler->reposition_window_relative(dx, dy);
+ gain_title->reposition_window_relative(dx, dy);
+ offset_title->reposition_window_relative(dx, dy);
+ normalize->reposition_window_relative(dx, dy);
+ history_title->reposition_window_relative(dx, dy);
+ history->reposition_window_relative(dx, dy);
+ xzoom_title->reposition_window_relative(dx, dy);
+ xzoom->reposition_window_relative(dx, dy);
+ peaks_title->reposition_window_relative(dx, dy);
+ peaks->reposition_window_relative(dx, dy);
+ damp_title->reposition_window_relative(dx, dy);
+ damp->reposition_window_relative(dx, dy);
+ cutoff_title->reposition_window_relative(dx, dy);
+ cutoff->reposition_window_relative(dx, dy);
+ freq_title->reposition_window_relative(dx, dy);
+ amplitude_title->reposition_window_relative(dx, dy);
+
+ flush();
+ plugin->w = w;
+ plugin->h = h;
+ return 0;
+}
+
+
+void EchoCancelWindow::calculate_frequency(int x, int y, int do_overlay)
+{
+ if( !do_overlay && probe_x == x && probe_y == y ) return;
+ if( do_overlay & 2 ) canvas->draw_overlay();
+ probe_x = x; probe_y = y;
+
+// Convert to coordinates in frame history
+ int need_config_update = 0;
+ double gain = plugin->config.gain;
+ double offset = plugin->config.offset;
+
+ if( x >= 0 && plugin->frame_history.size() ) {
+ int ww = canvas->get_w();
+ int freq_pixel = x;
+ if( freq_pixel < 0 ) freq_pixel = 0;
+ if( freq_pixel > ww ) freq_pixel = ww;
+ int time_pixel = 0;
+ int idx = plugin->frame_history.size()-1 - time_pixel;
+ EchoCancelFrame *frm = plugin->frame_history[idx];
+ int pixels = canvas->get_w();
+ int half_window = plugin->header.window_size / 2;
+ offset = (double)half_window * freq_pixel/(pixels * plugin->config.xzoom);
+ if( offset <= 1e-10 ) offset = 1;
+ int freq = plugin->header.sample_rate / offset;
+ int msecs = 1000. / freq;
+ char string[BCTEXTLEN];
+ sprintf(string, "%d Hz, %d ms (%d))", freq, msecs, (int)offset);
+ freq_title->update(string);
+ int frm_sz1 = frm->size()-1;
+ if( freq_pixel > frm_sz1 ) freq_pixel = frm_sz1;
+ float *frame_data = frm->samples();
+ double level = frame_data[freq_pixel];
+ double scale = frm->scale();
+ sprintf(string, "Amplitude: %.3f (%.6g)", level, scale);
+ amplitude_title->update(string);
+ }
+ if( y >= 0 ) {
+ int hh = canvas->get_h()/2;
+ int gain_pixel = hh - y;
+ if( gain_pixel < 0 ) gain_pixel = 0;
+ if( gain_pixel > hh ) gain_pixel = hh;
+ gain = (double)gain_pixel / hh;
+ }
+ if( plugin->config.gain != gain ) {
+ gain_title->update(plugin->config.gain = gain);
+ need_config_update = 1;
+ }
+ if( offset > 0 && plugin->config.offset != offset ) {
+ offset_title->update(plugin->config.offset = offset);
+ need_config_update = 1;
+ }
+ if( need_config_update )
+ plugin->send_configure_change();
+
+ if( do_overlay & 1 ) canvas->draw_overlay();
+ if( do_overlay ) canvas->flash();
+}
+
+void EchoCancelWindow::update_gui()
+{
+ level->update(plugin->config.level);
+ char string[BCTEXTLEN];
+ sprintf(string, "%d", plugin->config.window_size);
+ window_size->set_text(string);
+ mode->set_text(EchoCancelMode::to_text(plugin->config.mode));
+ history->update(plugin->config.history_size);
+ normalize->set_value(plugin->config.normalize);
+ gain_title->update(plugin->config.gain);
+ offset_title->update(plugin->config.offset);
+}
+
+EchoCancel::EchoCancel(PluginServer *server)
+ : PluginAClient(server)
+{
+ reset();
+ timer = new Timer;
+ w = 640;
+ h = 480;
+}
+
+EchoCancel::~EchoCancel()
+{
+ delete fft;
+ delete audio_buffer;
+ delete data;
+ delete_buffers();
+ frame_buffer.remove_all_objects();
+ frame_history.remove_all_objects();
+ delete timer;
+}
+
+
+void EchoCancel::reset()
+{
+ thread = 0;
+ window_size = 0;
+ half_window = 0;
+ interrupted = 0;
+ fft = 0;
+ audio_buffer = 0;
+ data = 0;
+ cor = 0;
+ aud_real = 0; aud_imag = 0;
+ env_real = 0; env_imag = 0;
+ envelope = 0; env_data = 0;
+ time_frame = 0;
+ memset(&header, 0, sizeof(header));
+}
+
+void DataBuffer::
+set_buffer(int ofs, int len)
+{
+ int needed = ofs + len;
+ if( needed > allocated ) {
+ DataHeader *new_header = new_data_header(needed);
+ if( data_header ) {
+ int old_size = sizeof(DataHeader) + allocated*sizeof(float);
+ memcpy(new_header, data_header, old_size);
+ delete [] (char *)data_header;
+ }
+ else
+ memset(new_header, 0, sizeof(*new_header));
+ data_header = new_header;
+ allocated = needed;
+ }
+ sample_data = &data_header->samples[ofs];
+ data_len = len;
+}
+
+void AudioBuffer::
+set_buffer(int ofs, int len)
+{
+ int needed = ofs + len;
+ if( needed > allocated ) {
+ Samples *new_samples = new Samples(needed);
+ if( samples ) {
+ double *old_data = samples->get_data();
+ double *new_data = new_samples->get_data();
+ memcpy(new_data, old_data, allocated*sizeof(old_data[0]));
+ delete samples;
+ }
+ samples = new_samples;
+ allocated = needed;
+ }
+ sample_data = samples->get_data() + ofs;
+ data_len = len;
+}
+
+void AudioBuffer::
+append(double *bfr, int len)
+{
+ set_buffer(buffer_size(), len);
+ memcpy(get_data(), bfr, len*sizeof(sample_data[0]));
+ data_len = len;
+}
+
+void AudioBuffer::
+remove(int len)
+{
+ double *data = samples->get_data();
+ int move_len = buffer_size() - len;
+ memmove(data, data+len, move_len*sizeof(double));
+ if( move_len < data_len ) {
+ sample_data = data;
+ data_len = move_len;
+ }
+ else
+ sample_data -= len;
+}
+
+const char* EchoCancel::plugin_title() { return _("EchoCancel"); }
+int EchoCancel::is_realtime() { return 1; }
+
+static inline double sqr(double v) { return v*v; }
+
+static inline void cx_product(int n, int sf, double *rp, double *ip,
+ double *arp, double *aip, double *brp, double *bip)
+{
+ int m = !sf ? n : n/2, i = 0;
+ while( i <= m ) {
+ double ar = arp[i], ai = aip[i];
+ double br = brp[i], bi = bip[i];
+ rp[i] = ar*br - ai*bi; // complex a*ib
+ ip[i] = ar*bi + ai*br;
+ ++i;
+ }
+ if( !sf ) return;
+ while( --m > 0 ) { rp[i] = rp[m]; ip[i] = -ip[m]; ++i; }
+}
+
+static inline void cj_product(int n, int sf, double *rp, double *ip,
+ double *arp, double *aip, double *brp, double *bip)
+{
+ int m = !sf ? n : n/2, i = 0;
+ while( i <= m ) {
+ double ar = arp[i], ai = aip[i];
+ double br = brp[i], bi = -bip[i];
+ rp[i] = ar*br - ai*bi; // complex a*ib'
+ ip[i] = ar*bi + ai*br;
+ ++i;
+ }
+ if( !sf ) return;
+ while( --m > 0 ) { rp[i] = rp[m]; ip[i] = -ip[m]; ++i; }
+}
+
+static inline void log_power(int n, int sf, double *rp, double *arp, double *aip)
+{
+ int m = !sf ? n : n/2, i = 0;
+ while( i <= m ) {
+ double sr = arp[i], si = aip[i];
+ double ss = sqr(sr) + sqr(si);
+ rp[i] = ss > 1e-20 ? log(ss) : -46;
+ ++i;
+ }
+ if( !sf ) return;
+ while( --m > 0 ) { rp[i] = rp[m]; ++i; }
+}
+
+/* Adapted from Marple: Digital Spectral Analysis with Applications
+ * Solves linear simultaneous equations by the Levinson algorithm.
+ * TX = Z
+ * Input:
+ * T[m,m] a nonsymmetric Toeplitz matrix,
+ * tc[0..m-1] left column, tr[0..m-1] top row
+ * Z[m] known right-hand-side column,
+ * Output:
+ * X[m] solution vector
+ * Returns: 1 if singular, 0 on success
+ */
+
+static inline int toeplitz(int m, double *tc, double *tr, double *z, double *x)
+{
+ double a[m], b[m], p = tc[0];
+ if( p != tr[0] || fabs(p) < 1e-20 ) return 1;
+ x[0] = z[0] / p;
+ for( int k0=0,k=1; k<m; k0=k++ ) {
+ double sc = tc[k], sr = tr[k], xc = x[0] * sc;
+ for( int i=1,j=k0; i<k; ++i,--j ) {
+ sc += a[i] * tc[j];
+ sr += b[i] * tr[j];
+ xc += x[i] * tc[j];
+ }
+ double ac = -sc / p, br = -sr / p;
+ p *= 1 - ac * br;
+ if( fabs(p) < 1e-20 ) return 1;
+ a[k] = ac; b[k] = br;
+ x[k] = (z[k] - xc) / p;
+ for( int i=1,j=k0; j>0; ++i,--j ) {
+ sc = a[i]; sr = b[j];
+ a[i] += ac * sr;
+ b[j] += br * sc;
+ }
+ double xk = x[k];
+ for( int i=0,j=k; j>0; ++i,--j )
+ x[i] += xk * b[j];
+ }
+ return 0;
+}
+
+
+void EchoCancel::cepstrum(double *audio)
+{
+//dfile_dump(TMPDIR "audio",audio,window_size);
+ fft->do_fft(window_size, 0, audio, 0, aud_real, aud_imag);
+//dfile_dump(TMPDIR "aud_real",aud_real,window_size);
+//dfile_dump(TMPDIR "aud_imag",aud_imag,window_size);
+ // half zero, half window of audio
+ int k = 0;
+ for( ; k<half_window; ++k ) cor[k] = 0;
+ for( ; k<window_size; ++k ) cor[k] = audio[k];
+ fft->do_fft(window_size, 0, cor, 0, env_real, env_imag);
+//dfile_dump(TMPDIR "env_real",env_real,window_size);
+//dfile_dump(TMPDIR "env_imag",env_imag,window_size);
+ cj_product(window_size, 1, env_real, env_imag,
+ env_real, env_imag, aud_real, aud_imag);
+//dfile_dump(TMPDIR "prd_real",env_real,window_size);
+//dfile_dump(TMPDIR "prd_imag",env_imag,window_size);
+ log_power(window_size, 1, aud_real, env_real, env_imag);
+ // a = zeros, last half audio buffer, b = audio buffer
+ // COR = fft(a)*conj(fft(b)), cor = IFFT(COR)
+ // cepstrum = IFFT(log(mag(COR)**2))
+ double *cept_real = aud_real, *cept_imag = aud_imag; // do in place
+ fft->do_fft(window_size, 1, aud_real, 0, cept_real, cept_imag);
+ fft->do_fft(window_size, 1, env_real, env_imag, cor, aud_imag);
+//dfile_dump(TMPDIR "cor",cor,half_window);
+ int damp = config.damp; // apply dampening
+ if( ++time_frames < damp ) damp = time_frames;
+ float wt = 1./damp, wt1 = 1. - wt;
+ float *dp = data->get_buffer();
+ for( int i=0; i<half_window; ++i ) {
+ double v = cept_real[i] * cor[i];
+ dp[i] = time_frame[i] = (wt1*time_frame[i] + wt*v);
+ }
+//ffile_dump(TMPDIR "time_frame",time_frame,half_window);
+}
+
+void EchoCancel::dfile_dump(const char *fn, double *dp, int n)
+{
+ FILE *fp = fopen(fn,"w");
+ if( !fp ) return;
+ for( int i=0; i<n; ++i ) fprintf(fp,"%f\n",dp[i]);
+ fclose(fp);
+}
+
+void EchoCancel::ffile_dump(const char *fn, float *dp, int n)
+{
+ FILE *fp = fopen(fn,"w");
+ if( !fp ) return;
+ for( int i=0; i<n; ++i ) fprintf(fp,"%f\n",dp[i]);
+ fclose(fp);
+}
+
+void EchoCancel::calculate_envelope(int sample_rate, int peaks, int bsz)
+{
+ int cutoff = sample_rate/config.cutoff;
+ bsz |= 1; // should be odd
+ int bsz2 = bsz/2+1;
+ double power = 0;
+ for( int i=bsz2; --i>0; ) power += 2*cor[i];
+ power += cor[0];
+ int damp = config.damp; // apply dampening
+ if( ++time_frames < damp ) damp = time_frames;
+ float wt = 1./damp, wt1 = 1. - wt;
+ for( int i=0; i<half_window; ++i ) envelope[i] *= wt1;
+ if( power < 1e-12 ) return;
+ double *dp = env_real, *ep = dp;
+ for( int i=0; i<half_window; ++i ) *ep++ = time_frame[i];
+//printf("envelope %.4f:\n", power);
+ while( --peaks >= 0 ) {
+ int x = cutoff;
+ double *sp = dp+x, *mp = sp, *xp = sp;
+ double sx = 0; // first bsz band
+ for( int i=bsz; --i>=0; ) sx += *xp++;
+ double mx = sx;
+ while( xp < ep ) {
+ sx += *xp++ - *sp++;
+ if( sx > mx ) { mx = sx; mp = sp; }
+ }
+ x = mp - dp;
+ double echo_signal = 0, *cp = cor+x;
+ for( int i=0; i<bsz; ++i ) echo_signal += cp[i];
+//printf(" %d(%.3f/%.3f):\n", x, mx, echo_signal/power);
+ // positive reflections only, not too small
+ if( echo_signal/power < 0.001 ) break;
+ double gain[bsz];
+ int ret = toeplitz(bsz, cor, cor, cp, gain);
+ if( !ret ) {
+ double ss = 0;
+ for( int i=0; i<bsz; ++i ) ss += sqr(gain[i]);
+ double rms = sqrt(ss / bsz);
+//printf("rms=%f ",rms);
+ if( rms > 2 ) ret = 1; // too wacky
+ }
+ if( ret ) { // misbehaved, use single pulse in band center
+//printf("** ");
+ memset(gain, 0, bsz*sizeof(gain[0]));
+ gain[bsz2] = echo_signal / power;
+ }
+//double g=0;
+ for( int i=0; i<bsz; ++i,++x ) {
+ envelope[x] += wt * gain[i];
+//printf("[%d]=%.3f,", x, envelope[x]); g += envelope[x];
+ }
+//printf(" : %.4f\n",g);
+ if( !peaks ) break;
+ // remove echo from search, prevent periodic reverbs
+ for(int k=x; k<half_window; k+=x ) {
+ // clear this band and adj bands
+ int j = k - bsz, n = 3*bsz;
+ if( j < 0 ) { n += j; j = 0; }
+ if( j+n > half_window ) n = half_window - j;
+ for( int i=n; --i>=0; ++j ) dp[j] = 0;
+ }
+ }
+//printf(" ==\n");
+}
+
+void EchoCancel::create_envelope(int sample_rate)
+{
+ memset(envelope,0,half_window*sizeof(envelope[0]));
+ int offset = config.offset;
+ if( offset < 0 || offset >= half_window ) return;
+ double gain = config.gain;
+ if( gain < 0 || gain >= 1 ) return;
+ envelope[offset] = gain;
+}
+
+int EchoCancel::cancel_echo(double *bp, int size)
+{
+ int n = half_window + size;
+ if( n < window_size ) n = window_size;
+ if( audio_buffer->buffer_size() < n ) return 1;
+ int k = 0;
+ // a = audio data, t = env data: convolution = ifft(fft(a)*conj(fft(t))),
+ // but ifft(fft(a)*conj(fft(t)))==ifft(fft(a)*fft([t[0]]+t[:0:-1]))
+ // removing conj reverses t, as echos are from the past
+ if( config.mode == CANCEL_MAN ) create_envelope(sample_rate);
+ for( int i=0; i<half_window; ++i ) env_data[k++] = envelope[i];
+ while( k < window_size ) env_data[k++] = 0;
+ fft->do_fft(window_size, 0, env_data, 0, env_real, env_imag);
+
+ n = half_window + size;
+ if( n < window_size ) n = window_size;
+ double *ap = audio_buffer->get_buffer(-n);
+ fft->do_fft(window_size, 0, ap, 0, aud_real, aud_imag);
+ cx_product(window_size, 1, aud_real, aud_imag,
+ aud_real, aud_imag, env_real, env_imag);
+ fft->do_fft(window_size, 1, aud_real, aud_imag);
+ n = size;
+ if( n > half_window ) n = half_window;
+ double *sp = &aud_real[window_size - n];
+ for( int i=n; --i>=0; ++bp,++sp ) *bp -= *sp;
+
+ return 0;
+}
+
+void EchoCancel::delete_buffers()
+{
+ delete [] cor; cor = 0;
+ delete [] aud_real; aud_real = 0;
+ delete [] aud_imag; aud_imag = 0;
+ delete [] envelope; envelope = 0;
+ delete [] env_real; env_real = 0;
+ delete [] env_imag; env_real = 0;
+ delete [] env_data; env_data = 0;
+ delete [] time_frame; time_frame = 0;
+}
+
+void EchoCancel::alloc_buffers(int fft_sz)
+{
+ window_size = fft_sz;
+ half_window = window_size / 2;
+ cor = new double[window_size];
+ aud_real = new double[window_size];
+ aud_imag = new double[window_size];
+ envelope = new double[half_window];
+ env_real = new double[window_size];
+ env_imag = new double[window_size];
+ env_data = new double[window_size];
+ time_frame = new float[half_window];
+}
+
+int EchoCancel::process_buffer(int64_t size, Samples *buffer,
+ int64_t start_position, int sample_rate)
+{
+// Pass through
+ read_samples(buffer, 0, sample_rate, start_position, size);
+// Reset audio buffer
+ load_configuration();
+ int fft_sz = config.window_size;
+ if( !fft_sz ) fft_sz = 1<<FFT::samples_to_bits(sample_rate);
+ if( window_size != fft_sz ) {
+ render_stop();
+ delete_buffers();
+ alloc_buffers(fft_sz);
+ time_frames = 0;
+ }
+ if( !time_frames ) {
+ memset(envelope, 0, half_window*sizeof(envelope[0]));
+ memset(time_frame, 0, half_window*sizeof(time_frame[0]));
+ }
+ if( !data ) data = new DataBuffer(window_size);
+ if( !audio_buffer ) audio_buffer = new AudioBuffer(window_size+2*size);
+ if( !fft ) fft = new FFT;
+
+// Accumulate audio
+ int past_audio = audio_buffer->buffer_size() - window_size;
+ if( past_audio > 0 ) audio_buffer->remove(past_audio);
+ audio_buffer->append(buffer->get_data(), size);
+
+// process half_windows
+ double *bp = buffer->get_data();
+ int total_windows = 0, audio_size = audio_buffer->buffer_size();
+ while( audio_size >= window_size ) {
+ int sample_offset = half_window * total_windows++;
+ data->set_buffer(sample_offset, half_window);
+ double *audio = audio_buffer->get_buffer(-audio_size);
+ cepstrum(audio);
+ if( config.mode == CANCEL_ON )
+ calculate_envelope(sample_rate, config.peaks, 7);
+ if( config.mode != CANCEL_OFF && size > 0 )
+ cancel_echo(bp, size);
+ bp += half_window;
+ size -= half_window;
+ audio_size -= half_window;
+ }
+
+ DataHeader *data_header = data->get_header();
+ data_header->window_size = window_size;
+ data_header->interrupted = interrupted;
+ data_header->sample_rate = sample_rate;
+ data_header->total_windows = total_windows;
+ data_header->level = DB::fromdb(config.level); // Linear output level
+ send_render_gui(data_header, data_header->size());
+ interrupted = 0;
+ return 0;
+}
+
+void EchoCancel::render_stop()
+{
+ interrupted = 1;
+ if( audio_buffer ) audio_buffer->set_buffer(0, 0);
+ if( data ) { delete data; data = 0; }
+}
+
+
+
+NEW_WINDOW_MACRO(EchoCancel, EchoCancelWindow)
+
+
+void EchoCancelCanvas::
+draw_frame(float *data, int len, int hh)
+{
+ int w = get_w(), h = get_h();
+ float y = data[0];
+ int h1 = h-1;
+ int y1 = hh - hh*y;
+ for(int x1=0,x2=1; x2<w; ++x2 ) {
+ if( x2 < len ) y = data[x2];
+ int y2 = hh - hh*y;
+ CLAMP(y2, 0, h1);
+ draw_line(x1, y1, x2, y2);
+ x1 = x2; y1 = y2;
+ }
+}
+
+void EchoCancel::update_gui()
+{
+ if( !thread ) return;
+ EchoCancelWindow *window = (EchoCancelWindow*)thread->get_window();
+ window->lock_window("EchoCancel::update_gui");
+ if( load_configuration() )
+ window->update_gui();
+// Shift frames into history
+ int new_frames = 0;
+ while( frame_buffer.size() > 0 ) {
+ while( frame_history.size() >= config.history_size )
+ frame_history.remove_object_number(0);
+ frame_history.append(frame_buffer[0]);
+ frame_buffer.remove_number(0);
+ ++new_frames;
+ }
+
+ if( new_frames > 0 ) {
+ int last_frame = frame_history.size()-1;
+// Draw frames from history
+ EchoCancelCanvas *canvas = (EchoCancelCanvas*)window->canvas;
+ canvas->clear_box(0, 0, canvas->get_w(), canvas->get_h());
+ canvas->set_line_width(1);
+ for( int frame=0; frame<last_frame; ++frame ) {
+ int luma = 0x80 * (frame+1)/last_frame + 0x40;
+ canvas->set_color(luma*0x010101);
+ EchoCancelFrame *frm = frame_history[frame];
+ canvas->draw_frame(frm->samples(), frm->size(), h/2);
+ }
+ canvas->set_line_width(2);
+ canvas->set_color(WHITE);
+ EchoCancelFrame *frm = frame_history[last_frame];
+ canvas->draw_frame(frm->samples(), frm->size(), h/2);
+ canvas->set_color(RED);
+ canvas->draw_line(frm->cut_offset,0, frm->cut_offset,10);
+// Recompute probe level
+ int do_overlay = canvas->is_dragging() ? 1 : 0;
+ window->calculate_frequency(window->probe_x, window->probe_y, do_overlay);
+ canvas->flash();
+ }
+
+ window->unlock_window();
+}
+
+void EchoCancel::render_gui(void *data, int size)
+{
+ if( !thread ) return;
+ DataHeader *data_header = (DataHeader *)data;
+ memcpy(&header, data_header, sizeof(header));
+ if( window_size != data_header->window_size ) {
+ window_size = data_header->window_size;
+ half_window = window_size / 2;
+ frame_buffer.remove_all_objects();
+ frame_history.remove_all_objects();
+ }
+ if( data_header->interrupted )
+ interrupted = 1;
+ EchoCancelWindow *window = (EchoCancelWindow *)thread->get_window();
+ int pixels = window->canvas->get_w();
+ float window_scale = (double)half_window / (pixels * config.xzoom);
+ int nwin = data_header->total_windows;
+ int iwin = nwin - config.history_size;
+ if( iwin < 0 ) iwin = 0;
+ for( ; iwin < nwin; ++iwin ) {
+ float *samples = data_header->samples + half_window*iwin;
+ EchoCancelFrame *frm = new EchoCancelFrame(pixels);
+ float *frm_data = frm->samples();
+ double cut_period = (double)data_header->sample_rate / config.cutoff;
+ frm->cut_offset = cut_period / window_scale;
+
+ for(int i = 0; i < pixels; i++) {
+ float start = i*window_scale, stop = start+window_scale;
+ int istart = (int)start, istop = (int)stop;
+ float fstart = start-istart, fstop = stop-istop;
+ fstart = istart == istop ? -fstart : 1 - fstart;
+ float sum = samples[istart++] * fstart;
+ while( istart < istop ) sum += samples[istart++];
+ if( fstop > 1e-10 ) sum += samples[istop] * fstop;
+ frm_data[i] = sum/window_scale;
+ }
+// Normalize
+ float scale = data_header->level;
+ if( config.normalize ) {
+ float max = 0;
+ for( int i=frm->cut_offset; i<pixels; ++i ) {
+ float dat = fabsf(frm_data[i]);
+ if( dat > max ) max = dat;
+ }
+ scale = max > 1e-10 ? scale / max : 1;
+ }
+ if( scale != 1 ) frm->rescale(scale);
+ while( frame_buffer.size() >= config.history_size )
+ frame_buffer.remove_object_number(0);
+ frame_buffer.append(frm);
+ }
+
+ timer->update();
+}
+
+int EchoCancel::load_configuration()
+{
+ KeyFrame *prev_keyframe = get_prev_keyframe(get_source_position());
+ EchoCancelConfig old_config;
+ old_config.copy_from(config);
+ read_data(prev_keyframe);
+ return !old_config.equivalent(config) ? 1 : 0;
+}
+
+void EchoCancel::read_data(KeyFrame *keyframe)
+{
+ int result;
+ FileXML input;
+ input.set_shared_input(keyframe->get_data(), strlen(keyframe->get_data()));
+
+ while(!(result = input.read_tag()) ) {
+ if( !input.tag.title_is("ECHOCANCEL")) continue;
+ config.level = input.tag.get_property("LEVEL", config.level);
+ config.normalize = input.tag.get_property("NORMALIZE", config.normalize);
+ config.window_size = input.tag.get_property("WINDOW_SIZE", config.window_size);
+ config.xzoom = input.tag.get_property("XZOOM", config.xzoom);
+ config.peaks = input.tag.get_property("PEAKS", config.peaks);
+ config.mode = input.tag.get_property("MODE", config.mode);
+ config.damp = input.tag.get_property("DAMP", config.damp);
+ config.history_size = input.tag.get_property("HISTORY_SIZE", config.history_size);
+ config.gain = input.tag.get_property("GAIN", config.gain);
+ config.offset = input.tag.get_property("OFFSET", config.offset);
+ if( !is_defaults() ) continue;
+ w = input.tag.get_property("W", w);
+ h = input.tag.get_property("H", h);
+ }
+}
+
+void EchoCancel::save_data(KeyFrame *keyframe)
+{
+ FileXML output;
+ output.set_shared_output(keyframe->get_data(), MESSAGESIZE);
+
+ output.tag.set_title("ECHOCANCEL");
+ output.tag.set_property("LEVEL", config.level);
+ output.tag.set_property("NORMALIZE", config.normalize);
+ output.tag.set_property("WINDOW_SIZE", config.window_size);
+ output.tag.set_property("MODE", config.mode);
+ output.tag.set_property("XZOOM", config.xzoom);
+ output.tag.set_property("PEAKS", config.peaks);
+ output.tag.set_property("DAMP", config.damp);
+ output.tag.set_property("HISTORY_SIZE", config.history_size);
+ output.tag.set_property("GAIN", config.gain);
+ output.tag.set_property("OFFSET", config.offset);
+ output.tag.set_property("W", w);
+ output.tag.set_property("H", h);
+ output.append_tag();
+ output.tag.set_title("/ECHOCANCEL");
+ output.append_tag();
+ output.append_newline();
+ output.terminate_string();
+}
+
+
+
+
+