Destroy Alignment when closing frame
[jalview.git] / 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