* Factored out dot graph generation into a separate file.

This commit is contained in:
Eelco Dolstra 2003-09-03 11:20:18 +00:00
parent 0d2bc68681
commit c0bbed0959
4 changed files with 76 additions and 61 deletions

View file

@ -4,7 +4,7 @@ check_PROGRAMS = test
AM_CXXFLAGS = -DSYSTEM=\"@host@\" -Wall -I.. -I../externals/inst/include $(CXXFLAGS) AM_CXXFLAGS = -DSYSTEM=\"@host@\" -Wall -I.. -I../externals/inst/include $(CXXFLAGS)
AM_LDFLAGS = -L../externals/inst/lib -ldb_cxx -lATerm $(LDFLAGS) AM_LDFLAGS = -L../externals/inst/lib -ldb_cxx -lATerm $(LDFLAGS)
nix_SOURCES = nix.cc nix_SOURCES = nix.cc dotgraph.cc
nix_LDADD = libshared.a libnix.a -ldb_cxx -lATerm nix_LDADD = libshared.a libnix.a -ldb_cxx -lATerm
nix_hash_SOURCES = nix-hash.cc nix_hash_SOURCES = nix-hash.cc

63
src/dotgraph.cc Normal file
View file

@ -0,0 +1,63 @@
#include "dotgraph.hh"
static string dotQuote(const string & s)
{
return "\"" + s + "\"";
}
void printDotGraph(const FSIds & roots)
{
FSIds workList(roots.begin(), roots.end());
FSIdSet doneSet;
cout << "digraph G {\n";
while (!workList.empty()) {
FSId id = workList.front();
workList.pop_front();
if (doneSet.find(id) == doneSet.end()) {
doneSet.insert(id);
FState fs = parseFState(termFromId(id));
string label, shape;
if (fs.type == FState::fsDerive) {
for (FSIdSet::iterator i = fs.derive.inputs.begin();
i != fs.derive.inputs.end(); i++)
{
workList.push_back(*i);
cout << dotQuote(*i) << " -> "
<< dotQuote(id) << ";\n";
}
label = "derive";
shape = "box";
for (StringPairs::iterator i = fs.derive.env.begin();
i != fs.derive.env.end(); i++)
if (i->first == "name") label = i->second;
}
else if (fs.type == FState::fsSlice) {
label = baseNameOf((*fs.slice.elems.begin()).first);
shape = "ellipse";
if (isHash(string(label, 0, Hash::hashSize * 2)) &&
label[Hash::hashSize * 2] == '-')
label = string(label, Hash::hashSize * 2 + 1);
}
else abort();
cout << dotQuote(id) << "[label = "
<< dotQuote(label)
<< ", shape = " << shape
<< "];\n";
}
}
cout << "}\n";
}

8
src/dotgraph.hh Normal file
View file

@ -0,0 +1,8 @@
#ifndef __DOTGRAPH_H
#define __DOTGRAPH_H
#include "fstate.hh"
void printDotGraph(const FSIds & roots);
#endif /* !__DOTGRAPH_H */

View file

@ -5,6 +5,7 @@
#include "normalise.hh" #include "normalise.hh"
#include "archive.hh" #include "archive.hh"
#include "shared.hh" #include "shared.hh"
#include "dotgraph.hh"
typedef void (* Operation) (Strings opFlags, Strings opArgs); typedef void (* Operation) (Strings opFlags, Strings opArgs);
@ -80,12 +81,6 @@ static void opAdd(Strings opFlags, Strings opArgs)
} }
static string dotQuote(const string & s)
{
return "\"" + s + "\"";
}
FSId maybeNormalise(const FSId & id, bool normalise) FSId maybeNormalise(const FSId & id, bool normalise)
{ {
return normalise ? normaliseFState(id) : id; return normalise ? normaliseFState(id) : id;
@ -170,62 +165,11 @@ static void opQuery(Strings opFlags, Strings opArgs)
} }
case qGraph: { case qGraph: {
FSIds roots;
FSIds workList;
for (Strings::iterator i = opArgs.begin(); for (Strings::iterator i = opArgs.begin();
i != opArgs.end(); i++) i != opArgs.end(); i++)
workList.push_back(argToId(*i)); roots.push_back(argToId(*i));
printDotGraph(roots);
FSIdSet doneSet;
cout << "digraph G {\n";
while (!workList.empty()) {
FSId id = workList.front();
workList.pop_front();
if (doneSet.find(id) == doneSet.end()) {
doneSet.insert(id);
FState fs = parseFState(termFromId(id));
string label, shape;
if (fs.type == FState::fsDerive) {
for (FSIdSet::iterator i = fs.derive.inputs.begin();
i != fs.derive.inputs.end(); i++)
{
workList.push_back(*i);
cout << dotQuote(*i) << " -> "
<< dotQuote(id) << ";\n";
}
label = "derive";
shape = "box";
for (StringPairs::iterator i = fs.derive.env.begin();
i != fs.derive.env.end(); i++)
if (i->first == "name") label = i->second;
}
else if (fs.type == FState::fsSlice) {
label = baseNameOf((*fs.slice.elems.begin()).first);
shape = "ellipse";
if (isHash(string(label, 0, Hash::hashSize * 2)) &&
label[Hash::hashSize * 2] == '-')
label = string(label, Hash::hashSize * 2 + 1);
}
else abort();
cout << dotQuote(id) << "[label = "
<< dotQuote(label)
<< ", shape = " << shape
<< "];\n";
}
}
cout << "}\n";
break; break;
} }