mirror of
https://github.com/f4exb/sdrangel.git
synced 2024-11-03 15:31:15 -05:00
91 lines
3.9 KiB
C++
91 lines
3.9 KiB
C++
///////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (C) 2022 Edouard Griffiths, F4EXB //
|
|
// //
|
|
// 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 as version 3 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 V3 for more details. //
|
|
// //
|
|
// You should have received a copy of the GNU General Public License //
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>. //
|
|
///////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef INCLUDE_DOA2CORR_H
|
|
#define INCLUDE_DOA2CORR_H
|
|
|
|
#include <complex>
|
|
#include <QObject>
|
|
|
|
#include "dsp/dsptypes.h"
|
|
#include "dsp/fftwindow.h"
|
|
#include "util/message.h"
|
|
|
|
#include "doa2settings.h"
|
|
|
|
class FFTEngine;
|
|
|
|
class DOA2Correlator : public QObject {
|
|
Q_OBJECT
|
|
public:
|
|
DOA2Correlator(int fftSize);
|
|
~DOA2Correlator();
|
|
|
|
void setCorrType(DOA2Settings::CorrelationType corrType) { m_corrType = corrType; }
|
|
DOA2Settings::CorrelationType getCorrType() const { return m_corrType; }
|
|
bool performCorr( //!< Returns true if results were produced
|
|
const SampleVector& data0,
|
|
unsigned int size0,
|
|
const SampleVector& data1,
|
|
unsigned int size1
|
|
);
|
|
int getFullFFTSize() const { return 2*m_fftSize; }
|
|
void setPhase(int phase);
|
|
|
|
SampleVector m_tcorr; //!< correlation result (time or spectrum inverse FFT) - Sample vector expected
|
|
std::vector<Complex> m_xcorr; //!< correlation result of inverse FFT of FFT prod with conjugate (performFFTProd) for DOA processing
|
|
int m_processed; //!< number of samples processed at the end of correlation
|
|
int m_remaining[2]; //!< number of samples remaining per member at the end of correlation
|
|
|
|
signals:
|
|
void dataReady(int start, int stop);
|
|
|
|
private:
|
|
bool performOpCorr( //!< Returns true if results were produced
|
|
const SampleVector& data0,
|
|
unsigned int size0,
|
|
const SampleVector& data1,
|
|
unsigned int size1,
|
|
Sample sampleOp(const Sample& a, const Sample& b)
|
|
);
|
|
bool performFFTProd( //!< Returns true if results were produced
|
|
const SampleVector& data0,
|
|
unsigned int size0,
|
|
const SampleVector& data1,
|
|
unsigned int size1
|
|
);
|
|
void adjustTCorrSize(int size);
|
|
void adjustXCorrSize(int size);
|
|
|
|
DOA2Settings::CorrelationType m_corrType;
|
|
unsigned int m_fftSize; //!< FFT length
|
|
FFTEngine *m_fft[2]; //!< FFT engines
|
|
FFTEngine *m_invFFT; //!< Inverse FFT engine
|
|
unsigned int m_fftSequences[2]; //!< FFT engines sequences
|
|
unsigned int m_invFFTSequence; //!< Inverse FFT engine sequence
|
|
FFTWindow m_window; //!< FFT window
|
|
std::complex<float> *m_dataj; //!< conjuate of FFT transform
|
|
SampleVector m_data1p; //!< data1 with phase correction
|
|
int m_tcorrSize; //!< time correlations vector size
|
|
int m_xcorrSize; //!< DOA correlations vector size
|
|
int m_phase; //!< phase correction
|
|
int64_t m_sin; //!< scaled sine of phase correction
|
|
int64_t m_cos; //!< scaled cosine of phase correction
|
|
};
|
|
|
|
#endif // INCLUDE_DOA2CORR_H
|