1 package org.opentrafficsim.demo;
2
3 import java.awt.Dimension;
4 import java.awt.event.ActionEvent;
5 import java.rmi.RemoteException;
6 import java.util.ArrayList;
7 import java.util.List;
8
9 import javax.naming.NamingException;
10 import javax.swing.Box;
11 import javax.swing.ButtonGroup;
12 import javax.swing.JLabel;
13 import javax.swing.JOptionPane;
14 import javax.swing.JPanel;
15 import javax.swing.JRadioButton;
16
17 import org.djunits.unit.util.UNITS;
18 import org.djunits.value.vdouble.scalar.Duration;
19 import org.djunits.value.vdouble.scalar.Time;
20 import org.opentrafficsim.core.dsol.OTSAnimator;
21 import org.opentrafficsim.core.dsol.OTSSimulatorInterface;
22 import org.opentrafficsim.core.gtu.GTUDirectionality;
23 import org.opentrafficsim.core.network.NetworkException;
24 import org.opentrafficsim.draw.core.OTSDrawingException;
25 import org.opentrafficsim.draw.graphs.GraphPath;
26 import org.opentrafficsim.draw.graphs.TrajectoryPlot;
27 import org.opentrafficsim.draw.graphs.road.GraphLaneUtil;
28 import org.opentrafficsim.kpi.sampling.KpiLaneDirection;
29 import org.opentrafficsim.road.network.OTSRoadNetwork;
30 import org.opentrafficsim.road.network.lane.LaneDirection;
31 import org.opentrafficsim.road.network.sampling.RoadSampler;
32 import org.opentrafficsim.swing.graphs.SwingPlot;
33 import org.opentrafficsim.swing.gui.OTSAnimationPanel;
34 import org.opentrafficsim.swing.gui.OTSSimulationApplication;
35
36 import nl.tudelft.simulation.dsol.SimRuntimeException;
37 import nl.tudelft.simulation.dsol.model.inputparameters.InputParameter;
38 import nl.tudelft.simulation.dsol.model.inputparameters.InputParameterMap;
39 import nl.tudelft.simulation.dsol.model.inputparameters.InputParameterSelectionMap;
40 import nl.tudelft.simulation.dsol.swing.gui.TablePanel;
41 import nl.tudelft.simulation.dsol.swing.gui.inputparameters.AbstractInputField;
42 import nl.tudelft.simulation.dsol.swing.gui.inputparameters.InputField;
43 import nl.tudelft.simulation.dsol.swing.gui.inputparameters.TabbedParameterDialog;
44 import nl.tudelft.simulation.language.DSOLException;
45
46
47
48
49
50
51
52
53
54
55
56 public class NetworksSwing extends OTSSimulationApplication<NetworksModel> implements UNITS
57 {
58
59 private static final long serialVersionUID = 1L;
60
61
62
63
64
65
66
67
68 public NetworksSwing(final String title, final OTSAnimationPanel panel, final NetworksModel model)
69 throws OTSDrawingException
70 {
71 super(model, panel);
72 OTSRoadNetwork network = model.getNetwork();
73 System.out.println(network.getLinkMap());
74 }
75
76
77 @Override
78 protected void addTabs()
79 {
80 addStatisticsTabs(getModel().getSimulator());
81 }
82
83
84
85
86
87 public static void main(final String[] args)
88 {
89 demo(true);
90 }
91
92
93
94
95
96 public static void demo(final boolean exitOnClose)
97 {
98 try
99 {
100 OTSAnimator simulator = new OTSAnimator("NetworksSwing");
101 final NetworksModelml#NetworksModel">NetworksModel otsModel = new NetworksModel(simulator);
102 if (NetworksParameterDialog.process(otsModel.getInputParameterMap()))
103 {
104 simulator.initialize(Time.ZERO, Duration.ZERO, Duration.instantiateSI(3600.0), otsModel);
105 OTSAnimationPanel animationPanel = new OTSAnimationPanel(otsModel.getNetwork().getExtent(), new Dimension(800,
106 600), simulator, otsModel, DEFAULT_COLORER, otsModel.getNetwork());
107 NetworksSwingng.html#NetworksSwing">NetworksSwing app = new NetworksSwing("Networks", animationPanel, otsModel);
108 app.setExitOnClose(exitOnClose);
109 }
110 else
111 {
112 if (exitOnClose)
113 {
114 System.exit(0);
115 }
116 }
117 }
118 catch (SimRuntimeException | NamingException | RemoteException | OTSDrawingException | DSOLException exception)
119 {
120 exception.printStackTrace();
121 }
122 }
123
124
125
126
127
128 protected final void addStatisticsTabs(final OTSSimulatorInterface simulator)
129 {
130 int graphCount = getModel().pathCount();
131 int columns = 1;
132 int rows = 0 == columns ? 0 : (int) Math.ceil(graphCount * 1.0 / columns);
133 TablePanel charts = new TablePanel(columns, rows);
134 RoadSampler sampler = new RoadSampler(getModel().getNetwork());
135 Duration updateInterval = Duration.instantiateSI(10.0);
136 for (int graphIndex = 0; graphIndex < graphCount; graphIndex++)
137 {
138 List<LaneDirection> start = new ArrayList<>();
139 start.add(new LaneDirection(getModel().getPath(graphIndex).get(0), GTUDirectionality.DIR_PLUS));
140 GraphPath<KpiLaneDirection> path;
141 try
142 {
143 path = GraphLaneUtil.createPath("name", start.get(0));
144 }
145 catch (NetworkException exception)
146 {
147 throw new RuntimeException(exception);
148 }
149 GraphPath.initRecording(sampler, path);
150 SwingPlot plot = new SwingPlot(new TrajectoryPlot("Trajectories on lane " + (graphIndex + 1), updateInterval,
151 simulator, sampler.getSamplerData(), path));
152 charts.setCell(plot.getContentPane(), graphIndex % columns, graphIndex / columns);
153 }
154
155 getAnimationPanel().getTabbedPane().addTab(getAnimationPanel().getTabbedPane().getTabCount(), "statistics ", charts);
156 }
157
158
159 private static class NetworksParameterDialog extends TabbedParameterDialog
160 {
161
162 private static final long serialVersionUID = 1L;
163
164
165
166
167 NetworksParameterDialog(final InputParameterMap inputParameterMap)
168 {
169 super(inputParameterMap);
170 }
171
172
173 @SuppressWarnings({"rawtypes", "unchecked"})
174 @Override
175 public void addParameterField(final JPanel panel, final InputParameter<?, ?> parameter)
176 {
177 if (parameter instanceof InputParameterSelectionMap<?, ?>)
178 {
179 this.fields.add(new InputFieldSelectionMapRadio(panel, (InputParameterSelectionMap<?, ?>) parameter));
180 }
181 else
182 {
183 super.addParameterField(panel, parameter);
184 }
185 }
186
187
188 @Override
189 public void actionPerformed(final ActionEvent e)
190 {
191 boolean ok = true;
192 try
193 {
194 for (InputField field : this.fields)
195 {
196 if (field instanceof InputFieldSelectionMapRadio<?, ?>)
197 {
198 InputFieldSelectionMapRadio<?, ?> f = (InputFieldSelectionMapRadio<?, ?>) field;
199 f.getParameter().setObjectValue(f.getValue());
200 }
201 }
202 }
203 catch (Exception exception)
204 {
205 JOptionPane.showMessageDialog(null, exception.getMessage(), "Data Entry Error", JOptionPane.ERROR_MESSAGE);
206 ok = false;
207 }
208 if (ok)
209 {
210 super.actionPerformed(e);
211 }
212 }
213
214
215
216
217
218
219 public static boolean process(final InputParameterMap inputParameterMap)
220 {
221 NetworksParameterDialog dialog = new NetworksParameterDialog(inputParameterMap);
222 return !dialog.stopped;
223 }
224 }
225
226
227
228
229
230
231 public static class InputFieldSelectionMapRadio<K, T> extends AbstractInputField
232 {
233
234 private List<JRadioButton> buttons = new ArrayList<>();
235
236
237 private List<T> values = new ArrayList<>();
238
239
240
241
242
243
244 public InputFieldSelectionMapRadio(final JPanel panel, final InputParameterSelectionMap<K, T> parameter)
245 {
246 super(parameter);
247 Box box = Box.createVerticalBox();
248 box.add(new JLabel(" "));
249 box.add(new JLabel(parameter.getShortName()));
250 ButtonGroup group = new ButtonGroup();
251 for (K option : parameter.getOptions().keySet())
252 {
253 String item = option.toString();
254 T value = parameter.getOptions().get(option);
255 JRadioButton button = new JRadioButton(item);
256 button.setActionCommand(item);
257 if (value.equals(parameter.getDefaultValue()))
258 {
259 button.setSelected(true);
260 }
261 group.add(button);
262 box.add(button);
263 this.buttons.add(button);
264 this.values.add(value);
265 }
266 panel.add(box);
267 }
268
269
270 @SuppressWarnings("unchecked")
271 @Override
272 public InputParameterSelectionMap<K, T> getParameter()
273 {
274 return (InputParameterSelectionMap<K, T>) super.getParameter();
275 }
276
277
278 public T getValue()
279 {
280 for (JRadioButton button : this.buttons)
281 {
282 if (button.isSelected())
283 {
284 System.out.println("SELECTED: " + this.values.get(this.buttons.indexOf(button)));
285 return this.values.get(this.buttons.indexOf(button));
286 }
287 }
288 return this.values.get(0);
289 }
290 }
291 }