ActivityCachedSim.java

package kb.howtokb.clustering;

import edu.stanford.nlp.util.Pair;
import kb.howtokb.utils.BijectiveMap;

public class ActivityCachedSim {

	private BijectiveMap<String, Pair<Integer, Integer>> ids;
	private ActivityComponentSim vSim, nSim;

	public ActivityCachedSim(BijectiveMap<String, Pair<Integer, Integer>> ids, ActivityComponentSim vSim,
			ActivityComponentSim nSim) {
		this.ids = ids;
		this.vSim = vSim;
		this.nSim = nSim;
	}

	private boolean sim(Pair<Integer, Integer> e1, Pair<Integer, Integer> e2) {
//		return
//		// v1 v2 are similar
//		vSim.simFromCache(e1.first, e2.first) &&
//		// n1 v2 are similar
//				nSim.simFromCache(e1.second, e2.second);
		
		if (!vSim.simFromCache(e1.first, e2.first) &&
		// n1 v2 are similar
				!nSim.simFromCache(e1.second, e2.second))
			return false;
		return true;
	}

	public boolean sim(String a1, String a2) {
		Pair<Integer, Integer> e1 = this.ids.getValueFromKey(a1);
		Pair<Integer, Integer> e2 = this.ids.getValueFromKey(a2);
		return sim(e1, e2);
	}
}