forked from abacusmodeling/abacus-develop
-
Notifications
You must be signed in to change notification settings - Fork 134
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Refactor:Replace the current fft with templates and polymorphism (#5410)
* add the basic func of the file * modify the Makefile * delete file * modify the position of the new fft * modify the Makefile * [pre-commit.ci lite] apply automatic fixes * add the cpu float in the fft floder * change the test file * [pre-commit.ci lite] apply automatic fixes * add the func in test * add the float fft * change ft into ft1 * add the file of the float_define and the device set * delete the memory allocate in the ft * [pre-commit.ci lite] apply automatic fixes * add the Smart Pointer and the logic gate * modify the position of the FFT * change fft_bundle name * save version of the pw_test and single version * fix complie bug and change the fftwf logic * add comments for the fft class * modify the fft name and add comments * modify the Makefile * update the file * update the format * update the shared_ptr * [pre-commit.ci lite] apply automatic fixes --------- Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com> Co-authored-by: Mohan Chen <[email protected]>
- Loading branch information
3 people
authored
Nov 12, 2024
1 parent
8d6e593
commit 7c46674
Showing
27 changed files
with
1,875 additions
and
114 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -6,7 +6,6 @@ list (APPEND LIBM_SRC | |
| libm/sincos.cpp | ||
| ) | ||
| endif() | ||
|
|
||
| add_library( | ||
| base | ||
| OBJECT | ||
|
|
||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,8 @@ | ||
| #include "fft_base.h" | ||
| namespace ModulePW | ||
| { | ||
| template FFT_BASE<float>::FFT_BASE(); | ||
| template FFT_BASE<double>::FFT_BASE(); | ||
| template FFT_BASE<float>::~FFT_BASE(); | ||
| template FFT_BASE<double>::~FFT_BASE(); | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,163 @@ | ||
| #include <complex> | ||
| #include <string> | ||
| #include "fftw3.h" | ||
| #ifndef FFT_BASE_H | ||
| #define FFT_BASE_H | ||
| namespace ModulePW | ||
| { | ||
| template <typename FPTYPE> | ||
| class FFT_BASE | ||
| { | ||
| public: | ||
|
|
||
| FFT_BASE(){}; | ||
| virtual ~FFT_BASE(){}; | ||
|
|
||
| /** | ||
| * @brief Initialize the fft parameters As virtual function. | ||
| * | ||
| * The function is used to initialize the fft parameters. | ||
| */ | ||
| virtual __attribute__((weak)) | ||
| void initfft(int nx_in, | ||
| int ny_in, | ||
| int nz_in, | ||
| int lixy_in, | ||
| int rixy_in, | ||
| int ns_in, | ||
| int nplane_in, | ||
| int nproc_in, | ||
| bool gamma_only_in, | ||
| bool xprime_in = true); | ||
|
|
||
| /** | ||
| * @brief Setup the fft Plan and data As pure virtual function. | ||
| * | ||
| * The function is set as pure virtual function.In order to | ||
| * override the function in the derived class.In the derived | ||
| * class, the function is used to setup the fft Plan and data. | ||
| */ | ||
| virtual void setupFFT()=0; | ||
|
|
||
| /** | ||
| * @brief Clean the fft Plan As pure virtual function. | ||
| * | ||
| * The function is set as pure virtual function.In order to | ||
| * override the function in the derived class.In the derived | ||
| * class, the function is used to clean the fft Plan. | ||
| */ | ||
| virtual void cleanFFT()=0; | ||
|
|
||
| /** | ||
| * @brief Clear the fft data As pure virtual function. | ||
| * | ||
| * The function is set as pure virtual function.In order to | ||
| * override the function in the derived class.In the derived | ||
| * class, the function is used to clear the fft data. | ||
| */ | ||
| virtual void clear()=0; | ||
|
|
||
| /** | ||
| * @brief Get the real space data in cpu-like fft | ||
| * | ||
| * The function is used to get the real space data.While the | ||
| * FFT_BASE is an abstract class,the function will be override, | ||
| * The attribute weak is used to avoid define the function. | ||
| */ | ||
| virtual __attribute__((weak)) | ||
| FPTYPE* get_rspace_data() const; | ||
|
|
||
| virtual __attribute__((weak)) | ||
| std::complex<FPTYPE>* get_auxr_data() const; | ||
|
|
||
| virtual __attribute__((weak)) | ||
| std::complex<FPTYPE>* get_auxg_data() const; | ||
|
|
||
| /** | ||
| * @brief Get the auxiliary real space data in 3D | ||
| * | ||
| * The function is used to get the auxiliary real space data in 3D. | ||
| * While the FFT_BASE is an abstract class,the function will be override, | ||
| * The attribute weak is used to avoid define the function. | ||
| */ | ||
| virtual __attribute__((weak)) | ||
| std::complex<FPTYPE>* get_auxr_3d_data() const; | ||
|
|
||
| //forward fft in x-y direction | ||
|
|
||
| /** | ||
| * @brief Forward FFT in x-y direction | ||
| * @param in input data | ||
| * @param out output data | ||
| * | ||
| * This function performs the forward FFT in the x-y direction. | ||
| * It involves two axes, x and y. The FFT is applied multiple times | ||
| * along the left and right boundaries in the primary direction(which is | ||
| * determined by the xprime flag).Notably, the Y axis operates in | ||
| * "many-many-FFT" mode. | ||
| */ | ||
| virtual __attribute__((weak)) | ||
| void fftxyfor(std::complex<FPTYPE>* in, | ||
| std::complex<FPTYPE>* out) const; | ||
|
|
||
| virtual __attribute__((weak)) | ||
| void fftxybac(std::complex<FPTYPE>* in, | ||
| std::complex<FPTYPE>* out) const; | ||
|
|
||
| /** | ||
| * @brief Forward FFT in z direction | ||
| * @param in input data | ||
| * @param out output data | ||
| * | ||
| * This function performs the forward FFT in the z direction. | ||
| * It involves only one axis, z. The FFT is applied only once. | ||
| * Notably, the Z axis operates in many FFT with nz*ns. | ||
| */ | ||
| virtual __attribute__((weak)) | ||
| void fftzfor(std::complex<FPTYPE>* in, | ||
| std::complex<FPTYPE>* out) const; | ||
|
|
||
| virtual __attribute__((weak)) | ||
| void fftzbac(std::complex<FPTYPE>* in, | ||
| std::complex<FPTYPE>* out) const; | ||
|
|
||
| /** | ||
| * @brief Forward FFT in x-y direction with real to complex | ||
| * @param in input data, real type | ||
| * @param out output data, complex type | ||
| * | ||
| * This function performs the forward FFT in the x-y direction | ||
| * with real to complex.There is no difference between fftxyfor. | ||
| */ | ||
| virtual __attribute__((weak)) | ||
| void fftxyr2c(FPTYPE* in, | ||
| std::complex<FPTYPE>* out) const; | ||
|
|
||
| virtual __attribute__((weak)) | ||
| void fftxyc2r(std::complex<FPTYPE>* in, | ||
| FPTYPE* out) const; | ||
|
|
||
| /** | ||
| * @brief Forward FFT in 3D | ||
| * @param in input data | ||
| * @param out output data | ||
| * | ||
| * This function performs the forward FFT for gpu-like fft. | ||
| * It involves three axes, x, y, and z. The FFT is applied multiple times | ||
| * for fft3D_forward. | ||
| */ | ||
| virtual __attribute__((weak)) | ||
| void fft3D_forward(std::complex<FPTYPE>* in, | ||
| std::complex<FPTYPE>* out) const; | ||
|
|
||
| virtual __attribute__((weak)) | ||
| void fft3D_backward(std::complex<FPTYPE>* in, | ||
| std::complex<FPTYPE>* out) const; | ||
|
|
||
| protected: | ||
| int nx=0; | ||
| int ny=0; | ||
| int nz=0; | ||
| }; | ||
| } | ||
| #endif // FFT_BASE_H |
Oops, something went wrong.