alpaqa 1.0.0a13
Nonconvex constrained optimization
Loading...
Searching...
No Matches
dl-problem.cpp
Go to the documentation of this file.
5
6#include <dlfcn.h>
7#include <algorithm>
8#include <cassert>
9#include <charconv>
10#include <list>
11#include <memory>
12#include <mutex>
13#include <stdexcept>
14
15namespace alpaqa::dl {
16
17namespace {
18
20 std::string s(16, '0');
21 auto begin = s.data(), end = begin + s.size();
22 auto [ptr, ec] = std::to_chars(begin, end, version, 16);
23 if (ec != std::errc())
24 throw std::logic_error(std::make_error_code(ec).message());
25 std::rotate(begin, ptr, end);
26 return s;
27}
28
29void check_abi_version(uint64_t abi_version) {
30 if (abi_version != ALPAQA_DL_ABI_VERSION) {
31 auto prob_version = format_abi_version(abi_version);
33 throw std::runtime_error(
34 "alpaqa::dl::DLProblem::DLProblem: "
35 "Incompatible problem definition (problem ABI version 0x" +
36 prob_version + ", this version of alpaqa supports 0x" +
37 alpaqa_version + ")");
38 }
39}
40
41std::shared_ptr<void> load_lib(const std::string &so_filename) {
42 assert(!so_filename.empty());
43 ::dlerror();
44 void *h = ::dlopen(so_filename.c_str(), RTLD_LOCAL | RTLD_NOW);
45 if (auto *err = ::dlerror())
46 throw std::runtime_error(err);
47 return std::shared_ptr<void>{h, &::dlclose};
48}
49
50template <class F>
51F *load_func(void *handle, const std::string &name) {
52 assert(handle);
53 ::dlerror();
54 auto *h = ::dlsym(handle, name.c_str());
55 if (auto *err = ::dlerror())
56 throw std::runtime_error("Unable to load function '" + name +
57 "': " + err);
58 // We can only hope that the user got the signature right ...
59 return reinterpret_cast<F *>(h);
60}
61
63std::list<std::shared_ptr<void>> leaked_modules;
64void leak_lib(std::shared_ptr<void> handle) {
65 std::lock_guard lck{leaked_modules_mutex};
66 leaked_modules.emplace_back(std::move(handle));
67}
68
69// clang-format off
70template <Config Conf>
73 switch (sp.kind) {
76 using Dense = sparsity::Dense<config_t>;
77 return Dense{
78 .rows = sp.dense.rows,
79 .cols = sp.dense.cols,
80 .symmetry = static_cast<Symmetry>(sp.dense.symmetry),
81 };
83 using SparseCSC = sparsity::SparseCSC<config_t, int>;
84 return SparseCSC{
85 .rows = sp.sparse_csc.rows,
86 .cols = sp.sparse_csc.cols,
87 .symmetry = static_cast<Symmetry>(sp.sparse_csc.symmetry),
88 .inner_idx = typename SparseCSC::index_vector_map_t{sp.sparse_csc.inner_idx, sp.sparse_csc.nnz},
89 .outer_ptr = typename SparseCSC::index_vector_map_t{sp.sparse_csc.outer_ptr, sp.sparse_csc.cols + 1},
90 .order = static_cast<SparseCSC::Order>(sp.sparse_csc.order),
91 };
94 return SparseCSCl{
95 .rows = sp.sparse_csc_l.rows,
96 .cols = sp.sparse_csc_l.cols,
97 .symmetry = static_cast<Symmetry>(sp.sparse_csc_l.symmetry),
98 .inner_idx = typename SparseCSCl::index_vector_map_t{sp.sparse_csc_l.inner_idx, sp.sparse_csc_l.nnz},
99 .outer_ptr = typename SparseCSCl::index_vector_map_t{sp.sparse_csc_l.outer_ptr, sp.sparse_csc_l.cols + 1},
100 .order = static_cast<SparseCSCl::Order>(sp.sparse_csc_l.order),
101 };
104 return SparseCSCll{
105 .rows = sp.sparse_csc_ll.rows,
106 .cols = sp.sparse_csc_ll.cols,
107 .symmetry = static_cast<Symmetry>(sp.sparse_csc_ll.symmetry),
108 .inner_idx = typename SparseCSCll::index_vector_map_t{sp.sparse_csc_ll.inner_idx, sp.sparse_csc_ll.nnz},
109 .outer_ptr = typename SparseCSCll::index_vector_map_t{sp.sparse_csc_ll.outer_ptr, sp.sparse_csc_ll.cols + 1},
110 .order = static_cast<SparseCSCll::Order>(sp.sparse_csc_ll.order),
111 };
113 using SparseCOO = sparsity::SparseCOO<config_t, int>;
114 return SparseCOO{
115 .rows = sp.sparse_coo.rows,
116 .cols = sp.sparse_coo.cols,
117 .symmetry = static_cast<Symmetry>(sp.sparse_coo.symmetry),
118 .row_indices = typename SparseCOO::index_vector_map_t{sp.sparse_coo.row_indices, sp.sparse_coo.nnz},
119 .col_indices = typename SparseCOO::index_vector_map_t{sp.sparse_coo.col_indices, sp.sparse_coo.nnz},
120 .order = static_cast<SparseCOO::Order>(sp.sparse_coo.order),
121 .first_index = sp.sparse_coo.first_index,
122 };
125 return SparseCOOl{
126 .rows = sp.sparse_coo_l.rows,
127 .cols = sp.sparse_coo_l.cols,
128 .symmetry = static_cast<Symmetry>(sp.sparse_coo_l.symmetry),
129 .row_indices = typename SparseCOOl::index_vector_map_t{sp.sparse_coo_l.row_indices, sp.sparse_coo_l.nnz},
130 .col_indices = typename SparseCOOl::index_vector_map_t{sp.sparse_coo_l.col_indices, sp.sparse_coo_l.nnz},
131 .order = static_cast<SparseCOOl::Order>(sp.sparse_coo_l.order),
132 .first_index = sp.sparse_coo_l.first_index,
133 };
136 return SparseCOOll{
137 .rows = sp.sparse_coo_ll.rows,
138 .cols = sp.sparse_coo_ll.cols,
139 .symmetry = static_cast<Symmetry>(sp.sparse_coo_ll.symmetry),
140 .row_indices = typename SparseCOOll::index_vector_map_t{sp.sparse_coo_ll.row_indices, sp.sparse_coo_ll.nnz},
141 .col_indices = typename SparseCOOll::index_vector_map_t{sp.sparse_coo_ll.col_indices, sp.sparse_coo_ll.nnz},
142 .order = static_cast<SparseCOOll::Order>(sp.sparse_coo_ll.order),
143 .first_index = sp.sparse_coo_ll.first_index,
144 };
145 default: throw std::invalid_argument("Invalid sparsity kind");
146 }
147}
148// clang-format on
149
150} // namespace
151
153 const std::string &function_name, void *user_param)
154 : BoxConstrProblem{0, 0} {
155 handle = load_lib(so_filename);
156 auto *register_func =
158 auto r = register_func(user_param);
159 // Avoid leaking if we throw (or if std::shared_ptr constructor throws)
160 std::unique_ptr<void, void (*)(void *)> unique_inst{r.instance, r.cleanup};
161 std::unique_ptr<alpaqa_function_dict_t> unique_extra{r.extra_functions};
162 std::unique_ptr<alpaqa_exception_ptr_t> unique_exception{r.exception};
163 check_abi_version(r.abi_version);
164 // Check exception thrown by plugin
165 if (unique_exception) {
166 // Here we're facing an interesting problem: the exception we throw
167 // might propagate upwards to a point where this instance is destroyed.
168 // This would cause the shared module to be closed using dlclose.
169 // However, the exception is still stored somewhere in the memory of
170 // that module, causing a segmentation fault when accessed.
171 // To get around this issue, we need to ensure that the shared module
172 // is not closed. Here we simply leak it by storing a shared_ptr to it
173 // in a global variable.
174 leak_lib(handle);
175 std::rethrow_exception(unique_exception->exc);
176 }
177 if (!r.functions)
178 throw std::logic_error("alpaqa::dl::DLProblem::DLProblem: plugin did "
179 "not return any functions");
180 // Store data returned by plugin
181 instance = std::shared_ptr<void>{std::move(unique_inst)};
182 functions = r.functions;
183 extra_funcs = std::shared_ptr<function_dict_t>{std::move(unique_extra)};
184
185 this->n = functions->n;
186 this->m = functions->m;
187 this->C = Box{this->n};
188 this->D = Box{this->m};
190 functions->initialize_box_C(instance.get(), this->C.lowerbound.data(),
191 this->C.upperbound.data());
193 functions->initialize_box_D(instance.get(), this->D.lowerbound.data(),
194 this->D.upperbound.data());
196 length_t = 0;
197 functions->initialize_l1_reg(instance.get(), nullptr, &);
198 if ( > 0) {
199 this->l1_reg.resize(nλ);
200 functions->initialize_l1_reg(instance.get(), this->l1_reg.data(),
201 &);
202 }
203 }
204}
205
207 rvec p) const -> real_t {
208 if (functions->eval_prox_grad_step)
209 return functions->eval_prox_grad_step(
210 instance.get(), γ, x.data(), grad_ψ.data(), x̂.data(), p.data());
211 return BoxConstrProblem<config_t>::eval_prox_grad_step(γ, x, grad_ψ, x̂, p);
212}
213
214// clang-format off
215auto DLProblem::eval_f(crvec x) const -> real_t { return functions->eval_f(instance.get(), x.data()); }
216auto DLProblem::eval_grad_f(crvec x, rvec grad_fx) const -> void { return functions->eval_grad_f(instance.get(), x.data(), grad_fx.data()); }
217auto DLProblem::eval_g(crvec x, rvec gx) const -> void { return functions->eval_g(instance.get(), x.data(), gx.data()); }
218auto DLProblem::eval_grad_g_prod(crvec x, crvec y, rvec grad_gxy) const -> void { return functions->eval_grad_g_prod(instance.get(), x.data(), y.data(), grad_gxy.data()); }
219auto DLProblem::eval_grad_gi(crvec x, index_t i, rvec grad_gi) const -> void { return functions->eval_grad_gi(instance.get(), x.data(), i, grad_gi.data()); }
220auto DLProblem::eval_jac_g(crvec x, rvec J_values) const -> void { return functions->eval_jac_g(instance.get(), x.data(), J_values.size() == 0 ? nullptr : J_values.data()); }
222auto DLProblem::eval_hess_L_prod(crvec x, crvec y, real_t scale, crvec v, rvec Hv) const -> void { return functions->eval_hess_L_prod(instance.get(), x.data(), y.data(), scale, v.data(), Hv.data()); }
223auto DLProblem::eval_hess_L(crvec x, crvec y, real_t scale, rvec H_values) const -> void { return functions->eval_hess_L(instance.get(), x.data(), y.data(), scale, H_values.size() == 0 ? nullptr : H_values.data()); }
225auto DLProblem::eval_hess_ψ_prod(crvec x, crvec y, crvec Σ, real_t scale, crvec v, rvec Hv) const -> void { return functions->eval_hess_ψ_prod(instance.get(), x.data(), y.data(), Σ.data(), scale, D.lowerbound.data(), D.upperbound.data(), v.data(), Hv.data()); }
226auto DLProblem::eval_hess_ψ(crvec x, crvec y, crvec Σ, real_t scale, rvec H_values) const -> void { return functions->eval_hess_ψ(instance.get(), x.data(), y.data(), Σ.data(), scale, D.lowerbound.data(), D.upperbound.data(), H_values.size() == 0 ? nullptr : H_values.data()); }
228auto DLProblem::eval_f_grad_f(crvec x, rvec grad_fx) const -> real_t { return functions->eval_f_grad_f(instance.get(), x.data(), grad_fx.data()); }
229auto DLProblem::eval_f_g(crvec x, rvec g) const -> real_t { return functions->eval_f_g(instance.get(), x.data(), g.data()); }
230auto DLProblem::eval_grad_f_grad_g_prod(crvec x, crvec y, rvec grad_f, rvec grad_gxy) const -> void { return functions->eval_grad_f_grad_g_prod(instance.get(), x.data(), y.data(), grad_f.data(), grad_gxy.data()); }
231auto DLProblem::eval_grad_L(crvec x, crvec y, rvec grad_L, rvec work_n) const -> void { return functions->eval_grad_L(instance.get(), x.data(), y.data(), grad_L.data(), work_n.data()); }
232auto DLProblem::eval_ψ(crvec x, crvec y, crvec Σ, rvec ŷ) const -> real_t { return functions->eval_ψ(instance.get(), x.data(), y.data(), Σ.data(), D.lowerbound.data(), D.upperbound.data(), ŷ.data()); }
233auto DLProblem::eval_grad_ψ(crvec x, crvec y, crvec Σ, rvec grad_ψ, rvec work_n, rvec work_m) const -> void { return functions->eval_grad_ψ(instance.get(), x.data(), y.data(), Σ.data(), D.lowerbound.data(), D.upperbound.data(), grad_ψ.data(), work_n.data(), work_m.data()); }
234auto DLProblem::eval_ψ_grad_ψ(crvec x, crvec y, crvec Σ, rvec grad_ψ, rvec work_n, rvec work_m) const -> real_t { return functions->eval_ψ_grad_ψ(instance.get(), x.data(), y.data(), Σ.data(), D.lowerbound.data(), D.upperbound.data(), grad_ψ.data(), work_n.data(), work_m.data()); }
235
236bool DLProblem::provides_eval_f() const { return functions->eval_f != nullptr; }
237bool DLProblem::provides_eval_grad_f() const { return functions->eval_grad_f != nullptr; }
238bool DLProblem::provides_eval_g() const { return functions->eval_g != nullptr; }
240bool DLProblem::provides_eval_jac_g() const { return functions->eval_jac_g != nullptr; }
242bool DLProblem::provides_eval_grad_gi() const { return functions->eval_grad_gi != nullptr; }
244bool DLProblem::provides_eval_hess_L() const { return functions->eval_hess_L != nullptr; }
247bool DLProblem::provides_eval_hess_ψ() const { return functions->eval_hess_ψ != nullptr; }
250bool DLProblem::provides_eval_f_g() const { return functions->eval_f_g != nullptr; }
252bool DLProblem::provides_eval_grad_L() const { return functions->eval_grad_L != nullptr; }
253bool DLProblem::provides_eval_ψ() const { return functions->eval_ψ != nullptr; }
254bool DLProblem::provides_eval_grad_ψ() const { return functions->eval_grad_ψ != nullptr; }
257// clang-format on
258
259#if ALPAQA_WITH_OCP
260
262 const std::string &function_name,
263 void *user_param) {
264 handle = load_lib(so_filename);
266 handle.get(), function_name);
267 auto r = register_func(user_param);
268 // Avoid leaking if we throw (or if std::shared_ptr constructor throws)
269 std::unique_ptr<void, void (*)(void *)> unique_inst{r.instance, r.cleanup};
270 std::unique_ptr<alpaqa_function_dict_t> unique_extra{r.extra_functions};
271 std::unique_ptr<alpaqa_exception_ptr_t> unique_exception{r.exception};
272 check_abi_version(r.abi_version);
273 // Check exception thrown by plugin
274 if (unique_exception) {
275 // Here we're facing an interesting problem: the exception we throw
276 // might propagate upwards to a point where this instance is destroyed.
277 // This would cause the shared module to be closed using dlclose.
278 // However, the exception is still stored somewhere in the memory of
279 // that module, causing a segmentation fault when accessed.
280 // To get around this issue, we need to ensure that the shared module
281 // is not closed. Here we simply leak it by storing a shared_ptr to it
282 // in a global variable.
283 leak_lib(handle);
284 std::rethrow_exception(unique_exception->exc);
285 }
286 if (!functions)
287 throw std::logic_error("alpaqa::dl::DLControlProblem::DLControlProblem:"
288 " plugin did not return any functions");
289 // Store data returned by plugin
290 instance = std::shared_ptr<void>{std::move(unique_inst)};
291 functions = r.functions;
292 extra_funcs = std::shared_ptr<function_dict_t>{std::move(unique_extra)};
293}
294
295// clang-format off
296auto DLControlProblem::get_U(Box &U) const -> void { return functions->get_U(instance.get(), U.lowerbound.data(), U.upperbound.data()); }
297auto DLControlProblem::get_D(Box &D) const -> void { return functions->get_D(instance.get(), D.lowerbound.data(), D.upperbound.data()); }
298auto DLControlProblem::get_D_N(Box &D) const -> void { return functions->get_D_N(instance.get(), D.lowerbound.data(), D.upperbound.data()); }
299auto DLControlProblem::get_x_init(rvec x_init) const -> void { return functions->get_x_init(instance.get(), x_init.data()); }
300auto DLControlProblem::eval_f(index_t timestep, crvec x, crvec u, rvec fxu) const -> void { return functions->eval_f(instance.get(), timestep, x.data(), u.data(), fxu.data()); }
301auto DLControlProblem::eval_jac_f(index_t timestep, crvec x, crvec u, rmat J_fxu) const -> void { return functions->eval_jac_f(instance.get(), timestep, x.data(), u.data(), J_fxu.data()); }
302auto DLControlProblem::eval_grad_f_prod(index_t timestep, crvec x, crvec u, crvec p, rvec grad_fxu_p) const -> void { return functions->eval_grad_f_prod(instance.get(), timestep, x.data(), u.data(), p.data(), grad_fxu_p.data()); }
303auto DLControlProblem::eval_h(index_t timestep, crvec x, crvec u, rvec h) const -> void { return functions->eval_h(instance.get(), timestep, x.data(), u.data(), h.data()); }
304auto DLControlProblem::eval_h_N(crvec x, rvec h) const -> void { return functions->eval_h_N(instance.get(), x.data(), h.data()); }
305auto DLControlProblem::eval_l(index_t timestep, crvec h) const -> real_t { return functions->eval_l(instance.get(), timestep, h.data()); }
306auto DLControlProblem::eval_l_N(crvec h) const -> real_t { return functions->eval_l_N(instance.get(), h.data()); }
307auto DLControlProblem::eval_qr(index_t timestep, crvec xu, crvec h, rvec qr) const -> void { return functions->eval_qr(instance.get(), timestep, xu.data(), h.data(), qr.data()); }
308auto DLControlProblem::eval_q_N(crvec x, crvec h, rvec q) const -> void { return functions->eval_q_N(instance.get(), x.data(), h.data(), q.data()); }
309auto DLControlProblem::eval_add_Q(index_t timestep, crvec xu, crvec h, rmat Q) const -> void { return functions->eval_add_Q(instance.get(), timestep, xu.data(), h.data(), Q.data()); }
310auto DLControlProblem::eval_add_Q_N(crvec x, crvec h, rmat Q) const -> void { return functions->eval_add_Q_N(instance.get(), x.data(), h.data(), Q.data()); }
311auto DLControlProblem::eval_add_R_masked(index_t timestep, crvec xu, crvec h, crindexvec mask, rmat R, rvec work) const -> void { return functions->eval_add_R_masked(instance.get(), timestep, xu.data(), h.data(), mask.data(), R.data(), work.data()); }
312auto DLControlProblem::eval_add_S_masked(index_t timestep, crvec xu, crvec h, crindexvec mask, rmat S, rvec work) const -> void { return functions->eval_add_S_masked(instance.get(), timestep, xu.data(), h.data(), mask.data(), S.data(), work.data()); }
313auto DLControlProblem::eval_add_R_prod_masked(index_t timestep, crvec xu, crvec h, crindexvec mask_J, crindexvec mask_K, crvec v, rvec out, rvec work) const -> void { return functions->eval_add_R_prod_masked(instance.get(), timestep, xu.data(), h.data(), mask_J.data(), mask_K.data(), v.data(), out.data(), work.data()); }
314auto DLControlProblem::eval_add_S_prod_masked(index_t timestep, crvec xu, crvec h, crindexvec mask_K, crvec v, rvec out, rvec work) const -> void { return functions->eval_add_S_prod_masked(instance.get(), timestep, xu.data(), h.data(), mask_K.data(), v.data(), out.data(), work.data()); }
317auto DLControlProblem::eval_constr(index_t timestep, crvec x, rvec c) const -> void { return functions->eval_constr(instance.get(), timestep, x.data(), c.data()); }
318auto DLControlProblem::eval_constr_N(crvec x, rvec c) const -> void { return functions->eval_constr_N(instance.get(), x.data(), c.data()); }
319auto DLControlProblem::eval_grad_constr_prod(index_t timestep, crvec x, crvec p, rvec grad_cx_p) const -> void { return functions->eval_grad_constr_prod(instance.get(), timestep, x.data(), p.data(), grad_cx_p.data()); }
320auto DLControlProblem::eval_grad_constr_prod_N(crvec x, crvec p, rvec grad_cx_p) const -> void { return functions->eval_grad_constr_prod_N(instance.get(), x.data(), p.data(), grad_cx_p.data()); }
321auto DLControlProblem::eval_add_gn_hess_constr(index_t timestep, crvec x, crvec M, rmat out) const -> void { return functions->eval_add_gn_hess_constr(instance.get(), timestep, x.data(), M.data(), out.data()); }
322auto DLControlProblem::eval_add_gn_hess_constr_N(crvec x, crvec M, rmat out) const -> void { return functions->eval_add_gn_hess_constr_N(instance.get(), x.data(), M.data(), out.data()); }
323
324bool DLControlProblem::provides_get_D() const { return functions->get_D != nullptr; }
325bool DLControlProblem::provides_get_D_N() const { return functions->get_D_N != nullptr; }
337// clang-format on
338
339#endif
340
341} // namespace alpaqa::dl
Implements common problem functions for minimization problems with box constraints.
Box C
Constraints of the decision variables, .
length_t m
Number of constraints, dimension of g(x) and z.
length_t n
Number of decision variables, dimension of x.
real_t eval_prox_grad_step(real_t γ, crvec x, crvec grad_ψ, rvec x̂, rvec p) const
vec l1_reg
(1-norm) regularization parameter.
void eval_add_S_masked(index_t timestep, crvec xu, crvec h, crindexvec mask, rmat S, rvec work) const
bool provides_eval_grad_constr_prod() const
void eval_add_Q_N(crvec x, crvec h, rmat Q) const
bool provides_eval_add_gn_hess_constr() const
DLControlProblem(const std::string &so_filename, const std::string &function_name="register_alpaqa_control_problem", void *user_param=nullptr)
Load a problem from a shared library.
void eval_q_N(crvec x, crvec h, rvec q) const
void eval_add_gn_hess_constr(index_t timestep, crvec x, crvec M, rmat out) const
void eval_grad_constr_prod(index_t timestep, crvec x, crvec p, rvec grad_cx_p) const
void eval_add_R_prod_masked(index_t timestep, crvec xu, crvec h, crindexvec mask_J, crindexvec mask_K, crvec v, rvec out, rvec work) const
bool provides_eval_add_R_prod_masked() const
length_t get_S_work_size() const
void eval_constr_N(crvec x, rvec c) const
void get_D_N(Box &D) const
bool provides_get_S_work_size() const
void eval_grad_f_prod(index_t timestep, crvec x, crvec u, crvec p, rvec grad_fxu_p) const
std::shared_ptr< void > instance
Problem instance created by the registration function, including the deleter to destroy it.
void eval_jac_f(index_t timestep, crvec x, crvec u, rmat J_fxu) const
real_t eval_l_N(crvec h) const
void eval_h_N(crvec x, rvec h) const
real_t eval_l(index_t timestep, crvec h) const
void eval_grad_constr_prod_N(crvec x, crvec p, rvec grad_cx_p) const
control_problem_functions_t * functions
Pointer to the struct of function pointers for evaluating the objective, constraints,...
void eval_add_gn_hess_constr_N(crvec x, crvec M, rmat out) const
void eval_f(index_t timestep, crvec x, crvec u, rvec fxu) const
bool provides_eval_grad_constr_prod_N() const
length_t get_R_work_size() const
void eval_add_S_prod_masked(index_t timestep, crvec xu, crvec h, crindexvec mask_K, crvec v, rvec out, rvec work) const
bool provides_eval_add_S_prod_masked() const
void get_x_init(rvec x_init) const
void eval_h(index_t timestep, crvec x, crvec u, rvec h) const
void eval_add_R_masked(index_t timestep, crvec xu, crvec h, crindexvec mask, rmat R, rvec work) const
std::shared_ptr< void > handle
Handle to the shared module defining the problem.
void eval_qr(index_t timestep, crvec xu, crvec h, rvec qr) const
ExtraFuncs extra_funcs
Dictionary of extra functions that were registered by the problem.
void eval_add_Q(index_t timestep, crvec xu, crvec h, rmat Q) const
bool provides_get_R_work_size() const
bool provides_eval_add_gn_hess_constr_N() const
void eval_constr(index_t timestep, crvec x, rvec c) const
bool provides_eval_hess_L() const
real_t eval_prox_grad_step(real_t γ, crvec x, crvec grad_ψ, rvec x̂, rvec p) const
real_t eval_ψ_grad_ψ(crvec x, crvec y, crvec Σ, rvec grad_ψ, rvec work_n, rvec work_m) const
void eval_grad_gi(crvec x, index_t i, rvec grad_gi) const
bool provides_get_hess_L_sparsity() const
void eval_hess_L_prod(crvec x, crvec y, real_t scale, crvec v, rvec Hv) const
bool provides_eval_hess_ψ_prod() const
void eval_hess_ψ_prod(crvec x, crvec y, crvec Σ, real_t scale, crvec v, rvec Hv) const
bool provides_eval_ψ_grad_ψ() const
bool provides_get_box_C() const
Sparsity get_jac_g_sparsity() const
real_t eval_f_g(crvec x, rvec g) const
void eval_grad_f(crvec x, rvec grad_fx) const
Sparsity get_hess_ψ_sparsity() const
bool provides_eval_jac_g() const
Sparsity get_hess_L_sparsity() const
void eval_grad_f_grad_g_prod(crvec x, crvec y, rvec grad_f, rvec grad_gxy) const
real_t eval_ψ(crvec x, crvec y, crvec Σ, rvec ŷ) const
std::shared_ptr< void > instance
Problem instance created by the registration function, including the deleter to destroy it.
bool provides_eval_g() const
void eval_grad_ψ(crvec x, crvec y, crvec Σ, rvec grad_ψ, rvec work_n, rvec work_m) const
void eval_hess_ψ(crvec x, crvec y, crvec Σ, real_t scale, rvec H_values) const
bool provides_eval_grad_f_grad_g_prod() const
DLProblem(const std::string &so_filename, const std::string &function_name="register_alpaqa_problem", void *user_param=nullptr)
Load a problem from a shared library.
bool provides_eval_f() const
bool provides_get_hess_ψ_sparsity() const
bool provides_eval_hess_L_prod() const
bool provides_eval_grad_g_prod() const
bool provides_get_jac_g_sparsity() const
real_t eval_f_grad_f(crvec x, rvec grad_fx) const
bool provides_eval_f_grad_f() const
bool provides_eval_grad_f() const
bool provides_eval_grad_gi() const
bool provides_eval_f_g() const
problem_functions_t * functions
Pointer to the struct of function pointers for evaluating the objective, constraints,...
void eval_jac_g(crvec x, rvec J_values) const
real_t eval_f(crvec x) const
void eval_hess_L(crvec x, crvec y, real_t scale, rvec H_values) const
bool provides_eval_grad_L() const
void eval_grad_g_prod(crvec x, crvec y, rvec grad_gxy) const
bool provides_eval_grad_ψ() const
std::shared_ptr< void > handle
Handle to the shared module defining the problem.
ExtraFuncs extra_funcs
Dictionary of extra functions that were registered by the problem.
bool provides_eval_hess_ψ() const
void eval_grad_L(crvec x, crvec y, rvec grad_L, rvec work_n) const
void eval_g(crvec x, rvec gx) const
bool provides_eval_ψ() const
#define USING_ALPAQA_CONFIG(Conf)
Definition config.hpp:56
#define ALPAQA_DL_ABI_VERSION
Definition dl-problem.h:9
std::string format_abi_version(uint64_t version)
std::list< std::shared_ptr< void > > leaked_modules
F * load_func(void *handle, const std::string &name)
Sparsity< Conf > convert_sparsity(alpaqa_sparsity_t sp)
void leak_lib(std::shared_ptr< void > handle)
Symmetry
Describes the symmetry of matrices.
Definition sparsity.hpp:12
Dense matrix structure.
Definition sparsity.hpp:21
typename Conf::rmat rmat
Definition config.hpp:74
typename Conf::real_t real_t
Definition config.hpp:65
typename Conf::index_t index_t
Definition config.hpp:77
typename Conf::length_t length_t
Definition config.hpp:76
constexpr const auto inf
Definition config.hpp:85
typename Conf::rvec rvec
Definition config.hpp:69
typename Conf::crvec crvec
Definition config.hpp:70
typename Conf::crindexvec crindexvec
Definition config.hpp:80
Sparse coordinate list structure (COO).
Definition sparsity.hpp:56
Sparse compressed-column structure (CCS or CSC).
Definition sparsity.hpp:29
Stores any of the supported sparsity patterns.
Definition sparsity.hpp:106
void(* eval_add_gn_hess_constr)(void *instance, alpaqa_index_t timestep, const alpaqa_real_t *x, const alpaqa_real_t *M, alpaqa_real_t *out)
Definition dl-problem.h:519
alpaqa_length_t(* get_S_work_size)(void *instance)
Definition dl-problem.h:497
void(* eval_grad_constr_prod_N)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *p, alpaqa_real_t *grad_cx_p)
Definition dl-problem.h:514
void(* get_D)(void *instance, alpaqa_real_t *lb, alpaqa_real_t *ub)
Definition dl-problem.h:391
void(* eval_grad_constr_prod)(void *instance, alpaqa_index_t timestep, const alpaqa_real_t *x, const alpaqa_real_t *p, alpaqa_real_t *grad_cx_p)
Definition dl-problem.h:508
void(* eval_add_R_prod_masked)(void *instance, alpaqa_index_t timestep, const alpaqa_real_t *xu, const alpaqa_real_t *h, const alpaqa_index_t *mask_J, const alpaqa_index_t *mask_K, const alpaqa_real_t *v, alpaqa_real_t *out, alpaqa_real_t *work)
Definition dl-problem.h:476
void(* eval_add_gn_hess_constr_N)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *M, alpaqa_real_t *out)
Definition dl-problem.h:525
alpaqa_length_t(* get_R_work_size)(void *instance)
Definition dl-problem.h:495
void(* get_D_N)(void *instance, alpaqa_real_t *lb, alpaqa_real_t *ub)
Definition dl-problem.h:395
void(* eval_constr)(void *instance, alpaqa_index_t timestep, const alpaqa_real_t *x, alpaqa_real_t *c)
Definition dl-problem.h:499
void(* eval_add_Q_N)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *h, alpaqa_real_t *Q)
Definition dl-problem.h:455
void(* eval_add_S_prod_masked)(void *instance, alpaqa_index_t timestep, const alpaqa_real_t *xu, const alpaqa_real_t *h, const alpaqa_index_t *mask_K, const alpaqa_real_t *v, alpaqa_real_t *out, alpaqa_real_t *work)
Definition dl-problem.h:486
void(* eval_constr_N)(void *instance, const alpaqa_real_t *x, alpaqa_real_t *c)
Definition dl-problem.h:504
void(* eval_grad_f)(void *instance, const alpaqa_real_t *x, alpaqa_real_t *grad_fx)
Gradient of the cost function.
Definition dl-problem.h:173
alpaqa_real_t(* eval_ψ)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, const alpaqa_real_t *Σ, const alpaqa_real_t *zl, const alpaqa_real_t *zu, alpaqa_real_t *ŷ)
Augmented Lagrangian.
Definition dl-problem.h:285
void(* eval_hess_ψ)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, const alpaqa_real_t *Σ, alpaqa_real_t scale, const alpaqa_real_t *zl, const alpaqa_real_t *zu, alpaqa_real_t *H_values)
Hessian of the augmented Lagrangian.
Definition dl-problem.h:242
void(* eval_hess_ψ_prod)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, const alpaqa_real_t *Σ, alpaqa_real_t scale, const alpaqa_real_t *zl, const alpaqa_real_t *zu, const alpaqa_real_t *v, alpaqa_real_t *Hv)
Hessian-vector product of the augmented Lagrangian.
Definition dl-problem.h:230
void(* initialize_box_C)(void *instance, alpaqa_real_t *lb, alpaqa_real_t *ub)
Provide the initial values for the bounds of alpaqa::BoxConstrProblem::C, i.e.
Definition dl-problem.h:331
alpaqa_real_t(* eval_prox_grad_step)(void *instance, alpaqa_real_t γ, const alpaqa_real_t *x, const alpaqa_real_t *grad_ψ, alpaqa_real_t *x̂, alpaqa_real_t *p)
Proximal gradient step.
Definition dl-problem.h:321
alpaqa_real_t(* eval_f_g)(void *instance, const alpaqa_real_t *x, alpaqa_real_t *g)
Cost and constraints.
Definition dl-problem.h:263
void(* eval_grad_f_grad_g_prod)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, alpaqa_real_t *grad_f, alpaqa_real_t *grad_gxy)
Gradient of the cost and gradient-vector product of the constraints.
Definition dl-problem.h:269
void(* eval_grad_L)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, alpaqa_real_t *grad_L, alpaqa_real_t *work_n)
Gradient of the Lagrangian.
Definition dl-problem.h:277
alpaqa_real_t(* eval_f_grad_f)(void *instance, const alpaqa_real_t *x, alpaqa_real_t *grad_fx)
Cost and its gradient.
Definition dl-problem.h:257
alpaqa_real_t(* eval_f)(void *instance, const alpaqa_real_t *x)
Cost function.
Definition dl-problem.h:168
alpaqa_sparsity_t(* get_hess_L_sparsity)(void *instance)
Get the sparsity pattern of the Hessian of the Lagrangian.
Definition dl-problem.h:226
alpaqa_sparsity_t(* get_hess_ψ_sparsity)(void *instance)
Get the sparsity pattern of the Hessian of the augmented Lagrangian.
Definition dl-problem.h:253
alpaqa_length_t m
Number of constraints.
Definition dl-problem.h:163
void(* eval_grad_ψ)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, const alpaqa_real_t *Σ, const alpaqa_real_t *zl, const alpaqa_real_t *zu, alpaqa_real_t *grad_ψ, alpaqa_real_t *work_n, alpaqa_real_t *work_m)
Gradient of the augmented Lagrangian.
Definition dl-problem.h:295
alpaqa_length_t n
Number of decision variables.
Definition dl-problem.h:160
void(* eval_grad_gi)(void *instance, const alpaqa_real_t *x, alpaqa_index_t i, alpaqa_real_t *grad_gi)
Gradient of specific constraint function.
Definition dl-problem.h:202
void(* initialize_l1_reg)(void *instance, alpaqa_real_t *lambda, alpaqa_length_t *size)
Provide the initial values for alpaqa::BoxConstrProblem::l1_reg, the ℓ₁-regularization factor.
Definition dl-problem.h:347
void(* eval_hess_L_prod)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, alpaqa_real_t scale, const alpaqa_real_t *v, alpaqa_real_t *Hv)
Hessian-vector product of the Lagrangian.
Definition dl-problem.h:209
void(* eval_g)(void *instance, const alpaqa_real_t *x, alpaqa_real_t *gx)
Constraints function.
Definition dl-problem.h:179
alpaqa_sparsity_t(* get_jac_g_sparsity)(void *instance)
Get the sparsity pattern of the Jacobian of the constraints function.
Definition dl-problem.h:198
void(* initialize_box_D)(void *instance, alpaqa_real_t *lb, alpaqa_real_t *ub)
Provide the initial values for the bounds of alpaqa::BoxConstrProblem::D, i.e.
Definition dl-problem.h:337
void(* eval_jac_g)(void *instance, const alpaqa_real_t *x, alpaqa_real_t *J_values)
Jacobian of the constraints function.
Definition dl-problem.h:192
alpaqa_real_t(* eval_ψ_grad_ψ)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, const alpaqa_real_t *Σ, const alpaqa_real_t *zl, const alpaqa_real_t *zu, alpaqa_real_t *grad_ψ, alpaqa_real_t *work_n, alpaqa_real_t *work_m)
Augmented Lagrangian and its gradient.
Definition dl-problem.h:307
void(* eval_hess_L)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, alpaqa_real_t scale, alpaqa_real_t *H_values)
Hessian of the Lagrangian.
Definition dl-problem.h:218
void(* eval_grad_g_prod)(void *instance, const alpaqa_real_t *x, const alpaqa_real_t *y, alpaqa_real_t *grad_gxy)
Gradient-vector product of the constraints function.
Definition dl-problem.h:185
Sparsity of matrices.
Definition dl-problem.h:134