alpaqa sparse
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, std::string symbol_prefix, std::string_view name) {
52 assert(handle);
54 full_name += '_';
55 full_name += name;
56 ::dlerror();
57 auto *h = ::dlsym(handle, full_name.c_str());
58 if (auto *err = ::dlerror())
59 throw std::runtime_error(err);
60 // We can only hope that the user got the signature right ...
61 return reinterpret_cast<F *>(h);
62}
63
65std::list<std::shared_ptr<void>> leaked_modules;
66void leak_lib(std::shared_ptr<void> handle) {
67 std::lock_guard lck{leaked_modules_mutex};
68 leaked_modules.emplace_back(std::move(handle));
69}
70
71// clang-format off
72template <Config Conf>
75 switch (sp.kind) {
78 using Dense = sparsity::Dense<config_t>;
79 return Dense{
80 .rows = sp.dense.rows,
81 .cols = sp.dense.cols,
82 .symmetry = static_cast<Symmetry>(sp.dense.symmetry),
83 };
85 using SparseCSC = sparsity::SparseCSC<config_t, int>;
86 return SparseCSC{
87 .rows = sp.sparse_csc.rows,
88 .cols = sp.sparse_csc.cols,
89 .symmetry = static_cast<Symmetry>(sp.sparse_csc.symmetry),
90 .inner_idx = typename SparseCSC::index_vector_map_t{sp.sparse_csc.inner_idx, sp.sparse_csc.nnz},
91 .outer_ptr = typename SparseCSC::index_vector_map_t{sp.sparse_csc.outer_ptr, sp.sparse_csc.cols + 1},
92 .order = static_cast<SparseCSC::Order>(sp.sparse_csc.order),
93 };
96 return SparseCSCl{
97 .rows = sp.sparse_csc_l.rows,
98 .cols = sp.sparse_csc_l.cols,
99 .symmetry = static_cast<Symmetry>(sp.sparse_csc_l.symmetry),
100 .inner_idx = typename SparseCSCl::index_vector_map_t{sp.sparse_csc_l.inner_idx, sp.sparse_csc_l.nnz},
101 .outer_ptr = typename SparseCSCl::index_vector_map_t{sp.sparse_csc_l.outer_ptr, sp.sparse_csc_l.cols + 1},
102 .order = static_cast<SparseCSCl::Order>(sp.sparse_csc_l.order),
103 };
106 return SparseCSCll{
107 .rows = sp.sparse_csc_ll.rows,
108 .cols = sp.sparse_csc_ll.cols,
109 .symmetry = static_cast<Symmetry>(sp.sparse_csc_ll.symmetry),
110 .inner_idx = typename SparseCSCll::index_vector_map_t{sp.sparse_csc_ll.inner_idx, sp.sparse_csc_ll.nnz},
111 .outer_ptr = typename SparseCSCll::index_vector_map_t{sp.sparse_csc_ll.outer_ptr, sp.sparse_csc_ll.cols + 1},
112 .order = static_cast<SparseCSCll::Order>(sp.sparse_csc_ll.order),
113 };
115 using SparseCOO = sparsity::SparseCOO<config_t, int>;
116 return SparseCOO{
117 .rows = sp.sparse_coo.rows,
118 .cols = sp.sparse_coo.cols,
119 .symmetry = static_cast<Symmetry>(sp.sparse_coo.symmetry),
120 .row_indices = typename SparseCOO::index_vector_map_t{sp.sparse_coo.row_indices, sp.sparse_coo.nnz},
121 .col_indices = typename SparseCOO::index_vector_map_t{sp.sparse_coo.col_indices, sp.sparse_coo.nnz},
122 .order = static_cast<SparseCOO::Order>(sp.sparse_coo.order),
123 .first_index = sp.sparse_coo.first_index,
124 };
127 return SparseCOOl{
128 .rows = sp.sparse_coo_l.rows,
129 .cols = sp.sparse_coo_l.cols,
130 .symmetry = static_cast<Symmetry>(sp.sparse_coo_l.symmetry),
131 .row_indices = typename SparseCOOl::index_vector_map_t{sp.sparse_coo_l.row_indices, sp.sparse_coo_l.nnz},
132 .col_indices = typename SparseCOOl::index_vector_map_t{sp.sparse_coo_l.col_indices, sp.sparse_coo_l.nnz},
133 .order = static_cast<SparseCOOl::Order>(sp.sparse_coo_l.order),
134 .first_index = sp.sparse_coo_l.first_index,
135 };
138 return SparseCOOll{
139 .rows = sp.sparse_coo_ll.rows,
140 .cols = sp.sparse_coo_ll.cols,
141 .symmetry = static_cast<Symmetry>(sp.sparse_coo_ll.symmetry),
142 .row_indices = typename SparseCOOll::index_vector_map_t{sp.sparse_coo_ll.row_indices, sp.sparse_coo_ll.nnz},
143 .col_indices = typename SparseCOOll::index_vector_map_t{sp.sparse_coo_ll.col_indices, sp.sparse_coo_ll.nnz},
144 .order = static_cast<SparseCOOll::Order>(sp.sparse_coo_ll.order),
145 .first_index = sp.sparse_coo_ll.first_index,
146 };
147 default: throw std::invalid_argument("Invalid sparsity kind");
148 }
149}
150// clang-format on
151
152} // namespace
153
154DLProblem::DLProblem(const std::string &so_filename, std::string symbol_prefix,
155 void *user_param)
156 : BoxConstrProblem{0, 0} {
157 handle = load_lib(so_filename);
159 handle.get(), std::move(symbol_prefix), "register");
160 auto r = register_func(user_param);
161 // Avoid leaking if we throw (or if std::shared_ptr constructor throws)
162 std::unique_ptr<void, void (*)(void *)> unique_inst{r.instance, r.cleanup};
163 std::unique_ptr<alpaqa_function_dict_t> unique_extra{r.extra_functions};
164 std::unique_ptr<alpaqa_exception_ptr_t> unique_exception{r.exception};
165 check_abi_version(r.abi_version);
166 // Check exception thrown by plugin
167 if (unique_exception) {
168 // Here we're facing an interesting problem: the exception we throw
169 // might propagate upwards to a point where this instance is destroyed.
170 // This would cause the shared module to be closed using dlclose.
171 // However, the exception is still stored somewhere in the memory of
172 // that module, causing a segmentation fault when accessed.
173 // To get around this issue, we need to ensure that the shared module
174 // is not closed. Here we simply leak it by storing a shared_ptr to it
175 // in a global variable.
176 leak_lib(handle);
177 std::rethrow_exception(unique_exception->exc);
178 }
179 if (!r.functions)
180 throw std::logic_error("alpaqa::dl::DLProblem::DLProblem: plugin did "
181 "not return any functions");
182 // Store data returned by plugin
183 instance = std::shared_ptr<void>{std::move(unique_inst)};
184 functions = r.functions;
185 extra_funcs = std::shared_ptr<function_dict_t>{std::move(unique_extra)};
186
187 this->n = functions->n;
188 this->m = functions->m;
189 this->C = Box{this->n};
190 this->D = Box{this->m};
192 functions->initialize_box_C(instance.get(), this->C.lowerbound.data(),
193 this->C.upperbound.data());
195 functions->initialize_box_D(instance.get(), this->D.lowerbound.data(),
196 this->D.upperbound.data());
198 length_t = 0;
199 functions->initialize_l1_reg(instance.get(), nullptr, &);
200 if ( > 0) {
201 this->l1_reg.resize(nλ);
202 functions->initialize_l1_reg(instance.get(), this->l1_reg.data(),
203 &);
204 }
205 }
206}
207
209 rvec p) const -> real_t {
210 if (functions->eval_prox_grad_step)
211 return functions->eval_prox_grad_step(
212 instance.get(), γ, x.data(), grad_ψ.data(), x̂.data(), p.data());
213 return BoxConstrProblem<config_t>::eval_prox_grad_step(γ, x, grad_ψ, x̂, p);
214}
215
216// clang-format off
217auto DLProblem::eval_f(crvec x) const -> real_t { return functions->eval_f(instance.get(), x.data()); }
218auto DLProblem::eval_grad_f(crvec x, rvec grad_fx) const -> void { return functions->eval_grad_f(instance.get(), x.data(), grad_fx.data()); }
219auto DLProblem::eval_g(crvec x, rvec gx) const -> void { return functions->eval_g(instance.get(), x.data(), gx.data()); }
220auto 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()); }
221auto 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()); }
222auto 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()); }
224auto 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()); }
225auto 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()); }
227auto 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()); }
228auto 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()); }
230auto 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()); }
231auto DLProblem::eval_f_g(crvec x, rvec g) const -> real_t { return functions->eval_f_g(instance.get(), x.data(), g.data()); }
232auto 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()); }
233auto 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()); }
234auto 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()); }
235auto 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()); }
236auto 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()); }
237
238bool DLProblem::provides_eval_f() const { return functions->eval_f != nullptr; }
239bool DLProblem::provides_eval_grad_f() const { return functions->eval_grad_f != nullptr; }
240bool DLProblem::provides_eval_g() const { return functions->eval_g != nullptr; }
242bool DLProblem::provides_eval_jac_g() const { return functions->eval_jac_g != nullptr; }
244bool DLProblem::provides_eval_grad_gi() const { return functions->eval_grad_gi != nullptr; }
246bool DLProblem::provides_eval_hess_L() const { return functions->eval_hess_L != nullptr; }
249bool DLProblem::provides_eval_hess_ψ() const { return functions->eval_hess_ψ != nullptr; }
252bool DLProblem::provides_eval_f_g() const { return functions->eval_f_g != nullptr; }
254bool DLProblem::provides_eval_grad_L() const { return functions->eval_grad_L != nullptr; }
255bool DLProblem::provides_eval_ψ() const { return functions->eval_ψ != nullptr; }
256bool DLProblem::provides_eval_grad_ψ() const { return functions->eval_grad_ψ != nullptr; }
259// clang-format on
260
261#if ALPAQA_WITH_OCP
262
264 std::string symbol_prefix,
265 void *user_param) {
266 handle = load_lib(so_filename);
268 handle.get(), std::move(symbol_prefix), "register");
269 auto r = register_func(user_param);
270 // Avoid leaking if we throw (or if std::shared_ptr constructor throws)
271 std::unique_ptr<void, void (*)(void *)> unique_inst{r.instance, r.cleanup};
272 std::unique_ptr<alpaqa_function_dict_t> unique_extra{r.extra_functions};
273 std::unique_ptr<alpaqa_exception_ptr_t> unique_exception{r.exception};
274 check_abi_version(r.abi_version);
275 // Check exception thrown by plugin
276 if (unique_exception) {
277 // Here we're facing an interesting problem: the exception we throw
278 // might propagate upwards to a point where this instance is destroyed.
279 // This would cause the shared module to be closed using dlclose.
280 // However, the exception is still stored somewhere in the memory of
281 // that module, causing a segmentation fault when accessed.
282 // To get around this issue, we need to ensure that the shared module
283 // is not closed. Here we simply leak it by storing a shared_ptr to it
284 // in a global variable.
285 leak_lib(handle);
286 std::rethrow_exception(unique_exception->exc);
287 }
288 if (!functions)
289 throw std::logic_error("alpaqa::dl::DLControlProblem::DLControlProblem:"
290 " plugin did not return any functions");
291 // Store data returned by plugin
292 instance = std::shared_ptr<void>{std::move(unique_inst)};
293 functions = r.functions;
294 extra_funcs = std::shared_ptr<function_dict_t>{std::move(unique_extra)};
295}
296
297// clang-format off
298auto DLControlProblem::get_U(Box &U) const -> void { return functions->get_U(instance.get(), U.lowerbound.data(), U.upperbound.data()); }
299auto DLControlProblem::get_D(Box &D) const -> void { return functions->get_D(instance.get(), D.lowerbound.data(), D.upperbound.data()); }
300auto DLControlProblem::get_D_N(Box &D) const -> void { return functions->get_D_N(instance.get(), D.lowerbound.data(), D.upperbound.data()); }
301auto DLControlProblem::get_x_init(rvec x_init) const -> void { return functions->get_x_init(instance.get(), x_init.data()); }
302auto 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()); }
303auto 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()); }
304auto 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()); }
305auto 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()); }
306auto DLControlProblem::eval_h_N(crvec x, rvec h) const -> void { return functions->eval_h_N(instance.get(), x.data(), h.data()); }
307auto DLControlProblem::eval_l(index_t timestep, crvec h) const -> real_t { return functions->eval_l(instance.get(), timestep, h.data()); }
308auto DLControlProblem::eval_l_N(crvec h) const -> real_t { return functions->eval_l_N(instance.get(), h.data()); }
309auto 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()); }
310auto 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()); }
311auto 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()); }
312auto 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()); }
313auto 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()); }
314auto 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()); }
315auto 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()); }
316auto 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()); }
319auto DLControlProblem::eval_constr(index_t timestep, crvec x, rvec c) const -> void { return functions->eval_constr(instance.get(), timestep, x.data(), c.data()); }
320auto DLControlProblem::eval_constr_N(crvec x, rvec c) const -> void { return functions->eval_constr_N(instance.get(), x.data(), c.data()); }
321auto 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()); }
322auto 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()); }
323auto 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()); }
324auto 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()); }
325
326bool DLControlProblem::provides_get_D() const { return functions->get_D != nullptr; }
327bool DLControlProblem::provides_get_D_N() const { return functions->get_D_N != nullptr; }
339// clang-format on
340
341#endif
342
343} // 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.
DLControlProblem(const std::string &so_filename, std::string symbol_prefix="alpaqa_control_problem", void *user_param=nullptr)
Load a problem from a shared library.
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
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
DLProblem(const std::string &so_filename, std::string symbol_prefix="alpaqa_problem", void *user_param=nullptr)
Load a problem from a shared library.
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
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)
F * load_func(void *handle, std::string symbol_prefix, std::string_view name)
std::list< std::shared_ptr< void > > leaked_modules
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:11
Dense matrix structure.
Definition sparsity.hpp:20
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:55
Sparse compressed-column structure (CCS or CSC).
Definition sparsity.hpp:28
Stores any of the supported sparsity patterns.
Definition sparsity.hpp:105
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