diff options
Diffstat (limited to 'src/net/sourceforge/plantuml/creole/AtomImg.java')
| -rw-r--r-- | src/net/sourceforge/plantuml/creole/AtomImg.java | 137 |
1 files changed, 137 insertions, 0 deletions
diff --git a/src/net/sourceforge/plantuml/creole/AtomImg.java b/src/net/sourceforge/plantuml/creole/AtomImg.java new file mode 100644 index 0000000..4e7c255 --- /dev/null +++ b/src/net/sourceforge/plantuml/creole/AtomImg.java @@ -0,0 +1,137 @@ +/* ========================================================================
+ * PlantUML : a free UML diagram generator
+ * ========================================================================
+ *
+ * (C) Copyright 2009-2014, Arnaud Roques
+ *
+ * Project Info: http://plantuml.sourceforge.net
+ *
+ * This file is part of PlantUML.
+ *
+ * Licensed under The MIT License (Massachusetts Institute of Technology License)
+ *
+ * See http://opensource.org/licenses/MIT
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included
+ * in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+ * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
+ * IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ *
+ *
+ * Original Author: Arnaud Roques
+ */
+package net.sourceforge.plantuml.creole;
+
+import java.awt.Font;
+import java.awt.geom.Dimension2D;
+import java.awt.image.BufferedImage;
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URL;
+import java.net.URLConnection;
+
+import javax.imageio.ImageIO;
+
+import net.sourceforge.plantuml.Dimension2DDouble;
+import net.sourceforge.plantuml.FileSystem;
+import net.sourceforge.plantuml.graphic.FontConfiguration;
+import net.sourceforge.plantuml.graphic.HtmlColorUtils;
+import net.sourceforge.plantuml.graphic.ImgValign;
+import net.sourceforge.plantuml.graphic.StringBounder;
+import net.sourceforge.plantuml.ugraphic.UFont;
+import net.sourceforge.plantuml.ugraphic.UGraphic;
+import net.sourceforge.plantuml.ugraphic.UImage;
+
+public class AtomImg implements Atom {
+
+ private final BufferedImage image;
+
+ private AtomImg(BufferedImage image) {
+ this.image = image;
+ }
+
+ public static Atom create(String src, final ImgValign valign, final int vspace) {
+ final UFont font = new UFont("Monospaced", Font.PLAIN, 14);
+ final FontConfiguration fc = new FontConfiguration(font, HtmlColorUtils.BLACK, HtmlColorUtils.BLUE, true);
+ try {
+ final File f = FileSystem.getInstance().getFile(src);
+ if (f.exists() == false) {
+ // Check if valid URL
+ if (src.startsWith("http:") || src.startsWith("https:")) {
+ final byte image[] = getFile(src);
+ final BufferedImage read = ImageIO.read(new ByteArrayInputStream(image));
+ if (read == null) {
+ return AtomText.create("(Cannot decode: " + src + ")", fc);
+ }
+ return new AtomImg(read);
+ }
+ return AtomText.create("(File not found: " + f + ")", fc);
+ }
+ if (f.getName().endsWith(".svg")) {
+ // return new AtomImg(new TileImageSvg(f));
+ throw new UnsupportedOperationException();
+ }
+ final BufferedImage read = ImageIO.read(f);
+ if (read == null) {
+ return AtomText.create("(Cannot decode: " + f + ")", fc);
+ }
+ return new AtomImg(ImageIO.read(f));
+ } catch (IOException e) {
+ return AtomText.create("ERROR " + e.toString(), fc);
+ }
+ }
+
+ // Added by Alain Corbiere
+ static byte[] getFile(String host) throws IOException {
+ final ByteArrayOutputStream image = new ByteArrayOutputStream();
+ InputStream input = null;
+ try {
+ final URL url = new URL(host);
+ final URLConnection connection = url.openConnection();
+ input = connection.getInputStream();
+ final byte[] buffer = new byte[1024];
+ int read;
+ while ((read = input.read(buffer)) > 0) {
+ image.write(buffer, 0, read);
+ }
+ image.close();
+ return image.toByteArray();
+ } finally {
+ if (input != null) {
+ input.close();
+ }
+ }
+ }
+
+ // End
+
+ public Dimension2D calculateDimension(StringBounder stringBounder) {
+ return new Dimension2DDouble(image.getWidth(), image.getHeight());
+ }
+
+ public double getStartingAltitude(StringBounder stringBounder) {
+ return 0;
+ }
+
+ public void drawU(UGraphic ug) {
+ // final double h = calculateDimension(ug.getStringBounder()).getHeight();
+ ug.draw(new UImage(image));
+ // tileImage.drawU(ug.apply(new UTranslate(0, -h)));
+ }
+
+}
|