diff options
Diffstat (limited to 'src/MemlessPoly.cpp')
| -rw-r--r-- | src/MemlessPoly.cpp | 45 |
1 files changed, 19 insertions, 26 deletions
diff --git a/src/MemlessPoly.cpp b/src/MemlessPoly.cpp index 28f4b4a..ccb3a7b 100644 --- a/src/MemlessPoly.cpp +++ b/src/MemlessPoly.cpp @@ -43,6 +43,7 @@ #include <iostream> #include <fstream> #include <memory> +#include <complex> using namespace std; @@ -79,7 +80,7 @@ MemlessPoly::MemlessPoly(const std::string& coefs_file, unsigned int num_threads void MemlessPoly::load_coefficients(const std::string &coefFile) { - std::vector<complexf> coefs; + std::vector<float> coefs; std::ifstream coef_fstream(coefFile.c_str()); if (!coef_fstream) { throw std::runtime_error("MemlessPoly: Could not open file with coefs!"); @@ -100,10 +101,9 @@ void MemlessPoly::load_coefficients(const std::string &coefFile) coefs.resize(n_coefs); for (int n = 0; n < n_coefs; n++) { - float a, b; + float a; coef_fstream >> a; - coef_fstream >> b; - coefs[n] = complexf(a, b); + coefs[n] = a; if (coef_fstream.eof()) { etiLog.log(error, "MemlessPoly: file %s should contains %d coefs, " @@ -125,44 +125,37 @@ void MemlessPoly::load_coefficients(const std::string &coefFile) */ static void apply_coeff( - const vector<complexf> &coefs, + const vector<float> &coefs, const complexf *__restrict in, size_t start, size_t stop, complexf *__restrict out) { for (size_t i = start; i < stop; i++) { /* Implement - a0 + a1*x + a2*x^2 + a3*x^3 + a4*x^4 + a5*x^5; + a1*x + a3*x*|x|^2 + a5*x*|x|^4 + a5*x*|x|^4 + a5*x*|x|^4 + a7*x*|x|^6 with less multiplications: a0 + x*(a1 + x*(a2 + x*(a3 + x*(a3 + x*(a4 + a5*x))))); */ /* Make sure to adapt NUM_COEFS when you change this */ -#if 1 // Complex polynomial, all operations are on complex values. // Usually this is the representation we use when speaking // about the real-valued passband signal that the PA receives. - out[i] = - coefs[0] + in[i] * - ( coefs[1] + in[i] * - ( coefs[2] + in[i] * - ( coefs[3] + in[i] * - ( coefs[4] + in[i] * - ( coefs[5] + in[i] ))))); -#else - // Polynomial on the magnitude only. All but one multiplication are - // on real values. This is what is usually used in papers for the - // baseband representation of the signal. - const float mag = std::abs(in[i]); + float in_mag = std::abs(in[i]); + float in_2 = in_mag * in_mag; + float in_4 = in_2 * in_2; + float in_6 = in_2 * in_4; + float in_8 = in_4 * in_4; + float in_10 = in_6 * in_4; out[i] = in[i] * - mag * (coefs[0] + - mag * (coefs[1] + - mag * (coefs[2] + - mag * (coefs[3] + - mag * (coefs[4] + - mag * coefs[5] ))))); -#endif + ( + coefs[0] * in_2 + + coefs[1] * in_4 + + coefs[2] * in_6 + + coefs[3] * in_8 + + coefs[4] * in_10 + ); } } |