1
0
mirror of https://github.com/f4exb/sdrangel.git synced 2024-11-22 08:04:49 -05:00
sdrangel/sdrbase/dsp/vkfftengine.h

75 lines
2.7 KiB
C
Raw Normal View History

2023-08-13 18:52:57 -04:00
///////////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2012 maintech GmbH, Otto-Hahn-Str. 15, 97204 Hoechberg, Germany //
// written by Christian Daniel //
// Copyright (C) 2015-2016, 2018, 2020 Edouard Griffiths, F4EXB <f4exb06@gmail.com> //
// Copyright (C) 2023 Jon Beniston, M7RCE <jon@beniston.com> //
2023-08-13 18:52:57 -04:00
// //
// 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_VKFFTENGINE_H
#define INCLUDE_VKFFTENGINE_H
#include <QMutex>
2023-08-14 05:37:27 -04:00
#include <vkFFT.h>
2023-08-13 18:52:57 -04:00
#include "dsp/fftengine.h"
#include "dsp/vkfftutils.h"
#include "export.h"
class SDRBASE_API vkFFTEngine : public FFTEngine {
public:
vkFFTEngine();
virtual ~vkFFTEngine();
virtual void configure(int n, bool inverse);
virtual Complex* in();
virtual Complex* out();
virtual void setReuse(bool reuse) { m_reuse = reuse; }
bool isAvailable() override;
protected:
static QMutex m_globalPlanMutex;
struct Plan {
Plan();
virtual ~Plan();
int n;
uint64_t m_bufferSize;
bool m_inverse;
VkFFTConfiguration* m_configuration;
VkFFTApplication* m_app;
Complex* m_in; // CPU memory
Complex* m_out;
};
QList<Plan *> m_plans;
Plan* m_currentPlan;
bool m_reuse;
VkGPU *vkGPU;
virtual VkFFTResult gpuInit() = 0;
virtual VkFFTResult gpuAllocateBuffers() = 0;
virtual VkFFTResult gpuConfigure() = 0;
virtual Plan *gpuAllocatePlan() = 0;
virtual void gpuDeallocatePlan(Plan *plan) = 0;
void freeAll();
};
#endif // INCLUDE_VKFFTENGINE_H