commit d9f0619ee02dd42a45fc61b102115c03f367c54c
parent 321a94d57562fe8bc2ca16268e98f9174e990431
Author: [email protected] <[email protected]>
Date: Thu, 22 Nov 2018 21:10:34 +0300
DFT C API
Diffstat:
3 files changed, 198 insertions(+), 1 deletion(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -89,5 +89,5 @@ add_library(kfr INTERFACE)
target_sources(kfr INTERFACE ${KFR_SRC})
target_include_directories(kfr INTERFACE include)
-add_library(kfr_dft include/kfr/dft/dft-src.cpp)
+add_library(kfr_dft include/kfr/dft/dft-src.cpp include/kfr/dft/dft_c.h)
target_link_libraries(kfr_dft kfr)
diff --git a/include/kfr/dft/dft-src.cpp b/include/kfr/dft/dft-src.cpp
@@ -24,6 +24,8 @@
See https://www.kfrlib.com for details.
*/
+#include "dft_c.h"
+
#include "../base/basic_expressions.hpp"
#include "../testo/assert.hpp"
#include "bitrev.hpp"
@@ -1253,6 +1255,105 @@ template void dft_plan_real<double>::to_fmt(kfr::complex<double>* out, kfr::dft_
} // namespace kfr
+extern "C"
+{
+
+ KFR_DFT_PLAN_F32* kfr_dft_create_plan_f32(size_t size)
+ {
+ return reinterpret_cast<KFR_DFT_PLAN_F32*>(new kfr::dft_plan<float>(size));
+ }
+ KFR_DFT_PLAN_F64* kfr_dft_create_plan_f64(size_t size)
+ {
+ return reinterpret_cast<KFR_DFT_PLAN_F64*>(new kfr::dft_plan<double>(size));
+ }
+
+ void kfr_dft_execute_f32(KFR_DFT_PLAN_F32* plan, size_t size, float* out, const float* in, uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<float>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<float>*>(out), reinterpret_cast<const kfr::complex<float>*>(in),
+ temp, kfr::cfalse);
+ }
+ void kfr_dft_execute_f64(KFR_DFT_PLAN_F64* plan, size_t size, double* out, const double* in,
+ uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<double>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<double>*>(out), reinterpret_cast<const kfr::complex<double>*>(in),
+ temp, kfr::cfalse);
+ }
+ void kfr_dft_execute_inverse_f32(KFR_DFT_PLAN_F32* plan, size_t size, float* out, const float* in,
+ uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<float>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<float>*>(out), reinterpret_cast<const kfr::complex<float>*>(in),
+ temp, kfr::ctrue);
+ }
+ void kfr_dft_execute_inverse_f64(KFR_DFT_PLAN_F64* plan, size_t size, double* out, const double* in,
+ uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<double>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<double>*>(out), reinterpret_cast<const kfr::complex<double>*>(in),
+ temp, kfr::ctrue);
+ }
+
+ void kfr_dft_delete_plan_f32(KFR_DFT_PLAN_F32* plan)
+ {
+ delete reinterpret_cast<kfr::dft_plan<float>*>(plan);
+ }
+ void kfr_dft_delete_plan_f64(KFR_DFT_PLAN_F64* plan)
+ {
+ delete reinterpret_cast<kfr::dft_plan<double>*>(plan);
+ }
+
+ // Real DFT plans
+
+ KFR_DFT_REAL_PLAN_F32* kfr_dft_create_real_plan_f32(size_t size)
+ {
+ return reinterpret_cast<KFR_DFT_REAL_PLAN_F32*>(new kfr::dft_plan_real<float>(size));
+ }
+ KFR_DFT_REAL_PLAN_F64* kfr_dft_create_real_plan_f64(size_t size)
+ {
+ return reinterpret_cast<KFR_DFT_REAL_PLAN_F64*>(new kfr::dft_plan_real<double>(size));
+ }
+
+ void kfr_dft_execute_real_f32(KFR_DFT_REAL_PLAN_F32* plan, size_t size, float* out, const float* in,
+ uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<float>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<float>*>(out), reinterpret_cast<const kfr::complex<float>*>(in),
+ temp, kfr::cfalse);
+ }
+ void kfr_dft_execute_real_f64(KFR_DFT_REAL_PLAN_F64* plan, size_t size, double* out, const double* in,
+ uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<double>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<double>*>(out), reinterpret_cast<const kfr::complex<double>*>(in),
+ temp, kfr::cfalse);
+ }
+ void kfr_dft_execute_real_inverse_f32(KFR_DFT_REAL_PLAN_F32* plan, size_t size, float* out,
+ const float* in, uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<float>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<float>*>(out), reinterpret_cast<const kfr::complex<float>*>(in),
+ temp, kfr::ctrue);
+ }
+ void kfr_dft_execute_real_inverse__f64(KFR_DFT_REAL_PLAN_F64* plan, size_t size, double* out,
+ const double* in, uint8_t* temp)
+ {
+ reinterpret_cast<kfr::dft_plan<double>*>(plan)->execute(
+ reinterpret_cast<kfr::complex<double>*>(out), reinterpret_cast<const kfr::complex<double>*>(in),
+ temp, kfr::ctrue);
+ }
+
+ void kfr_dft_delete_real_plan_f32(KFR_DFT_REAL_PLAN_F32* plan)
+ {
+ delete reinterpret_cast<kfr::dft_plan_real<float>*>(plan);
+ }
+ void kfr_dft_delete_real_plan_f64(KFR_DFT_REAL_PLAN_F64* plan)
+ {
+ delete reinterpret_cast<kfr::dft_plan_real<double>*>(plan);
+ }
+}
+
CMT_PRAGMA_GNU(GCC diagnostic pop)
CMT_PRAGMA_MSVC(warning(pop))
diff --git a/include/kfr/dft/dft_c.h b/include/kfr/dft/dft_c.h
@@ -0,0 +1,96 @@
+/** @addtogroup dft
+ * @{
+ */
+/*
+ Copyright (C) 2016 D Levin (https://www.kfrlib.com)
+ This file is part of KFR
+
+ KFR 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 3 of the License, or
+ (at your option) any later version.
+
+ KFR 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 KFR.
+
+ If GPL is not suitable for your project, you must purchase a commercial license to use KFR.
+ Buying a commercial license is mandatory as soon as you develop commercial activities without
+ disclosing the source code of your own applications.
+ See https://www.kfrlib.com for details.
+ */
+#pragma once
+
+#include <stdint.h>
+
+#ifdef __cplusplus
+extern "C"
+{
+#endif
+
+ typedef struct KFR_DFT_PLAN_F32
+ {
+ size_t size;
+ size_t temp_size;
+ } KFR_DFT_PLAN_F32;
+ typedef struct KFR_DFT_PLAN_F64
+ {
+ size_t size;
+ size_t temp_size;
+ } KFR_DFT_PLAN_F64;
+
+ typedef struct KFR_DFT_REAL_PLAN_F32
+ {
+ size_t dummy;
+ size_t temp_size;
+ size_t size;
+ } KFR_DFT_REAL_PLAN_F32;
+ typedef struct KFR_DFT_REAL_PLAN_F64
+ {
+ size_t dummy;
+ size_t temp_size;
+ size_t size;
+ } KFR_DFT_REAL_PLAN_F64;
+
+ // Complex DFT plans
+
+ KFR_DFT_PLAN_F32* kfr_dft_create_plan_f32(size_t size);
+ KFR_DFT_PLAN_F64* kfr_dft_create_plan_f64(size_t size);
+
+ void kfr_dft_execute_f32(KFR_DFT_PLAN_F32* plan, size_t size, float* out, const float* in, uint8_t* temp);
+ void kfr_dft_execute_f64(KFR_DFT_PLAN_F64* plan, size_t size, double* out, const double* in,
+ uint8_t* temp);
+
+ void kfr_dft_execute_inverse_f32(KFR_DFT_PLAN_F32* plan, size_t size, float* out, const float* in,
+ uint8_t* temp);
+ void kfr_dft_execute_inverse_f64(KFR_DFT_PLAN_F64* plan, size_t size, double* out, const double* in,
+ uint8_t* temp);
+
+ void kfr_dft_delete_plan_f32(KFR_DFT_PLAN_F32* plan);
+ void kfr_dft_delete_plan_f64(KFR_DFT_PLAN_F64* plan);
+
+ // Real DFT plans
+
+ KFR_DFT_REAL_PLAN_F32* kfr_dft_create_real_plan_f32(size_t size);
+ KFR_DFT_REAL_PLAN_F64* kfr_dft_create_real_plan_f64(size_t size);
+
+ void kfr_dft_execute_real_f32(KFR_DFT_REAL_PLAN_F32* plan, size_t size, float* out, const float* in,
+ uint8_t* temp);
+ void kfr_dft_execute_real_f64(KFR_DFT_REAL_PLAN_F64* plan, size_t size, double* out, const double* in,
+ uint8_t* temp);
+
+ void kfr_dft_execute_real_inverse_f32(KFR_DFT_REAL_PLAN_F32* plan, size_t size, float* out,
+ const float* in, uint8_t* temp);
+ void kfr_dft_execute_real_inverse_f64(KFR_DFT_REAL_PLAN_F64* plan, size_t size, double* out,
+ const double* in, uint8_t* temp);
+
+ void kfr_dft_delete_real_plan_f32(KFR_DFT_REAL_PLAN_F32* plan);
+ void kfr_dft_delete_real_plan_f64(KFR_DFT_REAL_PLAN_F64* plan);
+
+#ifdef __cplusplus
+}
+#endif
