package arkref.data;
import java.io.Serializable;
import java.util.HashMap;
import java.util.Map;
/** build this up through the pipeline stages **/
public class RefGraph implements Serializable {
private static final long serialVersionUID = 2746461702540825024L;
// public Map<Mention, ArrayList<Mention>> refCandidates;
private Map<Mention, Mention> finalResolutions;
public RefGraph(){ finalResolutions=new HashMap<Mention, Mention>(); }
public void setNullRef(Mention m) {
finalResolutions.put(m, null);
}
public void setRef(Mention m, Mention m2) {
finalResolutions.put(m, m2);
}
public Map<Mention, Mention> getFinalResolutions() {
return finalResolutions;
}
public boolean needsReso(Mention m) {
return !finalResolutions.containsKey(m);
}
}