Source file `TacCompute.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
(**************************************************************************)
(*                                                                        *)
(*  This file is part of WP plug-in of Frama-C.                           *)
(*                                                                        *)
(*  Copyright (C) 2007-2023                                               *)
(*    CEA (Commissariat a l'energie atomique et aux energies              *)
(*         alternatives)                                                  *)
(*                                                                        *)
(*  you can redistribute it and/or modify it under the terms of the GNU   *)
(*  Lesser General Public License as published by the Free Software       *)
(*  Foundation, version 2.1.                                              *)
(*                                                                        *)
(*  It 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 Lesser General Public License for more details.                   *)
(*                                                                        *)
(*  See the GNU Lesser General Public License version 2.1                 *)
(*  for more details (enclosed in the file licenses/LGPLv2.1).            *)
(*                                                                        *)
(**************************************************************************)

open Lang
open Tactical

(* -------------------------------------------------------------------------- *)
(* --- Compute Term                                                       --- *)
(* -------------------------------------------------------------------------- *)

exception Split of F.term * F.term * F.term
exception Unfold of F.term * Lang.lfun * F.term

let head e =
  if F.lc_closed e && not @@ F.is_primitive e then
    match F.repr e with
    | If(c,a,b) -> raise (Split(c,a,b))
    | Fun(f,es) -> raise (Unfold(e,f,TacUnfold.unfold f es))
    | _ -> ()

let process (fd : Tactical.feedback) ?at e =
  try
    let q = Queue.create () in
    let m = ref F.Tset.empty in
    let lookup e =
      head e ;
      m := F.Tset.add e !m ;
      F.lc_iter (fun e -> Queue.push e q) e
    in
    lookup e ;
    while not @@ Queue.is_empty q do
      let e = Queue.pop q in
      if not @@ F.Tset.mem e !m then lookup e ;
    done ;
    Not_applicable
  with
  | Split(a,b,c) ->
    let p = F.p_bool c in
    let q = F.p_not p in
    fd#set_title "Compute (split)" ;
    fd#set_descr "Split on if-then-else condition" ;
    Applicable (Tactical.rewrite ?at ["If-Then",p,e,a;"If-Else",q,e,b])
  | Unfold(e,f,d) ->
    fd#set_title "Compute (def)" ;
    fd#set_descr "Unfold definition of '%s'" (Lang.name_of_lfun f) ;
    Applicable (Tactical.rewrite ?at ["Definition",F.p_true,e,d])

class compute : Tactical.tactical =
  object
    inherit Tactical.make ~id:"Wp.compute"
        ~title:"Compute"
        ~descr:"Unfold definitions and split on conditions"
        ~params:[]
    method select feedback (s : Tactical.selection) =
      process feedback ?at:(Tactical.at s) (Tactical.selected s)
  end

let tactical = Tactical.export (new compute)
let strategy = Strategy.make tactical ~arguments:[]

(* -------------------------------------------------------------------------- *)