]> matita.cs.unibo.it Git - helm.git/blobdiff - helm/ocaml/metadata/metadataConstraints.ml
- changed metadata type so that positions contains depth
[helm.git] / helm / ocaml / metadata / metadataConstraints.ml
index 314ceba5345e616b18a80120d4f3f9cffc460712..2e5250f6208df28f07346803fa60d97c598eacf6 100644 (file)
@@ -69,18 +69,20 @@ let mk_positions positions cur_tbl =
 let add_card_constr tbl (n,from,where) = function
   | None -> (n,from,where)
   | Some (Eq card) ->
+      let cur_tbl = tbln n in
       (n+1,
-       (sprintf "%s as %s" tbl (tbln n) :: from),
-       (sprintf "no=%d" card ::
+       (sprintf "%s as %s" tbl cur_tbl :: from),
+       (sprintf "%s.no=%d" cur_tbl card ::
         (if n=0 then []
-        else [sprintf "table0.source = %s.source" (tbln n)]) @
+        else [sprintf "table0.source = %s.source" cur_tbl]) @
         where))
   | Some (Gt card) ->
+      let cur_tbl = tbln n in
       (n+1,
-       (sprintf "%s as %s" tbl (tbln n) :: from),
-       (sprintf "no>%d" card ::
+       (sprintf "%s as %s" tbl cur_tbl :: from),
+       (sprintf "%s.no>%d" cur_tbl card ::
         (if n=0 then []
-        else [sprintf "table0.source = %s.source" (tbln n)]) @
+        else [sprintf "table0.source = %s.source" cur_tbl]) @
         where))
 
 let at_least ~(dbh:Dbi.connection) ?concl_card ?full_card
@@ -135,7 +137,6 @@ let at_least ~(dbh:Dbi.connection) ?concl_card ?full_card
   let from = String.concat ", " from in
   let where = String.concat " and " where in
   let query = sprintf "select table0.source from %s where %s" from where in
-prerr_endline query;
   let query = dbh#prepare query in
   query#execute [];
   List.map (function [`String s] -> s | _ -> assert false) (query#fetchall ())
@@ -170,8 +171,8 @@ and add_root n root childs =
       (SetSet.singleton (StringSet.singleton root))
 
 and inspect_conclusion n t = 
-if n = 0 then SetSet.empty
-else match t with
+  if n = 0 then SetSet.empty
+  else match t with
       Cic.Rel _                    
     | Cic.Meta _                     
     | Cic.Sort _ 
@@ -362,17 +363,30 @@ let union l =
     | u::l -> u::(drop_repetitions l) in
   drop_repetitions (List.sort Pervasives.compare (List.concat l))
 
-let must_of_prefix m s =
-  let s' = List.map (fun u -> `Obj (u, [`InConclusion])) s in
+let must_of_prefix ?(where = `Conclusion) m s =
+  let positions =
+    match where with
+    | `Conclusion -> [`InConclusion]
+    | `Statement -> [`InConclusion; `InHypothesis; `MainHypothesis None]
+  in
+  let s' = List.map (fun u -> `Obj (u, positions)) s in
   `Obj (m, [`MainConclusion None]) :: s'
 
 let escape = Str.global_replace (Str.regexp_string "\'") "\\'"
 
-let get_inconcl (dbh:Dbi.connection) uri =
+let get_constants (dbh:Dbi.connection) ~where uri =
   let uri = escape uri in
+  let positions =
+    match where with
+    | `Conclusion -> ["\"MainConclusion\""; "\"InConclusion\""]
+    | `Statement ->
+        ["\"MainConclusion\""; "\"InConclusion\""; "\"InHypothesis\"";
+         "\"MainHypothesis\""]
+  in
   let query = 
-    dbh#prepare (sprintf "select h_occurrence from refObj where source=\"%s\" and (h_position=\"MainConclusion\" or h_position=\"InConclusion\")"
-      uri)
+    dbh#prepare
+      (sprintf "select h_occurrence from refObj where source=\"%s\" and (%s)"
+        uri (String.concat " or " positions))
   in
   query#execute [];
   query#fold_left (* transform the result in a set *)
@@ -381,8 +395,8 @@ let get_inconcl (dbh:Dbi.connection) uri =
       StringSet.add uri set)
     StringSet.empty
 
