public IDEPTreeDatum getDEPTreeDatum(
IDEPTreeDatumFactory treeFactory, IDEPNodeDatumFactory nodeFactory) {
IDEPTreeDatum datum = treeFactory.createDEPTreeDatum();
List<IDEPNodeDatum> nodeData = Lists.newArrayList();
int i, size = size();
DEPNode node;
for (i = 1; i < size; i++) {
node = get(i);
nodeData.add(node.getDEPNodeDatum(nodeFactory));
}
datum.setDEPNodeData(nodeData);
return datum;
}
public static DEPTree buildFrom(IDEPTreeDatum treeDatum) {
List<IDEPNodeDatum> nodeData = treeDatum.getDEPNodeData();
DEPTree tree = new DEPTree();
int i, size = nodeData.size();
IDEPNodeDatum nd;
DEPNode node;
for (i = 0; i < size; i++) {
nd = nodeData.get(i);
tree.add(
new DEPNode(
nd.getID(),
nd.getForm(),
nd.getLemma(),
nd.getPOS(),
nd.getNamedEntity(),
new DEPFeat(nd.getFeats())));
}
for (i = 0; i < size; i++) {
nd = nodeData.get(i);
node = tree.get(i + 1);
node.initSHeads();
node.setHead(new DEPArc(tree, nd.getSyntacticHead()));
node.addSHeads(DEPLib.getSRLArcs(tree, nd.getSemanticHeads()));
}
tree.resetDependents();
return tree;
}