src/ast/generator-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 generateVisitorHeader(const char * working_directory)
  11 {
  12   assert(working_directory);
  13
  14   std::fstream out;
  15   std::string path(working_directory);
  16
  17   path.append("/visitor.hh");
  18   out.open(path.c_str(), std::fstream::out | std::fstream::trunc);
  19   assert(out.is_open() && out.good());
  20
  21   ag::appendGeneratedWarning(out);
  22   out << "#ifndef AST_VISITOR_HH" NL
  23       << "# define AST_VISITOR_HH" NL
  24       NL
  25       << "#include <ast/ast.hh>" NL
  26       NL
  27       << "namespace ast" NL
  28       << "{" NL;
  29
  30   out << "  /** @defgroup Visitors */" NL
  31       << "  /**" NL
  32       << "   * @brief The Visitor" NL
  33       << "   * @ingroup Visitors" NL
  34       << "   */" NL
  35       << "  class Visitor" NL
  36       << "  {" NL
  37       << "  public:" NL
  38       << "    virtual ~Visitor() {};" NL
  39       NL;
  40
  41   const ag::Node *node;
  42   for (node = ag::nodes; !node->name.empty() > 0; node++)
  43     out << "    /** The visit method, see the Visitor design pattern. */" NL
  44         << "    virtual void visit(" << node->name << " & node) = 0;"
  45         NL NL;
  46   out << "  };" NL NL;
  47
  48   out << "  /**" NL
  49       << "   * @brief The const Visitor" NL
  50       << "   * @ingroup Visitors" NL
  51       << "   */" NL
  52       << "  class ConstVisitor" NL
  53       << "  {" NL
  54       << "  public:" NL
  55       << "    virtual ~ConstVisitor() {};" NL
  56       NL;
  57
  58   for (node = ag::nodes; !node->name.empty() > 0; node++)
  59     out << "    /** The visit method, see the Visitor design pattern. */" NL
  60         << "    virtual void visit(const " << node->name << " & node) = 0;"
  61         NL NL;
  62   out << "  };" NL;
  63
  64   out << "} // namespace ast" NL
  65       << "#endif // !AST_VISITOR_HH" NL;
  66   out.close();
  67 }
  68
  69 int main(int argc, char **argv)
  70 {
  71   assert(argc == 2);
  72   generateVisitorHeader(argv[1]);
  73   return 0;
  74 }