src/ast/generator-default-visitor.cc

   1 #include <iostream>
   2 #include <fstream>
   3
   4 #include <core/assert.hh>
   5 #include "generator.hh"
   6
   7 namespace ag = ast::generator;
   8
   9 static void
  10 generateDefaultVisitorHeader(const char * working_directory)
  11 {
  12   assert(working_directory);
  13
  14   std::fstream out;
  15   std::string path(working_directory);
  16
  17   path.append("/default-visitor.hh");
  18   out.open(path.c_str(), std::fstream::out | std::fstream::trunc);
  19   assert(out.is_open() && out.good());
  20
  21   const ag::Node *node;
  22
  23   ag::appendGeneratedWarning(out);
  24   out << "#ifndef AST_DEFAULT_VISITOR_HH" NL
  25       << "# define AST_DEFAULT_VISITOR_HH" NL
  26       NL
  27       << "#include <ast/visitor.hh>" NL
  28       NL
  29       << "namespace ast" NL
  30       << "{" NL;
  31
  32 #define DEFAULT_VISITOR_HH(Prefix, Const)                               \
  33   out << "  /** @defgroup Visitors */" NL                               \
  34       << "  /**" NL                                                     \
  35       << "   * @brief The default visitor" NL                           \
  36       << "   * @ingroup Visitors" NL                                    \
  37       << "   */" NL                                                     \
  38       << "  class "Prefix"DefaultVisitor : public "Prefix"Visitor" NL   \
  39       << "  {" NL                                                       \
  40       << "  public:" NL                                                 \
  41       << "    virtual ~"Prefix"DefaultVisitor();" NL                    \
  42       NL;                                                               \
  43                                                                         \
  44   for (node = ag::nodes; !node->name.empty() > 0; node++)               \
  45     out << "    virtual void visit("Const << node->name << " & node);"  \
  46         NL;                                                             \
  47   out << "  };" NL NL;
  48
  49   DEFAULT_VISITOR_HH("", "");
  50   DEFAULT_VISITOR_HH("Const", "const ");
  51
  52   out << "} // namespace ast" NL
  53       << "#endif // !AST_DEFAULT_VISITOR_HH" NL;
  54   out.close();
  55 }
  56
  57 static void
  58 generateDefaultVisitorSource(const char * working_directory)
  59 {
  60   assert(working_directory);
  61
  62   std::fstream out;
  63   std::string path(working_directory);
  64
  65   path.append("/default-visitor.cc");
  66   out.open(path.c_str(), std::fstream::out | std::fstream::trunc);
  67   assert(out.is_open() && out.good());
  68
  69   const ag::Node *node;
  70
  71   ag::appendGeneratedWarning(out);
  72   out << "#include <ast/default-visitor.hh>" NL NL
  73       << "#define NL << std::endl" NL NL
  74       << "namespace ast" NL
  75       << "{" NL;
  76
  77 #define DEFAULT_VISITOR_CC(Prefix, Const)                               \
  78   out << "  "Prefix"DefaultVisitor::~"Prefix"DefaultVisitor() {}" NL; \
  79                                                                         \
  80   for (node = ag::nodes; !node->name.empty() > 0; node++)                   \
  81     out << "  void "Prefix"DefaultVisitor::visit("Const << node->name \
  82         << " &) {}" NL NL;
  83
  84   DEFAULT_VISITOR_CC("", "");
  85   DEFAULT_VISITOR_CC("Const", "const ");
  86
  87   out << "} // namespace ast" NL;
  88   out.close();
  89 }
  90
  91 int main(int argc, char **argv)
  92 {
  93   assert(argc == 2);
  94   generateDefaultVisitorHeader(argv[1]);
  95   generateDefaultVisitorSource(argv[1]);
  96   return 0;
  97 }