- final static void collapseSpeciesSpecificSubtrees( final Phylogeny phy ) {
- boolean inferred = false;
- for( final PhylogenyNodeIterator it = phy.iteratorPreorder(); it.hasNext(); ) {
- final PhylogenyNode n = it.next();
- if ( !n.isExternal() && !n.isCollapse() && ( n.getNumberOfDescendants() > 1 ) ) {
- final Set<Taxonomy> taxs = obtainDistinctTaxonomies( n );
- if ( ( taxs != null ) && ( taxs.size() == 1 ) ) {
- AptxUtil.collapseSubtree( n, true );
- if ( !n.getNodeData().isHasTaxonomy() ) {
- n.getNodeData().setTaxonomy( ( Taxonomy ) n.getAllExternalDescendants().get( 0 ).getNodeData()
- .getTaxonomy().copy() );
- }
- inferred = true;
- }
- else {
- n.setCollapse( false );
- }
- }
- }
- if ( inferred ) {
- phy.setRerootable( false );
- }
- }
-
- final static void collapseSubtree( final PhylogenyNode node, final boolean collapse ) {
- node.setCollapse( collapse );
- if ( node.isExternal() ) {
- return;
- }
- final PhylogenyNodeIterator it = new PreorderTreeIterator( node );
- while ( it.hasNext() ) {
- it.next().setCollapse( collapse );
- }
- }
-
- final static void colorPhylogenyAccordingToConfidenceValues( final Phylogeny tree, final TreePanel tree_panel ) {
- double max_conf = 0.0;
- for( final PhylogenyNodeIterator it = tree.iteratorPreorder(); it.hasNext(); ) {
- final PhylogenyNode n = it.next();
- n.getBranchData().setBranchColor( null );
- if ( n.getBranchData().isHasConfidences() ) {
- final double conf = PhylogenyMethods.getConfidenceValue( n );
- if ( conf > max_conf ) {
- max_conf = conf;
- }
- }
- }
- if ( max_conf > 0.0 ) {
- final Color bg = tree_panel.getTreeColorSet().getBackgroundColor();
- final Color br = tree_panel.getTreeColorSet().getBranchColor();
- for( final PhylogenyNodeIterator it = tree.iteratorPreorder(); it.hasNext(); ) {
- final PhylogenyNode n = it.next();
- if ( n.getBranchData().isHasConfidences() ) {
- final double conf = PhylogenyMethods.getConfidenceValue( n );
- final BranchColor c = new BranchColor( ForesterUtil.calcColor( conf, 0.0, max_conf, bg, br ) );
- colorizeSubtree( n, c );
- }
- }
- }
- }
-
- final static void colorPhylogenyAccordingToExternalTaxonomy( final Phylogeny tree, final TreePanel tree_panel ) {
- for( final PhylogenyNodeIterator it = tree.iteratorPreorder(); it.hasNext(); ) {
- it.next().getBranchData().setBranchColor( null );
- }
- for( final PhylogenyNodeIterator it = tree.iteratorPreorder(); it.hasNext(); ) {
- final PhylogenyNode n = it.next();
- if ( !n.getBranchData().isHasBranchColor() ) {
- final Taxonomy tax = PhylogenyMethods.getExternalDescendantsTaxonomy( n );
- if ( tax != null ) {
- n.getBranchData().setBranchColor( new BranchColor( tree_panel.calculateTaxonomyBasedColor( tax ) ) );
- final List<PhylogenyNode> descs = PhylogenyMethods.getAllDescendants( n );
- for( final PhylogenyNode desc : descs ) {
- desc.getBranchData()
- .setBranchColor( new BranchColor( tree_panel.calculateTaxonomyBasedColor( tax ) ) );
- }
- }
- }
- }
- }
-
- final static int colorPhylogenyAccordingToRanks( final Phylogeny tree, final String rank, final TreePanel tree_panel ) {
- final Map<String, Color> true_lineage_to_color_map = new HashMap<String, Color>();
- int colorizations = 0;
- for( final PhylogenyNodeIterator it = tree.iteratorPostorder(); it.hasNext(); ) {
- final PhylogenyNode n = it.next();
- if ( n.getNodeData().isHasTaxonomy()
- && ( !ForesterUtil.isEmpty( n.getNodeData().getTaxonomy().getScientificName() )
- || !ForesterUtil.isEmpty( n.getNodeData().getTaxonomy().getCommonName() ) || !ForesterUtil
- .isEmpty( n.getNodeData().getTaxonomy().getTaxonomyCode() ) ) ) {
- if ( !ForesterUtil.isEmpty( n.getNodeData().getTaxonomy().getRank() )
- && n.getNodeData().getTaxonomy().getRank().equalsIgnoreCase( rank ) ) {
- final BranchColor c = new BranchColor( tree_panel.calculateTaxonomyBasedColor( n.getNodeData()
- .getTaxonomy() ) );
- colorizeSubtree( n, c );
- ++colorizations;
- if ( !ForesterUtil.isEmpty( n.getNodeData().getTaxonomy().getScientificName() ) ) {
- true_lineage_to_color_map.put( n.getNodeData().getTaxonomy().getScientificName(), c.getValue() );
- }
- }
- }
- }
- for( final PhylogenyNodeIterator it = tree.iteratorPostorder(); it.hasNext(); ) {
- final PhylogenyNode node = it.next();
- if ( ( node.getBranchData().getBranchColor() == null ) && node.getNodeData().isHasTaxonomy()
- && !ForesterUtil.isEmpty( node.getNodeData().getTaxonomy().getLineage() ) ) {
- boolean success = false;
- if ( !true_lineage_to_color_map.isEmpty() ) {
- for( final String lin : node.getNodeData().getTaxonomy().getLineage() ) {
- if ( true_lineage_to_color_map.containsKey( lin ) ) {
- colorizeSubtree( node, new BranchColor( true_lineage_to_color_map.get( lin ) ) );
- ++colorizations;
- success = true;
- break;
- }
- }
- }
- if ( !success ) {
- final Map<String, String> lineage_to_rank_map = MainPanel.getLineageToRankMap();
- for( final String lin : node.getNodeData().getTaxonomy().getLineage() ) {
- final Taxonomy temp_tax = new Taxonomy();
- temp_tax.setScientificName( lin );
- if ( lineage_to_rank_map.containsKey( lin )
- && !ForesterUtil.isEmpty( lineage_to_rank_map.get( lin ) )
- && lineage_to_rank_map.get( lin ).equalsIgnoreCase( rank ) ) {
- final BranchColor c = new BranchColor( tree_panel.calculateTaxonomyBasedColor( temp_tax ) );
- colorizeSubtree( node, c );
- ++colorizations;
- true_lineage_to_color_map.put( lin, c.getValue() );
- break;
- }
- else {
- UniProtTaxonomy up = null;
- try {
- up = TaxonomyDataManager.obtainUniProtTaxonomy( temp_tax, null, null );
- }
- catch ( final Exception e ) {
- e.printStackTrace();
- }
- if ( ( up != null ) && !ForesterUtil.isEmpty( up.getRank() ) ) {
- lineage_to_rank_map.put( lin, up.getRank() );
- if ( up.getRank().equalsIgnoreCase( rank ) ) {
- final BranchColor c = new BranchColor( tree_panel.calculateTaxonomyBasedColor( temp_tax ) );
- colorizeSubtree( node, c );
- ++colorizations;
- true_lineage_to_color_map.put( lin, c.getValue() );
- break;
- }
- }
- }
- }
- }
- }
- }
- return colorizations;
- }
-
- final static String createAnnotationString( final SortedSet<Annotation> annotations ) {
- final SortedMap<String, List<Annotation>> m = new TreeMap<String, List<Annotation>>();
- for( final Annotation an : annotations ) {
- final String ref_source = ForesterUtil.isEmpty( an.getRefSource() ) ? "?" : an.getRefSource();
- if ( !m.containsKey( ref_source ) ) {
- m.put( ref_source, new ArrayList<Annotation>() );
- }
- m.get( ref_source ).add( an );
- }
- final StringBuilder sb = new StringBuilder();
- for( final Entry<String, List<Annotation>> e : m.entrySet() ) {
- final String ref_source = e.getKey();
- final List<Annotation> ans = e.getValue();
- if ( m.size() > 1 ) {
- sb.append( "[" );
- }
- if ( !ref_source.equals( "?" ) ) {
- sb.append( ref_source );
- sb.append( ": " );
- }
- for( int i = 0; i < ans.size(); ++i ) {
- final Annotation an = ans.get( i );
- if ( !ForesterUtil.isEmpty( an.getRefValue() ) ) {
- sb.append( an.getRefValue() );
- sb.append( " " );
- }
- if ( !ForesterUtil.isEmpty( an.getDesc() ) ) {
- sb.append( an.getDesc() );