CristalCaveGem » LustreC

module TopologicalDepGraph = Topological.Make(IdentDepGraph)

(* A module to sort dependencies among local variables when relying on clocked declarations *)

module VarClockDep =

struct

  let rec get_clock_dep ck =

    match ck.Clocks.cdesc with

    | Clocks.Con (ck ,c ,l) -> l::(get_clock_dep ck)

    | Clocks.Clink ck' 

    | Clocks.Ccarrying (_, ck') -> get_clock_dep ck'

    | _ -> []

  let sort locals =

    let g = new_graph () in

    let g = List.fold_left (

      fun g var_decl ->

	let deps = get_clock_dep var_decl.var_clock in

	add_edges [var_decl.var_id] deps g

    ) g locals

    in

    let sorted, no_deps =

      TopologicalDepGraph.fold (fun vid (accu, remaining) -> (

	let select v = v.var_id = vid in

	let selected, not_selected = List.partition select remaining in

	selected@accu, not_selected

      )) g ([],locals)

    in

    no_deps @ sorted

end

  let prog = SortProg.sort_nodes_locals prog in

	Causality.TopologicalDepGraph.fold 

let sort_node_locals nd =

  { nd with node_locals = Causality.VarClockDep.sort nd.node_locals}

let sort_nodes_locals prog =

  List.map

    (fun top ->

      match top.top_decl_desc with

      | Node nd -> {top with top_decl_desc = Node (sort_node_locals nd)}

      | _ -> top

    )

    prog