blob: 8723e53b0582aa6d02c601343ea2155540f1b74b [file] [log] [blame]
package opennlp.tools.parse_thicket;
import java.util.List;
import edu.stanford.nlp.trees.Tree;
public class ParseThicket {
// parse trees
private List<Tree> sentenceTrees;
// there should be an arc for each sentence
private List<WordWordInterSentenceRelationArc> arcs;
// lists of nodes for each sentence
// then list for all sentences
private List<List<ParseTreeNode>> sentenceNodes;
private List<Float> sentimentProfile;
private String origText;
private List<List<ParseTreeNode>> phrases;
public List<Tree> getSentenceTrees() {
return sentenceTrees;
}
public void setSentenceTrees(List<Tree> sentenceTrees) {
this.sentenceTrees = sentenceTrees;
}
public List<List<ParseTreeNode>> getSentenceNodes() {
return sentenceNodes;
}
public void setSentenceNodes(List<List<ParseTreeNode>> sentenceNodes) {
this.sentenceNodes = sentenceNodes;
}
public String getOrigText() {
return origText;
}
public void setOrigText(String origText) {
this.origText = origText;
}
public List<Tree> getSentences() {
return sentenceTrees;
}
public void setSentences(List<Tree> sentences) {
this.sentenceTrees = sentences;
}
public List<WordWordInterSentenceRelationArc> getArcs() {
return arcs;
}
public void setArcs(List<WordWordInterSentenceRelationArc> arcs) {
this.arcs = arcs;
}
public List<List<ParseTreeNode>> getNodesThicket() {
return sentenceNodes;
}
public void setNodesThicket(List<List<ParseTreeNode>> nodesThicket) {
this.sentenceNodes = nodesThicket;
}
public ParseThicket(String paragraph){
ParseCorefsBuilder builder = ParseCorefsBuilder.getInstance();
ParseThicket res = builder.buildParseThicket(paragraph);
this.sentenceTrees= res.sentenceTrees;
this.arcs = res.arcs;
}
public ParseThicket(List<Tree> ptTrees,
List<WordWordInterSentenceRelationArc> barcs) {
this.sentenceTrees= ptTrees;
this.arcs = barcs;
}
public String toString(){
return this.sentenceTrees+"\n"+this.arcs;
}
public void setPhrases(List<List<ParseTreeNode>> phrs) {
this.phrases = phrs;
}
public List<List<ParseTreeNode>> getPhrases() {
return phrases;
}
public List<Float> getSentimentProfile() {
return sentimentProfile;
}
public void setSentimentProfile(List<Float> sentimentProfile) {
this.sentimentProfile = sentimentProfile;
}
}