/* nag_tsa_transf_prelim_fit (g13bdc) Example Program.
 *
 * Copyright 2001 Numerical Algorithms Group.
 *
 * Mark 7, 2001.
 */

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

int main(void)
{
  /* Scalars */
  double r0, s;
  Integer exit_status, i, iwa, nl, nwds;

  /* Arrays */
  double *r=0, *wa=0, *wds=0;
  Integer isf[3];
  Nag_TransfOrder transfv;
  NagError fail;

  INIT_FAIL(fail);
  exit_status = 0;

  Vprintf("g13bdc Example Program Results\n");

  /* Skip heading in data file */
  Vscanf("%*[^\n] ");

  Vscanf("%ld%*[^\n] ", &nl);
  Vscanf("%lf%*[^\n] ", &r0);

  if (nl > 0)
    {
      /* Allocate array r */
      if ( !(r = NAG_ALLOC(nl, double)))
        {
          Vprintf("Allocation failure\n");
          exit_status = -1;
          goto END;
        }

      for (i = 1; i <= nl; ++i)
        Vscanf("%lf", &r[i-1]);
      Vscanf("%*[^\n] ");

      Vscanf("%ld %ld %ld", &transfv.nag_b, &transfv.nag_q, &transfv.nag_p);
      Vscanf("%lf%*[^\n] ", &s);

      nwds = transfv.nag_q + transfv.nag_p + 1;
      iwa = transfv.nag_p * (transfv.nag_p + 1);

      /* Allocate arrays wa and wds */
      if ( !(wa = NAG_ALLOC(iwa, double)) ||
           !(wds = NAG_ALLOC(nwds, double)))
        {
          Vprintf("Allocation failure\n");
          exit_status = -1;
          goto END;
        }

      g13bdc(r0, r, nl, &transfv, s, wds, isf, &fail);
      if (fail.code != NE_NOERROR)
        {
          Vprintf("Error from g13bdc.\n%s\n",   fail.message);
          exit_status = 1;
          goto END;
        }

      Vprintf("\n");
      Vprintf("Success/failure indicator%4ld%4ld\n", isf[0], isf[1]);
      Vprintf("\n");
      Vprintf("Transfer function model B, Q, P =");

      Vprintf("%4ld %4ld %4ld\n",
              transfv.nag_b, transfv.nag_q, transfv.nag_p);

      Vprintf("\n");
      Vprintf("Parameter initial estimates\n");

      for (i = 1; i <= nwds; ++i)
        Vprintf("%10.4f", wds[i-1]);
      Vprintf("\n");
    }

 END:
  if (r) NAG_FREE(r);
  if (wa) NAG_FREE(wa);
  if (wds) NAG_FREE(wds);

  return exit_status;
}