include Ast

(* Default tree transformation
 * (node -> node) -> node -> node *)
let rec transform visitor node =
    let trav = visitor in
    let trav_all nodes = List.map trav nodes in
    match node with
    | Program (decls) ->
        Program (trav_all decls)
    | FunDec (ret_type, name, params) ->
        FunDec (ret_type, name, trav_all params)
    | FunDef (export, ret_type, name, params, body) ->
        FunDef (export, ret_type, name, trav_all params, trav_all body)
    | GlobalDec (ctype, name) ->
        GlobalDec (ctype, name)
    | GlobalDef (export, ctype, name, Some init) ->
        GlobalDef (export, ctype, name, Some (trav init))

    | VarDec (ctype, name, Some init) ->
        VarDec (ctype, name, Some (trav init))
    | Assign (name, value) ->
        Assign (name, trav value)
    | Return (value) ->
        Return (trav value)
    | If (cond, body) ->
        If (trav cond, trav_all body)
    | IfElse (cond, true_body, false_body) ->
        IfElse (trav cond, trav_all true_body, trav_all false_body)
    | While (cond, body) ->
        While (trav cond, trav_all body)
    | DoWhile (cond, body) ->
        DoWhile (trav cond, trav_all body)
    | For (counter, start, stop, step, body) ->
        For (counter, trav start, trav stop, trav step, trav_all body)
    | Expr (value) ->
        Expr (trav value)

    | Monop (op, value) ->
        Monop (op, trav value)
    | Binop (op, left, right) ->
        Binop (op, trav left, trav right)
    | Cond (cond, true_expr, false_expr) ->
        Cond (trav cond, trav true_expr, trav false_expr)
    | TypeCast (ctype, value) ->
        TypeCast (ctype, trav value)
    | FunCall (name, args) ->
        FunCall (name, trav_all args)

    | _ -> node

(*
(* Visit
 *  *)
let rec visit visitor = function
    let trav = visit visitor in
    let trav_all nodes = List.map trav nodes in
    | Program (decls) -> List.map visitor decls;
    | Param (ctype, name) ->
    | FunDec (ret_type, name, params) ->
    | FunDef (export, ret_type, name, params, body) ->
    | GlobalDec (ctype, name) ->
    | GlobalDef (export, ctype, name, None) ->
    | GlobalDef (export, ctype, name, Some init) ->

    | VarDec (ctype, name, None) ->
    | VarDec (ctype, name, Some init) ->
    | Assign (name, value) ->
    | Return (value) ->
    | If (cond, body) ->
    | IfElse (cond, true_body, false_body) ->
    | While (cond, body) ->
    | DoWhile (cond, body) ->
    | For (counter, start, stop, step, body) ->
    | Expr (value) ->

    | BoolConst (value) ->
    | IntConst (value) ->
    | FloatConst (value) ->
    | Var (name) ->
    | Monop (op, value) ->
    | Binop (op, left, right) ->
    | Cond (cond, true_expr, false_expr) ->
    | TypeCast (ctype, value) ->
    | FunCall (name, args) ->
*)