heptagon/compiler/obc/transformations/simplify.ml

51 lines
1.8 KiB
OCaml

(**************************************************************************)
(* *)
(* Heptagon *)
(* *)
(* Author : Marc Pouzet *)
(* Organization : Demons, LRI, University of Paris-Sud, Orsay *)
(* *)
(**************************************************************************)
(** This module simplify static expression of the program and deal with :
(0^n)[3] ==> 0
[3,4,5][2] ==> 5
*)
open Names
open Types
open Static
open Obc
open Obc_mapfold
let extvaluedesc funs acc evd = match evd with
| Wconst se ->
Wconst (simplify QualEnv.empty se), acc
| Warray (ev,e) ->
let ev, acc = extvalue_it funs acc ev in
(match ev.w_desc with
| Wconst { se_desc = Sarray_power (sv, [_]) } ->
Wconst sv, acc
| Wconst ({ se_desc = Sarray_power (sv, _::idx) } as arr) ->
Wconst {arr with se_desc = Sarray_power (sv, idx)}, acc
| Wconst { se_desc = Sarray sv_l } ->
(match e.e_desc with
| Eextvalue { w_desc = Wconst i } ->
(try
let indice = int_of_static_exp QualEnv.empty i in
Wconst (Misc.nth_of_list (indice+1) sv_l), acc
with _ -> raise Errors.Fallback)
| _ -> raise Errors.Fallback
)
| _ -> raise Errors.Fallback
)
| _ -> raise Errors.Fallback
let program p =
let funs = { defaults with evdesc = extvaluedesc } in
let p, _ = program_it funs [] p in
p