prettyprint.ml

open Ast 

(* Pretty-printing functions *)

let rec string_of_op = function
    Add -> "+"
  | Sub -> "-"
  | Mult -> "*"
  | Div -> "/"
  | Mod -> "%"
  | Asn -> "="
  | Eq -> "=="
  | Neq -> "!="
  | Lt -> "<"
  | Leq -> "<="
  | Gt -> ">"
  | Geq -> ">="
  | And -> "and"
  | Or -> "or"

and string_of_unop o e = match o with
    Not -> "!" ^ string_of_expr e
  | Neg -> "-" ^ string_of_expr e
  | Mag -> "|" ^ string_of_expr e ^ "|"

and string_of_typ = function
    Int -> "int"
  | Double -> "double"
  | Char -> "char"
  | String -> "string"
  | Bool -> "bool"
  | Void -> "void"

and  str_brackets d str = if d > 0 then str_brackets (d - 1) ("[]" ^ str) else str

and string_of_array t d = string_of_typ t  ^ (str_brackets d "")


and string_of_datatype = function 
    UnsizedArray(t, d) -> string_of_array t d
  | SizedArray(t, el)   -> string_of_typ t ^ "[" ^ String.concat "][" (List.map string_of_int el) ^ "]"
  | Datatype(t) -> string_of_typ t

and string_of_expr = function
    IntLit(l) -> string_of_int l
  | DoubleLit(l) -> string_of_float l
  | StrLit(l) -> "\"" ^ l ^ "\""
  | CharLit(l) -> "'" ^ Char.escaped l ^ "'"
  | BoolLit(l) -> string_of_bool l
  | Id(s) -> s
  | Binop(e1, o, e2) -> "\t" ^ string_of_expr e1 ^ " " ^ string_of_op o ^ " " ^ string_of_expr e2
  | Unop(o, e) -> string_of_unop o e
  | ArrayListInit(l) -> "[" ^ String.concat ", " (List.map string_of_expr l) ^ "]"
  | ArrayAccess(id, dl) -> "\t" ^ id ^ "[" ^ String.concat "][" (List.map string_of_expr dl) ^ "]"
  | FuncCall(n, p) -> n ^ "(" ^ String.concat ", " (List.map string_of_expr p) ^ ")"
  | Noexpr -> ""

let string_of_argdecl a = string_of_datatype a.argdeclType ^ " " ^ a.argdeclID

let string_of_vardecl_simple vdecl =
    "\t" ^ string_of_datatype vdecl.declTyp ^ " " ^
    vdecl.declID ^ 
    string_of_expr vdecl.declInit ^ ";\n"

let string_of_init_vardecl vdecl =
    "\t" ^ string_of_datatype vdecl.declTyp ^ " " ^
    vdecl.declID ^ " = " ^
    string_of_expr vdecl.declInit ^ ";\n"

let string_of_vardecl vdecl = match vdecl.declInit with
    Noexpr ->  string_of_vardecl_simple vdecl
  | _ -> string_of_init_vardecl vdecl

let rec string_of_stmt = function
    Block(stmts) ->
      "{\n" ^ String.concat "" (List.map string_of_stmt stmts) ^ "}\n"
  | Expr(expr) -> string_of_expr expr ^ ";\n"   
  | Decl(decl) -> string_of_vardecl decl  
  | Return(expr) -> "return" ^ " " ^ string_of_expr expr ^ ";\n"
  | If(e, s1, s2) -> "if (" ^ string_of_expr e ^ ") {\n" ^ string_of_stmt s1 ^ "}\n else {\n" ^ string_of_stmt s2 ^ "}\n"
  | For(e1, e2, e3, s) -> "for (" ^ string_of_expr e1 ^ ";" ^ string_of_expr e2 ^ ";" ^ string_of_expr e3 ^ ")\n" ^ string_of_stmt s ^ ";\n"
  | While(e, s) -> "while (" ^ string_of_expr e ^ ")\n" ^ string_of_stmt s ^ ";\n"

let string_of_funcdecl fdecl =
  string_of_datatype fdecl.typ ^ " " ^
  fdecl.fname ^ "(" ^
  String.concat ", " (List.map string_of_argdecl fdecl.args) ^ ")\n{\n" ^
  String.concat "" (List.map string_of_stmt fdecl.body) ^ "}\n"

let string_of_prog (vars, funcs) = 
    String.concat "" (List.map string_of_vardecl vars) ^ "\n" ^
    String.concat "\n" (List.map string_of_funcdecl funcs)