bundles/org.simantics.g2d/src/org/simantics/g2d/routing/test/TestStopSet.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.g2d.routing.test;\r
  13 \r
  14 import java.awt.BasicStroke;\r
  15 import java.awt.Color;\r
  16 import java.awt.Frame;\r
  17 import java.awt.Graphics;\r
  18 import java.awt.Graphics2D;\r
  19 import java.awt.Image;\r
  20 import java.awt.RenderingHints;\r
  21 import java.awt.event.MouseEvent;\r
  22 import java.awt.event.MouseListener;\r
  23 import java.awt.event.MouseMotionListener;\r
  24 import java.awt.event.WindowEvent;\r
  25 import java.awt.event.WindowListener;\r
  26 import java.awt.geom.Rectangle2D;\r
  27 import java.util.ArrayList;\r
  28 import java.util.Collection;\r
  29 import java.util.HashMap;\r
  30 import java.util.Map;\r
  31 \r
  32 import org.simantics.g2d.routing.algorithm1.Penalty;\r
  33 import org.simantics.g2d.routing.algorithm1.Rectangle;\r
  34 import org.simantics.g2d.routing.algorithm1.StopSet;\r
  35 import org.simantics.g2d.routing.algorithm1.StopSet.IStopProcedure;\r
  36 import org.simantics.g2d.routing.algorithm1.StopSet.Line;\r
  37 import org.simantics.g2d.routing.algorithm1.StopSet.Stop;\r
  38 \r
  39 public class TestStopSet extends Frame {\r
  40 \r
  41         private static final long serialVersionUID = 2181877722124429003L;\r
  42         \r
  43         Collection<Rectangle> rectangles = new ArrayList<Rectangle>();\r
  44         Rectangle source;\r
  45         Rectangle target;\r
  46                 \r
  47         double startX, startY;\r
  48         double curX, curY;\r
  49         int mouseButtons = 0;\r
  50 \r
  51         public TestStopSet() {\r
  52                 addWindowListener(new WindowListener() {\r
  53 \r
  54                         @Override\r
  55                         public void windowActivated(WindowEvent e) {\r
  56                         }\r
  57 \r
  58                         @Override\r
  59                         public void windowClosed(WindowEvent e) {               \r
  60                         }\r
  61 \r
  62                         @Override\r
  63                         public void windowClosing(WindowEvent e) {\r
  64                                 System.exit(0);                 \r
  65                         }\r
  66 \r
  67                         @Override\r
  68                         public void windowDeactivated(WindowEvent e) {\r
  69                         }\r
  70 \r
  71                         @Override\r
  72                         public void windowDeiconified(WindowEvent e) {\r
  73                         }\r
  74 \r
  75                         @Override\r
  76                         public void windowIconified(WindowEvent e) {\r
  77                         }\r
  78 \r
  79                         @Override\r
  80                         public void windowOpened(WindowEvent e) {\r
  81                         }\r
  82                         \r
  83                 });\r
  84                 \r
  85                 addMouseListener(new MouseListener() {\r
  86 \r
  87                         @Override\r
  88                         public void mouseClicked(MouseEvent e) {\r
  89                                 // TODO Auto-generated method stub\r
  90                                 \r
  91                         }\r
  92 \r
  93                         @Override\r
  94                         public void mouseEntered(MouseEvent e) {\r
  95                                 // TODO Auto-generated method stub\r
  96                                 \r
  97                         }\r
  98 \r
  99                         @Override\r
 100                         public void mouseExited(MouseEvent e) {\r
 101                                 // TODO Auto-generated method stub\r
 102                                 \r
 103                         }\r
 104 \r
 105                         @Override\r
 106                         public void mousePressed(MouseEvent e) {\r
 107                                 mouseButtons = e.getButton();\r
 108                                 curX = startX = e.getX();\r
 109                                 curY = startY = e.getY();\r
 110                                 if(e.getButton() == MouseEvent.BUTTON2)\r
 111                                         source = null;\r
 112                                 else if(e.getButton() == MouseEvent.BUTTON3)\r
 113                                         target = null;\r
 114                                 repaint();\r
 115                         }\r
 116 \r
 117                         @Override\r
 118                         public void mouseReleased(MouseEvent e) {\r
 119                                 mouseButtons = 0;\r
 120                                 double x0 = startX;\r
 121                                 double y0 = startY;\r
 122                                 double x1 = e.getX();\r
 123                                 double y1 = e.getY();\r
 124                                 if(x0 > x1) {\r
 125                                         double temp = x0;\r
 126                                         x0 = x1;\r
 127                                         x1 = temp;\r
 128                                 }\r
 129                                 if(y0 > y1) {\r
 130                                         double temp = y0;\r
 131                                         y0 = y1;\r
 132                                         y1 = temp;\r
 133                                 }\r
 134                                 Rectangle rect = new Rectangle(x0, y0, x1, y1);\r
 135                                 \r
 136                                 if(e.getButton() == MouseEvent.BUTTON1)                         \r
 137                                         rectangles.add(rect);\r
 138                                 else if(e.getButton() == MouseEvent.BUTTON2)\r
 139                                         source = rect;\r
 140                                 else if(e.getButton() == MouseEvent.BUTTON3)\r
 141                                         target = rect;\r
 142                                 \r
 143                                 repaint();                              \r
 144                         }\r
 145                         \r
 146                 });\r
 147                 \r
 148                 addMouseMotionListener(new MouseMotionListener() {\r
 149 \r
 150                         @Override\r
 151                         public void mouseDragged(MouseEvent e) {\r
 152                                 curX = e.getX();\r
 153                                 curY = e.getY();\r
 154                                 repaint();\r
 155                         }\r
 156 \r
 157                         @Override\r
 158                         public void mouseMoved(MouseEvent e) {                                          \r
 159                         }\r
 160                         \r
 161                 });\r
 162                 \r
 163                 setSize(640, 480);\r
 164                 setVisible(true);\r
 165         }\r
 166         \r
 167         @Override\r
 168         public void paint(Graphics _g) {\r
 169                 final Graphics2D g = (Graphics2D)_g;\r
 170                 Map<Object, Object> hints = new HashMap<Object, Object>();\r
 171                 hints.put(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);\r
 172                 hints.put(RenderingHints.KEY_RENDERING, RenderingHints.VALUE_RENDER_QUALITY);\r
 173                 g.addRenderingHints(hints);\r
 174                 \r
 175                 /*Path2D path = new Path2D.Double();\r
 176                 path.moveTo(0.0, 0.0);\r
 177                 path.lineTo(100.0, 100.0);\r
 178                 path.lineTo(400.0, 100.0);\r
 179                                 \r
 180                 g.draw(path);\r
 181                 */\r
 182                 \r
 183                 g.setColor(Color.GRAY);\r
 184                 g.setStroke(new BasicStroke(1.0f));\r
 185                 for(Rectangle rect : rectangles)\r
 186                         drawRectangle(g, rect.x0, rect.y0, rect.x1, rect.y1);           \r
 187                 \r
 188                 if(source != null) {\r
 189                         g.setColor(Color.BLUE);\r
 190                         drawRectangle(g, source.x0, source.y0, source.x1, source.y1);\r
 191                 }\r
 192                 if(target != null) {\r
 193                         g.setColor(Color.GREEN);\r
 194                         drawRectangle(g, target.x0, target.y0, target.x1, target.y1);\r
 195                 }\r
 196                 if(mouseButtons != 0) {\r
 197                         if(mouseButtons == 1)\r
 198                                 g.setColor(Color.BLACK);\r
 199                         else if(mouseButtons == 2)\r
 200                                 g.setColor(Color.BLUE);\r
 201                         else if(mouseButtons == 3)\r
 202                                 g.setColor(Color.GREEN);\r
 203                         drawRectangle(g, startX, startY, curX, curY);\r
 204                 }\r
 205                 \r
 206                 Collection<Stop> stops = new ArrayList<Stop>(rectangles.size());\r
 207                 Penalty penalty = new Penalty(100.0, 0.0);\r
 208                 for(Rectangle rect : rectangles)\r
 209                         stops.add(new Stop(penalty, rect.x0, rect.x1, rect.y0));\r
 210                 StopSet ss = new StopSet(stops);\r
 211                 \r
 212                 if(target != null) {\r
 213                         g.setColor(Color.GREEN);\r
 214                         g.setStroke(new BasicStroke(1.0f));\r
 215                         drawFront(g, ss, target.x0, target.x1, target.y1);\r
 216                 }\r
 217         }\r
 218         \r
 219         public final double PADDING = 2.0;\r
 220         \r
 221         void drawFront(final Graphics2D g, final StopSet ss, final double x0, final double x1, final double y) {\r
 222                 ss.findStops(x0, x1, y, new IStopProcedure() {\r
 223 \r
 224                         @Override\r
 225                         public void blockEnd(double y1) {\r
 226                                 if(y1 == Double.POSITIVE_INFINITY)\r
 227                                         y1 = 100000.0;\r
 228                                 Rectangle2D rect = new Rectangle2D.Double(x0 + PADDING, y + PADDING, x1-x0 - PADDING*2, y1-y - PADDING*2);\r
 229                                 g.draw(rect);                                                   \r
 230                         }\r
 231 \r
 232                         @Override\r
 233                         public void continuation(double min, double max, int pos, Line line) {\r
 234                                 drawFront(g, ss, pos, line, min, max);\r
 235                         }\r
 236                         \r
 237                 });\r
 238         }\r
 239         \r
 240         void drawFront(final Graphics2D g, final StopSet ss, int pos, final Line line, final double x0, final double x1) {\r
 241                 StopSet.continueStop(pos, line, x0, x1, new IStopProcedure() {\r
 242 \r
 243                         @Override\r
 244                         public void blockEnd(double y1) {\r
 245                                 if(y1 == Double.POSITIVE_INFINITY)\r
 246                                         y1 = 100000.0;\r
 247                                 Rectangle2D rect = new Rectangle2D.Double(x0 + PADDING, line.y + PADDING, x1-x0 - PADDING*2, y1-line.y - PADDING*2);\r
 248                                 g.draw(rect);                                                   \r
 249                         }\r
 250 \r
 251                         @Override\r
 252                         public void continuation(double min, double max, int pos, Line line) {\r
 253                                 drawFront(g, ss, pos, line, min, max);\r
 254                         }\r
 255                         \r
 256                 });\r
 257         }\r
 258         \r
 259         static void drawRectangle(Graphics2D g, double x0, double y0, double x1, double y1) {\r
 260                 if(x0 > x1) {\r
 261                         double temp = x0;\r
 262                         x0 = x1;\r
 263                         x1 = temp;\r
 264                 }\r
 265                 if(y0 > y1) {\r
 266                         double temp = y0;\r
 267                         y0 = y1;\r
 268                         y1 = temp;\r
 269                 }\r
 270                 Rectangle2D rect = new Rectangle2D.Double(x0, y0, x1-x0, y1-y0);\r
 271                 g.fill(rect);                   \r
 272         }\r
 273         \r
 274         Image offScreenImage;\r
 275         int offScreenImageWidth;\r
 276         int offScreenImageHeight;\r
 277 \r
 278         public void update( Graphics g ) {\r
 279                 if ( offScreenImage == null || getWidth() != offScreenImageWidth || getHeight() != offScreenImageHeight) {\r
 280                         offScreenImageWidth = getWidth();\r
 281                         offScreenImageHeight = getHeight();\r
 282                         offScreenImage = createImage(offScreenImageWidth, offScreenImageHeight);\r
 283                 }\r
 284 \r
 285                 Graphics gOffScreenImage = offScreenImage.getGraphics();\r
 286 \r
 287                 gOffScreenImage.clearRect(0, 0, getWidth(), getHeight());\r
 288                 paint( gOffScreenImage );\r
 289 \r
 290                 g.drawImage(offScreenImage, 0, 0, this);\r
 291 \r
 292                 gOffScreenImage.dispose();\r
 293         } \r
 294         \r
 295         public static void main(String[] args) {\r
 296                 new TestStopSet();\r
 297         }\r
 298         \r
 299 }\r