summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/GainControl.cpp110
-rw-r--r--src/GainControl.h21
2 files changed, 101 insertions, 30 deletions
diff --git a/src/GainControl.cpp b/src/GainControl.cpp
index cf0247d..81ae06e 100644
--- a/src/GainControl.cpp
+++ b/src/GainControl.cpp
@@ -43,6 +43,7 @@ union __u128 {
using namespace std;
+static float var_variance;
GainControl::GainControl(size_t framesize,
GainMode mode,
@@ -56,39 +57,24 @@ GainControl::GainControl(size_t framesize,
m_frameSize(framesize),
#endif
m_digGain(digGain),
- m_normalise(normalise)
+ m_normalise(normalise),
+ m_var_variance_rc(4.0f),
+ m_gainmode(mode),
+ m_mutex()
{
PDEBUG("GainControl::GainControl(%zu, %zu) @ %p\n", framesize, (size_t)mode, this);
/* register the parameters that can be remote controlled */
RC_ADD_PARAMETER(digital, "Digital Gain");
-
- switch(mode) {
- case GainMode::GAIN_FIX:
- PDEBUG("Gain mode: fix\n");
- computeGain = computeGainFix;
- break;
- case GainMode::GAIN_MAX:
- PDEBUG("Gain mode: max\n");
- computeGain = computeGainMax;
- break;
- case GainMode::GAIN_VAR:
- PDEBUG("Gain mode: var\n");
- computeGain = computeGainVar;
- break;
- default:
- throw std::runtime_error(
- "GainControl::GainControl invalid computation gain mode!");
- }
+ RC_ADD_PARAMETER(mode, "Gainmode (fix|max|var)");
+ RC_ADD_PARAMETER(var, "Variance setting for gainmode var (default: 4)");
}
-
GainControl::~GainControl()
{
PDEBUG("GainControl::~GainControl() @ %p\n", this);
}
-
int GainControl::internal_process(Buffer* const dataIn, Buffer* dataOut)
{
PDEBUG("GainControl::process"
@@ -98,12 +84,41 @@ int GainControl::internal_process(Buffer* const dataIn, Buffer* dataOut)
dataOut->setLength(dataIn->getLength());
#ifdef __SSE__
+ __m128 (*computeGain)(const __m128* in, size_t sizeIn);
+#else
+ float (*computeGain)(const complexf* in, size_t sizeIn);
+#endif
+ {
+ std::lock_guard<std::mutex> lock(m_mutex);
+
+ var_variance = m_var_variance_rc;
+
+ switch (m_gainmode) {
+ case GainMode::GAIN_FIX:
+ PDEBUG("Gain mode: fix\n");
+ computeGain = computeGainFix;
+ break;
+ case GainMode::GAIN_MAX:
+ PDEBUG("Gain mode: max\n");
+ computeGain = computeGainMax;
+ break;
+ case GainMode::GAIN_VAR:
+ PDEBUG("Gain mode: var\n");
+ computeGain = computeGainVar;
+ break;
+ default:
+ throw std::logic_error("Internal error: invalid gainmode");
+ }
+ }
+
+#ifdef __SSE__
const __m128* in = reinterpret_cast<const __m128*>(dataIn->getData());
__m128* out = reinterpret_cast<__m128*>(dataOut->getData());
size_t sizeIn = dataIn->getLength() / sizeof(__m128);
size_t sizeOut = dataOut->getLength() / sizeof(__m128);
__u128 gain128;
+
if ((sizeIn % m_frameSize) != 0) {
PDEBUG("%zu != %zu\n", sizeIn, m_frameSize);
throw std::runtime_error(
@@ -287,7 +302,7 @@ __m128 GainControl::computeGainVar(const __m128* in, size_t sizeIn)
var128.m = _mm_sqrt_ps(var128.m);
PDEBUG("********** Var: %10f + %10fj, %10f + %10fj **********\n",
var128.f[0], var128.f[1], var128.f[2], var128.f[3]);
- var128.m = _mm_mul_ps(var128.m, _mm_set1_ps(4.0f));
+ var128.m = _mm_mul_ps(var128.m, _mm_set1_ps(var_variance));
PDEBUG("********** 4*Var: %10f + %10fj, %10f + %10fj **********\n",
var128.f[0], var128.f[1], var128.f[2], var128.f[3]);
@@ -446,7 +461,7 @@ float GainControl::computeGainVar(const complexf* in, size_t sizeIn)
complexf var(sqrt(tmpvar.real()), sqrt(tmpvar.imag()));
PDEBUG("********** Var: %10f + %10fj **********\n", var.real(), var.imag());
- var = var * 4.0f;
+ var = var * var_variance;
PDEBUG("********** 4*Var: %10f + %10fj **********\n", var.real(), var.imag());
////////////////////////////////////////////////////////////////////////////
@@ -480,6 +495,39 @@ void GainControl::set_parameter(const string& parameter, const string& value)
ss >> new_factor;
m_digGain = new_factor;
}
+ else if (parameter == "mode") {
+ string new_mode;
+ ss >> new_mode;
+ std::transform(new_mode.begin(), new_mode.end(), new_mode.begin(),
+ [](const char c) { return std::tolower(c); } );
+
+ GainMode m;
+ if (new_mode == "fix") {
+ m = GainMode::GAIN_FIX;
+ }
+ else if (new_mode == "max") {
+ m = GainMode::GAIN_MAX;
+ }
+ else if (new_mode == "var") {
+ m = GainMode::GAIN_VAR;
+ }
+ else {
+ throw ParameterError("Gainmode " + new_mode + " unknown");
+ }
+
+ {
+ std::lock_guard<std::mutex> lock(m_mutex);
+ m_gainmode = m;
+ }
+ }
+ else if (parameter == "var") {
+ float newvar = 0;
+ ss >> newvar;
+ {
+ std::lock_guard<std::mutex> lock(m_mutex);
+ m_var_variance_rc = newvar;
+ }
+ }
else {
stringstream ss;
ss << "Parameter '" << parameter
@@ -494,6 +542,22 @@ const string GainControl::get_parameter(const string& parameter) const
if (parameter == "digital") {
ss << std::fixed << m_digGain;
}
+ else if (parameter == "mode") {
+ switch (m_gainmode) {
+ case GainMode::GAIN_FIX:
+ ss << "fix";
+ break;
+ case GainMode::GAIN_MAX:
+ ss << "max";
+ break;
+ case GainMode::GAIN_VAR:
+ ss << "var";
+ break;
+ }
+ }
+ else if (parameter == "var") {
+ ss << std::fixed << m_var_variance_rc;
+ }
else {
ss << "Parameter '" << parameter <<
"' is not exported by controllable " << get_rc_name();
diff --git a/src/GainControl.h b/src/GainControl.h
index d686402..8935fc3 100644
--- a/src/GainControl.h
+++ b/src/GainControl.h
@@ -37,6 +37,7 @@
#include <sys/types.h>
#include <complex>
#include <string>
+#include <mutex>
#ifdef __SSE__
# include <xmmintrin.h>
#endif
@@ -58,29 +59,35 @@ class GainControl : public PipelinedModCodec, public RemoteControllable
GainControl(const GainControl&);
GainControl& operator=(const GainControl&);
- const char* name() { return "GainControl"; }
+ const char* name() override { return "GainControl"; }
/* Functions for the remote control */
/* Base function to set parameters. */
virtual void set_parameter(const std::string& parameter,
- const std::string& value);
+ const std::string& value) override;
/* Getting a parameter always returns a string. */
- virtual const std::string get_parameter(const std::string& parameter) const;
+ virtual const std::string get_parameter(
+ const std::string& parameter) const override;
protected:
- virtual int internal_process(Buffer* const dataIn, Buffer* dataOut) override;
+ virtual int internal_process(
+ Buffer* const dataIn, Buffer* dataOut) override;
size_t m_frameSize;
float& m_digGain;
- float m_normalise;
+ float m_normalise;
+
+ // The following variables are accessed from the RC thread
+ float m_var_variance_rc;
+ GainMode m_gainmode;
+ mutable std::mutex m_mutex;
+
#ifdef __SSE__
- __m128 (*computeGain)(const __m128* in, size_t sizeIn);
__m128 static computeGainFix(const __m128* in, size_t sizeIn);
__m128 static computeGainMax(const __m128* in, size_t sizeIn);
__m128 static computeGainVar(const __m128* in, size_t sizeIn);
#else
- float (*computeGain)(const complexf* in, size_t sizeIn);
float static computeGainFix(const complexf* in, size_t sizeIn);
float static computeGainMax(const complexf* in, size_t sizeIn);
float static computeGainVar(const complexf* in, size_t sizeIn);
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-15 02:59:07 +0000