problem.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "constraint.hpp"
 
namespace cvx
{
 
    class OptimizationProblem
    {
    public:
        OptimizationProblem() = default;
 
        Scalar addVariable(const std::string &name);
 
        VectorX addVariable(const std::string &name,
                            size_t rows);
 
        MatrixX addVariable(const std::string &name,
                            size_t rows,
                            size_t cols);
 
        void addConstraint(const Constraint &constraint);
 
        void addConstraint(const std::vector<Constraint> &constraints);
 
        void addCostTerm(const Scalar &term);
 
        void getVariableValue(const std::string &name, double &var);
 
        void getVariableValue(const std::string &name, Eigen::VectorXd &var);
 
        void getVariableValue(const std::string &name, Eigen::MatrixXd &var);
 
        void getVariable(const std::string &name, Scalar &var);
 
        void getVariable(const std::string &name, VectorX &var);
 
        void getVariable(const std::string &name, MatrixX &var);
 
        double getOptimalValue() const;
 
        size_t getNumVariables() const;
 
        friend std::ostream &operator<<(std::ostream &os, const OptimizationProblem &socp);
        friend internal::SOCPWrapperBase;
        friend internal::QPWrapperBase;
 
    private:
        OptimizationProblem(const OptimizationProblem &other);
 
        Scalar costFunction;
 
        std::vector<internal::EqualityConstraint> equality_constraints;
        std::vector<internal::PositiveConstraint> positive_constraints;
        std::vector<internal::BoxConstraint> box_constraints;
        std::vector<internal::SecondOrderConeConstraint> second_order_cone_constraints;
 
        std::map<std::string, Scalar> scalar_variables;
        std::map<std::string, VectorX> vector_variables;
        std::map<std::string, MatrixX> matrix_variables;
    };
 
} // namespace cvx