src/eval.ml

   1 (***********************************************************************)
   2 (*                                                                     *)
   3 (*                               TAToo                                 *)
   4 (*                                                                     *)
   5 (*                     Kim Nguyen, LRI UMR8623                         *)
   6 (*                   Université Paris-Sud & CNRS                       *)
   7 (*                                                                     *)
   8 (*  Copyright 2010-2013 Université Paris-Sud and Centre National de la *)
   9 (*  Recherche Scientifique. All rights reserved.  This file is         *)
  10 (*  distributed under the terms of the GNU Lesser General Public       *)
  11 (*  License, with the special exception on linking described in file   *)
  12 (*  ../LICENSE.                                                        *)
  13 (*                                                                     *)
  14 (***********************************************************************)
  15
  16 (*
  17   Time-stamp: <Last modified on 2013-04-24 23:24:11 CEST by Kim Nguyen>
  18 *)
  19
  20 INCLUDE "utils.ml"
  21 open Format
  22
  23 module Make (T : Tree.S) :
  24   sig
  25     val eval : Ata.t -> T.t -> T.node -> T.node list
  26   end
  27  = struct
  28
  29
  30 IFDEF HTMLTRACE
  31   THEN
  32 DEFINE TRACE(e) = (e)
  33   ELSE
  34 DEFINE TRACE(e) = ()
  35 END
  36
  37    let html tree node i config msg =
  38      let config = config.Ata.Config.node in
  39      let oldi = config.Ata.round in
  40      Html.trace (T.preorder tree node) i oldi
  41        "node: %i<br/>%s<br/>sat: %a<br/>unsat: %a<br/>todo: %around: %i<br/>"
  42        ((T.preorder tree node):>int)
  43        msg
  44        StateSet.print config.Ata.sat
  45        StateSet.print config.Ata.unsat
  46        (Ata.TransList.print ~sep:"<br/>") config.Ata.todo oldi
  47
  48   type cache = StateSet.t Cache.N1.t
  49   let get c t n = Cache.N1.find c (T.preorder t n)
  50
  51   let set c t n v i =
  52     v.Ata.Config.node.Ata.round <- i;
  53     Cache.N1.add c (T.preorder t n) v
  54
  55
  56   let top_down_run auto tree node cache _i =
  57     let rec loop node =
  58       if node == T.nil then false else begin
  59         let parent = T.parent tree node in
  60         let fc = T.first_child tree node in
  61         let ns = T.next_sibling tree node in
  62         let tag = T.tag tree node in
  63         let _, config0 =
  64           let c = get cache tree node in
  65           if c == Cache.N1.dummy cache then
  66            true,Ata.Config.make
  67              { c.Ata.Config.node with
  68                Ata.todo = Ata.get_trans auto tag auto.Ata.states;
  69                summary = Ata.node_summary
  70                  (node == T.first_child tree parent) (* is_left *)
  71                  (node == T.next_sibling tree parent) (* is_right *)
  72                  (fc != T.nil) (* has_left *)
  73                  (ns != T.nil) (* has_right *)
  74                  (T.kind tree node) (* kind *)
  75              }
  76           else c
  77         in
  78
  79         TRACE(html tree node _i config0 "Entering node");
  80
  81         let _, ps = get cache tree parent in
  82         let old_unstable_left, fcs = get cache tree fc in
  83         let old_unstable_right, nss = get cache tree ns in
  84         let config1 = Ata.eval_trans auto fcs nss ps config0 in
  85
  86         TRACE(html tree node _i config1 "Updating transitions");
  87
  88         if config0 != config1 then set cache tree node (true,config1) _i;
  89         let unstable_left = old_unstable_left && loop fc in
  90         let _, fcs1 = get cache tree fc in
  91         let config2 = Ata.eval_trans auto fcs1 nss ps config1 in
  92
  93         TRACE(html tree node _i config2 "Updating transitions (after first-child)");
  94
  95         if config1 != config2 then set cache tree node (true, config2) _i;
  96         let unstable_right = old_unstable_right && loop ns in
  97         let _, nss1 = get cache tree ns in
  98         let config3 = Ata.eval_trans auto fcs1 nss1 ps config2 in
  99
 100         TRACE(html tree node _i config3 "Updating transitions (after next-sibling)");
 101
 102         if config2 != config3 then set cache tree node (true, config3) _i;
 103         let unstable =
 104           unstable_left
 105           || unstable_right
 106           || Ata.(TransList.nil != config3.Config.node.todo)
 107         in
 108         if Ata.(config3.Config.node.unstable_subtree) && not unstable then
 109           Ata.(config3.Config.node.unstable_subtree <- false);
 110         unstable
 111       end
 112     in
 113     loop node
 114
 115   let get_results auto tree node cache =
 116     let rec loop node acc =
 117       if node == T.nil then acc
 118       else
 119         let acc0 = loop (T.next_sibling tree node) acc in
 120         let acc1 = loop (T.first_child tree node) acc0 in
 121
 122         if StateSet.intersect
 123           (get cache tree node).Ata.Config.node.Ata.sat
 124           auto.Ata.selection_states then node::acc1
 125         else acc1
 126     in
 127     loop node []
 128
 129   let eval auto tree node =
 130     let cache = Cache.N1.create
 131       true , Ata.(Config.make { sat = StateSet.empty;
 132                               unsat = StateSet.empty;
 133                               todo = TransList.nil;
 134                               summary = dummy_summary;
 135                               round = ~-1
 136                             })
 137     in
 138     let redo = ref true in
 139     let iter = ref 0 in
 140     Ata.reset auto;
 141     while !redo do
 142       redo := false;
 143       Ata.reset auto; (* prevents the .cache2 and .cache4 memoization tables from growing too much *)
 144       redo := top_down_run auto tree node cache !iter;
 145       incr iter;
 146     done;
 147     at_exit (fun () -> eprintf "@[STATS: %i iterations@]@." !iter);
 148     let r = get_results auto tree node cache in
 149     TRACE(Html.gen_trace (module T : Tree.S with type t = T.t) (tree));
 150     r
 151
 152 end