// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
// SPDX-FileCopyrightText: Bradley M. Bell <bradbell@seanet.com>
// SPDX-FileContributor: 2003-22 Bradley M. Bell
// ----------------------------------------------------------------------------
// adolc examples should suppress conversion warnings
# include <cppad/wno_conversion.hpp>

# include <adolc/adouble.h>
# include <adolc/taping.h>
# include <adolc/interfaces.h>

// adouble definitions not in Adolc distribution and
// required in order to use CppAD::AD<adouble>
# include <cppad/example/base_adolc.hpp>

# include <cppad/cppad.hpp>

bool base_adolc(void)
{  bool ok = true;                   // initialize test result

   typedef adouble             ADdouble;  // for first level of taping
   typedef CppAD::AD<ADdouble> ADDdouble; // for second level of taping
   size_t n = 4;                          // number independent variables

   CPPAD_TESTVECTOR(ADdouble)   a_x(n);
   CPPAD_TESTVECTOR(ADDdouble) aa_x(n);

   // value of the independent variables
   short tag = 0;      // Adolc setup
   int keep = 1;
   trace_on(tag, keep);
   size_t j;
   for(j = 0; j < n; j++)
      a_x[j] <<= double(n - j);  // a_x is independent for ADdouble
   for(j = 0; j < n; j++)
      aa_x[j] = a_x[j];          // track how aa_x depends on a_x
   CppAD::Independent(aa_x);          // aa_x is independent for ADDdouble


   // compute function
   size_t m = 5;
   CPPAD_TESTVECTOR(ADDdouble) aa_f(m);    // dependent variables

   // different cases of conditional expressions
   aa_f[0] = CondExpLt(aa_x[0], aa_x[1], aa_x[2], aa_x[3]);
   aa_f[1] = CondExpLe(aa_x[0], aa_x[1], aa_x[2], aa_x[3]);
   aa_f[2] = CondExpEq(aa_x[0], aa_x[1], aa_x[2], aa_x[3]);
   aa_f[3] = CondExpGe(aa_x[0], aa_x[1], aa_x[2], aa_x[3]);
   aa_f[4] = CondExpGt(aa_x[0], aa_x[1], aa_x[2], aa_x[3]);

   // declare inner function (just to stop inner taping)
   CppAD::ADFun<ADdouble> a_F(aa_x, aa_f);

   // set values for outer function same as inner function
   // (corresponding to the tape of adobule operations)
   double f_j;
   for(j = 0; j < m; j++)
      Value(aa_f[j]) >>= f_j;
   trace_off();

   // arrays for Adolc zos_forward
   double *x = nullptr, *y = nullptr;
   x = CPPAD_TRACK_NEW_VEC(n, x);
   y = CPPAD_TRACK_NEW_VEC(m, y);

   // switch order of arguments from when taping was done
   for(j = 0; j < n; j++)
      x[j] = double(j);
   zos_forward(tag, int(m), int(n), keep, x, y);
   // CondExpLt(0, 1, 2, 3)
   ok &= (y[0] == double(2));
   // CondExpLe(0, 1, 2, 3)
   ok &= (y[1] == double(2));
   // CondExpEq(0, 1, 2, 3)
   ok &= (y[2] == double(3));
   // CondExpGe(0, 1, 2, 3)
   ok &= (y[3] == double(3));
   // CondExpGt(0, 1, 2, 3)
   ok &= (y[4] == double(3));

   // set left = right and true < false
   x[0] = x[1] = 1.;
   x[2] = 2.;
   x[3] = 3.;
   zos_forward(tag, int(m), int(n), keep, x, y);
   // CondExpLt(1, 1, 2, 3)
   ok &= (y[0] == double(3));
   // CondExpLe(1, 1, 2, 3)
   ok &= (y[1] == double(2));
   // CondExpEq(1, 1, 2, 3)
   ok &= (y[2] == double(2));
   // CondExpGe(1, 1, 2, 3)
   ok &= (y[3] == double(2));
   // CondExpGt(1, 1, 2, 3)
   ok &= (y[4] == double(3));

   CPPAD_TRACK_DEL_VEC(x);
   CPPAD_TRACK_DEL_VEC(y);
   return ok;
}
// END PROGRAM
