blob: 085957c1c688476a4c5e461566fff8648ec876cd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
#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 {
std::cerr << "Error: Can't call this function on this class" << std::endl;
(void)var;
}
};
class ast_ParamList : public ast_Base {
private:
mutable std::vector<const ast_Base*> param_list;
public:
ast_ParamList() {}
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
|