1   package org.opentrafficsim.draw.road;
2   
3   import java.awt.BasicStroke;
4   import java.awt.Color;
5   import java.awt.Graphics2D;
6   import java.awt.Stroke;
7   import java.awt.geom.AffineTransform;
8   import java.awt.image.ImageObserver;
9   import java.io.Serializable;
10  import java.rmi.RemoteException;
11  
12  import javax.naming.NamingException;
13  
14  import org.djunits.unit.LengthUnit;
15  import org.djunits.value.vdouble.scalar.Length;
16  import org.opentrafficsim.draw.core.PaintPolygons;
17  import org.opentrafficsim.road.network.lane.conflict.Conflict;
18  import org.opentrafficsim.road.network.lane.conflict.ConflictType;
19  
20  import nl.tudelft.simulation.dsol.simulators.SimulatorInterface;
21  
22  
23  
24  
25  
26  
27  
28  
29  
30  
31  
32  
33  public class ConflictAnimation extends AbstractLineAnimation<Conflict> implements Serializable
34  {
35      
36  
37      
38      private static final long serialVersionUID = 20161207L;
39  
40      
41  
42  
43  
44  
45  
46      public ConflictAnimation(final Conflict source, final SimulatorInterface.TimeDoubleUnit simulator)
47              throws NamingException, RemoteException
48      {
49          super(source, simulator, .9, new Length(0.5, LengthUnit.SI));
50      }
51  
52      
53      @Override
54      public final void paint(final Graphics2D graphics, final ImageObserver observer) throws RemoteException
55      {
56          Conflict conflict = this.getSource();
57  
58  
59  
60  
61  
62          Color fillColor;
63          switch (conflict.conflictPriority())
64          {
65              case SPLIT:
66                  fillColor = Color.blue;
67                  break;
68  
69              case PRIORITY:
70                  fillColor = Color.green;
71                  break;
72  
73              case YIELD:
74                  fillColor = Color.orange;
75                  break;
76  
77              default:
78                  
79                  fillColor = Color.red;
80                  break;
81          }
82  
83          graphics.setColor(fillColor);
84          super.paint(graphics, observer);
85  
86          Stroke oldStroke = graphics.getStroke();
87  
88          BasicStroke stroke;
89          float factor = conflict.isPermitted() ? .5f : 1f;
90          if (conflict.getConflictType().equals(ConflictType.CROSSING))
91          {
92              stroke = new BasicStroke(.1f, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND, 1.0f,
93                      new float[] { factor * 1.0f, factor * 2.0f }, 0.0f);
94          }
95          else
96          {
97              stroke = new BasicStroke(.1f, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND, 1.0f,
98                      new float[] { factor * 1.0f, factor * 0.95f, factor * 0.1f, factor * 0.95f }, 0.0f);
99          }
100         graphics.setStroke(stroke);
101         AffineTransform saveAT = graphics.getTransform();
102         double angle = -getSource().getLocation().getRotZ();
103         if (isRotate() && angle != 0.0)
104         {
105             graphics.rotate(-angle);
106         }
107         if (conflict.getGeometry() != null)
108         {
109             PaintPolygons.paintMultiPolygon(graphics, fillColor, conflict.getLocation(), conflict.getGeometry(), false);
110             
111             
112             
113             
114             
115             
116             
117             
118             
119             
120         }
121         if (isRotate() && angle != 0.0)
122         {
123             graphics.rotate(+angle);
124         }
125         graphics.setStroke(oldStroke);
126         graphics.setTransform(saveAT);
127     }
128 
129     
130     @Override
131     public final String toString()
132     {
133         return "ConflictAnimation [getSource()=" + getSource() + "]";
134     }
135 
136 }