int main(int argc, char *argv[])
{
// Parse command line arguments.
boost::filesystem::path input("."), output(".");
if (argc >= 2) input = argv[1];
if (argc >= 3) output = argv[2];
// Create the output directory if required.
if (!boost::filesystem::exists(output)) {
cout << "Creating output directory " << output << endl;
boost::filesystem::create_directories(output);
}
// Build the stage list.
typedef vector<string> Stages;
Stages stages;
stages.push_back("BuildGraph");
stages.push_back("BindIdentifiers");
stages.push_back("BindNamedTypes");
stages.push_back("EvalTypeExprs");
stages.push_back("InitRootTypes");
stages.push_back("GenImplicitAccessors");
stages.push_back("ConfigureCalls");
stages.push_back("FindCallCandidates");
stages.push_back("CalcPossibleTypes");
stages.push_back("CalcRequiredTypes");
stages.push_back("CalcActualTypes");
stages.push_back("NarrowCallCandidates");
stages.push_back("SelectCallCandidate");
// Generate the stages header file.
boost::filesystem::path hpath = output;
hpath /= "stages.hpp";
ofstream h(hpath.c_str());
h << "/* Automatically generated by ast-gen. DO NOT MODIFY. */\n";
h << "#pragma once\n";
h << "#include \"Stage.hpp\"\n\n";
h << "namespace stage {\n\n";
h << "using ast::NodePtr;\n";
h << "using ast::NodeVector;\n";
h << "using ast::Repository;\n\n";
int id = 1;
for (Stages::iterator it = stages.begin(); it != stages.end(); it++, id++) {
h << "class " << *it << " : public Stage\n";
h << "{\n";
h << "protected:\n";
h << "\tvirtual void process(const NodePtr& node);\n";
h << "public:\n";
h << "\tstring getName() const { return \"" << *it << "\"; }\n";
h << "\tint getId() const { return " << id << "; }\n";
h << "\t" << *it << "(Repository& r) : Stage(r) {}\n";
// Custom header fields.
boost::filesystem::path cpath = input;
cpath /= *it + ".in.hpp";
ifstream fin(cpath.c_str());
if (fin.good()) {
h << "\n";
h << string((std::istreambuf_iterator<char>(fin)), std::istreambuf_iterator<char>());
h << "\n";
}
h << "};\n\n";
}
h << "} // namespace stage\n";
// Generate the stage manager header file.
boost::filesystem::path mpath = output;
mpath /= "StageManager.hpp";
ofstream mh(mpath.c_str());
mh << "/* Automatically generated by ast-gen. DO NOT MODIFY. */\n";
mh << "#pragma once\n";
mh << "#include \"stages.hpp\"\n";
mh << "#include <string>\n";
mh << "#include <map>\n";
mh << "#include <vector>\n";
mh << "\n";
mh << "namespace stage {\n\n";
mh << "using std::string;\nusing std::map;\nusing std::vector;\n\n";
mh << "class StageManager\n{\npublic:\n";
mh << "\tRepository& repository;\n\n";
// Generate the constructor.
mh << "\tStageManager(Repository& r) : repository(r)";
for (Stages::iterator it = stages.begin(); it != stages.end(); it++) {
mh << ",\n\t\tstage" << *it << "(r)";
}
mh << "\n";
mh << "\t{\n";
for (Stages::iterator it = stages.begin(); it != stages.end(); it++) {
mh << "\t\tstageIndices[\"" << *it << "\"] = stages.size();\n";
mh << "\t\tstagesByName[\"" << *it << "\"] = &stage" << *it << ";\n";
mh << "\t\tstages.push_back(&stage" << *it << ");\n";
}
mh << "\t}\n\n";
// Stages vector.
mh << "\ttypedef vector<Stage*> Stages;\n";
mh << "\ttypedef map<string, int> StageIndices;\n";
mh << "\ttypedef map<string, Stage*> StagesByName;\n";
mh << "\tStages stages;\n";
mh << "\tStageIndices stageIndices;\n";
mh << "\tStagesByName stagesByName;\n\n";
// Stage ids.
mh << "\tstatic int getIdOfStage(const string& name)\n\t{\n";
id = 1;
for (Stages::iterator it = stages.begin(); it != stages.end(); it++, id++) {
mh << "\t\tif (name == \"" << *it << "\") return " << id << ";\n";
}
mh << "\t\tthrow std::runtime_error(\"Stage \" + name + \" does not exist.\");\n";
mh << "\t}\n";
mh << "protected:\n";
for (Stages::iterator it = stages.begin(); it != stages.end(); it++) {
mh << "\t" << *it << " stage" << *it << ";\n";
}
mh << "};\n\n";
mh << "} // namespace stage\n";
return 0;
}
void stages() {
os<<"Stage Statistics:\n";
// Find all stages
int stage=0;
size_t maxWidth = 0;
typedef std::vector<string> Stages;
Stages stages;
vl_unordered_map<string,int> stageInt;
typedef std::multimap<string,const V3Statistic*> ByName;
ByName byName;
// * is always first
for (StatColl::iterator it = s_allStats.begin(); it!=s_allStats.end(); ++it) {
const V3Statistic* repp = &(*it);
if (repp->stage() != "*" && repp->printit()) {
if (maxWidth < repp->name().length()) maxWidth = repp->name().length();
if (stageInt.find(repp->stage()) == stageInt.end()) {
stageInt.insert(make_pair(repp->stage(), stage++));
stages.push_back(repp->stage());
}
byName.insert(make_pair(repp->name(), repp));
}
}
// Header
os<<" Stat "<<std::left<<std::setw(maxWidth-5-2)<<"";
for (Stages::iterator it = stages.begin(); it!=stages.end(); ++it) {
os<<" "<<std::left<<std::setw(9)<<*it;
}
os<<endl;
os<<" -------- "<<std::left<<std::setw(maxWidth-5-2)<<"";
for (Stages::iterator it = stages.begin(); it!=stages.end(); ++it) {
os<<" "<<std::left<<std::setw(9)<<"-------";
}
//os<<endl;
// Print organized by stage
string lastName = "__NONE__";
string lastCommaName = "__NONE__";
unsigned col=0;
for (ByName::iterator it = byName.begin(); it!=byName.end(); ++it) {
const V3Statistic* repp = it->second;
if (lastName != repp->name()) {
lastName = repp->name();
{
string commaName = lastName;
string::size_type pos;
if ((pos=commaName.find(",")) != string::npos) {
commaName.erase(pos);
}
if (lastCommaName != commaName) {
lastCommaName = commaName;
os<<endl;
}
}
os<<endl;
col = 0;
os<<" "<<std::left<<std::setw(maxWidth)<<repp->name();
}
while (col<stages.size() && stages.at(col) != repp->stage()) {
os<<std::setw(11)<<"";
col++;
}
repp->dump(os);
col++;
}
os<<endl;
}
