aboutsummaryrefslogtreecommitdiffstats
path: root/c_compiler/include/function.hpp
blob: 6fbcdee0cdd6c08dc613ace69fc9de8abd08520b (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 Function : public Base {
private:
    std::string id;
    const Base* param;
    const Base* comp_statement;
public:
    Function(const std::string& _id, const Base* _param, const 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 Base* var) const {
        std::cerr << "Error: Can't call this function on this class" << std::endl;
	(void)var;
    }
};

class ParamList : public Base {
private:
    mutable std::vector<const Base*> param_list;

public:
    ParamList() {}
    
    ParamList(const 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 Base* _var) const {
	param_list.push_back(_var);
    }
};

#endif