/*
 o-----------------------------------------------------------------------------o
 |
 | TPSA Example 5
 |
 | Methodical Accelerator Design - Copyright (c) 2016+
 | Support: http://cern.ch/mad  - mad at cern.ch
 | Authors: L. Deniau, laurent.deniau at cern.ch
 | Contrib: -
 |
 o-----------------------------------------------------------------------------o
 | You can redistribute this file and/or modify it under the terms of the GNU
 | General Public License GPLv3 (or later), as published by the Free Software
 | Foundation. This file is distributed in the hope that it will be useful, but
 | WITHOUT ANY WARRANTY OF ANY KIND. See http://gnu.org/licenses for details.
 o-----------------------------------------------------------------------------o

  WARNING: this file is not part of MAD, it must be used as an example with the
           standalone library, see mad/src/REAME.GTPSA for more info.
*/

#include "mad_cst.h"
#include "mad_tpsa.h"

int main(void)
{
  // descriptor for TPSA with 100 variables of order 2 without parameters
  const desc_t *d2 = mad_desc_newv(100, 2);
  printf("d2 length=%4d coefs\n", mad_desc_maxlen(d2, mad_tpsa_default));
  mad_desc_del(d2); d2 = 0; // not used anymore.

  // descriptor for TPSA with 6  variables  of order 2 and
  //                          94 parameters of order 1
  const desc_t *d = mad_desc_newvp(6, 2, 94, 1);
  printf("d  length=%4d coefs\n", mad_desc_maxlen(d, mad_tpsa_default));

  // two TPSAs, t2 is same as t1
  tpsa_t *t1 = mad_tpsa_newd(d, mad_tpsa_default);
  tpsa_t *t2 = mad_tpsa_new(t1, mad_tpsa_same);

  // set order 0 and 1 (quick and dirty!)
  num_t vec[101]={M_PI/6}; for (int i=1; i<=100; i++) vec[i]=1;
  mad_tpsa_setv(t1, 0, 1+100, vec);
  mad_tpsa_print(t1, "ini", 0,0,0);

  // t2=sin(t1)
  mad_tpsa_sin(t1, t2);
  mad_tpsa_print(t2, "sin", 0,0,0);
  mad_tpsa_del(t1);

  // tpsa functions and operators support aliasing (i.e. src == dst)
  mad_tpsa_asin(t2, t2);             // asin(x) = -i*ln(i*x + sqrt(1-x^2))
  mad_tpsa_print(t2, "asin", 0,0,0); // see the accuracy of asin(sin)
  mad_tpsa_del(t2);

  // destroy all created descriptors (optional cleanup)
  mad_desc_del(0);
  return 0;
}