summaryrefslogtreecommitdiff
path: root/decls_cgen.c
blob: bbdb8a3bbe5480bc1218d4ed3c1ff8fbb1a5fd57 (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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
static bool cgen_decls_stmt(CGenerator *g, Statement *s);
static bool cgen_decls_block(CGenerator *g, Block *b);

static bool cgen_decls_expr(CGenerator *g, Expression *e) {
	switch (e->kind) {
	case EXPR_UNARY_OP:
		if (!cgen_decls_expr(g, e->unary.of))
			return false;
		break;
	case EXPR_BINARY_OP:
		if (!cgen_decls_expr(g, e->binary.lhs)
			|| !cgen_decls_expr(g, e->binary.rhs))
			return false;
		break;
	case EXPR_CAST:
		if (!cgen_decls_expr(g, e->cast.expr))
			return false;
		break;
	case EXPR_CALL:
		if (!cgen_decls_expr(g, e->call.fn))
			return false;
		arr_foreach(e->call.arg_exprs, Expression, a)
			if (!cgen_decls_expr(g, a))
				return false;
		break;
	case EXPR_BLOCK:
		if (!cgen_decls_block(g, &e->block))
			return false;
		break;
	case EXPR_IF:
		if (e->if_.cond)
			if (!cgen_decls_expr(g, e->if_.cond))
				return false;
		if (!cgen_decls_block(g, &e->if_.body))
			return false;
		if (e->if_.next_elif)
			if (!cgen_decls_expr(g, e->if_.next_elif))
				return false;
		break;
	case EXPR_WHILE:
		if (e->while_.cond)
			if (!cgen_decls_expr(g, e->while_.cond))
				return false;
		if (!cgen_decls_block(g, &e->while_.body))
			return false;
		break;
	case EXPR_TUPLE:
		arr_foreach(e->tuple, Expression, x)
			if (!cgen_decls_expr(g, x))
				return false;
		break;
	case EXPR_FN:
		e->fn.c.name = NULL;
	    e->fn.c.id = g->ident_counter++;
		if (!cgen_fn_header(g, &e->fn, e->where))
			return false;
		cgen_write(g, ";\n");
		if (!cgen_decls_block(g, &e->fn.body))
			return false;
		break;
	case EXPR_DIRECT:
	case EXPR_NEW:
	case EXPR_IDENT:
	case EXPR_LITERAL_BOOL:
	case EXPR_LITERAL_INT:
	case EXPR_LITERAL_STR:
	case EXPR_LITERAL_CHAR:
	case EXPR_LITERAL_FLOAT:
		break;
	}
	return true;
}

static bool cgen_decls_block(CGenerator *g, Block *b) {
	Block *prev = g->block;
	cgen_block_enter(g, b);
	arr_foreach(b->stmts, Statement, s)
		cgen_decls_stmt(g, s);
	cgen_block_exit(g, prev);
	return true;
}

static bool cgen_decls_decl(CGenerator *g, Declaration *d) {
	if (cgen_is_fn_direct(g, d)) {
		d->expr.fn.c.name = d->idents[0];
		if (!cgen_fn_header(g, &d->expr.fn, d->where))
			return false;
		cgen_write(g, ";\n");
		if (!cgen_decls_block(g, &d->expr.fn.body))
			return false;
	} else if (d->flags & DECL_FLAG_HAS_EXPR) {
		if (!cgen_decls_expr(g, &d->expr))
			return false;
	}
	return true;
}

static bool cgen_decls_stmt(CGenerator *g, Statement *s) {
	switch (s->kind) {
	case STMT_DECL:
		if (!cgen_decls_decl(g, &s->decl))
			return false;
		break;
	case STMT_EXPR:
		break;
	case STMT_RET:
		break;
	}
	return true;
}

static bool cgen_decls_file(CGenerator *g, ParsedFile *f) {
	cgen_write(g, "/* declarations */\n");
	arr_foreach(f->stmts, Statement, s) {
		if (!cgen_decls_stmt(g, s))
			return false;
	}
	return true;
}