Odin/src/printer.cpp

gb_inline void print_indent(isize indent) {
	while (indent --> 0)
		gb_printf("  ");
}

void print_ast(AstNode *node, isize indent) {
	if (node == NULL)
		return;

	switch (node->kind) {
	case AstNode_BasicLit:
		print_indent(indent);
		print_token(node->basic_lit);
		break;
	case AstNode_Ident:
		print_indent(indent);
		print_token(node->ident.token);
		break;
	case AstNode_ProcLit:
		print_indent(indent);
		gb_printf("(proc lit)\n");
		print_ast(node->proc_lit.type, indent+1);
		print_ast(node->proc_lit.body, indent+1);
		break;

	case AstNode_CompoundLit:
		print_indent(indent);
		gb_printf("(compound lit)\n");
		print_ast(node->compound_lit.type, indent+1);
		print_ast(node->compound_lit.elem_list, indent+1);
		break;


	case AstNode_TagExpr:
		print_indent(indent);
		gb_printf("(tag)\n");
		print_indent(indent+1);
		print_token(node->tag_expr.name);
		print_ast(node->tag_expr.expr, indent+1);
		break;

	case AstNode_UnaryExpr:
		print_indent(indent);
		print_token(node->unary_expr.op);
		print_ast(node->unary_expr.expr, indent+1);
		break;
	case AstNode_BinaryExpr:
		print_indent(indent);
		print_token(node->binary_expr.op);
		print_ast(node->binary_expr.left, indent+1);
		print_ast(node->binary_expr.right, indent+1);
		break;
	case AstNode_CallExpr:
		print_indent(indent);
		gb_printf("(call)\n");
		print_ast(node->call_expr.proc, indent+1);
		print_ast(node->call_expr.arg_list, indent+1);
		break;
	case AstNode_SelectorExpr:
		print_indent(indent);
		gb_printf(".\n");
		print_ast(node->selector_expr.expr,  indent+1);
		print_ast(node->selector_expr.selector, indent+1);
		break;
	case AstNode_IndexExpr:
		print_indent(indent);
		gb_printf("([])\n");
		print_ast(node->index_expr.expr, indent+1);
		print_ast(node->index_expr.index, indent+1);
		break;
	case AstNode_CastExpr:
		print_indent(indent);
		gb_printf("(cast)\n");
		print_ast(node->cast_expr.type, indent+1);
		print_ast(node->cast_expr.expr, indent+1);
		break;
	case AstNode_DerefExpr:
		print_indent(indent);
		gb_printf("(deref)\n");
		print_ast(node->deref_expr.expr, indent+1);
		break;


	case AstNode_ExprStmt:
		print_ast(node->expr_stmt.expr, indent);
		break;
	case AstNode_IncDecStmt:
		print_indent(indent);
		print_token(node->inc_dec_stmt.op);
		print_ast(node->inc_dec_stmt.expr, indent+1);
		break;
	case AstNode_AssignStmt:
		print_indent(indent);
		print_token(node->assign_stmt.op);
		print_ast(node->assign_stmt.lhs_list, indent+1);
		print_ast(node->assign_stmt.rhs_list, indent+1);
		break;
	case AstNode_BlockStmt:
		print_indent(indent);
		gb_printf("(block)\n");
		print_ast(node->block_stmt.list, indent+1);
		break;

	case AstNode_IfStmt:
		print_indent(indent);
		gb_printf("(if)\n");
		print_ast(node->if_stmt.cond, indent+1);
		print_ast(node->if_stmt.body, indent+1);
		if (node->if_stmt.else_stmt) {
			print_indent(indent);
			gb_printf("(else)\n");
			print_ast(node->if_stmt.else_stmt, indent+1);
		}
		break;
	case AstNode_ReturnStmt:
		print_indent(indent);
		gb_printf("(return)\n");
		print_ast(node->return_stmt.result_list, indent+1);
		break;
	case AstNode_ForStmt:
		print_indent(indent);
		gb_printf("(for)\n");
		print_ast(node->for_stmt.init, indent+1);
		print_ast(node->for_stmt.cond, indent+1);
		print_ast(node->for_stmt.end, indent+1);
		print_ast(node->for_stmt.body, indent+1);
		break;
	case AstNode_DeferStmt:
		print_indent(indent);
		gb_printf("(defer)\n");
		print_ast(node->defer_stmt.stmt, indent+1);
		break;


	case AstNode_VarDecl:
		print_indent(indent);
		if (node->var_decl.kind == Declaration_Mutable)
			gb_printf("(decl:var,mutable)\n");
		else if (node->var_decl.kind == Declaration_Immutable)
			gb_printf("(decl:var,immutable)\n");
		print_ast(node->var_decl.name_list, indent+1);
		print_ast(node->var_decl.type, indent+1);
		print_ast(node->var_decl.value_list, indent+1);
		break;
	case AstNode_ProcDecl:
		print_indent(indent);
		if (node->proc_decl.kind == Declaration_Mutable)
			gb_printf("(decl:proc,mutable)\n");
		else if (node->proc_decl.kind == Declaration_Immutable)
			gb_printf("(decl:proc,immutable)\n");
		print_ast(node->proc_decl.type, indent+1);
		print_ast(node->proc_decl.body, indent+1);
		print_ast(node->proc_decl.tag_list, indent+1);
		break;

	case AstNode_TypeDecl:
		print_indent(indent);
		gb_printf("(type)\n");
		print_ast(node->type_decl.name, indent+1);
		print_ast(node->type_decl.type, indent+1);
		break;

	case AstNode_AliasDecl:
		print_indent(indent);
		gb_printf("(alias)\n");
		print_ast(node->alias_decl.name, indent+1);
		print_ast(node->alias_decl.type, indent+1);
		break;


	case AstNode_ProcType:
		print_indent(indent);
		gb_printf("(type:proc)(%td -> %td)\n", node->proc_type.param_count, node->proc_type.result_count);
		print_ast(node->proc_type.param_list, indent+1);
		if (node->proc_type.result_list) {
			print_indent(indent+1);
			gb_printf("->\n");
			print_ast(node->proc_type.result_list, indent+1);
		}
		break;
	case AstNode_Field:
		print_ast(node->field.name_list, indent);
		print_ast(node->field.type, indent);
		break;
	case AstNode_PointerType:
		print_indent(indent);
		print_token(node->pointer_type.token);
		print_ast(node->pointer_type.type, indent+1);
		break;
	case AstNode_ArrayType:
		print_indent(indent);
		gb_printf("[]\n");
		print_ast(node->array_type.count, indent+1);
		print_ast(node->array_type.elem, indent+1);
		break;
	case AstNode_StructType:
		print_indent(indent);
		gb_printf("(struct)\n");
		print_ast(node->struct_type.field_list, indent+1);
		break;
	}

	if (node->next)
		print_ast(node->next, indent);
}