aboutsummaryrefslogtreecommitdiffstats
path: root/c_parser/include/ast_function.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'c_parser/include/ast_function.hpp')
-rw-r--r--c_parser/include/ast_function.hpp48
1 files changed, 48 insertions, 0 deletions
diff --git a/c_parser/include/ast_function.hpp b/c_parser/include/ast_function.hpp
new file mode 100644
index 0000000..86230d1
--- /dev/null
+++ b/c_parser/include/ast_function.hpp
@@ -0,0 +1,48 @@
+#ifndef AST_FUNCTION_HPP
+#define AST_FUNCTION_HPP
+
+#include "ast.hpp"
+
+#include <string>
+#include <iostream>
+
+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, 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;
+ }
+
+ virtual void push(const ast_Base* var) const {}
+};
+
+class ast_ParamList : public ast_Base {
+private:
+ mutable std::vector<const ast_Base*> param_list;
+
+public:
+ ast_ParamList(const ast_Base* param) {
+ param_list.push_back(param);
+ }
+
+ virtual void print() const {
+ for(size_t i = 0; i < param_list.size(); ++i) {
+ param_list[i]->print();
+ }
+ }
+
+ virtual void push(const ast_Base* _var) const {
+ param_list.push_back(_var);
+ }
+};
+
+#endif