@Override public int compare(Entity entity1, Entity entity2) { switch (sortingBy) { case CONFIDENCE: return Double.compare(entity1.getConfidence(), entity2.getConfidence()); case TEXT: return entity1.getText().compareTo(entity2.getText()); case OCCURRENCE: return Integer.compare(entity1.getSpan().getTokenStart(), entity2.getSpan().getTokenStart()); } throw new IllegalArgumentException("Invalid sort type."); }
/** * Create a new entity from an existing entity. * @param entity An existing entity. */ public Entity(Entity entity) { setConfidence(entity.getConfidence()); setMetadata(entity.getMetadata()); setType(entity.getType()); setLanguageCode(entity.getLanguageCode()); setSpan(entity.getSpan()); setText(entity.getText()); setUri(entity.getUri()); }