You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
3.0 KiB
OCaml
59 lines
3.0 KiB
OCaml
(***********************************************************************)
|
|
(* *)
|
|
(* Heptagon *)
|
|
(* *)
|
|
(* Gwenael Delaval, LIG/INRIA, UJF *)
|
|
(* Leonard Gerard, Parkas, ENS *)
|
|
(* Adrien Guatto, Parkas, ENS *)
|
|
(* Cedric Pasteur, Parkas, ENS *)
|
|
(* *)
|
|
(* Copyright 2012 ENS, INRIA, UJF *)
|
|
(* *)
|
|
(* This file is part of the Heptagon compiler. *)
|
|
(* *)
|
|
(* Heptagon is free software: you can redistribute it and/or modify it *)
|
|
(* under the terms of the GNU General Public License as published by *)
|
|
(* the Free Software Foundation, either version 3 of the License, or *)
|
|
(* (at your option) any later version. *)
|
|
(* *)
|
|
(* Heptagon is distributed in the hope that it will be useful, *)
|
|
(* but WITHOUT ANY WARRANTY; without even the implied warranty of *)
|
|
(* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *)
|
|
(* GNU General Public License for more details. *)
|
|
(* *)
|
|
(* You should have received a copy of the GNU General Public License *)
|
|
(* along with Heptagon. If not, see <http://www.gnu.org/licenses/> *)
|
|
(* *)
|
|
(***********************************************************************)
|
|
open Heptagon
|
|
|
|
|
|
(* Transform [f (...) every e]
|
|
into [f (...) every r] and add an equation [r=e] *)
|
|
|
|
let is_var = function
|
|
| { e_desc = Evar _ } -> true
|
|
| _ -> false
|
|
|
|
let block funs acc b =
|
|
let b, (v, acc_eq_list) = Hept_mapfold.block funs ([], []) b in
|
|
{ b with b_local = v @ b.b_local; b_equs = acc_eq_list@b.b_equs }, acc
|
|
|
|
let edesc funs (v,acc_eq_list) ed =
|
|
let ed, (v, acc_eq_list) = Hept_mapfold.edesc funs (v,acc_eq_list) ed in
|
|
match ed with
|
|
| Eapp (op, e_list, Some re) when not (is_var re) ->
|
|
let re, vre, eqre = Reset.reset_var_from_exp re in
|
|
Eapp(op, e_list, Some re), (vre::v, eqre::acc_eq_list)
|
|
| Eiterator(it, op, n, pe_list, e_list, Some re) when not (is_var re) ->
|
|
let re, vre, eqre = Reset.reset_var_from_exp re in
|
|
Eiterator(it, op, n, pe_list, e_list, Some re),
|
|
(vre::v, eqre::acc_eq_list)
|
|
| _ -> ed, (v, acc_eq_list)
|
|
|
|
let program p =
|
|
let funs = { Hept_mapfold.defaults
|
|
with Hept_mapfold.edesc = edesc; Hept_mapfold.block = block } in
|
|
let p, _ = Hept_mapfold.program_it funs ([],[]) p in
|
|
p
|