From 83c59dc79524d36f4a01b99b70b89b7259afe863 Mon Sep 17 00:00:00 2001
From: Josh Berdine <jjb@fb.com>
Date: Wed, 8 Jan 2020 08:03:39 -0800
Subject: [PATCH] [sledge] Simplify Sh.fold_terms to fold_vars

Summary:
Also, previous code was sometimes inconsistent regarding whether to
enumerate all subterms or only toplevel terms.

Reviewed By: ngorogiannis

Differential Revision: D19221873

fbshipit-source-id: e8644098b
---
 sledge/src/llair/term.ml  |  4 ++--
 sledge/src/symbheap/sh.ml | 17 +++++++----------
 2 files changed, 9 insertions(+), 12 deletions(-)

diff --git a/sledge/src/llair/term.ml b/sledge/src/llair/term.ml
index f88f5f9ef..fcd80dd78 100644
--- a/sledge/src/llair/term.ml
+++ b/sledge/src/llair/term.ml
@@ -999,8 +999,8 @@ let fold_terms e ~init ~f =
   fix fold_terms_ (fun _ s -> s) e init
 
 let fold_vars e ~init ~f =
-  fold_terms e ~init ~f:(fun z -> function
-    | Var _ as v -> f z (v :> Var.t) | _ -> z )
+  fold_terms e ~init ~f:(fun s -> function
+    | Var _ as v -> f s (v :> Var.t) | _ -> s )
 
 (** Query *)
 
diff --git a/sledge/src/symbheap/sh.ml b/sledge/src/symbheap/sh.ml
index 9549d7586..449c8959d 100644
--- a/sledge/src/symbheap/sh.ml
+++ b/sledge/src/symbheap/sh.ml
@@ -107,7 +107,7 @@ let pp = pp Var.Set.empty Var.Set.empty
 let pp_djn = pp_djn Var.Set.empty Var.Set.empty
 
 let fold_terms_seg {loc; bas; len; siz; arr} ~init ~f =
-  let f b z = Term.fold_terms b ~init:z ~f in
+  let f b s = f s b in
   f loc (f bas (f len (f siz (f arr init))))
 
 let fold_vars_seg seg ~init ~f =
@@ -115,20 +115,17 @@ let fold_vars_seg seg ~init ~f =
 
 let fv_seg seg = fold_vars_seg seg ~f:Set.add ~init:Var.Set.empty
 
-let fold_terms fold_terms {us= _; xs= _; cong; pure; heap; djns} ~init ~f =
-  Equality.fold_terms ~init cong ~f
+let fold_vars fold_vars {us= _; xs= _; cong; pure; heap; djns} ~init ~f =
+  Equality.fold_terms ~init cong ~f:(fun init -> Term.fold_vars ~f ~init)
   |> fun init ->
-  List.fold ~init pure ~f:(fun init -> Term.fold_terms ~f ~init)
+  List.fold ~init pure ~f:(fun init -> Term.fold_vars ~f ~init)
   |> fun init ->
-  List.fold ~init heap ~f:(fun init -> fold_terms_seg ~f ~init)
+  List.fold ~init heap ~f:(fun init -> fold_vars_seg ~f ~init)
   |> fun init ->
-  List.fold ~init djns ~f:(fun init -> List.fold ~init ~f:fold_terms)
+  List.fold ~init djns ~f:(fun init -> List.fold ~init ~f:fold_vars)
 
 let rec fv_union init q =
-  Set.diff
-    (fold_terms fv_union q ~init ~f:(fun init ->
-         Term.fold_vars ~f:Set.add ~init ))
-    q.xs
+  Set.diff (fold_vars fv_union q ~init ~f:Set.add) q.xs
 
 let fv q = fv_union Var.Set.empty q