com.articulate.sigma.KBcache Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sigma-component Show documentation
Show all versions of sigma-component Show documentation
Sigma knowledge engineering system is an system for developing, viewing and debugging theories in first
order logic. It works with Knowledge Interchange Format (KIF) and is optimized for the Suggested Upper Merged
Ontology (SUMO) www.ontologyportal.org.
The newest version!
/** This code is copyright Articulate Software (c) 2003. Some
portions copyright Teknowledge (c) 2003 and reused under the termsof the GNU
license. This software is released under the GNU Public License
. Users of this code also consent,
by use of this code, to credit Articulate Software and Teknowledge in any
writings, briefings, publications, presentations, or other representations
of any software which incorporates, builds on, or uses this code. Please
cite the following article in any publication with references:
Pease, A., (2003). The Sigma Ontology Development Environment, in Working
Notes of the IJCAI-2003 Workshop on Ontology and Distributed Systems,
August 9, Acapulco, Mexico. see also
http://sigmakee.sourceforge.net
Note that this class, and therefore, Sigma, depends upon several terms
being present in the ontology in order to function as intended. They are:
domain
domainSubclass
Entity
instance
Relation
subclass
subrelation
TransitiveRelation
*/
/*************************************************************************************************/
package com.articulate.sigma;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import java.io.*;
import java.util.*;
public class KBcache {
public KB kb = null;
private static boolean debug = false;
/** The String constant that is the suffix for files of cached assertions. */
public static final String _cacheFileSuffix = "_Cache.kif";
/** all the relations in the kb **/
public HashSet relations = new HashSet();
/** all the transitive relations in the kb **/
public HashSet transRels = new HashSet();
/** all the transitive relations between instances in the kb **/
public HashSet instTransRels = new HashSet();
/** All the cached "parent" relations of all transitive relations
* meaning the relations between all first arguments and the
* transitive closure of second arguments. The external HashMap
* pairs relation name String keys to values that are the parent
* relationships. The interior HashMap is the set of terms and
* their transitive closure of parents. */
public HashMap>> parents =
new HashMap>>();
/** Parent relations from instances, including those that are
* transitive through (instance,instance) relations, such as
* subAttribute and subrelation */
public HashMap> instances =
new HashMap>();
/** A temporary list of instances built during creation of the
children map, in order to efficiently create the instances map **/
// TODO: make private
public HashSet insts = new HashSet();
/** All the cached "child" relations of all transitive relations
* meaning the relations between all first arguments and the
* transitive closure of second arguments. The external HashMap
* pairs relation name String keys to values that are the child
* relationships. The interior HashMap is the set of terms and
* their transitive closure of children. */
public HashMap>> children =
new HashMap>>();
/** Relation name keys and argument types with 0th arg always "".
Variable arity relations may have a type for the last argument,
which will be the type repeated for all extended arguments.
Note that types can be functions, rather than just terms. **/
public HashMap> signatures =
new HashMap>();
/** The number of arguments to each relation. Variable arity is -1 **/
public HashMap valences = new HashMap();
/** Disjoint relations which were explicitly defined in "partition", "disjoint",
"disjointDecomposition" and "exhaustiveDecomposition" expressions **/
public HashMap> explicitDisjointRelations = new HashMap<>();
/** ***************************************************************
* empty constructor for testing only
*/
public KBcache() {
}
/** ***************************************************************
*/
public KBcache(KB kb) {
this.kb = kb;
}
/** ***************************************************************
*/
public KBcache(KBcache kbCacheIn, KB kbIn) {
this.kb = kbIn;
if (kbCacheIn.relations != null) {
this.relations = Sets.newHashSet(kbCacheIn.relations);
}
if (kbCacheIn.transRels != null) {
this.transRels = Sets.newHashSet(kbCacheIn.transRels);
}
if (kbCacheIn.instTransRels != null) {
this.instTransRels = Sets.newHashSet(kbCacheIn.instTransRels);
}
if (kbCacheIn.parents != null) {
for (Map.Entry>> outerEntry : kbCacheIn.parents.entrySet()) {
String outerKey = outerEntry.getKey();
HashMap> newInnerMap = Maps.newHashMap();
HashMap> oldInnerMap = outerEntry.getValue();
for (Map.Entry> innerEntry : oldInnerMap.entrySet()) {
String innerKey = innerEntry.getKey();
HashSet newInnerSet = Sets.newHashSet(innerEntry.getValue());
newInnerMap.put(innerKey, newInnerSet);
}
this.parents.put(outerKey, newInnerMap);
}
}
if (kbCacheIn.instances != null) {
for (Map.Entry> entry : kbCacheIn.instances.entrySet()) {
String key = entry.getKey();
HashSet newSet = Sets.newHashSet(entry.getValue());
this.instances.put(key, newSet);
}
}
if (kbCacheIn.insts != null) {
this.insts = Sets.newHashSet(kbCacheIn.insts);
}
if (kbCacheIn.children != null) {
for (Map.Entry>> outerEntry : kbCacheIn.children.entrySet()) {
String outerKey = outerEntry.getKey();
HashMap> newInnerMap = Maps.newHashMap();
HashMap> oldInnerMap = outerEntry.getValue();
for (Map.Entry> innerEntry : oldInnerMap.entrySet()) {
String innerKey = innerEntry.getKey();
HashSet newInnerSet = Sets.newHashSet(innerEntry.getValue());
newInnerMap.put(innerKey, newInnerSet);
}
this.children.put(outerKey, newInnerMap);
}
}
if (kbCacheIn.signatures != null) {
for (Map.Entry> entry : kbCacheIn.signatures.entrySet()) {
String key = entry.getKey();
ArrayList newSet = Lists.newArrayList(entry.getValue());
this.signatures.put(key, newSet);
}
}
if (kbCacheIn.valences != null) {
this.valences = Maps.newHashMap(kbCacheIn.valences);
}
if (kbCacheIn.explicitDisjointRelations != null) {
for (Map.Entry> entry : kbCacheIn.explicitDisjointRelations.entrySet()) {
String key = entry.getKey();
HashSet newSet = Sets.newHashSet(entry.getValue());
this.explicitDisjointRelations.put(key, newSet);
}
}
}
/** ***************************************************************
* An ArrayList utility method
*/
private void arrayListReplace(ArrayList al, int index, String newEl) {
if (index > al.size()) {
System.out.println("Error in KBcache.arrayListReplace(): index " + index +
" out of bounds.");
return;
}
al.remove(index);
al.add(index,newEl);
}
/** ***************************************************************
* Find whether the given child has the given parent for the given
* transitive relation. Return false if they are equal
*/
public boolean childOfP(String rel, String parent, String child) {
if (parent.equals(child)) {
return false;
}
HashMap> childMap = children.get(rel);
HashSet childSet = (childMap != null) ? childMap.get(parent) : null;
if (childSet == null) {
if (debug) System.out.println("INFO in KBcache.childOfP(): null childset for relation, parent, child: "
+ rel + " " + parent + " " + child);
return false;
}
if (childSet.contains(child))
return true;
else
return false;
}
/** *************************************************************
* Returns true if i is an instance of c, else returns false.
*
* @param i A String denoting an instance.
* @param c A String denoting a Class.
* @return true or false.
*/
public boolean isInstanceOf(String i, String c) {
if (instances.containsKey(i)) {
HashSet hashSet = instances.get(i);
if (hashSet.contains(c))
return true;
else
return false;
}
else
return false;
}
/** ***************************************************************
* Find whether the given instance has the given parent class.
* Include paths the have transitive relations between instances such
* as an Attribute that is a subAttribute of another instance, which
* in turn then is an instance of the given class.
* Return false if they are equal.
*/
public boolean transInstOf(String child, String parent) {
HashSet prents = instances.get(child);
if (prents != null)
return prents.contains(parent);
else
return false;
}
/** ***************************************************************
* Find whether the given class has the given parent class.
*/
public boolean subclassOf(String child, String parent) {
HashMap> prentsForRel = parents.get("subclass");
if (prentsForRel != null) {
HashSet prents = prentsForRel.get(child);
if (prents != null)
return prents.contains(parent);
else
return false;
}
return false;
}
/** ***************************************************************
* Find whether the given class is the subAttribute of the given parent class.
*/
public boolean subAttributeOf(String child, String parent) {
HashMap> prentsForRel = parents.get("subAttribute");
if (prentsForRel != null) {
HashSet prents = prentsForRel.get(child);
if (prents != null)
return prents.contains(parent);
else
return false;
}
return false;
}
/** ***************************************************************
* Record instances and their explicitly defined parent classes
*/
void buildDirectInstances() {
ArrayList forms = kb.ask("arg", 0, "instance");
for (int i = 0; i < forms.size(); i++) {
Formula f = forms.get(i);
String child = f.getArgument(1);
String parent = f.getArgument(2);
HashMap> superclasses = parents.get("subclass");
HashSet iset = new HashSet();
if (instances.get(child) != null)
iset = instances.get(child);
iset.add(parent);
if (superclasses != null && superclasses.get(parent) != null)
iset.addAll(superclasses.get(parent));
instances.put(child, iset);
}
}
/** ***************************************************************
* build a disjoint-relations-map which were explicitly defined in
* "partition", "exhaustiveDecomposition", "disjointDecomposition"
* and "disjoint" expressions;
*/
public void buildDisjointRelationsMap() {
ArrayList explicitDisjontFormulae = new ArrayList();
explicitDisjontFormulae.addAll(kb.ask("arg", 0, "partition"));
explicitDisjontFormulae.addAll(kb.ask("arg", 0, "disjoint"));
explicitDisjontFormulae.addAll(kb.ask("arg", 0, "disjointDecomposition"));
explicitDisjontFormulae.addAll(kb.ask("arg", 0, "exhaustiveDecomposition"));
for (Formula f : explicitDisjontFormulae) {
ArrayList arguments = f.argumentsToArrayList(0);
if (arguments != null && !arguments.isEmpty()) {
int i = 2;
if (f.getArgument(0).equals("disjoint")) {
i = 1;
}
for ( ; i < arguments.size(); i++) {
String key = arguments.get(i);
int j = 2;
if (f.getArgument(0).equals("disjoint"))
j = 1;
for ( ; j < arguments.size(); j++) {
if (j != i) {
String val = arguments.get(j);
if (!explicitDisjointRelations.containsKey(key)) {
HashSet vals = new HashSet();
vals.add(val);
explicitDisjointRelations.put(key, vals);
}
else {
HashSet vals = explicitDisjointRelations.get(key);
vals.add(val);
explicitDisjointRelations.put(key, vals);
}
}
}
}
}
}
}
/** ***************************************************************
* check if there are any two types in typeSet are disjoint or not;
*/
public static boolean checkDisjoint(KB kb, HashSet typeSet) {
ArrayList typeList = new ArrayList<>(typeSet);
int size = typeList.size();
for (int i = 0; i < size; i++) {
String rel1 = typeList.get(i);
for (int j = i+1; j < size; j++) {
String rel2 = typeList.get(j);
if (checkDisjoint(kb, rel1, rel2) == true)
return true;
}
}
return false;
}
/** ***************************************************************
* check if rel1 and rel2 are disjoint
* return true if rel1 and rel2 are disjoint; otherwise return false.
*/
public static boolean checkDisjoint(KB kb, String rel1, String rel2) {
HashSet ancestors_rel1 = kb.kbCache.getParentClasses(rel1);
HashSet ancestors_rel2 = kb.kbCache.getParentClasses(rel2);
if (ancestors_rel1 == null || ancestors_rel2 == null)
return false;
ancestors_rel1.add(rel1);
ancestors_rel2.add(rel2);
for (String s1 : ancestors_rel1) {
for (String s2 : ancestors_rel2) {
if (kb.kbCache.isExplicitDisjoint(kb.kbCache.explicitDisjointRelations, s1, s2)) {
if (debug)
System.out.println(rel1 + " and " + rel2 +
" are disjoint relations, because of " + s1 + " and " + s2);
return true;
}
}
}
return false;
}
/** ***************************************************************
* return true if rel1 and rel2 are explicitly defined as disjoint
* relations; otherwise return false.
*/
public boolean isExplicitDisjoint(HashMap> explicitDisjointRelations,
String rel1, String rel2) {
if (explicitDisjointRelations.containsKey(rel1)) {
return explicitDisjointRelations.get(rel1).contains(rel2);
}
else if (explicitDisjointRelations.containsKey(rel2)) {
return explicitDisjointRelations.get(rel2).contains(rel1);
}
else
return false;
}
/** ***************************************************************
* Cache whether a given instance has a given parent class.
* Include paths the have transitive relations between instances such
* as an Attribute that is a subAttribute of another instance, which
* in turn then is an instance of the given class.
* TODO: make sure that direct instances are recorded too
*/
void buildTransInstOf() {
Iterator titer = insts.iterator(); // Iterate through the temporary list of instances built
// during creation of the @see children map
while (titer.hasNext()) {
String child = titer.next();
ArrayList forms = kb.ask("arg",1,child);
for (int i = 0; i < forms.size(); i++) {
Formula f = forms.get(i);
String rel = f.getArgument(0);
if (instTransRels.contains(rel) && !rel.equals("subclass")) {
HashMap> prentList = parents.get(rel);
if (prentList != null) {
HashSet prents = prentList.get(f.getArgument(1)); // include all parents of the child
if (prents != null) {
Iterator it = prents.iterator();
while (it.hasNext()) {
String p = it.next();
ArrayList forms2 = kb.askWithRestriction(0,"instance",1,p);
for (int j = 0; j < forms2.size(); j++) {
Formula f2 = forms2.get(j);
String cl = f2.getArgument(2);
HashMap> superclasses = parents.get("subclass");
HashSet pset = new HashSet();
if (instances.get(child) != null)
pset = instances.get(child);
pset.add(cl);
pset.addAll(superclasses.get(cl));
instances.put(child, pset);
}
}
}
}
}
else if (rel.equals("instance")) {
if (child.equals("exhaustiveAttribute"))
System.out.println("INFO in KBcache.buildTransInstOf(): f: " + f);
String cl = f.getArgument(2);
HashMap> superclasses = parents.get("subclass");
HashSet iset = new HashSet();
if (instances.get(child) != null)
iset = instances.get(child);
iset.add(cl);
iset.addAll(superclasses.get(cl));
instances.put(child, iset);
}
}
}
buildDirectInstances();
}
/** ***************************************************************
* return parent classes for the given cl from subclass expressions.
*/
public HashSet getParentClasses(String cl) {
HashMap> ps = parents.get("subclass");
if (ps != null)
return ps.get(cl);
else
return null;
}
/** ***************************************************************
* return child classes for the given cl from subclass expressions.
*/
public HashSet getChildClasses(String cl) {
HashMap> ps = children.get("subclass");
if (ps != null)
return ps.get(cl);
else
return null;
}
/** ***************************************************************
* return classes for the given instance cl.
*
* For example, if we know (instance UnitedStates Nation), then
* getParentClassesOfInstances(UnitedStates) returns Nation and its
* super claasses from subclass expressions.
*/
public HashSet getParentClassesOfInstance(String cl) {
HashSet ps = instances.get(cl);
if (ps != null)
return ps;
else
return new HashSet();
}
/** ***************************************************************
* Get all instances for the given input class
*
* For example, given the class "Nation", getInstancesForType(Nation)
* returns all instances, like "America", "Austria", "Albania", etc.
*/
public HashSet getInstancesForType(String cl) {
HashSet instancesForType = new HashSet<>();
for (String inst : instances.keySet()) {
HashSet parents = instances.get(inst);
if (parents.contains(cl))
instancesForType.add(inst);
}
return instancesForType;
}
/** ***************************************************************
* Get the HashSet of the given arguments from an ArrayList of Formulas.
*/
public static HashSet collectArgFromFormulas(int arg, ArrayList forms) {
HashSet subs = new HashSet();
for (int i = 0; i < forms.size(); i++) {
Formula f = forms.get(i);
String sub = f.getArgument(arg);
subs.add(sub);
}
return subs;
}
/** ***************************************************************
* Do a proper search for relations (including Functions), utilizing
* the formal definitions, rather than the convention of initial
* lower case letter. This means getting any instance of Relation
* tracing back through subclasses as well.
*/
public void buildTransitiveRelationsSet() {
HashSet rels = new HashSet();
rels.add("TransitiveRelation");
while (!rels.isEmpty()) {
HashSet relSubs = new HashSet();
Iterator it = rels.iterator();
while (it.hasNext()){
String rel = it.next();
relSubs = new HashSet();
ArrayList forms = kb.askWithRestriction(0, "subclass", 2, rel);
if (forms != null) {
if (debug) System.out.println("INFO in KBcache.buildTransitiveRelationsSet(): subclasses: " + forms);
relSubs.addAll(collectArgFromFormulas(1,forms));
}
else
if (debug) System.out.println("INFO in KBcache.buildTransitiveRelationsSet(): no subclasses for : " + rels);
forms = kb.askWithRestriction(0,"instance",2,rel);
if (forms != null)
transRels.addAll(collectArgFromFormulas(1,forms));
forms = kb.askWithRestriction(0,"subrelation",2,rel);
if (forms != null)
transRels.addAll(collectArgFromFormulas(1,forms));
}
rels = new HashSet();
rels.addAll(relSubs);
}
}
/** ***************************************************************
* Do a proper search for relations (including Functions), utilizing
* the formal definitions, rather than the convention of initial
* lower case letter. This means getting any instance of Relation
* tracing back through subclasses as well.
*/
public void buildRelationsSet() {
HashSet rels = new HashSet();
rels.add("Relation");
while (!rels.isEmpty()) {
HashSet relSubs = new HashSet();
Iterator it = rels.iterator();
while (it.hasNext()) {
String rel = it.next();
ArrayList forms = kb.askWithRestriction(0,"subclass",2,rel);
if (forms != null)
relSubs.addAll(collectArgFromFormulas(1,forms));
forms = kb.askWithRestriction(0,"instance",2,rel);
if (forms != null) {
relations.addAll(collectArgFromFormulas(1,forms));
relSubs.addAll(collectArgFromFormulas(1,forms));
}
forms = kb.askWithRestriction(0,"subrelation",2,rel);
if (forms != null) {
relations.addAll(collectArgFromFormulas(1,forms));
relSubs.addAll(collectArgFromFormulas(1,forms));
}
}
rels = new HashSet();
rels.addAll(relSubs);
}
}
/** ***************************************************************
* Find the parent "roots" of any transitive relation - terms that
* appear only as argument 2
*/
private HashSet findRoots(String rel) {
HashSet result = new HashSet();
ArrayList forms = kb.ask("arg",0,rel);
HashSet arg1s = collectArgFromFormulas(1,forms);
HashSet arg2s = collectArgFromFormulas(2,forms);
arg2s.removeAll(arg1s);
result.addAll(arg2s);
return result;
}
/** ***************************************************************
* Find the child "roots" of any transitive relation - terms that
* appear only as argument 1
*/
private HashSet findLeaves(String rel) {
HashSet result = new HashSet();
ArrayList forms = kb.ask("arg",0,rel);
HashSet arg1s = collectArgFromFormulas(1,forms);
HashSet arg2s = collectArgFromFormulas(2,forms);
arg1s.removeAll(arg2s);
result.addAll(arg1s);
return result;
}
/** ***************************************************************
* Build "parent" relations based on breadth first search algorithm.
*/
private void breadthFirstBuildParents(String root, String rel) {
HashMap> relParents = parents.get(rel);
if (relParents == null) {
System.out.println("Error in KBcache.breadthFirstBuildParents(): no relation " + rel);
return;
}
int threshold = 10; // maximum time that a term can be traversed in breadthFirstBuildParents()
HashMap appearanceCount = new HashMap<>(); // for each term, we count how many times it has been traversed
ArrayDeque Q = new ArrayDeque();
Q.add(root);
while (!Q.isEmpty()) {
String t = Q.remove();
//System.out.println("visiting " + t);
ArrayList forms = kb.askWithRestriction(0,rel,2,t);
if (forms != null) {
HashSet relSubs = collectArgFromFormulas(1,forms);
Iterator it = relSubs.iterator();
while (it.hasNext()) {
String newTerm = it.next();
HashSet newParents = new HashSet();
HashSet oldParents = relParents.get(t);
if (oldParents == null) {
oldParents = new HashSet();
relParents.put(t, oldParents);
}
newParents.addAll(oldParents);
newParents.add(t);
HashSet newTermParents = relParents.get(newTerm);
if (newTermParents != null)
newParents.addAll(newTermParents);
relParents.put(newTerm, newParents);
if (appearanceCount.get(newTerm) == null) {
appearanceCount.put(newTerm, 1);
Q.addFirst(newTerm);
}
else if (appearanceCount.get(newTerm) <= threshold) {
appearanceCount.put(newTerm, appearanceCount.get(newTerm)+1);
Q.addFirst(newTerm);
}
}
}
}
}
/** ***************************************************************
* Build "children" relations based on breadth first search algorithm.
*/
private void breadthFirstBuildChildren(String root, String rel) {
HashMap> relChildren = children.get(rel);
if (relChildren == null) {
System.out.println("Error in KBcache.breadthFirstBuildChildren(): no relation " + rel);
return;
}
if (debug) System.out.println("INFO in KBcache.breadthFirst(): trying relation " + rel);
ArrayDeque Q = new ArrayDeque();
HashSet V = new HashSet();
Q.add(root);
V.add(root);
while (!Q.isEmpty()) {
String t = Q.remove();
if (debug) System.out.println("visiting " + t);
ArrayList forms = kb.askWithRestriction(0,rel,1,t);
if (forms != null) {
HashSet relSubs = collectArgFromFormulas(2,forms);
if (debug) System.out.println("visiting subs of t: " + relSubs);
Iterator it = relSubs.iterator();
while (it.hasNext()) {
String newTerm = it.next();
HashSet newChildren = new HashSet();
HashSet oldChildren = relChildren.get(t);
if (oldChildren == null) {
oldChildren = new HashSet();
relChildren.put(t, oldChildren);
}
newChildren.addAll(oldChildren);
newChildren.add(t);
HashSet newTermChildren = relChildren.get(newTerm);
if (newTermChildren != null)
newChildren.addAll(newTermChildren);
relChildren.put(newTerm, newChildren);
if (!V.contains(newTerm)) {
V.add(newTerm);
Q.addFirst(newTerm);
}
}
}
}
insts.addAll(relChildren.keySet());
}
/** ***************************************************************
* For each transitive relation, find its transitive closure. If
* rel is transitive, and (rel A B) and (rel B C) then the entry for
* rel is a HashMap where the key A has value ArrayList of {B,C}.
*/
public void buildParents() {
Iterator it = transRels.iterator();
while (it.hasNext()) {
String rel = it.next();
HashMap> value = new HashMap>();
HashSet roots = findRoots(rel);
parents.put(rel, value);
Iterator it1 = roots.iterator();
while (it1.hasNext()) {
String root = it1.next();
breadthFirstBuildParents(root,rel);
}
}
}
/** ***************************************************************
* For each transitive relation, find its transitive closure. If
* rel is transitive, and (rel A B) and (rel B C) then the entry for
* rel is a HashMap where the key A has value ArrayList of {B,C}.
*/
public void buildChildren() {
Iterator it = transRels.iterator();
while (it.hasNext()) {
String rel = it.next();
HashMap> value = new HashMap>();
HashSet leaves = findLeaves(rel);
children.put(rel, value);
Iterator it1 = leaves.iterator();
while (it1.hasNext()) {
String root = it1.next();
breadthFirstBuildChildren(root, rel);
}
}
}
/** ***************************************************************
* Fill an array of String with the specified String up to but
* not including the index, starting from the 1st argument and
* ignoring the 0th argument.
*/
private static void fillArray(String st, String[] ar, int start, int end) {
for (int i = start; i < end; i++)
if (StringUtil.emptyString(ar[i]))
ar[i] = st;
}
/** ***************************************************************
* Fill an array of String with the specified String up to but
* not including the index, starting from the end of the array
*/
private static void fillArrayList(String st, ArrayList ar, int start, int end) {
for (int i = start; i < end; i++)
if (i > ar.size()-1 || StringUtil.emptyString(ar.get(i)))
ar.add(st);
}
/** ***************************************************************
* Build the argument type list for every relation. If the argument
* is a domain subclass, append a "+" to the argument type. If
* no domain is defined for the given relation and argument position,
* inherit it from the parent. If there is no argument type, send
* an error to the Sigma error list.
* Relation name keys and argument types with 0th arg always ""
* public HashMap> signatures =
* new HashMap>();
*
* TODO: Function range
* TODO: Variable arity relations, which requires a new predicate
* in SUMO that will define the type of a row variable
*/
public void collectDomains() {
Iterator it = relations.iterator();
while (it.hasNext()) {
String rel = it.next();
String[] domainArray = new String[Formula.MAX_PREDICATE_ARITY];
int maxIndex = 0;
domainArray[0] = "";
ArrayList forms = kb.askWithRestriction(0,"domain",1,rel);
if (forms != null) {
for (int i = 0; i < forms.size(); i++) {
Formula form = forms.get(i);
//System.out.println("INFO in KBcache.collectDomains(): form " + form);
int arg = Integer.valueOf(form.getArgument(2));
String type = form.getArgument(3);
domainArray[arg] = type;
if (arg > maxIndex)
maxIndex = arg;
}
}
forms = kb.askWithRestriction(0,"domainSubclass",1,rel);
if (forms != null) {
for (int i = 0; i < forms.size(); i++) {
Formula form = forms.get(i);
int arg = Integer.valueOf(form.getArgument(2));
String type = form.getArgument(3);
domainArray[arg] = type + "+";
if (arg > maxIndex)
maxIndex = arg;
}
}
fillArray("Entity",domainArray,1,maxIndex);
ArrayList domains = new ArrayList();
for (int i = 0; i <= maxIndex; i++)
domains.add(domainArray[i]);
signatures.put(rel,domains);
valences.put(rel, new Integer(maxIndex));
}
inheritDomains();
}
/** ***************************************************************
* Note that this routine forces child relations to have arguments
* that are the same or more specific than their parent relations.
*/
private void breadthFirstInheritDomains(String root) {
String rel = "subrelation";
HashMap> relParents = parents.get("subrelation");
if (relParents == null) {
System.out.println("Error in KBcache.breadthFirst(): no relation subrelation");
return;
}
ArrayDeque Q = new ArrayDeque();
HashSet V = new HashSet();
Q.add(root);
V.add(root);
while (!Q.isEmpty()) {
String t = Q.remove();
ArrayList tdomains = signatures.get(t);
ArrayList forms = kb.askWithRestriction(0,rel,2,t);
if (forms != null) {
HashSet relSubs = collectArgFromFormulas(1,forms);
Iterator it = relSubs.iterator();
while (it.hasNext()) {
String newTerm = it.next();
ArrayList newDomains = signatures.get(newTerm);
if (valences.get(newTerm) == null || valences.get(newTerm) < valences.get(t)) {
fillArrayList("Entity",newDomains,valences.get(newTerm)+1,valences.get(t)+1);
valences.put(newTerm, valences.get(t));
}
for (int i = 1; i < valences.get(t); i++) {
String childArgType = newDomains.get(i);
String parentArgType = tdomains.get(i);
// If child-relation does not have definition of argument-type, we use parent-relation's argument-type
// TODO: if parent-relation does not have definition of argument-type, we continue to find its parent until we find the definition of argument-type
if (kb.askWithTwoRestrictions(0, "domain", 1, newTerm, 3, childArgType).isEmpty()) {
arrayListReplace(newDomains,i,parentArgType);
}
}
if (!V.contains(newTerm)) {
V.add(newTerm);
Q.addFirst(newTerm);
}
}
}
}
}
/** *************************************************************
* Delete and writes the cache .kif file then call addConstituent() so
* that the file can be processed and loaded by the inference engine.
*/
public void writeCacheFile() {
FileWriter fw = null;
try {
File dir = new File(KBmanager.getMgr().getPref("kbDir"));
File f = new File(dir, (kb.name + _cacheFileSuffix));
if (f.exists())
f.delete();
String filename = f.getCanonicalPath();
fw = new FileWriter(f, true);
Iterator it = parents.keySet().iterator();
while (it.hasNext()) {
String rel = it.next();
HashMap> valSet = parents.get(rel);
Iterator it2 = valSet.keySet().iterator();
while (it2.hasNext()) {
String child = it2.next();
HashSet prents = valSet.get(child);
Iterator it3 = prents.iterator();
while (it3.hasNext()) {
String parent = it3.next();
String tuple = "(" + rel + " " + child + " " + parent + ")";
if (!kb.formulaMap.containsKey(tuple)) {
fw.write(tuple);
fw.write(System.getProperty("line.separator"));
}
}
}
}
if (fw != null) {
fw.close();
fw = null;
}
kb.constituents.remove(filename);
kb.addConstituent(filename);
//kb.addConstituent(filename, false, false, false);
//KBmanager.getMgr().writeConfiguration();
}
catch (Exception ex) {
ex.printStackTrace();
}
finally {
try {
if (fw != null)
fw.close();
}
catch (Exception ex) {
ex.printStackTrace();
}
}
}
/** ***************************************************************
* Find domain and domainSubclass definitions that impact a child
* relation. If the type of an argument is less specific than
* the same type of a parent's argument, use that of the parent.
*/
public void inheritDomains() {
HashSet roots = findRoots("subrelation");
Iterator it = roots.iterator();
while (it.hasNext()) {
String root = it.next();
breadthFirstInheritDomains(root);
}
}
/** ***************************************************************
* Compile the set of transitive relations that are between instances
*/
public void buildInstTransRels() {
Iterator it = transRels.iterator();
while (it.hasNext()) {
String rel = it.next();
boolean instrel = true;
ArrayList sig = signatures.get(rel);
if (sig == null) {
System.out.println("Error in KBcache.buildInstTransRels(): Error " + rel + " not found.");
}
else {
for (int i = 0; i < sig.size(); i++) {
if (sig.get(i).endsWith("+")) {
instrel = false;
break;
}
}
if (instrel)
instTransRels.add(rel);
}
}
}
/** ***************************************************************
* Main entry point for the class.
*/
public void buildCaches() {
buildRelationsSet();
buildTransitiveRelationsSet();
buildParents();
buildChildren(); // note that buildTransInstOf() depends on this
collectDomains(); // note that buildInstTransRels() depends on this
buildInstTransRels();
buildDirectInstances();
buildDisjointRelationsMap(); // find relations under partition definition
System.out.println("INFO in KBcache.buildCaches(): size: " + instances.keySet().size());
}
/** *************************************************************
*/
public void showState() {
System.out.println("-------------- relations ----------------");
Iterator it = this.relations.iterator();
while (it.hasNext())
System.out.print(it.next() + " ");
System.out.println();
//nkbc.buildTransitiveRelationsSet();
System.out.println("-------------- transitives ----------------");
it = this.transRels.iterator();
while (it.hasNext())
System.out.print(it.next() + " ");
System.out.println();
System.out.println("-------------- parents ----------------");
//nkbc.buildParents();
it = this.parents.keySet().iterator();
while (it.hasNext()) {
String rel = it.next();
System.out.println("Relation: " + rel);
HashMap> relmap = this.parents.get(rel);
Iterator it2 = relmap.keySet().iterator();
while (it2.hasNext()) {
String term = it2.next();
System.out.println(term + ": " + relmap.get(term));
}
System.out.println();
}
System.out.println();
System.out.println("-------------- children ----------------");
//nkbc.buildChildren();
it = this.children.keySet().iterator();
while (it.hasNext()) {
String rel = it.next();
System.out.println("Relation: " + rel);
HashMap> relmap = this.children.get(rel);
Iterator it2 = relmap.keySet().iterator();
while (it2.hasNext()) {
String term = it2.next();
System.out.println(term + ": " + relmap.get(term));
}
System.out.println();
}
System.out.println();
System.out.println("-------------- domains ----------------");
//nkbc.collectDomains();
Iterator it3 = this.relations.iterator();
while (it3.hasNext()) {
String rel = it3.next();
ArrayList domains = this.signatures.get(rel);
System.out.println(rel + ": " + domains);
}
System.out.println();
System.out.println("-------------- valences ----------------");
Iterator it4 = this.valences.keySet().iterator();
while (it4.hasNext()) {
String rel = it4.next();
Integer arity = this.valences.get(rel);
System.out.println(rel + ": " + arity);
}
System.out.println();
System.out.println("-------------- insts ----------------");
Iterator it5 = this.insts.iterator();
while (it5.hasNext()) {
String inst = it5.next();
System.out.print(inst + ", ");
}
System.out.println();
System.out.println();
System.out.println("-------------- instances ----------------");
Iterator it6 = this.instances.keySet().iterator();
while (it6.hasNext()) {
String inst = it6.next();
System.out.println(inst + ": " + this.instances.get(inst));
}
}
/** *************************************************************
*/
public static void main(String[] args) {
KBmanager.getMgr().initializeOnce();
KB kb = KBmanager.getMgr().getKB("SUMO");
System.out.println("**** Finished loading KB ***");
KBcache nkbc = kb.kbCache;
//nkbc.buildCaches();
//nkbc.buildRelationsSet();
nkbc.showState();
}
}