This commit is contained in:
Eelco Dolstra 2016-01-28 16:01:01 +01:00
parent 9f14d7d33a
commit 26566cd28e

View file

@ -20,8 +20,8 @@ static string nextColour()
{ {
static int n = 0; static int n = 0;
static string colours[] = static string colours[] =
{ "black", "red", "green", "blue" { "black", "red", "green", "blue"
, "magenta", "burlywood" }; , "magenta", "burlywood" };
return colours[n++ % (sizeof(colours) / sizeof(string))]; return colours[n++ % (sizeof(colours) / sizeof(string))];
} }
@ -29,7 +29,7 @@ static string nextColour()
static string makeEdge(const string & src, const string & dst) static string makeEdge(const string & src, const string & dst)
{ {
format f = format("%1% -> %2% [color = %3%];\n") format f = format("%1% -> %2% [color = %3%];\n")
% dotQuote(src) % dotQuote(dst) % dotQuote(nextColour()); % dotQuote(src) % dotQuote(dst) % dotQuote(nextColour());
return f.str(); return f.str();
} }
@ -38,8 +38,8 @@ static string makeNode(const string & id, const string & label,
const string & colour) const string & colour)
{ {
format f = format("%1% [label = %2%, shape = box, " format f = format("%1% [label = %2%, shape = box, "
"style = filled, fillcolor = %3%];\n") "style = filled, fillcolor = %3%];\n")
% dotQuote(id) % dotQuote(label) % dotQuote(colour); % dotQuote(id) % dotQuote(label) % dotQuote(colour);
return f.str(); return f.str();
} }
@ -65,30 +65,30 @@ void printClosure(const Path & nePath, const StoreExpr & fs)
PathSet doneSet; PathSet doneSet;
for (PathSet::iterator i = workList.begin(); i != workList.end(); ++i) { for (PathSet::iterator i = workList.begin(); i != workList.end(); ++i) {
cout << makeEdge(pathLabel(nePath, *i), nePath); cout << makeEdge(pathLabel(nePath, *i), nePath);
} }
while (!workList.empty()) { while (!workList.empty()) {
Path path = *(workList.begin()); Path path = *(workList.begin());
workList.erase(path); workList.erase(path);
if (doneSet.find(path) == doneSet.end()) { if (doneSet.find(path) == doneSet.end()) {
doneSet.insert(path); doneSet.insert(path);
ClosureElems::const_iterator elem = fs.closure.elems.find(path); ClosureElems::const_iterator elem = fs.closure.elems.find(path);
if (elem == fs.closure.elems.end()) if (elem == fs.closure.elems.end())
throw Error(format("bad closure, missing path %1%") % path); throw Error(format("bad closure, missing path %1%") % path);
for (StringSet::const_iterator i = elem->second.refs.begin(); for (StringSet::const_iterator i = elem->second.refs.begin();
i != elem->second.refs.end(); ++i) i != elem->second.refs.end(); ++i)
{ {
workList.insert(*i); workList.insert(*i);
cout << makeEdge(pathLabel(nePath, *i), pathLabel(nePath, path)); cout << makeEdge(pathLabel(nePath, *i), pathLabel(nePath, path));
} }
cout << makeNode(pathLabel(nePath, path), cout << makeNode(pathLabel(nePath, path),
symbolicName(path), "#ff0000"); symbolicName(path), "#ff0000");
} }
} }
} }
#endif #endif
@ -102,10 +102,10 @@ void printDotGraph(const PathSet & roots)
cout << "digraph G {\n"; cout << "digraph G {\n";
while (!workList.empty()) { while (!workList.empty()) {
Path path = *(workList.begin()); Path path = *(workList.begin());
workList.erase(path); workList.erase(path);
if (doneSet.find(path) != doneSet.end()) continue; if (doneSet.find(path) != doneSet.end()) continue;
doneSet.insert(path); doneSet.insert(path);
cout << makeNode(path, symbolicName(path), "#ff0000"); cout << makeNode(path, symbolicName(path), "#ff0000");
@ -124,34 +124,34 @@ void printDotGraph(const PathSet & roots)
#if 0 #if 0
StoreExpr ne = storeExprFromPath(path); StoreExpr ne = storeExprFromPath(path);
string label, colour; string label, colour;
if (ne.type == StoreExpr::neDerivation) { if (ne.type == StoreExpr::neDerivation) {
for (PathSet::iterator i = ne.derivation.inputs.begin(); for (PathSet::iterator i = ne.derivation.inputs.begin();
i != ne.derivation.inputs.end(); ++i) i != ne.derivation.inputs.end(); ++i)
{ {
workList.insert(*i); workList.insert(*i);
cout << makeEdge(*i, path); cout << makeEdge(*i, path);
} }
label = "derivation"; label = "derivation";
colour = "#00ff00"; colour = "#00ff00";
for (StringPairs::iterator i = ne.derivation.env.begin(); for (StringPairs::iterator i = ne.derivation.env.begin();
i != ne.derivation.env.end(); ++i) i != ne.derivation.env.end(); ++i)
if (i->first == "name") label = i->second; if (i->first == "name") label = i->second;
} }
else if (ne.type == StoreExpr::neClosure) { else if (ne.type == StoreExpr::neClosure) {
label = "<closure>"; label = "<closure>";
colour = "#00ffff"; colour = "#00ffff";
printClosure(path, ne); printClosure(path, ne);
} }
else abort(); else abort();
cout << makeNode(path, label, colour); cout << makeNode(path, label, colour);
#endif #endif
} }