1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
// 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
// ----------------------------------------------------------------------------
/*
{xrst_begin sacado_poly.cpp}
Sacado Speed: Second Derivative of a Polynomial
###############################################
Specifications
**************
See :ref:`link_poly-name` .
Implementation
**************
{xrst_spell_off}
{xrst_code cpp} */
// suppress conversion warnings before other includes
# include <cppad/wno_conversion.hpp>
//
# include <Sacado.hpp>
# include <cppad/utility/vector.hpp>
# include <cppad/utility/poly.hpp>
# include <cppad/speed/uniform_01.hpp>
// list of possible options
# include <map>
extern std::map<std::string, bool> global_option;
bool link_poly(
size_t size ,
size_t repeat ,
CppAD::vector<double> &a , // coefficients of polynomial
CppAD::vector<double> &z , // polynomial argument value
CppAD::vector<double> &ddp ) // second derivative w.r.t z
{
if( global_option["atomic"] )
return false;
if( global_option["memory"] || global_option["onetape"] || global_option["optimize"] )
return false;
// -----------------------------------------------------
// setup
typedef Sacado::Tay::Taylor<double> ADScalar;
CppAD::vector<ADScalar> A(size);
size_t i; // temporary index
ADScalar Z; // domain space AD value
ADScalar P; // range space AD value
int order = 2; // order of Taylor coefficients
Z.resize(order+1, false);
P.resize(order+1, false);
// choose the polynomial coefficients
CppAD::uniform_01(size, a);
// AD copy of the polynomial coefficients
for(i = 0; i < size; i++)
A[i] = a[i];
// ------------------------------------------------------
while(repeat--)
{ // get the next argument value
CppAD::uniform_01(1, z);
// independent variable value
Z.fastAccessCoeff(0) = z[0]; // argument value
Z.fastAccessCoeff(1) = 1.; // first order coefficient
Z.fastAccessCoeff(2) = 0.; // second order coefficient
// AD computation of the dependent variable
P = CppAD::Poly(0, A, Z);
// second derivative is twice second order Taylor coefficient
ddp[0] = 2. * P.fastAccessCoeff(2);
}
// ------------------------------------------------------
return true;
}
/* {xrst_code}
{xrst_spell_on}
{xrst_end sacado_poly.cpp}
*/
|
