1 # ifndef CPPAD_LOCAL_ATAN_OP_HPP
2 # define CPPAD_LOCAL_ATAN_OP_HPP
16 namespace CppAD {
namespace local {
55 Base* x = taylor + i_x * cap_order;
56 Base* z = taylor + i_z * cap_order;
57 Base* b = z - cap_order;
61 { z[0] =
atan( x[0] );
62 b[0] = Base(1.0) + x[0] * x[0];
65 for(
size_t j = p; j <= q; j++)
67 b[j] = Base(2.0) * x[0] * x[j];
69 for(k = 1; k < j; k++)
70 { b[j] += x[k] * x[j-k];
71 z[j] -= Base(
double(k)) * z[k] * b[j-k];
73 z[j] /= Base(
double(j));
111 size_t num_taylor_per_var = (cap_order-1) * r + 1;
112 Base* x = taylor + i_x * num_taylor_per_var;
113 Base* z = taylor + i_z * num_taylor_per_var;
114 Base* b = z - num_taylor_per_var;
116 size_t m = (q-1) * r + 1;
117 for(
size_t ell = 0; ell < r; ell++)
118 { b[m+ell] = Base(2.0) * x[m+ell] * x[0];
119 z[m+ell] = Base(
double(q)) * x[m+ell];
120 for(
size_t k = 1; k < q; k++)
121 { b[m+ell] += x[(k-1)*r+1+ell] * x[(q-k-1)*r+1+ell];
122 z[m+ell] -= Base(
double(k)) * z[(k-1)*r+1+ell] * b[(q-k-1)*r+1+ell];
124 z[m+ell] /= ( Base(
double(q)) * b[0] );
143 template <
class Base>
156 Base* x = taylor + i_x * cap_order;
157 Base* z = taylor + i_z * cap_order;
158 Base* b = z - cap_order;
161 b[0] = Base(1.0) + x[0] * x[0];
179 template <
class Base>
196 const Base* x = taylor + i_x * cap_order;
197 Base* px = partial + i_x * nc_partial;
200 const Base* z = taylor + i_z * cap_order;
201 Base* pz = partial + i_z * nc_partial;
204 const Base* b = z - cap_order;
205 Base* pb = pz - nc_partial;
207 Base inv_b0 = Base(1.0) / b[0];
214 pz[j] =
azmul(pz[j], inv_b0);
217 pb[0] -=
azmul(pz[j], z[j]);
218 px[j] += pz[j] +
azmul(pb[j], x[0]);
219 px[0] +=
azmul(pb[j], x[j]);
222 pz[j] /= Base(
double(j));
224 for(k = 1; k < j; k++)
225 { pb[j-k] -= Base(
double(k)) *
azmul(pz[j], z[k]);
226 pz[k] -= Base(
double(k)) *
azmul(pz[j], b[j-k]);
227 px[k] +=
azmul(pb[j], x[j-k]);
231 px[0] +=
azmul(pz[0], inv_b0) + Base(2.0) *
azmul(pb[0], x[0]);
AD< Base > azmul(const AD< Base > &x, const AD< Base > &y)
void reverse_atan_op(size_t d, size_t i_z, size_t i_x, size_t cap_order, const Base *taylor, size_t nc_partial, Base *partial)
Reverse mode partial derivatives for result of op = AtanOp.
void forward_atan_op_dir(size_t q, size_t r, size_t i_z, size_t i_x, size_t cap_order, Base *taylor)
Multiple direction Taylor coefficient for op = AtanOp.
std::complex< double > atan(const std::complex< double > &x)
size_t NumArg(OpCode op)
Number of arguments for a specified operator.
void forward_atan_op_0(size_t i_z, size_t i_x, size_t cap_order, Base *taylor)
Zero order forward mode Taylor coefficient for result of op = AtanOp.
size_t NumRes(OpCode op)
Number of variables resulting from the specified operation.
void forward_atan_op(size_t p, size_t q, size_t i_z, size_t i_x, size_t cap_order, Base *taylor)
Forward mode Taylor coefficient for result of op = AtanOp.
#define CPPAD_ASSERT_UNKNOWN(exp)
Check that exp is true, if not terminate execution.