mirror of
				https://github.com/f4exb/sdrangel.git
				synced 2025-10-30 20:40:20 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			282 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			282 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*  bandpass.c
 | |
| 
 | |
| This file is part of a program that implements a Software-Defined Radio.
 | |
| 
 | |
| Copyright (C) 2013, 2016, 2017 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 <QRecursiveMutex>
 | |
| 
 | |
| #include "comm.hpp"
 | |
| #include "bps.hpp"
 | |
| #include "fir.hpp"
 | |
| #include "bandpass.hpp"
 | |
| #include "RXA.hpp"
 | |
| #include "TXA.hpp"
 | |
| 
 | |
| namespace WDSP {
 | |
| 
 | |
| /********************************************************************************************************
 | |
| *                                                                                                       *
 | |
| *                                       Overlap-Save Bandpass                                           *
 | |
| *                                                                                                       *
 | |
| ********************************************************************************************************/
 | |
| 
 | |
| void BPS::calc_bps (BPS *a)
 | |
| {
 | |
|     double* impulse;
 | |
|     a->infilt = new double[2 * a->size * 2]; // (double *)malloc0(2 * a->size * sizeof(dcomplex));
 | |
|     a->product = new double[2 * a->size * 2]; // (double *)malloc0(2 * a->size * sizeof(dcomplex));
 | |
|     impulse = FIR::fir_bandpass(a->size + 1, a->f_low, a->f_high, a->samplerate, a->wintype, 1, 1.0 / (double)(2 * a->size));
 | |
|     a->mults = FIR::fftcv_mults(2 * a->size, impulse);
 | |
|     a->CFor = fftw_plan_dft_1d(2 * a->size, (fftw_complex *)a->infilt, (fftw_complex *)a->product, FFTW_FORWARD, FFTW_PATIENT);
 | |
|     a->CRev = fftw_plan_dft_1d(2 * a->size, (fftw_complex *)a->product, (fftw_complex *)a->out, FFTW_BACKWARD, FFTW_PATIENT);
 | |
|     delete[](impulse);
 | |
| }
 | |
| 
 | |
| void BPS::decalc_bps (BPS *a)
 | |
| {
 | |
|     fftw_destroy_plan(a->CRev);
 | |
|     fftw_destroy_plan(a->CFor);
 | |
|     delete[] (a->mults);
 | |
|     delete[] (a->product);
 | |
|     delete[] (a->infilt);
 | |
| }
 | |
| 
 | |
| BPS* BPS::create_bps (int run, int position, int size, double* in, double* out,
 | |
|     double f_low, double f_high, int samplerate, int wintype, double gain)
 | |
| {
 | |
|     BPS *a = new BPS;
 | |
|     a->run = run;
 | |
|     a->position = position;
 | |
|     a->size = size;
 | |
|     a->samplerate = (double)samplerate;
 | |
|     a->wintype = wintype;
 | |
|     a->gain = gain;
 | |
|     a->in = in;
 | |
|     a->out = out;
 | |
|     a->f_low = f_low;
 | |
|     a->f_high = f_high;
 | |
|     calc_bps (a);
 | |
|     return a;
 | |
| }
 | |
| 
 | |
| void BPS::destroy_bps (BPS *a)
 | |
| {
 | |
|     decalc_bps (a);
 | |
|     delete a;
 | |
| }
 | |
| 
 | |
| void BPS::flush_bps (BPS *a)
 | |
| {
 | |
|     memset (a->infilt, 0, 2 * a->size * sizeof (dcomplex));
 | |
| }
 | |
| 
 | |
| void BPS::xbps (BPS *a, int pos)
 | |
| {
 | |
|     int i;
 | |
|     double I, Q;
 | |
|     if (a->run && pos == a->position)
 | |
|     {
 | |
|         memcpy (&(a->infilt[2 * a->size]), a->in, a->size * sizeof (dcomplex));
 | |
|         fftw_execute (a->CFor);
 | |
|         for (i = 0; i < 2 * a->size; i++)
 | |
|         {
 | |
|             I = a->gain * a->product[2 * i + 0];
 | |
|             Q = a->gain * a->product[2 * i + 1];
 | |
|             a->product[2 * i + 0] = I * a->mults[2 * i + 0] - Q * a->mults[2 * i + 1];
 | |
|             a->product[2 * i + 1] = I * a->mults[2 * i + 1] + Q * a->mults[2 * i + 0];
 | |
|         }
 | |
|         fftw_execute (a->CRev);
 | |
|         memcpy (a->infilt, &(a->infilt[2 * a->size]), a->size * sizeof(dcomplex));
 | |
|     }
 | |
|     else if (a->in != a->out)
 | |
|         memcpy (a->out, a->in, a->size * sizeof (dcomplex));
 | |
| }
 | |
| 
 | |
| void BPS::setBuffers_bps (BPS *a, double* in, double* out)
 | |
| {
 | |
|     decalc_bps (a);
 | |
|     a->in = in;
 | |
|     a->out = out;
 | |
|     calc_bps (a);
 | |
| }
 | |
| 
 | |
| void BPS::setSamplerate_bps (BPS *a, int rate)
 | |
| {
 | |
|     decalc_bps (a);
 | |
|     a->samplerate = rate;
 | |
|     calc_bps (a);
 | |
| }
 | |
| 
 | |
| void BPS::setSize_bps (BPS *a, int size)
 | |
| {
 | |
|     decalc_bps (a);
 | |
|     a->size = size;
 | |
|     calc_bps (a);
 | |
| }
 | |
| 
 | |
| void BPS::setFreqs_bps (BPS *a, double f_low, double f_high)
 | |
| {
 | |
|     decalc_bps (a);
 | |
|     a->f_low = f_low;
 | |
|     a->f_high = f_high;
 | |
|     calc_bps (a);
 | |
| }
 | |
| 
 | |
| /********************************************************************************************************
 | |
| *                                                                                                       *
 | |
| *                               Overlap-Save Bandpass:  RXA Properties                                  *
 | |
| *                                                                                                       *
 | |
| ********************************************************************************************************/
 | |
| 
 | |
| void BPS::SetBPSRun (RXA& rxa, int run)
 | |
| {
 | |
|     rxa.csDSP.lock();
 | |
|     rxa.bp1.p->run = run;
 | |
|     rxa.csDSP.unlock();
 | |
| }
 | |
| 
 | |
| void BPS::SetBPSFreqs (RXA& rxa, double f_low, double f_high)
 | |
| {
 | |
|     double* impulse;
 | |
|     BPS *a1;
 | |
|     rxa.csDSP.lock();
 | |
|     a1 = rxa.bps1.p;
 | |
|     if ((f_low != a1->f_low) || (f_high != a1->f_high))
 | |
|     {
 | |
|         a1->f_low = f_low;
 | |
|         a1->f_high = f_high;
 | |
|         delete[] (a1->mults);
 | |
|         impulse = FIR::fir_bandpass(a1->size + 1, f_low, f_high, a1->samplerate, a1->wintype, 1, 1.0 / (double)(2 * a1->size));
 | |
|         a1->mults = FIR::fftcv_mults (2 * a1->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     rxa.csDSP.unlock();
 | |
| }
 | |
| 
 | |
| void BPS::SetBPSWindow (RXA& rxa, int wintype)
 | |
| {
 | |
|     double* impulse;
 | |
|     BPS *a1;
 | |
|     rxa.csDSP.lock();
 | |
|     a1 = rxa.bps1.p;
 | |
|     if ((a1->wintype != wintype))
 | |
|     {
 | |
|         a1->wintype = wintype;
 | |
|         delete[] (a1->mults);
 | |
|         impulse = FIR::fir_bandpass(a1->size + 1, a1->f_low, a1->f_high, a1->samplerate, a1->wintype, 1, 1.0 / (double)(2 * a1->size));
 | |
|         a1->mults = FIR::fftcv_mults (2 * a1->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     rxa.csDSP.unlock();
 | |
| }
 | |
| 
 | |
| /********************************************************************************************************
 | |
| *                                                                                                       *
 | |
| *                                           TXA Properties                                              *
 | |
| *                                                                                                       *
 | |
| ********************************************************************************************************/
 | |
| // UNCOMMENT properties when pointers in place in txa
 | |
| void BPS::SetBPSRun (TXA& txa, int run)
 | |
| {
 | |
|     txa.csDSP.lock();
 | |
|     txa.bp1.p->run = run;
 | |
|     txa.csDSP.unlock();
 | |
| }
 | |
| 
 | |
| void BPS::SetBPSFreqs (TXA& txa, double f_low, double f_high)
 | |
| {
 | |
|     double* impulse;
 | |
|     BPS *a;
 | |
|     txa.csDSP.lock();
 | |
|     a = txa.bps0.p;
 | |
|     if ((f_low != a->f_low) || (f_high != a->f_high))
 | |
|     {
 | |
|         a->f_low = f_low;
 | |
|         a->f_high = f_high;
 | |
|         delete[] (a->mults);
 | |
|         impulse = FIR::fir_bandpass(a->size + 1, f_low, f_high, a->samplerate, a->wintype, 1, 1.0 / (double)(2 * a->size));
 | |
|         a->mults = FIR::fftcv_mults (2 * a->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     a = txa.bps1.p;
 | |
|     if ((f_low != a->f_low) || (f_high != a->f_high))
 | |
|     {
 | |
|         a->f_low = f_low;
 | |
|         a->f_high = f_high;
 | |
|         delete[] (a->mults);
 | |
|         impulse = FIR::fir_bandpass(a->size + 1, f_low, f_high, a->samplerate, a->wintype, 1, 1.0 / (double)(2 * a->size));
 | |
|         a->mults = FIR::fftcv_mults (2 * a->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     a = txa.bps2.p;
 | |
|     if ((f_low != a->f_low) || (f_high != a->f_high))
 | |
|     {
 | |
|         a->f_low = f_low;
 | |
|         a->f_high = f_high;
 | |
|         delete[] (a->mults);
 | |
|         impulse = FIR::fir_bandpass(a->size + 1, f_low, f_high, a->samplerate, a->wintype, 1, 1.0 / (double)(2 * a->size));
 | |
|         a->mults = FIR::fftcv_mults (2 * a->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     txa.csDSP.unlock();
 | |
| }
 | |
| 
 | |
| void BPS::SetBPSWindow (TXA& txa, int wintype)
 | |
| {
 | |
|     double* impulse;
 | |
|     BPS *a;
 | |
|     txa.csDSP.lock();
 | |
|     a = txa.bps0.p;
 | |
|     if (a->wintype != wintype)
 | |
|     {
 | |
|         a->wintype = wintype;
 | |
|         delete[] (a->mults);
 | |
|         impulse = FIR::fir_bandpass(a->size + 1, a->f_low, a->f_high, a->samplerate, a->wintype, 1, 1.0 / (double)(2 * a->size));
 | |
|         a->mults = FIR::fftcv_mults (2 * a->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     a = txa.bps1.p;
 | |
|     if (a->wintype != wintype)
 | |
|     {
 | |
|         a->wintype = wintype;
 | |
|         delete[] (a->mults);
 | |
|         impulse = FIR::fir_bandpass(a->size + 1, a->f_low, a->f_high, a->samplerate, a->wintype, 1, 1.0 / (double)(2 * a->size));
 | |
|         a->mults = FIR::fftcv_mults (2 * a->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     a = txa.bps2.p;
 | |
|     if (a->wintype != wintype)
 | |
|     {
 | |
|         a->wintype = wintype;
 | |
|         delete[] (a->mults);
 | |
|         impulse = FIR::fir_bandpass (a->size + 1, a->f_low, a->f_high, a->samplerate, a->wintype, 1, 1.0 / (double)(2 * a->size));
 | |
|         a->mults = FIR::fftcv_mults (2 * a->size, impulse);
 | |
|         delete[] (impulse);
 | |
|     }
 | |
|     txa.csDSP.unlock();
 | |
| }
 | |
| 
 | |
| } // namespace WDSP
 |