Dekker1/on-restart-benchmarks
Archived
1
0
Fork 0
Code
This repository has been archived on 2025-03-06. You can view files and clone it, but cannot push or open issues or pull requests.
on-restart-benchmarks/solvers/MIP/MIP_gurobi_wrap.cpp
Jip J. Dekker f2a1c4e389 Squashed 'software/mza/' content from commit f970a59b17 
git-subtree-dir: software/mza
git-subtree-split: f970a59b177c13ca3dd8aaef8cc6681d83b7e813
2021-07-11 16:34:30 +10:00

816 lines
32 KiB
C++
Raw Blame History

// * -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */
/*
* Main authors:
* Gleb Belov <gleb.belov@monash.edu>
*/
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifdef _MSC_VER
#define _CRT_SECURE_NO_WARNINGS
#endif
#include <minizinc/config.hh>
#include <minizinc/exception.hh>
#include <minizinc/file_utils.hh>
#include <minizinc/utils_savestream.hh>
#include <cmath>
#include <cstring>
#include <ctime>
#include <fstream>
#include <iomanip>
#include <iostream>
#include <sstream>
#include <stdexcept>
#include <string>
#ifdef GUROBI_PLUGIN
#ifdef HAS_DLFCN_H
#include <dlfcn.h>
#elif defined HAS_WINDOWS_H
#include <Windows.h>
#endif
#endif
using namespace std;
#include <minizinc/solvers/MIP/MIP_gurobi_wrap.hh>
#include <minizinc/utils.hh>
string MIP_gurobi_wrapper::getDescription(MiniZinc::SolverInstanceBase::Options* opt) {
ostringstream oss;
oss << "MIP wrapper for Gurobi library " << getVersion();
oss << ". Compiled " __DATE__ " " __TIME__;
return oss.str();
}
string MIP_gurobi_wrapper::getVersion(MiniZinc::SolverInstanceBase::Options* opt) {
ostringstream oss;
MIP_gurobi_wrapper mgw(nullptr); // to avoid opening the env
try {
mgw.checkDLL();
int major, minor, technical;
mgw.dll_GRBversion(&major, &minor, &technical);
oss << major << '.' << minor << '.' << technical;
return oss.str();
} catch (MiniZinc::InternalError&) {
return "<unknown version>";
}
}
string MIP_gurobi_wrapper::needDllFlag() {
MIP_gurobi_wrapper mgw(NULL);
try {
mgw.checkDLL();
return "";
} catch (MiniZinc::InternalError&) {
return "--gurobi-dll";
}
}
string MIP_gurobi_wrapper::getId() { return "gurobi"; }
string MIP_gurobi_wrapper::getName() { return "Gurobi"; }
vector<string> MIP_gurobi_wrapper::getTags() { return {"mip", "float", "api"}; }
vector<string> MIP_gurobi_wrapper::getStdFlags() { return {"-a", "-n", "-p", "-s"}; }
const vector<string>& gurobiDLLs(void) {
static const vector<string> sGurobiDLLs = {"gurobi90", "gurobi85", "gurobi81", "gurobi80",
"gurobi75", "gurobi70", "gurobi65"};
return sGurobiDLLs;
}
void MIP_gurobi_wrapper::Options::printHelp(ostream& os) {
os << "GUROBI MIP wrapper options:"
<< std::endl
// -s print statistics
// << " --readParam <file> read GUROBI parameters from file
// << "--writeParam <file> write GUROBI parameters to file
// << "--tuneParam instruct GUROBI to tune parameters instead of solving
<< " -f\n free search (default)" << std::endl
<< " --fixed-search\n fixed search (approximation of the model's one by branching "
"priorities)"
<< std::endl
<< " --uniform-search\n 'more fixed' search (all variables in the search anns get "
"priority 1)"
<< std::endl
<< " --mipfocus <n>\n 1: feasibility, 2: optimality, 3: move bound (default is 0, "
"balanced)"
<< std::endl
<< " -a\n print intermediate solutions (use for optimization problems only TODO)"
<< std::endl
<< " -p <N>\n use N threads, default: 1."
<< std::endl
// << " --nomippresolve disable MIP presolving NOT IMPL" << std::endl
<< " --solver-time-limit <N>, --solver-time\n"
" stop search after N milliseconds wall time"
<< std::endl
<< " --solver-time-limit-feas <N>, --solver-tlf\n"
" stop search after N milliseconds wall time after the first feasible solution"
<< std::endl
<< " -n <N>, --num-solutions <N>\n"
" stop search after N solutions"
<< std::endl
<< " --workmem <N>, --nodefilestart <N>\n"
" maximal RAM for node tree used before writing to node file, GB, default: 3"
<< std::endl
<< " --writeModel <file>\n write model to <file> (.lp, .mps, .sav, ...)" << std::endl
<< " --readParam <file>\n read GUROBI parameters from file" << std::endl
<< " --writeParam <file>\n write GUROBI parameters to file"
<< std::endl
// << " --tuneParam instruct GUROBI to tune parameters instead of solving NOT
// IMPL"
<< "\n --absGap <n>\n absolute gap |primal-dual| to stop" << std::endl
<< " --relGap <n>\n relative gap |primal-dual|/<solver-dep> to stop. Default 1e-8, set <0 "
"to use backend's default"
<< std::endl
<< " --feasTol <n>\n primal feasibility tolerance. Default 1e-8" << std::endl
<< " --intTol <n>\n integrality tolerance for a variable. Gurobi recommends at least "
"feasTol. Default 1e-8"
<< std::endl
// << " --objDiff <n> objective function discretization. Default 1.0" << std::endl
<< "\n --gurobi-dll <file> or <basename>\n Gurobi DLL, or base name, such as gurobi75, "
"when using plugin. Default range tried: "
<< gurobiDLLs().front() << " .. " << gurobiDLLs().back() << std::endl
<< std::endl;
}
bool MIP_gurobi_wrapper::Options::processOption(int& i, std::vector<std::string>& argv) {
MiniZinc::CLOParser cop(i, argv);
if (string(argv[i]) == "-a" || string(argv[i]) == "--all" ||
string(argv[i]) == "--all-solutions") {
flag_all_solutions = true;
} else if (string(argv[i]) == "-f") {
} else if (string(argv[i]) == "--fixed-search") {
nFreeSearch = 0;
} else if (string(argv[i]) == "--uniform-search") {
nFreeSearch = 2;
} else if (cop.get("--mipfocus --mipFocus --MIPFocus --MIPfocus", &nMIPFocus)) {
} else if (cop.get("--writeModel", &sExportModel)) {
} else if (cop.get("-p", &nThreads)) {
} else if (cop.get("--solver-time-limit --solver-time", &nTimeout1000)) {
} else if (cop.get("--solver-time-limit-feas --solver-tlf", &nTimeoutFeas1000)) {
} else if (cop.get("-n --num-solutions", &nSolLimit)) {
} else if (cop.get("--workmem --nodefilestart", &nWorkMemLimit)) {
} else if (cop.get("--readParam", &sReadParams)) {
} else if (cop.get("--writeParam", &sWriteParams)) {
} else if (cop.get("--absGap", &absGap)) {
} else if (cop.get("--relGap", &relGap)) {
} else if (cop.get("--feasTol", &feasTol)) {
} else if (cop.get("--intTol", &intTol)) {
} else if (cop.get("--gurobi-dll", &sGurobiDLL)) {
// } else if ( cop.get( "--objDiff", &objDiff ) ) {
} else
return false;
return true;
}
void MIP_gurobi_wrapper::wrap_assert(bool cond, string msg, bool fTerm) {
if (!cond) {
gurobi_buffer = "[NO ERROR STRING GIVEN]";
if (error) {
gurobi_buffer = dll_GRBgeterrormsg(env);
}
string msgAll =
(" MIP_gurobi_wrapper runtime error: " + gurobi_buffer + "\nMessage from caller: " + msg);
cerr << msgAll << "\nGurobi error code: " << error << endl;
if (fTerm) {
cerr << "TERMINATING." << endl;
throw runtime_error(msgAll);
}
}
}
#ifdef GUROBI_PLUGIN
namespace {
void* dll_open(const char* file) {
#ifdef HAS_DLFCN_H
if (MiniZinc::FileUtils::is_absolute(file)) {
return dlopen(file, RTLD_NOW);
} else {
return dlopen((std::string("lib") + file + ".so").c_str(), RTLD_NOW);
}
#else
if (MiniZinc::FileUtils::is_absolute(file)) {
return LoadLibrary(file);
} else {
return LoadLibrary((std::string(file) + ".dll").c_str());
}
#endif
}
void* dll_sym(void* dll, const char* sym) {
#ifdef HAS_DLFCN_H
void* ret = dlsym(dll, sym);
#else
void* ret = GetProcAddress((HMODULE)dll, sym);
#endif
if (ret == NULL)
throw MiniZinc::InternalError("cannot load symbol " + string(sym) + " from gurobi dll");
return ret;
}
void dll_close(void* dll) {
#ifdef HAS_DLFCN_H
dlclose(dll);
#else
FreeLibrary((HMODULE)dll);
#endif
}
} // namespace
#endif
void MIP_gurobi_wrapper::checkDLL() {
#ifdef GUROBI_PLUGIN
gurobi_dll = NULL;
if (options && options->sGurobiDLL.size()) {
gurobi_dll = dll_open(options->sGurobiDLL.c_str());
} else {
for (const auto& s : gurobiDLLs()) {
gurobi_dll = dll_open(s.c_str());
if (NULL != gurobi_dll) {
break;
}
}
}
if (gurobi_dll == NULL) {
if (options == NULL || options->sGurobiDLL.empty()) {
throw MiniZinc::InternalError("cannot load gurobi dll, specify --gurobi-dll");
} else {
throw MiniZinc::InternalError("cannot load gurobi dll `" + options->sGurobiDLL + "'");
}
}
*(void**)(&dll_GRBversion) = dll_sym(gurobi_dll, "GRBversion");
*(void**)(&dll_GRBaddconstr) = dll_sym(gurobi_dll, "GRBaddconstr");
*(void**)(&dll_GRBaddgenconstrIndicator) = dll_sym(gurobi_dll, "GRBaddgenconstrIndicator");
*(void**)(&dll_GRBaddvars) = dll_sym(gurobi_dll, "GRBaddvars");
*(void**)(&dll_GRBcbcut) = dll_sym(gurobi_dll, "GRBcbcut");
*(void**)(&dll_GRBcbget) = dll_sym(gurobi_dll, "GRBcbget");
*(void**)(&dll_GRBcblazy) = dll_sym(gurobi_dll, "GRBcblazy");
*(void**)(&dll_GRBfreeenv) = dll_sym(gurobi_dll, "GRBfreeenv");
*(void**)(&dll_GRBfreemodel) = dll_sym(gurobi_dll, "GRBfreemodel");
*(void**)(&dll_GRBgetdblattr) = dll_sym(gurobi_dll, "GRBgetdblattr");
*(void**)(&dll_GRBgetdblattrarray) = dll_sym(gurobi_dll, "GRBgetdblattrarray");
*(void**)(&dll_GRBgetenv) = dll_sym(gurobi_dll, "GRBgetenv");
*(void**)(&dll_GRBgeterrormsg) = dll_sym(gurobi_dll, "GRBgeterrormsg");
*(void**)(&dll_GRBgetintattr) = dll_sym(gurobi_dll, "GRBgetintattr");
*(void**)(&dll_GRBloadenv) = dll_sym(gurobi_dll, "GRBloadenv");
*(void**)(&dll_GRBnewmodel) = dll_sym(gurobi_dll, "GRBnewmodel");
*(void**)(&dll_GRBoptimize) = dll_sym(gurobi_dll, "GRBoptimize");
*(void**)(&dll_GRBreadparams) = dll_sym(gurobi_dll, "GRBreadparams");
*(void**)(&dll_GRBsetcallbackfunc) = dll_sym(gurobi_dll, "GRBsetcallbackfunc");
*(void**)(&dll_GRBsetdblparam) = dll_sym(gurobi_dll, "GRBsetdblparam");
*(void**)(&dll_GRBsetintattr) = dll_sym(gurobi_dll, "GRBsetintattr");
*(void**)(&dll_GRBsetintattrlist) = dll_sym(gurobi_dll, "GRBsetintattrlist");
*(void**)(&dll_GRBsetdblattrelement) = dll_sym(gurobi_dll, "GRBsetdblattrelement");
*(void**)(&dll_GRBsetdblattrlist) = dll_sym(gurobi_dll, "GRBsetdblattrlist");
*(void**)(&dll_GRBsetintparam) = dll_sym(gurobi_dll, "GRBsetintparam");
*(void**)(&dll_GRBsetstrparam) = dll_sym(gurobi_dll, "GRBsetstrparam");
*(void**)(&dll_GRBterminate) = dll_sym(gurobi_dll, "GRBterminate");
*(void**)(&dll_GRBupdatemodel) = dll_sym(gurobi_dll, "GRBupdatemodel");
*(void**)(&dll_GRBwrite) = dll_sym(gurobi_dll, "GRBwrite");
*(void**)(&dll_GRBwriteparams) = dll_sym(gurobi_dll, "GRBwriteparams");
#else
dll_GRBversion = GRBversion;
dll_GRBaddconstr = GRBaddconstr;
dll_GRBaddgenconstrIndicator = GRBaddgenconstrIndicator;
dll_GRBaddvars = GRBaddvars;
dll_GRBcbcut = GRBcbcut;
dll_GRBcbget = GRBcbget;
dll_GRBcblazy = GRBcblazy;
dll_GRBfreeenv = GRBfreeenv;
dll_GRBfreemodel = GRBfreemodel;
dll_GRBgetdblattr = GRBgetdblattr;
dll_GRBgetdblattrarray = GRBgetdblattrarray;
dll_GRBgetenv = GRBgetenv;
dll_GRBgeterrormsg = GRBgeterrormsg;
dll_GRBgetintattr = GRBgetintattr;
dll_GRBloadenv = GRBloadenv;
dll_GRBnewmodel = GRBnewmodel;
dll_GRBoptimize = GRBoptimize;
dll_GRBreadparams = GRBreadparams;
dll_GRBsetcallbackfunc = GRBsetcallbackfunc;
dll_GRBsetdblparam = GRBsetdblparam;
dll_GRBsetintattr = GRBsetintattr;
dll_GRBsetintattrlist = GRBsetintattrlist;
dll_GRBsetdblattrelement = GRBsetdblattrelement;
dll_GRBsetdblattrlist = GRBsetdblattrlist;
dll_GRBsetintparam = GRBsetintparam;
dll_GRBsetstrparam = GRBsetstrparam;
dll_GRBterminate = GRBterminate;
dll_GRBupdatemodel = GRBupdatemodel;
dll_GRBwrite = GRBwrite;
dll_GRBwriteparams = GRBwriteparams;
#endif
}
void MIP_gurobi_wrapper::openGUROBI() {
checkDLL();
/* Initialize the GUROBI environment */
{
// cout << "% " << flush; // Gurobi 7.5.2 prints "Academic License..."
MiniZinc::StreamRedir redirStdout(stdout, stderr);
error = dll_GRBloadenv(&env, "mzn-gurobi.log");
}
wrap_assert(!error, "Could not open GUROBI environment.");
error = dll_GRBsetintparam(env, "OutputFlag", 0); // Switch off output
// error = dll_GRBsetintparam(env, "LogToConsole",
// fVerbose ? 1 : 0); // also when flag_all_solutions? TODO
/* Create the problem. */
error = dll_GRBnewmodel(env, &model, "mzn_gurobi", 0, NULL, NULL, NULL, NULL, NULL);
wrap_assert(model != NULL, "Failed to create LP.");
}
void MIP_gurobi_wrapper::closeGUROBI() {
/* Free model */
// If not allocated, skip
if (nullptr != model) {
/* Free up the problem as allocated by GRB_createprob, if necessary */
dll_GRBfreemodel(model);
model = 0;
}
/* Free environment */
if (nullptr != env) dll_GRBfreeenv(env);
/// and at last:
// MIP_wrapper::cleanup();
#ifdef GUROBI_PLUGIN
// dll_close(gurobi_dll); // Is called too many times, disabling. 2019-05-06
#endif
}
void MIP_gurobi_wrapper::doAddVars(size_t n, double* obj, double* lb, double* ub,
MIP_wrapper::VarType* vt, string* names) {
/// Convert var types:
vector<char> ctype(n);
vector<char*> pcNames(n);
for (size_t i = 0; i < n; ++i) {
pcNames[i] = (char*)names[i].c_str();
switch (vt[i]) {
case REAL:
ctype[i] = GRB_CONTINUOUS;
break;
case INT:
ctype[i] = GRB_INTEGER;
break;
case BINARY:
ctype[i] = GRB_BINARY;
break;
default:
throw runtime_error(" MIP_wrapper: unknown variable type");
}
}
error = dll_GRBaddvars(model, static_cast<int>(n), 0, NULL, NULL, NULL, obj, lb, ub, &ctype[0],
&pcNames[0]);
wrap_assert(!error, "Failed to declare variables.");
error = dll_GRBupdatemodel(model);
wrap_assert(!error, "Failed to update model.");
}
static char getGRBSense(MIP_wrapper::LinConType s) {
switch (s) {
case MIP_wrapper::LQ:
return GRB_LESS_EQUAL;
case MIP_wrapper::EQ:
return GRB_EQUAL;
case MIP_wrapper::GQ:
return GRB_GREATER_EQUAL;
default:
throw runtime_error(" MIP_gurobi_wrapper: unknown constraint sense");
}
}
void MIP_gurobi_wrapper::addRow(int nnz, int* rmatind, double* rmatval,
MIP_wrapper::LinConType sense, double rhs, int mask,
string rowName) {
//// Make sure in order to notice the indices of lazy constr:
++nRows;
/// Convert var types:
char ssense = getGRBSense(sense);
const char* pRName = rowName.c_str();
error = dll_GRBaddconstr(model, nnz, rmatind, rmatval, ssense, rhs, pRName);
wrap_assert(!error, "Failed to add constraint.");
int nLazyAttr = 0;
const bool fUser = (MaskConsType_Usercut & mask) != 0;
const bool fLazy = (MaskConsType_Lazy & mask) != 0;
/// Gurobi 6.5.2 has lazyness 1-3.
if (fUser) {
if (fLazy)
nLazyAttr = 2; // just active lazy
else
nLazyAttr = 3; // even LP-active
} else if (fLazy)
nLazyAttr = 1; // very lazy
if (nLazyAttr) {
nLazyIdx.push_back(nRows - 1);
nLazyValue.push_back(nLazyAttr);
}
}
void MIP_gurobi_wrapper::addIndicatorConstraint(int iBVar, int bVal, int nnz, int* rmatind,
double* rmatval, MIP_wrapper::LinConType sense,
double rhs, string rowName) {
wrap_assert(0 <= bVal && 1 >= bVal, "Gurobi: addIndicatorConstraint: bVal not 0/1");
//// Make sure in order to notice the indices of lazy constr: also here? TODO
++nRows;
char ssense = getGRBSense(sense);
error = dll_GRBaddgenconstrIndicator(model, rowName.c_str(), iBVar, bVal, nnz, rmatind, rmatval,
ssense, rhs);
wrap_assert(!error, "Failed to add indicator constraint.");
}
bool MIP_gurobi_wrapper::addSearch(const std::vector<VarId>& vars, const std::vector<int> pri) {
assert(vars.size() == pri.size());
static_assert(sizeof(VarId) == sizeof(int), "VarId should be (u)int currently");
error = dll_GRBsetintattrlist(model, "BranchPriority", static_cast<int>(vars.size()),
(int*)vars.data(), (int*)pri.data());
wrap_assert(!error, "Failed to add branching priorities");
return true;
}
int MIP_gurobi_wrapper::getFreeSearch() { return options->nFreeSearch; }
bool MIP_gurobi_wrapper::addWarmStart(const std::vector<VarId>& vars,
const std::vector<double> vals) {
assert(vars.size() == vals.size());
static_assert(sizeof(VarId) == sizeof(int), "VarId should be (u)int currently");
// error = GRBsetdblattrelement(model, "Start", 0, 1.0);
error = dll_GRBsetdblattrlist(model, "Start", static_cast<int>(vars.size()), (int*)vars.data(),
(double*)vals.data());
wrap_assert(!error, "Failed to add warm start");
return true;
}
void MIP_gurobi_wrapper::setVarBounds(int iVar, double lb, double ub) {
wrap_assert(lb <= ub, "mzn-gurobi: setVarBounds: lb>ub");
error = dll_GRBsetdblattrelement(model, GRB_DBL_ATTR_LB, iVar, lb);
wrap_assert(!error, "mzn-gurobi: failed to set var lb.");
error = dll_GRBsetdblattrelement(model, GRB_DBL_ATTR_UB, iVar, ub);
wrap_assert(!error, "mzn-gurobi: failed to set var ub.");
}
void MIP_gurobi_wrapper::setVarLB(int iVar, double lb) {
error = dll_GRBsetdblattrelement(model, GRB_DBL_ATTR_LB, iVar, lb);
wrap_assert(!error, "mzn-gurobi: failed to set var lb.");
}
void MIP_gurobi_wrapper::setVarUB(int iVar, double ub) {
error = dll_GRBsetdblattrelement(model, GRB_DBL_ATTR_UB, iVar, ub);
wrap_assert(!error, "mzn-gurobi: failed to set var ub.");
}
/// SolutionCallback ------------------------------------------------------------------------
/// Gurobi ensures thread-safety
static int __stdcall solcallback(GRBmodel* model, void* cbdata, int where, void* usrdata) {
MIP_wrapper::CBUserInfo* info = (MIP_wrapper::CBUserInfo*)usrdata;
MIP_gurobi_wrapper* gw = static_cast<MIP_gurobi_wrapper*>(info->wrapper);
double nodecnt = 0.0, actnodes = 0.0, objVal = 0.0;
int solcnt = 0;
int newincumbent = 0;
if (GRB_CB_MIP == where) {
/* General MIP callback */
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIP_OBJBND, &info->pOutput->bestBound);
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIP_NODLFT, &actnodes);
info->pOutput->nOpenNodes = static_cast<int>(actnodes);
/// Check time after the 1st feas
if (-1e100 != info->nTime1Feas) {
double tNow;
gw->dll_GRBcbget(cbdata, where, GRB_CB_RUNTIME, (void*)&tNow);
if (tNow - info->nTime1Feas >= info->nTimeoutFeas) gw->dll_GRBterminate(model);
}
} else if (GRB_CB_MESSAGE == where) {
/* Message callback */
if (info->fVerb) {
char* msg;
gw->dll_GRBcbget(cbdata, where, GRB_CB_MSG_STRING, &msg);
cerr << msg << flush;
}
} else if (GRB_CB_MIPSOL == where) {
/* MIP solution callback */
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIPSOL_NODCNT, &nodecnt);
info->pOutput->nNodes = static_cast<int>(nodecnt);
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIPSOL_OBJ, &objVal);
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIPSOL_SOLCNT, &solcnt);
if (fabs(info->pOutput->objVal - objVal) > 1e-12 * (1.0 + fabs(objVal))) {
newincumbent = 1;
// Not confirmed yet, see lazy cuts
// info->pOutput->objVal = objVal;
// info->pOutput->status = MIP_wrapper::SAT;
// info->pOutput->statusName = "feasible from a callback";
}
if (newincumbent) {
assert(info->pOutput->x);
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIPSOL_SOL, (void*)info->pOutput->x);
info->pOutput->dWallTime = std::chrono::duration<double>(std::chrono::steady_clock::now() -
info->pOutput->dWallTime0)
.count();
info->pOutput->dCPUTime = double(std::clock() - info->pOutput->cCPUTime0) / CLOCKS_PER_SEC;
}
/// Callback for lazy cuts
/// Before printing
if (info->cutcbfn && info->cutMask & MIP_wrapper::MaskConsType_Lazy) {
MIP_wrapper::CutInput cutInput;
cerr << " GRB: GRB_CB_MIPSOL (" << objVal << ") -> cut callback " << endl;
info->cutcbfn(*info->pOutput, cutInput, info->psi, true);
for (auto& cd : cutInput) {
// assert( cd.mask & MIP_wrapper::MaskConsType_Lazy );
if (cd.mask & MIP_wrapper::MaskConsType_Lazy) { // take only lazy constr generators
int error =
gw->dll_GRBcblazy(cbdata, static_cast<int>(cd.rmatind.size()), cd.rmatind.data(),
cd.rmatval.data(), getGRBSense(cd.sense), cd.rhs);
if (error)
cerr << " GRB_wrapper: failed to add lazy cut. " << endl;
else
newincumbent = -1;
// info->pOutput->objVal = 1e100; // to mark that we can get a new incumbent
// which should be printed
}
}
}
if (solcnt >= 0 /*This is solution number for Gurobi*/ && newincumbent >= 0) {
if (fabs(info->pOutput->objVal - objVal) > 1e-12 * (1.0 + fabs(objVal))) {
newincumbent = 1;
info->pOutput->objVal = objVal;
info->pOutput->status = MIP_wrapper::SAT;
info->pOutput->statusName = "feasible from a callback";
}
}
if (newincumbent > 0) {
info->pOutput->dCPUTime = double(std::clock() - info->pOutput->cCPUTime0) / CLOCKS_PER_SEC;
/// Set time for the 1st feas
if (0 <= info->nTimeoutFeas && -1e100 == info->nTime1Feas)
gw->dll_GRBcbget(cbdata, where, GRB_CB_RUNTIME, (void*)&info->nTime1Feas);
/// Call the user function:
if (info->solcbfn) (*info->solcbfn)(*info->pOutput, info->psi);
if (0 == info->nTimeoutFeas) gw->dll_GRBterminate(model); // Straight after feas
}
} else if (GRB_CB_MIPNODE == where) {
int status;
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIPNODE_STATUS, &status);
if (status == GRB_OPTIMAL && info->cutcbfn) { // if cut handler given
MIP_wrapper::Output outpRlx;
outpRlx.x = info->pOutput->x; // using the sol output storage TODO?
outpRlx.nCols = info->pOutput->nCols;
assert(outpRlx.x && outpRlx.nCols);
// dll_GRBcbget(cbdata, where, GRB_CB_MIPNODE_RELOBJ, outpRlx.objVal);
gw->dll_GRBcbget(cbdata, where, GRB_CB_MIPNODE_REL, (void*)outpRlx.x);
// cerr << " GRB: GRB_CB_MIPNODE -> cut callback " << endl;
MIP_wrapper::CutInput cutInput;
info->cutcbfn(outpRlx, cutInput, info->psi, false);
// static int nCuts=0;
// nCuts += cutInput.size();
// if ( cutInput.size() )
// cerr << "\n N CUTS: " << nCuts << endl;
for (auto& cd : cutInput) {
if (!(cd.mask & (MIP_wrapper::MaskConsType_Usercut | MIP_wrapper::MaskConsType_Lazy)))
throw runtime_error("Cut callback: should be user/lazy");
if (cd.mask & MIP_wrapper::MaskConsType_Usercut) {
int error =
gw->dll_GRBcbcut(cbdata, static_cast<int>(cd.rmatind.size()), cd.rmatind.data(),
cd.rmatval.data(), getGRBSense(cd.sense), cd.rhs);
if (error) cerr << " GRB_wrapper: failed to add user cut. " << endl;
}
if (cd.mask & MIP_wrapper::MaskConsType_Lazy) {
int error =
gw->dll_GRBcblazy(cbdata, static_cast<int>(cd.rmatind.size()), cd.rmatind.data(),
cd.rmatval.data(), getGRBSense(cd.sense), cd.rhs);
if (error) cerr << " GRB_wrapper: failed to add lazy cut. " << endl;
}
}
}
}
return 0;
} /* END logcallback */
// end SolutionCallback ---------------------------------------------------------------------
MIP_gurobi_wrapper::Status MIP_gurobi_wrapper::convertStatus(int gurobiStatus) {
Status s = Status::UNKNOWN;
ostringstream oss;
/* Converting the status. */
if (gurobiStatus == GRB_OPTIMAL) {
s = Status::OPT;
oss << "Optimal";
} else if (gurobiStatus == GRB_INF_OR_UNBD) {
s = Status::UNSATorUNBND;
oss << "Infeasible or unbounded";
} else if (gurobiStatus == GRB_INFEASIBLE) {
s = Status::UNSAT;
oss << "Infeasible";
} else if (gurobiStatus == GRB_UNBOUNDED) {
oss << "Unbounded";
s = Status::UNBND;
} else {
int solcount = 0;
error = dll_GRBgetintattr(model, "SolCount", &solcount);
wrap_assert(!error, " Failure to access solution count.", false);
if (solcount) s = Status::SAT;
oss << "Gurobi stopped with status " << gurobiStatus;
}
output.statusName = gurobi_status_buffer = oss.str();
return s;
}
void MIP_gurobi_wrapper::solve() { // Move into ancestor?
if (options->flag_all_solutions && 0 == nProbType)
cerr << "WARNING. --all-solutions for SAT problems not implemented." << endl;
error = dll_GRBupdatemodel(model); // for model export
wrap_assert(!error, "Failed to update model.");
/// ADDING LAZY CONSTRAINTS IF ANY
if (nLazyIdx.size()) {
assert(nLazyIdx.size() == nLazyValue.size());
if (fVerbose)
cerr << " MIP_gurobi_wrapper: marking " << nLazyIdx.size() << " lazy cuts." << endl;
error = dll_GRBsetintattrlist(model, "Lazy", static_cast<int>(nLazyIdx.size()), nLazyIdx.data(),
nLazyValue.data());
wrap_assert(!error, "Failed to set constraint attribute.");
nLazyIdx.clear();
nLazyValue.clear();
error = dll_GRBupdatemodel(model); // for model export
wrap_assert(!error, "Failed to update model after modifying some constraint attr.");
}
/////////////// Last-minute solver options //////////////////
/* Turn on output to file */
error = dll_GRBsetstrparam(dll_GRBgetenv(model), "LogFile",
""); // FAILS to switch off in Ubuntu 15.04
/* Turn on output to the screen */
error = dll_GRBsetintparam(dll_GRBgetenv(model), "OutputFlag",
/*fVerbose ? 1 :*/ 0); // switch off, redirect in callback
// error = dll_GRBsetintparam(dll_GRBgetenv(model), "LogToConsole",
// fVerbose ? 1 : 0); // also when flag_all_solutions? TODO
wrap_assert(!error, " GUROBI Warning: Failure to switch screen indicator.", false);
// error = dll_GRB_setintparam (env, GRB_PARAM_ClockType, 1); // CPU time
// error = dll_GRB_setintparam (env, GRB_PARAM_MIP_Strategy_CallbackReducedLP, GRB__OFF); //
// Access original model
if (options->sExportModel.size()) {
error = dll_GRBwrite(model, options->sExportModel.c_str());
wrap_assert(!error, "Failed to write LP to disk.", false);
}
/// TODO
// if(all_solutions && obj.getImpl()) {
// IloNum lastObjVal = (obj.getSense() == IloObjective::Minimize ) ?
// _ilogurobi->use(SolutionCallback(_iloenv, lastObjVal, *this));
// Turn off GUROBI logging
if (options->nThreads > 0) {
error = dll_GRBsetintparam(dll_GRBgetenv(model), GRB_INT_PAR_THREADS, options->nThreads);
// int nn; // THE SETTING FAILS TO WORK IN 6.0.5.
// error = dll_getintparam(env, GRB_INT_PAR_THREADS, &nn);
// cerr << "Set " << nThreads << " threads, reported " << nn << endl;
wrap_assert(!error, "Failed to set GRB_INT_PAR_THREADS.", false);
}
if (options->nTimeout1000 > 0) {
error = dll_GRBsetdblparam(dll_GRBgetenv(model), GRB_DBL_PAR_TIMELIMIT,
static_cast<double>(options->nTimeout1000) / 1000.0);
wrap_assert(!error, "Failed to set GRB_PARAM_TimeLimit.", false);
}
if (options->nSolLimit > 0) {
error = dll_GRBsetintparam(dll_GRBgetenv(model), GRB_INT_PAR_SOLUTIONLIMIT, options->nSolLimit);
wrap_assert(!error, "Failed to set GRB_INT_PAR_SOLLIMIT.", false);
}
if (options->nWorkMemLimit > 0 && options->nWorkMemLimit < 1e200) {
error = dll_GRBsetdblparam(dll_GRBgetenv(model), "NodefileStart", options->nWorkMemLimit);
wrap_assert(!error, "Failed to set NodefileStart.", false);
}
if (options->absGap >= 0.0) {
error = dll_GRBsetdblparam(dll_GRBgetenv(model), "MIPGapAbs", options->absGap);
wrap_assert(!error, "Failed to set MIPGapAbs.", false);
}
if (options->nMIPFocus > 0) {
error = dll_GRBsetintparam(dll_GRBgetenv(model), GRB_INT_PAR_MIPFOCUS, options->nMIPFocus);
wrap_assert(!error, "Failed to set GRB_INT_PAR_MIPFOCUS.", false);
}
if (options->relGap >= 0.0) {
error = dll_GRBsetdblparam(dll_GRBgetenv(model), "MIPGap", options->relGap);
wrap_assert(!error, "Failed to set MIPGap.", false);
}
if (options->intTol >= 0.0) {
error = dll_GRBsetdblparam(dll_GRBgetenv(model), "IntFeasTol", options->intTol);
wrap_assert(!error, "Failed to set IntFeasTol.", false);
}
if (options->feasTol >= 0.0) {
error = dll_GRBsetdblparam(dll_GRBgetenv(model), "FeasibilityTol", options->feasTol);
wrap_assert(!error, "Failed to set FeasTol.", false);
}
/// Solution callback
output.nCols = static_cast<int>(colObj.size());
x.resize(output.nCols);
output.x = &x[0];
SolCallbackFn solcbfn = cbui.solcbfn;
if (true) { // Need for logging
cbui.fVerb = fVerbose;
cbui.nTimeoutFeas = options->nTimeoutFeas1000 / 1000.0;
if (!options->flag_all_solutions) cbui.solcbfn = 0;
if (cbui.cutcbfn) {
assert(cbui.cutMask & (MaskConsType_Usercut | MaskConsType_Lazy));
if (cbui.cutMask & MaskConsType_Usercut) {
// For user cuts, needs to keep some info after presolve
if (fVerbose)
cerr << " MIP_gurobi_wrapper: user cut callback enabled, setting PreCrush=1" << endl;
error = dll_GRBsetintparam(dll_GRBgetenv(model), GRB_INT_PAR_PRECRUSH, 1);
wrap_assert(!error, "Failed to set GRB_INT_PAR_PRECRUSH.", false);
}
if (cbui.cutMask & MaskConsType_Lazy) {
// For lazy cuts, Gurobi disables some presolves
if (fVerbose)
cerr << " MIP_gurobi_wrapper: lazy cut callback enabled, setting LazyConstraints=1"
<< endl;
error = dll_GRBsetintparam(dll_GRBgetenv(model), GRB_INT_PAR_LAZYCONSTRAINTS, 1);
wrap_assert(!error, "Failed to set GRB_INT_PAR_LAZYCONSTRAINTS.", false);
}
}
error = dll_GRBsetcallbackfunc(model, solcallback, (void*)&cbui);
wrap_assert(!error, "Failed to set callback", false);
}
/// after all modifs
if (options->sReadParams.size()) {
error = dll_GRBreadparams(dll_GRBgetenv(model), options->sReadParams.c_str());
wrap_assert(!error, "Failed to read GUROBI parameters.", false);
}
if (options->sWriteParams.size()) {
error = dll_GRBwriteparams(dll_GRBgetenv(model), options->sWriteParams.c_str());
wrap_assert(!error, "Failed to write GUROBI parameters.", false);
}
cbui.pOutput->dWallTime0 = output.dWallTime0 = std::chrono::steady_clock::now();
output.dCPUTime = cbui.pOutput->cCPUTime0 = std::clock();
/* Optimize the problem and obtain solution. */
error = dll_GRBoptimize(model);
wrap_assert(!error, "Failed to optimize MIP.");
output.dWallTime =
std::chrono::duration<double>(std::chrono::steady_clock::now() - output.dWallTime0).count();
output.dCPUTime = (std::clock() - output.dCPUTime) / CLOCKS_PER_SEC;
int solstat;
error = dll_GRBgetintattr(model, GRB_INT_ATTR_STATUS, &solstat);
wrap_assert(!error, "Failed to get MIP status.", false);
output.status = convertStatus(solstat);
/// Continuing to fill the output object:
if (Status::OPT == output.status || Status::SAT == output.status) {
error = dll_GRBgetdblattr(model, GRB_DBL_ATTR_OBJVAL, &output.objVal);
wrap_assert(!error, "No MIP objective value available.");
// int cur_numrows = dll_GRB_getnumrows (env, lp);
int cur_numcols = getNCols();
assert(cur_numcols == colObj.size());
x.resize(cur_numcols);
output.x = &x[0];
error = dll_GRBgetdblattrarray(model, GRB_DBL_ATTR_X, 0, cur_numcols, (double*)output.x);
wrap_assert(!error, "Failed to get variable values.");
if (!options->flag_all_solutions && solcbfn) {
solcbfn(output, cbui.psi);
}
}
output.bestBound = 1e308;
error = dll_GRBgetdblattr(model, GRB_DBL_ATTR_OBJBOUNDC, &output.bestBound);
wrap_assert(!error, "Failed to get the best bound.", false);
double nNodes = -1;
error = dll_GRBgetdblattr(model, GRB_DBL_ATTR_NODECOUNT, &nNodes);
output.nNodes = static_cast<int>(nNodes);
output.nOpenNodes = 0;
}
void MIP_gurobi_wrapper::setObjSense(int s) {
error = dll_GRBsetintattr(model, GRB_INT_ATTR_MODELSENSE, s > 0 ? GRB_MAXIMIZE : GRB_MINIMIZE);
wrap_assert(!error, "Failed to set obj sense.");
}
View Git Blame Copy Permalink