1 files changed, 42 insertions, 0 deletions
diff --git a/tiger-compiler/src/inlining/pruner.hh b/tiger-compiler/src/inlining/pruner.hh
new file mode 100644
index 0000000..30a0609
--- /dev/null
+++ b/tiger-compiler/src/inlining/pruner.hh
@@ -0,0 +1,42 @@
+/**
+ ** \file inlining/pruner.hh
+ ** \brief Declaration of inlining::Pruner.
+ */
+
+#pragma once
+
+#include <map>
+
+#include <astclone/cloner.hh>
+#include <misc/set.hh>
+
+namespace inlining
+{
+  /// Prune useless function declarations within an AST.
+  class Pruner : public astclone::Cloner
+  {
+  public:
+    using super_type = astclone::Cloner;
+
+    // Import overloaded virtual functions.
+    using super_type::operator();
+
+    using func_vect = std::vector<ast::FunctionDec*>;
+    using func_count = std::map<misc::symbol, int>;
+
+    /// \name Visit methods.
+    /// \{
+    // FIXME: Some code was deleted here.
+    /// \}
+
+    ast::FunctionChunk* prune(ast::FunctionChunk& e);
+
+  private:
+    // Keep track of current nested functions
+    std::vector<const ast::FunctionDec*> current_functions_;
+
+    // Associate function dec with the number of CallExp calling it
+    func_count called_functions_;
+  };
+
+} // namespace inlining