39 static void reconstruct(
char* filename,
int N,
int M,
int Z,
int iteration,
int weight)
49 double epsilon=0.0000003;
52 unsigned infft_flags = CGNR | PRECOMPUTE_DAMP;
56 my_N[0]=Z; my_n[0]=ceil(Z*1.2);
57 my_N[1]=N; my_n[1]=ceil(N*1.2);
58 my_N[2]=N; my_n[2]=ceil(N*1.2);
59 nfft_init_guru(&my_plan, 3, my_N, M, my_n, 6,
60 PRE_PHI_HUT| PRE_PSI |MALLOC_X| MALLOC_F_HAT|
61 MALLOC_F| FFTW_INIT| FFT_OUT_OF_PLACE,
62 FFTW_MEASURE| FFTW_DESTROY_INPUT);
65 if(my_plan.
flags & PRE_LIN_PSI)
66 nfft_precompute_lin_psi(&my_plan);
69 infft_flags = infft_flags | PRECOMPUTE_WEIGHT;
75 if(my_iplan.
flags & PRECOMPUTE_WEIGHT)
77 fin=fopen(
"weights.dat",
"r");
80 fscanf(fin,
"%le ",&my_iplan.
w[j]);
86 if(my_iplan.
flags & PRECOMPUTE_DAMP)
94 double r=sqrt(j2*j2+k2*k2+z2*z2);
96 my_iplan.
w_hat[z*N*N+j*N+k]=0.0;
98 my_iplan.
w_hat[z*N*N+j*N+k]=1.0;
105 fin=fopen(filename,
"r");
108 fout_real=fopen(
"output_real.dat",
"w");
109 fout_imag=fopen(
"output_imag.dat",
"w");
114 fscanf(fin,
"%le %le %le %le %le ",&my_plan.
x[3*j+1],&my_plan.
x[3*j+2], &my_plan.
x[3*j+0],
116 my_iplan.
y[j] = real + _Complex_I*imag;
120 if(my_plan.
flags & PRE_PSI)
121 nfft_precompute_psi(&my_plan);
124 if(my_plan.
flags & PRE_FULL_PSI)
125 nfft_precompute_full_psi(&my_plan);
132 solver_before_loop_complex(&my_iplan);
133 for(l=0;l<iteration;l++)
138 fprintf(stderr,
"%e, %i of %i\n",sqrt(my_iplan.
dot_r_iter),
140 solver_loop_one_step_complex(&my_iplan);
148 fprintf(fout_real,
"%le ",creal(my_iplan.
f_hat_iter[ k+N*N*l ]));
149 fprintf(fout_imag,
"%le ",cimag(my_iplan.
f_hat_iter[ k+N*N*l ]));
151 fprintf(fout_real,
"\n");
152 fprintf(fout_imag,
"\n");
158 solver_finalize_complex(&my_iplan);
159 nfft_finalize(&my_plan);
162 int main(
int argc,
char **argv)
165 printf(
"usage: ./reconstruct3D FILENAME N M Z ITER WEIGHTS\n");
169 reconstruct(argv[1],atoi(argv[2]),atoi(argv[3]),atoi(argv[4]),atoi(argv[5]),atoi(argv[6]));
double * w
weighting factors
unsigned flags
iteration type
double dot_r_iter
weighted dotproduct of r_iter
data structure for an NFFT (nonequispaced fast Fourier transform) plan with double precision ...
NFFT_INT N_total
Total number of Fourier coefficients.
static void reconstruct(char *filename, int N, int M, int Z, int iteration, int weight)
reconstruct makes an inverse 3d-nfft
fftw_complex * y
right hand side, samples
double * x
Nodes in time/spatial domain, size is doubles.
unsigned flags
Flags for precomputation, (de)allocation, and FFTW usage, default setting is PRE_PHI_HUT | PRE_PSI | ...
data structure for an inverse NFFT plan with double precision
double * w_hat
damping factors
fftw_complex * f_hat_iter
iterative solution