From e83990754f49ec6eba7b9e1a5ff80a6add5de3a1 Mon Sep 17 00:00:00 2001
From: Nagi5Yeq <wu@x-nagi.com>
Date: Tue, 26 May 2020 00:26:57 +0800
Subject: bugfix & test

---
 node/BaseNode.hh | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'node/BaseNode.hh')

diff --git a/node/BaseNode.hh b/node/BaseNode.hh
index 16ee82d..71b0925 100644
--- a/node/BaseNode.hh
+++ b/node/BaseNode.hh
@@ -25,6 +25,21 @@ template <typename T, class Base = BaseNode> class ListNode : public Base {
   private:
     std::list<T> childs_;
 };
+
+class FunctionNode;
+
+class ProgramNode : public BaseNode {
+  public:
+    ProgramNode(Context* context, const std::string& name,
+                std::list<std::shared_ptr<BaseNode>> GlobalDeclarations,
+                std::list<std::shared_ptr<FunctionNode>> functions);
+    virtual Value CodeGen() override;
+
+  protected:
+    std::string name_;
+    std::list<std::shared_ptr<BaseNode>> GlobalDeclarations_;
+    std::list<std::shared_ptr<FunctionNode>> functions_;
+};
 } // namespace pcc
 
 #endif
-- 
cgit