/* E01BE_A1W_F C++ Header Example Program.
 *
 * Copyright 2017 Numerical Algorithms Group.
 * Mark 26.2, 2017.
 */

#include <nag.h>
#include <nagad.h>
#include <stdio.h>
#include <math.h>
#include <nag_stdlib.h>
#include <iostream>
using namespace std;

int main(void)
{
  // Scalars
  int               exit_status = 0;
  
  cout << "E01BE_A1W_F C++ Header Example Program Results\n\n";

  // Skip first line of data file
  string mystr;
  getline (cin, mystr);
  // Read number of data points
  Integer n;
  cin >> n;

  // Allocate arrays for data and interpolant
  nagad_a1w_w_rtype *x = 0, *f = 0, *d = 0;
  if (!(x = NAG_ALLOC(n, nagad_a1w_w_rtype)) ||
      !(f = NAG_ALLOC(n, nagad_a1w_w_rtype)) ||
      !(d = NAG_ALLOC(n, nagad_a1w_w_rtype))) {
    printf("Allocation failure\n");
    exit_status = -2;
  }
  if (exit_status==0) {
   // Create AD tape
    nagad_a1w_ir_create();

    // Create AD configuration data object
    Integer ifail = 0;
    void    *ad_handle = 0;
    x10aa_a1w_f_(ad_handle,ifail);

    // Read data and register variables
    for (int i=0; i<n; i++) {
      double xr, fr;
      cin >> xr >> fr;
      x[i].value = xr;
      x[i].id = 0;
      f[i].value = fr;
      f[i].id = 0;
      nagad_a1w_ir_register_variable(&x[i]);
      nagad_a1w_ir_register_variable(&f[i]);
    }
    
    // Call the AD routine
    ifail = 0;
    e01be_a1w_f_(ad_handle,n,x,f,d,ifail);

    // Evaluate interpolant and derivatives at a mid-point
    const Integer     m = 1;
    nagad_a1w_w_rtype px[m], pf[m], pd[m];
    double            xint;
    xint = 0.5*(nagad_a1w_get_value(x[n/2-1]) + nagad_a1w_get_value(x[n/2]));
    px[0].value = xint;
    px[0].id = 0;
    
    ifail = 0;
    e01bg_a1w_f_(ad_handle,n,x,f,d,m,px,pf,pd,ifail);

    cout << "\n Value of interpolant at x = " << xint;
    cout.precision(5);
    cout << " is: " << nagad_a1w_get_value(pf[0]) << endl;

    // Setup evaluation of derivatives via adjoints.
    double inc = 1.0;
    nagad_a1w_inc_derivative(&pf[0],inc);

    ifail = 0;
    nagad_a1w_ir_interpret_adjoint(ifail);

    cout << "\n Derivatives calculated: First order adjoints\n";
    cout << " Computational mode    : algorithmic\n";
  
    // Get derivatives
    cout << "\n Derivatives of fitted value w.r.t. data points:\n\n";
    cout << "    i     d/dx         d/df\n";
    cout.setf(ios::scientific,ios::floatfield);
    cout.precision(4);
    for (int j=0; j < n; j++) {
      double dx = nagad_a1w_get_derivative(x[j]);
      double dy = nagad_a1w_get_derivative(f[j]);
      cout.width(5); cout << j+1;
      cout.width(12); cout << dx;
      cout.width(12); cout << dy << endl;
    }

    // Remove computational data object and tape
    x10ab_a1w_f_(ad_handle,ifail);
    nagad_a1w_ir_remove();
  }
  NAG_FREE(x);
  NAG_FREE(f);
  NAG_FREE(d);
  
  return exit_status;
}