X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=forester%2Fjava%2Fsrc%2Forg%2Fforester%2Farchaeopteryx%2FMainFrame.java;h=28398f7058c12962a52a05b0fe4d7836c5e053b4;hb=0b39293d1577083551ba528d0ac7afcacbf69dde;hp=9708688ee320230c525d9b18b199ab8d478a8d5e;hpb=fafb6815569eda661ee8dbd81adcd9e9e7e72c19;p=jalview.git diff --git a/forester/java/src/org/forester/archaeopteryx/MainFrame.java b/forester/java/src/org/forester/archaeopteryx/MainFrame.java index 9708688..28398f7 100644 --- a/forester/java/src/org/forester/archaeopteryx/MainFrame.java +++ b/forester/java/src/org/forester/archaeopteryx/MainFrame.java @@ -273,13 +273,12 @@ public abstract class MainFrame extends JFrame implements ActionListener { JMenuItem _phyloxml_ref_item; JMenuItem _aptx_ref_item; // - JFileChooser _writetopdf_filechooser; File _current_dir; + JFileChooser _writetopdf_filechooser; JFileChooser _save_filechooser; JFileChooser _writetographics_filechooser; // process menu: JMenu _process_menu; - // Handy pointers to child components: MainPanel _mainpanel; Container _contentpane; final LinkedList _textframes = new LinkedList(); ; @@ -293,15 +292,27 @@ public abstract class MainFrame extends JFrame implements ActionListener { MainFrame() { _process_pool = ProcessPool.createInstance(); _writetopdf_filechooser = new JFileChooser(); + _writetopdf_filechooser.setMultiSelectionEnabled( false ); + _writetopdf_filechooser.addChoosableFileFilter( pdffilter ); + _writetographics_filechooser = new JFileChooser(); + _writetographics_filechooser.setMultiSelectionEnabled( false ); + _writetographics_filechooser.addChoosableFileFilter( graphicsfilefilter ); _save_filechooser = new JFileChooser(); - _save_filechooser.setCurrentDirectory( new File( "." ) ); _save_filechooser.setMultiSelectionEnabled( false ); - _save_filechooser.setFileFilter( MainFrame.xmlfilter ); + _save_filechooser.setFileFilter( xmlfilter ); _save_filechooser.addChoosableFileFilter( nhfilter ); _save_filechooser.addChoosableFileFilter( nexusfilter ); _save_filechooser.addChoosableFileFilter( _save_filechooser.getAcceptAllFileFilter() ); - _writetographics_filechooser = new JFileChooser(); - _writetographics_filechooser.addChoosableFileFilter( MainFrame.graphicsfilefilter ); + try { + final String home_dir = System.getProperty( "user.home" ); + _save_filechooser.setCurrentDirectory( new File( home_dir ) ); + _writetopdf_filechooser.setCurrentDirectory( new File( home_dir ) ); + _writetographics_filechooser.setCurrentDirectory( new File( home_dir ) ); + } + catch ( final Exception e ) { + e.printStackTrace(); + // Do nothing. Not important. + } } /** @@ -611,34 +622,92 @@ public abstract class MainFrame extends JFrame implements ActionListener { } } else if ( o == _write_to_pdf_item ) { - writeToPdf( _mainpanel.getCurrentPhylogeny() ); + final File curr_dir = writeToPdf( _mainpanel.getCurrentPhylogeny(), + getMainPanel(), + _writetopdf_filechooser, + _current_dir, + getContentPane(), + this ); + if ( curr_dir != null ) { + setCurrentDir( curr_dir ); + } } else if ( o == _save_all_item ) { writeAllToFile(); } else if ( o == _write_to_jpg_item ) { - writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), GraphicsExportType.JPG ); + final File new_dir = writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), + GraphicsExportType.JPG, + _mainpanel, + _writetographics_filechooser, + this, + getContentPane(), + _current_dir ); + if ( new_dir != null ) { + setCurrentDir( new_dir ); + } } else if ( o == _write_to_gif_item ) { - writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), GraphicsExportType.GIF ); + final File new_dir = writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), + GraphicsExportType.GIF, + _mainpanel, + _writetographics_filechooser, + this, + getContentPane(), + _current_dir ); + if ( new_dir != null ) { + setCurrentDir( new_dir ); + } } else if ( o == _write_to_tif_item ) { - writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), GraphicsExportType.TIFF ); + final File new_dir = writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), + GraphicsExportType.TIFF, + _mainpanel, + _writetographics_filechooser, + this, + getContentPane(), + _current_dir ); + if ( new_dir != null ) { + setCurrentDir( new_dir ); + } } else if ( o == _write_to_bmp_item ) { - writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), GraphicsExportType.BMP ); + final File new_dir = writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), + GraphicsExportType.BMP, + _mainpanel, + _writetographics_filechooser, + this, + getContentPane(), + _current_dir ); + if ( new_dir != null ) { + setCurrentDir( new_dir ); + } + } + else if ( o == _write_to_png_item ) { + final File new_dir = writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), + GraphicsExportType.PNG, + _mainpanel, + _writetographics_filechooser, + this, + getContentPane(), + _current_dir ); + if ( new_dir != null ) { + setCurrentDir( new_dir ); + } } else if ( o == _print_item ) { - print( getCurrentTreePanel(), - getOptions(), - - this ); + print( getCurrentTreePanel(), getOptions(), this ); } else if ( o == _save_item ) { - writeToFile( _mainpanel.getCurrentPhylogeny() ); - } - else if ( o == _write_to_png_item ) { - writeToGraphicsFile( _mainpanel.getCurrentPhylogeny(), GraphicsExportType.PNG ); + final File new_dir = writeToFile( _mainpanel.getCurrentPhylogeny(), + getMainPanel(), + _save_filechooser, + _current_dir, + getContentPane(), + this ); + if ( new_dir != null ) { + setCurrentDir( new_dir ); + } } else if ( o == _graphics_export_visible_only_cbmi ) { updateOptions( getOptions() ); @@ -734,154 +803,101 @@ public abstract class MainFrame extends JFrame implements ActionListener { } ); } - void activateSaveAllIfNeeded() { - if ( ( getMainPanel().getTabbedPane() != null ) && ( getMainPanel().getTabbedPane().getTabCount() > 1 ) ) { - _save_all_item.setEnabled( true ); - } - else { - _save_all_item.setEnabled( false ); - } - } - - void buildFileMenu() { - _file_jmenu = MainFrame.createMenu( "File", getConfiguration() ); - _file_jmenu.add( _save_item = new JMenuItem( "Save Tree As..." ) ); - _file_jmenu.addSeparator(); - _file_jmenu.add( _write_to_pdf_item = new JMenuItem( "Export to PDF file ..." ) ); - if ( AptxUtil.canWriteFormat( "tif" ) || AptxUtil.canWriteFormat( "tiff" ) || AptxUtil.canWriteFormat( "TIF" ) ) { - _file_jmenu.add( _write_to_tif_item = new JMenuItem( "Export to TIFF file..." ) ); - } - _file_jmenu.add( _write_to_png_item = new JMenuItem( "Export to PNG file..." ) ); - _file_jmenu.add( _write_to_jpg_item = new JMenuItem( "Export to JPG file..." ) ); - if ( AptxUtil.canWriteFormat( "gif" ) ) { - _file_jmenu.add( _write_to_gif_item = new JMenuItem( "Export to GIF file..." ) ); - } - if ( AptxUtil.canWriteFormat( "bmp" ) ) { - _file_jmenu.add( _write_to_bmp_item = new JMenuItem( "Export to BMP file..." ) ); - } - _file_jmenu.addSeparator(); - _file_jmenu.add( _print_item = new JMenuItem( "Print..." ) ); - _file_jmenu.addSeparator(); - _file_jmenu.add( _exit_item = new JMenuItem( "Exit" ) ); - customizeJMenuItem( _save_item ); - customizeJMenuItem( _write_to_pdf_item ); - customizeJMenuItem( _write_to_png_item ); - customizeJMenuItem( _write_to_jpg_item ); - customizeJMenuItem( _write_to_gif_item ); - customizeJMenuItem( _write_to_tif_item ); - customizeJMenuItem( _write_to_bmp_item ); - customizeJMenuItem( _print_item ); - customizeJMenuItem( _exit_item ); - _jmenubar.add( _file_jmenu ); - } - - void buildFontSizeMenu() { - _font_size_menu = createMenu( FONT_SIZE_MENU_LABEL, getConfiguration() ); - _font_size_menu.add( _super_tiny_fonts_item = new JMenuItem( "Super Tiny Fonts" ) ); - _font_size_menu.add( _tiny_fonts_item = new JMenuItem( "Tiny Fonts" ) ); - _font_size_menu.add( _small_fonts_item = new JMenuItem( "Small Fonts" ) ); - _font_size_menu.add( _medium_fonts_item = new JMenuItem( "Medium Fonts" ) ); - _font_size_menu.add( _large_fonts_item = new JMenuItem( "Large Fonts" ) ); - customizeJMenuItem( _super_tiny_fonts_item ); - customizeJMenuItem( _tiny_fonts_item ); - customizeJMenuItem( _small_fonts_item ); - customizeJMenuItem( _medium_fonts_item ); - customizeJMenuItem( _large_fonts_item ); - _jmenubar.add( _font_size_menu ); - } - - void buildHelpMenu() { - _help_jmenu = createMenu( "Help", getConfiguration() ); - _help_jmenu.add( _help_item = new JMenuItem( "Documentation" ) ); - _help_jmenu.addSeparator(); - _help_jmenu.add( _website_item = new JMenuItem( "Archaeopteryx Home" ) ); - _aptx_ref_item = new JMenuItem( "Archaeopteryx Reference" ); //TODO need to add this... - _help_jmenu.add( _phyloxml_website_item = new JMenuItem( "phyloXML Home" ) ); - _help_jmenu.add( _phyloxml_ref_item = new JMenuItem( "phyloXML Reference" ) ); - _help_jmenu.addSeparator(); - _help_jmenu.add( _about_item = new JMenuItem( "About" ) ); - customizeJMenuItem( _help_item ); - customizeJMenuItem( _website_item ); - customizeJMenuItem( _phyloxml_website_item ); - customizeJMenuItem( _aptx_ref_item ); - customizeJMenuItem( _phyloxml_ref_item ); - customizeJMenuItem( _about_item ); - _phyloxml_ref_item.setToolTipText( PHYLOXML_REF_TOOL_TIP ); - _aptx_ref_item.setToolTipText( APTX_REF_TOOL_TIP ); - _jmenubar.add( _help_jmenu ); - } - - void buildTypeMenu() { - _type_menu = createMenu( TYPE_MENU_HEADER, getConfiguration() ); - _type_menu.add( _rectangular_type_cbmi = new JCheckBoxMenuItem( MainFrame.RECTANGULAR_TYPE_CBMI_LABEL ) ); - _type_menu.add( _euro_type_cbmi = new JCheckBoxMenuItem( MainFrame.EURO_TYPE_CBMI_LABEL ) ); - _type_menu.add( _rounded_type_cbmi = new JCheckBoxMenuItem( MainFrame.ROUNDED_TYPE_CBMI_LABEL ) ); - _type_menu.add( _curved_type_cbmi = new JCheckBoxMenuItem( MainFrame.CURVED_TYPE_CBMI_LABEL ) ); - _type_menu.add( _triangular_type_cbmi = new JCheckBoxMenuItem( MainFrame.TRIANGULAR_TYPE_CBMI_LABEL ) ); - _type_menu.add( _convex_type_cbmi = new JCheckBoxMenuItem( MainFrame.CONVEX_TYPE_CBMI_LABEL ) ); - _type_menu.add( _unrooted_type_cbmi = new JCheckBoxMenuItem( MainFrame.UNROOTED_TYPE_CBMI_LABEL ) ); - _type_menu.add( _circular_type_cbmi = new JCheckBoxMenuItem( MainFrame.CIRCULAR_TYPE_CBMI_LABEL ) ); - customizeCheckBoxMenuItem( _rectangular_type_cbmi, false ); - customizeCheckBoxMenuItem( _triangular_type_cbmi, false ); - customizeCheckBoxMenuItem( _euro_type_cbmi, false ); - customizeCheckBoxMenuItem( _rounded_type_cbmi, false ); - customizeCheckBoxMenuItem( _curved_type_cbmi, false ); - customizeCheckBoxMenuItem( _convex_type_cbmi, false ); - customizeCheckBoxMenuItem( _unrooted_type_cbmi, false ); - customizeCheckBoxMenuItem( _circular_type_cbmi, false ); - _unrooted_type_cbmi.setToolTipText( MainFrame.USE_MOUSEWHEEL_SHIFT_TO_ROTATE ); - _circular_type_cbmi.setToolTipText( MainFrame.USE_MOUSEWHEEL_SHIFT_TO_ROTATE ); - initializeTypeMenu( getOptions() ); - _jmenubar.add( _type_menu ); - } - - void buildViewMenu() { - _view_jmenu = createMenu( "View", getConfiguration() ); - _view_jmenu.add( _display_basic_information_item = new JMenuItem( SHOW_BASIC_TREE_INFORMATION_LABEL ) ); - _view_jmenu.addSeparator(); - _view_jmenu.add( _view_as_XML_item = new JMenuItem( "as phyloXML" ) ); - _view_jmenu.add( _view_as_NH_item = new JMenuItem( "as Newick" ) ); - _view_jmenu.add( _view_as_nexus_item = new JMenuItem( "as Nexus" ) ); - customizeJMenuItem( _display_basic_information_item ); - customizeJMenuItem( _view_as_NH_item ); - customizeJMenuItem( _view_as_XML_item ); - customizeJMenuItem( _view_as_nexus_item ); - _jmenubar.add( _view_jmenu ); - } - - void checkTextFrames() { - if ( _textframes.size() > 5 ) { - try { - if ( _textframes.getFirst() != null ) { - _textframes.getFirst().removeMe(); - } - else { - _textframes.removeFirst(); - } + private void annotateSequences() { + if ( getCurrentTreePanel() != null ) { + List nodes = null; + if ( ( getCurrentTreePanel().getFoundNodes0() != null ) + || ( getCurrentTreePanel().getFoundNodes1() != null ) ) { + nodes = getCurrentTreePanel().getFoundNodesAsListOfPhylogenyNodes(); } - catch ( final NoSuchElementException e ) { - // Ignore. + if ( ( nodes == null ) || nodes.isEmpty() ) { + JOptionPane + .showMessageDialog( this, + "Need to select nodes, either via direct selection or via the \"Search\" function", + "No nodes selected for annotation", + JOptionPane.ERROR_MESSAGE ); + return; + } + final Phylogeny phy = getMainPanel().getCurrentPhylogeny(); + if ( ( phy != null ) && !phy.isEmpty() ) { + final JTextField ref_field = new JTextField( 10 ); + final JTextField desc_filed = new JTextField( 20 ); + ref_field.setText( ForesterUtil.isEmpty( getPreviousNodeAnnotationReference() ) ? "" + : getPreviousNodeAnnotationReference() ); + final JPanel my_panel = new JPanel(); + my_panel.add( new JLabel( "Reference " ) ); + my_panel.add( ref_field ); + my_panel.add( Box.createHorizontalStrut( 15 ) ); + my_panel.add( new JLabel( "Description " ) ); + my_panel.add( desc_filed ); + final int result = JOptionPane.showConfirmDialog( null, + my_panel, + "Enter the sequence annotation(s) for the " + + nodes.size() + " selected nodes", + JOptionPane.OK_CANCEL_OPTION ); + if ( result == JOptionPane.OK_OPTION ) { + String ref = ref_field.getText(); + String desc = desc_filed.getText(); + if ( !ForesterUtil.isEmpty( ref ) ) { + ref = ref.trim(); + ref = ref.replaceAll( "\\s+", " " ); + if ( ( ref.indexOf( ':' ) < 1 ) || ( ref.indexOf( ':' ) > ( ref.length() - 2 ) ) + || ( ref.length() < 3 ) ) { + JOptionPane.showMessageDialog( this, + "Reference needs to be in the form of \"GO:1234567\"", + "Illegal Format for Annotation Reference", + JOptionPane.ERROR_MESSAGE ); + return; + } + } + if ( ref != null ) { + setPreviousNodeAnnotationReference( ref ); + } + if ( desc != null ) { + desc = desc.trim(); + desc = desc.replaceAll( "\\s+", " " ); + } + if ( !ForesterUtil.isEmpty( ref ) || !ForesterUtil.isEmpty( desc ) ) { + for( final PhylogenyNode n : nodes ) { + ForesterUtil.ensurePresenceOfSequence( n ); + final Annotation ann = ForesterUtil.isEmpty( ref ) ? new Annotation() + : new Annotation( ref ); + if ( !ForesterUtil.isEmpty( desc ) ) { + ann.setDesc( desc ); + } + n.getNodeData().getSequence().addAnnotation( ann ); + } + } + getMainPanel().getControlPanel().showAnnotations(); + } } } } - void choosePdfWidth() { - final String s = ( String ) JOptionPane.showInputDialog( this, - "Please enter the default line width for PDF export.\n" - + "[current value: " - + getOptions().getPrintLineWidth() + "]\n", - "Line Width for PDF Export", - JOptionPane.QUESTION_MESSAGE, - null, - null, - getOptions().getPrintLineWidth() ); + private void chooseFont() { + final FontChooser fc = new FontChooser(); + fc.setFont( getMainPanel().getTreeFontSet().getLargeFont() ); + fc.showDialog( this, "Select the Base Font" ); + getMainPanel().getTreeFontSet().setBaseFont( fc.getFont() ); + } + + private void chooseMinimalConfidence() { + final String s = ( String ) JOptionPane + .showInputDialog( this, + "Please enter the minimum for confidence values to be displayed.\n" + + "[current value: " + getOptions().getMinConfidenceValue() + "]\n", + "Minimal Confidence Value", + JOptionPane.QUESTION_MESSAGE, + null, + null, + getOptions().getMinConfidenceValue() ); if ( !ForesterUtil.isEmpty( s ) ) { boolean success = true; - float f = 0.0f; + double m = 0.0; final String m_str = s.trim(); if ( !ForesterUtil.isEmpty( m_str ) ) { try { - f = Float.parseFloat( m_str ); + m = Double.parseDouble( m_str ); } catch ( final Exception ex ) { success = false; @@ -890,1365 +906,1111 @@ public abstract class MainFrame extends JFrame implements ActionListener { else { success = false; } - if ( success && ( f > 0.0 ) ) { - getOptions().setPrintLineWidth( f ); + if ( success && ( m >= 0.0 ) ) { + getOptions().setMinConfidenceValue( m ); } } } - void choosePrintSize() { - final String s = ( String ) JOptionPane.showInputDialog( this, - "Please enter values for width and height,\nseparated by a comma.\n" - + "[current values: " - + getOptions().getPrintSizeX() + ", " - + getOptions().getPrintSizeY() + "]\n" - + "[A4: " + Constants.A4_SIZE_X + ", " - + Constants.A4_SIZE_Y + "]\n" + "[US Letter: " - + Constants.US_LETTER_SIZE_X + ", " - + Constants.US_LETTER_SIZE_Y + "]", - "Default Size for Graphics Export", - JOptionPane.QUESTION_MESSAGE, - null, - null, - getOptions().getPrintSizeX() + ", " - + getOptions().getPrintSizeY() ); - if ( !ForesterUtil.isEmpty( s ) && ( s.indexOf( ',' ) > 0 ) ) { - boolean success = true; - int x = 0; - int y = 0; - final String[] str_ary = s.split( "," ); - if ( str_ary.length == 2 ) { - final String x_str = str_ary[ 0 ].trim(); - final String y_str = str_ary[ 1 ].trim(); - if ( !ForesterUtil.isEmpty( x_str ) && !ForesterUtil.isEmpty( y_str ) ) { - try { - x = Integer.parseInt( x_str ); - y = Integer.parseInt( y_str ); - } - catch ( final Exception ex ) { - success = false; + private void deleteSelectedNodes( final boolean delete ) { + final Phylogeny phy = getMainPanel().getCurrentPhylogeny(); + if ( ( phy == null ) || ( phy.getNumberOfExternalNodes() < 2 ) ) { + return; + } + final List nodes = new ArrayList(); + if ( ( getCurrentTreePanel().getFoundNodes0() != null ) || ( getCurrentTreePanel().getFoundNodes1() != null ) ) { + final List all_selected_nodes = getCurrentTreePanel().getFoundNodesAsListOfPhylogenyNodes(); + for( final PhylogenyNode n : all_selected_nodes ) { + if ( n.isExternal() ) { + nodes.add( n ); + } + } + } + String function = "Retain"; + if ( delete ) { + function = "Delete"; + } + if ( ( nodes == null ) || nodes.isEmpty() ) { + JOptionPane + .showMessageDialog( this, + "Need to select external nodes, either via direct selection or via the \"Search\" function", + "No external nodes selected to " + function.toLowerCase(), + JOptionPane.ERROR_MESSAGE ); + return; + } + final int todo = nodes.size(); + final int ext = phy.getNumberOfExternalNodes(); + int res = todo; + if ( delete ) { + res = ext - todo; + } + if ( res < 1 ) { + JOptionPane.showMessageDialog( this, + "Cannot delete all nodes", + "Attempt to delete all nodes ", + JOptionPane.ERROR_MESSAGE ); + return; + } + final int result = JOptionPane.showConfirmDialog( null, function + " " + todo + + " external node(s), from a total of " + ext + " external nodes," + "\nresulting in tree with " + res + + " nodes?", function + " external nodes", JOptionPane.OK_CANCEL_OPTION ); + if ( result == JOptionPane.OK_OPTION ) { + if ( !delete ) { + final List to_delete = new ArrayList(); + for( final PhylogenyNodeIterator it = phy.iteratorExternalForward(); it.hasNext(); ) { + final PhylogenyNode n = it.next(); + if ( !nodes.contains( n ) ) { + to_delete.add( n ); } } - else { - success = false; + for( final PhylogenyNode n : to_delete ) { + phy.deleteSubtree( n, true ); } } else { - success = false; - } - if ( success && ( x > 1 ) && ( y > 1 ) ) { - getOptions().setPrintSizeX( x ); - getOptions().setPrintSizeY( y ); + for( final PhylogenyNode n : nodes ) { + phy.deleteSubtree( n, true ); + } } + resetSearch(); + getCurrentTreePanel().setNodeInPreorderToNull(); + phy.externalNodesHaveChanged(); + phy.clearHashIdToNodeMap(); + phy.recalculateNumberOfExternalDescendants( true ); + getCurrentTreePanel().resetNodeIdToDistToLeafMap(); + getCurrentTreePanel().setEdited( true ); + repaint(); } } - void close() { - removeAllTextFrames(); - if ( _mainpanel != null ) { - _mainpanel.terminate(); - } - if ( _contentpane != null ) { - _contentpane.removeAll(); + private void doUpdateProcessMenu() { + if ( _process_pool.size() > 0 ) { + if ( _process_menu == null ) { + _process_menu = createMenu( "", getConfiguration() ); + _process_menu.setForeground( Color.RED ); + } + _process_menu.removeAll(); + final String text = "processes running: " + _process_pool.size(); + _process_menu.setText( text ); + _jmenubar.add( _process_menu ); + for( int i = 0; i < _process_pool.size(); ++i ) { + final ProcessRunning p = _process_pool.getProcessByIndex( i ); + _process_menu.add( customizeJMenuItem( new JMenuItem( p.getName() + " [" + p.getStart() + "]" ) ) ); + } } - setVisible( false ); - dispose(); - } - - void colorRank() { - if ( _mainpanel.getCurrentTreePanel() != null ) { - final String[] ranks = AptxUtil.getAllPossibleRanks(); - final String rank = ( String ) JOptionPane - .showInputDialog( this, - "What rank should the colorization be based on", - "Rank Selection", - JOptionPane.QUESTION_MESSAGE, - null, - ranks, - null ); - if ( !ForesterUtil.isEmpty( rank ) ) { - _mainpanel.getCurrentTreePanel().colorRank( rank ); + else { + if ( _process_menu != null ) { + _process_menu.removeAll(); + _jmenubar.remove( _process_menu ); } } + _jmenubar.validate(); + _jmenubar.repaint(); + repaint(); } - void confColor() { - if ( _mainpanel.getCurrentTreePanel() != null ) { - _mainpanel.getCurrentTreePanel().confColor(); - } + private String getPreviousNodeAnnotationReference() { + return _previous_node_annotation_ref; } - void customizeCheckBoxMenuItem( final JCheckBoxMenuItem item, final boolean is_selected ) { - if ( item != null ) { - item.setFont( MainFrame.menu_font ); - if ( !getConfiguration().isUseNativeUI() ) { - item.setBackground( getConfiguration().getGuiMenuBackgroundColor() ); - item.setForeground( getConfiguration().getGuiMenuTextColor() ); - } - item.setSelected( is_selected ); - item.addActionListener( this ); + private void removeBranchColors() { + if ( getMainPanel().getCurrentPhylogeny() != null ) { + AptxUtil.removeBranchColors( getMainPanel().getCurrentPhylogeny() ); } } - JMenuItem customizeJMenuItem( final JMenuItem jmi ) { - if ( jmi != null ) { - jmi.setFont( MainFrame.menu_font ); - if ( !getConfiguration().isUseNativeUI() ) { - jmi.setBackground( getConfiguration().getGuiMenuBackgroundColor() ); - jmi.setForeground( getConfiguration().getGuiMenuTextColor() ); - } - jmi.addActionListener( this ); + private void removeVisualStyles() { + if ( getMainPanel().getCurrentPhylogeny() != null ) { + AptxUtil.removeVisualStyles( getMainPanel().getCurrentPhylogeny() ); } - return jmi; } - void customizeRadioButtonMenuItem( final JRadioButtonMenuItem item, final boolean is_selected ) { - if ( item != null ) { - item.setFont( MainFrame.menu_font ); - if ( !getConfiguration().isUseNativeUI() ) { - item.setBackground( getConfiguration().getGuiMenuBackgroundColor() ); - item.setForeground( getConfiguration().getGuiMenuTextColor() ); - } - item.setSelected( is_selected ); - item.addActionListener( this ); - } + private void setPreviousNodeAnnotationReference( final String previous_node_annotation_ref ) { + _previous_node_annotation_ref = previous_node_annotation_ref; } - void displayBasicInformation( final File treefile ) { - if ( ( _mainpanel.getCurrentPhylogeny() != null ) && !_mainpanel.getCurrentPhylogeny().isEmpty() ) { - String title = "Basic Information"; - if ( !ForesterUtil.isEmpty( _mainpanel.getCurrentPhylogeny().getName() ) ) { - title = title + " for \"" + _mainpanel.getCurrentPhylogeny().getName() + "\""; + private void writeAllToFile() { + if ( ( getMainPanel().getTabbedPane() == null ) || ( getMainPanel().getTabbedPane().getTabCount() < 1 ) ) { + return; + } + final File my_dir = getCurrentDir(); + if ( my_dir != null ) { + _save_filechooser.setCurrentDirectory( my_dir ); + } + _save_filechooser.setSelectedFile( new File( "" ) ); + final int result = _save_filechooser.showSaveDialog( _contentpane ); + final File file = _save_filechooser.getSelectedFile(); + setCurrentDir( _save_filechooser.getCurrentDirectory() ); + if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { + if ( file.exists() ) { + final int i = JOptionPane.showConfirmDialog( this, + file + " already exists. Overwrite?", + "Warning", + JOptionPane.OK_CANCEL_OPTION, + JOptionPane.WARNING_MESSAGE ); + if ( i != JOptionPane.OK_OPTION ) { + return; + } + else { + try { + file.delete(); + } + catch ( final Exception e ) { + JOptionPane.showMessageDialog( this, + "Failed to delete: " + file, + "Error", + JOptionPane.WARNING_MESSAGE ); + } + } + } + final int count = getMainPanel().getTabbedPane().getTabCount(); + final List trees = new ArrayList(); + for( int i = 0; i < count; ++i ) { + final Phylogeny phy = getMainPanel().getPhylogeny( i ); + if ( ForesterUtil.isEmpty( phy.getName() ) + && !ForesterUtil.isEmpty( getMainPanel().getTabbedPane().getTitleAt( i ) ) ) { + phy.setName( getMainPanel().getTabbedPane().getTitleAt( i ) ); + } + trees.add( phy ); + getMainPanel().getTreePanels().get( i ).setEdited( false ); + } + final PhylogenyWriter writer = new PhylogenyWriter(); + try { + writer.toPhyloXML( file, trees, 0, ForesterUtil.LINE_SEPARATOR ); + } + catch ( final IOException e ) { + JOptionPane.showMessageDialog( this, + "Failed to write to: " + file, + "Error", + JOptionPane.WARNING_MESSAGE ); } - showTextFrame( AptxUtil.createBasicInformation( _mainpanel.getCurrentPhylogeny(), treefile ), title ); } } - void exceptionOccuredDuringOpenFile( final Exception e ) { - try { - _mainpanel.getCurrentTreePanel().setArrowCursor(); + void activateSaveAllIfNeeded() { + if ( ( getMainPanel().getTabbedPane() != null ) && ( getMainPanel().getTabbedPane().getTabCount() > 1 ) ) { + _save_all_item.setEnabled( true ); } - catch ( final Exception ex ) { - // Do nothing. + else { + _save_all_item.setEnabled( false ); } - JOptionPane.showMessageDialog( this, - ForesterUtil.wordWrap( e.getLocalizedMessage(), 80 ), - "Error during File|Open", - JOptionPane.ERROR_MESSAGE ); } - void exceptionOccuredDuringSaveAs( final Exception e ) { - try { - _mainpanel.getCurrentTreePanel().setArrowCursor(); - } - catch ( final Exception ex ) { - // Do nothing. - } - JOptionPane.showMessageDialog( this, "Exception" + e, "Error during File|SaveAs", JOptionPane.ERROR_MESSAGE ); - } - - void executeGSDI() { - if ( !isOKforSDI( false, true ) ) { - return; - } - if ( !_mainpanel.getCurrentPhylogeny().isRooted() ) { - JOptionPane.showMessageDialog( this, - "Gene tree is not rooted.", - "Cannot execute GSDI", - JOptionPane.ERROR_MESSAGE ); - return; - } - final Phylogeny gene_tree = _mainpanel.getCurrentPhylogeny().copy(); - gene_tree.setAllNodesToNotCollapse(); - gene_tree.recalculateNumberOfExternalDescendants( false ); - GSDI gsdi = null; - final Phylogeny species_tree = getSpeciesTree().copy(); - try { - gsdi = new GSDI( gene_tree, species_tree, false, true, true, true ); - } - catch ( final SDIException e ) { - JOptionPane.showMessageDialog( this, - e.getLocalizedMessage(), - "Error during GSDI", - JOptionPane.ERROR_MESSAGE ); - return; - } - catch ( final Exception e ) { - AptxUtil.unexpectedException( e ); - return; - } - gene_tree.setRerootable( false ); - gene_tree.clearHashIdToNodeMap(); - gene_tree.recalculateNumberOfExternalDescendants( true ); - _mainpanel.addPhylogenyInNewTab( gene_tree, getConfiguration(), "gene tree", null ); - getMainPanel().getControlPanel().setShowEvents( true ); - showWhole(); - final int selected = _mainpanel.getTabbedPane().getSelectedIndex(); - _mainpanel.addPhylogenyInNewTab( species_tree, getConfiguration(), "species tree", null ); - showWhole(); - _mainpanel.getTabbedPane().setSelectedIndex( selected ); - showWhole(); - _mainpanel.getCurrentTreePanel().setEdited( true ); - final int poly = PhylogenyMethods.countNumberOfPolytomies( species_tree ); - if ( gsdi.getStrippedExternalGeneTreeNodes().size() > 0 ) { - JOptionPane.showMessageDialog( this, - "Duplications: " + gsdi.getDuplicationsSum() + "\n" - + "Potential duplications: " - + gsdi.getSpeciationOrDuplicationEventsSum() + "\n" - + "Speciations: " + gsdi.getSpeciationsSum() + "\n" - + "Stripped gene tree nodes: " - + gsdi.getStrippedExternalGeneTreeNodes().size() + "\n" - + "Taxonomy linkage based on: " + gsdi.getTaxCompBase() + "\n" - + "Number of polytomies in species tree used: " + poly + "\n", - "GSDI successfully completed", - JOptionPane.WARNING_MESSAGE ); - } - else { - JOptionPane.showMessageDialog( this, - "Duplications: " + gsdi.getDuplicationsSum() + "\n" - + "Potential duplications: " - + gsdi.getSpeciationOrDuplicationEventsSum() + "\n" - + "Speciations: " + gsdi.getSpeciationsSum() + "\n" - + "Stripped gene tree nodes: " - + gsdi.getStrippedExternalGeneTreeNodes().size() + "\n" - + "Taxonomy linkage based on: " + gsdi.getTaxCompBase() + "\n" - + "Number of polytomies in species tree used: " + poly + "\n", - "GSDI successfully completed", - JOptionPane.INFORMATION_MESSAGE ); - } - } - - void executeGSDIR() { - if ( !isOKforSDI( false, false ) ) { - return; - } - final int p = PhylogenyMethods.countNumberOfPolytomies( _mainpanel.getCurrentPhylogeny() ); - if ( ( p > 0 ) - && !( ( p == 1 ) && ( _mainpanel.getCurrentPhylogeny().getRoot().getNumberOfDescendants() == 3 ) ) ) { - JOptionPane.showMessageDialog( this, - "Gene tree is not completely binary", - "Cannot execute GSDI", - JOptionPane.ERROR_MESSAGE ); - return; - } - final Phylogeny gene_tree = _mainpanel.getCurrentPhylogeny().copy(); - gene_tree.setAllNodesToNotCollapse(); - gene_tree.recalculateNumberOfExternalDescendants( false ); - GSDIR gsdir = null; - final Phylogeny species_tree = getSpeciesTree().copy(); - try { - gsdir = new GSDIR( gene_tree, species_tree, true, true, true ); - } - catch ( final SDIException e ) { - JOptionPane.showMessageDialog( this, - e.getLocalizedMessage(), - "Error during GSDIR", - JOptionPane.ERROR_MESSAGE ); - return; - } - catch ( final Exception e ) { - AptxUtil.unexpectedException( e ); - return; - } - final Phylogeny result_gene_tree = gsdir.getMinDuplicationsSumGeneTree(); - result_gene_tree.setRerootable( false ); - result_gene_tree.clearHashIdToNodeMap(); - result_gene_tree.recalculateNumberOfExternalDescendants( true ); - PhylogenyMethods.orderAppearance( result_gene_tree.getRoot(), true, true, DESCENDANT_SORT_PRIORITY.NODE_NAME ); - _mainpanel.addPhylogenyInNewTab( result_gene_tree, getConfiguration(), "gene tree", null ); - getMainPanel().getControlPanel().setShowEvents( true ); - showWhole(); - final int selected = _mainpanel.getTabbedPane().getSelectedIndex(); - _mainpanel.addPhylogenyInNewTab( species_tree, getConfiguration(), "species tree", null ); - showWhole(); - _mainpanel.getTabbedPane().setSelectedIndex( selected ); - showWhole(); - _mainpanel.getCurrentTreePanel().setEdited( true ); - final int poly = PhylogenyMethods.countNumberOfPolytomies( species_tree ); - if ( gsdir.getStrippedExternalGeneTreeNodes().size() > 0 ) { - JOptionPane.showMessageDialog( this, - "Minimal duplications: " + gsdir.getMinDuplicationsSum() + "\n" - + "Speciations: " + gsdir.getSpeciationsSum() + "\n" - + "Stripped gene tree nodes: " - + gsdir.getStrippedExternalGeneTreeNodes().size() + "\n" - + "Taxonomy linkage based on: " + gsdir.getTaxCompBase() + "\n" - + "Number of polytomies in species tree used: " + poly + "\n", - "GSDIR successfully completed", - JOptionPane.WARNING_MESSAGE ); - } - else { - JOptionPane.showMessageDialog( this, - "Minimal duplications: " + gsdir.getMinDuplicationsSum() + "\n" - + "Speciations: " + gsdir.getSpeciationsSum() + "\n" - + "Stripped gene tree nodes: " - + gsdir.getStrippedExternalGeneTreeNodes().size() + "\n" - + "Taxonomy linkage based on: " + gsdir.getTaxCompBase() + "\n" - + "Number of polytomies in species tree used: " + poly + "\n", - "GSDIR successfully completed", - JOptionPane.INFORMATION_MESSAGE ); - } - } - - boolean GAndSDoHaveMoreThanOneSpeciesInComman( final Phylogeny gene_tree ) { - if ( ( gene_tree == null ) || gene_tree.isEmpty() ) { - JOptionPane.showMessageDialog( this, - "Gene tree and species tree have no species in common.", - "Error during SDI", - JOptionPane.ERROR_MESSAGE ); - return false; - } - else if ( gene_tree.getNumberOfExternalNodes() < 2 ) { - JOptionPane.showMessageDialog( this, - "Gene tree and species tree have only one species in common.", - "Error during SDI", - JOptionPane.ERROR_MESSAGE ); - return false; - } - else { - return true; - } - } - - ControlPanel getControlPanel() { - return getMainPanel().getControlPanel(); - } - - File getCurrentDir() { - if ( ( _current_dir == null ) || !_current_dir.canRead() ) { - if ( ForesterUtil.isWindows() ) { - try { - _current_dir = new File( WindowsUtils.getCurrentUserDesktopPath() ); - } - catch ( final Exception e ) { - _current_dir = null; - } - } - } - if ( ( _current_dir == null ) || !_current_dir.canRead() ) { - if ( System.getProperty( "user.home" ) != null ) { - _current_dir = new File( System.getProperty( "user.home" ) ); - } - else if ( System.getProperty( "user.dir" ) != null ) { - _current_dir = new File( System.getProperty( "user.dir" ) ); - } - } - return _current_dir; - } - - TreePanel getCurrentTreePanel() { - return getMainPanel().getCurrentTreePanel(); - } - - JMenu getHelpMenu() { - return _help_jmenu; - } - - JCheckBoxMenuItem getlabelDirectionCbmi() { - return _label_direction_cbmi; - } - - JMenuBar getMenuBarOfMainFrame() { - return _jmenubar; - } - - final Phylogeny getSpeciesTree() { - return _species_tree; - } - - void initializeTypeMenu( final Options options ) { - setTypeMenuToAllUnselected(); - switch ( options.getPhylogenyGraphicsType() ) { - case CONVEX: - _convex_type_cbmi.setSelected( true ); - break; - case CURVED: - _curved_type_cbmi.setSelected( true ); - break; - case EURO_STYLE: - _euro_type_cbmi.setSelected( true ); - break; - case ROUNDED: - _rounded_type_cbmi.setSelected( true ); - break; - case TRIANGULAR: - _triangular_type_cbmi.setSelected( true ); - break; - case UNROOTED: - _unrooted_type_cbmi.setSelected( true ); - break; - case CIRCULAR: - _circular_type_cbmi.setSelected( true ); - break; - default: - _rectangular_type_cbmi.setSelected( true ); - break; - } - } - - boolean isOKforSDI( final boolean species_tree_has_to_binary, final boolean gene_tree_has_to_binary ) { - if ( ( _mainpanel.getCurrentPhylogeny() == null ) || _mainpanel.getCurrentPhylogeny().isEmpty() ) { - return false; - } - else if ( ( getSpeciesTree() == null ) || getSpeciesTree().isEmpty() ) { - JOptionPane.showMessageDialog( this, - "No species tree loaded", - "Cannot execute GSDI", - JOptionPane.ERROR_MESSAGE ); - return false; - } - else if ( species_tree_has_to_binary && !getSpeciesTree().isCompletelyBinary() ) { - JOptionPane.showMessageDialog( this, - "Species tree is not completely binary", - "Cannot execute GSDI", - JOptionPane.ERROR_MESSAGE ); - return false; - } - else if ( gene_tree_has_to_binary && !_mainpanel.getCurrentPhylogeny().isCompletelyBinary() ) { - JOptionPane.showMessageDialog( this, - "Gene tree is not completely binary", - "Cannot execute GSDI", - JOptionPane.ERROR_MESSAGE ); - return false; - } - else { - return true; - } - } - - boolean isSubtreeDisplayed() { - if ( getCurrentTreePanel() != null ) { - if ( getCurrentTreePanel().isCurrentTreeIsSubtree() ) { - JOptionPane - .showMessageDialog( this, - "This operation can only be performed on a complete tree, not on the currently displayed sub-tree only.", - "Operation can not be exectuted on a sub-tree", - JOptionPane.WARNING_MESSAGE ); - return true; - } - } - return false; - } - - void midpointRoot() { - if ( _mainpanel.getCurrentTreePanel() != null ) { - _mainpanel.getCurrentTreePanel().midpointRoot(); - } - } - - void printPhylogenyToPdf( final String file_name ) { - if ( !getOptions().isPrintUsingActualSize() ) { - getCurrentTreePanel() - .calcParametersForPainting( getOptions().getPrintSizeX(), getOptions().getPrintSizeY() ); - getCurrentTreePanel().resetPreferredSize(); - getCurrentTreePanel().repaint(); - } - String pdf_written_to = ""; - boolean error = false; - try { - if ( getOptions().isPrintUsingActualSize() ) { - pdf_written_to = PdfExporter.writePhylogenyToPdf( file_name, - getCurrentTreePanel(), - getCurrentTreePanel().getWidth(), - getCurrentTreePanel().getHeight() ); - } - else { - pdf_written_to = PdfExporter.writePhylogenyToPdf( file_name, getCurrentTreePanel(), getOptions() - .getPrintSizeX(), getOptions().getPrintSizeY() ); - } - } - catch ( final IOException e ) { - error = true; - JOptionPane.showMessageDialog( this, e.getMessage(), "Error", JOptionPane.ERROR_MESSAGE ); - } - if ( !error ) { - if ( !ForesterUtil.isEmpty( pdf_written_to ) ) { - JOptionPane.showMessageDialog( this, - "Wrote PDF to: " + pdf_written_to, - "Information", - JOptionPane.INFORMATION_MESSAGE ); - } - else { - JOptionPane.showMessageDialog( this, - "There was an unknown problem when attempting to write to PDF file: \"" - + file_name + "\"", - "Error", - JOptionPane.ERROR_MESSAGE ); - } - } - if ( !getOptions().isPrintUsingActualSize() ) { - getControlPanel().showWhole(); + void buildFileMenu() { + _file_jmenu = MainFrame.createMenu( "File", getConfiguration() ); + _file_jmenu.add( _save_item = new JMenuItem( "Save Tree As..." ) ); + _file_jmenu.addSeparator(); + _file_jmenu.add( _write_to_pdf_item = new JMenuItem( "Export to PDF file ..." ) ); + if ( AptxUtil.canWriteFormat( "tif" ) || AptxUtil.canWriteFormat( "tiff" ) || AptxUtil.canWriteFormat( "TIF" ) ) { + _file_jmenu.add( _write_to_tif_item = new JMenuItem( "Export to TIFF file..." ) ); } - } - - void readPhylogeniesFromWebservice( final int i ) { - final UrlTreeReader reader = new UrlTreeReader( this, i ); - new Thread( reader ).start(); - } - - void removeAllTextFrames() { - for( final TextFrame tf : _textframes ) { - if ( tf != null ) { - tf.close(); - } + _file_jmenu.add( _write_to_png_item = new JMenuItem( "Export to PNG file..." ) ); + _file_jmenu.add( _write_to_jpg_item = new JMenuItem( "Export to JPG file..." ) ); + if ( AptxUtil.canWriteFormat( "gif" ) ) { + _file_jmenu.add( _write_to_gif_item = new JMenuItem( "Export to GIF file..." ) ); } - _textframes.clear(); - } - - void resetSearch() { - getMainPanel().getCurrentTreePanel().setFoundNodes0( null ); - getMainPanel().getCurrentTreePanel().setFoundNodes1( null ); - getMainPanel().getControlPanel().setSearchFoundCountsOnLabel0( 0 ); - getMainPanel().getControlPanel().getSearchFoundCountsLabel0().setVisible( false ); - getMainPanel().getControlPanel().getSearchTextField0().setText( "" ); - getMainPanel().getControlPanel().getSearchResetButton0().setEnabled( false ); - getMainPanel().getControlPanel().getSearchResetButton0().setVisible( false ); - getMainPanel().getControlPanel().setSearchFoundCountsOnLabel1( 0 ); - getMainPanel().getControlPanel().getSearchFoundCountsLabel1().setVisible( false ); - getMainPanel().getControlPanel().getSearchTextField1().setText( "" ); - getMainPanel().getControlPanel().getSearchResetButton1().setEnabled( false ); - getMainPanel().getControlPanel().getSearchResetButton1().setVisible( false ); - } - - void setConfiguration( final Configuration configuration ) { - _configuration = configuration; - } - - void setCurrentDir( final File current_dir ) { - _current_dir = current_dir; - } - - void setInferenceManager( final InferenceManager i ) { - _inference_manager = i; - } - - void setOptions( final Options options ) { - _options = options; - } - - void setSelectedTypeInTypeMenu( final PHYLOGENY_GRAPHICS_TYPE type ) { - setTypeMenuToAllUnselected(); - switch ( type ) { - case CIRCULAR: - _circular_type_cbmi.setSelected( true ); - break; - case CONVEX: - _convex_type_cbmi.setSelected( true ); - break; - case CURVED: - _curved_type_cbmi.setSelected( true ); - break; - case EURO_STYLE: - _euro_type_cbmi.setSelected( true ); - break; - case ROUNDED: - _rounded_type_cbmi.setSelected( true ); - break; - case RECTANGULAR: - _rectangular_type_cbmi.setSelected( true ); - break; - case TRIANGULAR: - _triangular_type_cbmi.setSelected( true ); - break; - case UNROOTED: - _unrooted_type_cbmi.setSelected( true ); - break; - default: - throw new IllegalArgumentException( "unknown type: " + type ); + if ( AptxUtil.canWriteFormat( "bmp" ) ) { + _file_jmenu.add( _write_to_bmp_item = new JMenuItem( "Export to BMP file..." ) ); } + _file_jmenu.addSeparator(); + _file_jmenu.add( _print_item = new JMenuItem( "Print..." ) ); + _file_jmenu.addSeparator(); + _file_jmenu.add( _exit_item = new JMenuItem( "Exit" ) ); + customizeJMenuItem( _save_item ); + customizeJMenuItem( _write_to_pdf_item ); + customizeJMenuItem( _write_to_png_item ); + customizeJMenuItem( _write_to_jpg_item ); + customizeJMenuItem( _write_to_gif_item ); + customizeJMenuItem( _write_to_tif_item ); + customizeJMenuItem( _write_to_bmp_item ); + customizeJMenuItem( _print_item ); + customizeJMenuItem( _exit_item ); + _jmenubar.add( _file_jmenu ); } - final void setSpeciesTree( final Phylogeny species_tree ) { - _species_tree = species_tree; - } - - void setTypeMenuToAllUnselected() { - _convex_type_cbmi.setSelected( false ); - _curved_type_cbmi.setSelected( false ); - _euro_type_cbmi.setSelected( false ); - _rounded_type_cbmi.setSelected( false ); - _triangular_type_cbmi.setSelected( false ); - _rectangular_type_cbmi.setSelected( false ); - _unrooted_type_cbmi.setSelected( false ); - _circular_type_cbmi.setSelected( false ); + void buildFontSizeMenu() { + _font_size_menu = createMenu( FONT_SIZE_MENU_LABEL, getConfiguration() ); + _font_size_menu.add( _super_tiny_fonts_item = new JMenuItem( "Super Tiny Fonts" ) ); + _font_size_menu.add( _tiny_fonts_item = new JMenuItem( "Tiny Fonts" ) ); + _font_size_menu.add( _small_fonts_item = new JMenuItem( "Small Fonts" ) ); + _font_size_menu.add( _medium_fonts_item = new JMenuItem( "Medium Fonts" ) ); + _font_size_menu.add( _large_fonts_item = new JMenuItem( "Large Fonts" ) ); + customizeJMenuItem( _super_tiny_fonts_item ); + customizeJMenuItem( _tiny_fonts_item ); + customizeJMenuItem( _small_fonts_item ); + customizeJMenuItem( _medium_fonts_item ); + customizeJMenuItem( _large_fonts_item ); + _jmenubar.add( _font_size_menu ); } - void switchColors() { - final TreeColorSet colorset = _mainpanel.getTreeColorSet(); - final ColorSchemeChooser csc = new ColorSchemeChooser( getMainPanel(), colorset ); - csc.setVisible( true ); + void buildHelpMenu() { + _help_jmenu = createMenu( "Help", getConfiguration() ); + _help_jmenu.add( _help_item = new JMenuItem( "Documentation" ) ); + _help_jmenu.addSeparator(); + _help_jmenu.add( _website_item = new JMenuItem( "Archaeopteryx Home" ) ); + _aptx_ref_item = new JMenuItem( "Archaeopteryx Reference" ); //TODO need to add this... + _help_jmenu.add( _phyloxml_website_item = new JMenuItem( "phyloXML Home" ) ); + _help_jmenu.add( _phyloxml_ref_item = new JMenuItem( "phyloXML Reference" ) ); + _help_jmenu.addSeparator(); + _help_jmenu.add( _about_item = new JMenuItem( "About" ) ); + customizeJMenuItem( _help_item ); + customizeJMenuItem( _website_item ); + customizeJMenuItem( _phyloxml_website_item ); + customizeJMenuItem( _aptx_ref_item ); + customizeJMenuItem( _phyloxml_ref_item ); + customizeJMenuItem( _about_item ); + _phyloxml_ref_item.setToolTipText( PHYLOXML_REF_TOOL_TIP ); + _aptx_ref_item.setToolTipText( APTX_REF_TOOL_TIP ); + _jmenubar.add( _help_jmenu ); } - void taxColor() { - if ( _mainpanel.getCurrentTreePanel() != null ) { - _mainpanel.getCurrentTreePanel().taxColor(); - } + void buildTypeMenu() { + _type_menu = createMenu( TYPE_MENU_HEADER, getConfiguration() ); + _type_menu.add( _rectangular_type_cbmi = new JCheckBoxMenuItem( MainFrame.RECTANGULAR_TYPE_CBMI_LABEL ) ); + _type_menu.add( _euro_type_cbmi = new JCheckBoxMenuItem( MainFrame.EURO_TYPE_CBMI_LABEL ) ); + _type_menu.add( _rounded_type_cbmi = new JCheckBoxMenuItem( MainFrame.ROUNDED_TYPE_CBMI_LABEL ) ); + _type_menu.add( _curved_type_cbmi = new JCheckBoxMenuItem( MainFrame.CURVED_TYPE_CBMI_LABEL ) ); + _type_menu.add( _triangular_type_cbmi = new JCheckBoxMenuItem( MainFrame.TRIANGULAR_TYPE_CBMI_LABEL ) ); + _type_menu.add( _convex_type_cbmi = new JCheckBoxMenuItem( MainFrame.CONVEX_TYPE_CBMI_LABEL ) ); + _type_menu.add( _unrooted_type_cbmi = new JCheckBoxMenuItem( MainFrame.UNROOTED_TYPE_CBMI_LABEL ) ); + _type_menu.add( _circular_type_cbmi = new JCheckBoxMenuItem( MainFrame.CIRCULAR_TYPE_CBMI_LABEL ) ); + customizeCheckBoxMenuItem( _rectangular_type_cbmi, false ); + customizeCheckBoxMenuItem( _triangular_type_cbmi, false ); + customizeCheckBoxMenuItem( _euro_type_cbmi, false ); + customizeCheckBoxMenuItem( _rounded_type_cbmi, false ); + customizeCheckBoxMenuItem( _curved_type_cbmi, false ); + customizeCheckBoxMenuItem( _convex_type_cbmi, false ); + customizeCheckBoxMenuItem( _unrooted_type_cbmi, false ); + customizeCheckBoxMenuItem( _circular_type_cbmi, false ); + _unrooted_type_cbmi.setToolTipText( MainFrame.USE_MOUSEWHEEL_SHIFT_TO_ROTATE ); + _circular_type_cbmi.setToolTipText( MainFrame.USE_MOUSEWHEEL_SHIFT_TO_ROTATE ); + initializeTypeMenu( getOptions() ); + _jmenubar.add( _type_menu ); } - void typeChanged( final Object o ) { - updateTypeCheckboxes( getOptions(), o ); - updateOptions( getOptions() ); - if ( getCurrentTreePanel() != null ) { - final PHYLOGENY_GRAPHICS_TYPE previous_type = getCurrentTreePanel().getPhylogenyGraphicsType(); - final PHYLOGENY_GRAPHICS_TYPE new_type = getOptions().getPhylogenyGraphicsType(); - if ( ( ( previous_type == PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) && ( new_type != PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) ) - || ( ( previous_type == PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) && ( new_type != PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) ) - || ( ( previous_type != PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) && ( new_type == PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) ) - || ( ( previous_type != PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) && ( new_type == PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) ) ) { - getCurrentTreePanel().getControlPanel().showWhole(); - } - if ( getCurrentTreePanel().isPhyHasBranchLengths() && ( new_type != PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) ) { - getCurrentTreePanel().getControlPanel().setDrawPhylogramEnabled( true ); - } - else { - getCurrentTreePanel().getControlPanel().setDrawPhylogramEnabled( false ); - } - getCurrentTreePanel().setPhylogenyGraphicsType( getOptions().getPhylogenyGraphicsType() ); - updateScreenTextAntialias( getMainPanel().getTreePanels() ); - if ( getCurrentTreePanel().getControlPanel().getDynamicallyHideData() != null ) { - if ( new_type == PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) { - getCurrentTreePanel().getControlPanel().getDynamicallyHideData().setEnabled( false ); - } - else { - getCurrentTreePanel().getControlPanel().getDynamicallyHideData().setEnabled( true ); - } - } - } + void buildViewMenu() { + _view_jmenu = createMenu( "View", getConfiguration() ); + _view_jmenu.add( _display_basic_information_item = new JMenuItem( SHOW_BASIC_TREE_INFORMATION_LABEL ) ); + _view_jmenu.addSeparator(); + _view_jmenu.add( _view_as_XML_item = new JMenuItem( "as phyloXML" ) ); + _view_jmenu.add( _view_as_NH_item = new JMenuItem( "as Newick" ) ); + _view_jmenu.add( _view_as_nexus_item = new JMenuItem( "as Nexus" ) ); + customizeJMenuItem( _display_basic_information_item ); + customizeJMenuItem( _view_as_NH_item ); + customizeJMenuItem( _view_as_XML_item ); + customizeJMenuItem( _view_as_nexus_item ); + _jmenubar.add( _view_jmenu ); } - void updateOptions( final Options options ) { - options.setAntialiasScreen( ( _screen_antialias_cbmi != null ) && _screen_antialias_cbmi.isSelected() ); - options.setBackgroundColorGradient( ( _background_gradient_cbmi != null ) - && _background_gradient_cbmi.isSelected() ); - options.setShowDomainLabels( ( _show_domain_labels != null ) && _show_domain_labels.isSelected() ); - options.setShowAnnotationRefSource( ( _show_annotation_ref_source != null ) - && _show_annotation_ref_source.isSelected() ); - options.setAbbreviateScientificTaxonNames( ( _abbreviate_scientific_names != null ) - && _abbreviate_scientific_names.isSelected() ); - options.setColorLabelsSameAsParentBranch( ( _color_labels_same_as_parent_branch != null ) - && _color_labels_same_as_parent_branch.isSelected() ); - options.setShowDefaultNodeShapesInternal( ( _show_default_node_shapes_internal_cbmi != null ) - && _show_default_node_shapes_internal_cbmi.isSelected() ); - options.setShowDefaultNodeShapesExternal( ( _show_default_node_shapes_external_cbmi != null ) - && _show_default_node_shapes_external_cbmi.isSelected() ); - options.setShowDefaultNodeShapesForMarkedNodes( ( _show_default_node_shapes_for_marked_cbmi != null ) - && _show_default_node_shapes_for_marked_cbmi.isSelected() ); - if ( ( _non_lined_up_cladograms_rbmi != null ) && ( _non_lined_up_cladograms_rbmi.isSelected() ) ) { - options.setCladogramType( CLADOGRAM_TYPE.NON_LINED_UP ); - } - else if ( ( _uniform_cladograms_rbmi != null ) && ( _uniform_cladograms_rbmi.isSelected() ) ) { - options.setCladogramType( CLADOGRAM_TYPE.TOTAL_NODE_SUM_DEP ); - } - else if ( ( _ext_node_dependent_cladogram_rbmi != null ) && ( _ext_node_dependent_cladogram_rbmi.isSelected() ) ) { - options.setCladogramType( CLADOGRAM_TYPE.EXT_NODE_SUM_DEP ); - } - options.setSearchCaseSensitive( ( _search_case_senstive_cbmi != null ) - && _search_case_senstive_cbmi.isSelected() ); - if ( ( _show_scale_cbmi != null ) && _show_scale_cbmi.isEnabled() ) { - options.setShowScale( _show_scale_cbmi.isSelected() ); + void checkTextFrames() { + if ( _textframes.size() > 5 ) { + try { + if ( _textframes.getFirst() != null ) { + _textframes.getFirst().removeMe(); + } + else { + _textframes.removeFirst(); + } + } + catch ( final NoSuchElementException e ) { + // Ignore. + } } - if ( _label_direction_cbmi != null ) { - if ( _label_direction_cbmi.isSelected() ) { - options.setNodeLabelDirection( NODE_LABEL_DIRECTION.RADIAL ); + } + + void choosePdfWidth() { + final String s = ( String ) JOptionPane.showInputDialog( this, + "Please enter the default line width for PDF export.\n" + + "[current value: " + + getOptions().getPrintLineWidth() + "]\n", + "Line Width for PDF Export", + JOptionPane.QUESTION_MESSAGE, + null, + null, + getOptions().getPrintLineWidth() ); + if ( !ForesterUtil.isEmpty( s ) ) { + boolean success = true; + float f = 0.0f; + final String m_str = s.trim(); + if ( !ForesterUtil.isEmpty( m_str ) ) { + try { + f = Float.parseFloat( m_str ); + } + catch ( final Exception ex ) { + success = false; + } } else { - options.setNodeLabelDirection( NODE_LABEL_DIRECTION.HORIZONTAL ); + success = false; + } + if ( success && ( f > 0.0 ) ) { + getOptions().setPrintLineWidth( f ); } } - options.setShowOverview( ( _show_overview_cbmi != null ) && _show_overview_cbmi.isSelected() ); - options.setShowConfidenceStddev( ( _show_confidence_stddev_cbmi != null ) - && _show_confidence_stddev_cbmi.isSelected() ); - if ( ( _color_by_taxonomic_group_cbmi != null ) && _color_by_taxonomic_group_cbmi.isEnabled() ) { - options.setColorByTaxonomicGroup( _color_by_taxonomic_group_cbmi.isSelected() ); - } - options.setPrintUsingActualSize( ( _print_using_actual_size_cbmi != null ) - && ( _print_using_actual_size_cbmi.isSelected() ) ); - options.setGraphicsExportUsingActualSize( ( _graphics_export_using_actual_size_cbmi != null ) - && ( _graphics_export_using_actual_size_cbmi.isSelected() ) ); - options.setAntialiasPrint( ( _antialias_print_cbmi != null ) && _antialias_print_cbmi.isSelected() ); - if ( ( _use_brackets_for_conf_in_nh_export_cbmi != null ) - && _use_brackets_for_conf_in_nh_export_cbmi.isSelected() ) { - options.setNhConversionSupportValueStyle( NH_CONVERSION_SUPPORT_VALUE_STYLE.IN_SQUARE_BRACKETS ); - } - else if ( ( _use_internal_names_for_conf_in_nh_export_cbmi != null ) - && _use_internal_names_for_conf_in_nh_export_cbmi.isSelected() ) { - options.setNhConversionSupportValueStyle( NH_CONVERSION_SUPPORT_VALUE_STYLE.AS_INTERNAL_NODE_NAMES ); - } - else { - options.setNhConversionSupportValueStyle( NH_CONVERSION_SUPPORT_VALUE_STYLE.NONE ); - } - options.setPrintBlackAndWhite( ( _print_black_and_white_cbmi != null ) - && _print_black_and_white_cbmi.isSelected() ); - options.setInternalNumberAreConfidenceForNhParsing( ( _internal_number_are_confidence_for_nh_parsing_cbmi != null ) - && _internal_number_are_confidence_for_nh_parsing_cbmi.isSelected() ); - if ( ( _extract_taxonomy_pfam_strict_rbmi != null ) && _extract_taxonomy_pfam_strict_rbmi.isSelected() ) { - options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.PFAM_STYLE_STRICT ); - } - else if ( ( _extract_taxonomy_pfam_relaxed_rbmi != null ) && _extract_taxonomy_pfam_relaxed_rbmi.isSelected() ) { - options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.PFAM_STYLE_RELAXED ); - } - else if ( ( _extract_taxonomy_agressive_rbmi != null ) && _extract_taxonomy_agressive_rbmi.isSelected() ) { - options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.AGGRESSIVE ); - } - else if ( ( _extract_taxonomy_no_rbmi != null ) && _extract_taxonomy_no_rbmi.isSelected() ) { - options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.NO ); - } - options.setReplaceUnderscoresInNhParsing( ( _replace_underscores_cbmi != null ) - && _replace_underscores_cbmi.isSelected() ); - options.setAllowErrorsInDistanceToParent( ( _allow_errors_in_distance_to_parent_cbmi != null ) - && _allow_errors_in_distance_to_parent_cbmi.isSelected() ); - options.setMatchWholeTermsOnly( ( _search_whole_words_only_cbmi != null ) - && _search_whole_words_only_cbmi.isSelected() ); - options.setSearchWithRegex( ( _search_with_regex_cbmi != null ) && _search_with_regex_cbmi.isSelected() ); - options.setInverseSearchResult( ( _inverse_search_result_cbmi != null ) - && _inverse_search_result_cbmi.isSelected() ); - if ( _graphics_export_visible_only_cbmi != null ) { - options.setGraphicsExportVisibleOnly( _graphics_export_visible_only_cbmi.isSelected() ); - if ( _graphics_export_visible_only_cbmi.isSelected() && ( _graphics_export_using_actual_size_cbmi != null ) ) { - _graphics_export_using_actual_size_cbmi.setSelected( true ); - _graphics_export_using_actual_size_cbmi.setEnabled( false ); + } + + void choosePrintSize() { + final String s = ( String ) JOptionPane.showInputDialog( this, + "Please enter values for width and height,\nseparated by a comma.\n" + + "[current values: " + + getOptions().getPrintSizeX() + ", " + + getOptions().getPrintSizeY() + "]\n" + + "[A4: " + Constants.A4_SIZE_X + ", " + + Constants.A4_SIZE_Y + "]\n" + "[US Letter: " + + Constants.US_LETTER_SIZE_X + ", " + + Constants.US_LETTER_SIZE_Y + "]", + "Default Size for Graphics Export", + JOptionPane.QUESTION_MESSAGE, + null, + null, + getOptions().getPrintSizeX() + ", " + + getOptions().getPrintSizeY() ); + if ( !ForesterUtil.isEmpty( s ) && ( s.indexOf( ',' ) > 0 ) ) { + boolean success = true; + int x = 0; + int y = 0; + final String[] str_ary = s.split( "," ); + if ( str_ary.length == 2 ) { + final String x_str = str_ary[ 0 ].trim(); + final String y_str = str_ary[ 1 ].trim(); + if ( !ForesterUtil.isEmpty( x_str ) && !ForesterUtil.isEmpty( y_str ) ) { + try { + x = Integer.parseInt( x_str ); + y = Integer.parseInt( y_str ); + } + catch ( final Exception ex ) { + success = false; + } + } + else { + success = false; + } } else { - _graphics_export_using_actual_size_cbmi.setEnabled( true ); + success = false; + } + if ( success && ( x > 1 ) && ( y > 1 ) ) { + getOptions().setPrintSizeX( x ); + getOptions().setPrintSizeY( y ); } } - if ( ( _rectangular_type_cbmi != null ) && _rectangular_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.RECTANGULAR ); - } - else if ( ( _triangular_type_cbmi != null ) && _triangular_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.TRIANGULAR ); - } - else if ( ( _curved_type_cbmi != null ) && _curved_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.CURVED ); - } - else if ( ( _convex_type_cbmi != null ) && _convex_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.CONVEX ); - } - else if ( ( _euro_type_cbmi != null ) && _euro_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.EURO_STYLE ); - } - else if ( ( _rounded_type_cbmi != null ) && _rounded_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.ROUNDED ); - } - else if ( ( _unrooted_type_cbmi != null ) && _unrooted_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.UNROOTED ); + } + + void close() { + removeAllTextFrames(); + if ( _mainpanel != null ) { + _mainpanel.terminate(); } - else if ( ( _circular_type_cbmi != null ) && _circular_type_cbmi.isSelected() ) { - options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ); + if ( _contentpane != null ) { + _contentpane.removeAll(); } - if ( ( _right_line_up_domains_cbmi != null ) && _right_line_up_domains_cbmi.isEnabled() ) { - options.setRightLineUpDomains( _right_line_up_domains_cbmi.isSelected() ); + setVisible( false ); + dispose(); + } + + void colorRank() { + if ( _mainpanel.getCurrentTreePanel() != null ) { + final String[] ranks = AptxUtil.getAllPossibleRanks(); + final String rank = ( String ) JOptionPane + .showInputDialog( this, + "What rank should the colorization be based on", + "Rank Selection", + JOptionPane.QUESTION_MESSAGE, + null, + ranks, + null ); + if ( !ForesterUtil.isEmpty( rank ) ) { + _mainpanel.getCurrentTreePanel().colorRank( rank ); + } } - if ( ( _line_up_renderable_data_cbmi != null ) && _line_up_renderable_data_cbmi.isEnabled() ) { - options.setLineUpRendarableNodeData( _line_up_renderable_data_cbmi.isSelected() ); + } + + void confColor() { + if ( _mainpanel.getCurrentTreePanel() != null ) { + _mainpanel.getCurrentTreePanel().confColor(); } } - void updateTypeCheckboxes( final Options options, final Object o ) { - setTypeMenuToAllUnselected(); - ( ( JCheckBoxMenuItem ) o ).setSelected( true ); + void customizeCheckBoxMenuItem( final JCheckBoxMenuItem item, final boolean is_selected ) { + if ( item != null ) { + item.setFont( MainFrame.menu_font ); + if ( !getConfiguration().isUseNativeUI() ) { + item.setBackground( getConfiguration().getGuiMenuBackgroundColor() ); + item.setForeground( getConfiguration().getGuiMenuTextColor() ); + } + item.setSelected( is_selected ); + item.addActionListener( this ); + } } - void viewAsNexus() { - if ( ( _mainpanel.getCurrentPhylogeny() != null ) && !_mainpanel.getCurrentPhylogeny().isEmpty() ) { - String title = "Nexus"; - if ( !ForesterUtil.isEmpty( _mainpanel.getCurrentPhylogeny().getName() ) ) { - title = "\"" + getMainPanel().getCurrentPhylogeny().getName() + "\" in " + title; + JMenuItem customizeJMenuItem( final JMenuItem jmi ) { + if ( jmi != null ) { + jmi.setFont( MainFrame.menu_font ); + if ( !getConfiguration().isUseNativeUI() ) { + jmi.setBackground( getConfiguration().getGuiMenuBackgroundColor() ); + jmi.setForeground( getConfiguration().getGuiMenuTextColor() ); } - showTextFrame( _mainpanel.getCurrentPhylogeny().toNexus( getOptions().getNhConversionSupportValueStyle() ), - title ); + jmi.addActionListener( this ); } + return jmi; } - void viewAsNH() { - if ( ( _mainpanel.getCurrentPhylogeny() != null ) && !_mainpanel.getCurrentPhylogeny().isEmpty() ) { - String title = "New Hampshire"; - if ( !ForesterUtil.isEmpty( _mainpanel.getCurrentPhylogeny().getName() ) ) { - title = "\"" + getMainPanel().getCurrentPhylogeny().getName() + "\" in " + title; + void customizeRadioButtonMenuItem( final JRadioButtonMenuItem item, final boolean is_selected ) { + if ( item != null ) { + item.setFont( MainFrame.menu_font ); + if ( !getConfiguration().isUseNativeUI() ) { + item.setBackground( getConfiguration().getGuiMenuBackgroundColor() ); + item.setForeground( getConfiguration().getGuiMenuTextColor() ); } - showTextFrame( _mainpanel.getCurrentPhylogeny().toNewHampshire( getOptions() - .getNhConversionSupportValueStyle() ), - title ); + item.setSelected( is_selected ); + item.addActionListener( this ); } } - void viewAsXML() { + void displayBasicInformation( final File treefile ) { if ( ( _mainpanel.getCurrentPhylogeny() != null ) && !_mainpanel.getCurrentPhylogeny().isEmpty() ) { - String title = "phyloXML"; + String title = "Basic Information"; if ( !ForesterUtil.isEmpty( _mainpanel.getCurrentPhylogeny().getName() ) ) { - title = "\"" + getMainPanel().getCurrentPhylogeny().getName() + "\" in " + title; + title = title + " for \"" + _mainpanel.getCurrentPhylogeny().getName() + "\""; } - showTextFrame( _mainpanel.getCurrentPhylogeny().toPhyloXML( 0 ), title ); + showTextFrame( AptxUtil.createBasicInformation( _mainpanel.getCurrentPhylogeny(), treefile ), title ); + } + } + + void exceptionOccuredDuringOpenFile( final Exception e ) { + try { + _mainpanel.getCurrentTreePanel().setArrowCursor(); + } + catch ( final Exception ex ) { + // Do nothing. } + JOptionPane.showMessageDialog( this, + ForesterUtil.wordWrap( e.getLocalizedMessage(), 80 ), + "Error during File|Open", + JOptionPane.ERROR_MESSAGE ); } - boolean writeAsNewHampshire( final Phylogeny t, boolean exception, final File file ) { - try { - final PhylogenyWriter writer = new PhylogenyWriter(); - writer.toNewHampshire( t, true, getOptions().getNhConversionSupportValueStyle(), file ); + void executeGSDI() { + if ( !isOKforSDI( false, true ) ) { + return; } - catch ( final Exception e ) { - exception = true; - exceptionOccuredDuringSaveAs( e ); + if ( !_mainpanel.getCurrentPhylogeny().isRooted() ) { + JOptionPane.showMessageDialog( this, + "Gene tree is not rooted.", + "Cannot execute GSDI", + JOptionPane.ERROR_MESSAGE ); + return; } - return exception; - } - - boolean writeAsNexus( final Phylogeny t, boolean exception, final File file ) { + final Phylogeny gene_tree = _mainpanel.getCurrentPhylogeny().copy(); + gene_tree.setAllNodesToNotCollapse(); + gene_tree.recalculateNumberOfExternalDescendants( false ); + GSDI gsdi = null; + final Phylogeny species_tree = getSpeciesTree().copy(); try { - final PhylogenyWriter writer = new PhylogenyWriter(); - writer.toNexus( file, t, getOptions().getNhConversionSupportValueStyle() ); + gsdi = new GSDI( gene_tree, species_tree, false, true, true, true ); + } + catch ( final SDIException e ) { + JOptionPane.showMessageDialog( this, + e.getLocalizedMessage(), + "Error during GSDI", + JOptionPane.ERROR_MESSAGE ); + return; } catch ( final Exception e ) { - exception = true; - exceptionOccuredDuringSaveAs( e ); + AptxUtil.unexpectedException( e ); + return; } - return exception; - } - - boolean writeAsPhyloXml( final Phylogeny t, boolean exception, final File file ) { - try { - final PhylogenyWriter writer = new PhylogenyWriter(); - writer.toPhyloXML( file, t, 0 ); + gene_tree.setRerootable( false ); + gene_tree.clearHashIdToNodeMap(); + gene_tree.recalculateNumberOfExternalDescendants( true ); + _mainpanel.addPhylogenyInNewTab( gene_tree, getConfiguration(), "gene tree", null ); + getMainPanel().getControlPanel().setShowEvents( true ); + showWhole(); + final int selected = _mainpanel.getTabbedPane().getSelectedIndex(); + _mainpanel.addPhylogenyInNewTab( species_tree, getConfiguration(), "species tree", null ); + showWhole(); + _mainpanel.getTabbedPane().setSelectedIndex( selected ); + showWhole(); + _mainpanel.getCurrentTreePanel().setEdited( true ); + final int poly = PhylogenyMethods.countNumberOfPolytomies( species_tree ); + if ( gsdi.getStrippedExternalGeneTreeNodes().size() > 0 ) { + JOptionPane.showMessageDialog( this, + "Duplications: " + gsdi.getDuplicationsSum() + "\n" + + "Potential duplications: " + + gsdi.getSpeciationOrDuplicationEventsSum() + "\n" + + "Speciations: " + gsdi.getSpeciationsSum() + "\n" + + "Stripped gene tree nodes: " + + gsdi.getStrippedExternalGeneTreeNodes().size() + "\n" + + "Taxonomy linkage based on: " + gsdi.getTaxCompBase() + "\n" + + "Number of polytomies in species tree used: " + poly + "\n", + "GSDI successfully completed", + JOptionPane.WARNING_MESSAGE ); } - catch ( final Exception e ) { - exception = true; - exceptionOccuredDuringSaveAs( e ); + else { + JOptionPane.showMessageDialog( this, + "Duplications: " + gsdi.getDuplicationsSum() + "\n" + + "Potential duplications: " + + gsdi.getSpeciationOrDuplicationEventsSum() + "\n" + + "Speciations: " + gsdi.getSpeciationsSum() + "\n" + + "Stripped gene tree nodes: " + + gsdi.getStrippedExternalGeneTreeNodes().size() + "\n" + + "Taxonomy linkage based on: " + gsdi.getTaxCompBase() + "\n" + + "Number of polytomies in species tree used: " + poly + "\n", + "GSDI successfully completed", + JOptionPane.INFORMATION_MESSAGE ); } - return exception; } - void writePhylogenyToGraphicsFile( final String file_name, final GraphicsExportType type ) { - _mainpanel.getCurrentTreePanel().calcParametersForPainting( _mainpanel.getCurrentTreePanel().getWidth(), - _mainpanel.getCurrentTreePanel().getHeight() ); - String file_written_to = ""; - boolean error = false; - try { - file_written_to = AptxUtil.writePhylogenyToGraphicsFile( file_name, - _mainpanel.getCurrentTreePanel().getWidth(), - _mainpanel.getCurrentTreePanel().getHeight(), - _mainpanel.getCurrentTreePanel(), - _mainpanel.getControlPanel(), - type, - getOptions() ); + void executeGSDIR() { + if ( !isOKforSDI( false, false ) ) { + return; } - catch ( final IOException e ) { - error = true; - JOptionPane.showMessageDialog( this, e.getMessage(), "Error", JOptionPane.ERROR_MESSAGE ); + final int p = PhylogenyMethods.countNumberOfPolytomies( _mainpanel.getCurrentPhylogeny() ); + if ( ( p > 0 ) + && !( ( p == 1 ) && ( _mainpanel.getCurrentPhylogeny().getRoot().getNumberOfDescendants() == 3 ) ) ) { + JOptionPane.showMessageDialog( this, + "Gene tree is not completely binary", + "Cannot execute GSDI", + JOptionPane.ERROR_MESSAGE ); + return; } - if ( !error ) { - if ( ( file_written_to != null ) && ( file_written_to.length() > 0 ) ) { - JOptionPane.showMessageDialog( this, - "Wrote image to: " + file_written_to, - "Graphics Export", - JOptionPane.INFORMATION_MESSAGE ); - } - else { - JOptionPane.showMessageDialog( this, - "There was an unknown problem when attempting to write to an image file: \"" - + file_name + "\"", - "Error", - JOptionPane.ERROR_MESSAGE ); - } + final Phylogeny gene_tree = _mainpanel.getCurrentPhylogeny().copy(); + gene_tree.setAllNodesToNotCollapse(); + gene_tree.recalculateNumberOfExternalDescendants( false ); + GSDIR gsdir = null; + final Phylogeny species_tree = getSpeciesTree().copy(); + try { + gsdir = new GSDIR( gene_tree, species_tree, true, true, true ); } - _contentpane.repaint(); - } - - void writeToFile( final Phylogeny t ) { - if ( t == null ) { + catch ( final SDIException e ) { + JOptionPane.showMessageDialog( this, + e.getLocalizedMessage(), + "Error during GSDIR", + JOptionPane.ERROR_MESSAGE ); return; } - String initial_filename = null; - if ( getMainPanel().getCurrentTreePanel().getTreeFile() != null ) { - try { - initial_filename = getMainPanel().getCurrentTreePanel().getTreeFile().getCanonicalPath(); - } - catch ( final IOException e ) { - initial_filename = null; - } + catch ( final Exception e ) { + AptxUtil.unexpectedException( e ); + return; } - if ( !ForesterUtil.isEmpty( initial_filename ) ) { - _save_filechooser.setSelectedFile( new File( initial_filename ) ); + final Phylogeny result_gene_tree = gsdir.getMinDuplicationsSumGeneTree(); + result_gene_tree.setRerootable( false ); + result_gene_tree.clearHashIdToNodeMap(); + result_gene_tree.recalculateNumberOfExternalDescendants( true ); + PhylogenyMethods.orderAppearance( result_gene_tree.getRoot(), true, true, DESCENDANT_SORT_PRIORITY.NODE_NAME ); + _mainpanel.addPhylogenyInNewTab( result_gene_tree, getConfiguration(), "gene tree", null ); + getMainPanel().getControlPanel().setShowEvents( true ); + showWhole(); + final int selected = _mainpanel.getTabbedPane().getSelectedIndex(); + _mainpanel.addPhylogenyInNewTab( species_tree, getConfiguration(), "species tree", null ); + showWhole(); + _mainpanel.getTabbedPane().setSelectedIndex( selected ); + showWhole(); + _mainpanel.getCurrentTreePanel().setEdited( true ); + final int poly = PhylogenyMethods.countNumberOfPolytomies( species_tree ); + if ( gsdir.getStrippedExternalGeneTreeNodes().size() > 0 ) { + JOptionPane.showMessageDialog( this, + "Minimal duplications: " + gsdir.getMinDuplicationsSum() + "\n" + + "Speciations: " + gsdir.getSpeciationsSum() + "\n" + + "Stripped gene tree nodes: " + + gsdir.getStrippedExternalGeneTreeNodes().size() + "\n" + + "Taxonomy linkage based on: " + gsdir.getTaxCompBase() + "\n" + + "Number of polytomies in species tree used: " + poly + "\n", + "GSDIR successfully completed", + JOptionPane.WARNING_MESSAGE ); } else { - _save_filechooser.setSelectedFile( new File( "" ) ); + JOptionPane.showMessageDialog( this, + "Minimal duplications: " + gsdir.getMinDuplicationsSum() + "\n" + + "Speciations: " + gsdir.getSpeciationsSum() + "\n" + + "Stripped gene tree nodes: " + + gsdir.getStrippedExternalGeneTreeNodes().size() + "\n" + + "Taxonomy linkage based on: " + gsdir.getTaxCompBase() + "\n" + + "Number of polytomies in species tree used: " + poly + "\n", + "GSDIR successfully completed", + JOptionPane.INFORMATION_MESSAGE ); } - final File my_dir = getCurrentDir(); - if ( my_dir != null ) { - _save_filechooser.setCurrentDirectory( my_dir ); + } + + boolean GAndSDoHaveMoreThanOneSpeciesInComman( final Phylogeny gene_tree ) { + if ( ( gene_tree == null ) || gene_tree.isEmpty() ) { + JOptionPane.showMessageDialog( this, + "Gene tree and species tree have no species in common.", + "Error during SDI", + JOptionPane.ERROR_MESSAGE ); + return false; } - final int result = _save_filechooser.showSaveDialog( _contentpane ); - final File file = _save_filechooser.getSelectedFile(); - setCurrentDir( _save_filechooser.getCurrentDirectory() ); - boolean exception = false; - if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { - if ( file.exists() ) { - final int i = JOptionPane.showConfirmDialog( this, - file + " already exists.\nOverwrite?", - "Overwrite?", - JOptionPane.OK_CANCEL_OPTION, - JOptionPane.QUESTION_MESSAGE ); - if ( i != JOptionPane.OK_OPTION ) { - return; + else if ( gene_tree.getNumberOfExternalNodes() < 2 ) { + JOptionPane.showMessageDialog( this, + "Gene tree and species tree have only one species in common.", + "Error during SDI", + JOptionPane.ERROR_MESSAGE ); + return false; + } + else { + return true; + } + } + + ControlPanel getControlPanel() { + return getMainPanel().getControlPanel(); + } + + File getCurrentDir() { + if ( ( _current_dir == null ) || !_current_dir.canRead() ) { + if ( ForesterUtil.isWindows() ) { + try { + _current_dir = new File( WindowsUtils.getCurrentUserDesktopPath() ); } - else { - final File to = new File( file.getAbsoluteFile().toString() + Constants.BACKUP_FILE_SUFFIX ); - try { - ForesterUtil.copyFile( file, to ); - } - catch ( final Exception e ) { - JOptionPane.showMessageDialog( this, - "Failed to create backup copy " + to, - "Failed to Create Backup Copy", - JOptionPane.WARNING_MESSAGE ); - } - try { - file.delete(); - } - catch ( final Exception e ) { - JOptionPane.showMessageDialog( this, - "Failed to delete: " + file, - "Failed to Delete", - JOptionPane.WARNING_MESSAGE ); - } + catch ( final Exception e ) { + _current_dir = null; } } - if ( _save_filechooser.getFileFilter() == MainFrame.nhfilter ) { - exception = writeAsNewHampshire( t, exception, file ); - } - else if ( _save_filechooser.getFileFilter() == MainFrame.xmlfilter ) { - exception = writeAsPhyloXml( t, exception, file ); - } - else if ( _save_filechooser.getFileFilter() == MainFrame.nexusfilter ) { - exception = writeAsNexus( t, exception, file ); - } - // "*.*": - else { - final String file_name = file.getName().trim().toLowerCase(); - if ( file_name.endsWith( ".nh" ) || file_name.endsWith( ".newick" ) || file_name.endsWith( ".phy" ) - || file_name.endsWith( ".tree" ) ) { - exception = writeAsNewHampshire( t, exception, file ); - } - else if ( file_name.endsWith( ".nex" ) || file_name.endsWith( ".nexus" ) ) { - exception = writeAsNexus( t, exception, file ); - } - // XML is default: - else { - exception = writeAsPhyloXml( t, exception, file ); - } + } + if ( ( _current_dir == null ) || !_current_dir.canRead() ) { + if ( System.getProperty( "user.home" ) != null ) { + _current_dir = new File( System.getProperty( "user.home" ) ); } - if ( !exception ) { - getMainPanel().setTitleOfSelectedTab( file.getName() ); - getMainPanel().getCurrentTreePanel().setTreeFile( file ); - getMainPanel().getCurrentTreePanel().setEdited( false ); + else if ( System.getProperty( "user.dir" ) != null ) { + _current_dir = new File( System.getProperty( "user.dir" ) ); } } + return _current_dir; } - void writeToGraphicsFile( final Phylogeny t, final GraphicsExportType type ) { - if ( ( t == null ) || t.isEmpty() ) { - return; - } - String initial_filename = ""; - if ( getMainPanel().getCurrentTreePanel().getTreeFile() != null ) { - initial_filename = getMainPanel().getCurrentTreePanel().getTreeFile().toString(); - } - if ( initial_filename.indexOf( '.' ) > 0 ) { - initial_filename = initial_filename.substring( 0, initial_filename.lastIndexOf( '.' ) ); - } - initial_filename = initial_filename + "." + type; - _writetographics_filechooser.setSelectedFile( new File( initial_filename ) ); - final File my_dir = getCurrentDir(); - if ( my_dir != null ) { - _writetographics_filechooser.setCurrentDirectory( my_dir ); - } - final int result = _writetographics_filechooser.showSaveDialog( _contentpane ); - File file = _writetographics_filechooser.getSelectedFile(); - setCurrentDir( _writetographics_filechooser.getCurrentDirectory() ); - if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { - if ( !file.toString().toLowerCase().endsWith( type.toString() ) ) { - file = new File( file.toString() + "." + type ); - } - if ( file.exists() ) { - final int i = JOptionPane.showConfirmDialog( this, - file + " already exists. Overwrite?", - "Warning", - JOptionPane.OK_CANCEL_OPTION, - JOptionPane.WARNING_MESSAGE ); - if ( i != JOptionPane.OK_OPTION ) { - return; - } - else { - try { - file.delete(); - } - catch ( final Exception e ) { - JOptionPane.showMessageDialog( this, - "Failed to delete: " + file, - "Error", - JOptionPane.WARNING_MESSAGE ); - } - } - } - writePhylogenyToGraphicsFile( file.toString(), type ); + TreePanel getCurrentTreePanel() { + return getMainPanel().getCurrentTreePanel(); + } + + JMenu getHelpMenu() { + return _help_jmenu; + } + + JCheckBoxMenuItem getlabelDirectionCbmi() { + return _label_direction_cbmi; + } + + JMenuBar getMenuBarOfMainFrame() { + return _jmenubar; + } + + final Phylogeny getSpeciesTree() { + return _species_tree; + } + + void initializeTypeMenu( final Options options ) { + setTypeMenuToAllUnselected(); + switch ( options.getPhylogenyGraphicsType() ) { + case CONVEX: + _convex_type_cbmi.setSelected( true ); + break; + case CURVED: + _curved_type_cbmi.setSelected( true ); + break; + case EURO_STYLE: + _euro_type_cbmi.setSelected( true ); + break; + case ROUNDED: + _rounded_type_cbmi.setSelected( true ); + break; + case TRIANGULAR: + _triangular_type_cbmi.setSelected( true ); + break; + case UNROOTED: + _unrooted_type_cbmi.setSelected( true ); + break; + case CIRCULAR: + _circular_type_cbmi.setSelected( true ); + break; + default: + _rectangular_type_cbmi.setSelected( true ); + break; } } - void writeToPdf( final Phylogeny t ) { - if ( ( t == null ) || t.isEmpty() ) { - return; + boolean isOKforSDI( final boolean species_tree_has_to_binary, final boolean gene_tree_has_to_binary ) { + if ( ( _mainpanel.getCurrentPhylogeny() == null ) || _mainpanel.getCurrentPhylogeny().isEmpty() ) { + return false; } - String initial_filename = ""; - if ( getMainPanel().getCurrentTreePanel().getTreeFile() != null ) { - initial_filename = getMainPanel().getCurrentTreePanel().getTreeFile().toString(); + else if ( ( getSpeciesTree() == null ) || getSpeciesTree().isEmpty() ) { + JOptionPane.showMessageDialog( this, + "No species tree loaded", + "Cannot execute GSDI", + JOptionPane.ERROR_MESSAGE ); + return false; } - if ( initial_filename.indexOf( '.' ) > 0 ) { - initial_filename = initial_filename.substring( 0, initial_filename.lastIndexOf( '.' ) ); + else if ( species_tree_has_to_binary && !getSpeciesTree().isCompletelyBinary() ) { + JOptionPane.showMessageDialog( this, + "Species tree is not completely binary", + "Cannot execute GSDI", + JOptionPane.ERROR_MESSAGE ); + return false; } - initial_filename = initial_filename + ".pdf"; - _writetopdf_filechooser.setSelectedFile( new File( initial_filename ) ); - final File my_dir = getCurrentDir(); - if ( my_dir != null ) { - _writetopdf_filechooser.setCurrentDirectory( my_dir ); + else if ( gene_tree_has_to_binary && !_mainpanel.getCurrentPhylogeny().isCompletelyBinary() ) { + JOptionPane.showMessageDialog( this, + "Gene tree is not completely binary", + "Cannot execute GSDI", + JOptionPane.ERROR_MESSAGE ); + return false; } - final int result = _writetopdf_filechooser.showSaveDialog( _contentpane ); - File file = _writetopdf_filechooser.getSelectedFile(); - setCurrentDir( _writetopdf_filechooser.getCurrentDirectory() ); - if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { - if ( !file.toString().toLowerCase().endsWith( ".pdf" ) ) { - file = new File( file.toString() + ".pdf" ); - } - if ( file.exists() ) { - final int i = JOptionPane.showConfirmDialog( this, - file + " already exists. Overwrite?", - "WARNING", - JOptionPane.OK_CANCEL_OPTION, - JOptionPane.WARNING_MESSAGE ); - if ( i != JOptionPane.OK_OPTION ) { - return; - } - } - printPhylogenyToPdf( file.toString() ); + else { + return true; } } - private void annotateSequences() { + boolean isSubtreeDisplayed() { if ( getCurrentTreePanel() != null ) { - List nodes = null; - if ( ( getCurrentTreePanel().getFoundNodes0() != null ) - || ( getCurrentTreePanel().getFoundNodes1() != null ) ) { - nodes = getCurrentTreePanel().getFoundNodesAsListOfPhylogenyNodes(); - } - if ( ( nodes == null ) || nodes.isEmpty() ) { + if ( getCurrentTreePanel().isCurrentTreeIsSubtree() ) { JOptionPane .showMessageDialog( this, - "Need to select nodes, either via direct selection or via the \"Search\" function", - "No nodes selected for annotation", - JOptionPane.ERROR_MESSAGE ); - return; + "This operation can only be performed on a complete tree, not on the currently displayed sub-tree only.", + "Operation can not be exectuted on a sub-tree", + JOptionPane.WARNING_MESSAGE ); + return true; } - final Phylogeny phy = getMainPanel().getCurrentPhylogeny(); - if ( ( phy != null ) && !phy.isEmpty() ) { - final JTextField ref_field = new JTextField( 10 ); - final JTextField desc_filed = new JTextField( 20 ); - ref_field.setText( ForesterUtil.isEmpty( getPreviousNodeAnnotationReference() ) ? "" - : getPreviousNodeAnnotationReference() ); - final JPanel my_panel = new JPanel(); - my_panel.add( new JLabel( "Reference " ) ); - my_panel.add( ref_field ); - my_panel.add( Box.createHorizontalStrut( 15 ) ); - my_panel.add( new JLabel( "Description " ) ); - my_panel.add( desc_filed ); - final int result = JOptionPane.showConfirmDialog( null, - my_panel, - "Enter the sequence annotation(s) for the " - + nodes.size() + " selected nodes", - JOptionPane.OK_CANCEL_OPTION ); - if ( result == JOptionPane.OK_OPTION ) { - String ref = ref_field.getText(); - String desc = desc_filed.getText(); - if ( !ForesterUtil.isEmpty( ref ) ) { - ref = ref.trim(); - ref = ref.replaceAll( "\\s+", " " ); - if ( ( ref.indexOf( ':' ) < 1 ) || ( ref.indexOf( ':' ) > ( ref.length() - 2 ) ) - || ( ref.length() < 3 ) ) { - JOptionPane.showMessageDialog( this, - "Reference needs to be in the form of \"GO:1234567\"", - "Illegal Format for Annotation Reference", - JOptionPane.ERROR_MESSAGE ); - return; - } - } - if ( ref != null ) { - setPreviousNodeAnnotationReference( ref ); - } - if ( desc != null ) { - desc = desc.trim(); - desc = desc.replaceAll( "\\s+", " " ); - } - if ( !ForesterUtil.isEmpty( ref ) || !ForesterUtil.isEmpty( desc ) ) { - for( final PhylogenyNode n : nodes ) { - ForesterUtil.ensurePresenceOfSequence( n ); - final Annotation ann = ForesterUtil.isEmpty( ref ) ? new Annotation() - : new Annotation( ref ); - if ( !ForesterUtil.isEmpty( desc ) ) { - ann.setDesc( desc ); - } - n.getNodeData().getSequence().addAnnotation( ann ); - } - } - getMainPanel().getControlPanel().showAnnotations(); - } + } + return false; + } + + void midpointRoot() { + if ( _mainpanel.getCurrentTreePanel() != null ) { + _mainpanel.getCurrentTreePanel().midpointRoot(); + } + } + + void readPhylogeniesFromWebservice( final int i ) { + final UrlTreeReader reader = new UrlTreeReader( this, i ); + new Thread( reader ).start(); + } + + void removeAllTextFrames() { + for( final TextFrame tf : _textframes ) { + if ( tf != null ) { + tf.close(); } } + _textframes.clear(); + } + + void resetSearch() { + getMainPanel().getCurrentTreePanel().setFoundNodes0( null ); + getMainPanel().getCurrentTreePanel().setFoundNodes1( null ); + getMainPanel().getControlPanel().setSearchFoundCountsOnLabel0( 0 ); + getMainPanel().getControlPanel().getSearchFoundCountsLabel0().setVisible( false ); + getMainPanel().getControlPanel().getSearchTextField0().setText( "" ); + getMainPanel().getControlPanel().getSearchResetButton0().setEnabled( false ); + getMainPanel().getControlPanel().getSearchResetButton0().setVisible( false ); + getMainPanel().getControlPanel().setSearchFoundCountsOnLabel1( 0 ); + getMainPanel().getControlPanel().getSearchFoundCountsLabel1().setVisible( false ); + getMainPanel().getControlPanel().getSearchTextField1().setText( "" ); + getMainPanel().getControlPanel().getSearchResetButton1().setEnabled( false ); + getMainPanel().getControlPanel().getSearchResetButton1().setVisible( false ); + } + + void setConfiguration( final Configuration configuration ) { + _configuration = configuration; + } + + void setCurrentDir( final File current_dir ) { + _current_dir = current_dir; + } + + void setInferenceManager( final InferenceManager i ) { + _inference_manager = i; + } + + void setOptions( final Options options ) { + _options = options; + } + + void setSelectedTypeInTypeMenu( final PHYLOGENY_GRAPHICS_TYPE type ) { + setTypeMenuToAllUnselected(); + switch ( type ) { + case CIRCULAR: + _circular_type_cbmi.setSelected( true ); + break; + case CONVEX: + _convex_type_cbmi.setSelected( true ); + break; + case CURVED: + _curved_type_cbmi.setSelected( true ); + break; + case EURO_STYLE: + _euro_type_cbmi.setSelected( true ); + break; + case ROUNDED: + _rounded_type_cbmi.setSelected( true ); + break; + case RECTANGULAR: + _rectangular_type_cbmi.setSelected( true ); + break; + case TRIANGULAR: + _triangular_type_cbmi.setSelected( true ); + break; + case UNROOTED: + _unrooted_type_cbmi.setSelected( true ); + break; + default: + throw new IllegalArgumentException( "unknown type: " + type ); + } } - private void chooseFont() { - final FontChooser fc = new FontChooser(); - fc.setFont( getMainPanel().getTreeFontSet().getLargeFont() ); - fc.showDialog( this, "Select the Base Font" ); - getMainPanel().getTreeFontSet().setBaseFont( fc.getFont() ); + final void setSpeciesTree( final Phylogeny species_tree ) { + _species_tree = species_tree; } - private void chooseMinimalConfidence() { - final String s = ( String ) JOptionPane - .showInputDialog( this, - "Please enter the minimum for confidence values to be displayed.\n" - + "[current value: " + getOptions().getMinConfidenceValue() + "]\n", - "Minimal Confidence Value", - JOptionPane.QUESTION_MESSAGE, - null, - null, - getOptions().getMinConfidenceValue() ); - if ( !ForesterUtil.isEmpty( s ) ) { - boolean success = true; - double m = 0.0; - final String m_str = s.trim(); - if ( !ForesterUtil.isEmpty( m_str ) ) { - try { - m = Double.parseDouble( m_str ); - } - catch ( final Exception ex ) { - success = false; - } + void setTypeMenuToAllUnselected() { + _convex_type_cbmi.setSelected( false ); + _curved_type_cbmi.setSelected( false ); + _euro_type_cbmi.setSelected( false ); + _rounded_type_cbmi.setSelected( false ); + _triangular_type_cbmi.setSelected( false ); + _rectangular_type_cbmi.setSelected( false ); + _unrooted_type_cbmi.setSelected( false ); + _circular_type_cbmi.setSelected( false ); + } + + void switchColors() { + final TreeColorSet colorset = _mainpanel.getTreeColorSet(); + final ColorSchemeChooser csc = new ColorSchemeChooser( getMainPanel(), colorset ); + csc.setVisible( true ); + } + + void taxColor() { + if ( _mainpanel.getCurrentTreePanel() != null ) { + _mainpanel.getCurrentTreePanel().taxColor(); + } + } + + void typeChanged( final Object o ) { + updateTypeCheckboxes( getOptions(), o ); + updateOptions( getOptions() ); + if ( getCurrentTreePanel() != null ) { + final PHYLOGENY_GRAPHICS_TYPE previous_type = getCurrentTreePanel().getPhylogenyGraphicsType(); + final PHYLOGENY_GRAPHICS_TYPE new_type = getOptions().getPhylogenyGraphicsType(); + if ( ( ( previous_type == PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) && ( new_type != PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) ) + || ( ( previous_type == PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) && ( new_type != PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) ) + || ( ( previous_type != PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) && ( new_type == PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) ) + || ( ( previous_type != PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) && ( new_type == PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) ) ) { + getCurrentTreePanel().getControlPanel().showWhole(); + } + if ( getCurrentTreePanel().isPhyHasBranchLengths() && ( new_type != PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ) ) { + getCurrentTreePanel().getControlPanel().setDrawPhylogramEnabled( true ); } else { - success = false; + getCurrentTreePanel().getControlPanel().setDrawPhylogramEnabled( false ); } - if ( success && ( m >= 0.0 ) ) { - getOptions().setMinConfidenceValue( m ); + getCurrentTreePanel().setPhylogenyGraphicsType( getOptions().getPhylogenyGraphicsType() ); + updateScreenTextAntialias( getMainPanel().getTreePanels() ); + if ( getCurrentTreePanel().getControlPanel().getDynamicallyHideData() != null ) { + if ( new_type == PHYLOGENY_GRAPHICS_TYPE.UNROOTED ) { + getCurrentTreePanel().getControlPanel().getDynamicallyHideData().setEnabled( false ); + } + else { + getCurrentTreePanel().getControlPanel().getDynamicallyHideData().setEnabled( true ); + } } } } - private void deleteSelectedNodes( final boolean delete ) { - final Phylogeny phy = getMainPanel().getCurrentPhylogeny(); - if ( ( phy == null ) || ( phy.getNumberOfExternalNodes() < 2 ) ) { - return; - } - final List nodes = new ArrayList(); - if ( ( getCurrentTreePanel().getFoundNodes0() != null ) || ( getCurrentTreePanel().getFoundNodes1() != null ) ) { - final List all_selected_nodes = getCurrentTreePanel().getFoundNodesAsListOfPhylogenyNodes(); - for( final PhylogenyNode n : all_selected_nodes ) { - if ( n.isExternal() ) { - nodes.add( n ); - } - } - } - String function = "Retain"; - if ( delete ) { - function = "Delete"; + void updateOptions( final Options options ) { + options.setAntialiasScreen( ( _screen_antialias_cbmi != null ) && _screen_antialias_cbmi.isSelected() ); + options.setBackgroundColorGradient( ( _background_gradient_cbmi != null ) + && _background_gradient_cbmi.isSelected() ); + options.setShowDomainLabels( ( _show_domain_labels != null ) && _show_domain_labels.isSelected() ); + options.setShowAnnotationRefSource( ( _show_annotation_ref_source != null ) + && _show_annotation_ref_source.isSelected() ); + options.setAbbreviateScientificTaxonNames( ( _abbreviate_scientific_names != null ) + && _abbreviate_scientific_names.isSelected() ); + options.setColorLabelsSameAsParentBranch( ( _color_labels_same_as_parent_branch != null ) + && _color_labels_same_as_parent_branch.isSelected() ); + options.setShowDefaultNodeShapesInternal( ( _show_default_node_shapes_internal_cbmi != null ) + && _show_default_node_shapes_internal_cbmi.isSelected() ); + options.setShowDefaultNodeShapesExternal( ( _show_default_node_shapes_external_cbmi != null ) + && _show_default_node_shapes_external_cbmi.isSelected() ); + options.setShowDefaultNodeShapesForMarkedNodes( ( _show_default_node_shapes_for_marked_cbmi != null ) + && _show_default_node_shapes_for_marked_cbmi.isSelected() ); + if ( ( _non_lined_up_cladograms_rbmi != null ) && ( _non_lined_up_cladograms_rbmi.isSelected() ) ) { + options.setCladogramType( CLADOGRAM_TYPE.NON_LINED_UP ); } - if ( ( nodes == null ) || nodes.isEmpty() ) { - JOptionPane - .showMessageDialog( this, - "Need to select external nodes, either via direct selection or via the \"Search\" function", - "No external nodes selected to " + function.toLowerCase(), - JOptionPane.ERROR_MESSAGE ); - return; + else if ( ( _uniform_cladograms_rbmi != null ) && ( _uniform_cladograms_rbmi.isSelected() ) ) { + options.setCladogramType( CLADOGRAM_TYPE.TOTAL_NODE_SUM_DEP ); } - final int todo = nodes.size(); - final int ext = phy.getNumberOfExternalNodes(); - int res = todo; - if ( delete ) { - res = ext - todo; + else if ( ( _ext_node_dependent_cladogram_rbmi != null ) && ( _ext_node_dependent_cladogram_rbmi.isSelected() ) ) { + options.setCladogramType( CLADOGRAM_TYPE.EXT_NODE_SUM_DEP ); } - if ( res < 1 ) { - JOptionPane.showMessageDialog( this, - "Cannot delete all nodes", - "Attempt to delete all nodes ", - JOptionPane.ERROR_MESSAGE ); - return; + options.setSearchCaseSensitive( ( _search_case_senstive_cbmi != null ) + && _search_case_senstive_cbmi.isSelected() ); + if ( ( _show_scale_cbmi != null ) && _show_scale_cbmi.isEnabled() ) { + options.setShowScale( _show_scale_cbmi.isSelected() ); } - final int result = JOptionPane.showConfirmDialog( null, function + " " + todo - + " external node(s), from a total of " + ext + " external nodes," + "\nresulting in tree with " + res - + " nodes?", function + " external nodes", JOptionPane.OK_CANCEL_OPTION ); - if ( result == JOptionPane.OK_OPTION ) { - if ( !delete ) { - final List to_delete = new ArrayList(); - for( final PhylogenyNodeIterator it = phy.iteratorExternalForward(); it.hasNext(); ) { - final PhylogenyNode n = it.next(); - if ( !nodes.contains( n ) ) { - to_delete.add( n ); - } - } - for( final PhylogenyNode n : to_delete ) { - phy.deleteSubtree( n, true ); - } + if ( _label_direction_cbmi != null ) { + if ( _label_direction_cbmi.isSelected() ) { + options.setNodeLabelDirection( NODE_LABEL_DIRECTION.RADIAL ); } else { - for( final PhylogenyNode n : nodes ) { - phy.deleteSubtree( n, true ); - } + options.setNodeLabelDirection( NODE_LABEL_DIRECTION.HORIZONTAL ); } - resetSearch(); - getCurrentTreePanel().setNodeInPreorderToNull(); - phy.externalNodesHaveChanged(); - phy.clearHashIdToNodeMap(); - phy.recalculateNumberOfExternalDescendants( true ); - getCurrentTreePanel().resetNodeIdToDistToLeafMap(); - getCurrentTreePanel().setEdited( true ); - repaint(); } - } - - private void doUpdateProcessMenu() { - if ( _process_pool.size() > 0 ) { - if ( _process_menu == null ) { - _process_menu = createMenu( "", getConfiguration() ); - _process_menu.setForeground( Color.RED ); + options.setShowOverview( ( _show_overview_cbmi != null ) && _show_overview_cbmi.isSelected() ); + options.setShowConfidenceStddev( ( _show_confidence_stddev_cbmi != null ) + && _show_confidence_stddev_cbmi.isSelected() ); + if ( ( _color_by_taxonomic_group_cbmi != null ) && _color_by_taxonomic_group_cbmi.isEnabled() ) { + options.setColorByTaxonomicGroup( _color_by_taxonomic_group_cbmi.isSelected() ); + } + options.setPrintUsingActualSize( ( _print_using_actual_size_cbmi != null ) + && ( _print_using_actual_size_cbmi.isSelected() ) ); + options.setGraphicsExportUsingActualSize( ( _graphics_export_using_actual_size_cbmi != null ) + && ( _graphics_export_using_actual_size_cbmi.isSelected() ) ); + options.setAntialiasPrint( ( _antialias_print_cbmi != null ) && _antialias_print_cbmi.isSelected() ); + if ( ( _use_brackets_for_conf_in_nh_export_cbmi != null ) + && _use_brackets_for_conf_in_nh_export_cbmi.isSelected() ) { + options.setNhConversionSupportValueStyle( NH_CONVERSION_SUPPORT_VALUE_STYLE.IN_SQUARE_BRACKETS ); + } + else if ( ( _use_internal_names_for_conf_in_nh_export_cbmi != null ) + && _use_internal_names_for_conf_in_nh_export_cbmi.isSelected() ) { + options.setNhConversionSupportValueStyle( NH_CONVERSION_SUPPORT_VALUE_STYLE.AS_INTERNAL_NODE_NAMES ); + } + else { + options.setNhConversionSupportValueStyle( NH_CONVERSION_SUPPORT_VALUE_STYLE.NONE ); + } + options.setPrintBlackAndWhite( ( _print_black_and_white_cbmi != null ) + && _print_black_and_white_cbmi.isSelected() ); + options.setInternalNumberAreConfidenceForNhParsing( ( _internal_number_are_confidence_for_nh_parsing_cbmi != null ) + && _internal_number_are_confidence_for_nh_parsing_cbmi.isSelected() ); + if ( ( _extract_taxonomy_pfam_strict_rbmi != null ) && _extract_taxonomy_pfam_strict_rbmi.isSelected() ) { + options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.PFAM_STYLE_STRICT ); + } + else if ( ( _extract_taxonomy_pfam_relaxed_rbmi != null ) && _extract_taxonomy_pfam_relaxed_rbmi.isSelected() ) { + options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.PFAM_STYLE_RELAXED ); + } + else if ( ( _extract_taxonomy_agressive_rbmi != null ) && _extract_taxonomy_agressive_rbmi.isSelected() ) { + options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.AGGRESSIVE ); + } + else if ( ( _extract_taxonomy_no_rbmi != null ) && _extract_taxonomy_no_rbmi.isSelected() ) { + options.setTaxonomyExtraction( TAXONOMY_EXTRACTION.NO ); + } + options.setReplaceUnderscoresInNhParsing( ( _replace_underscores_cbmi != null ) + && _replace_underscores_cbmi.isSelected() ); + options.setAllowErrorsInDistanceToParent( ( _allow_errors_in_distance_to_parent_cbmi != null ) + && _allow_errors_in_distance_to_parent_cbmi.isSelected() ); + options.setMatchWholeTermsOnly( ( _search_whole_words_only_cbmi != null ) + && _search_whole_words_only_cbmi.isSelected() ); + options.setSearchWithRegex( ( _search_with_regex_cbmi != null ) && _search_with_regex_cbmi.isSelected() ); + options.setInverseSearchResult( ( _inverse_search_result_cbmi != null ) + && _inverse_search_result_cbmi.isSelected() ); + if ( _graphics_export_visible_only_cbmi != null ) { + options.setGraphicsExportVisibleOnly( _graphics_export_visible_only_cbmi.isSelected() ); + if ( _graphics_export_visible_only_cbmi.isSelected() && ( _graphics_export_using_actual_size_cbmi != null ) ) { + _graphics_export_using_actual_size_cbmi.setSelected( true ); + _graphics_export_using_actual_size_cbmi.setEnabled( false ); } - _process_menu.removeAll(); - final String text = "processes running: " + _process_pool.size(); - _process_menu.setText( text ); - _jmenubar.add( _process_menu ); - for( int i = 0; i < _process_pool.size(); ++i ) { - final ProcessRunning p = _process_pool.getProcessByIndex( i ); - _process_menu.add( customizeJMenuItem( new JMenuItem( p.getName() + " [" + p.getStart() + "]" ) ) ); + else { + _graphics_export_using_actual_size_cbmi.setEnabled( true ); } } - else { - if ( _process_menu != null ) { - _process_menu.removeAll(); - _jmenubar.remove( _process_menu ); - } + if ( ( _rectangular_type_cbmi != null ) && _rectangular_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.RECTANGULAR ); } - _jmenubar.validate(); - _jmenubar.repaint(); - repaint(); - } - - private String getPreviousNodeAnnotationReference() { - return _previous_node_annotation_ref; - } - - static void print( final TreePanel tp, - final Options op, - - final Component c - ) { - if ( ( tp == null ) || ( tp.getPhylogeny() == null ) - || tp.getPhylogeny().isEmpty() ) { - return; + else if ( ( _triangular_type_cbmi != null ) && _triangular_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.TRIANGULAR ); } - if ( !op.isPrintUsingActualSize() ) { - tp.calcParametersForPainting( op.getPrintSizeX() - 80, - op.getPrintSizeY() - 140 ); - tp.resetPreferredSize(); - tp.repaint(); + else if ( ( _curved_type_cbmi != null ) && _curved_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.CURVED ); } - final String job_name = Constants.PRG_NAME; - boolean error = false; - String printer_name = null; - try { - printer_name = Printer.print( tp, job_name ); + else if ( ( _convex_type_cbmi != null ) && _convex_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.CONVEX ); } - catch ( final Exception e ) { - error = true; - JOptionPane.showMessageDialog( c, e.getMessage(), "Printing Error", JOptionPane.ERROR_MESSAGE ); + else if ( ( _euro_type_cbmi != null ) && _euro_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.EURO_STYLE ); } - if ( !error && ( printer_name != null ) ) { - String msg = "Printing data sent to printer"; - if ( printer_name.length() > 1 ) { - msg += " [" + printer_name + "]"; - } - JOptionPane.showMessageDialog( c, msg, "Printing...", JOptionPane.INFORMATION_MESSAGE ); + else if ( ( _rounded_type_cbmi != null ) && _rounded_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.ROUNDED ); } - if ( !op.isPrintUsingActualSize() ) { - tp.getControlPanel().showWhole(); + else if ( ( _unrooted_type_cbmi != null ) && _unrooted_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.UNROOTED ); + } + else if ( ( _circular_type_cbmi != null ) && _circular_type_cbmi.isSelected() ) { + options.setPhylogenyGraphicsType( PHYLOGENY_GRAPHICS_TYPE.CIRCULAR ); + } + if ( ( _right_line_up_domains_cbmi != null ) && _right_line_up_domains_cbmi.isEnabled() ) { + options.setRightLineUpDomains( _right_line_up_domains_cbmi.isSelected() ); + } + if ( ( _line_up_renderable_data_cbmi != null ) && _line_up_renderable_data_cbmi.isEnabled() ) { + options.setLineUpRendarableNodeData( _line_up_renderable_data_cbmi.isSelected() ); } } - private void removeBranchColors() { - if ( getMainPanel().getCurrentPhylogeny() != null ) { - AptxUtil.removeBranchColors( getMainPanel().getCurrentPhylogeny() ); - } + void updateTypeCheckboxes( final Options options, final Object o ) { + setTypeMenuToAllUnselected(); + ( ( JCheckBoxMenuItem ) o ).setSelected( true ); } - private void removeVisualStyles() { - if ( getMainPanel().getCurrentPhylogeny() != null ) { - AptxUtil.removeVisualStyles( getMainPanel().getCurrentPhylogeny() ); + void viewAsNexus() { + if ( ( _mainpanel.getCurrentPhylogeny() != null ) && !_mainpanel.getCurrentPhylogeny().isEmpty() ) { + String title = "Nexus"; + if ( !ForesterUtil.isEmpty( _mainpanel.getCurrentPhylogeny().getName() ) ) { + title = "\"" + getMainPanel().getCurrentPhylogeny().getName() + "\" in " + title; + } + showTextFrame( _mainpanel.getCurrentPhylogeny().toNexus( getOptions().getNhConversionSupportValueStyle() ), + title ); } } - private void setPreviousNodeAnnotationReference( final String previous_node_annotation_ref ) { - _previous_node_annotation_ref = previous_node_annotation_ref; + void viewAsNH() { + if ( ( _mainpanel.getCurrentPhylogeny() != null ) && !_mainpanel.getCurrentPhylogeny().isEmpty() ) { + String title = "New Hampshire"; + if ( !ForesterUtil.isEmpty( _mainpanel.getCurrentPhylogeny().getName() ) ) { + title = "\"" + getMainPanel().getCurrentPhylogeny().getName() + "\" in " + title; + } + showTextFrame( _mainpanel.getCurrentPhylogeny().toNewHampshire( getOptions() + .getNhConversionSupportValueStyle() ), + title ); + } } - private void writeAllToFile() { - if ( ( getMainPanel().getTabbedPane() == null ) || ( getMainPanel().getTabbedPane().getTabCount() < 1 ) ) { - return; - } - final File my_dir = getCurrentDir(); - if ( my_dir != null ) { - _save_filechooser.setCurrentDirectory( my_dir ); + void viewAsXML() { + if ( ( _mainpanel.getCurrentPhylogeny() != null ) && !_mainpanel.getCurrentPhylogeny().isEmpty() ) { + String title = "phyloXML"; + if ( !ForesterUtil.isEmpty( _mainpanel.getCurrentPhylogeny().getName() ) ) { + title = "\"" + getMainPanel().getCurrentPhylogeny().getName() + "\" in " + title; + } + showTextFrame( _mainpanel.getCurrentPhylogeny().toPhyloXML( 0 ), title ); } - _save_filechooser.setSelectedFile( new File( "" ) ); - final int result = _save_filechooser.showSaveDialog( _contentpane ); - final File file = _save_filechooser.getSelectedFile(); - setCurrentDir( _save_filechooser.getCurrentDirectory() ); - if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { - if ( file.exists() ) { - final int i = JOptionPane.showConfirmDialog( this, - file + " already exists. Overwrite?", - "Warning", - JOptionPane.OK_CANCEL_OPTION, - JOptionPane.WARNING_MESSAGE ); - if ( i != JOptionPane.OK_OPTION ) { - return; + } + + private static void cycleNodeDataReturn( final Options op, final Configuration conf ) { + switch ( op.getExtDescNodeDataToReturn() ) { + case UNKNOWN: + op.setExtDescNodeDataToReturn( NodeDataField.DOMAINS_ALL ); + break; + case DOMAINS_ALL: + op.setExtDescNodeDataToReturn( NodeDataField.DOMAINS_COLLAPSED_PER_PROTEIN ); + break; + case DOMAINS_COLLAPSED_PER_PROTEIN: + op.setExtDescNodeDataToReturn( NodeDataField.SEQ_ANNOTATIONS ); + break; + case SEQ_ANNOTATIONS: + op.setExtDescNodeDataToReturn( NodeDataField.GO_TERM_IDS ); + break; + case GO_TERM_IDS: + op.setExtDescNodeDataToReturn( NodeDataField.SEQUENCE_MOL_SEQ_FASTA ); + break; + case SEQUENCE_MOL_SEQ_FASTA: + if ( ( conf != null ) && ( conf.getExtDescNodeDataToReturn() != null ) + && ( conf.getExtDescNodeDataToReturn() != NodeDataField.DOMAINS_ALL ) + && ( conf.getExtDescNodeDataToReturn() != NodeDataField.DOMAINS_COLLAPSED_PER_PROTEIN ) + && ( conf.getExtDescNodeDataToReturn() != NodeDataField.SEQ_ANNOTATIONS ) + && ( conf.getExtDescNodeDataToReturn() != NodeDataField.GO_TERM_IDS ) + && ( conf.getExtDescNodeDataToReturn() != NodeDataField.SEQUENCE_MOL_SEQ_FASTA ) ) { + op.setExtDescNodeDataToReturn( conf.getExtDescNodeDataToReturn() ); } else { - try { - file.delete(); - } - catch ( final Exception e ) { - JOptionPane.showMessageDialog( this, - "Failed to delete: " + file, - "Error", - JOptionPane.WARNING_MESSAGE ); - } - } - } - final int count = getMainPanel().getTabbedPane().getTabCount(); - final List trees = new ArrayList(); - for( int i = 0; i < count; ++i ) { - final Phylogeny phy = getMainPanel().getPhylogeny( i ); - if ( ForesterUtil.isEmpty( phy.getName() ) - && !ForesterUtil.isEmpty( getMainPanel().getTabbedPane().getTitleAt( i ) ) ) { - phy.setName( getMainPanel().getTabbedPane().getTitleAt( i ) ); + op.setExtDescNodeDataToReturn( NodeDataField.UNKNOWN ); } - trees.add( phy ); - getMainPanel().getTreePanels().get( i ).setEdited( false ); - } - final PhylogenyWriter writer = new PhylogenyWriter(); - try { - writer.toPhyloXML( file, trees, 0, ForesterUtil.LINE_SEPARATOR ); - } - catch ( final IOException e ) { - JOptionPane.showMessageDialog( this, - "Failed to write to: " + file, - "Error", - JOptionPane.WARNING_MESSAGE ); - } + break; + default: + op.setExtDescNodeDataToReturn( NodeDataField.UNKNOWN ); } } @@ -2363,38 +2125,125 @@ public abstract class MainFrame extends JFrame implements ActionListener { } } - static void cycleNodeShape( final Options op ) { - switch ( op.getDefaultNodeShape() ) { - case CIRCLE: - op.setDefaultNodeShape( NodeShape.RECTANGLE ); - break; - case RECTANGLE: - op.setDefaultNodeShape( NodeShape.CIRCLE ); - break; - default: - throw new RuntimeException( "unknown shape: " + op.getDefaultNodeShape() ); + static void cycleNodeShape( final Options op ) { + switch ( op.getDefaultNodeShape() ) { + case CIRCLE: + op.setDefaultNodeShape( NodeShape.RECTANGLE ); + break; + case RECTANGLE: + op.setDefaultNodeShape( NodeShape.CIRCLE ); + break; + default: + throw new RuntimeException( "unknown shape: " + op.getDefaultNodeShape() ); + } + } + + static void cycleOverview( final Options op, final TreePanel tree_panel ) { + switch ( op.getOvPlacement() ) { + case LOWER_LEFT: + op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.UPPER_LEFT ); + break; + case LOWER_RIGHT: + op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.LOWER_LEFT ); + break; + case UPPER_LEFT: + op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.UPPER_RIGHT ); + break; + case UPPER_RIGHT: + op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.LOWER_RIGHT ); + break; + default: + throw new RuntimeException( "unknown placement: " + op.getOvPlacement() ); + } + if ( tree_panel != null ) { + tree_panel.updateOvSettings(); + } + } + + static void exceptionOccuredDuringSaveAs( final Exception e, final TreePanel tp, final Component comp ) { + try { + tp.setArrowCursor(); + } + catch ( final Exception ex ) { + // Do nothing. + } + JOptionPane.showMessageDialog( comp, "Exception" + e, "Error during File|SaveAs", JOptionPane.ERROR_MESSAGE ); + } + + static void print( final TreePanel tp, final Options op, final Component c ) { + if ( ( tp == null ) || ( tp.getPhylogeny() == null ) || tp.getPhylogeny().isEmpty() ) { + return; + } + if ( !op.isPrintUsingActualSize() ) { + tp.calcParametersForPainting( op.getPrintSizeX() - 80, op.getPrintSizeY() - 140 ); + tp.resetPreferredSize(); + tp.repaint(); + } + final String job_name = Constants.PRG_NAME; + boolean error = false; + String printer_name = null; + try { + printer_name = Printer.print( tp, job_name ); + } + catch ( final Exception e ) { + error = true; + JOptionPane.showMessageDialog( c, e.getMessage(), "Printing Error", JOptionPane.ERROR_MESSAGE ); + } + if ( !error && ( printer_name != null ) ) { + String msg = "Printing data sent to printer"; + if ( printer_name.length() > 1 ) { + msg += " [" + printer_name + "]"; + } + JOptionPane.showMessageDialog( c, msg, "Printing...", JOptionPane.INFORMATION_MESSAGE ); + } + if ( !op.isPrintUsingActualSize() ) { + tp.getControlPanel().showWhole(); + } + } + + static void printPhylogenyToPdf( final String file_name, + final Options opts, + final TreePanel tp, + final Component comp ) { + if ( !opts.isPrintUsingActualSize() ) { + tp.calcParametersForPainting( opts.getPrintSizeX(), opts.getPrintSizeY() ); + tp.resetPreferredSize(); + tp.repaint(); } - } - - static void cycleOverview( final Options op, final TreePanel tree_panel ) { - switch ( op.getOvPlacement() ) { - case LOWER_LEFT: - op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.UPPER_LEFT ); - break; - case LOWER_RIGHT: - op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.LOWER_LEFT ); - break; - case UPPER_LEFT: - op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.UPPER_RIGHT ); - break; - case UPPER_RIGHT: - op.setOvPlacement( Options.OVERVIEW_PLACEMENT_TYPE.LOWER_RIGHT ); - break; - default: - throw new RuntimeException( "unknown placement: " + op.getOvPlacement() ); + String pdf_written_to = ""; + boolean error = false; + try { + if ( opts.isPrintUsingActualSize() ) { + pdf_written_to = PdfExporter.writePhylogenyToPdf( file_name, tp, tp.getWidth(), tp.getHeight() ); + } + else { + pdf_written_to = PdfExporter.writePhylogenyToPdf( file_name, + tp, + opts.getPrintSizeX(), + opts.getPrintSizeY() ); + } } - if ( tree_panel != null ) { - tree_panel.updateOvSettings(); + catch ( final IOException e ) { + error = true; + JOptionPane.showMessageDialog( comp, e.getMessage(), "Error", JOptionPane.ERROR_MESSAGE ); + } + if ( !error ) { + if ( !ForesterUtil.isEmpty( pdf_written_to ) ) { + JOptionPane.showMessageDialog( comp, + "Wrote PDF to: " + pdf_written_to, + "Information", + JOptionPane.INFORMATION_MESSAGE ); + } + else { + JOptionPane.showMessageDialog( comp, + "There was an unknown problem when attempting to write to PDF file: \"" + + file_name + "\"", + "Error", + JOptionPane.ERROR_MESSAGE ); + } + } + if ( !opts.isPrintUsingActualSize() ) { + tp.getControlPanel().showWhole(); } } @@ -2482,39 +2331,281 @@ public abstract class MainFrame extends JFrame implements ActionListener { } } - private static void cycleNodeDataReturn( final Options op, final Configuration conf ) { - switch ( op.getExtDescNodeDataToReturn() ) { - case UNKNOWN: - op.setExtDescNodeDataToReturn( NodeDataField.DOMAINS_ALL ); - break; - case DOMAINS_ALL: - op.setExtDescNodeDataToReturn( NodeDataField.DOMAINS_COLLAPSED_PER_PROTEIN ); - break; - case DOMAINS_COLLAPSED_PER_PROTEIN: - op.setExtDescNodeDataToReturn( NodeDataField.SEQ_ANNOTATIONS ); - break; - case SEQ_ANNOTATIONS: - op.setExtDescNodeDataToReturn( NodeDataField.GO_TERM_IDS ); - break; - case GO_TERM_IDS: - op.setExtDescNodeDataToReturn( NodeDataField.SEQUENCE_MOL_SEQ_FASTA ); - break; - case SEQUENCE_MOL_SEQ_FASTA: - if ( ( conf != null ) && ( conf.getExtDescNodeDataToReturn() != null ) - && ( conf.getExtDescNodeDataToReturn() != NodeDataField.DOMAINS_ALL ) - && ( conf.getExtDescNodeDataToReturn() != NodeDataField.DOMAINS_COLLAPSED_PER_PROTEIN ) - && ( conf.getExtDescNodeDataToReturn() != NodeDataField.SEQ_ANNOTATIONS ) - && ( conf.getExtDescNodeDataToReturn() != NodeDataField.GO_TERM_IDS ) - && ( conf.getExtDescNodeDataToReturn() != NodeDataField.SEQUENCE_MOL_SEQ_FASTA ) ) { - op.setExtDescNodeDataToReturn( conf.getExtDescNodeDataToReturn() ); + static boolean writeAsNewHampshire( final TreePanel tp, final Options op, boolean exception, final File file ) { + try { + final PhylogenyWriter writer = new PhylogenyWriter(); + writer.toNewHampshire( tp.getPhylogeny(), true, op.getNhConversionSupportValueStyle(), file ); + } + catch ( final Exception e ) { + exception = true; + exceptionOccuredDuringSaveAs( e, tp, tp ); + } + return exception; + } + + static boolean writeAsNexus( final TreePanel tp, final Options op, boolean exception, final File file ) { + try { + final PhylogenyWriter writer = new PhylogenyWriter(); + writer.toNexus( file, tp.getPhylogeny(), op.getNhConversionSupportValueStyle() ); + } + catch ( final Exception e ) { + exception = true; + exceptionOccuredDuringSaveAs( e, tp, tp ); + } + return exception; + } + + static boolean writeAsPhyloXml( final TreePanel tp, final Options op, boolean exception, final File file ) { + try { + final PhylogenyWriter writer = new PhylogenyWriter(); + writer.toPhyloXML( file, tp.getPhylogeny(), 0 ); + } + catch ( final Exception e ) { + exception = true; + exceptionOccuredDuringSaveAs( e, tp, tp ); + } + return exception; + } + + static void writePhylogenyToGraphicsFile( final String file_name, + final GraphicsExportType type, + final MainPanel mp, + final Component comp, + final Container contentpane ) { + mp.getCurrentTreePanel().calcParametersForPainting( mp.getCurrentTreePanel().getWidth(), + mp.getCurrentTreePanel().getHeight() ); + String file_written_to = ""; + boolean error = false; + try { + file_written_to = AptxUtil.writePhylogenyToGraphicsFile( file_name, + mp.getCurrentTreePanel().getWidth(), + mp.getCurrentTreePanel().getHeight(), + mp.getCurrentTreePanel(), + mp.getControlPanel(), + type, + mp.getOptions() ); + } + catch ( final IOException e ) { + error = true; + JOptionPane.showMessageDialog( comp, e.getMessage(), "Error", JOptionPane.ERROR_MESSAGE ); + } + if ( !error ) { + if ( ( file_written_to != null ) && ( file_written_to.length() > 0 ) ) { + JOptionPane.showMessageDialog( comp, + "Wrote image to: " + file_written_to, + "Graphics Export", + JOptionPane.INFORMATION_MESSAGE ); + } + else { + JOptionPane.showMessageDialog( comp, + "There was an unknown problem when attempting to write to an image file: \"" + + file_name + "\"", + "Error", + JOptionPane.ERROR_MESSAGE ); + } + } + contentpane.repaint(); + } + + static File writeToFile( final Phylogeny t, + final MainPanel mp, + final JFileChooser save_filechooser, + final File current_dir, + final Container contentpane, + final Component comp ) { + File new_file = null; + if ( t == null ) { + return null; + } + String initial_filename = null; + if ( mp.getCurrentTreePanel().getTreeFile() != null ) { + try { + initial_filename = mp.getCurrentTreePanel().getTreeFile().getCanonicalPath(); + } + catch ( final IOException e ) { + initial_filename = null; + } + } + if ( !ForesterUtil.isEmpty( initial_filename ) ) { + save_filechooser.setSelectedFile( new File( initial_filename ) ); + } + else { + save_filechooser.setSelectedFile( new File( "" ) ); + } + final File my_dir = current_dir; + if ( my_dir != null ) { + save_filechooser.setCurrentDirectory( my_dir ); + } + final int result = save_filechooser.showSaveDialog( contentpane ); + final File file = save_filechooser.getSelectedFile(); + new_file = save_filechooser.getCurrentDirectory(); + boolean exception = false; + if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { + if ( file.exists() ) { + final int i = JOptionPane.showConfirmDialog( comp, + file + " already exists.\nOverwrite?", + "Overwrite?", + JOptionPane.OK_CANCEL_OPTION, + JOptionPane.QUESTION_MESSAGE ); + if ( i != JOptionPane.OK_OPTION ) { + return null; } else { - op.setExtDescNodeDataToReturn( NodeDataField.UNKNOWN ); + final File to = new File( file.getAbsoluteFile().toString() + Constants.BACKUP_FILE_SUFFIX ); + try { + ForesterUtil.copyFile( file, to ); + } + catch ( final Exception e ) { + JOptionPane.showMessageDialog( comp, + "Failed to create backup copy " + to, + "Failed to Create Backup Copy", + JOptionPane.WARNING_MESSAGE ); + } + try { + file.delete(); + } + catch ( final Exception e ) { + JOptionPane.showMessageDialog( comp, + "Failed to delete: " + file, + "Failed to Delete", + JOptionPane.WARNING_MESSAGE ); + } } - break; - default: - op.setExtDescNodeDataToReturn( NodeDataField.UNKNOWN ); + } + if ( save_filechooser.getFileFilter() == MainFrame.nhfilter ) { + exception = writeAsNewHampshire( mp.getCurrentTreePanel(), mp.getOptions(), exception, file ); + } + else if ( save_filechooser.getFileFilter() == MainFrame.xmlfilter ) { + exception = writeAsPhyloXml( mp.getCurrentTreePanel(), mp.getOptions(), exception, file ); + } + else if ( save_filechooser.getFileFilter() == MainFrame.nexusfilter ) { + exception = writeAsNexus( mp.getCurrentTreePanel(), mp.getOptions(), exception, file ); + } + // "*.*": + else { + final String file_name = file.getName().trim().toLowerCase(); + if ( file_name.endsWith( ".nh" ) || file_name.endsWith( ".newick" ) || file_name.endsWith( ".phy" ) + || file_name.endsWith( ".tree" ) ) { + exception = writeAsNewHampshire( mp.getCurrentTreePanel(), mp.getOptions(), exception, file ); + } + else if ( file_name.endsWith( ".nex" ) || file_name.endsWith( ".nexus" ) ) { + exception = writeAsNexus( mp.getCurrentTreePanel(), mp.getOptions(), exception, file ); + } + // XML is default: + else { + exception = writeAsPhyloXml( mp.getCurrentTreePanel(), mp.getOptions(), exception, file ); + } + } + if ( !exception ) { + mp.setTitleOfSelectedTab( file.getName() ); + mp.getCurrentTreePanel().setTreeFile( file ); + mp.getCurrentTreePanel().setEdited( false ); + } + } + return new_file; + } + + static File writeToGraphicsFile( final Phylogeny t, + final GraphicsExportType type, + final MainPanel mp, + final JFileChooser writetographics_filechooser, + final Component component, + final Container contentpane, + final File current_dir ) { + File new_dir = null; + if ( ( t == null ) || t.isEmpty() ) { + return null; + } + String initial_filename = ""; + if ( mp.getCurrentTreePanel().getTreeFile() != null ) { + initial_filename = mp.getCurrentTreePanel().getTreeFile().toString(); + } + if ( initial_filename.indexOf( '.' ) > 0 ) { + initial_filename = initial_filename.substring( 0, initial_filename.lastIndexOf( '.' ) ); + } + initial_filename = initial_filename + "." + type; + writetographics_filechooser.setSelectedFile( new File( initial_filename ) ); + final File my_dir = current_dir; + if ( my_dir != null ) { + writetographics_filechooser.setCurrentDirectory( my_dir ); + } + final int result = writetographics_filechooser.showSaveDialog( contentpane ); + File file = writetographics_filechooser.getSelectedFile(); + //setCurrentDir( writetographics_filechooser.getCurrentDirectory() ); + new_dir = writetographics_filechooser.getCurrentDirectory(); + if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { + if ( !file.toString().toLowerCase().endsWith( type.toString() ) ) { + file = new File( file.toString() + "." + type ); + } + if ( file.exists() ) { + final int i = JOptionPane.showConfirmDialog( component, + file + " already exists. Overwrite?", + "Warning", + JOptionPane.OK_CANCEL_OPTION, + JOptionPane.WARNING_MESSAGE ); + if ( i != JOptionPane.OK_OPTION ) { + return null; + } + else { + try { + file.delete(); + } + catch ( final Exception e ) { + JOptionPane.showMessageDialog( component, + "Failed to delete: " + file, + "Error", + JOptionPane.WARNING_MESSAGE ); + } + } + } + writePhylogenyToGraphicsFile( file.toString(), type, mp, component, contentpane ); + } + return new_dir; + } + + static File writeToPdf( final Phylogeny t, + final MainPanel mp, + final JFileChooser writetopdf_filechooser, + final File curr_dir, + final Container contentpane, + final Component component ) { + if ( ( t == null ) || t.isEmpty() ) { + return null; + } + String initial_filename = ""; + if ( mp.getCurrentTreePanel().getTreeFile() != null ) { + initial_filename = mp.getCurrentTreePanel().getTreeFile().toString(); + } + if ( initial_filename.indexOf( '.' ) > 0 ) { + initial_filename = initial_filename.substring( 0, initial_filename.lastIndexOf( '.' ) ); + } + initial_filename = initial_filename + ".pdf"; + writetopdf_filechooser.setSelectedFile( new File( initial_filename ) ); + final File my_dir = curr_dir; + if ( my_dir != null ) { + writetopdf_filechooser.setCurrentDirectory( my_dir ); + } + final int result = writetopdf_filechooser.showSaveDialog( contentpane ); + File file = writetopdf_filechooser.getSelectedFile(); + // setCurrentDir( writetopdf_filechooser.getCurrentDirectory() ); + final File new_current_dir = writetopdf_filechooser.getCurrentDirectory(); + if ( ( file != null ) && ( result == JFileChooser.APPROVE_OPTION ) ) { + if ( !file.toString().toLowerCase().endsWith( ".pdf" ) ) { + file = new File( file.toString() + ".pdf" ); + } + if ( file.exists() ) { + final int i = JOptionPane.showConfirmDialog( component, + file + " already exists. Overwrite?", + "WARNING", + JOptionPane.OK_CANCEL_OPTION, + JOptionPane.WARNING_MESSAGE ); + if ( i != JOptionPane.OK_OPTION ) { + return null; + } + } + printPhylogenyToPdf( file.toString(), mp.getOptions(), mp.getCurrentTreePanel(), component ); } + return new_current_dir; } }