1 (* Copyright (C) 2004, HELM Team.
3 * This file is part of HELM, an Hypertextual, Electronic
4 * Library of Mathematics, developed at the Computer Science
5 * Department, University of Bologna, Italy.
7 * HELM is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version 2
10 * of the License, or (at your option) any later version.
12 * HELM is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with HELM; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
22 * For details, see the HELM World-Wide-Web page,
23 * http://helm.cs.unibo.it/
29 let critical_value = 7
32 module UriManagerSet = UriManager.UriSet
33 module SetSet = Set.Make (UriManagerSet)
35 type term_signature = (UriManager.uri * UriManager.uri list) option * UriManagerSet.t
37 type cardinality_condition =
42 type rating_criterion =
43 [ `Hits (** order by number of hits, most used objects first *)
47 (library_obj_tbl,library_rel_tbl,library_sort_tbl,library_count_tbl)
49 let current_tables () =
50 (obj_tbl (),rel_tbl (),sort_tbl (), count_tbl ())
52 let tbln n = "table" ^ string_of_int n
55 let add_depth_constr depth_opt cur_tbl where =
58 | Some depth -> (sprintf "%s.h_depth = %d" cur_tbl depth) :: where
61 let mk_positions positions cur_tbl =
66 let pos_str = MetadataPp.pp_position_tag pos in
71 | `MainConclusion None
72 | `MainHypothesis None ->
73 sprintf "%s.h_position = \"%s\"" cur_tbl pos_str
74 | `MainConclusion (Some r)
75 | `MainHypothesis (Some r) ->
76 let depth = MetadataPp.pp_relation r in
77 sprintf "(%s.h_position = \"%s\" and %s.h_depth %s)"
78 cur_tbl pos_str cur_tbl depth)
79 (positions :> MetadataTypes.position list)) ^
82 let explode_card_constr = function
83 | Eq card -> "=", card
84 | Gt card -> ">", card
85 | Lt card -> "<", card
87 let add_card_constr tbl col where = function
90 let op, card = explode_card_constr constr in
91 (* count(_utente).hypothesis = 3 *)
92 (sprintf "%s.%s %s %d" tbl col op card :: where)
94 let add_diff_constr tbl where = function
97 let op, card = explode_card_constr constr in
98 (sprintf "%s.hypothesis - %s.conclusion %s %d" tbl tbl op card :: where)
100 let add_all_constr ?(tbl=library_count_tbl) (n,from,where) concl full diff =
101 match (concl, full, diff) with
102 | None, None, None -> (n,from,where)
104 let cur_tbl = tbln n in
105 let from = (sprintf "%s as %s" tbl cur_tbl) :: from in
106 let where = add_card_constr cur_tbl "conclusion" where concl in
107 let where = add_card_constr cur_tbl "statement" where full in
108 let where = add_diff_constr cur_tbl where diff in
111 sprintf "table0.source = %s.source" cur_tbl :: where
116 let add_constraint ?(start=0) ?(tables=default_tables) (n,from,where) metadata =
117 let obj_tbl,rel_tbl,sort_tbl,count_tbl = tables
119 let cur_tbl = tbln n in
120 let start_table = tbln start in
122 | `Obj (uri, positions) ->
123 let from = (sprintf "%s as %s" obj_tbl cur_tbl) :: from in
125 (sprintf "(%s.h_occurrence = \"%s\")" cur_tbl (UriManager.string_of_uri uri)) ::
126 mk_positions positions cur_tbl ::
128 else [sprintf "%s.source = %s.source" start_table cur_tbl]) @
133 let from = (sprintf "%s as %s" rel_tbl cur_tbl) :: from in
135 mk_positions positions cur_tbl ::
137 else [sprintf "%s.source = %s.source" start_table cur_tbl]) @
141 | `Sort (sort, positions) ->
142 let sort_str = CicPp.ppsort sort in
143 let from = (sprintf "%s as %s" sort_tbl cur_tbl) :: from in
145 (sprintf "%s.h_sort = \"%s\"" cur_tbl sort_str ) ::
146 mk_positions positions cur_tbl ::
150 [sprintf "%s.source = %s.source" start_table cur_tbl ]) @ where
154 let exec ~(dbd:Mysql.dbd) ?rating (n,from,where) =
155 let from = String.concat ", " from in
156 let where = String.concat " and " where in
159 | None -> sprintf "select table0.source from %s where %s" from where
162 ("select table0.source from %s, hits where %s
163 and table0.source = hits.source order by hits.no desc")
166 (* prerr_endline query; *)
167 let result = Mysql.exec dbd query in
169 (fun row -> match row.(0) with Some s -> UriManager.uri_of_string s | _ -> assert false)
172 let at_least ~(dbd:Mysql.dbd) ?concl_card ?full_card ?diff ?rating tables
173 (metadata: MetadataTypes.constr list)
175 let obj_tbl,rel_tbl,sort_tbl, count_tbl = tables
177 if (metadata = []) && concl_card = None && full_card = None then
178 failwith "MetadataQuery.at_least: no constraints given";
180 List.fold_left (add_constraint ~tables) (0,[],[]) metadata
183 add_all_constr ~tbl:count_tbl (n,from,where) concl_card full_card diff
185 exec ~dbd ?rating (n,from,where)
188 ~(dbd:Mysql.dbd) ?concl_card ?full_card ?diff ?rating
189 (metadata: MetadataTypes.constr list)
191 if are_tables_ownerized () then
193 ~dbd ?concl_card ?full_card ?diff ?rating default_tables metadata) @
195 ~dbd ?concl_card ?full_card ?diff ?rating (current_tables ()) metadata)
198 ~dbd ?concl_card ?full_card ?diff ?rating default_tables metadata
201 (** Prefix handling *)
203 let filter_by_card n =
204 SetSet.filter (fun t -> (UriManagerSet.cardinal t) <= n)
207 let init = SetSet.union a b in
208 let merge_single_set s1 b =
210 (fun s2 res -> SetSet.add (UriManagerSet.union s1 s2) res)
213 SetSet.fold (fun s1 res -> SetSet.union (merge_single_set s1 b) res) a init
217 let string_of_uriref i =
218 UriManager.uri_of_string (UriManager.string_of_uriref i)
220 let rec inspect_children n childs =
222 (fun res term -> merge n (inspect_conclusion n term) res)
225 and add_root n root childs =
226 let childunion = inspect_children n childs in
227 let addroot = UriManagerSet.add root in
229 (fun child newsets -> SetSet.add (addroot child) newsets)
231 (SetSet.singleton (UriManagerSet.singleton root))
233 and inspect_conclusion n t =
234 if n = 0 then SetSet.empty
239 | Cic.Implicit _ -> SetSet.empty
240 | Cic.Var (u,exp_named_subst) -> SetSet.empty
241 | Cic.Const (u,exp_named_subst) ->
242 SetSet.singleton (UriManagerSet.singleton u)
243 | Cic.MutInd (u, t, exp_named_subst) ->
244 SetSet.singleton (UriManagerSet.singleton
245 (string_of_uriref (u, [t])))
246 | Cic.MutConstruct (u, t, c, exp_named_subst) ->
247 SetSet.singleton (UriManagerSet.singleton
248 (string_of_uriref (u, [t; c])))
249 | Cic.Cast (t, _) -> inspect_conclusion n t
250 | Cic.Prod (_, s, t) ->
251 merge n (inspect_conclusion n s) (inspect_conclusion n t)
252 | Cic.Lambda (_, s, t) ->
253 merge n (inspect_conclusion n s) (inspect_conclusion n t)
254 | Cic.LetIn (_, s, t) ->
255 merge n (inspect_conclusion n s) (inspect_conclusion n t)
256 | Cic.Appl ((Cic.Const (u,exp_named_subst))::l) ->
258 add_root (n-1) suri l
259 | Cic.Appl ((Cic.MutInd (u, t, exp_named_subst))::l) ->
260 let suri = string_of_uriref (u, [t]) in
261 add_root (n-1) suri l
262 | Cic.Appl ((Cic.MutConstruct (u, t, c, exp_named_subst))::l) ->
263 let suri = string_of_uriref (u, [t; c]) in
264 add_root (n-1) suri l
267 | Cic.MutCase (u, t, tt, uu, m) ->
271 | Cic.CoFix (_, m) ->
274 let rec inspect_term n t =
282 | Cic.Implicit _ -> None, SetSet.empty
283 | Cic.Var (u,exp_named_subst) -> None, SetSet.empty
284 | Cic.Const (u,exp_named_subst) ->
286 | Cic.MutInd (u, t, exp_named_subst) ->
287 let uri = string_of_uriref (u, [t]) in
288 Some uri, SetSet.empty
289 | Cic.MutConstruct (u, t, c, exp_named_subst) ->
290 let uri = string_of_uriref (u, [t; c]) in
291 Some uri, SetSet.empty
292 | Cic.Cast (t, _) -> inspect_term n t
293 | Cic.Prod (_, _, t) -> inspect_term n t
294 | Cic.LetIn (_, _, t) -> inspect_term n t
295 | Cic.Appl ((Cic.Const (u,exp_named_subst))::l) ->
297 let childunion = inspect_children (n-1) l in
298 Some suri, childunion
299 | Cic.Appl ((Cic.MutInd (u, t, exp_named_subst))::l) ->
300 let suri = string_of_uriref (u, [t]) in
301 if u = HelmLibraryObjects.Logic.eq_URI && n>1 then
302 (* equality is handled in a special way: in particular,
303 the type, if defined, is always added to the prefix,
304 and n is not decremented - it should have been n-2 *)
306 Cic.Const (u1,exp_named_subst1)::l1 ->
308 let inconcl = add_root (n-1) suri1 l1 in
310 | Cic.MutInd (u1, t1, exp_named_subst1)::l1 ->
311 let suri1 = string_of_uriref (u1, [t1]) in
312 let inconcl = add_root (n-1) suri1 l1 in
314 | Cic.MutConstruct (u1, t1, c1, exp_named_subst1)::l1 ->
315 let suri1 = string_of_uriref (u1, [t1; c1]) in
316 let inconcl = add_root (n-1) suri1 l1 in
318 | _ :: _ -> Some suri, SetSet.empty
319 | _ -> assert false (* args number must be > 0 *)
321 let childunion = inspect_children (n-1) l in
322 Some suri, childunion
323 | Cic.Appl ((Cic.MutConstruct (u, t, c, exp_named_subst))::l) ->
324 let suri = string_of_uriref (u, [t; c]) in
325 let childunion = inspect_children (n-1) l in
326 Some suri, childunion
327 | _ -> None, SetSet.empty
329 let add_cardinality s =
330 let l = SetSet.elements s in
334 let el = UriManagerSet.elements set in
335 (List.length el, el)) l in
336 (* ordered by descending cardinality *)
337 List.sort (fun (n,_) (m,_) -> m - n) ((0,[])::res)
340 match inspect_term n t with
341 Some a, set -> Some a, add_cardinality set
342 | None, set when (SetSet.is_empty set) -> None, []
343 | _, _ -> assert false
346 let rec add children =
348 (fun acc t -> UriManagerSet.union (signature_concl t) acc)
349 (UriManagerSet.empty) children
351 (* this function creates the set of all different constants appearing in
352 the conclusion of the term *)
353 and signature_concl =
358 | Cic.Implicit _ -> UriManagerSet.empty
359 | Cic.Var (u,exp_named_subst) -> UriManagerSet.empty
360 | Cic.Const (u,exp_named_subst) ->
361 UriManagerSet.singleton u
362 | Cic.MutInd (u, t, exp_named_subst) ->
363 let uri = string_of_uriref (u, [t]) in
364 UriManagerSet.singleton uri
365 | Cic.MutConstruct (u, t, c, exp_named_subst) ->
366 let uri = string_of_uriref (u, [t;c]) in
367 UriManagerSet.singleton uri
368 | Cic.Cast (t, _) -> signature_concl t
369 | Cic.Prod (_, s, t) ->
370 UriManagerSet.union (signature_concl s) (signature_concl t)
371 | Cic.Lambda (_, s, t) ->
372 UriManagerSet.union (signature_concl s) (signature_concl t)
373 | Cic.LetIn (_, s, t) ->
374 UriManagerSet.union (signature_concl s) (signature_concl t)
375 | Cic.Appl l -> add l
381 let rec signature_of = function
382 | Cic.Cast (t, _) -> signature_of t
383 | Cic.Prod (_, _, t) -> signature_of t
384 | Cic.LetIn (_, _, t) -> signature_of t
385 | Cic.Appl ((Cic.Const (u,exp_named_subst))::l) ->
387 Some (suri, []), add l
388 | Cic.Appl ((Cic.MutInd (u, t, exp_named_subst))::l) ->
389 let suri = string_of_uriref (u, [t]) in
390 if u = HelmLibraryObjects.Logic.eq_URI then
391 (* equality is handled in a special way: in particular,
392 the type, if defined, is always added to the prefix,
393 and n is not decremented - it should have been n-2 *)
395 Cic.Const (u1,exp_named_subst1)::l1 ->
397 let inconcl = UriManagerSet.remove suri1 (add l1) in
398 Some (suri, [suri1]), inconcl
399 | Cic.MutInd (u1, t1, exp_named_subst1)::l1 ->
400 let suri1 = string_of_uriref (u1, [t1]) in
401 let inconcl = UriManagerSet.remove suri1 (add l1) in
402 Some (suri, [suri1]), inconcl
403 | Cic.MutConstruct (u1, t1, c1, exp_named_subst1)::l1 ->
404 let suri1 = string_of_uriref (u1, [t1;c1]) in
405 let inconcl = UriManagerSet.remove suri1 (add l1) in
406 Some (suri, [suri1]), inconcl
407 | _ :: _ -> Some (suri, []), UriManagerSet.empty
408 | _ -> assert false (* args number must be > 0 *)
410 Some (suri, []), add l
411 | Cic.Appl ((Cic.MutConstruct (u, t, c, exp_named_subst))::l) ->
412 let suri = string_of_uriref (u, [t;c]) in
413 Some (suri, []), add l
414 | t -> None, signature_concl t
416 (* takes a list of lists and returns the list of all elements
417 without repetitions *)
419 let rec drop_repetitions = function
422 | u1::u2::l when u1 = u2 -> drop_repetitions (u2::l)
423 | u::l -> u::(drop_repetitions l) in
424 drop_repetitions (List.sort Pervasives.compare (List.concat l))
426 let must_of_prefix ?(where = `Conclusion) m s =
429 | `Conclusion -> [`InConclusion]
430 | `Statement -> [`InConclusion; `InHypothesis; `MainHypothesis None]
432 let s' = List.map (fun u -> `Obj (u, positions)) s in
433 `Obj (m, [`MainConclusion None]) :: s'
435 let escape = Str.global_replace (Str.regexp_string "\'") "\\'"
437 let get_constants (dbd:Mysql.dbd) ~where uri =
438 let uri = escape (UriManager.string_of_uri uri) in
441 | `Conclusion -> [ MetadataTypes.mainconcl_pos; MetadataTypes.inconcl_pos ]
443 [ MetadataTypes.mainconcl_pos; MetadataTypes.inconcl_pos;
444 MetadataTypes.inhyp_pos; MetadataTypes.mainhyp_pos ]
449 (List.map (fun pos -> sprintf "(h_position = \"%s\")" pos) positions)
451 sprintf ("SELECT h_occurrence FROM %s WHERE source=\"%s\" AND (%s) UNION "^^
452 "SELECT h_occurrence FROM %s WHERE source=\"%s\" AND (%s)")
453 (MetadataTypes.obj_tbl ()) uri pos_predicate
454 MetadataTypes.library_obj_tbl uri pos_predicate
457 let result = Mysql.exec dbd query in
458 let set = ref UriManagerSet.empty in
462 | Some uri -> set := UriManagerSet.add (UriManager.uri_of_string uri) !set
463 | _ -> assert false);
466 let at_most ~(dbd:Mysql.dbd) ?(where = `Conclusion) only u =
467 let inconcl = get_constants dbd ~where u in
468 UriManagerSet.subset inconcl only
470 (* Special handling of equality. The problem is filtering out theorems just
471 * containing variables (e.g. all the theorems in cic:/Coq/Ring/). Really
472 * ad-hoc, no better solution found at the moment *)
473 let myspeciallist_of_facts =
474 [0,UriManager.uri_of_string "cic:/Coq/Init/Logic/eq.ind#xpointer(1/1/1)"]
476 [0,UriManager.uri_of_string "cic:/Coq/Init/Logic/eq.ind#xpointer(1/1/1)";
477 (* 0,UriManager.uri_of_string "cic:/Coq/Init/Logic/sym_eq.con"; *)
478 0,UriManager.uri_of_string "cic:/Coq/Init/Logic/trans_eq.con";
479 0,UriManager.uri_of_string "cic:/Coq/Init/Logic/f_equal.con";
480 0,UriManager.uri_of_string "cic:/Coq/Init/Logic/f_equal2.con";
481 0,UriManager.uri_of_string "cic:/Coq/Init/Logic/f_equal3.con"]
484 let compute_exactly ~(dbd:Mysql.dbd) ?(facts=false) ~where main prefixes =
488 if ((m = 0) && (main = UriManager.uri_of_string HelmLibraryObjects.Logic.eq_XURI)) then
489 (if facts then myspeciallist_of_facts
493 let must = must_of_prefix ~where main s in
495 | `Conclusion -> at_least ~dbd ~concl_card:(Eq (m+1)) must
496 | `Statement -> at_least ~dbd ~full_card:(Eq (m+1)) must
498 List.map (fun uri -> (m, uri)) res)
501 (* critical value reached, fallback to "only" constraints *)
503 let compute_with_only ~(dbd:Mysql.dbd) ?(facts=false) ?(where = `Conclusion)
504 main prefixes constants
506 let max_prefix_length =
509 | (max,_)::_ -> max in
510 let maximal_prefixes =
511 let rec filter res = function
513 | (n,s)::l when n = max_prefix_length -> filter ((n,s)::res) l
515 filter [] prefixes in
521 let must = must_of_prefix ~where main s in
524 | `Conclusion -> at_least ~dbd ~concl_card:(Gt (m+1)) must
525 | `Statement -> at_least ~dbd ~full_card:(Gt (m+1)) must
527 (* we tag the uri with m+1, for sorting purposes *)
528 List.map (fun uri -> (m+1, uri)) res))
531 List.filter (function (_,uri) -> at_most ~dbd ~where constants uri) all in
532 let equal_to = compute_exactly ~dbd ~facts ~where main prefixes in
533 greater_than @ equal_to
535 (* real match query implementation *)
537 let cmatch ~(dbd:Mysql.dbd) ?(facts=false) t =
538 let (main, constants) = signature_of t in
541 | Some (main, types) ->
542 (* the type of eq is not counted in constants_no *)
543 let types_no = List.length types in
544 let constants_no = UriManagerSet.cardinal constants in
545 if (constants_no > critical_value) then
546 let prefixes = prefixes just_factor t in
548 | Some main, all_concl ->
550 List.fold_right UriManagerSet.add types (UriManagerSet.add main constants)
552 compute_with_only ~dbd ~facts main all_concl all_constants
555 (* in this case we compute all prefixes, and we do not need
556 to apply the only constraints *)
558 if constants_no = 0 then
559 (if types_no = 0 then
562 Some main, [0, []; types_no, types])
564 prefixes (constants_no+types_no+1) t
567 Some main, all_concl ->
568 compute_exactly ~dbd ~facts ~where:`Conclusion main all_concl
573 let must = must_of_prefix ~where:`Conclusion main s in
574 let res = at_least ~dbd ~concl_card:(Eq (m+1)) must in
575 List.map (fun uri -> (m, uri)) res)
579 let power_upto upto consts =
580 let l = UriManagerSet.elements consts in
581 List.sort (fun (n,_) (m,_) -> m - n)
584 List.filter (function (n,l) -> n <= upto)
585 res@(List.map (function (n,l) -> (n+1,a::l)) res))
589 let l = UriManagerSet.elements consts in
590 List.sort (fun (n,_) (m,_) -> m - n)
592 (fun res a -> res@(List.map (function (n,l) -> (n+1,a::l)) res))
595 type where = [ `Conclusion | `Statement ]
597 let sigmatch ~(dbd:Mysql.dbd)
598 ?(facts=false) ?(where = `Conclusion) (main, constants) =
601 | Some (main, types) ->
602 let constants_no = UriManagerSet.cardinal constants in
603 if (constants_no > critical_value) then
605 let subsets = power_upto just_factor constants in
606 let types_no = List.length types in
607 List.map (function (n,l) -> (n+types_no,types@l)) subsets
610 List.fold_right UriManagerSet.add types (UriManagerSet.add main constants)
612 compute_with_only ~dbd ~where main subsets all_constants
615 let subsets = power constants in
616 let types_no = List.length types in
618 (0,[]) :: List.map (function (n,l) -> (n+types_no,types@l)) subsets
621 compute_exactly ~dbd ~facts ~where main subsets
623 (* match query wrappers *)
627 let cmatch ~dbd ?(facts=false) term =
630 (fun x y -> Pervasives.compare (fst y) (fst x))
631 (cmatch' ~dbd ~facts term))
633 let constants_of = signature_concl