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 import jalview.util.imagemaker.BitmapImageSizing;
  38
  39 /**
  40  * A class that marshals steps in exporting a view in image graphics format
  41  * <ul>
  42  * <li>prompts the user for the output file, if not already specified</li>
  43  * <li>prompts the user for Text or Lineart character rendering, if
  44  * necessary</li>
  45  * <li>instantiates an ImageMaker to create the appropriate Graphics output
  46  * context for the image format</li>
  47  * <li>invokes a callback to do the work of writing to the graphics</li>
  48  * </ul>
  49  *
  50  * @author gmcarstairs
  51  *
  52  */
  53 public class ImageExporter
  54 {
  55   // todo move interface to jalview.api? or replace with lambda?
  56   /**
  57    * An interface for the callback that can be run to write the image on to the
  58    * graphics object. The callback should throw any exceptions arising so they
  59    * can be reported by this class.
  60    */
  61   public interface ImageWriterI
  62   {
  63     void exportImage(Graphics g) throws Exception;
  64   }
  65
  66   private IProgressIndicator messageBoard;
  67
  68   private ImageWriterI imageWriter;
  69
  70   TYPE imageType;
  71
  72   private String title;
  73
  74   /**
  75    * Constructor given a callback handler to write graphics data, an (optional)
  76    * target for status messages, image type and (optional) title for output file
  77    *
  78    * @param writer
  79    * @param statusBar
  80    * @param type
  81    * @param fileTitle
  82    */
  83   public ImageExporter(ImageWriterI writer, IProgressIndicator statusBar,
  84           TYPE type, String fileTitle)
  85   {
  86     this.imageWriter = writer;
  87     this.messageBoard = statusBar;
  88     this.imageType = type;
  89     this.title = fileTitle;
  90   }
  91
  92   /**
  93    * Prompts the user for output file and Text/Lineart options as required,
  94    * configures a Graphics context for output, and makes a callback to the
  95    * client code to perform the image output
  96    *
  97    * @param file
  98    *          output file (if null, user is prompted to choose)
  99    * @param parent
 100    *          parent component for any dialogs shown
 101    * @param width
 102    * @param height
 103    * @param imageSource
 104    *          what the image is of e.g. Tree, Alignment
 105    */
 106   public void doExport(File file, Component parent, int width, int height,
 107           String imageSource)
 108   {
 109     doExport(file, parent, width, height, imageSource, null,
 110             BitmapImageSizing.nullBitmapImageSizing());
 111   }
 112
 113   public void doExport(File file, Component parent, int width, int height,
 114           String imageSource, String renderer, BitmapImageSizing userBis)
 115   {
 116     final long messageId = System.currentTimeMillis();
 117     setStatus(
 118             MessageManager.formatMessage(
 119                     "status.exporting_alignment_as_x_file", imageType),
 120             messageId);
 121
 122     /*
 123      * prompt user for output file if not provided
 124      */
 125     if (file == null && !Jalview.isHeadlessMode())
 126     {
 127       JalviewFileChooser chooser = imageType.getFileChooser();
 128       chooser.setFileView(new JalviewFileView());
 129       MessageManager.formatMessage("label.create_image_of",
 130               imageType.getName(), imageSource);
 131       String title = "Create " + imageType.getName()
 132               + " image from alignment";
 133       chooser.setDialogTitle(title);
 134       chooser.setToolTipText(MessageManager.getString("action.save"));
 135       int value = chooser.showSaveDialog(parent);
 136       if (value != JalviewFileChooser.APPROVE_OPTION)
 137       {
 138         String msg = MessageManager.formatMessage(
 139                 "status.cancelled_image_export_operation", imageType.name);
 140         setStatus(msg, messageId);
 141         return;
 142       }
 143       Cache.setProperty("LAST_DIRECTORY",
 144               chooser.getSelectedFile().getParent());
 145       file = chooser.getSelectedFile();
 146     }
 147
 148     /*
 149      * Prompt for Text or Lineart (EPS/SVG) unless a preference is already set
 150      * for this as EPS_RENDERING / SVG_RENDERING
 151      * Always set to Text for JalviewJS as Lineart (glyph fonts) not available
 152      */
 153     String renderStyle = renderer == null
 154             ? Cache.getDefault(imageType.getName() + "_RENDERING",
 155                     LineartOptions.PROMPT_EACH_TIME)
 156             : renderer;
 157     if (Platform.isJS())
 158     {
 159       renderStyle = "Text";
 160     }
 161     AtomicBoolean textSelected = new AtomicBoolean(
 162             !"Lineart".equals(renderStyle));
 163     if ((imageType == TYPE.EPS || imageType == TYPE.SVG)
 164             && LineartOptions.PROMPT_EACH_TIME.equals(renderStyle)
 165             && !Jalview.isHeadlessMode())
 166     {
 167       final File chosenFile = file;
 168       Callable<Void> okAction = () -> {
 169         exportImage(chosenFile, !textSelected.get(), width, height,
 170                 messageId, userBis);
 171         return null;
 172       };
 173       LineartOptions epsOption = new LineartOptions(TYPE.EPS.getName(),
 174               textSelected);
 175       epsOption.setResponseAction(1, new Callable<Void>()
 176       {
 177         @Override
 178         public Void call()
 179         {
 180           setStatus(MessageManager.formatMessage(
 181                   "status.cancelled_image_export_operation",
 182                   imageType.getName()), messageId);
 183           return null;
 184         }
 185       });
 186       epsOption.setResponseAction(0, okAction);
 187       epsOption.showDialog();
 188       /* no code here - JalviewJS cannot execute it */
 189     }
 190     else
 191     {
 192       /*
 193        * character rendering not required, or preference already set
 194        * - just do the export
 195        */
 196       exportImage(file, !textSelected.get(), width, height, messageId,
 197               userBis);
 198     }
 199   }
 200
 201   /**
 202    * Constructs a suitable graphics context and passes it to the callback
 203    * handler for the image to be written. Shows status messages for export in
 204    * progress, complete, or failed as appropriate.
 205    *
 206    * @param chosenFile
 207    * @param asLineart
 208    * @param width
 209    * @param height
 210    * @param messageId
 211    */
 212   protected void exportImage(File chosenFile, boolean asLineart, int width,
 213           int height, long messageId, BitmapImageSizing userBis)
 214   {
 215     String type = imageType.getName();
 216     try
 217     {
 218       // setStatus(
 219       // MessageManager.formatMessage(
 220       // "status.exporting_alignment_as_x_file", type),
 221       // messageId);
 222       ImageMaker im = new ImageMaker(imageType, width, height, chosenFile,
 223               title, asLineart, userBis);
 224       imageWriter.exportImage(im.getGraphics());
 225       im.writeImage();
 226       setStatus(
 227               MessageManager.formatMessage("status.export_complete", type),
 228               messageId);
 229     } catch (Exception e)
 230     {
 231       System.out.println(String.format("Error creating %s file: %s", type,
 232               e.toString()));
 233       setStatus(MessageManager.formatMessage("info.error_creating_file",
 234               type), messageId);
 235     }
 236   }
 237
 238   /**
 239    * Asks the callback to show a status message with given id
 240    *
 241    * @param msg
 242    * @param id
 243    */
 244   void setStatus(String msg, long id)
 245   {
 246     if (messageBoard != null && !Jalview.isHeadlessMode())
 247     {
 248       messageBoard.setProgressBar(msg, id);
 249     }
 250   }
 251
 252 }