/** * returns all SofaTextMention objects with type "problem" in this SofaText object * * @return */ public Collection<? extends SofaTextMention> getProblems() { List<SofaTextMention> problems = new ArrayList<SofaTextMention>(); for (SofaTextMention stm : sofaTextMention) { if (stm.getMentionType().equals("problem")) problems.add(stm); } return problems; }
/** * returns all SofaTextMention objects with type "treatment" in this SofaText object * * @return */ public Collection<? extends SofaTextMention> getTreatments() { List<SofaTextMention> treatments = new ArrayList<SofaTextMention>(); for (SofaTextMention stm : sofaTextMention) { if (stm.getMentionType().equals("treatment")) treatments.add(stm); } return treatments; }
/** * Helper method to generate HTML, will be moved to controller in future, the html contains the * text of the SofaText object with each mention wrapped in a span tag * * @return */ public String getAnnotatedHTML() { String html = new String(text); String tagged; for (SofaTextMention m : sofaTextMention) { tagged = wrapInMentionTypeTag(m.getMentionText(), m.getMentionType()); // System.out.println(tagged); if (!m.getSofaTextMentionConcept().isEmpty()) tagged = wrapInConceptTag(tagged, m); html = html.replace(m.getMentionText(), tagged); // html = html.replaceAll("\\n", "<br/>"); } return html; }