2024-04-27 22:42:42 -07:00
|
|
|
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
|
|
|
|
|
From: Tyler Veness <calcmogul@gmail.com>
|
|
|
|
|
Date: Wed, 24 Apr 2024 15:56:06 -0700
|
2024-07-07 06:43:44 -07:00
|
|
|
Subject: [PATCH 1/5] Remove "using enum" declarations
|
2024-04-27 22:42:42 -07:00
|
|
|
|
|
|
|
|
---
|
|
|
|
|
include/sleipnir/autodiff/Expression.hpp | 161 +++++++-----------
|
|
|
|
|
.../optimization/SolverExitCondition.hpp | 22 ++-
|
|
|
|
|
2 files changed, 73 insertions(+), 110 deletions(-)
|
|
|
|
|
|
|
|
|
|
diff --git a/include/sleipnir/autodiff/Expression.hpp b/include/sleipnir/autodiff/Expression.hpp
|
2024-08-04 06:26:05 -07:00
|
|
|
index dd53755ccae88e3975d1b5e6b13ac464bd4efcce..51070613e82cdf5e4105519f39632deb5d2bf19e 100644
|
2024-04-27 22:42:42 -07:00
|
|
|
--- a/include/sleipnir/autodiff/Expression.hpp
|
|
|
|
|
+++ b/include/sleipnir/autodiff/Expression.hpp
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -203,8 +203,6 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
friend SLEIPNIR_DLLEXPORT ExpressionPtr operator*(const ExpressionPtr& lhs,
|
|
|
|
|
const ExpressionPtr& rhs) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (lhs->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -219,20 +217,22 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (lhs->type == kConstant && rhs->type == kConstant) {
|
|
|
|
|
+ if (lhs->type == ExpressionType::kConstant &&
|
|
|
|
|
+ rhs->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(lhs->value * rhs->value);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate expression type
|
|
|
|
|
ExpressionType type;
|
|
|
|
|
- if (lhs->type == kConstant) {
|
|
|
|
|
+ if (lhs->type == ExpressionType::kConstant) {
|
|
|
|
|
type = rhs->type;
|
|
|
|
|
- } else if (rhs->type == kConstant) {
|
|
|
|
|
+ } else if (rhs->type == ExpressionType::kConstant) {
|
|
|
|
|
type = lhs->type;
|
|
|
|
|
- } else if (lhs->type == kLinear && rhs->type == kLinear) {
|
|
|
|
|
- type = kQuadratic;
|
|
|
|
|
+ } else if (lhs->type == ExpressionType::kLinear &&
|
|
|
|
|
+ rhs->type == ExpressionType::kLinear) {
|
|
|
|
|
+ type = ExpressionType::kQuadratic;
|
|
|
|
|
} else {
|
|
|
|
|
- type = kNonlinear;
|
|
|
|
|
+ type = ExpressionType::kNonlinear;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -258,8 +258,6 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
friend SLEIPNIR_DLLEXPORT ExpressionPtr operator/(const ExpressionPtr& lhs,
|
|
|
|
|
const ExpressionPtr& rhs) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (lhs->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -269,16 +267,17 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (lhs->type == kConstant && rhs->type == kConstant) {
|
|
|
|
|
+ if (lhs->type == ExpressionType::kConstant &&
|
|
|
|
|
+ rhs->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(lhs->value / rhs->value);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate expression type
|
|
|
|
|
ExpressionType type;
|
|
|
|
|
- if (rhs->type == kConstant) {
|
|
|
|
|
+ if (rhs->type == ExpressionType::kConstant) {
|
|
|
|
|
type = lhs->type;
|
|
|
|
|
} else {
|
|
|
|
|
- type = kNonlinear;
|
|
|
|
|
+ type = ExpressionType::kNonlinear;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -306,8 +305,6 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
friend SLEIPNIR_DLLEXPORT ExpressionPtr operator+(const ExpressionPtr& lhs,
|
|
|
|
|
const ExpressionPtr& rhs) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (lhs == nullptr || lhs->IsConstant(0.0)) {
|
|
|
|
|
return rhs;
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -316,7 +313,8 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (lhs->type == kConstant && rhs->type == kConstant) {
|
|
|
|
|
+ if (lhs->type == ExpressionType::kConstant &&
|
|
|
|
|
+ rhs->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(lhs->value + rhs->value);
|
|
|
|
|
}
|
|
|
|
|
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -340,8 +338,6 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
friend SLEIPNIR_DLLEXPORT ExpressionPtr operator-(const ExpressionPtr& lhs,
|
|
|
|
|
const ExpressionPtr& rhs) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (lhs->IsConstant(0.0)) {
|
|
|
|
|
if (rhs->IsConstant(0.0)) {
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -355,7 +351,8 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (lhs->type == kConstant && rhs->type == kConstant) {
|
|
|
|
|
+ if (lhs->type == ExpressionType::kConstant &&
|
|
|
|
|
+ rhs->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(lhs->value - rhs->value);
|
|
|
|
|
}
|
|
|
|
|
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -377,8 +374,6 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
* @param lhs Operand of unary minus.
|
|
|
|
|
*/
|
|
|
|
|
friend SLEIPNIR_DLLEXPORT ExpressionPtr operator-(const ExpressionPtr& lhs) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (lhs->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -386,7 +381,7 @@ struct SLEIPNIR_DLLEXPORT Expression {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (lhs->type == kConstant) {
|
|
|
|
|
+ if (lhs->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(-lhs->value);
|
|
|
|
|
}
|
|
|
|
|
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -469,8 +464,6 @@ inline constexpr void IntrusiveSharedPtrDecRefCount(Expression* expr) {
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr abs( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -478,12 +471,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr abs( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::abs(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::abs(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::abs(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
if (x < 0.0) {
|
|
|
|
|
return -parentAdjoint;
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -514,20 +507,18 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr abs( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr acos( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
return MakeExpressionPtr(std::numbers::pi / 2.0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::acos(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::acos(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::acos(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return -parentAdjoint / std::sqrt(1.0 - x * x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -546,8 +537,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr acos( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr asin( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -555,12 +544,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr asin( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::asin(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::asin(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::asin(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint / std::sqrt(1.0 - x * x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -579,8 +568,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr asin( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr atan( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -588,12 +575,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr atan( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::atan(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::atan(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::atan(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint / (1.0 + x * x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -612,8 +599,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr atan( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr atan2( // NOLINT
|
|
|
|
|
const ExpressionPtr& y, const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (y->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -623,12 +608,14 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr atan2( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (y->type == kConstant && x->type == kConstant) {
|
|
|
|
|
+ if (y->type == ExpressionType::kConstant &&
|
|
|
|
|
+ x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::atan2(y->value, x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double y, double x) { return std::atan2(y, x); },
|
|
|
|
|
+ ExpressionType::kNonlinear,
|
|
|
|
|
+ [](double y, double x) { return std::atan2(y, x); },
|
|
|
|
|
[](double y, double x, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * x / (y * y + x * x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -653,20 +640,18 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr atan2( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr cos( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
return MakeExpressionPtr(1.0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::cos(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::cos(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::cos(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return -parentAdjoint * std::sin(x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -684,20 +669,18 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr cos( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr cosh( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
return MakeExpressionPtr(1.0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::cosh(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::cosh(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::cosh(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * std::sinh(x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -715,8 +698,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr cosh( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr erf( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -724,12 +705,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr erf( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::erf(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::erf(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::erf(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * 2.0 * std::numbers::inv_sqrtpi *
|
|
|
|
|
std::exp(-x * x);
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -750,20 +731,18 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr erf( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr exp( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
return MakeExpressionPtr(1.0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::exp(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::exp(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::exp(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * std::exp(x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -782,8 +761,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr exp( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr hypot( // NOLINT
|
|
|
|
|
const ExpressionPtr& x, const ExpressionPtr& y) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
return y;
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -792,12 +769,14 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr hypot( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant && y->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant &&
|
|
|
|
|
+ y->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::hypot(x->value, y->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double y) { return std::hypot(x, y); },
|
|
|
|
|
+ ExpressionType::kNonlinear,
|
|
|
|
|
+ [](double x, double y) { return std::hypot(x, y); },
|
|
|
|
|
[](double x, double y, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * x / std::hypot(x, y);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -822,8 +801,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr hypot( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr log( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -831,12 +808,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr log( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::log(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::log(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::log(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) { return parentAdjoint / x; },
|
|
|
|
|
[](const ExpressionPtr& x, const ExpressionPtr&,
|
|
|
|
|
const ExpressionPtr& parentAdjoint) { return parentAdjoint / x; },
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -850,8 +827,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr log( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr log10( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -859,12 +834,13 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr log10( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::log10(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::log10(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear,
|
|
|
|
|
+ [](double x, double) { return std::log10(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint / (std::numbers::ln10 * x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -883,8 +859,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr log10( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr pow( // NOLINT
|
|
|
|
|
const ExpressionPtr& base, const ExpressionPtr& power) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (base->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -899,12 +873,15 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr pow( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (base->type == kConstant && power->type == kConstant) {
|
|
|
|
|
+ if (base->type == ExpressionType::kConstant &&
|
|
|
|
|
+ power->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::pow(base->value, power->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- base->type == kLinear && power->IsConstant(2.0) ? kQuadratic : kNonlinear,
|
|
|
|
|
+ base->type == ExpressionType::kLinear && power->IsConstant(2.0)
|
|
|
|
|
+ ? ExpressionType::kQuadratic
|
|
|
|
|
+ : ExpressionType::kNonlinear,
|
|
|
|
|
[](double base, double power) { return std::pow(base, power); },
|
|
|
|
|
[](double base, double power, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * std::pow(base, power - 1) * power;
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -945,10 +922,8 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr pow( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
* @param x The argument.
|
|
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr sign(const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
if (x->value < 0.0) {
|
|
|
|
|
return MakeExpressionPtr(-1.0);
|
|
|
|
|
} else if (x->value == 0.0) {
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -960,7 +935,7 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sign(const ExpressionPtr& x) {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear,
|
|
|
|
|
+ ExpressionType::kNonlinear,
|
|
|
|
|
[](double x, double) {
|
|
|
|
|
if (x < 0.0) {
|
|
|
|
|
return -1.0;
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -985,8 +960,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sign(const ExpressionPtr& x) {
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr sin( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -994,12 +967,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sin( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::sin(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::sin(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::sin(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * std::cos(x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1016,8 +989,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sin( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
* @param x The argument.
|
|
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr sinh(const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1025,12 +996,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sinh(const ExpressionPtr& x) {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::sinh(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::sinh(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::sinh(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint * std::cosh(x);
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1048,10 +1019,8 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sinh(const ExpressionPtr& x) {
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr sqrt( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
if (x->value == 0.0) {
|
|
|
|
|
// Return zero
|
|
|
|
|
return x;
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1063,7 +1032,7 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sqrt( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::sqrt(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::sqrt(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint / (2.0 * std::sqrt(x));
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1082,8 +1051,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr sqrt( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr tan( // NOLINT
|
|
|
|
|
const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1091,12 +1058,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr tan( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::tan(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::tan(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::tan(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint / (std::cos(x) * std::cos(x));
|
|
|
|
|
},
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1114,8 +1081,6 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr tan( // NOLINT
|
2024-04-27 22:42:42 -07:00
|
|
|
* @param x The argument.
|
|
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT inline ExpressionPtr tanh(const ExpressionPtr& x) {
|
|
|
|
|
- using enum ExpressionType;
|
|
|
|
|
-
|
|
|
|
|
// Prune expression
|
|
|
|
|
if (x->IsConstant(0.0)) {
|
|
|
|
|
// Return zero
|
2024-08-04 06:26:05 -07:00
|
|
|
@@ -1123,12 +1088,12 @@ SLEIPNIR_DLLEXPORT inline ExpressionPtr tanh(const ExpressionPtr& x) {
|
2024-04-27 22:42:42 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Evaluate constant
|
|
|
|
|
- if (x->type == kConstant) {
|
|
|
|
|
+ if (x->type == ExpressionType::kConstant) {
|
|
|
|
|
return MakeExpressionPtr(std::tanh(x->value));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return MakeExpressionPtr(
|
|
|
|
|
- kNonlinear, [](double x, double) { return std::tanh(x); },
|
|
|
|
|
+ ExpressionType::kNonlinear, [](double x, double) { return std::tanh(x); },
|
|
|
|
|
[](double x, double, double parentAdjoint) {
|
|
|
|
|
return parentAdjoint / (std::cosh(x) * std::cosh(x));
|
|
|
|
|
},
|
|
|
|
|
diff --git a/include/sleipnir/optimization/SolverExitCondition.hpp b/include/sleipnir/optimization/SolverExitCondition.hpp
|
|
|
|
|
index 7d1445297e33e3c62bcdf9d03eebeaad20af9a1c..734cd3d127327e8ce01e1a42fe74ccc81fea1f90 100644
|
|
|
|
|
--- a/include/sleipnir/optimization/SolverExitCondition.hpp
|
|
|
|
|
+++ b/include/sleipnir/optimization/SolverExitCondition.hpp
|
|
|
|
|
@@ -46,31 +46,29 @@ enum class SolverExitCondition : int8_t {
|
|
|
|
|
*/
|
|
|
|
|
SLEIPNIR_DLLEXPORT constexpr std::string_view ToMessage(
|
|
|
|
|
const SolverExitCondition& exitCondition) {
|
|
|
|
|
- using enum SolverExitCondition;
|
|
|
|
|
-
|
|
|
|
|
switch (exitCondition) {
|
|
|
|
|
- case kSuccess:
|
|
|
|
|
+ case SolverExitCondition::kSuccess:
|
|
|
|
|
return "solved to desired tolerance";
|
|
|
|
|
- case kSolvedToAcceptableTolerance:
|
|
|
|
|
+ case SolverExitCondition::kSolvedToAcceptableTolerance:
|
|
|
|
|
return "solved to acceptable tolerance";
|
|
|
|
|
- case kCallbackRequestedStop:
|
|
|
|
|
+ case SolverExitCondition::kCallbackRequestedStop:
|
|
|
|
|
return "callback requested stop";
|
|
|
|
|
- case kTooFewDOFs:
|
|
|
|
|
+ case SolverExitCondition::kTooFewDOFs:
|
|
|
|
|
return "problem has too few degrees of freedom";
|
|
|
|
|
- case kLocallyInfeasible:
|
|
|
|
|
+ case SolverExitCondition::kLocallyInfeasible:
|
|
|
|
|
return "problem is locally infeasible";
|
|
|
|
|
- case kFeasibilityRestorationFailed:
|
|
|
|
|
+ case SolverExitCondition::kFeasibilityRestorationFailed:
|
|
|
|
|
return "solver failed to reach the desired tolerance, and feasibility "
|
|
|
|
|
"restoration failed to converge";
|
|
|
|
|
- case kNonfiniteInitialCostOrConstraints:
|
|
|
|
|
+ case SolverExitCondition::kNonfiniteInitialCostOrConstraints:
|
|
|
|
|
return "solver encountered nonfinite initial cost or constraints and "
|
|
|
|
|
"gave up";
|
|
|
|
|
- case kDivergingIterates:
|
|
|
|
|
+ case SolverExitCondition::kDivergingIterates:
|
|
|
|
|
return "solver encountered diverging primal iterates xₖ and/or sₖ and "
|
|
|
|
|
"gave up";
|
|
|
|
|
- case kMaxIterationsExceeded:
|
|
|
|
|
+ case SolverExitCondition::kMaxIterationsExceeded:
|
|
|
|
|
return "solution returned after maximum iterations exceeded";
|
|
|
|
|
- case kTimeout:
|
|
|
|
|
+ case SolverExitCondition::kTimeout:
|
|
|
|
|
return "solution returned after maximum wall clock time exceeded";
|
|
|
|
|
default:
|
|
|
|
|
return "unknown";
|