3 files changed, 42 insertions, 1 deletions
diff --git a/c_parser/include/ast.hpp b/c_parser/include/ast.hpp
index 7292af2..787e928 100644
--- a/c_parser/include/ast.hpp
+++ b/c_parser/include/ast.hpp
@@ -2,6 +2,7 @@
 #define AST_HPP
 
 #include "ast_base.hpp"
+#include "ast_statement.hpp"
 #include "ast_function.hpp"
 #include "ast_declaration.hpp"
 #include "ast_primitives.hpp"
diff --git a/c_parser/include/ast_function.hpp b/c_parser/include/ast_function.hpp
index 9383601..86230d1 100644
--- a/c_parser/include/ast_function.hpp
+++ b/c_parser/include/ast_function.hpp
@@ -10,12 +10,15 @@ class ast_Function : public ast_Base {
 private:
     std::string id;
     const ast_Base* param;
+    const ast_Base* comp_statement;
 public:
-    ast_Function(const std::string& _id, const ast_Base* _param) : id(_id), param(_param) {}
+    ast_Function(const std::string& _id, const ast_Base* _param, const ast_Base* _comp_statement) :
+	id(_id), param(_param), comp_statement(_comp_statement) {}
 
     virtual void print() const {
 	std::cout << "<Function id=\"" << id << "\">" << std::endl;
         param->print();
+	comp_statement->print();
 	std::cout << "</Function>" << std::endl;
     }
 
diff --git a/c_parser/include/ast_statement.hpp b/c_parser/include/ast_statement.hpp
new file mode 100644
index 0000000..725308b
--- /dev/null
+++ b/c_parser/include/ast_statement.hpp
@@ -0,0 +1,37 @@
+#ifndef AST_STATEMENT_HPP
+#define AST_STATEMENT_HPP
+
+class ast_Statement : public ast_Base {
+protected:
+    mutable std::vector<const ast_Base*> ast_list;
+    
+public:
+    ast_Statement(const ast_Base* _el) {
+	ast_list.push_back(_el);
+    }
+
+    virtual void print() const {
+	for(size_t i = 0; i < ast_list.size(); ++i) {
+	    ast_list[i]->print();
+	}
+    }
+
+    virtual void push(const ast_Base* _var) const {
+	ast_list.push_back(_var);
+    }
+};
+
+class ast_CompoundStatement : public ast_Statement {
+public:
+    ast_CompoundStatement(const ast_Base* _el) : ast_Statement(_el) {}
+
+    virtual void print() const override {
+	std::cout << "<Scope>" << std::endl;
+	for(size_t i = 0; i < ast_list.size(); ++i) {
+	    ast_list[i]->print();
+	}
+	std::cout << "</Scope>" << std::endl;
+    }
+};
+
+#endif