/* nag_fft_multid_single (c06pfc) Example Program. * * Copyright 2002 Numerical Algorithms Group. * * Mark 7, 2002. */ #include #include #include #include #include #include int main(void) { /* Scalars */ Integer i, l, n, ndim; Integer exit_status = 0; NagError fail; /* Arrays */ Complex *x = 0; Integer *nd = 0; INIT_FAIL(fail); printf("nag_fft_multid_single (c06pfc) Example Program Results\n"); /* Skip heading in data file */ scanf("%*[^\n]"); scanf("%ld%ld%ld", &ndim, &l, &n); if (n >= 1) { /* Allocate memory */ if (!(x = NAG_ALLOC(n, Complex)) || !(nd = NAG_ALLOC(ndim, Integer))) { printf("Allocation failure\n"); exit_status = -1; goto END; } for (i = 0; i < ndim; ++i) { scanf("%ld", &nd[i]); } /* Read in complex data and print out. */ scanf("%*[^\n]"); for (i = 0; i < n; ++i) { scanf(" ( %lf, %lf ) ", &x[i].re, &x[i].im); } scanf("%*[^\n]"); printf("\n"); /* nag_gen_complx_mat_print_comp (x04dbc). * Print complex general matrix (comprehensive) */ fflush(stdout); nag_gen_complx_mat_print_comp(Nag_ColMajor, Nag_GeneralMatrix, Nag_NonUnitDiag, nd[0], n/nd[0], x, nd[0], Nag_BracketForm, "%6.3f", "Original data", Nag_NoLabels, 0, Nag_NoLabels, 0, 80, 0, 0, &fail); if (fail.code != NE_NOERROR) { printf( "Error from nag_gen_complx_mat_print_comp (x04dbc).\n%s\n", fail.message); exit_status = 1; goto END; } /* Compute transform */ /* nag_fft_multid_single (c06pfc). * One-dimensional complex discrete Fourier transform of * multi-dimensional data (using complex data type) */ nag_fft_multid_single(Nag_ForwardTransform, ndim, l, nd, n, x, &fail); if (fail.code != NE_NOERROR) { printf("Error from nag_fft_multid_single (c06pfc).\n%s\n", fail.message); exit_status = 1; goto END; } printf("\n"); /* nag_gen_complx_mat_print_comp (x04dbc), see above. */ fflush(stdout); nag_gen_complx_mat_print_comp(Nag_ColMajor, Nag_GeneralMatrix, Nag_NonUnitDiag, nd[0], n/nd[0], x, nd[0], Nag_BracketForm, "%6.3f", "Discrete Fourier transform of variable 2", Nag_NoLabels, 0, Nag_NoLabels, 0, 80, 0, 0, &fail); if (fail.code != NE_NOERROR) { printf( "Error from nag_gen_complx_mat_print_comp (x04dbc).\n%s\n", fail.message); exit_status = 1; goto END; } /* Compute inverse transform */ /* nag_fft_multid_single (c06pfc), see above. */ nag_fft_multid_single(Nag_BackwardTransform, ndim, l, nd, n, x, &fail); if (fail.code != NE_NOERROR) { printf("Error from nag_fft_multid_single (c06pfc).\n%s\n", fail.message); exit_status = 1; goto END; } printf("\n"); /* nag_gen_complx_mat_print_comp (x04dbc), see above. */ fflush(stdout); nag_gen_complx_mat_print_comp(Nag_ColMajor, Nag_GeneralMatrix, Nag_NonUnitDiag, nd[0], n/nd[0], x, nd[0], Nag_BracketForm, "%6.3f", "Original data as restored by inverse" " transform", Nag_NoLabels, 0, Nag_NoLabels, 0, 80, 0, 0, &fail); if (fail.code != NE_NOERROR) { printf( "Error from nag_gen_complx_mat_print_comp (x04dbc).\n%s\n", fail.message); exit_status = 1; goto END; } } else printf("\nInvalid value of n.\n"); END: if (x) NAG_FREE(x); if (nd) NAG_FREE(nd); return exit_status; }