final boolean most_parsimonious_duplication_model,
final boolean strip_gene_tree,
final boolean strip_species_tree ) throws SDIException {
+ this( gene_tree, species_tree, most_parsimonious_duplication_model, strip_gene_tree, strip_species_tree, true );
+ }
+
+ public GSDI( final Phylogeny gene_tree,
+ final Phylogeny species_tree,
+ final boolean most_parsimonious_duplication_model,
+ final boolean strip_gene_tree,
+ final boolean strip_species_tree,
+ final boolean transfer_taxonomy ) throws SDIException {
_most_parsimonious_duplication_model = most_parsimonious_duplication_model;
if ( gene_tree.getRoot().getNumberOfDescendants() == 3 ) {
gene_tree.reRoot( gene_tree.getRoot().getChildNode( 2 ) );
_tax_comp_base = nodes_linking_result.getTaxCompBase();
PhylogenyMethods.preOrderReId( species_tree );
final GSDIsummaryResult gsdi_summary_result = geneTreePostOrderTraversal( gene_tree,
- _most_parsimonious_duplication_model );
+ _most_parsimonious_duplication_model,
+ transfer_taxonomy );
_speciation_or_duplication_events_sum = gsdi_summary_result.getSpeciationOrDuplicationEventsSum();
_speciations_sum = gsdi_summary_result.getSpeciationsSum();
_duplications_sum = gsdi_summary_result.getDuplicationsSum();
* Preconditions: Mapping M for external nodes must have been calculated and
* the species tree must be labeled in preorder.
* <p>
- * @return
- * @throws SDIException
- *
+ * @param transfer_taxonomy
+ * @return
+ * @throws SDIException
+ *
*/
final static GSDIsummaryResult geneTreePostOrderTraversal( final Phylogeny gene_tree,
- final boolean most_parsimonious_duplication_model )
- throws SDIException {
+ final boolean most_parsimonious_duplication_model,
+ final boolean transfer_taxonomy ) throws SDIException {
final GSDIsummaryResult res = new GSDIsummaryResult();
for( final PhylogenyNodeIterator it = gene_tree.iteratorPostorder(); it.hasNext(); ) {
final PhylogenyNode g = it.next();
if ( g.isInternal() ) {
if ( g.getNumberOfDescendants() != 2 ) {
throw new SDIException( "gene tree contains internal node with " + g.getNumberOfDescendants()
- + " descendents" );
+ + " descendents" );
}
PhylogenyNode s1 = g.getChildNode1().getLink();
PhylogenyNode s2 = g.getChildNode2().getLink();
g.setLink( s1 );
determineEvent( s1, g, most_parsimonious_duplication_model, res );
}
+ if ( transfer_taxonomy ) {
+ transferTaxonomy( g );
+ }
}
return res;
}
if ( g.isInternal() ) {
if ( g.getNumberOfDescendants() != 2 ) {
throw new SDIException( "gene tree contains internal node with " + g.getNumberOfDescendants()
- + " descendents" );
+ + " descendents" );
}
PhylogenyNode s1 = g.getChildNode1().getLink();
PhylogenyNode s2 = g.getChildNode2().getLink();
* This allows for linking of internal nodes of the species tree (as opposed
* to just external nodes, as in the method it overrides.
* If TaxonomyComparisonBase is null, it will try to determine it.
- * @throws SDIException
- *
+ * @throws SDIException
+ *
*/
final static NodesLinkingResult linkNodesOfG( final Phylogeny gene_tree,
final Phylogeny species_tree,
}
else {
throw new SDIException( "taxonomy \"" + g.getNodeData().getTaxonomy()
- + "\" not present in species tree" );
+ + "\" not present in species tree" );
}
}
else {
return res;
}
+ static final void transferTaxonomy( final PhylogenyNode g ) {
+ if ( g == null ) {
+ throw new IllegalArgumentException( "gene tree node is null" );
+ }
+ final PhylogenyNode s = g.getLink();
+ if ( s == null ) {
+ throw new IllegalArgumentException( "mapped species tree node is null" );
+ }
+ if ( s.getNodeData().isHasTaxonomy() ) {
+ g.getNodeData().setTaxonomy( s.getNodeData().getTaxonomy() );
+ if ( g.isInternal() ) {
+ if ( g.getChildNode1().isInternal() && g.getChildNode1().getNodeData().isHasTaxonomy()
+ && ( g.getChildNode1().getNodeData().getTaxonomy() == s.getNodeData().getTaxonomy() ) ) {
+ g.getChildNode1().getNodeData().setTaxonomy( null );
+ }
+ if ( g.getChildNode2().isInternal() && g.getChildNode2().getNodeData().isHasTaxonomy()
+ && ( g.getChildNode2().getNodeData().getTaxonomy() == s.getNodeData().getTaxonomy() ) ) {
+ g.getChildNode2().getNodeData().setTaxonomy( null );
+ }
+ }
+ }
+ else if ( ForesterUtil.isEmpty( g.getName() ) && !ForesterUtil.isEmpty( s.getName() ) ) {
+ g.setName( s.getName() );
+ if ( g.isInternal() ) {
+ if ( g.getChildNode1().isInternal() && ( g.getChildNode1().getName() == s.getName() ) ) {
+ g.getChildNode1().setName( "" );
+ }
+ if ( g.getChildNode2().isInternal() && ( g.getChildNode2().getName() == s.getName() ) ) {
+ g.getChildNode2().setName( "" );
+ }
+ }
+ }
+ }
+
private final static void addScientificNamesMappedToReducedSpecificity( final String s1,
final String s2,
final SortedSet<String> scientific_names_mapped_to_reduced_specificity ) {
final Set<PhylogenyNode> set = new HashSet<PhylogenyNode>();
for( PhylogenyNode n : g.getChildNode1().getAllExternalDescendants() ) {
n = n.getLink();
- while ( n.getParent() != s ) {
+ while ( ( n.getParent() != s ) && ( n.getParent() != null ) ) {
n = n.getParent();
if ( n.isRoot() ) {
break;
boolean multiple = false;
for( PhylogenyNode n : g.getChildNode2().getAllExternalDescendants() ) {
n = n.getLink();
- while ( n.getParent() != s ) {
+ while ( ( n.getParent() != s ) && ( n.getParent() != null ) ) {
n = n.getParent();
if ( n.isRoot() ) {
break;