131 lines
2.3 KiB
C++
131 lines
2.3 KiB
C++
#include <cassert>
|
|
|
|
#include <iostream>
|
|
#include <sstream>
|
|
|
|
#include "LiftedUtils.h"
|
|
#include "ConstraintTree.h"
|
|
|
|
|
|
namespace LiftedUtils {
|
|
|
|
|
|
unordered_map<string, unsigned> symbolDict;
|
|
|
|
|
|
Symbol
|
|
getSymbol (const string& symbolName)
|
|
{
|
|
unordered_map<string, unsigned>::iterator it
|
|
= symbolDict.find (symbolName);
|
|
if (it != symbolDict.end()) {
|
|
return it->second;
|
|
} else {
|
|
symbolDict[symbolName] = symbolDict.size() - 1;
|
|
return symbolDict.size() - 1;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
void
|
|
printSymbolDictionary (void)
|
|
{
|
|
unordered_map<string, unsigned>::const_iterator it
|
|
= symbolDict.begin();
|
|
while (it != symbolDict.end()) {
|
|
cout << it->first << " -> " << it->second << endl;
|
|
++ it;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ostream& operator<< (ostream &os, const Symbol& s)
|
|
{
|
|
unordered_map<string, unsigned>::const_iterator it
|
|
= LiftedUtils::symbolDict.begin();
|
|
while (it != LiftedUtils::symbolDict.end() && it->second != s) {
|
|
++ it;
|
|
}
|
|
assert (it != LiftedUtils::symbolDict.end());
|
|
os << it->first;
|
|
return os;
|
|
}
|
|
|
|
|
|
|
|
ostream& operator<< (ostream &os, const LogVar& X)
|
|
{
|
|
const string labels[] = {
|
|
"A", "B", "C", "D", "E", "F",
|
|
"G", "H", "I", "J", "K", "M" };
|
|
(X >= 12) ? os << "X_" << X.id_ : os << labels[X];
|
|
return os;
|
|
}
|
|
|
|
|
|
|
|
ostream& operator<< (ostream &os, const Tuple& t)
|
|
{
|
|
os << "(" ;
|
|
for (size_t i = 0; i < t.size(); i++) {
|
|
os << ((i != 0) ? "," : "") << t[i];
|
|
}
|
|
os << ")" ;
|
|
return os;
|
|
}
|
|
|
|
|
|
|
|
ostream& operator<< (ostream &os, const Ground& gr)
|
|
{
|
|
os << gr.functor();
|
|
os << "(" ;
|
|
for (size_t i = 0; i < gr.args().size(); i++) {
|
|
if (i != 0) os << ", " ;
|
|
os << gr.args()[i];
|
|
}
|
|
os << ")" ;
|
|
return os;
|
|
}
|
|
|
|
|
|
|
|
LogVars
|
|
Substitution::getDiscardedLogVars (void) const
|
|
{
|
|
LogVars discardedLvs;
|
|
set<LogVar> doneLvs;
|
|
unordered_map<LogVar, LogVar>::const_iterator it;
|
|
it = subs_.begin();
|
|
while (it != subs_.end()) {
|
|
if (Util::contains (doneLvs, it->second)) {
|
|
discardedLvs.push_back (it->first);
|
|
} else {
|
|
doneLvs.insert (it->second);
|
|
}
|
|
++ it;
|
|
}
|
|
return discardedLvs;
|
|
}
|
|
|
|
|
|
|
|
ostream& operator<< (ostream &os, const Substitution& theta)
|
|
{
|
|
unordered_map<LogVar, LogVar>::const_iterator it;
|
|
os << "[" ;
|
|
it = theta.subs_.begin();
|
|
while (it != theta.subs_.end()) {
|
|
if (it != theta.subs_.begin()) os << ", " ;
|
|
os << it->first << "->" << it->second ;
|
|
++ it;
|
|
}
|
|
os << "]" ;
|
|
return os;
|
|
}
|
|
|