ast.ml

type vtype =
	Int | Char | Double | Bool | String | Any

type expr =
	Tree of expr * expr list
|	IntLit of string
|	ChrLit of string
|	FltLit of string
|	StrLit of string
|	GetBranch of expr * expr
|	Void
|	FunCall of string * expr
|	Eq of expr * expr
|	Neq of expr * expr
|	Lt of expr * expr
|	Leq of expr * expr
|	Gt of expr * expr
|	Geq of expr * expr
|	Add of expr * expr
|	Minus of expr * expr
|	Mul of expr * expr
|	Div of expr * expr
|	Mod of expr * expr
|	Cast of vtype * expr
|	GetWidth of expr
|	Id of string

type stmt =
	While of expr * stmt
|	If of expr * stmt
|	IfElse of expr * stmt * stmt
|	FuncDec of string * vtype * string * stmt
|	VarDec of vtype * string * expr
|	Assn of string * expr
|	Expr of expr
|	Return of expr
|	Seq of stmt list

type program = stmt list

let string_of_vtype = function
	Int -> "int"
| Char -> "char"
| Double -> "double"
| Bool -> "bool"
| String -> "string"
| Any -> "any"

let rec string_of_expr = function
	Tree(e,l) -> string_of_expr e ^ " {{ " ^
		String.concat ", " (List.map string_of_expr l) ^ "}}"
| IntLit(s) -> s
| ChrLit(s) -> s
| FltLit(s) -> s
| StrLit(s) -> s
| GetBranch(e1,e2) -> string_of_expr e1 ^ "." ^ string_of_expr e2
| GetWidth(e1) -> string_of_expr e1
| Void -> "void"
| FunCall(s,e) -> s ^ "(" ^ string_of_expr e ^ ")"
| Eq(e1, e2) -> string_of_expr e1 ^ "==" ^ string_of_expr e2
| Neq(e1, e2) -> string_of_expr e1 ^ "!=" ^ string_of_expr e2
| Lt(e1, e2) -> string_of_expr e1 ^ "<" ^ string_of_expr e2
| Leq(e1, e2) -> string_of_expr e1 ^ "<=" ^ string_of_expr e2
| Gt(e1, e2) -> string_of_expr e1 ^ ">" ^ string_of_expr e2
| Geq(e1, e2) -> string_of_expr e1 ^ ">=" ^ string_of_expr e2
| Add(e1, e2) -> string_of_expr e1 ^ "+" ^ string_of_expr e2
| Minus(e1, e2) -> string_of_expr e1 ^ "-" ^ string_of_expr e2
| Mul(e1, e2) -> string_of_expr e1 ^ "*" ^ string_of_expr e2
| Div(e1, e2) -> string_of_expr e1 ^ "/" ^ string_of_expr e2
| Mod(e1, e2) -> string_of_expr e1 ^ "%" ^ string_of_expr e2
| Cast(vt,e) -> string_of_vtype vt ^ " to " ^ string_of_expr e
| Id(s) -> s

(* mixing c-like and pltree-like syntax. not on purpose *)
let rec string_of_stmt = function
	While(e,s) -> "While(" ^ string_of_expr e ^ ") {" ^ string_of_stmt s ^ "}"
| If(e,s) -> "If(" ^ string_of_expr e ^ ") {" ^ string_of_stmt s ^ "}"
| IfElse(e,s1,s2) -> "If(" ^ string_of_expr e ^ ") {" ^ string_of_stmt s1 ^ "} Else {" ^ string_of_stmt s2 ^ "}"
| FuncDec(s,vt,vn,l) -> s ^ "[" ^ string_of_stmt l ^ "]"
| VarDec(v,s,e) -> s ^ " = " ^ string_of_expr e
| Assn(s,e) -> s ^ " = " ^ string_of_expr e
| Expr(e) -> string_of_expr e
| Return(e) -> "Return(" ^ string_of_expr e ^ ")"
| Seq(l) ->  String.concat ", " (List.map string_of_stmt l)

let string_of_program stmts =
  String.concat " " (List.map string_of_stmt stmts)