civicaml/phases/expand_dims.ml

open Types
open Util

let rec expand_dims = function
    (* Flatten Block nodes returned by transformations below *)
    | FunDef (export, ret_type, name, params, body, ann) ->
        let params = flatten_blocks (List.map expand_dims params) in
        FunDef (export, ret_type, name, params, expand_dims body, ann)

    | FunDec (ret_type, name, params, ann) ->
        let params = flatten_blocks (List.map expand_dims params) in
        FunDec (ret_type, name, params, ann)

    | FunUse (dec, params, ann) ->
        FunUse (dec, flatten_blocks (List.map expand_dims params), ann)

    (* Add additional parameters for array dimensions *)
    | Param (Array (_,dims) as ctype, name, ann) ->
        let rec do_expand = function
            | [] ->
                [Param (ctype, name, ann)]
            | Dim (name, ann) :: tail ->
                Param (Int, name, ann) :: (do_expand tail)
            | _ -> raise InvalidNode
        in
        Block (do_expand dims)

    (* Add additional function arguments for array dimensions *)
    | Arg (VarUse (VarDec (Array (_, dims), _, _, _), None, _)) as node ->
        let rec do_expand = function
            | []       -> [node]
            | hd :: tl -> Arg (VarUse (hd, None, [])) :: (do_expand tl)
        in
        Block (do_expand dims)

    | node -> transform_children expand_dims node

let phase = function
    | Ast node -> Ast (expand_dims node)
    | _ -> raise (InvalidInput "expand dimensions")