(*
    ||M||  This file is part of HELM, an Hypertextual, Electronic        
    ||A||  Library of Mathematics, developed at the Computer Science     
    ||T||  Department, University of Bologna, Italy.                     
    ||I||                                                                
    ||T||  HELM is free software; you can redistribute it and/or         
    ||A||  modify it under the terms of the GNU General Public License   
    \   /  version 2 or (at your option) any later version.      
     \ /   This software is distributed as is, NO WARRANTY.     
      V_______________________________________________________________ *)

(* $Id: index.mli 9822 2009-06-03 15:37:06Z tassi $ *)

module Superposition (B : Terms.Blob) = 
  struct
    module IDX = Index.Index(B)
    module Unif = FoUnif.Founif(B)
    module Subst = FoSubst.Subst(B)
    module Order = Orderings.Orderings(B)
    module Utils = FoUtils.Utils(B)
    module Pp = Pp.Pp(B)

    let all_positions pos ctx t f =
      let rec aux pos ctx = function
      | Terms.Leaf _ as t -> f t pos ctx 
      | Terms.Var _ -> []
      | Terms.Node l as t-> 
          let acc, _, _ = 
            List.fold_left
            (fun (acc,pre,post) t -> (* Invariant: pre @ [t] @ post = l *)
                let newctx = fun x -> ctx (Terms.Node (pre@[x]@post)) in
                let acc = aux (List.length pre :: pos) newctx t @ acc in
                if post = [] then acc, l, []
                else acc, pre @ [t], List.tl post)
             (f t pos ctx, [], List.tl l) l
          in
           acc
      in
        aux pos ctx t
    ;;

    let superposition_right table varlist subterm pos context =
      let cands = IDX.DT.retrieve_unifiables table subterm in
      HExtlib.filter_map
        (fun (dir, (id,lit,vl,_ (*as uc*))) ->
           match lit with
           | Terms.Predicate _ -> assert false
           | Terms.Equation (l,r,_,o) ->
               assert(o <> Terms.Eq);
               let side, newside = if dir=Terms.Left2Right then l,r else r,l in
               try 
                 let subst, varlist = 
                   Unif.unification (varlist@vl) [] subterm side 
                 in
                 if o = Terms.Incomparable then
                   let side = Subst.apply_subst subst side in
                   let newside = Subst.apply_subst subst newside in
                   let o = Order.compare_terms side newside in
                   (* XXX: check Riazanov p. 33 (iii) *)
                   if o <> Terms.Lt && o <> Terms.Eq then  
                     Some (context newside, subst, varlist, id, pos, dir)
                   else 
                     ((*prerr_endline ("Filtering: " ^ 
                        Pp.pp_foterm side ^ " =(< || =)" ^ 
                        Pp.pp_foterm newside ^ " coming from " ^ 
                        Pp.pp_unit_clause uc );*)None)
                 else
                   Some (context newside, subst, varlist, id, pos, dir)
               with FoUnif.UnificationFailure _ -> None)
        (IDX.ClauseSet.elements cands)
    ;;

    let build_new_clause bag maxvar filter t subst vl id id2 pos dir =
      let maxvar, vl, relocsubst = Utils.relocate maxvar vl in
      let subst = Subst.concat relocsubst subst in
      let proof = Terms.Step(Terms.SuperpositionRight,id,id2,dir,pos,subst) in
      let t = Subst.apply_subst subst t in
      if filter t then
        let literal = 
          match t with
          | Terms.Node [ Terms.Leaf eq ; ty; l; r ] when B.eq B.eqP eq ->
               let o = Order.compare_terms l r in
               Terms.Equation (l, r, ty, o)
          | t -> Terms.Predicate t
        in
        let bag, uc = 
          Utils.add_to_bag bag (0, literal, vl, proof)
        in
        Some (bag, maxvar, uc)
      else
        ((*prerr_endline ("Filtering: " ^ Pp.pp_foterm t);*)None)
    ;;

    let fold_build_new_clause bag maxvar id filter res =
      let maxvar, bag, new_clauses = 
        List.fold_left
          (fun (maxvar, bag, acc) (t,subst,vl,id2,pos,dir) -> 
             match build_new_clause bag maxvar filter t subst vl id id2 pos dir
             with Some (bag, maxvar, uc) -> maxvar, bag, uc::acc
                | None -> maxvar, bag, acc)
          (maxvar, bag, []) res
      in
      bag, maxvar, new_clauses
    ;;

    let superposition_right_with_table bag maxvar (id,selected,vl,_) table =
      match selected with 
      | Terms.Predicate _ -> assert false
      | Terms.Equation (l,r,ty,Terms.Lt) ->
          fold_build_new_clause bag maxvar id (fun _ -> true)
            (all_positions [3] 
              (fun x -> Terms.Node [ Terms.Leaf B.eqP; ty; l; x ])
              r (superposition_right table vl))          
      | Terms.Equation (l,r,ty,Terms.Gt) ->
          fold_build_new_clause bag maxvar id (fun _ -> true)
            (all_positions [2] 
              (fun x -> Terms.Node [ Terms.Leaf B.eqP; ty; x; r ])
              l (superposition_right table vl))
      | Terms.Equation (l,r,ty,Terms.Incomparable) -> 
          fold_build_new_clause bag maxvar id
            (function (* Riazanov: p.33 condition (iv) *)
              | Terms.Node [Terms.Leaf eq; ty; l; r ] when B.eq B.eqP eq -> 
                  Order.compare_terms l r <> Terms.Eq
              | _ -> assert false)
            ((all_positions [3] 
               (fun x -> Terms.Node [ Terms.Leaf B.eqP; ty; l; x ])
               r (superposition_right table vl)) @         
             (all_positions [2] 
               (fun x -> Terms.Node [ Terms.Leaf B.eqP; ty; x; r ])
               l (superposition_right table vl)))
      | _ -> assert false
    ;;

    (* the current equation is normal w.r.t. demodulation with atable
     * (and is not the identity) *)
    let superposition_right bag maxvar current (alist,atable) = 
      let ctable = IDX.index_unit_clause IDX.DT.empty current in
      let bag, maxvar, new_clauses =
        List.fold_left 
          (fun (bag, maxvar, acc) active ->
             let bag, maxvar, newc = 
               superposition_right_with_table bag maxvar active ctable 
             in
             bag, maxvar, newc @ acc)
          (bag, maxvar, []) alist
      in
      let alist, atable = 
        current :: alist, IDX.index_unit_clause atable current
      in
      let fresh_current, maxvar = Utils.fresh_unit_clause maxvar current in
      let bag, maxvar, additional_new_clauses =
        superposition_right_with_table bag maxvar fresh_current atable 
      in
      bag, maxvar, (alist, atable), new_clauses @ additional_new_clauses
    ;;
          
  end