1 files changed, 87 insertions, 0 deletions
diff --git a/tiger-compiler/src/parse/tweast.cc b/tiger-compiler/src/parse/tweast.cc
new file mode 100644
index 0000000..fbb21ac
--- /dev/null
+++ b/tiger-compiler/src/parse/tweast.cc
@@ -0,0 +1,87 @@
+/**
+ ** \file parse/tweast.cc
+ ** \brief Implementation of parse::Tweast.
+ */
+
+#include <sstream>
+
+#include <boost/algorithm/string/replace.hpp>
+
+#include <parse/tweast.hh>
+
+namespace parse
+{
+  unsigned Tweast::count_ = 0;
+
+  Tweast::Tweast()
+    : Tweast("")
+  {}
+
+  Tweast::Tweast(const std::string& str)
+    : MetavarMap<ast::Exp>::MetavarMap("exp")
+    , MetavarMap<ast::Var>::MetavarMap("lvalue")
+    , MetavarMap<ast::NameTy>::MetavarMap("namety")
+    , MetavarMap<ast::ChunkList>::MetavarMap("chunks")
+    , MetavarMap<Tweast>::MetavarMap("tweast")
+    , input_(str)
+  {}
+
+  void Tweast::flatten()
+  {
+    using tweasts_type = MetavarMap<Tweast>;
+    using tweasts_map_type = tweasts_type::map_type;
+
+    // Recursively flatten each Tweast metavariable.
+    for (const tweasts_map_type::value_type& t : tweasts_type::map_)
+      t.second->flatten();
+
+    // Grab each non-Tweast metavariable of Tweast metavariables, and
+    // renumber the metavariables.
+    for (const tweasts_map_type::value_type& t : tweasts_type::map_)
+      {
+        unsigned tweast_index = t.first;
+        Tweast& tweast = *t.second;
+
+        // Tweast input.
+        std::string tweast_input = tweast.input_get();
+        // Move metavariables to THIS and rename them.
+        move_metavars_<ast::Exp>(tweast, tweast_input);
+        move_metavars_<ast::Var>(tweast, tweast_input);
+        move_metavars_<ast::NameTy>(tweast, tweast_input);
+        move_metavars_<ast::ChunkList>(tweast, tweast_input);
+
+        // Replace the tweast metavariable reference with its input string.
+        std::string tweast_var = tweasts_type::show(tweast_index);
+        std::string input = input_.str();
+        boost::algorithm::replace_first(input, tweast_var, tweast_input);
+        input_.str(input);
+      }
+
+    // Delete aggregated Tweasts, and remove their associated
+    // metavariable from the map of Tweasts.
+    for (const tweasts_map_type::value_type& t : tweasts_type::map_)
+      delete t.second;
+    tweasts_type::map_.clear();
+  }
+
+  std::string Tweast::input_get() const { return input_.str(); }
+
+  std::ostream& Tweast::dump(std::ostream& ostr) const
+  {
+    ostr << "input = \"" << input_.str() << "\"" << misc::iendl;
+
+    MetavarMap<ast::Exp>::dump(ostr);
+    MetavarMap<ast::Var>::dump(ostr);
+    MetavarMap<ast::NameTy>::dump(ostr);
+    MetavarMap<ast::ChunkList>::dump(ostr);
+    MetavarMap<Tweast>::dump(ostr);
+
+    return ostr << misc::decendl;
+  }
+
+  std::ostream& operator<<(std::ostream& ostr, const Tweast& in)
+  {
+    return in.dump(ostr);
+  }
+
+} // namespace parse