-
Notifications
You must be signed in to change notification settings - Fork 18
/
fftw++.cc
105 lines (88 loc) · 2.18 KB
/
fftw++.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#include <cstring>
#include <sstream>
#include "fftw++.h"
using namespace std;
using namespace utils;
namespace fftwpp {
const double fftw::twopi=2.0*acos(-1.0);
bool fftw::wiser=false;
// User settings:
size_t fftw::effort=FFTW_MEASURE;
string wisdomName="wisdom3.txt";
ostringstream wisdomTemp;
size_t fftw::maxthreads=1;
char *epiphany=NULL;
fftw_plan (*fftw::planner)(fftw *f, Complex *in, Complex *out)=Planner;
const char *fftw::oddshift="Shift is not implemented for odd nx";
const char *inout=
"constructor and call must be both in place or both out of place";
Mfft1d::Table Mfft1d::threadtable;
Mrcfft1d::Table Mrcfft1d::threadtable;
Mcrfft1d::Table Mcrfft1d::threadtable;
static bool Wise=false;
void loadWisdom()
{
if(!Wise) {
wisdomTemp << wisdomName << "_" << getpid();
ifstream ifWisdom;
ifWisdom.open(wisdomName);
ostringstream wisdom;
wisdom << ifWisdom.rdbuf();
ifWisdom.close();
const string& s=wisdom.str();
fftw_import_wisdom_from_string(s.c_str());
Wise=true;
}
}
void recall() {
if(epiphany) {
fftw_import_wisdom_from_string(epiphany);
fftw_free(epiphany);
epiphany=NULL;
}
}
void saveWisdom()
{
if(fftw::wiser) {
fftw_forget_wisdom();
Wise=false;
loadWisdom();
recall();
char *wisdom=fftw_export_wisdom_to_string();
ofstream ofWisdom;
ofWisdom.open(wisdomTemp.str().c_str());
ofWisdom << wisdom;
fftw_free(wisdom);
ofWisdom.close();
rename(wisdomTemp.str().c_str(),wisdomName.c_str());
fftw::wiser=false;
}
}
fftw_plan Planner(fftw *F, Complex *in, Complex *out)
{
loadWisdom();
fftw::effort |= FFTW_WISDOM_ONLY;
fftw_plan plan=F->Plan(in,out);
fftw::effort &= ~FFTW_WISDOM_ONLY;
if(!plan) {
char *experience=fftw_export_wisdom_to_string();
fftw_forget_wisdom();
plan=F->Plan(in,out);
recall();
epiphany=fftw_export_wisdom_to_string();
fftw_import_wisdom_from_string(experience);
fftw_free(experience);
static bool first=true;
if(first) {
atexit(saveWisdom);
first=false;
}
fftw::wiser=true;
}
return plan;
}
ThreadBase::ThreadBase() {threads=fftw::maxthreads;}
}
namespace utils {
size_t defaultmpithreads=1;
}