X-Git-Url: http://git.nguyen.vg/gitweb/?p=tatoo.git;a=blobdiff_plain;f=src%2Fata.ml;h=190c4c72443ec18c6d76186eda0c96f326d37bd3;hp=565dfc8bf1595a7c2a75315a239f5dde8cebdce7;hb=6844beef9142975d1bf4a2b3fb72b93dfa3402dc;hpb=5f1d396f4a04c9a0c6bb3da4176fd47a4473d35d

diff --git a/src/ata.ml b/src/ata.ml
index 565dfc8..190c4c7 100644
--- a/src/ata.ml
+++ b/src/ata.ml
@@ -445,14 +445,24 @@ let compute_dependencies auto =
 
   edges
 
+let state_prerequisites dir auto q =
+  Hashtbl.fold (fun q' trans acc ->
+    List.fold_left (fun acc (_, phi) ->
+      let m_phi = Formula.get_states_by_move phi in
+      if StateSet.mem q (Move.get m_phi dir)
+      then StateSet.add q' acc else acc)
+      acc trans) auto.transitions StateSet.empty
 
 let compute_rank auto =
   let dependencies = compute_dependencies auto in
   let upward = [ `Stay ; `Parent ; `Previous_sibling ] in
   let downward = [ `Stay; `First_child; `Next_sibling ] in
   let swap dir = if dir == upward then downward else upward in
-  let is_satisfied q t =
-    Move.for_all (fun _ set -> StateSet.(is_empty (remove q set))) t
+  let is_satisfied dir q t =
+    Move.for_all (fun d set ->
+      if List.mem d dir then
+        StateSet.(is_empty (remove q set))
+      else StateSet.is_empty set) t
   in
   let update_dependencies dir initacc =
     let rec loop acc =
@@ -464,7 +474,7 @@ let compute_rank auto =
               Move.set deps m (StateSet.diff (Move.get deps m) to_remove)
             )
             dir;
-          if is_satisfied q deps then StateSet.add q acc else acc
+          if is_satisfied dir q deps then StateSet.add q acc else acc
         ) dependencies acc
       in
       if acc == new_acc then new_acc else loop new_acc