lin_expr.hpp

#pragma once
 
#include <vector>
 
#include "scippp/var.hpp"
 
namespace scippp {
 
class LinExpr {
    friend class Model;
    double m_constant { 0.0 };
    std::vector<Var> m_vars {};
    std::vector<double> m_coeffs {};
 
public:
    LinExpr() = default;
    template <typename Arithmetic, std::enable_if_t<std::is_arithmetic_v<Arithmetic>, bool> = true>
    LinExpr(Arithmetic constant)
        : m_constant { static_cast<double>(constant) }
    {
    }
    LinExpr(const Var& var);
 
    [[nodiscard]] double getConstant() const;
 
    LinExpr& operator+=(const LinExpr& expr);
    LinExpr& operator-=(const LinExpr& expr);
    LinExpr& operator*=(double factor);
};
 
LinExpr operator*(double factor, LinExpr rhs);
 
LinExpr operator+(LinExpr lhs, const LinExpr& rhs);
 
LinExpr operator-(LinExpr lhs, const LinExpr& rhs);
 
}