/* nag_tsa_gain_phase_bivar (g13cfc) Example Program.
 *
 * Copyright 1996 Numerical Algorithms Group.
 *
 * Mark 4, 1996.
 * Mark 8 revised, 2004.
 *
 */

#include <nag.h>
#include <nagx04.h>
#include <stdio.h>
#include <nag_stdlib.h>
#include <naga02.h>
#include <nagg13.h>

#define L      80
#define KC     8*L
#define NGMAX  KC
#define NXYMAX 300

int main(int argc, char *argv[])
{
  FILE     *fpin, *fpout;

  Complex  *xyg;
  Integer  exit_status = 0, i, is, j, kc = KC, l = L, mw, ng, nxy;
  NagError fail;
  double   *gn = 0, *gnlw = 0, *gnup = 0, *ph = 0, *phlw = 0, *phup = 0, pw,
            pxy, *stats = 0;
  double   *x = 0, *xg, *y = 0, *yg;

  INIT_FAIL(fail);

  /* Check for command-line IO options */
  fpin = nag_example_file_io(argc, argv, "-data", NULL);
  fpout = nag_example_file_io(argc, argv, "-results", NULL);
  fprintf(fpout, "nag_tsa_gain_phase_bivar (g13cfc) Example Program Results\n");

  /* Skip heading in data file */
  fscanf(fpin, "%*[^\n] ");
  fscanf(fpin, "%ld ", &nxy);
  if (nxy > 0 && nxy <= NXYMAX)
    {
      if (!(stats = NAG_ALLOC(4, double)) ||
          !(x = NAG_ALLOC(KC, double)) ||
          !(y = NAG_ALLOC(KC, double)) ||
          !(gnlw = NAG_ALLOC(NGMAX, double)) ||
          !(gnup = NAG_ALLOC(NGMAX, double)) ||
          !(phlw = NAG_ALLOC(NGMAX, double)) ||
          !(phup = NAG_ALLOC(NGMAX, double)) ||
          !(gn = NAG_ALLOC(NGMAX, double)) ||
          !(ph = NAG_ALLOC(NGMAX, double)))
        {
          fprintf(fpout, "Allocation failure\n");
          exit_status = -1;
          goto END;
        }
      for (i = 1; i <= nxy; ++i)
        fscanf(fpin, "%lf ", &x[i - 1]);
      for (i = 1; i <= nxy; ++i)
        fscanf(fpin, "%lf ", &y[i - 1]);

      /* Set parameters for call to nag_tsa_spectrum_univar (g13cbc) and g13cdc
       *  with mean correction and 10 percent taper
       */
      pxy = 0.1;
      /* Window shape parameter and zero covariance at lag 16 */
      pw = 0.5;
      mw = 16;
      /* Alignment shift of 3 */
      is = 3;

      /* Obtain univariate spectrum for the x and the y series */
      /* nag_tsa_spectrum_univar (g13cbc).
       * Univariate time series, smoothed sample spectrum using
       * spectral smoothing by the trapezium frequency (Daniell)
       * window
       */
      nag_tsa_spectrum_univar(nxy, Nag_Mean, pxy, mw, pw, l, kc, Nag_Unlogged,
                              x, &xg, &ng, stats, &fail);
      if (fail.code != NE_NOERROR)
        {
          fprintf(fpout, "Error from nag_tsa_spectrum_univar (g13cbc).\n%s\n",
                  fail.message);
          exit_status = 1;
          goto END;
        }

      /* nag_tsa_spectrum_univar (g13cbc), see above. */
      nag_tsa_spectrum_univar(nxy, Nag_Mean, pxy, mw, pw, l, kc, Nag_Unlogged,
                              y, &yg, &ng, stats, &fail);
      if (fail.code != NE_NOERROR)
        {
          fprintf(fpout, "Error from nag_tsa_spectrum_univar (g13cbc).\n%s\n",
                  fail.message);
          exit_status = 1;
          goto END;
        }

      /* Obtain cross spectrum of the bivariate series */
      /* nag_tsa_spectrum_bivar (g13cdc).
       * Multivariate time series, smoothed sample cross spectrum
       * using spectral smoothing by the trapezium frequency
       * (Daniell) window
       */
      nag_tsa_spectrum_bivar(nxy, Nag_Mean, pxy, mw, is, pw, l, kc, x, y, &xyg,
                             &ng, &fail);
      if (fail.code != NE_NOERROR)
        {
          fprintf(fpout, "Error from nag_tsa_spectrum_bivar (g13cdc).\n%s\n",
                  fail.message);
          exit_status = 1;
          goto END;
        }

      /* nag_tsa_gain_phase_bivar (g13cfc).
       * Multivariate time series, gain, phase, bounds, univariate
       * and bivariate (cross) spectra
       */
      nag_tsa_gain_phase_bivar(xg, yg, xyg, ng, stats, gn, gnlw, gnup, ph,
                               phlw, phup, &fail);
      if (fail.code != NE_NOERROR)
        {
          fprintf(fpout, "Error from nag_tsa_gain_phase_bivar (g13cfc).\n%s\n",
                  fail.message);
          exit_status = 1;
          goto END;
        }

      fprintf(fpout, "\n");
      fprintf(fpout, "               The gain\n\n");
      fprintf(fpout, "                       Lower     Upper\n");
      fprintf(fpout, "           Value       bound     bound\n\n");
      for (j = 1; j <= ng; ++j)
        fprintf(fpout, "%6ld %10.4f %10.4f %10.4f\n",
                j - 1, gn[j - 1], gnlw[j - 1], gnup[j - 1]);
      fprintf(fpout, "\n              The phase\n\n");
      fprintf(fpout, "                       Lower     Upper\n");
      fprintf(fpout, "           Value       bound     bound\n\n");
      for (j = 1; j <= ng; ++j)
        fprintf(fpout, "%6ld %10.4f %10.4f %10.4f\n",
                j - 1, ph[j - 1], phlw[j - 1], phup[j - 1]);
    }
  NAG_FREE(xg);
  NAG_FREE(yg);
  NAG_FREE(xyg);
 END:
  if (fpin != stdin) fclose(fpin);
  if (fpout != stdout) fclose(fpout);
  if (stats) NAG_FREE(stats);
  if (x) NAG_FREE(x);
  if (y) NAG_FREE(y);
  if (gnlw) NAG_FREE(gnlw);
  if (gnup) NAG_FREE(gnup);
  if (phlw) NAG_FREE(phlw);
  if (phup) NAG_FREE(phup);
  if (gn) NAG_FREE(gn);
  if (ph) NAG_FREE(ph);
  return exit_status;
}