Inequality.hpp

Go to the documentation of this file.

#ifndef INEQUALITY_HPP_
#define INEQUALITY_HPP_
 
#include <vector>
 
namespace ZonoOpt
{
 
struct IneqTerm
{
    int idx; 
 
    zono_float coeff; 
};
 
enum IneqType
{
    LESS_OR_EQUAL = -1, // less than or equal to
    GREATER_OR_EQUAL = 1, // greater than or equal to
    EQUAL = 0, // equal to
    LESS = -2, // strictly less than
    GREATER = 2 // strictly greater than
};
 
class Inequality {
public:
    explicit Inequality(const int n_dims) : n_dims(n_dims) {}
 
    Inequality(const int n_dims, const std::vector<IneqTerm>& terms, const zono_float rhs, const IneqType type) :
        n_dims(n_dims), terms(terms), rhs(rhs), type(type) {}
 
    void add_term(const int idx, const zono_float coeff)
    {
        IneqTerm term{};
        term.idx = idx;
        term.coeff = coeff;
        this->terms.push_back(term);
    }
 
    void set_rhs(const zono_float rhs)
    {
        this->rhs = rhs;
    }
 
    void set_ineq_type(const IneqType type)
    {
        this->type = type;
    }
 
    const std::vector<IneqTerm>& get_terms() const
    {
        return this->terms;
    }
 
    zono_float get_rhs() const
    {
        return this->rhs;
    }
 
    IneqType get_ineq_type() const
    {
        return this->type;
    }
 
    int get_n_dims() const
    {
        return n_dims;
    }
 
private:
    int n_dims;
 
    std::vector<IneqTerm> terms;
 
    zono_float rhs = 0;
 
    IneqType type = EQUAL;
};
 
}
 
#endif