ast.h

#include <stdlib.h>
#include <llvm-c/Core.h>

#include "y.tab.h"

const char *type_name(enum value_type t);

enum expr_type {
  BOOL_LIT,
  LITERAL,
  DECIMAL,
  VARIABLE,
  BIN_OP,
};

struct expr {
  enum expr_type type;
  union {
    int value; // for type == LITERAL || type == BOOL_LIT
    int value_dec; // for type == DECIMAL
    size_t id; // for type == VARIABLE
    struct {
      struct expr *lhs;
      struct expr *rhs;
      int op;
    } binop; // for type == BIN_OP
  };
};

struct expr* bool_lit(int v);
struct expr* literal(int v);
struct expr* decimal(float v);
struct expr* variable(size_t id);
struct expr* binop(struct expr *lhs, int op, struct expr *rhs);

void print_expr(struct expr *expr);
void emit_stack_machine(struct expr *expr);
int emit_reg_machine(struct expr *expr);

enum value_type check_types(struct expr *expr);

void free_expr(struct expr *expr);

enum stmt_type {
  STMT_SEQ,
  STMT_ASSIGN,
  STMT_IF,
  STMT_WHILE,
  STMT_DOWHILE,
  STMT_FOR,
  STMT_PRINT,
};

struct stmt {
  enum stmt_type type;
  union {
    struct {
      size_t id;
      struct expr *expr;
    } assign; // for type == STMT_ASSIGN
    struct {
      struct stmt *fst, *snd;
    } seq; // for type == STMT_SEQ
    struct {
      struct expr *cond;
      struct stmt *if_body, *else_body;
    } ifelse; // for type == STMT_IF
    struct {
      struct expr *cond;
      struct stmt *body;
    } while_, dowhile; // for type == STMT_WHILE, STMT_DOWHILE
    struct {
      struct stmt *init;
      struct expr *cond;
      struct stmt *body;
      struct stmt *incr;
    } for_; // for type == STMT_FOR
    struct {
      struct expr *expr;
    } print; // for type == STMT_PRINT
  };
};

struct stmt* make_seq(struct stmt *fst, struct stmt *snd);
struct stmt* make_assign(size_t id, struct expr *e);
struct stmt* make_while(struct expr *e, struct stmt *body);
struct stmt* make_dowhile(struct stmt *body, struct expr *e);
struct stmt* make_for(struct stmt *init, struct expr *cond, struct stmt *incr, struct stmt *body);
struct stmt* make_ifelse(struct expr *e, struct stmt *if_body, struct stmt *else_body);
struct stmt* make_if(struct expr *e, struct stmt *body);
struct stmt* make_print(struct expr *e);
void free_stmt(struct stmt *stmt);
void print_stmt(struct stmt *stmt, int indent);
int valid_stmt(struct stmt *stmt);

LLVMValueRef codegen_expr(struct expr *expr, LLVMModuleRef module, LLVMBuilderRef builder);
void codegen_stmt(struct stmt *stmt, LLVMModuleRef module, LLVMBuilderRef builder);