bundles/org.simantics.diagram/src/org/simantics/diagram/elements/TextGridNode.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.diagram.elements;\r
  13 \r
  14 import gnu.trove.map.TIntObjectMap;\r
  15 import gnu.trove.map.TMap;\r
  16 import gnu.trove.map.hash.THashMap;\r
  17 import gnu.trove.map.hash.TIntObjectHashMap;\r
  18 import gnu.trove.procedure.TObjectProcedure;\r
  19 \r
  20 import java.awt.Color;\r
  21 import java.awt.Font;\r
  22 import java.awt.Graphics2D;\r
  23 import java.awt.geom.AffineTransform;\r
  24 import java.awt.geom.Point2D;\r
  25 import java.awt.geom.Rectangle2D;\r
  26 import java.util.ArrayList;\r
  27 import java.util.List;\r
  28 \r
  29 import org.simantics.datatypes.literal.Vec2d;\r
  30 import org.simantics.db.layer0.variable.RVI;\r
  31 import org.simantics.diagram.profile.MonitorTextGridResult;\r
  32 import org.simantics.scenegraph.ExportableWidget.OutputWidget;\r
  33 import org.simantics.scenegraph.g2d.G2DParentNode;\r
  34 import org.simantics.scenegraph.g2d.G2DSceneGraph;\r
  35 import org.simantics.scenegraph.g2d.events.EventTypes;\r
  36 import org.simantics.scenegraph.g2d.events.KeyEvent.KeyPressedEvent;\r
  37 import org.simantics.scenegraph.g2d.events.MouseEvent;\r
  38 import org.simantics.scenegraph.g2d.events.MouseEvent.MouseButtonReleasedEvent;\r
  39 import org.simantics.scenegraph.g2d.events.MouseEvent.MouseDragBegin;\r
  40 import org.simantics.scenegraph.g2d.events.MouseEvent.MouseMovedEvent;\r
  41 import org.simantics.scenegraph.g2d.nodes.Decoration;\r
  42 import org.simantics.scenegraph.utils.NodeUtil;\r
  43 import org.simantics.scl.runtime.function.Function1;\r
  44 \r
  45 /**\r
  46  * TODO: does not work remotely, e.g. rowIds \r
  47  */\r
  48 @OutputWidget("textGrid")\r
  49 public class TextGridNode extends G2DParentNode implements Decoration {\r
  50 \r
  51     public static class A extends TextNode {\r
  52 \r
  53         MonitorTextGridResult cache = null;\r
  54         \r
  55         private static final long serialVersionUID = -4519849713591842241L;\r
  56 \r
  57         public MonitorTextGridResult getCache() {\r
  58                 return cache;\r
  59         }\r
  60         \r
  61         public void setCache(MonitorTextGridResult cache) {\r
  62                 this.cache = cache;\r
  63         }\r
  64 \r
  65     }\r
  66 \r
  67     private static final int CACHED_COLUMNS = 4;\r
  68     private static final int CACHED_ROWS = 10;\r
  69 \r
  70     private static Cell[][] cache;\r
  71 \r
  72     static {\r
  73         cache = new Cell[CACHED_ROWS][];\r
  74         for (int row = 0; row < CACHED_ROWS; ++row) {\r
  75             cache[row] = new Cell[CACHED_COLUMNS];\r
  76             for (int col = 0; col < CACHED_COLUMNS; ++col) {\r
  77                 cache[row][col] = new Cell(col+1, row+1);\r
  78             }\r
  79         }\r
  80     }\r
  81 \r
  82     private static String makeId(int x, int y) {\r
  83         return x + "#" + y;\r
  84     }\r
  85 \r
  86     static class Cell {\r
  87 \r
  88         public final int x;\r
  89         public final int y;\r
  90 \r
  91         public Cell(int x, int y) {\r
  92             this.x = x;\r
  93             this.y = y;\r
  94         }\r
  95 \r
  96         @Override\r
  97         public boolean equals(Object object) {\r
  98             if (this == object)\r
  99                 return true;\r
 100             else if (object == null)\r
 101                 return false;\r
 102             else if (Cell.class != object.getClass())\r
 103                 return false;\r
 104             Cell p = (Cell)object;\r
 105             return x == p.x && y == p.y;\r
 106         }\r
 107 \r
 108         @Override\r
 109         public int hashCode() {\r
 110             return x *31 + y;\r
 111         }\r
 112 \r
 113         @Override\r
 114         public String toString() {\r
 115             return makeId(x, y);\r
 116         }\r
 117 \r
 118         public static Cell make(int x, int y) {\r
 119             if (x >= 1 && x <= CACHED_COLUMNS && y >= 1 && y <= CACHED_ROWS)\r
 120                 return cache[y-1][x-1];\r
 121             return new Cell(x,y);\r
 122         }\r
 123     }\r
 124 \r
 125     private static final long serialVersionUID = 7015425802228571055L;\r
 126 \r
 127     private TMap<Cell, A> nodes = new THashMap<Cell, A>();\r
 128     private TIntObjectMap<String> rowIds = new TIntObjectHashMap<>();\r
 129     private boolean up = true;\r
 130 \r
 131     static class MaxY implements TObjectProcedure<Cell> {\r
 132         int max = 0;\r
 133         @Override\r
 134         public boolean execute(Cell key) {\r
 135             if (key.y > max)\r
 136                 max = key.y;\r
 137             return true;\r
 138         }\r
 139     }\r
 140 \r
 141     private int computeRows() {\r
 142         MaxY maxy = new MaxY();\r
 143         nodes.forEachKey(maxy);\r
 144         return maxy.max;\r
 145     }\r
 146 \r
 147     private List<Cell> peekRowCells(int y) {\r
 148         ArrayList<Cell> row = new ArrayList<Cell>(4);\r
 149         for (Cell key : nodes.keySet())\r
 150             if (key.y == y)\r
 151                 row.add(key);\r
 152         return row;\r
 153     }\r
 154     \r
 155     public void setUp(boolean up) {\r
 156         this.up = up;\r
 157     }\r
 158 \r
 159     public A get(int x, int y) {\r
 160         Cell p = Cell.make(x, y);\r
 161         A node = nodes.get(p);\r
 162         if(node == null) {\r
 163             //System.out.println(" create(" + x + "," + y + ")");\r
 164             node = getOrCreateNode(p.toString(), A.class);\r
 165             node.setZIndex(x + (y-1)*100);\r
 166             nodes.put(p, node);\r
 167         } else {\r
 168             //System.out.println(" get(" + x + "," + y + ")");\r
 169         }\r
 170         return node;\r
 171     }\r
 172 \r
 173     public MonitorTextGridResult getCache(int x, int y) {\r
 174         return get(x,y).getCache();\r
 175     }\r
 176     \r
 177     public void setCache(int x, int y, MonitorTextGridResult cache) {\r
 178         get(x,y).setCache(cache);\r
 179     }\r
 180 \r
 181     public void setTransform(int x, int y, AffineTransform transform) {\r
 182         get(x,y).setTransform(transform);\r
 183         dragBegin = null;\r
 184         currentDrag = null;\r
 185     }\r
 186 \r
 187     public void setHorizontalAlignment(int x, int y, byte horizontalAlignment) {\r
 188         get(x,y).setHorizontalAlignment(horizontalAlignment);\r
 189     }\r
 190 \r
 191     public void setVerticalAlignment(int x, int y, byte verticalAlignment) {\r
 192         get(x,y).setVerticalAlignment(verticalAlignment);\r
 193     }\r
 194 \r
 195     public void setEditable(int x, int y, boolean editable) {\r
 196         get(x,y).setEditable(editable);\r
 197     }\r
 198     \r
 199     public void setTextListener(int x, int y, ITextListener listener) {\r
 200         get(x,y).setTextListener(listener);\r
 201     }\r
 202 \r
 203     public void setInputValidator(int x, int y, Function1<String, String> validator) {\r
 204         get(x,y).setValidator(validator);\r
 205     }\r
 206     \r
 207     public void setTranslator(Function1<Vec2d, Boolean> translator) {\r
 208         this.translator = translator;\r
 209     }\r
 210 \r
 211     public void setContentFilter(int x, int y, ITextContentFilter filter) {\r
 212         get(x,y).setContentFilter(filter);\r
 213     }\r
 214 \r
 215     public void setRVI(int x, int y, RVI rvi) {\r
 216         get(x,y).setRVI(rvi);\r
 217     }\r
 218 \r
 219     public void setBackgroundColor(int x, int y, Color color) {\r
 220         get(x,y).setBackgroundColor(color);\r
 221     }\r
 222 \r
 223     @SyncField("font")\r
 224     public void setFont(int x, int y, Font font) {\r
 225         get(x,y).setFont(font);\r
 226     }\r
 227 \r
 228     @SyncField("text")\r
 229     public void setText(int x, int y, String text) {\r
 230         get(x,y).setText(text);\r
 231     }\r
 232 \r
 233     @SyncField("text")\r
 234     public void setPending(int x, int y, boolean pending) {\r
 235         get(x,y).setPending(pending);\r
 236     }\r
 237 \r
 238     @SyncField("color")\r
 239     public void setColor(int x, int y, Color color) {\r
 240         get(x,y).setColor(color);\r
 241     }\r
 242 \r
 243     public void removeRow(int y) {\r
 244         List<Cell> row = peekRowCells(y);\r
 245         if (row.isEmpty())\r
 246             return;\r
 247         //System.out.println("removeRow(" + y + "): removing " + row.size() + " cells");\r
 248         for (Cell cell : row) {\r
 249             nodes.remove(cell);\r
 250             removeNode(cell.toString());\r
 251         }\r
 252     }\r
 253 \r
 254     public void setRowId(int y, String id) {\r
 255         rowIds.put(y, id);\r
 256     }\r
 257 \r
 258     public String getRowId(int y) {\r
 259         return rowIds.get(y);\r
 260     }\r
 261     \r
 262     @Override\r
 263     public void render(Graphics2D g2d) {\r
 264         Vec2d delta = getDelta(FACTOR);\r
 265         if(delta != null)\r
 266                 setTransform(AffineTransform.getTranslateInstance(delta.x, delta.y-2.1*computeRows()*(up ? 1.0 : 0.0)));\r
 267         else\r
 268                 setTransform(AffineTransform.getTranslateInstance(0, -2.1*computeRows()*(up ? 1.0 : 0.0)));\r
 269         super.render(g2d);\r
 270     }\r
 271     \r
 272     @Override\r
 273     public Rectangle2D getBoundsInLocal(boolean b) {\r
 274         return null;\r
 275     }\r
 276     \r
 277     @Override\r
 278     public Rectangle2D getBoundsInLocal() {\r
 279         return null;\r
 280     }\r
 281     \r
 282     @Override\r
 283     public Rectangle2D getBounds() {\r
 284         return null;\r
 285     }\r
 286     \r
 287     @Override\r
 288     public int getEventMask() {\r
 289         return EventTypes.MouseDragBeginMask | EventTypes.KeyPressedMask;\r
 290     }\r
 291     \r
 292     private static boolean isEventDummy(MouseDragBegin e) {\r
 293         if (e.controlPosition.distance(0, 0) == 0 \r
 294                         && e.screenPosition.distance(0, 0) == 0\r
 295                         && e.buttons == 0) {\r
 296                 return true;\r
 297         } else {\r
 298                 return false;\r
 299         }\r
 300     }\r
 301     \r
 302     private boolean dragging = false;\r
 303     private Point2D dragBegin = null;\r
 304     private Point2D currentDrag = null;\r
 305     private Function1<Vec2d, Boolean> translator = null;\r
 306     \r
 307     private static double FACTOR = 1.0; \r
 308     private static double FACTOR2 = 7.0;\r
 309     \r
 310     private Vec2d getDelta(double factor) {\r
 311             if(dragBegin != null && currentDrag != null) {\r
 312                 double dx =  factor * (currentDrag.getX() - dragBegin.getX());\r
 313                 double dy = factor * (currentDrag.getY() - dragBegin.getY());\r
 314                 return new Vec2d(dx, dy);\r
 315             } else {\r
 316                 return null;\r
 317             }\r
 318     }\r
 319     \r
 320     @Override\r
 321     protected boolean keyPressed(KeyPressedEvent e) {\r
 322         if (dragging && e.keyCode == java.awt.event.KeyEvent.VK_ESCAPE) {\r
 323                 dragBegin = null;\r
 324                 currentDrag = null;\r
 325                 dragging = false;\r
 326                 repaint();\r
 327                 return true;\r
 328         }\r
 329         return false;\r
 330     }\r
 331 \r
 332     protected boolean mouseMoved(MouseMovedEvent event) {\r
 333         \r
 334         if (dragging) {\r
 335                 currentDrag = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());\r
 336                 repaint();\r
 337         }\r
 338         \r
 339         return false;\r
 340         \r
 341     }\r
 342     \r
 343     protected boolean hitTest(MouseEvent event, double tolerance) {\r
 344         \r
 345         Rectangle2D bounds = super.getBoundsInLocal(false);\r
 346         if(bounds == null) return false;\r
 347         Point2D localPos = NodeUtil.worldToLocal(this, event.controlPosition, new Point2D.Double());\r
 348         double x = localPos.getX();\r
 349         double y = localPos.getY()+2.1*computeRows()*(up ? 1.0 : 0.0);\r
 350         boolean hit = bounds.contains(x, y);\r
 351         return hit;\r
 352         \r
 353     }\r
 354     \r
 355     @Override\r
 356     protected boolean mouseDragged(MouseDragBegin e) {\r
 357         \r
 358         // Get rid of dummy events from dragGestureRecognized\r
 359         if (isEventDummy(e)) {\r
 360                 return false;\r
 361         }\r
 362         \r
 363         G2DSceneGraph sg = NodeUtil.getRootNode(this);\r
 364         Boolean b = sg.getGlobalProperty(G2DSceneGraph.IGNORE_FOCUS, false);\r
 365         if(!b) return false;\r
 366         \r
 367         if(!e.hasAnyButton(MouseEvent.LEFT_MASK) || e.hasAnyModifier(MouseEvent.ALL_MODIFIERS_MASK) || !hitTest(e, 0.0)) return false;\r
 368         \r
 369         dragBegin = NodeUtil.worldToLocal(this, e.controlPosition, new Point2D.Double());\r
 370         dragging = true;\r
 371         return true;\r
 372         \r
 373     }\r
 374     \r
 375     protected boolean mouseButtonReleased(MouseButtonReleasedEvent e) {\r
 376         \r
 377         if(!dragging) return false;\r
 378         \r
 379         Vec2d delta = getDelta(FACTOR2);\r
 380         if(delta != null && translator != null) {\r
 381                 translator.apply(delta); \r
 382         } else {\r
 383                 dragBegin = null;\r
 384                 currentDrag = null;\r
 385         }\r
 386         dragging = false;\r
 387         return false;\r
 388     }\r
 389     \r
 390     @Override\r
 391     public void init() {\r
 392         super.init();\r
 393         addEventHandler(this);\r
 394     }\r
 395 \r
 396     @Override\r
 397     public void cleanup() {\r
 398         removeEventHandler(this);\r
 399         super.cleanup();\r
 400     }\r
 401     \r
 402 }\r