1 files changed, 51 insertions, 0 deletions
diff --git a/tiger-compiler/src/ast/assert-visitor.hh b/tiger-compiler/src/ast/assert-visitor.hh
new file mode 100644
index 0000000..0488c1c
--- /dev/null
+++ b/tiger-compiler/src/ast/assert-visitor.hh
@@ -0,0 +1,51 @@
+/**
+ ** \file ast/object-visitor.hh
+ ** \brief Provide default visits for assertion nodes.
+ */
+
+#pragma once
+
+#include <ast/visitor.hh>
+
+namespace ast
+{
+  template <template <typename> class Const>
+  class GenAssertVisitor : virtual public GenVisitor<Const>
+  {
+  public:
+    /// Super class type.
+    using super_type = GenVisitor<Const>;
+
+    // Import overloaded virtual functions.
+    using super_type::operator();
+
+    /// Convenient abbreviation.
+    template <typename Type> using const_t = typename Const<Type>::type;
+
+    /** \name Ctor & dtor.
+     ** \{ */
+    /// Construct an object visitor.
+    GenAssertVisitor();
+    /// Destroy an object visitor.
+    virtual ~GenAssertVisitor();
+    /** \} */
+
+    /// \name Object-related visits.
+    /// \{
+    void operator()(const_t<AssertExp>& e) override;
+    /// \}
+  };
+
+  /// Shorthand for a const visitor.
+  using AssertConstVisitor = GenAssertVisitor<misc::constify_traits>;
+  /// Shorthand for a non const visitor.
+  using AssertVisitor = GenAssertVisitor<misc::id_traits>;
+
+#ifdef SWIG
+  %template(AssertConstVisitor) GenAssertVisitor<misc::constify_traits>;
+  %template(AssertVisitor) GenAssertVisitor<misc::id_traits>;
+#endif
+
+} // namespace ast
+
+#include <ast/assert-visitor.hxx>