src/jalview/util/ImageMaker.java

   1 /*\r
   2  * Jalview - A Sequence Alignment Editor and Viewer\r
   3  * Copyright (C) 2007 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
   4  *\r
   5  * This program is free software; you can redistribute it and/or\r
   6  * modify it under the terms of the GNU General Public License\r
   7  * as published by the Free Software Foundation; either version 2\r
   8  * of the License, or (at your option) any later version.\r
   9  *\r
  10  * This program is distributed in the hope that it will be useful,\r
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  13  * GNU General Public License for more details.\r
  14  *\r
  15  * You should have received a copy of the GNU General Public License\r
  16  * along with this program; if not, write to the Free Software\r
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA\r
  18  */\r
  19 package jalview.util;\r
  20 \r
  21 import java.io.*;\r
  22 import javax.imageio.*;\r
  23 \r
  24 import java.awt.*;\r
  25 import java.awt.image.*;\r
  26 \r
  27 import org.jibble.epsgraphics.*;\r
  28 import jalview.gui.*;\r
  29 import jalview.io.*;\r
  30 \r
  31 public class ImageMaker\r
  32 {\r
  33   public static final int EPS = 0;\r
  34   public static final int PNG = 1;\r
  35   int type = -1;\r
  36 \r
  37   EpsGraphics2D pg;\r
  38   Graphics graphics;\r
  39   FileOutputStream out;\r
  40   BufferedImage bi;\r
  41 \r
  42   public ImageMaker(Component parent, int type, String title,\r
  43                     int width, int height, File file, String EPStitle)\r
  44   {\r
  45     this.type = type;\r
  46 \r
  47     if (file == null)\r
  48     {\r
  49       JalviewFileChooser chooser;\r
  50       chooser = type == EPS ? getEPSChooser() : getPNGChooser();\r
  51 \r
  52       chooser.setFileView(new jalview.io.JalviewFileView());\r
  53       chooser.setDialogTitle(title);\r
  54       chooser.setToolTipText("Save");\r
  55 \r
  56       int value = chooser.showSaveDialog(parent);\r
  57 \r
  58       if (value == jalview.io.JalviewFileChooser.APPROVE_OPTION)\r
  59       {\r
  60         jalview.bin.Cache.setProperty("LAST_DIRECTORY",\r
  61                                       chooser.getSelectedFile().getParent());\r
  62 \r
  63         file = chooser.getSelectedFile();\r
  64       }\r
  65     }\r
  66 \r
  67     if (file != null)\r
  68     {\r
  69       try\r
  70       {\r
  71         out = new FileOutputStream(file);\r
  72 \r
  73         if (type == EPS)\r
  74         {\r
  75           setupEPS(width, height, EPStitle);\r
  76         }\r
  77         else\r
  78         {\r
  79           setupPNG(width, height);\r
  80         }\r
  81       }\r
  82       catch (Exception ex)\r
  83       {\r
  84         System.out.println("Error creating " + (type == EPS ? "EPS" : "PNG") +\r
  85                            " file.");\r
  86       }\r
  87     }\r
  88   }\r
  89 \r
  90   public Graphics getGraphics()\r
  91   {\r
  92     return graphics;\r
  93   }\r
  94 \r
  95   void setupPNG(int width, int height)\r
  96   {\r
  97     bi = new BufferedImage(width, height,\r
  98                            BufferedImage.TYPE_INT_RGB);\r
  99     graphics = bi.getGraphics();\r
 100     Graphics2D ig2 = (Graphics2D) graphics;\r
 101     ig2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,\r
 102                          RenderingHints.VALUE_ANTIALIAS_ON);\r
 103   }\r
 104 \r
 105   public void writeImage()\r
 106   {\r
 107     try\r
 108     {\r
 109       switch (type)\r
 110       {\r
 111         case EPS:\r
 112           pg.flush();\r
 113           pg.close();\r
 114           break;\r
 115         case PNG:\r
 116           ImageIO.write(bi, "png", out);\r
 117           out.close();\r
 118           break;\r
 119       }\r
 120     }\r
 121     catch (Exception ex)\r
 122     {\r
 123       ex.printStackTrace();\r
 124     }\r
 125   }\r
 126 \r
 127   void setupEPS(int width, int height, String title)\r
 128   {\r
 129     boolean accurateText = true;\r
 130 \r
 131     String renderStyle = jalview.bin.Cache.getDefault("EPS_RENDERING",\r
 132         "Prompt each time");\r
 133 \r
 134     // If we need to prompt, and if the GUI is visible then\r
 135     // Prompt for EPS rendering style\r
 136     if (renderStyle.equalsIgnoreCase("Prompt each time")\r
 137         && !\r
 138         (System.getProperty("java.awt.headless") != null\r
 139          && System.getProperty("java.awt.headless").equals("true")))\r
 140     {\r
 141       EPSOptions eps = new EPSOptions();\r
 142       renderStyle = eps.getValue();\r
 143 \r
 144       if (renderStyle == null || eps.cancelled)\r
 145       {\r
 146         return;\r
 147       }\r
 148     }\r
 149 \r
 150     if (renderStyle.equalsIgnoreCase("text"))\r
 151     {\r
 152       accurateText = false;\r
 153     }\r
 154 \r
 155     try\r
 156     {\r
 157       pg = new EpsGraphics2D(title, out, 0, 0, width,\r
 158                              height);\r
 159       Graphics2D ig2 = (Graphics2D) pg;\r
 160       ig2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,\r
 161                            RenderingHints.VALUE_ANTIALIAS_ON);\r
 162 \r
 163       pg.setAccurateTextMode(accurateText);\r
 164 \r
 165       graphics = pg;\r
 166     }\r
 167     catch (Exception ex)\r
 168     {}\r
 169   }\r
 170 \r
 171   JalviewFileChooser getPNGChooser()\r
 172   {\r
 173     return new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(\r
 174         "LAST_DIRECTORY"), new String[]\r
 175                                              {"png"},\r
 176                                              new String[]\r
 177                                              {"Portable network graphics"},\r
 178                                              "Portable network graphics");\r
 179   }\r
 180 \r
 181   JalviewFileChooser getEPSChooser()\r
 182   {\r
 183     return new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(\r
 184         "LAST_DIRECTORY"), new String[]\r
 185                                              {"eps"},\r
 186                                              new String[]\r
 187                                              {"Encapsulated Postscript"},\r
 188                                              "Encapsulated Postscript");\r
 189   }\r
 190 }\r