/* nag_ftt_multid_single (c06pfc) Example Program.
 *
 * Copyright 2002 Numerical Algorithms Group.
 *
 * Mark 7, 2002.
 */

#include <nag.h>
#include <stdio.h>
#include <nag_stdlib.h>
#include <nagc06.h>
#include <nagx04.h>

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);
  Vprintf("nag_fft_multid_single (c06pfc) Example Program Results\n");
  /* Skip heading in data file */
  Vscanf("%*[^\n]");
  Vscanf("%ld%ld%ld", &ndim, &l, &n);
  if (n >= 1)
    {
      /* Allocate memory */
      if ( !(x = NAG_ALLOC(n, Complex)) ||
           !(nd = NAG_ALLOC(ndim, Integer)) )
        {
          Vprintf("Allocation failure\n");
          exit_status = -1;
          goto END;
        }
      for (i = 0; i < ndim; ++i)
        {
          Vscanf("%ld",&nd[i]);
        }

      /* Read in complex data and print out. */
      Vscanf("%*[^\n]");
      for (i = 0; i<n; ++i)
        {
          Vscanf(" ( %lf, %lf ) ", &x[i].re, &x[i].im);
        }
      Vscanf("%*[^\n]");
      Vprintf("\n");
      /* nag_gen_complx_mat_print_comp (x04dbc).
       * Print complex general matrix (comprehensive)
       */
      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, 90, 0, 0,
                                    &fail);
      if (fail.code != NE_NOERROR)
        {
          Vprintf("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)
        {
          Vprintf("Error from nag_fft_multid_single (c06pfc).\n%s\n",
                  fail.message);
          exit_status = 1;
          goto END;
        }
      Vprintf("\n");
      /* nag_gen_complx_mat_print_comp (x04dbc), see above. */
      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, 90, 0, 0,
                                    &fail);
      if (fail.code != NE_NOERROR)
        {
          Vprintf("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)
        {
          Vprintf("Error from nag_fft_multid_single (c06pfc).\n%s\n",
                  fail.message);
          exit_status = 1;
          goto END;
        }
      Vprintf("\n");
      /* nag_gen_complx_mat_print_comp (x04dbc), see above. */
      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, 90, 0, 0, &fail);
      if (fail.code != NE_NOERROR)
        {
          Vprintf("Error from nag_gen_complx_mat_print_comp (x04dbc).\n%s\n",
                  fail.message);
          exit_status = 1;
          goto END;
        }
    }
  else
    Vfprintf(stderr,"\nInvalid value of n.\n");

 END:
  if (x) NAG_FREE(x);
  if (nd) NAG_FREE(nd);

  return exit_status;
}