26 #ifndef CASADI_CALCULUS_HPP
27 #define CASADI_CALCULUS_HPP
34 #include "casadi_common.hpp"
44 const double pi = M_PI;
46 const double pi = 3.14159265358979323846;
50 const double inf = std::numeric_limits<double>::infinity();
53 const double nan = std::numeric_limits<double>::quiet_NaN();
56 const double eps = std::numeric_limits<double>::epsilon();
65 OP_ADD, OP_SUB, OP_MUL, OP_DIV,
66 OP_NEG, OP_EXP, OP_LOG, OP_POW, OP_CONSTPOW,
67 OP_SQRT, OP_SQ, OP_TWICE,
68 OP_SIN, OP_COS, OP_TAN,
69 OP_ASIN, OP_ACOS, OP_ATAN,
70 OP_LT, OP_LE, OP_EQ, OP_NE, OP_NOT, OP_AND, OP_OR,
71 OP_FLOOR, OP_CEIL, OP_FMOD, OP_FABS, OP_SIGN, OP_COPYSIGN, OP_IF_ELSE_ZERO,
72 OP_ERF, OP_FMIN, OP_FMAX,
74 OP_SINH, OP_COSH, OP_TANH,
75 OP_ASINH, OP_ACOSH, OP_ATANH,
154 OP_GETNONZEROS_PARAM,
160 OP_ADDNONZEROS_PARAM,
166 OP_SETNONZEROS_PARAM,
178 OP_NORM2, OP_NORM1, OP_NORMINF, OP_NORMF,
213 #define NUM_BUILT_IN_OPS (OP_REMAINDER+1)
243 using std::remainder;
264 inline double sign(
double x) {
return x<0 ? -1 : x>0 ? 1 : x;}
271 inline double simplify(
double x) {
return x;}
272 inline double constpow(
double x,
double y) {
return pow(x, y);}
273 inline double printme(
double x,
double y) {
274 std::ios::fmtflags f(
uout().flags());
275 uout() <<
"|> " << y <<
" : ";
276 uout() << std::setprecision(std::numeric_limits<double>::digits10 + 1) << std::scientific;
277 uout() << x << std::endl;
281 inline bool is_equal(
double x,
double y, casadi_int depth=0) {
return x==y;}
285 inline casadi_int casadi_max(casadi_int x, casadi_int y) {
return std::max(x, y);}
286 inline casadi_int casadi_min(casadi_int x, casadi_int y) {
return std::min(x, y);}
289 inline double if_else_zero(
double x,
double y) {
return x==0 ? 0 : y;}
290 inline double if_else(
double x,
double y,
double z) {
return x==0 ? z : y;}
294 inline double erfinv(
double x)
throw() {
297 return x==1 ? inf : nan;
299 return x==-1 ? -inf : nan;
301 double z = sqrt(-log((1.0+x)/2.0));
302 return -(((1.641345311*z+3.429567803)*z-1.624906493)*z-1.970840454)/
303 ((1.637067800*z+3.543889200)*z+1.0);
308 y = x*(((-0.140543331*z+0.914624893)*z-1.645349621)*z+0.886226899)/
309 ((((-0.329097515*z+0.012229801)*z+1.442710462)*z-2.118377725)*z+1.0);
311 double z = sqrt(-log((1.0-x)/2.0));
312 y = (((1.641345311*z+3.429567803)*z-1.624906493)*z-1.970840454)/
313 ((1.637067800*z+3.543889200)*z+1.0);
317 y = y - (erf(y) - x) / (2.0/sqrt(pi) * exp(-y*y));
318 y = y - (erf(y) - x) / (2.0/sqrt(pi) * exp(-y*y));
326 T twice(
const T& x) {
335 template<casadi_
int I>
336 struct UnaryOperation {
338 template<
typename T>
static inline void fcn(
const T& x, T& f);
341 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d);
344 template<casadi_
int I>
345 struct BinaryOperation {
347 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) {
348 UnaryOperation<I>::fcn(x, f);}
351 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
352 UnaryOperation<I>::der(x, f, d); d[1]=0; }
355 template<casadi_
int I>
356 struct BinaryOperationE {
358 template<
typename T>
static inline T fcn(
const T& x,
const T& y) {
360 BinaryOperation<I>::fcn(x, y, ret);
366 template<casadi_
int I>
367 struct DerBinaryOperation {
369 template<
typename T>
static inline void derf(
const T& x,
const T& y, T& f, T* d) {
377 BinaryOperation<I>::fcn(x, y, tmp);
380 BinaryOperation<I>::der(x, y, tmp, d);
388 template<casadi_
int I>
389 struct BinaryOperationSS {
391 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f, casadi_int n) {
392 BinaryOperation<I>::fcn(x, y, f);
396 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d,
398 BinaryOperation<I>::der(x, y, f, d);
404 template<casadi_
int I>
405 struct BinaryOperationVV {
407 template<
typename T>
static inline void fcn(
const T* x,
const T* y, T* f, casadi_int n) {
408 for (casadi_int i=0; i<n; ++i) {
409 BinaryOperation<I>::fcn(*x++, *y++, *f++);
414 template<
typename T>
static inline void der(
const T* x,
const T* y,
415 const T* f, T* d, casadi_int n) {
416 for (casadi_int i=0; i<n; ++i, d+=2) {
417 BinaryOperation<I>::der(*x++, *y++, *f++, d);
423 template<casadi_
int I>
424 struct BinaryOperationVS {
426 template<
typename T>
static inline void fcn(
const T* x,
const T& y, T* f, casadi_int n) {
427 for (casadi_int i=0; i<n; ++i) {
428 BinaryOperation<I>::fcn(*x++, y, *f++);
433 template<
typename T>
static inline void der(
const T* x,
const T& y,
434 const T* f, T* d, casadi_int n) {
435 for (casadi_int i=0; i<n; ++i, d+=2) {
436 BinaryOperation<I>::der(*x++, y, *f++, d);
442 template<casadi_
int I>
443 struct BinaryOperationSV {
445 template<
typename T>
static inline void fcn(
const T& x,
const T* y, T* f, casadi_int n) {
446 for (casadi_int i=0; i<n; ++i) {
447 BinaryOperation<I>::fcn(x, *y++, *f++);
452 template<
typename T>
static inline void der(
const T& x,
const T* y,
453 const T* f, T* d, casadi_int n) {
454 for (casadi_int i=0; i<n; ++i, d+=2) {
455 BinaryOperation<I>::der(x, *y++, *f++, d);
462 template<casadi_
int I>
struct SmoothChecker {
static const bool check=
true;};
463 template<>
struct SmoothChecker<OP_LT>{
static const bool check=
false;};
464 template<>
struct SmoothChecker<OP_LE>{
static const bool check=
false;};
465 template<>
struct SmoothChecker<OP_FLOOR>{
static const bool check=
false;};
466 template<>
struct SmoothChecker<OP_CEIL>{
static const bool check=
false;};
467 template<>
struct SmoothChecker<OP_FMOD>{
static const bool check=
false;};
468 template<>
struct SmoothChecker<OP_REMAINDER>{
static const bool check=
false;};
469 template<>
struct SmoothChecker<OP_EQ>{
static const bool check=
false;};
470 template<>
struct SmoothChecker<OP_NE>{
static const bool check=
false;};
471 template<>
struct SmoothChecker<OP_SIGN>{
static const bool check=
false;};
472 template<>
struct SmoothChecker<OP_COPYSIGN>{
static const bool check=
false;};
473 template<>
struct SmoothChecker<OP_NOT>{
static const bool check=
false;};
474 template<>
struct SmoothChecker<OP_AND>{
static const bool check=
false;};
475 template<>
struct SmoothChecker<OP_OR>{
static const bool check=
false;};
476 template<>
struct SmoothChecker<OP_IF_ELSE_ZERO>{
static const bool check=
false;};
481 template<casadi_
int I>
struct F0XChecker {
static const bool check=
false;};
482 template<>
struct F0XChecker<OP_ASSIGN>{
static const bool check=
true;};
483 template<>
struct F0XChecker<OP_MUL>{
static const bool check=
true;};
484 template<>
struct F0XChecker<OP_DIV>{
static const bool check=
true;};
485 template<>
struct F0XChecker<OP_NEG>{
static const bool check=
true;};
486 template<>
struct F0XChecker<OP_SQRT>{
static const bool check=
true;};
487 template<>
struct F0XChecker<OP_SQ>{
static const bool check=
true;};
488 template<>
struct F0XChecker<OP_TWICE>{
static const bool check=
true;};
489 template<>
struct F0XChecker<OP_SIN>{
static const bool check=
true;};
490 template<>
struct F0XChecker<OP_TAN>{
static const bool check=
true;};
491 template<>
struct F0XChecker<OP_ATAN>{
static const bool check=
true;};
492 template<>
struct F0XChecker<OP_ASIN>{
static const bool check=
true;};
493 template<>
struct F0XChecker<OP_FLOOR>{
static const bool check=
true;};
494 template<>
struct F0XChecker<OP_CEIL>{
static const bool check=
true;};
495 template<>
struct F0XChecker<OP_FMOD>{
static const bool check=
true;};
496 template<>
struct F0XChecker<OP_REMAINDER>{
static const bool check=
true;};
497 template<>
struct F0XChecker<OP_FABS>{
static const bool check=
true;};
498 template<>
struct F0XChecker<OP_SIGN>{
static const bool check=
true;};
499 template<>
struct F0XChecker<OP_COPYSIGN>{
static const bool check=
true;};
500 template<>
struct F0XChecker<OP_ERF>{
static const bool check=
true;};
501 template<>
struct F0XChecker<OP_SINH>{
static const bool check=
true;};
502 template<>
struct F0XChecker<OP_TANH>{
static const bool check=
true;};
503 template<>
struct F0XChecker<OP_ASINH>{
static const bool check=
true;};
504 template<>
struct F0XChecker<OP_ATANH>{
static const bool check=
true;};
505 template<>
struct F0XChecker<OP_ERFINV>{
static const bool check=
true;};
506 template<>
struct F0XChecker<OP_AND>{
static const bool check=
true;};
507 template<>
struct F0XChecker<OP_IF_ELSE_ZERO>{
static const bool check=
true;};
508 template<>
struct F0XChecker<OP_LOG1P>{
static const bool check=
true;};
509 template<>
struct F0XChecker<OP_EXPM1>{
static const bool check=
true;};
514 template<casadi_
int I>
struct FX0Checker {
static const bool check=
false;};
515 template<>
struct FX0Checker<OP_MUL>{
static const bool check=
true;};
516 template<>
struct FX0Checker<OP_AND>{
static const bool check=
true;};
517 template<>
struct FX0Checker<OP_IF_ELSE_ZERO>{
static const bool check=
true;};
522 template<casadi_
int I>
struct F00Checker {
523 static const bool check=F0XChecker<I>::check || FX0Checker<I>::check;
525 template<>
struct F00Checker<OP_ADD>{
static const bool check=
true;};
526 template<>
struct F00Checker<OP_SUB>{
static const bool check=
true;};
527 template<>
struct F00Checker<OP_FMIN>{
static const bool check=
true;};
528 template<>
struct F00Checker<OP_FMAX>{
static const bool check=
true;};
529 template<>
struct F00Checker<OP_AND>{
static const bool check=
true;};
530 template<>
struct F00Checker<OP_OR>{
static const bool check=
true;};
531 template<>
struct F00Checker<OP_COPYSIGN>{
static const bool check=
true;};
532 template<>
struct F00Checker<OP_LT>{
static const bool check=
true;};
533 template<>
struct F00Checker<OP_HYPOT>{
static const bool check=
true;};
538 template<casadi_
int I>
struct CommChecker {
static const bool check=
false;};
539 template<>
struct CommChecker<OP_ADD>{
static const bool check=
true;};
540 template<>
struct CommChecker<OP_MUL>{
static const bool check=
true;};
541 template<>
struct CommChecker<OP_EQ>{
static const bool check=
true;};
542 template<>
struct CommChecker<OP_NE>{
static const bool check=
true;};
543 template<>
struct CommChecker<OP_AND>{
static const bool check=
true;};
544 template<>
struct CommChecker<OP_OR>{
static const bool check=
true;};
545 template<>
struct CommChecker<OP_HYPOT>{
static const bool check=
true;};
550 template<casadi_
int I>
struct NonnegativeChecker {
static const bool check=
false;};
551 template<>
struct NonnegativeChecker<OP_SQRT>{
static const bool check=
true;};
552 template<>
struct NonnegativeChecker<OP_SQ>{
static const bool check=
true;};
553 template<>
struct NonnegativeChecker<OP_EXP>{
static const bool check=
true;};
554 template<>
struct NonnegativeChecker<OP_LT>{
static const bool check=
true;};
555 template<>
struct NonnegativeChecker<OP_LE>{
static const bool check=
true;};
556 template<>
struct NonnegativeChecker<OP_EQ>{
static const bool check=
true;};
557 template<>
struct NonnegativeChecker<OP_NE>{
static const bool check=
true;};
558 template<>
struct NonnegativeChecker<OP_NOT>{
static const bool check=
true;};
559 template<>
struct NonnegativeChecker<OP_AND>{
static const bool check=
true;};
560 template<>
struct NonnegativeChecker<OP_OR>{
static const bool check=
true;};
561 template<>
struct NonnegativeChecker<OP_HYPOT>{
static const bool check=
true;};
566 template<casadi_
int I>
struct NargChecker {
static const casadi_int check=1;};
567 template<>
struct NargChecker<OP_ADD>{
static const casadi_int check=2;};
568 template<>
struct NargChecker<OP_SUB>{
static const casadi_int check=2;};
569 template<>
struct NargChecker<OP_MUL>{
static const casadi_int check=2;};
570 template<>
struct NargChecker<OP_DIV>{
static const casadi_int check=2;};
571 template<>
struct NargChecker<OP_POW>{
static const casadi_int check=2;};
572 template<>
struct NargChecker<OP_CONSTPOW>{
static const casadi_int check=2;};
573 template<>
struct NargChecker<OP_EQ>{
static const casadi_int check=2;};
574 template<>
struct NargChecker<OP_NE>{
static const casadi_int check=2;};
575 template<>
struct NargChecker<OP_AND>{
static const casadi_int check=2;};
576 template<>
struct NargChecker<OP_OR>{
static const casadi_int check=2;};
577 template<>
struct NargChecker<OP_FMIN>{
static const casadi_int check=2;};
578 template<>
struct NargChecker<OP_FMAX>{
static const casadi_int check=2;};
579 template<>
struct NargChecker<OP_PRINTME>{
static const casadi_int check=2;};
580 template<>
struct NargChecker<OP_ATAN2>{
static const casadi_int check=2;};
581 template<>
struct NargChecker<OP_IF_ELSE_ZERO>{
static const casadi_int check=2;};
582 template<>
struct NargChecker<OP_FMOD>{
static const casadi_int check=2;};
583 template<>
struct NargChecker<OP_REMAINDER>{
static const casadi_int check=2;};
584 template<>
struct NargChecker<OP_COPYSIGN>{
static const casadi_int check=2;};
585 template<>
struct NargChecker<OP_CONST>{
static const casadi_int check=0;};
586 template<>
struct NargChecker<OP_PARAMETER>{
static const casadi_int check=0;};
587 template<>
struct NargChecker<OP_INPUT>{
static const casadi_int check=0;};
588 template<>
struct NargChecker<OP_HYPOT>{
static const casadi_int check=2;};
593 struct UnaryOperation<OP_ASSIGN>{
595 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = x;}
596 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 1; }
601 struct BinaryOperation<OP_ADD>{
603 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x+y;}
604 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
610 struct BinaryOperation<OP_SUB>{
612 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x-y;}
613 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
619 struct BinaryOperation<OP_MUL>{
621 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x*y;}
622 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
628 struct BinaryOperation<OP_DIV>{
630 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x/y;}
631 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
632 d[0]=1/y; d[1]=-f/y;}
637 struct UnaryOperation<OP_NEG>{
639 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = -x;}
640 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=-1;}
645 struct UnaryOperation<OP_EXP>{
647 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = exp(x);}
648 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=f;}
653 struct UnaryOperation<OP_LOG>{
655 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = log(x);}
656 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=1/x;}
661 struct BinaryOperation<OP_POW>{
663 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = pow(x, y);}
665 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
666 d[0]=y*pow(x, y-1); d[1]=log(x)*f;}
671 struct BinaryOperation<OP_CONSTPOW>{
673 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = pow(x, y);}
674 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
675 d[0]=y*pow(x, y-1); d[1]=0;}
680 struct UnaryOperation<OP_SQRT>{
682 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = sqrt(x);}
683 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=1/(twice(f));}
688 struct UnaryOperation<OP_SQ>{
690 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = sq(x);}
691 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=twice(x);}
696 struct UnaryOperation<OP_TWICE>{
697 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = 2.*x;}
698 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 2; }
703 struct UnaryOperation<OP_SIN>{
705 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = sin(x);}
706 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=cos(x);}
711 struct UnaryOperation<OP_COS>{
713 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = cos(x);}
714 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=-sin(x);}
719 struct UnaryOperation<OP_TAN>{
721 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = tan(x);}
722 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d)
723 { d[0] = 1/sq(cos(x));}
728 struct UnaryOperation<OP_ASIN>{
730 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = asin(x);}
731 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=1/sqrt(1-x*x);}
736 struct UnaryOperation<OP_ACOS>{
738 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = acos(x);}
739 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d)
740 { d[0]=-1/sqrt(1-x*x);}
745 struct UnaryOperation<OP_ATAN>{
747 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = atan(x);}
748 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 1/(1+x*x);}
753 struct BinaryOperation<OP_LT>{
755 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x < y;}
756 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
762 struct BinaryOperation<OP_LE>{
764 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x <= y;}
765 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
771 struct UnaryOperation<OP_FLOOR>{
773 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = floor(x);}
774 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 0;}
779 struct UnaryOperation<OP_CEIL>{
781 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = ceil(x);}
782 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 0;}
787 struct BinaryOperation<OP_FMOD>{
788 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = fmod(x, y);}
789 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
790 d[0]=1; d[1]=(f-x)/y;}
795 struct BinaryOperation<OP_REMAINDER>{
796 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) {
797 f = remainder(x, y);}
798 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
799 d[0]=1; d[1]=(f-x)/y;}
804 struct BinaryOperation<OP_EQ>{
805 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x==y;}
806 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
812 struct BinaryOperation<OP_NE>{
813 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x!=y;}
814 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
820 struct UnaryOperation<OP_NOT>{
822 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = !x;}
823 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 0;}
828 struct BinaryOperation<OP_AND>{
829 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x && y;}
830 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
836 struct BinaryOperation<OP_OR>{
837 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x || y;}
838 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
844 struct UnaryOperation<OP_ERF>{
845 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = erf(x);}
846 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) {
847 d[0] = (2/sqrt(pi))*exp(-x*x);}
852 struct UnaryOperation<OP_FABS>{
853 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = fabs(x);}
854 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) {
860 struct UnaryOperation<OP_SIGN>{
861 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = sign(x);}
862 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0]=0;}
867 struct BinaryOperation<OP_COPYSIGN>{
868 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = copysign(x, y);}
869 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
870 T e = 1; d[0]=copysign(e, y); d[1]=0;}
875 struct BinaryOperation<OP_FMIN>{
876 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = fmin(x, y);}
877 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
886 struct BinaryOperation<OP_FMAX>{
887 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = fmax(x, y);}
888 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
897 struct UnaryOperation<OP_INV>{
898 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = 1./x;}
899 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = -f*f; }
904 struct UnaryOperation<OP_SINH>{
905 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = sinh(x);}
906 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = cosh(x); }
911 struct UnaryOperation<OP_COSH>{
912 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = cosh(x);}
913 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = sinh(x); }
918 struct UnaryOperation<OP_TANH>{
919 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = tanh(x);}
920 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 1-f*f; }
925 struct UnaryOperation<OP_ASINH>{
926 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = asinh(x);}
927 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) {
928 d[0] = 1/sqrt(1+x*x); }
933 struct UnaryOperation<OP_ACOSH>{
934 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = acosh(x);}
935 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) {
936 d[0] = 1/sqrt(x-1)/sqrt(x+1); }
941 struct UnaryOperation<OP_ATANH>{
942 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = atanh(x);}
943 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) { d[0] = 1/(1-x*x); }
948 struct UnaryOperation<OP_ERFINV>{
949 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = erfinv(x);}
950 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) {
951 d[0] = (sqrt(pi)/2)*exp(f*f); }
956 struct UnaryOperation<OP_LOG1P>{
957 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = log1p(x);}
958 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) {
964 struct UnaryOperation<OP_EXPM1>{
965 template<
typename T>
static inline void fcn(
const T& x, T& f) { f = expm1(x);}
966 template<
typename T>
static inline void der(
const T& x,
const T& f, T* d) {
972 struct BinaryOperation<OP_PRINTME>{
973 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) {f = printme(x, y); }
974 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
980 struct BinaryOperation<OP_ATAN2>{
982 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = atan2(x, y);}
983 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
984 T t = x*x+y*y; d[0]=y/t; d[1]=-x/t;}
989 struct BinaryOperation<OP_IF_ELSE_ZERO>{
991 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) {
992 f = if_else_zero(x, y);}
993 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
999 struct BinaryOperation<OP_LIFT>{
1000 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = x;}
1001 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
1002 d[0] = 1; d[1] = 0; }
1007 struct BinaryOperation<OP_HYPOT>{
1008 template<
typename T>
static inline void fcn(
const T& x,
const T& y, T& f) { f = hypot(x, y);}
1009 template<
typename T>
static inline void der(
const T& x,
const T& y,
const T& f, T* d) {
1010 d[0] = x/f; d[1] = y/f; }
1013 template<
template<casadi_
int>
class F,
typename T>
1014 T operation_getter(casadi_int op) {
1015 switch (
static_cast<Operation
>(op)) {
1016 case OP_ASSIGN:
return F<OP_ASSIGN>::check;
1017 case OP_ADD:
return F<OP_ADD>::check;
1018 case OP_SUB:
return F<OP_SUB>::check;
1019 case OP_MUL:
return F<OP_MUL>::check;
1020 case OP_DIV:
return F<OP_DIV>::check;
1021 case OP_NEG:
return F<OP_NEG>::check;
1022 case OP_EXP:
return F<OP_EXP>::check;
1023 case OP_LOG:
return F<OP_LOG>::check;
1024 case OP_POW:
return F<OP_POW>::check;
1025 case OP_CONSTPOW:
return F<OP_CONSTPOW>::check;
1026 case OP_SQRT:
return F<OP_SQRT>::check;
1027 case OP_SQ:
return F<OP_SQ>::check;
1028 case OP_TWICE:
return F<OP_TWICE>::check;
1029 case OP_SIN:
return F<OP_SIN>::check;
1030 case OP_COS:
return F<OP_COS>::check;
1031 case OP_TAN:
return F<OP_TAN>::check;
1032 case OP_ASIN:
return F<OP_ASIN>::check;
1033 case OP_ACOS:
return F<OP_ACOS>::check;
1034 case OP_ATAN:
return F<OP_ATAN>::check;
1035 case OP_LT:
return F<OP_LT>::check;
1036 case OP_LE:
return F<OP_LE>::check;
1037 case OP_EQ:
return F<OP_EQ>::check;
1038 case OP_NE:
return F<OP_NE>::check;
1039 case OP_NOT:
return F<OP_NOT>::check;
1040 case OP_AND:
return F<OP_AND>::check;
1041 case OP_OR:
return F<OP_OR>::check;
1042 case OP_FLOOR:
return F<OP_FLOOR>::check;
1043 case OP_CEIL:
return F<OP_CEIL>::check;
1044 case OP_FMOD:
return F<OP_FMOD>::check;
1045 case OP_REMAINDER:
return F<OP_REMAINDER>::check;
1046 case OP_FABS:
return F<OP_FABS>::check;
1047 case OP_SIGN:
return F<OP_SIGN>::check;
1048 case OP_COPYSIGN:
return F<OP_COPYSIGN>::check;
1049 case OP_IF_ELSE_ZERO:
return F<OP_IF_ELSE_ZERO>::check;
1050 case OP_ERF:
return F<OP_ERF>::check;
1051 case OP_FMIN:
return F<OP_FMIN>::check;
1052 case OP_FMAX:
return F<OP_FMAX>::check;
1053 case OP_INV:
return F<OP_INV>::check;
1054 case OP_SINH:
return F<OP_SINH>::check;
1055 case OP_COSH:
return F<OP_COSH>::check;
1056 case OP_TANH:
return F<OP_TANH>::check;
1057 case OP_ASINH:
return F<OP_ASINH>::check;
1058 case OP_ACOSH:
return F<OP_ACOSH>::check;
1059 case OP_ATANH:
return F<OP_ATANH>::check;
1060 case OP_ATAN2:
return F<OP_ATAN2>::check;
1061 case OP_CONST:
return F<OP_CONST>::check;
1062 case OP_INPUT:
return F<OP_INPUT>::check;
1063 case OP_OUTPUT:
return F<OP_OUTPUT>::check;
1064 case OP_PARAMETER:
return F<OP_PARAMETER>::check;
1065 case OP_CALL:
return F<OP_CALL>::check;
1066 case OP_FIND:
return F<OP_FIND>::check;
1067 case OP_LOW:
return F<OP_LOW>::check;
1068 case OP_MAP:
return F<OP_MAP>::check;
1069 case OP_MTIMES:
return F<OP_MTIMES>::check;
1070 case OP_SOLVE:
return F<OP_SOLVE>::check;
1071 case OP_TRANSPOSE:
return F<OP_TRANSPOSE>::check;
1072 case OP_DETERMINANT:
return F<OP_DETERMINANT>::check;
1073 case OP_INVERSE:
return F<OP_INVERSE>::check;
1074 case OP_DOT:
return F<OP_DOT>::check;
1075 case OP_BILIN:
return F<OP_BILIN>::check;
1076 case OP_RANK1:
return F<OP_RANK1>::check;
1077 case OP_HORZCAT:
return F<OP_HORZCAT>::check;
1078 case OP_VERTCAT:
return F<OP_VERTCAT>::check;
1079 case OP_DIAGCAT:
return F<OP_DIAGCAT>::check;
1080 case OP_HORZSPLIT:
return F<OP_HORZSPLIT>::check;
1081 case OP_VERTSPLIT:
return F<OP_VERTSPLIT>::check;
1082 case OP_DIAGSPLIT:
return F<OP_DIAGSPLIT>::check;
1083 case OP_RESHAPE:
return F<OP_RESHAPE>::check;
1084 case OP_SPARSITY_CAST:
return F<OP_SPARSITY_CAST>::check;
1085 case OP_SUBREF:
return F<OP_SUBREF>::check;
1086 case OP_SUBASSIGN:
return F<OP_SUBASSIGN>::check;
1087 case OP_GETNONZEROS:
return F<OP_GETNONZEROS>::check;
1088 case OP_GETNONZEROS_PARAM:
return F<OP_GETNONZEROS_PARAM>::check;
1089 case OP_ADDNONZEROS:
return F<OP_ADDNONZEROS>::check;
1090 case OP_ADDNONZEROS_PARAM:
return F<OP_ADDNONZEROS>::check;
1091 case OP_SETNONZEROS:
return F<OP_SETNONZEROS>::check;
1092 case OP_SETNONZEROS_PARAM:
return F<OP_SETNONZEROS>::check;
1093 case OP_PROJECT:
return F<OP_PROJECT>::check;
1094 case OP_ASSERTION:
return F<OP_ASSERTION>::check;
1095 case OP_MONITOR:
return F<OP_MONITOR>::check;
1096 case OP_NORM2:
return F<OP_NORM2>::check;
1097 case OP_NORM1:
return F<OP_NORM1>::check;
1098 case OP_NORMINF:
return F<OP_NORMINF>::check;
1099 case OP_NORMF:
return F<OP_NORMF>::check;
1100 case OP_MMIN:
return F<OP_MMIN>::check;
1101 case OP_MMAX:
return F<OP_MMAX>::check;
1102 case OP_HORZREPMAT:
return F<OP_HORZREPMAT>::check;
1103 case OP_HORZREPSUM:
return F<OP_HORZREPSUM>::check;
1104 case OP_ERFINV:
return F<OP_ERFINV>::check;
1105 case OP_PRINTME:
return F<OP_PRINTME>::check;
1106 case OP_LIFT:
return F<OP_LIFT>::check;
1107 case OP_EINSTEIN:
return F<OP_EINSTEIN>::check;
1108 case OP_BSPLINE:
return F<OP_BSPLINE>::check;
1109 case OP_CONVEXIFY:
return F<OP_CONVEXIFY>::check;
1110 case OP_LOG1P:
return F<OP_LOG1P>::check;
1111 case OP_EXPM1:
return F<OP_EXPM1>::check;
1112 case OP_HYPOT:
return F<OP_HYPOT>::check;
1113 case OP_LOGSUMEXP:
return F<OP_LOGSUMEXP>::check;
1118 template<
template<casadi_
int>
class F>
1119 bool operation_checker(casadi_int op) {
1120 return operation_getter<F, bool>(op);
1124 template<
typename T>
1125 struct casadi_math {
1130 static inline void fun(
unsigned char op,
const T& x,
const T& y, T& f);
1135 static inline void fun(
unsigned char op,
const T* x,
const T* y, T* f, casadi_int n);
1140 static inline void fun(
unsigned char op,
const T* x,
const T& y, T* f, casadi_int n);
1145 static inline void fun(
unsigned char op,
const T& x,
const T* y, T* f, casadi_int n);
1150 static inline void der(
unsigned char op,
const T& x,
const T& y,
const T& f, T* d);
1155 static inline void derF(
unsigned char op,
const T& x,
const T& y, T& f, T* d);
1160 static inline void fun_linear(
unsigned char op,
const T*x,
const T* y, T* f);
1165 static inline bool is_binary(
unsigned char op);
1170 static inline bool is_unary(
unsigned char op);
1175 static inline casadi_int ndeps(
unsigned char op);
1180 static inline std::string print(
unsigned char op,
const std::string& x,
1181 const std::string& y);
1182 static inline std::string print(
unsigned char op,
const std::string& x);
1183 static inline std::string name(
unsigned char op);
1184 static inline std::string pre(
unsigned char op);
1185 static inline std::string sep(
unsigned char op);
1186 static inline std::string post(
unsigned char op);
1191 struct casadi_math<casadi_int>{
1196 static inline void fun(
unsigned char op,
const casadi_int& x,
1197 const casadi_int& y, casadi_int& f) {
1199 casadi_math<double>::fun(op,
static_cast<double>(x),
static_cast<double>(y), ff);
1200 f =
static_cast<casadi_int
>(ff);
1203 static inline void fun(
unsigned char op,
const casadi_int* x,
const casadi_int* y,
1204 casadi_int* f, casadi_int n) {
1205 for (casadi_int i=0; i<n; ++i) {
1207 casadi_math<double>::fun(op,
static_cast<double>(*x++),
static_cast<double>(*y++), ff);
1208 *f++ =
static_cast<casadi_int
>(ff);
1212 static inline void fun(
unsigned char op,
const casadi_int* x,
const casadi_int& y,
1213 casadi_int* f, casadi_int n) {
1214 for (casadi_int i=0; i<n; ++i) {
1216 casadi_math<double>::fun(op,
static_cast<double>(*x++),
static_cast<double>(y), ff);
1217 *f++ =
static_cast<casadi_int
>(ff);
1221 static inline void fun(
unsigned char op,
const casadi_int& x,
const casadi_int* y,
1222 casadi_int* f, casadi_int n) {
1223 for (casadi_int i=0; i<n; ++i) {
1225 casadi_math<double>::fun(op,
static_cast<double>(x),
static_cast<double>(*y++), ff);
1226 *f++ =
static_cast<casadi_int
>(ff);
1233 static inline void der(
unsigned char op,
const casadi_int& x,
const casadi_int& y,
1234 const casadi_int& f, casadi_int* d) {
1235 double d_real[2] = {
static_cast<double>(d[0]),
static_cast<double>(d[1])};
1236 casadi_math<double>::der(op,
static_cast<double>(x),
static_cast<double>(y),
1237 static_cast<double>(f), d_real);
1238 d[0] =
static_cast<casadi_int
>(d_real[0]);
1239 d[1] =
static_cast<casadi_int
>(d_real[1]);
1245 static inline void derF(
unsigned char op,
const casadi_int& x,
const casadi_int& y,
1246 casadi_int& f, casadi_int* d) {
1247 double d_real[2] = {
static_cast<double>(d[0]),
static_cast<double>(d[1])};
1248 double f_real =
static_cast<double>(f);
1249 casadi_math<double>::derF(op,
static_cast<double>(x),
static_cast<double>(y), f_real, d_real);
1250 f =
static_cast<casadi_int
>(f_real);
1251 d[0] =
static_cast<casadi_int
>(d_real[0]);
1252 d[1] =
static_cast<casadi_int
>(d_real[1]);
1258 static inline casadi_int ndeps(
unsigned char op) {
1259 return casadi_math<double>::ndeps(op);
1265 static inline std::string print(
unsigned char op,
const std::string& x,
1266 const std::string& y) {
1267 return casadi_math<double>::print(op, x, y);
1269 static inline std::string print(
unsigned char op,
const std::string& x) {
1270 return casadi_math<double>::print(op, x);
1272 static inline std::string pre(
unsigned char op) {
1273 return casadi_math<double>::pre(op);
1275 static inline std::string name(
unsigned char op) {
1276 return casadi_math<double>::name(op);
1278 static inline std::string sep(
unsigned char op) {
1279 return casadi_math<double>::sep(op);
1281 static inline std::string post(
unsigned char op) {
1282 return casadi_math<double>::post(op);
1288 template<
typename T>
1289 inline void casadi_math<T>::fun(
unsigned char op,
const T& x,
const T& y, T& f) {
1292 #define CASADI_MATH_FUN_BUILTIN_GEN(CNAME, X, Y, F, N) \
1293 case OP_ASSIGN: CNAME<OP_ASSIGN>::fcn(X, Y, F, N); break; \
1294 case OP_ADD: CNAME<OP_ADD>::fcn(X, Y, F, N); break; \
1295 case OP_SUB: CNAME<OP_SUB>::fcn(X, Y, F, N); break; \
1296 case OP_MUL: CNAME<OP_MUL>::fcn(X, Y, F, N); break; \
1297 case OP_DIV: CNAME<OP_DIV>::fcn(X, Y, F, N); break; \
1298 case OP_NEG: CNAME<OP_NEG>::fcn(X, Y, F, N); break; \
1299 case OP_EXP: CNAME<OP_EXP>::fcn(X, Y, F, N); break; \
1300 case OP_LOG: CNAME<OP_LOG>::fcn(X, Y, F, N); break; \
1301 case OP_POW: CNAME<OP_POW>::fcn(X, Y, F, N); break; \
1302 case OP_CONSTPOW: CNAME<OP_CONSTPOW>::fcn(X, Y, F, N); break; \
1303 case OP_SQRT: CNAME<OP_SQRT>::fcn(X, Y, F, N); break; \
1304 case OP_SQ: CNAME<OP_SQ>::fcn(X, Y, F, N); break; \
1305 case OP_TWICE: CNAME<OP_TWICE>::fcn(X, Y, F, N); break; \
1306 case OP_SIN: CNAME<OP_SIN>::fcn(X, Y, F, N); break; \
1307 case OP_COS: CNAME<OP_COS>::fcn(X, Y, F, N); break; \
1308 case OP_TAN: CNAME<OP_TAN>::fcn(X, Y, F, N); break; \
1309 case OP_ASIN: CNAME<OP_ASIN>::fcn(X, Y, F, N); break; \
1310 case OP_ACOS: CNAME<OP_ACOS>::fcn(X, Y, F, N); break; \
1311 case OP_ATAN: CNAME<OP_ATAN>::fcn(X, Y, F, N); break; \
1312 case OP_LT: CNAME<OP_LT>::fcn(X, Y, F, N); break; \
1313 case OP_LE: CNAME<OP_LE>::fcn(X, Y, F, N); break; \
1314 case OP_EQ: CNAME<OP_EQ>::fcn(X, Y, F, N); break; \
1315 case OP_NE: CNAME<OP_NE>::fcn(X, Y, F, N); break; \
1316 case OP_NOT: CNAME<OP_NOT>::fcn(X, Y, F, N); break; \
1317 case OP_AND: CNAME<OP_AND>::fcn(X, Y, F, N); break; \
1318 case OP_OR: CNAME<OP_OR>::fcn(X, Y, F, N); break; \
1319 case OP_IF_ELSE_ZERO: CNAME<OP_IF_ELSE_ZERO>::fcn(X, Y, F, N); break; \
1320 case OP_FLOOR: CNAME<OP_FLOOR>::fcn(X, Y, F, N); break; \
1321 case OP_CEIL: CNAME<OP_CEIL>::fcn(X, Y, F, N); break; \
1322 case OP_FMOD: CNAME<OP_FMOD>::fcn(X, Y, F, N); break; \
1323 case OP_REMAINDER: CNAME<OP_REMAINDER>::fcn(X, Y, F, N); break; \
1324 case OP_FABS: CNAME<OP_FABS>::fcn(X, Y, F, N); break; \
1325 case OP_SIGN: CNAME<OP_SIGN>::fcn(X, Y, F, N); break; \
1326 case OP_COPYSIGN: CNAME<OP_COPYSIGN>::fcn(X, Y, F, N); break; \
1327 case OP_ERF: CNAME<OP_ERF>::fcn(X, Y, F, N); break; \
1328 case OP_FMIN: CNAME<OP_FMIN>::fcn(X, Y, F, N); break; \
1329 case OP_FMAX: CNAME<OP_FMAX>::fcn(X, Y, F, N); break; \
1330 case OP_INV: CNAME<OP_INV>::fcn(X, Y, F, N); break; \
1331 case OP_SINH: CNAME<OP_SINH>::fcn(X, Y, F, N); break; \
1332 case OP_COSH: CNAME<OP_COSH>::fcn(X, Y, F, N); break; \
1333 case OP_TANH: CNAME<OP_TANH>::fcn(X, Y, F, N); break; \
1334 case OP_ASINH: CNAME<OP_ASINH>::fcn(X, Y, F, N); break; \
1335 case OP_ACOSH: CNAME<OP_ACOSH>::fcn(X, Y, F, N); break; \
1336 case OP_ATANH: CNAME<OP_ATANH>::fcn(X, Y, F, N); break; \
1337 case OP_ATAN2: CNAME<OP_ATAN2>::fcn(X, Y, F, N); break; \
1338 case OP_ERFINV: CNAME<OP_ERFINV>::fcn(X, Y, F, N); break; \
1339 case OP_LIFT: CNAME<OP_LIFT>::fcn(X, Y, F, N); break; \
1340 case OP_PRINTME: CNAME<OP_PRINTME>::fcn(X, Y, F, N); break; \
1341 case OP_LOG1P: CNAME<OP_LOG1P>::fcn(X, Y, F, N); break; \
1342 case OP_EXPM1: CNAME<OP_EXPM1>::fcn(X, Y, F, N); break; \
1343 case OP_HYPOT: CNAME<OP_HYPOT>::fcn(X, Y, F, N); break;
1345 #define CASADI_MATH_FUN_BUILTIN(X, Y, F) CASADI_MATH_FUN_BUILTIN_GEN(BinaryOperationSS, X, Y, F, 1)
1348 CASADI_MATH_FUN_BUILTIN(x, y, f)
1352 template<
typename T>
1353 inline void casadi_math<T>::fun(
unsigned char op,
const T* x,
const T* y, T* f, casadi_int n) {
1355 CASADI_MATH_FUN_BUILTIN_GEN(BinaryOperationVV, x, y, f, n)
1359 template<
typename T>
1360 inline void casadi_math<T>::fun(
unsigned char op,
const T* x,
const T& y, T* f, casadi_int n) {
1362 CASADI_MATH_FUN_BUILTIN_GEN(BinaryOperationVS, x, y, f, n)
1366 template<
typename T>
1367 inline void casadi_math<T>::fun(
unsigned char op,
const T& x,
const T* y, T* f, casadi_int n) {
1369 CASADI_MATH_FUN_BUILTIN_GEN(BinaryOperationSV, x, y, f, n)
1374 template<
typename T>
1375 inline void casadi_math<T>::der(
unsigned char op,
const T& x,
const T& y,
const T& f, T* d) {
1378 #define CASADI_MATH_DER_BUILTIN(X, Y, F, D) \
1379 case OP_ASSIGN: BinaryOperation<OP_ASSIGN>::der(X, Y, F, D); break; \
1380 case OP_ADD: BinaryOperation<OP_ADD>::der(X, Y, F, D); break; \
1381 case OP_SUB: BinaryOperation<OP_SUB>::der(X, Y, F, D); break; \
1382 case OP_MUL: BinaryOperation<OP_MUL>::der(X, Y, F, D); break; \
1383 case OP_DIV: BinaryOperation<OP_DIV>::der(X, Y, F, D); break; \
1384 case OP_NEG: BinaryOperation<OP_NEG>::der(X, Y, F, D); break; \
1385 case OP_EXP: BinaryOperation<OP_EXP>::der(X, Y, F, D); break; \
1386 case OP_LOG: BinaryOperation<OP_LOG>::der(X, Y, F, D); break; \
1387 case OP_POW: BinaryOperation<OP_POW>::der(X, Y, F, D); break; \
1388 case OP_CONSTPOW: BinaryOperation<OP_CONSTPOW>::der(X, Y, F, D); break; \
1389 case OP_SQRT: BinaryOperation<OP_SQRT>::der(X, Y, F, D); break; \
1390 case OP_SQ: BinaryOperation<OP_SQ>::der(X, Y, F, D); break; \
1391 case OP_TWICE: BinaryOperation<OP_TWICE>::der(X, Y, F, D); break; \
1392 case OP_SIN: BinaryOperation<OP_SIN>::der(X, Y, F, D); break; \
1393 case OP_COS: BinaryOperation<OP_COS>::der(X, Y, F, D); break; \
1394 case OP_TAN: BinaryOperation<OP_TAN>::der(X, Y, F, D); break; \
1395 case OP_ASIN: BinaryOperation<OP_ASIN>::der(X, Y, F, D); break; \
1396 case OP_ACOS: BinaryOperation<OP_ACOS>::der(X, Y, F, D); break; \
1397 case OP_ATAN: BinaryOperation<OP_ATAN>::der(X, Y, F, D); break; \
1398 case OP_LT: BinaryOperation<OP_LT>::der(X, Y, F, D); break; \
1399 case OP_LE: BinaryOperation<OP_LE>::der(X, Y, F, D); break; \
1400 case OP_EQ: BinaryOperation<OP_EQ>::der(X, Y, F, D); break; \
1401 case OP_NE: BinaryOperation<OP_NE>::der(X, Y, F, D); break; \
1402 case OP_NOT: BinaryOperation<OP_NOT>::der(X, Y, F, D); break; \
1403 case OP_AND: BinaryOperation<OP_AND>::der(X, Y, F, D); break; \
1404 case OP_OR: BinaryOperation<OP_OR>::der(X, Y, F, D); break; \
1405 case OP_IF_ELSE_ZERO: BinaryOperation<OP_IF_ELSE_ZERO>::der(X, Y, F, D); break; \
1406 case OP_FLOOR: BinaryOperation<OP_FLOOR>::der(X, Y, F, D); break; \
1407 case OP_CEIL: BinaryOperation<OP_CEIL>::der(X, Y, F, D); break; \
1408 case OP_FMOD: BinaryOperation<OP_FMOD>::der(X, Y, F, D); break; \
1409 case OP_REMAINDER: BinaryOperation<OP_REMAINDER>::der(X, Y, F, D); break; \
1410 case OP_FABS: BinaryOperation<OP_FABS>::der(X, Y, F, D); break; \
1411 case OP_SIGN: BinaryOperation<OP_SIGN>::der(X, Y, F, D); break; \
1412 case OP_COPYSIGN: BinaryOperation<OP_COPYSIGN>::der(X, Y, F, D); break; \
1413 case OP_ERF: BinaryOperation<OP_ERF>::der(X, Y, F, D); break; \
1414 case OP_FMIN: BinaryOperation<OP_FMIN>::der(X, Y, F, D); break; \
1415 case OP_FMAX: BinaryOperation<OP_FMAX>::der(X, Y, F, D); break; \
1416 case OP_INV: BinaryOperation<OP_INV>::der(X, Y, F, D); break; \
1417 case OP_SINH: BinaryOperation<OP_SINH>::der(X, Y, F, D); break; \
1418 case OP_COSH: BinaryOperation<OP_COSH>::der(X, Y, F, D); break; \
1419 case OP_TANH: BinaryOperation<OP_TANH>::der(X, Y, F, D); break; \
1420 case OP_ASINH: BinaryOperation<OP_ASINH>::der(X, Y, F, D); break; \
1421 case OP_ACOSH: BinaryOperation<OP_ACOSH>::der(X, Y, F, D); break; \
1422 case OP_ATANH: BinaryOperation<OP_ATANH>::der(X, Y, F, D); break; \
1423 case OP_ATAN2: BinaryOperation<OP_ATAN2>::der(X, Y, F, D); break; \
1424 case OP_ERFINV: BinaryOperation<OP_ERFINV>::der(X, Y, F, D); break; \
1425 case OP_LIFT: BinaryOperation<OP_LIFT>::der(X, Y, F, D); break; \
1426 case OP_PRINTME: BinaryOperation<OP_PRINTME>::der(X, Y, F, D); break; \
1427 case OP_LOG1P: BinaryOperation<OP_LOG1P>::der(X, Y, F, D); break; \
1428 case OP_EXPM1: BinaryOperation<OP_EXPM1>::der(X, Y, F, D); break; \
1429 case OP_HYPOT: BinaryOperation<OP_HYPOT>::der(X, Y, F, D); break;
1431 CASADI_MATH_DER_BUILTIN(x, y, f, d)
1436 template<
typename T>
1437 inline void casadi_math<T>::derF(
unsigned char op,
const T& x,
const T& y, T& f, T* d) {
1440 #define CASADI_MATH_DERF_BUILTIN(X, Y, F, D) \
1441 case OP_ASSIGN: DerBinaryOperation<OP_ASSIGN>::derf(X, Y, F, D); break; \
1442 case OP_ADD: DerBinaryOperation<OP_ADD>::derf(X, Y, F, D); break; \
1443 case OP_SUB: DerBinaryOperation<OP_SUB>::derf(X, Y, F, D); break; \
1444 case OP_MUL: DerBinaryOperation<OP_MUL>::derf(X, Y, F, D); break; \
1445 case OP_DIV: DerBinaryOperation<OP_DIV>::derf(X, Y, F, D); break; \
1446 case OP_NEG: DerBinaryOperation<OP_NEG>::derf(X, Y, F, D); break; \
1447 case OP_EXP: DerBinaryOperation<OP_EXP>::derf(X, Y, F, D); break; \
1448 case OP_LOG: DerBinaryOperation<OP_LOG>::derf(X, Y, F, D); break; \
1449 case OP_POW: DerBinaryOperation<OP_POW>::derf(X, Y, F, D); break; \
1450 case OP_CONSTPOW: DerBinaryOperation<OP_CONSTPOW>::derf(X, Y, F, D); break; \
1451 case OP_SQRT: DerBinaryOperation<OP_SQRT>::derf(X, Y, F, D); break; \
1452 case OP_SQ: DerBinaryOperation<OP_SQ>::derf(X, Y, F, D); break; \
1453 case OP_TWICE: DerBinaryOperation<OP_TWICE>::derf(X, Y, F, D); break; \
1454 case OP_SIN: DerBinaryOperation<OP_SIN>::derf(X, Y, F, D); break; \
1455 case OP_COS: DerBinaryOperation<OP_COS>::derf(X, Y, F, D); break; \
1456 case OP_TAN: DerBinaryOperation<OP_TAN>::derf(X, Y, F, D); break; \
1457 case OP_ASIN: DerBinaryOperation<OP_ASIN>::derf(X, Y, F, D); break; \
1458 case OP_ACOS: DerBinaryOperation<OP_ACOS>::derf(X, Y, F, D); break; \
1459 case OP_ATAN: DerBinaryOperation<OP_ATAN>::derf(X, Y, F, D); break; \
1460 case OP_LT: DerBinaryOperation<OP_LT>::derf(X, Y, F, D); break; \
1461 case OP_LE: DerBinaryOperation<OP_LE>::derf(X, Y, F, D); break; \
1462 case OP_EQ: DerBinaryOperation<OP_EQ>::derf(X, Y, F, D); break; \
1463 case OP_NE: DerBinaryOperation<OP_NE>::derf(X, Y, F, D); break; \
1464 case OP_NOT: DerBinaryOperation<OP_NOT>::derf(X, Y, F, D); break; \
1465 case OP_AND: DerBinaryOperation<OP_AND>::derf(X, Y, F, D); break; \
1466 case OP_OR: DerBinaryOperation<OP_OR>::derf(X, Y, F, D); break; \
1467 case OP_IF_ELSE_ZERO: DerBinaryOperation<OP_IF_ELSE_ZERO>::derf(X, Y, F, D); break; \
1468 case OP_FLOOR: DerBinaryOperation<OP_FLOOR>::derf(X, Y, F, D); break; \
1469 case OP_CEIL: DerBinaryOperation<OP_CEIL>::derf(X, Y, F, D); break; \
1470 case OP_FMOD: DerBinaryOperation<OP_FMOD>::derf(X, Y, F, D); break; \
1471 case OP_REMAINDER: DerBinaryOperation<OP_REMAINDER>::derf(X, Y, F, D); break; \
1472 case OP_FABS: DerBinaryOperation<OP_FABS>::derf(X, Y, F, D); break; \
1473 case OP_SIGN: DerBinaryOperation<OP_SIGN>::derf(X, Y, F, D); break; \
1474 case OP_COPYSIGN: DerBinaryOperation<OP_COPYSIGN>::derf(X, Y, F, D); break; \
1475 case OP_ERF: DerBinaryOperation<OP_ERF>::derf(X, Y, F, D); break; \
1476 case OP_FMIN: DerBinaryOperation<OP_FMIN>::derf(X, Y, F, D); break; \
1477 case OP_FMAX: DerBinaryOperation<OP_FMAX>::derf(X, Y, F, D); break; \
1478 case OP_INV: DerBinaryOperation<OP_INV>::derf(X, Y, F, D); break; \
1479 case OP_SINH: DerBinaryOperation<OP_SINH>::derf(X, Y, F, D); break; \
1480 case OP_COSH: DerBinaryOperation<OP_COSH>::derf(X, Y, F, D); break; \
1481 case OP_TANH: DerBinaryOperation<OP_TANH>::derf(X, Y, F, D); break; \
1482 case OP_ASINH: DerBinaryOperation<OP_ASINH>::derf(X, Y, F, D); break; \
1483 case OP_ACOSH: DerBinaryOperation<OP_ACOSH>::derf(X, Y, F, D); break; \
1484 case OP_ATANH: DerBinaryOperation<OP_ATANH>::derf(X, Y, F, D); break; \
1485 case OP_ATAN2: DerBinaryOperation<OP_ATAN2>::derf(X, Y, F, D); break; \
1486 case OP_ERFINV: DerBinaryOperation<OP_ERFINV>::derf(X, Y, F, D); break; \
1487 case OP_LIFT: DerBinaryOperation<OP_LIFT>::derf(X, Y, F, D); break; \
1488 case OP_PRINTME: DerBinaryOperation<OP_PRINTME>::derf(X, Y, F, D); break; \
1489 case OP_LOG1P: DerBinaryOperation<OP_LOG1P>::derf(X, Y, F, D); break; \
1490 case OP_EXPM1: DerBinaryOperation<OP_EXPM1>::derf(X, Y, F, D); break; \
1491 case OP_HYPOT: DerBinaryOperation<OP_HYPOT>::derf(X, Y, F, D); break;
1493 CASADI_MATH_DERF_BUILTIN(x, y, f, d)
1497 #define CASADI_MATH_BINARY_BUILTIN \
1512 case OP_REMAINDER: \
1520 #define CASADI_MATH_UNARY_BUILTIN \
1551 template<
typename T>
1552 inline void casadi_math<T>::fun_linear(
unsigned char op,
const T* x,
const T* y, T* f) {
1553 if (op==OP_ADD || op==OP_SUB) {
1554 for (
int i=0;i<3;++i) {
1555 f[i] = T::binary(op, x[i], y[i]);
1557 }
else if (op==OP_TWICE || op==OP_NEG) {
1558 for (
int i=0;i<3;++i) {
1559 f[i] = T::unary(op, x[i]);
1561 }
else if (op==OP_MUL) {
1571 }
else if (op==OP_DIV) {
1572 bool const_argy = y[1].is_zero() && y[2].is_zero();
1578 f[2] = (x[0]+x[1]+x[2])/(y[0]+y[1]+y[2]);
1580 }
else if (casadi_math<T>::is_unary(op)) {
1581 bool const_arg = x[1].is_zero() && x[2].is_zero();
1583 f[0] = T::unary(op, x[0]);
1585 f[2] = T::unary(op, x[0]+x[1]+x[2]);
1588 }
else if (casadi_math<T>::is_binary(op)) {
1589 bool const_argx = x[1].is_zero() && x[2].is_zero();
1590 bool const_argy = y[1].is_zero() && y[2].is_zero();
1591 if (const_argx && const_argy) {
1592 f[0] = T::binary(op, x[0], y[0]);
1594 f[2] = T::binary(op, x[0]+x[1]+x[2], y[0]+y[1]+y[2]);
1597 casadi_error(
"Not implemented");
1601 template<
typename T>
1602 bool casadi_math<T>::is_binary(
unsigned char op) {
1604 CASADI_MATH_BINARY_BUILTIN
1605 case OP_IF_ELSE_ZERO:
1612 template<
typename T>
1613 bool casadi_math<T>::is_unary(
unsigned char op) {
1615 CASADI_MATH_UNARY_BUILTIN
1622 template<
typename T>
1623 inline casadi_int casadi_math<T>::ndeps(
unsigned char op) {
1629 CASADI_MATH_BINARY_BUILTIN
1630 case OP_IF_ELSE_ZERO:
1639 template<
typename T>
1641 casadi_math<T>::print(
unsigned char op,
1642 const std::string& x,
const std::string& y) {
1643 casadi_assert_dev(ndeps(op)==2);
1644 return pre(op) + x + sep(op) + y + post(op);
1647 template<
typename T>
1649 casadi_math<T>::print(
unsigned char op,
const std::string& x) {
1650 casadi_assert_dev(ndeps(op)==1);
1651 return pre(op) + x + post(op);
1654 template<
typename T>
1655 inline std::string casadi_math<T>::name(
unsigned char op) {
1657 case OP_ASSIGN:
return "assign";
1658 case OP_ADD:
return "add";
1659 case OP_SUB:
return "sub";
1660 case OP_MUL:
return "mul";
1661 case OP_DIV:
return "div";
1662 case OP_NEG:
return "neg";
1663 case OP_EXP:
return "exp";
1664 case OP_LOG:
return "log";
1666 case OP_POW:
return "pow";
1667 case OP_SQRT:
return "sqrt";
1668 case OP_SQ:
return "sq";
1669 case OP_TWICE:
return "twice";
1670 case OP_SIN:
return "sin";
1671 case OP_COS:
return "cos";
1672 case OP_TAN:
return "tan";
1673 case OP_ASIN:
return "asin";
1674 case OP_ACOS:
return "acos";
1675 case OP_ATAN:
return "atan";
1676 case OP_LT:
return "lt";
1677 case OP_LE:
return "le";
1678 case OP_EQ:
return "eq";
1679 case OP_NE:
return "ne";
1680 case OP_NOT:
return "not";
1681 case OP_AND:
return "and";
1682 case OP_OR:
return "or";
1683 case OP_FLOOR:
return "floor";
1684 case OP_CEIL:
return "ceil";
1685 case OP_FMOD:
return "fmod";
1686 case OP_REMAINDER:
return "remainder";
1687 case OP_FABS:
return "fabs";
1688 case OP_SIGN:
return "sign";
1689 case OP_COPYSIGN:
return "copysign";
1690 case OP_IF_ELSE_ZERO:
return "if_else_zero";
1691 case OP_ERF:
return "erf";
1692 case OP_FMIN:
return "fmin";
1693 case OP_FMAX:
return "fmax";
1694 case OP_INV:
return "inv";
1695 case OP_SINH:
return "sinh";
1696 case OP_COSH:
return "cosh";
1697 case OP_TANH:
return "tanh";
1698 case OP_ASINH:
return "asinh";
1699 case OP_ACOSH:
return "acosh";
1700 case OP_ATANH:
return "atanh";
1701 case OP_ATAN2:
return "atan2";
1702 case OP_CONST:
return "const";
1703 case OP_INPUT:
return "input";
1704 case OP_OUTPUT:
return "output";
1705 case OP_PARAMETER:
return "parameter";
1706 case OP_CALL:
return "call";
1707 case OP_MTIMES:
return "mtimes";
1708 case OP_SOLVE:
return "solve";
1709 case OP_TRANSPOSE:
return "transpose";
1710 case OP_DETERMINANT:
return "determinant";
1711 case OP_INVERSE:
return "inverse";
1712 case OP_DOT:
return "dot";
1713 case OP_HORZCAT:
return "horzcat";
1714 case OP_VERTCAT:
return "vertcat";
1715 case OP_DIAGCAT:
return "diagcat";
1716 case OP_HORZSPLIT:
return "horzsplit";
1717 case OP_VERTSPLIT:
return "vertsplit";
1718 case OP_DIAGSPLIT:
return "diagsplit";
1719 case OP_RESHAPE:
return "reshape";
1720 case OP_SPARSITY_CAST:
return "sparsity_cast";
1721 case OP_SUBREF:
return "subref";
1722 case OP_SUBASSIGN:
return "subassign";
1723 case OP_GETNONZEROS:
return "getnonzeros";
1724 case OP_GETNONZEROS_PARAM:
return "getnonzeros_param";
1725 case OP_ADDNONZEROS:
return "addnonzeros";
1726 case OP_ADDNONZEROS_PARAM:
return "addnonzeros_param";
1727 case OP_SETNONZEROS:
return "setnonzeros";
1728 case OP_SETNONZEROS_PARAM:
return "setnonzeros_param";
1729 case OP_PROJECT:
return "project";
1730 case OP_ASSERTION:
return "assertion";
1731 case OP_NORM2:
return "norm2";
1732 case OP_NORM1:
return "norm1";
1733 case OP_NORMINF:
return "norminf";
1734 case OP_NORMF:
return "normf";
1735 case OP_ERFINV:
return "erfinv";
1736 case OP_PRINTME:
return "printme";
1737 case OP_LIFT:
return "lift";
1738 case OP_EINSTEIN:
return "einstein";
1739 case OP_BSPLINE:
return "bspline";
1740 case OP_CONVEXIFY:
return "convexify";
1741 case OP_LOG1P:
return "log1p";
1742 case OP_EXPM1:
return "expm1";
1743 case OP_HYPOT:
return "hypot";
1744 case OP_LOGSUMEXP:
return "logsumexp";
1746 return "<invalid-op>";
1749 template<
typename T>
1750 inline std::string casadi_math<T>::pre(
unsigned char op) {
1752 case OP_ASSIGN:
return "";
1753 case OP_ADD:
return "(";
1754 case OP_SUB:
return "(";
1755 case OP_MUL:
return "(";
1756 case OP_DIV:
return "(";
1757 case OP_NEG:
return "(-";
1758 case OP_TWICE:
return "(2.*";
1759 case OP_LT:
return "(";
1760 case OP_LE:
return "(";
1761 case OP_EQ:
return "(";
1762 case OP_NE:
return "(";
1763 case OP_NOT:
return "(!";
1764 case OP_AND:
return "(";
1765 case OP_OR:
return "(";
1766 case OP_IF_ELSE_ZERO:
return "(";
1767 case OP_INV:
return "(1./";
1768 default:
return name(op) +
"(";
1772 template<
typename T>
1773 inline std::string casadi_math<T>::sep(
unsigned char op) {
1775 case OP_ADD:
return "+";
1776 case OP_SUB:
return "-";
1777 case OP_MUL:
return "*";
1778 case OP_DIV:
return "/";
1779 case OP_LT:
return "<";
1780 case OP_LE:
return "<=";
1781 case OP_EQ:
return "==";
1782 case OP_NE:
return "!=";
1783 case OP_AND:
return "&&";
1784 case OP_OR:
return "||";
1785 case OP_IF_ELSE_ZERO:
return "?";
1786 default:
return ",";
1790 template<
typename T>
1791 inline std::string casadi_math<T>::post(
unsigned char op) {
1793 case OP_ASSIGN:
return "";
1794 case OP_IF_ELSE_ZERO:
return ":0)";
1795 default:
return ")";
CASADI_EXPORT std::ostream & uout()