types.ml

open Ast
(*On note ET pour Expression type et TE pour Typed expression*)
type expr_type =
  | ET_void
  | ET_int
  | ET_char
  | ET_struct of string
  | ET_union of string
  | ET_star of expr_type
  | ET_null


module Env = Map.Make(struct type t=string
  let compare = compare end)

type field = expr_type * string

type type_signature =
  | UnionSig of field list
  | StructSig of field list

type prototype = { return : expr_type;
  name : string;
  args : expr_type list }

type env_expr = (expr_type) Env.t
type env_sig = (type_signature) Env.t
type env_proto = (prototype) Env.t

let num_type = function
  | ET_int | ET_char | ET_null -> true
  | _ -> false

let pointer_type = function
  | ET_null | ET_star _ -> true
  | _ -> false

let rec string_of_type = function
  | ET_void -> "void"
  | ET_int -> "int"
  | ET_char -> "char"
  | ET_struct s -> Printf.sprintf "struct %s" s
  | ET_union s -> Printf.sprintf "union %s" s
  | ET_star e -> Printf.sprintf "%s*" (string_of_type e)
  | ET_null -> "nulltype"

type tident = expr_type * aident 

type wdecl_vars = tident list

type wexpr =
  | TE_int of int32
  | TE_char of char
  | TE_str of string
  | TE_ident of aident
  | TE_star of texpr
  | TE_dot of texpr * aident
  | TE_gets of texpr * texpr
  | TE_call of aident * (texpr list)
  | TE_incr of aincr * int (* step, sizeof *) * texpr
  | TE_unop of aunop * texpr
  | TE_binop of abinop * texpr * texpr
  and texpr = expr_type * wexpr

  and winstr = 
    | VT_none
    | VT_inst of texpr
    | VT_if of texpr*winstr
    | VT_if_else of texpr*winstr*winstr
    | VT_while of texpr*winstr
    | VT_for of (texpr list)*texpr option*(texpr list)*winstr
    | VT_bloc of wbloc
    | VT_return of texpr option
    and wbloc = wdecl_vars*(winstr list)

type wdecl =
  | Tdecl_vars of wdecl_vars
  | Tdecl_typ of (bool (* is_union *) * aident * (field list))
  | Tdecl_fct of (expr_type * aident * (tident list) * wbloc)

type wfichier = wdecl list