X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FAnnotationExporter.java;h=81960e08a22ba15256ba12bf4750b8ce73576842;hb=3da878124135ff033f42d19d8733891b09e953cd;hp=a48c03066dc2cb84090e7528788472a217b91666;hpb=1ec8fc3e8bc982115e5ba1b16a101d239a2df591;p=jalview.git diff --git a/src/jalview/gui/AnnotationExporter.java b/src/jalview/gui/AnnotationExporter.java index a48c030..81960e0 100644 --- a/src/jalview/gui/AnnotationExporter.java +++ b/src/jalview/gui/AnnotationExporter.java @@ -20,7 +20,8 @@ */ package jalview.gui; -import jalview.api.FeatureColourI; +import jalview.api.FeatureRenderer; +import jalview.bin.Cache; import jalview.datamodel.AlignmentAnnotation; import jalview.datamodel.SequenceI; import jalview.io.AnnotationFile; @@ -34,7 +35,8 @@ import java.awt.Color; import java.awt.FlowLayout; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; -import java.util.Map; +import java.io.FileWriter; +import java.io.PrintWriter; import javax.swing.BorderFactory; import javax.swing.ButtonGroup; @@ -56,18 +58,22 @@ import javax.swing.SwingConstants; */ public class AnnotationExporter extends JPanel { - JInternalFrame frame; + private JInternalFrame frame; - AlignmentPanel ap; + private AlignmentPanel ap; - boolean features = true; + /* + * true if exporting features, false if exporting annotations + */ + private boolean exportFeatures = true; private AlignmentAnnotation[] annotations; private boolean wholeView; - public AnnotationExporter() + public AnnotationExporter(AlignmentPanel panel) { + this.ap = panel; try { jbInit(); @@ -83,62 +89,69 @@ public class AnnotationExporter extends JPanel frame.getPreferredSize().height); } - public void exportFeatures(AlignmentPanel ap) + /** + * Configures the diglog for options to export visible features + */ + public void exportFeatures() { - this.ap = ap; - features = true; + exportFeatures = true; CSVFormat.setVisible(false); frame.setTitle(MessageManager.getString("label.export_features")); } - public void exportAnnotations(AlignmentPanel ap) + /** + * Configures the dialog for options to export all visible annotations + */ + public void exportAnnotations() { - this.ap = ap; - annotations = ap.av.isShowAnnotation() ? null : ap.av.getAlignment() - .getAlignmentAnnotation(); - wholeView = true; - startExportAnnotation(); + boolean showAnnotation = ap.av.isShowAnnotation(); + exportAnnotation(showAnnotation ? null + : ap.av.getAlignment().getAlignmentAnnotation(), true); } - public void exportAnnotations(AlignmentPanel alp, - AlignmentAnnotation[] toExport) + /** + * Configures the dialog for options to export the given annotation row + * + * @param toExport + */ + public void exportAnnotation(AlignmentAnnotation toExport) { - ap = alp; - annotations = toExport; - wholeView = false; - startExportAnnotation(); + exportAnnotation(new AlignmentAnnotation[] { toExport }, false); } - private void startExportAnnotation() + private void exportAnnotation(AlignmentAnnotation[] toExport, + boolean forWholeView) { - features = false; + wholeView = forWholeView; + annotations = toExport; + exportFeatures = false; GFFFormat.setVisible(false); CSVFormat.setVisible(true); frame.setTitle(MessageManager.getString("label.export_annotations")); } - public void toFile_actionPerformed(ActionEvent e) + private void toFile_actionPerformed() { + // TODO: JAL-3048 JalviewFileChooser - Save option JalviewFileChooser chooser = new JalviewFileChooser( - jalview.bin.Cache.getProperty("LAST_DIRECTORY")); + Cache.getProperty("LAST_DIRECTORY")); chooser.setFileView(new JalviewFileView()); - chooser.setDialogTitle(features ? MessageManager - .getString("label.save_features_to_file") : MessageManager - .getString("label.save_annotation_to_file")); + chooser.setDialogTitle(exportFeatures + ? MessageManager.getString("label.save_features_to_file") + : MessageManager.getString("label.save_annotation_to_file")); chooser.setToolTipText(MessageManager.getString("action.save")); int value = chooser.showSaveDialog(this); if (value == JalviewFileChooser.APPROVE_OPTION) { - String text = getFileContents(); + String text = getText(); try { - java.io.PrintWriter out = new java.io.PrintWriter( - new java.io.FileWriter(chooser.getSelectedFile())); - + PrintWriter out = new PrintWriter( + new FileWriter(chooser.getSelectedFile())); out.print(text); out.close(); } catch (Exception ex) @@ -147,85 +160,106 @@ public class AnnotationExporter extends JPanel } } - close_actionPerformed(null); + close_actionPerformed(); + } + + /** + * Answers the text to output for either Features (in GFF or Jalview format) or + * Annotations (in CSV or Jalview format) + * + * @return + */ + private String getText() + { + return exportFeatures ? getFeaturesText() : getAnnotationsText(); } - private String getFileContents() + /** + * Returns the text contents for output of annotations in either CSV or Jalview + * format + * + * @return + */ + private String getAnnotationsText() { - String text = MessageManager - .getString("label.no_features_on_alignment"); - if (features) + String text; + if (CSVFormat.isSelected()) { - FeaturesFile formatter = new FeaturesFile(); - SequenceI[] sequences = ap.av.getAlignment().getSequencesArray(); - Map featureColours = ap.getFeatureRenderer() - .getDisplayedFeatureCols(); - boolean includeNonPositional = ap.av.isShowNPFeats(); - if (GFFFormat.isSelected()) - { - text = formatter.printGffFormat(sequences, featureColours, - includeNonPositional); - } - else - { - text = formatter.printJalviewFormat(sequences, featureColours, - includeNonPositional); - } + text = new AnnotationFile().printCSVAnnotations(annotations); } else { - if (CSVFormat.isSelected()) + if (wholeView) { - text = new AnnotationFile().printCSVAnnotations(annotations); + text = new AnnotationFile().printAnnotationsForView(ap.av); } else { - if (wholeView) - { - text = new AnnotationFile().printAnnotationsForView(ap.av); - } - else - { - text = new AnnotationFile().printAnnotations(annotations, null, - null); - } + text = new AnnotationFile().printAnnotations(annotations, null, + null); } } return text; } - public void toTextbox_actionPerformed(ActionEvent e) + /** + * Returns the text contents for output of features in either GFF or Jalview + * format + * + * @return + */ + private String getFeaturesText() + { + String text; + SequenceI[] sequences = ap.av.getAlignment().getSequencesArray(); + boolean includeNonPositional = ap.av.isShowNPFeats(); + + FeaturesFile formatter = new FeaturesFile(); + final FeatureRenderer fr = ap.getFeatureRenderer(); + if (GFFFormat.isSelected()) + { + text = formatter.printGffFormat(sequences, fr, includeNonPositional); + } + else + { + text = formatter.printJalviewFormat(sequences, fr, + includeNonPositional); + } + return text; + } + + private void toTextbox_actionPerformed() { CutAndPasteTransfer cap = new CutAndPasteTransfer(); try { - String text = getFileContents(); + String text = getText(); cap.setText(text); - Desktop.addInternalFrame( - cap, - (features ? MessageManager.formatMessage( - "label.features_for_params", - new String[] { ap.alignFrame.getTitle() }) - : MessageManager.formatMessage( - "label.annotations_for_params", - new String[] { ap.alignFrame.getTitle() })), + Desktop.addInternalFrame(cap, (exportFeatures ? MessageManager + .formatMessage("label.features_for_params", new String[] + { ap.alignFrame.getTitle() }) + : MessageManager.formatMessage("label.annotations_for_params", + new String[] + { ap.alignFrame.getTitle() })), 600, 500); } catch (OutOfMemoryError oom) { - new OOMWarning((features ? MessageManager.formatMessage( - "label.generating_features_for_params", - new String[] { ap.alignFrame.getTitle() }) + new OOMWarning((exportFeatures ? MessageManager.formatMessage( + "label.generating_features_for_params", new String[] + { ap.alignFrame.getTitle() }) : MessageManager.formatMessage( "label.generating_annotations_for_params", - new String[] { ap.alignFrame.getTitle() })), oom); + new String[] + { ap.alignFrame.getTitle() })), + oom); cap.dispose(); } - close_actionPerformed(null); + close_actionPerformed(); } - public void close_actionPerformed(ActionEvent e) + private void close_actionPerformed() { try { @@ -245,7 +279,7 @@ public class AnnotationExporter extends JPanel @Override public void actionPerformed(ActionEvent e) { - toFile_actionPerformed(e); + toFile_actionPerformed(); } }); toTextbox.setText(MessageManager.getString("label.to_textbox")); @@ -254,7 +288,7 @@ public class AnnotationExporter extends JPanel @Override public void actionPerformed(ActionEvent e) { - toTextbox_actionPerformed(e); + toTextbox_actionPerformed(); } }); close.setText(MessageManager.getString("action.close")); @@ -263,7 +297,7 @@ public class AnnotationExporter extends JPanel @Override public void actionPerformed(ActionEvent e) { - close_actionPerformed(e); + close_actionPerformed(); } }); jalviewFormat.setOpaque(false);