+/*******************************************************************************\r
+ * Copyright (c) 2007- VTT Technical Research Centre of Finland.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ * VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.image.ui.editor;\r
+\r
+import java.awt.geom.AffineTransform;\r
+import java.awt.geom.Point2D;\r
+import java.awt.geom.Rectangle2D;\r
+import java.awt.image.BufferedImage;\r
+import java.awt.image.DirectColorModel;\r
+import java.awt.image.IndexColorModel;\r
+import java.awt.image.WritableRaster;\r
+import java.io.ByteArrayInputStream;\r
+\r
+import org.eclipse.jface.layout.GridDataFactory;\r
+import org.eclipse.jface.resource.JFaceResources;\r
+import org.eclipse.jface.resource.LocalResourceManager;\r
+import org.eclipse.jface.resource.ResourceManager;\r
+import org.eclipse.swt.SWT;\r
+import org.eclipse.swt.SWTError;\r
+import org.eclipse.swt.SWTException;\r
+import org.eclipse.swt.graphics.GC;\r
+import org.eclipse.swt.graphics.Image;\r
+import org.eclipse.swt.graphics.ImageData;\r
+import org.eclipse.swt.graphics.PaletteData;\r
+import org.eclipse.swt.graphics.Point;\r
+import org.eclipse.swt.graphics.RGB;\r
+import org.eclipse.swt.graphics.Rectangle;\r
+import org.eclipse.swt.widgets.Canvas;\r
+import org.eclipse.swt.widgets.Composite;\r
+import org.eclipse.swt.widgets.Display;\r
+import org.eclipse.swt.widgets.Event;\r
+import org.eclipse.swt.widgets.Listener;\r
+import org.simantics.Simantics;\r
+import org.simantics.databoard.Bindings;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.request.ParametrizedRead;\r
+import org.simantics.db.common.request.ResourceRead;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.request.combinations.Combinators;\r
+import org.simantics.image2.ontology.ImageResource;\r
+import org.simantics.scenegraph.ScenegraphUtils;\r
+import org.simantics.scenegraph.utils.GeometryUtils;\r
+import org.simantics.ui.workbench.IResourceEditorInput;\r
+import org.simantics.ui.workbench.ResourceEditorPart;\r
+import org.simantics.ui.workbench.editor.input.InputValidationCombinators;\r
+import org.simantics.utils.ui.ErrorLogger;\r
+import org.simantics.utils.ui.LayoutUtils;\r
+\r
+import com.kitfox.svg.SVGDiagram;\r
+import com.kitfox.svg.SVGException;\r
+import com.kitfox.svg.SVGUniverse;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class ImageEditor extends ResourceEditorPart {\r
+\r
+ public static final String EDITOR_ID = "org.simantics.wiki.ui.image.editor";\r
+\r
+ protected boolean disposed = false;\r
+\r
+ protected Image image;\r
+ protected SVGDiagram svgDiagram;\r
+ private ResourceManager resourceManager;\r
+ private Canvas canvas;\r
+\r
+ private double zoomLevel = 1;\r
+ private Point2D previousTranslate = new Point2D.Double();\r
+ private Point2D translate = new Point2D.Double();\r
+ private boolean zoomToFit = false;\r
+ private SVGUniverse svgUniverse = new SVGUniverse();\r
+\r
+ static ImageData convertToSWT(BufferedImage bufferedImage) {\r
+ if (bufferedImage.getColorModel() instanceof DirectColorModel) {\r
+ DirectColorModel colorModel = (DirectColorModel) bufferedImage\r
+ .getColorModel();\r
+ PaletteData palette = new PaletteData(colorModel.getRedMask(),\r
+ colorModel.getGreenMask(), colorModel.getBlueMask());\r
+ ImageData data = new ImageData(bufferedImage.getWidth(),\r
+ bufferedImage.getHeight(), colorModel.getPixelSize(),\r
+ palette);\r
+ WritableRaster raster = bufferedImage.getRaster();\r
+ int[] pixelArray = new int[4];\r
+ for (int y = 0; y < data.height; y++) {\r
+ for (int x = 0; x < data.width; x++) {\r
+ raster.getPixel(x, y, pixelArray);\r
+ int pixel = palette.getPixel(new RGB(pixelArray[0],\r
+ pixelArray[1], pixelArray[2]));\r
+ data.setPixel(x, y, pixel);\r
+ }\r
+ }\r
+ return data;\r
+ } else if (bufferedImage.getColorModel() instanceof IndexColorModel) {\r
+ IndexColorModel colorModel = (IndexColorModel) bufferedImage\r
+ .getColorModel();\r
+ int size = colorModel.getMapSize();\r
+ byte[] reds = new byte[size];\r
+ byte[] greens = new byte[size];\r
+ byte[] blues = new byte[size];\r
+ colorModel.getReds(reds);\r
+ colorModel.getGreens(greens);\r
+ colorModel.getBlues(blues);\r
+ RGB[] rgbs = new RGB[size];\r
+ for (int i = 0; i < rgbs.length; i++) {\r
+ rgbs[i] = new RGB(reds[i] & 0xFF, greens[i] & 0xFF,\r
+ blues[i] & 0xFF);\r
+ }\r
+ PaletteData palette = new PaletteData(rgbs);\r
+ ImageData data = new ImageData(bufferedImage.getWidth(),\r
+ bufferedImage.getHeight(), colorModel.getPixelSize(),\r
+ palette);\r
+ data.transparentPixel = colorModel.getTransparentPixel();\r
+ WritableRaster raster = bufferedImage.getRaster();\r
+ int[] pixelArray = new int[1];\r
+ for (int y = 0; y < data.height; y++) {\r
+ for (int x = 0; x < data.width; x++) {\r
+ raster.getPixel(x, y, pixelArray);\r
+ data.setPixel(x, y, pixelArray[0]);\r
+ }\r
+ }\r
+ return data;\r
+ }\r
+ return null;\r
+ }\r
+\r
+ ParametrizedRead<IResourceEditorInput, Boolean> INPUT_VALIDATOR =\r
+ Combinators.compose(\r
+ InputValidationCombinators.hasURI(),\r
+ InputValidationCombinators.extractInputResource()\r
+ );\r
+\r
+ @Override\r
+ protected ParametrizedRead<IResourceEditorInput, Boolean> getInputValidator() {\r
+ return INPUT_VALIDATOR;\r
+ }\r
+\r
+ @Override\r
+ public void createPartControl(final Composite parent) {\r
+ this.resourceManager = new LocalResourceManager(JFaceResources.getResources(), parent);\r
+\r
+ parent.setLayout(LayoutUtils.createNoBorderGridLayout(1));\r
+\r
+ canvas = new Canvas(parent, SWT.DOUBLE_BUFFERED);\r
+ canvas.setBackground(resourceManager.createColor(new RGB(255, 255, 255)));\r
+ GridDataFactory.fillDefaults().grab(true, true).applyTo(canvas);\r
+\r
+ canvas.addListener(SWT.Paint, new Listener() {\r
+ @Override\r
+ public void handleEvent(Event event) {\r
+ if (svgDiagram != null) {\r
+ Rectangle2D r = svgDiagram.getViewRect();\r
+ if (r.isEmpty())\r
+ return;\r
+\r
+ // FIXME: this is unsafe, renders unnecessarily large\r
+ // buffered images when only control size is ultimately needed.\r
+\r
+ AffineTransform tr = AffineTransform.getScaleInstance(zoomLevel, zoomLevel);\r
+ tr.translate(translate.getX(), translate.getY());\r
+ try {\r
+ BufferedImage bi = ScenegraphUtils.paintSVG(svgDiagram, tr, 0f);\r
+ Image img = new Image(parent.getDisplay(), convertToSWT(bi));\r
+ drawImage(event.gc, img, false);\r
+ img.dispose();\r
+ } catch (SVGException e) {\r
+ }\r
+ } else if (image != null) {\r
+ drawImage(event.gc, image, true);\r
+ }\r
+ }\r
+ private void drawImage(GC gc, Image image, boolean fitToCanvas) {\r
+ Rectangle r = image.getBounds();\r
+ if (r.isEmpty())\r
+ return;\r
+\r
+ Point destSize = canvas.getSize();\r
+ int xSpace = destSize.x - r.width;\r
+ int ySpace = destSize.y - r.height;\r
+ boolean fitsX = xSpace >= 0;\r
+ boolean fitsY = ySpace >= 0;\r
+ boolean fitsCanvas = fitsX && fitsY;\r
+\r
+ // if the image is larger than the canvas, zoom it to fit\r
+ if ((!fitsCanvas && fitToCanvas) || zoomToFit) {\r
+ gc.setAntialias(SWT.ON);\r
+\r
+ // Zoom to fit propertionally\r
+ int leftMargin = 0;\r
+ int topMargin = 0;\r
+ if (xSpace > ySpace) {\r
+ double yr = (double) destSize.y / r.height;\r
+ double xo = (int)(r.width * yr);\r
+ leftMargin = (int) Math.round(((destSize.x - r.width) + (r.width - xo)) * 0.5);\r
+ gc.drawImage(image, 0, 0, r.width, r.height, leftMargin, topMargin, (int) xo, destSize.y);\r
+ } else {\r
+ double xr = (double) destSize.x / r.width;\r
+ double yo = (int)(r.height * xr);\r
+ topMargin = (int) Math.round(((destSize.y - r.height) + (r.height - yo)) * 0.5);\r
+ gc.drawImage(image, 0, 0, r.width, r.height, leftMargin, topMargin, destSize.x, (int) yo);\r
+ }\r
+ } else {\r
+ if (!fitsCanvas) {\r
+ int srcX = fitsX ? 0 : (int) Math.round(-xSpace * 0.5);\r
+ int srcY = fitsY ? 0 : (int) Math.round(-ySpace * 0.5);\r
+ int srcW = fitsX ? r.width : r.width + xSpace;\r
+ int srcH = fitsY ? r.height : r.height + ySpace;\r
+ int destX = fitsX ? (int) Math.round(xSpace * 0.5) : 0;\r
+ int destY = fitsY ? (int) Math.round(ySpace * 0.5) : 0;\r
+ int destW = fitsX ? r.width : destSize.x;\r
+ int destH = fitsY ? r.height : destSize.y;\r
+ gc.drawImage(image, srcX, srcY, srcW, srcH, destX, destY, destW, destH);\r
+ } else {\r
+ // Center on the canvas.\r
+ int leftMargin = (int) Math.round((destSize.x - r.width) * 0.5);\r
+ int topMargin = (int) Math.round((destSize.y - r.height) * 0.5);\r
+ gc.drawImage(image, 0, 0, r.width, r.height, leftMargin, topMargin, r.width, r.height);\r
+ }\r
+ }\r
+ }\r
+ });\r
+ canvas.addListener(SWT.MouseDoubleClick, new Listener() {\r
+ @Override\r
+ public void handleEvent(Event event) {\r
+ zoomToFit ^= true;\r
+ canvas.redraw();\r
+ }\r
+ });\r
+\r
+ // Mouse tracking support\r
+ Listener listener = new Listener() {\r
+ boolean pan = false;\r
+ Point panStart = new Point(0, 0); \r
+ @Override\r
+ public void handleEvent(Event e) {\r
+ switch (e.type) {\r
+ case SWT.MouseUp:\r
+ if (e.button == 3) {\r
+ pan = false;\r
+ previousTranslate.setLocation(translate);\r
+ }\r
+ break;\r
+ case SWT.MouseDown:\r
+ if (e.button == 3) {\r
+ panStart.x = e.x;\r
+ panStart.y = e.y;\r
+ pan = true;\r
+ previousTranslate.setLocation(translate);\r
+ }\r
+ break;\r
+ case SWT.MouseMove:\r
+ if (pan) {\r
+ int dx = e.x - panStart.x;\r
+ int dy = e.y - panStart.y;\r
+ translate.setLocation(\r
+ previousTranslate.getX() + dx / zoomLevel,\r
+ previousTranslate.getY() + dy / zoomLevel);\r
+ canvas.redraw();\r
+ }\r
+ break;\r
+ case SWT.MouseVerticalWheel:\r
+ double scroll = Math.min(0.9, -e.count / 20.0);\r
+ double z = 1 - scroll;\r
+ zoomLevel = limitScaleFactor(zoomLevel, z);\r
+ canvas.redraw();\r
+ break;\r
+ }\r
+ }\r
+ private double limitScaleFactor(double zoomLevel, double scaleFactor) {\r
+ double inLimit = 200.0;\r
+ double outLimit = 10;\r
+\r
+ AffineTransform view = AffineTransform.getScaleInstance(zoomLevel, zoomLevel);\r
+ double currentScale = GeometryUtils.getScale(view) * 100.0;\r
+ double newScale = currentScale * scaleFactor;\r
+\r
+ if (newScale > currentScale && newScale > inLimit) {\r
+ if (currentScale < inLimit)\r
+ scaleFactor = inLimit / currentScale;\r
+ else\r
+ return inLimit / 100.0;\r
+ } else if (newScale < currentScale && newScale < outLimit) {\r
+ if (currentScale > outLimit)\r
+ scaleFactor = outLimit / currentScale;\r
+ else\r
+ return outLimit / 100.0;\r
+ }\r
+ return zoomLevel * scaleFactor;\r
+ }\r
+ };\r
+ canvas.addListener(SWT.MouseUp, listener);\r
+ canvas.addListener(SWT.MouseDown, listener);\r
+ canvas.addListener(SWT.MouseMove, listener);\r
+ canvas.addListener(SWT.MouseWheel, listener);\r
+\r
+ // Start tracking editor input validity.\r
+ activateValidation();\r
+\r
+ loadAndTrackInput();\r
+ }\r
+\r
+ private void loadAndTrackInput() {\r
+ final Resource input = getInputResource();\r
+ Simantics.getSession().asyncRequest(new ResourceRead<Object>(input) {\r
+ @Override\r
+ public Object perform(ReadGraph graph) throws DatabaseException {\r
+ ImageResource img = ImageResource.getInstance(graph);\r
+ if (graph.isInstanceOf(input, img.SvgImage)) {\r
+ String text = graph.getPossibleValue(input, Bindings.STRING);\r
+ return text;\r
+ } else if (graph.isInstanceOf(input, img.Image)) {\r
+ byte data[] = graph.getPossibleValue(input, Bindings.BYTE_ARRAY);\r
+ return data;\r
+ }\r
+ return null;\r
+ }\r
+ }, new org.simantics.db.procedure.Listener<Object>() {\r
+ @Override\r
+ public void execute(Object result) {\r
+ if (result instanceof String) {\r
+ // svg text\r
+ try {\r
+ svgDiagram = ScenegraphUtils.loadSVGDiagram(svgUniverse, (String) result);\r
+ scheduleRedraw();\r
+ } catch (SVGException e) {\r
+ ErrorLogger.defaultLogError(e);\r
+ }\r
+ } else if (result instanceof byte[]) {\r
+ // bitmap image data\r
+ Display display = canvas.getDisplay();\r
+ if (!display.isDisposed()) {\r
+ try {\r
+ image = new Image(canvas.getDisplay(), new ByteArrayInputStream((byte[]) result));\r
+ scheduleRedraw();\r
+ } catch (SWTException e) {\r
+ ErrorLogger.defaultLogError(e);\r
+ } catch (SWTError e) {\r
+ ErrorLogger.defaultLogError(e);\r
+ }\r
+ }\r
+ }\r
+ }\r
+ private void scheduleRedraw() {\r
+ Display d = canvas.getDisplay();\r
+ if (!d.isDisposed())\r
+ d.asyncExec(new Runnable() {\r
+ @Override\r
+ public void run() {\r
+ if (!canvas.isDisposed())\r
+ canvas.redraw();\r
+ }\r
+ });\r
+ }\r
+ @Override\r
+ public void exception(Throwable t) {\r
+ ErrorLogger.defaultLogError(t);\r
+ }\r
+ @Override\r
+ public boolean isDisposed() {\r
+ return disposed;\r
+ }\r
+ });\r
+ }\r
+\r
+ @Override\r
+ public void dispose() {\r
+ disposed = true;\r
+ if (image != null) {\r
+ image.dispose();\r
+ image = null;\r
+ }\r
+ svgUniverse.clear();\r
+ }\r
+\r
+ @Override\r
+ public void setFocus() {\r
+ if (canvas != null)\r
+ canvas.setFocus();\r
+ }\r
+\r
+ @SuppressWarnings("rawtypes")\r
+ @Override\r
+ public Object getAdapter(Class adapter) {\r
+ return null;\r
+ }\r
+\r
+}\r