Clarabel.cpp/include/cpp/DefaultSolution.hpp at a42179aa67f6f8ba42168668b8ab3f3ab2cb4c43 · oxfordcontrol/Clarabel.cpp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#pragma once

#include <Eigen/Eigen>
#include <cstdint>
#include <type_traits>

namespace clarabel
{

enum class SolverStatus
{
    Unsolved,
    Solved,
    PrimalInfeasible,
    DualInfeasible,
    AlmostSolved,
    AlmostPrimalInfeasible,
    AlmostDualInfeasible,
    MaxIterations,
    MaxTime,
    NumericalError,
    InsufficientProgress,
    CallbackTerminated,
};

template<typename T = double>
class DefaultSolution
{
    static_assert(std::is_same<T, float>::value || std::is_same<T, double>::value, "T must be float or double");

  public:
    struct ClarabelDefaultSolution
    {
        T *x;
        uintptr_t x_length;
        T *z;
        uintptr_t z_length;
        T *s;
        uintptr_t s_length;
        SolverStatus status;
        T obj_val;
        T obj_val_dual;
        double solve_time;
        uint32_t iterations;
        T r_prim;
        T r_dual;
    };

    Eigen::Map<Eigen::VectorX<T>> x, z, s;
    SolverStatus status;
    T obj_val;
    T obj_val_dual;
    double solve_time;
    uint32_t iterations;
    T r_prim;
    T r_dual;

    DefaultSolution(ClarabelDefaultSolution &solution)
        :
        x(solution.x, solution.x_length),
        z(solution.z, solution.z_length),
        s(solution.s, solution.s_length),
        status(solution.status),
        obj_val(solution.obj_val),
        obj_val_dual(solution.obj_val_dual),
        solve_time(solution.solve_time),
        iterations(solution.iterations),
        r_prim(solution.r_prim),
        r_dual(solution.r_dual)
    {
    }
};

// Instantiate the templates
template struct DefaultSolution<double>;
template struct DefaultSolution<float>;

} // namespace clarabel