-let test_only ~(dbh:Dbi.connection) only u =
-  let inconcl = get_inconcl dbh u in
+let at_most ~(dbh:Dbi.connection) ?(where = `Conclusion) only u =
+  let inconcl = get_constants dbh ~where u in
   StringSet.subset inconcl only
 
   (* Special handling of equality. The problem is filtering out theorems just
@@ -396,7 +410,7 @@ let myspeciallist =
    0,"cic:/Coq/Init/Logic/f_equal2.con";
    0,"cic:/Coq/Init/Logic/f_equal3.con"]
 
-let compute_exactly ~(dbh:Dbi.connection) main prefixes =
+let compute_exactly ~(dbh:Dbi.connection) where main prefixes =
   List.concat
     (List.map 
       (fun (m,s) -> 
@@ -404,13 +418,18 @@ let compute_exactly ~(dbh:Dbi.connection) main prefixes =
           myspeciallist
         else
           let res =
-            at_least ~dbh ~concl_card:(Eq (m+1)) (must_of_prefix main s)
+            let must = must_of_prefix ~where main s in
+            match where with
+            | `Conclusion -> at_least ~dbh ~concl_card:(Eq (m+1)) must
+            | `Statement -> at_least ~dbh ~full_card:(Eq (m+1)) must
           in
           List.map (fun uri -> (m, uri)) res)
       prefixes)
 
   (* critical value reached, fallback to "only" constraints *)
-let compute_with_only ~(dbh:Dbi.connection) main prefixes constants =
+let compute_with_only ~(dbh:Dbi.connection) ?(where = `Conclusion) main
+  prefixes constants
+=
   let max_prefix_length = 
     match prefixes with
     | [] -> assert false 
@@ -426,15 +445,18 @@ let compute_with_only ~(dbh:Dbi.connection) main prefixes constants =
       union
         (List.map 
           (fun (m,s) -> 
+            let must = must_of_prefix ~where main s in
             (let res = 
-              at_least ~dbh ~concl_card:(Gt (m+1)) (must_of_prefix main s)
+              match where with
+              | `Conclusion -> at_least ~dbh ~concl_card:(Gt (m+1)) must
+              | `Statement -> at_least ~dbh ~full_card:(Gt (m+1)) must
             in
             (* we tag the uri with m+1, for sorting purposes *)
             List.map (fun uri -> (m+1, uri)) res))
           maximal_prefixes)
     in
-    List.filter (function (_,uri) -> test_only ~dbh constants uri) all in
-    let equal_to = compute_exactly ~dbh main prefixes in
+    List.filter (function (_,uri) -> at_most ~dbh ~where constants uri) all in
+    let equal_to = compute_exactly ~dbh where main prefixes in
     greater_than @ equal_to
 
   (* real match query implementation *)
@@ -454,23 +476,25 @@ let cmatch ~(dbh:Dbi.connection) t =
             in
             compute_with_only ~dbh main all_concl all_constants
          | _, _ -> [])
-      else if constants_no = 0 then []
       else
         (* in this case we compute all prefixes, and we do not need
            to apply the only constraints *)
-        let prefixes = prefixes constants_no t in
+        let prefixes =
+          if constants_no = 0 then
+            Some main, [0, []; List.length types, types]
+          else
+            prefixes constants_no t
+        in
         (match prefixes with
-             Some main, all_concl ->
-               List.concat
-                 (List.map 
-                    (fun (m,s) -> 
-                      (let res = 
-                        at_least ~dbh ~concl_card:(Eq (m+1))
-                          (must_of_prefix main s)
-                      in
-                      List.map (fun uri -> (m, uri)) res))
-                    all_concl)
-           | _, _ -> [])
+           Some main, all_concl ->
+             List.concat
+               (List.map 
+                  (fun (m,s) -> 
+                    let must = must_of_prefix ~where:`Conclusion main s in
+                    let res = at_least ~dbh ~concl_card:(Eq (m+1)) must in
+                    List.map (fun uri -> (m, uri)) res)
+                  all_concl)
+         | _, _ -> [])
 
 let power_upto upto consts =
   let l = StringSet.elements consts in
@@ -488,7 +512,9 @@ let power consts =
     (fun res a -> res@(List.map (function (n,l) -> (n+1,a::l)) res)) 
      [(0,[])] l)
 
-let sigmatch ~(dbh:Dbi.connection) (main, constants) =
+type where = [ `Conclusion | `Statement ]
+
+let sigmatch ~(dbh:Dbi.connection) ?(where = `Conclusion) (main, constants) =
   match main with
     None -> []
   | Some (main, types) ->
@@ -502,14 +528,14 @@ let sigmatch ~(dbh:Dbi.connection) (main, constants) =
         let all_constants = 
           List.fold_right StringSet.add types (StringSet.add main constants)
         in
-        compute_with_only ~dbh main subsets all_constants
+        compute_with_only ~dbh ~where main subsets all_constants
       else
         let subsets = 
           let subsets = power constants in
           let types_no = List.length types in
-          List.map (function (n,l) -> (n+types_no,types@l)) subsets
+          (0,[]) :: List.map (function (n,l) -> (n+types_no,types@l)) subsets
         in
-        compute_exactly ~dbh main subsets
+        compute_exactly ~dbh where main subsets
 
   (* match query wrappers *)
 let cmatch' = cmatch