GurobiLpSolver.h

Go to the documentation of this file.

#ifndef STORM_SOLVER_GUROBILPSOLVER
#define STORM_SOLVER_GUROBILPSOLVER
 
#include <map>
#include <optional>
#include "storm/solver/LpSolver.h"
// To detect whether the usage of Gurobi is possible, this include is neccessary.
#include "storm-config.h"
 
#ifdef STORM_HAVE_GUROBI
extern "C" {
#include "gurobi_c.h"
 
int __stdcall GRBislp(GRBenv**, const char*, const char*, const char*, const char*);
}
#endif
 
namespace storm {
namespace solver {
 
class GurobiEnvironment {
   public:
    GurobiEnvironment() = default;
    GurobiEnvironment(GurobiEnvironment const&) = delete;
    GurobiEnvironment& operator=(GurobiEnvironment const&) = delete;
    virtual ~GurobiEnvironment();
    void initialize();
    void setOutput(bool set = false);
#ifdef STORM_HAVE_GUROBI
    GRBenv* operator*();
#endif
   private:
    bool initialized = false;
#ifdef STORM_HAVE_GUROBI
    GRBenv* env = nullptr;
#endif
};
 
template<typename ValueType, bool RawMode = false>
class GurobiLpSolver : public LpSolver<ValueType, RawMode> {
   public:
    using VariableType = typename LpSolver<ValueType, RawMode>::VariableType;
    using Variable = typename LpSolver<ValueType, RawMode>::Variable;
    using Constant = typename LpSolver<ValueType, RawMode>::Constant;
    using Constraint = typename LpSolver<ValueType, RawMode>::Constraint;
 
    GurobiLpSolver(std::shared_ptr<GurobiEnvironment> const& environment, std::string const& name, OptimizationDirection const& optDir);
 
    GurobiLpSolver(std::shared_ptr<GurobiEnvironment> const& environment, std::string const& name);
 
    GurobiLpSolver(std::shared_ptr<GurobiEnvironment> const& environment, OptimizationDirection const& optDir);
 
    GurobiLpSolver(std::shared_ptr<GurobiEnvironment> const& environment);
 
    GurobiLpSolver(GurobiLpSolver<ValueType> const& other);
 
    virtual ~GurobiLpSolver();
 
    // Methods to add variables.
    virtual Variable addVariable(std::string const& name, VariableType const& type, std::optional<ValueType> const& lowerBound = std::nullopt,
                                 std::optional<ValueType> const& upperBound = std::nullopt, ValueType objectiveFunctionCoefficient = 0) override;
 
    // Methods to incorporate recent changes.
    virtual void update() const override;
 
    // Methods to add constraints
    virtual void addConstraint(std::string const& name, Constraint const& constraint) override;
    virtual void addIndicatorConstraint(std::string const& name, Variable indicatorVariable, bool indicatorValue, Constraint const& constraint) override;
 
    // Methods to optimize and retrieve optimality status.
    virtual void optimize() const override;
    virtual bool isInfeasible() const override;
    virtual bool isUnbounded() const override;
    virtual bool isOptimal() const override;
 
    // Methods to retrieve values of variables and the objective function in the optimal solutions.
    virtual ValueType getContinuousValue(Variable const& name) const override;
    virtual int_fast64_t getIntegerValue(Variable const& name) const override;
    virtual bool getBinaryValue(Variable const& name) const override;
    virtual ValueType getObjectiveValue() const override;
    // Methods to print the LP problem to a file.
    virtual void writeModelToFile(std::string const& filename) const override;
 
    virtual void push() override;
    virtual void pop() override;
 
    virtual void setMaximalMILPGap(ValueType const& gap, bool relative) override;
    virtual ValueType getMILPGap(bool relative) const override;
 
    // Methods to retrieve values of sub-optimal solutions found along the way.
    void setMaximalSolutionCount(uint64_t value);  // How many solutions will be stored (at max)
    uint64_t getSolutionCount() const;             // How many solutions have been found
    ValueType getContinuousValue(Variable const& name, uint64_t const& solutionIndex) const;
    int_fast64_t getIntegerValue(Variable const& name, uint64_t const& solutionIndex) const;
    bool getBinaryValue(Variable const& name, uint64_t const& solutionIndex) const;
    ValueType getObjectiveValue(uint64_t solutionIndex) const;
 
   private:
#ifdef STORM_HAVE_GUROBI
    // The Gurobi model.
    GRBmodel* model;
#endif
 
    std::shared_ptr<GurobiEnvironment> environment;
 
    // The index of the next variable.
    int nextVariableIndex;
 
    // A mapping from variables to their indices.
    std::map<storm::expressions::Variable, int> variableToIndexMap;
 
    struct IncrementalLevel {
        std::vector<storm::expressions::Variable> variables;
        // Gurobi considers a different set of indices for linear constraints and general constraints.
        // The term "general constraints" is used by Gurobi to refer to "non-standard" constraints, e.g., indicator constraint.
        int firstConstraintIndex;
        int firstGenConstraintIndex;
    };
    std::vector<IncrementalLevel> incrementalData;
};
 
enum class GurobiSolverMethod { AUTOMATIC = -1, PRIMALSIMPLEX = 0, DUALSIMPLEX = 1, BARRIER = 2, CONCURRENT = 3, DETCONCURRENT = 4, DETCONCURRENTSIMPLEX = 5 };
 
std::string toString(GurobiSolverMethod const& method);
std::optional<GurobiSolverMethod> gurobiSolverMethodFromString(std::string const&);
std::vector<GurobiSolverMethod> getGurobiSolverMethods();
 
}  // namespace solver
}  // namespace storm
 
#endif /* STORM_SOLVER_GUROBILPSOLVER */