Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
5.3 kB · 177 lines
C++
at develop
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178/* -*- 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/flatten_internal.hh>#include <minizinc/pathfileprinter.hh>
#include <sstream>
namespace MiniZinc {
using std::string;using std::vector;
PathFilePrinter::PathFilePrinter(std::ostream& o, EnvI&) : os(o), constraint_index(0) {}
void PathFilePrinter::addBetterName(Id* id, string name, string path, bool overwrite = false) { string oname; string opath;
NameMap::iterator it = betternames.find(id); if (it != betternames.end()) { oname = it->second.first; opath = it->second.second; }
if (!name.empty() && (overwrite || oname.empty())) oname = name; if (!path.empty() && (overwrite || opath.empty())) opath = path;
betternames[id] = NamePair(oname, opath);}
string path2name(string path) { std::stringstream name;
size_t idpos = path.rfind("id:"); if (idpos != string::npos) { idpos += 3; size_t semi = path.find(";", idpos); if (semi != string::npos) { // Variable name name << path.substr(idpos, semi - idpos);
// Check for array int dim = 0; size_t ilpos = semi - idpos; do { ilpos = path.find("il:", ilpos); if (ilpos != string::npos) { ilpos += 3; semi = path.find(";", ilpos); if (semi != string::npos) { if (dim == 0) name << "["; else name << ","; name << path.substr(ilpos, semi - ilpos); dim++; } } } while (ilpos != string::npos);
if (dim > 0) name << "?]";
// Check for anon if (path.find(":anon") != string::npos || path.find("=") != string::npos) { name.str(""); name.clear(); } } }
return name.str();}
void PathFilePrinter::print(Model* m) { // Build map for (VarDeclIterator vdit = m->begin_vardecls(); vdit != m->end_vardecls(); ++vdit) { VarDecl* e = vdit->e(); for (ExpressionSetIter it = e->ann().begin(); it != e->ann().end(); ++it) { if (Call* ca = (*it)->dyn_cast<Call>()) { ASTString cid = ca->id(); if (cid == constants().ann.output_array) { if (ArrayLit* rhs = e->e()->dyn_cast<ArrayLit>()) { for (unsigned int ind = 0; ind < rhs->size(); ind++) { if (Id* id = (*rhs)[ind]->dyn_cast<Id>()) { std::stringstream bettername; bettername << *e->id() << "[";
// Array of sets ArrayLit& dimsets = *ca->arg(0)->cast<ArrayLit>(); vector<IntVal> dims(dimsets.size(), 1); for (unsigned int i = 0; i < dimsets.size(); i++) { SetLit* sl = dimsets[i]->cast<SetLit>(); dims[i] = sl->isv()->card(); } vector<IntVal> dimspan(dims.size(), 1); for (unsigned int i = 0; i < dims.size(); i++) for (unsigned int j = i + 1; j < dims.size(); j++) dimspan[i] *= dims[j];
IntVal curind = ind; for (unsigned int i = 0; i < dims.size() - 1; i++) { IntVal thisind = curind / dimspan[i]; curind -= thisind * dimspan[i]; bettername << dimsets[i]->cast<SetLit>()->isv()->min() + thisind << ","; } bettername << dimsets[dimsets.size() - 1]->cast<SetLit>()->isv()->min() + curind << "]";
addBetterName(id, bettername.str(), "", true); } } } } else if (ca->id() == constants().ann.mzn_path) { StringLit* sl = ca->arg(0)->cast<StringLit>(); addBetterName(e->id(), path2name(sl->v().str()), sl->v().str()); } } } }
// Print values for (Item* item : *m) print(item);}
void PathFilePrinter::print(Item* item) { if (VarDeclI* vdi = item->dyn_cast<VarDeclI>()) { Id* id = vdi->e()->id(); NamePair np = betternames[id]; if (!np.first.empty() || !np.second.empty()) { // FlatZinc name os << *id << "\t";
// Nice name if (np.first.empty()) { os << *id << "\t"; } else { string name = np.first; os << name; if (name.find("?") != string::npos) os << "(" << *id << ")"; os << "\t"; }
// Path os << np.second << std::endl; } } else if (ConstraintI* ci = item->dyn_cast<ConstraintI>()) { StringLit* sl = nullptr; Expression* e = ci->e(); for (ExpressionSetIter it = e->ann().begin(); it != e->ann().end(); ++it) { if (Call* ca = (*it)->dyn_cast<Call>()) { ASTString cid = ca->id(); if (cid == constants().ann.mzn_path) { sl = ca->arg(0)->cast<StringLit>(); } } }
os << constraint_index << "\t"; os << constraint_index << "\t"; if (sl) { os << sl->v(); } else { os << ""; } os << std::endl; constraint_index++; }}} // namespace MiniZinc