20 bp::register_ptr_to_python<shared_ptr<StageData>>();
21 StdVectorPythonVisitor<std::vector<shared_ptr<StageData>>,
true>::expose(
24 bp::class_<StageData>(
"StageData",
"Data struct for StageModel objects.",
25 bp::init<const StageModel &>())
26 .def_readonly(
"cost_data", &StageData::cost_data)
27 .def_readwrite(
"dynamics_data", &StageData::dynamics_data)
28 .def_readwrite(
"constraint_data", &StageData::constraint_data);
37 using PolyCost = xyz::polymorphic<context::CostAbstract>;
38 using PolyDynamics = xyz::polymorphic<context::ExplicitDynamics>;
39 using PolyFunction = xyz::polymorphic<context::StageFunction>;
40 using PolyCstrSet = xyz::polymorphic<ConstraintSet>;
41 using PolyStage = xyz::polymorphic<StageModel>;
45 using StageVec = std::vector<PolyStage>;
46 StdVectorPythonVisitor<StageVec, true>::expose(
48 eigenpy::details::overload_base_get_item_for_std_vector<StageVec>());
50#pragma GCC diagnostic push
51#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
52 bp::class_<StageModel>(
54 "A stage of the control problem. Holds costs, dynamics, and constraints.",
56 .def(bp::init<const PolyCost &, const PolyDynamics &>(
57 (
"self"_a,
"cost",
"dynamics")))
58 .def<void (StageModel::*)(
const context::StageConstraint &)>(
59 "addConstraint", &StageModel::addConstraint,
60 eigenpy::deprecated_member<>(
"This method has been deprecated since "
61 "StageConstraint is deprecated."),
62 (
"self"_a,
"constraint"),
"Add an existing constraint to the stage.")
63 .def<void (StageModel::*)(
const PolyFunction &,
const PolyCstrSet &)>(
64 "addConstraint", &StageModel::addConstraint,
65 (
"self"_a,
"func",
"cstr_set"),
66 "Constructs a new constraint (from the underlying function and set) "
67 "and adds it to the stage.")
68 .def_readonly(
"constraints", &StageModel::constraints_,
69 "Get the set of constraints.")
70 .def_readonly(
"dynamics", &StageModel::dynamics_,
"Stage dynamics.")
71 .add_property(
"xspace",
72 bp::make_getter(&StageModel::xspace_,
73 bp::return_internal_reference<>()),
74 "State space for the current state :math:`x_k`.")
75 .add_property(
"xspace_next",
76 bp::make_getter(&StageModel::xspace_next_,
77 bp::return_internal_reference<>()),
78 "State space corresponding to next state :math:`x_{k+1}`.")
79 .add_property(
"uspace",
80 bp::make_getter(&StageModel::uspace_,
81 bp::return_internal_reference<>()),
84 bp::make_getter(&StageModel::cost_,
85 bp::return_internal_reference<>()),
87 .def(
"evaluate", &StageModel::evaluate, (
"self"_a,
"x",
"u",
"data"),
88 "Evaluate the stage cost, dynamics, constraints.")
89 .def(
"computeFirstOrderDerivatives",
90 &StageModel::computeFirstOrderDerivatives,
91 (
"self"_a,
"x",
"u",
"data"),
92 "Compute gradients of the stage cost and jacobians of the dynamics "
95 .def(
"computeSecondOrderDerivatives",
96 &StageModel::computeSecondOrderDerivatives,
97 (
"self"_a,
"x",
"u",
"data"),
"Compute Hessians of the stage cost.")
98 .add_property(
"ndx1", &StageModel::ndx1)
99 .add_property(
"ndx2", &StageModel::ndx2)
100 .add_property(
"nu", &StageModel::nu,
"Control space dimension.")
101 .add_property(
"num_dual", &StageModel::numDual,
102 "Number of dual variables.")
107#pragma GCC diagnostic pop
StageModelTpl< Scalar > StageModel
ManifoldAbstractTpl< Scalar > Manifold
StageDataTpl< Scalar > StageData
ConstraintSetTpl< Scalar > ConstraintSet