bundles/org.simantics.scenegraph/src/org/simantics/scenegraph/utils/BufferedImage.java

   1 /*******************************************************************************
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management
   3  * in Industry THTH ry.
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Eclipse Public License v1.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *     VTT Technical Research Centre of Finland - initial API and implementation
  11  *******************************************************************************/
  12 package org.simantics.scenegraph.utils;
  13
  14 import java.awt.Color;
  15 import java.awt.Graphics2D;
  16 import java.awt.Point;
  17 import java.awt.RenderingHints;
  18 import java.awt.geom.AffineTransform;
  19 import java.awt.geom.Rectangle2D;
  20 import java.util.Objects;
  21
  22 import org.simantics.scenegraph.g2d.G2DRenderingHints;
  23 import org.simantics.scenegraph.g2d.color.ColorFilter;
  24 import org.simantics.scenegraph.g2d.color.Graphics2DWithColorFilter;
  25
  26 import com.kitfox.svg.SVGDiagram;
  27 import com.kitfox.svg.SVGException;
  28
  29 /**
  30  * @author J-P Laine
  31  */
  32 public class BufferedImage  {
  33     SVGDiagram                 source;
  34     Rectangle2D                imageBounds;
  35     Point                      referenceSize;
  36
  37     java.awt.image.BufferedImage   buffer = null;
  38     AffineTransform previousTransform = null;
  39     ColorFilter previousColorFilter = null;
  40
  41     /**
  42      * @param original
  43      * @param imageBounds
  44      * @param referenceSize a reference size for the rasterized images or
  45      *        <code>null</code> to not specify one in which case a default
  46      *        resolution is used
  47      */
  48     public BufferedImage(SVGDiagram original, Rectangle2D imageBounds, Point referenceSize) {
  49         this.source = original;
  50         this.imageBounds = imageBounds;
  51         this.referenceSize = referenceSize;
  52     }
  53
  54     public BufferedImage(SVGDiagram original) {
  55         this.source = original;
  56         this.imageBounds = original.getViewRect();
  57         this.referenceSize = null;
  58     }
  59
  60     public java.awt.image.BufferedImage getBuffer() {
  61         return buffer;
  62     }
  63
  64     @Override
  65     public void finalize() throws Throwable {
  66         //System.out.println("BufferedImage.finalize()");
  67         releaseRaster();
  68         super.finalize();
  69     }
  70
  71     public synchronized void releaseRaster() {
  72         source = null;
  73         if (buffer != null) {
  74             buffer.flush();
  75             buffer = null;
  76         }
  77     }
  78
  79     public void paintToBuffer(AffineTransform transform, float margin) throws SVGException {
  80         paintToBuffer(transform, margin);
  81     }
  82
  83     public void paintToBuffer(AffineTransform transform, ColorFilter colorFilter, float margin) throws SVGException {
  84
  85         int w = (int)((imageBounds.getWidth()+margin*2) * transform.getScaleX());
  86         int h = (int)((imageBounds.getHeight()+margin*2) * transform.getScaleY());
  87         if(w < 1) w = 1;
  88         if(h < 1) h = 1; // FIXME
  89
  90         buffer = new java.awt.image.BufferedImage(w, h, java.awt.image.BufferedImage.TYPE_INT_ARGB);
  91
  92         Graphics2D b2g = (Graphics2D)buffer.getGraphics();
  93
  94         b2g.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_SPEED);
  95         b2g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
  96         b2g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_BILINEAR);
  97
  98         b2g.setBackground(Color.WHITE);
  99         b2g.clearRect(0, 0, w, h);
 100         b2g.translate(margin, margin);
 101         b2g.scale(transform.getScaleX(), transform.getScaleY());
 102         b2g.translate(-imageBounds.getMinX(), -imageBounds.getMinY());
 103         if (colorFilter != null) {
 104             source.render(new Graphics2DWithColorFilter(b2g, colorFilter));
 105         } else {
 106             source.render(b2g);
 107         }
 108
 109     }
 110
 111     public void paint(Graphics2D g) {
 112         float margin = 5;
 113
 114         ColorFilter colorFilter = (ColorFilter) g.getRenderingHint(G2DRenderingHints.KEY_COLOR_FILTER);
 115         if(previousTransform == null || previousTransform.getScaleX() != g.getTransform().getScaleX() || previousTransform.getScaleY() != g.getTransform().getScaleY() || !Objects.equals(colorFilter, previousColorFilter)) {
 116 //              setupSourceRender(g);
 117             try {
 118                 previousColorFilter = colorFilter;
 119                 previousTransform = (AffineTransform)g.getTransform().clone();
 120                 paintToBuffer(previousTransform, previousColorFilter, margin);
 121             } catch (SVGException e) {
 122                 // TODO Auto-generated catch block
 123                 e.printStackTrace();
 124             }
 125         }
 126
 127         g.setRenderingHint(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_SPEED);
 128         g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_OFF);
 129         g.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_OFF);
 130         g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, RenderingHints.VALUE_INTERPOLATION_NEAREST_NEIGHBOR);
 131
 132         AffineTransform at = (AffineTransform)g.getTransform().clone();
 133         try {
 134             g.translate(imageBounds.getMinX(), imageBounds.getMinY());
 135             g.scale(1/previousTransform.getScaleX(), 1/previousTransform.getScaleY());
 136             g.translate(-margin, -margin);
 137
 138             g.drawImage(buffer, 0, 0, null);
 139         } finally {
 140             g.setTransform(at);
 141         }
 142     }
 143 }