src/jalview/gui/ImageExporter.java

   1 /*
   2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
   3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
   4  *
   5  * This file is part of Jalview.
   6  *
   7  * Jalview is free software: you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation, either version 3
  10  * of the License, or (at your option) any later version.
  11  *
  12  * Jalview is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty
  14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15  * PURPOSE.  See the GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
  19  * The Jalview Authors are detailed in the 'AUTHORS' file.
  20  */
  21 package jalview.gui;
  22
  23 import java.awt.Component;
  24 import java.awt.Graphics;
  25 import java.io.File;
  26 import java.util.concurrent.Callable;
  27 import java.util.concurrent.atomic.AtomicBoolean;
  28
  29 import jalview.bin.Cache;
  30 import jalview.bin.Jalview;
  31 import jalview.io.JalviewFileChooser;
  32 import jalview.io.JalviewFileView;
  33 import jalview.util.ImageMaker;
  34 import jalview.util.ImageMaker.TYPE;
  35 import jalview.util.MessageManager;
  36 import jalview.util.Platform;
  37
  38 /**
  39  * A class that marshals steps in exporting a view in image graphics format
  40  * <ul>
  41  * <li>prompts the user for the output file, if not already specified</li>
  42  * <li>prompts the user for Text or Lineart character rendering, if
  43  * necessary</li>
  44  * <li>instantiates an ImageMaker to create the appropriate Graphics output
  45  * context for the image format</li>
  46  * <li>invokes a callback to do the work of writing to the graphics</li>
  47  * </ul>
  48  *
  49  * @author gmcarstairs
  50  *
  51  */
  52 public class ImageExporter
  53 {
  54   // todo move interface to jalview.api? or replace with lambda?
  55   /**
  56    * An interface for the callback that can be run to write the image on to the
  57    * graphics object. The callback should throw any exceptions arising so they
  58    * can be reported by this class.
  59    */
  60   public interface ImageWriterI
  61   {
  62     void exportImage(Graphics g) throws Exception;
  63   }
  64
  65   private IProgressIndicator messageBoard;
  66
  67   private ImageWriterI imageWriter;
  68
  69   TYPE imageType;
  70
  71   private String title;
  72
  73   /**
  74    * Constructor given a callback handler to write graphics data, an (optional)
  75    * target for status messages, image type and (optional) title for output file
  76    *
  77    * @param writer
  78    * @param statusBar
  79    * @param type
  80    * @param fileTitle
  81    */
  82   public ImageExporter(ImageWriterI writer, IProgressIndicator statusBar,
  83           TYPE type, String fileTitle)
  84   {
  85     this.imageWriter = writer;
  86     this.messageBoard = statusBar;
  87     this.imageType = type;
  88     this.title = fileTitle;
  89   }
  90
  91   /**
  92    * Prompts the user for output file and Text/Lineart options as required,
  93    * configures a Graphics context for output, and makes a callback to the
  94    * client code to perform the image output
  95    *
  96    * @param file
  97    *          output file (if null, user is prompted to choose)
  98    * @param parent
  99    *          parent component for any dialogs shown
 100    * @param width
 101    * @param height
 102    * @param imageSource
 103    *          what the image is of e.g. Tree, Alignment
 104    */
 105   public void doExport(File file, Component parent, int width, int height,
 106           String imageSource)
 107   {
 108     final long messageId = System.currentTimeMillis();
 109     setStatus(
 110             MessageManager.formatMessage(
 111                     "status.exporting_alignment_as_x_file", imageType),
 112             messageId);
 113
 114     /*
 115      * prompt user for output file if not provided
 116      */
 117     if (file == null && !Jalview.isHeadlessMode())
 118     {
 119       JalviewFileChooser chooser = imageType.getFileChooser();
 120       chooser.setFileView(new JalviewFileView());
 121       MessageManager.formatMessage("label.create_image_of",
 122               imageType.getName(), imageSource);
 123       String title = "Create " + imageType.getName()
 124               + " image from alignment";
 125       chooser.setDialogTitle(title);
 126       chooser.setToolTipText(MessageManager.getString("action.save"));
 127       int value = chooser.showSaveDialog(parent);
 128       if (value != JalviewFileChooser.APPROVE_OPTION)
 129       {
 130         String msg = MessageManager.formatMessage(
 131                 "status.cancelled_image_export_operation", imageType.name);
 132         setStatus(msg, messageId);
 133         return;
 134       }
 135       Cache.setProperty("LAST_DIRECTORY",
 136               chooser.getSelectedFile().getParent());
 137       file = chooser.getSelectedFile();
 138     }
 139
 140     /*
 141      * Prompt for Text or Lineart (EPS/SVG) unless a preference is already set
 142      * for this as EPS_RENDERING / SVG_RENDERING
 143      * Always set to Text for JalviewJS as Lineart (glyph fonts) not available
 144      */
 145     String renderStyle = Cache.getDefault(
 146             imageType.getName() + "_RENDERING",
 147             LineartOptions.PROMPT_EACH_TIME);
 148     if (Platform.isJS())
 149     {
 150       renderStyle = "Text";
 151     }
 152     AtomicBoolean textSelected = new AtomicBoolean(
 153             !"Lineart".equals(renderStyle));
 154     if ((imageType == TYPE.EPS || imageType == TYPE.SVG)
 155             && LineartOptions.PROMPT_EACH_TIME.equals(renderStyle)
 156             && !Jalview.isHeadlessMode())
 157     {
 158       final File chosenFile = file;
 159       Callable<Void> okAction = () -> {
 160         exportImage(chosenFile, !textSelected.get(), width, height,
 161                 messageId);
 162         return null;
 163       };
 164       LineartOptions epsOption = new LineartOptions(TYPE.EPS.getName(),
 165               textSelected);
 166       epsOption.setResponseAction(1, new Callable<Void>()
 167       {
 168         @Override
 169         public Void call()
 170         {
 171           setStatus(MessageManager.formatMessage(
 172                   "status.cancelled_image_export_operation",
 173                   imageType.getName()), messageId);
 174           return null;
 175         }
 176       });
 177       epsOption.setResponseAction(0, okAction);
 178       epsOption.showDialog();
 179       /* no code here - JalviewJS cannot execute it */
 180     }
 181     else
 182     {
 183       /*
 184        * character rendering not required, or preference already set
 185        * - just do the export
 186        */
 187       exportImage(file, !textSelected.get(), width, height, messageId);
 188     }
 189   }
 190
 191   /**
 192    * Constructs a suitable graphics context and passes it to the callback
 193    * handler for the image to be written. Shows status messages for export in
 194    * progress, complete, or failed as appropriate.
 195    *
 196    * @param chosenFile
 197    * @param asLineart
 198    * @param width
 199    * @param height
 200    * @param messageId
 201    */
 202   protected void exportImage(File chosenFile, boolean asLineart, int width,
 203           int height, long messageId)
 204   {
 205     String type = imageType.getName();
 206     try
 207     {
 208       // setStatus(
 209       // MessageManager.formatMessage(
 210       // "status.exporting_alignment_as_x_file", type),
 211       // messageId);
 212       ImageMaker im = new ImageMaker(imageType, width, height, chosenFile,
 213               title, asLineart);
 214       imageWriter.exportImage(im.getGraphics());
 215       im.writeImage();
 216       setStatus(
 217               MessageManager.formatMessage("status.export_complete", type),
 218               messageId);
 219     } catch (Exception e)
 220     {
 221       System.out.println(String.format("Error creating %s file: %s", type,
 222               e.toString()));
 223       setStatus(MessageManager.formatMessage("info.error_creating_file",
 224               type), messageId);
 225     }
 226   }
 227
 228   /**
 229    * Asks the callback to show a status message with given id
 230    *
 231    * @param msg
 232    * @param id
 233    */
 234   void setStatus(String msg, long id)
 235   {
 236     if (messageBoard != null && !Jalview.isHeadlessMode())
 237     {
 238       messageBoard.setProgressBar(msg, id);
 239     }
 240   }
 241
 242 }