Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
6.6 kB · 184 lines
C++
at develop
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185/* -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */
/* * Main authors: * Guido Tack <guido.tack@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/. */
#include <minizinc/eval_par.hh>#include <minizinc/solver_instance_base.hh>
#ifdef _MSC_VER#define _CRT_SECURE_NO_WARNINGS#undef ERROR // MICROsoft.#undef min#undef max#endif
namespace MiniZinc {
SolverInstanceBase::Status SolverInstanceBase::solve(void) { return SolverInstance__ERROR; }
void SolverInstanceBase::reset(void) { assert(false); }
void SolverInstanceBase::resetWithConstraints(Model::iterator begin, Model::iterator end) { assert(false);}
void SolverInstanceBase::processPermanentConstraints(Model::iterator begin, Model::iterator end) { assert(false);}
void Registry::add(const std::string& name, poster p) { _registry.insert(std::make_pair(name, p)); }void Registry::post(std::string name, Constraint* c) { std::unordered_map<std::string, poster>::iterator it = _registry.find(name); if (it == _registry.end()) { GCLock lock; throw InternalError("Error: solver backend cannot handle constraint: " + name + "\n"); } it->second(_base, c);}
void SolverInstanceBase::printSolution() { std::ostringstream oss;
if (_options->printStatistics) printStatistics(1); // Insert stats before sol separator if (0 == pS2Out) { getEnv()->evalOutput(std::cout); // deprecated std::cout << oss.str(); if (oss.str().size() && '\n' != oss.str().back()) std::cout << '\n'; std::cout << "----------" << std::endl; } else getSolns2Out()->evalOutput(oss.str());}
void SolverInstanceBase2::printSolution() { GCLock lock; assignSolutionToOutput(); SolverInstanceBase::printSolution();}
// void// SolverInstanceBase::assignSolutionToOutput(void) {// for (VarDeclIterator it = getEnv()->output()->begin_vardecls(); it !=// getEnv()->output()->end_vardecls(); ++it) {// if (it->e()->e() == NULL) {// it->e()->e(getSolutionValue(it->e()->id()));// }// }// }
void SolverInstanceBase2::assignSolutionToOutput() { GCLock lock;
MZN_ASSERT_HARD_MSG( 0 != pS2Out, "Setup a Solns2Out object to use default solution extraction/reporting procs");
if (_varsWithOutput.empty()) { for (VarDeclIterator it = getEnv()->flat()->begin_vardecls(); it != getEnv()->flat()->end_vardecls(); ++it) { if (!it->removed()) { VarDecl* vd = it->e(); if (!vd->ann().isEmpty()) { if (vd->ann().containsCall(constants().ann.output_array.aststr()) || vd->ann().contains(constants().ann.output_var)) { _varsWithOutput.push_back(vd); } } } } }
pS2Out->declNewOutput(); // Even for empty output decl
// iterate over set of ids that have an output annotation && obtain their right hand side from the // flat model for (unsigned int i = 0; i < _varsWithOutput.size(); i++) { VarDecl* vd = _varsWithOutput[i]; // std::cout << "DEBUG: Looking at var-decl with output-annotation: " << *vd << std::endl; if (Call* output_array_ann = Expression::dyn_cast<Call>( getAnnotation(vd->ann(), constants().ann.output_array.aststr()))) { assert(vd->e());
if (ArrayLit* al = vd->e()->dyn_cast<ArrayLit>()) { std::vector<Expression*> array_elems; ArrayLit& array = *al; for (unsigned int j = 0; j < array.size(); j++) { if (Id* id = array[j]->dyn_cast<Id>()) { // std::cout << "DEBUG: getting solution value from " << *id << " : " << id->v() << // std::endl; array_elems.push_back(getSolutionValue(id)); } else if (FloatLit* floatLit = array[j]->dyn_cast<FloatLit>()) { array_elems.push_back(floatLit); } else if (IntLit* intLit = array[j]->dyn_cast<IntLit>()) { array_elems.push_back(intLit); } else if (BoolLit* boolLit = array[j]->dyn_cast<BoolLit>()) { array_elems.push_back(boolLit); } else if (SetLit* setLit = array[j]->dyn_cast<SetLit>()) { array_elems.push_back(setLit); } else if (StringLit* strLit = array[j]->dyn_cast<StringLit>()) { array_elems.push_back(strLit); } else { std::ostringstream oss; oss << "Error: array element " << *array[j] << " is not an id nor a literal"; throw InternalError(oss.str()); } } GCLock lock; ArrayLit* dims; Expression* e = output_array_ann->arg(0); if (ArrayLit* al = e->dyn_cast<ArrayLit>()) { dims = al; } else if (Id* id = e->dyn_cast<Id>()) { dims = id->decl()->e()->cast<ArrayLit>(); } else { throw -1; } std::vector<std::pair<int, int> > dims_v; for (int i = 0; i < dims->length(); i++) { IntSetVal* isv = eval_intset(getEnv()->envi(), (*dims)[i]); if (isv->size() == 0) { dims_v.push_back(std::pair<int, int>(1, 0)); } else { dims_v.push_back(std::pair<int, int>(static_cast<int>(isv->min().toInt()), static_cast<int>(isv->max().toInt()))); } } ArrayLit* array_solution = new ArrayLit(Location(), array_elems, dims_v); KeepAlive ka(array_solution); auto& de = getSolns2Out()->findOutputVar(vd->id()->str().str()); de.first->e(array_solution); } } else if (vd->ann().contains(constants().ann.output_var)) { Expression* sol = getSolutionValue(vd->id()); vd->e(sol); auto& de = getSolns2Out()->findOutputVar(vd->id()->str().str()); de.first->e(sol); } }}
void SolverInstanceBase::flattenSearchAnnotations(const Annotation& ann, std::vector<Expression*>& out) { for (ExpressionSetIter i = ann.begin(); i != ann.end(); ++i) { Expression* e = *i; if (e->isa<Call>() && (e->cast<Call>()->id().str() == "seq_search" || e->cast<Call>()->id().str() == "warm_start_array")) { Call* c = e->cast<Call>(); auto* anns = c->arg(0)->cast<ArrayLit>(); for (unsigned int i = 0; i < anns->size(); i++) { Annotation subann; subann.add((*anns)[i]); flattenSearchAnnotations(subann, out); } } else { out.push_back(*i); } }}
} // namespace MiniZinc