1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-09-21 12:26:34 -04:00
sdrangel/wdsp/phrot.cpp
2024-08-01 00:31:28 +02:00

166 lines
4.1 KiB
C++

/* phrot.c
This file is part of a program that implements a Software-Defined Radio.
Copyright (C) 2014, 2022, 2023 Warren Pratt, NR0V
Copyright (C) 2024 Edouard Griffiths, F4EXB Adapted to SDRangel
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
The author can be reached by email at
warren@wpratt.com
*/
#include "comm.hpp"
#include "phrot.hpp"
#include "RXA.hpp"
#include "TXA.hpp"
namespace WDSP {
/********************************************************************************************************
* *
* Phase Rotator *
* *
********************************************************************************************************/
void PHROT::calc()
{
double g;
x0.resize(nstages); // (float *) malloc0 (nstages * sizeof (float));
x1.resize(nstages); // (float *) malloc0 (nstages * sizeof (float));
y0.resize(nstages); // (float *) malloc0 (nstages * sizeof (float));
y1.resize(nstages); // (float *) malloc0 (nstages * sizeof (float));
g = tan (PI * fc / (float)rate);
b0 = (g - 1.0) / (g + 1.0);
b1 = 1.0;
a1 = b0;
}
PHROT::PHROT(
int _run,
int _size,
float* _in,
float* _out,
int _rate,
double _fc,
int _nstages
) :
reverse(0),
run(_run),
size(_size),
in(_in),
out(_out),
rate(_rate),
fc(_fc),
nstages(_nstages)
{
calc();
}
void PHROT::flush()
{
std::fill (x0.begin(), x0.end(), 0);
std::fill (x1.begin(), x1.end(), 0);
std::fill (y0.begin(), y0.end(), 0);
std::fill (y1.begin(), y1.end(), 0);
}
void PHROT::execute()
{
if (reverse)
{
for (int i = 0; i < size; i++)
in[2 * i + 0] = -in[2 * i + 0];
}
if (run)
{
for (int i = 0; i < size; i++)
{
x0[0] = in[2 * i + 0];
for (int n = 0; n < nstages; n++)
{
if (n > 0) x0[n] = y0[n - 1];
y0[n] = b0 * x0[n]
+ b1 * x1[n]
- a1 * y1[n];
y1[n] = y0[n];
x1[n] = x0[n];
}
out[2 * i + 0] = y0[nstages - 1];
}
}
else if (out != in)
{
std::copy( in, in + size * 2, out);
}
}
void PHROT::setBuffers(float* _in, float* _out)
{
in = _in;
out = _out;
}
void PHROT::setSamplerate(int _rate)
{
rate = _rate;
calc();
}
void PHROT::setSize(int _size)
{
size = _size;
flush();
}
/********************************************************************************************************
* *
* TXA Properties *
* *
********************************************************************************************************/
void PHROT::setRun(int run)
{
run = run;
if (run)
flush();
}
void PHROT::setCorner(double corner)
{
fc = corner;
calc();
}
void PHROT::setNstages(int _nstages)
{
nstages = _nstages;
calc();
}
void PHROT::setReverse(int _reverse)
{
reverse = _reverse;
}
} // namespace WDSP