(**************************************************************************)
(*                                                                        *)
(*  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