1 package org.opentrafficsim.draw.graphs;
2
3 import java.awt.Color;
4 import java.awt.Font;
5 import java.awt.Graphics2D;
6 import java.awt.geom.AffineTransform;
7 import java.awt.geom.Rectangle2D;
8 import java.awt.image.BufferedImage;
9 import java.io.IOException;
10 import java.util.LinkedHashSet;
11 import java.util.Set;
12 import java.util.UUID;
13
14 import org.djunits.value.vdouble.scalar.Duration;
15 import org.djunits.value.vdouble.scalar.Time;
16 import org.djutils.event.EventType;
17 import org.jfree.chart.ChartUtils;
18 import org.jfree.chart.JFreeChart;
19 import org.jfree.chart.plot.XYPlot;
20 import org.jfree.chart.title.TextTitle;
21 import org.jfree.data.general.Dataset;
22 import org.jfree.data.general.DatasetChangeEvent;
23 import org.jfree.data.general.DatasetChangeListener;
24 import org.jfree.data.general.DatasetGroup;
25 import org.opentrafficsim.base.Identifiable;
26 import org.opentrafficsim.core.dsol.OTSSimulatorInterface;
27
28 import nl.tudelft.simulation.dsol.SimRuntimeException;
29 import nl.tudelft.simulation.dsol.formalisms.eventscheduling.SimEventInterface;
30 import nl.tudelft.simulation.dsol.simtime.SimTimeDoubleUnit;
31
32
33
34
35
36
37
38
39
40
41
42
43
44 public abstract class AbstractPlot implements Identifiable, Dataset
45 {
46
47
48
49
50
51 public static final EventType GRAPH_ADD_EVENT = new EventType("GRAPH.ADD");
52
53
54
55
56
57 public static final EventType GRAPH_REMOVE_EVENT = new EventType("GRAPH.REMOVE");
58
59
60 public static final Time DEFAULT_INITIAL_UPPER_TIME_BOUND = Time.instantiateSI(300.0);
61
62
63 private final OTSSimulatorInterface simulator;
64
65
66 private final String id = UUID.randomUUID().toString();
67
68
69 private final String caption;
70
71
72 private JFreeChart chart;
73
74
75 private Set<DatasetChangeListener> listeners = new LinkedHashSet<>();
76
77
78 private final Duration delay;
79
80
81 private Time updateTime;
82
83
84 private int updates = 0;
85
86
87 private Duration updateInterval;
88
89
90 private SimEventInterface<SimTimeDoubleUnit> updateEvent;
91
92
93
94
95
96
97
98
99 public AbstractPlot(final OTSSimulatorInterface simulator, final String caption, final Duration updateInterval,
100 final Duration delay)
101 {
102 this.simulator = simulator;
103 this.caption = caption;
104 this.updateInterval = updateInterval;
105 this.delay = delay;
106 this.updates = (int) (simulator.getSimulatorTime().si / updateInterval.si);
107 update();
108 }
109
110
111
112
113
114 @SuppressWarnings("methodlength")
115 protected void setChart(final JFreeChart chart)
116 {
117 this.chart = chart;
118
119
120 chart.setTitle(new TextTitle(chart.getTitle().getText(), new Font("SansSerif", java.awt.Font.BOLD, 16)));
121
122
123 chart.getPlot().setBackgroundPaint(Color.LIGHT_GRAY);
124 chart.setBackgroundPaint(Color.WHITE);
125 if (chart.getPlot() instanceof XYPlot)
126 {
127 chart.getXYPlot().setDomainGridlinePaint(Color.WHITE);
128 chart.getXYPlot().setRangeGridlinePaint(Color.WHITE);
129 }
130
131 }
132
133
134
135
136
137
138
139
140
141 public byte[] encodeAsPng(final int width, final int height, final double fontSize) throws IOException
142 {
143
144
145 double baseWidth = width / (fontSize / 16);
146 double baseHeight = height / (fontSize / 16);
147
148 BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
149 Graphics2D g2 = image.createGraphics();
150
151 AffineTransform saved = g2.getTransform();
152 g2.transform(AffineTransform.getScaleInstance(width / baseWidth, height / baseHeight));
153 getChart().draw(g2, new Rectangle2D.Double(0, 0, baseWidth, baseHeight), null, null);
154 g2.setTransform(saved);
155 g2.dispose();
156 return ChartUtils.encodeAsPNG(image);
157 }
158
159
160 @Override
161 public final DatasetGroup getGroup()
162 {
163 return null;
164 }
165
166
167 @Override
168 public final void setGroup(final DatasetGroup group)
169 {
170
171 }
172
173
174
175
176
177
178
179
180 public void setAutoBoundDomain(final XYPlot plot)
181 {
182
183 }
184
185
186
187
188
189
190
191
192 public void setAutoBoundRange(final XYPlot plot)
193 {
194
195 }
196
197
198
199
200
201 public abstract GraphType getGraphType();
202
203
204
205
206
207
208
209 public abstract String getStatusLabel(double domainValue, double rangeValue);
210
211
212
213
214
215 protected abstract void increaseTime(Time time);
216
217
218
219
220 public final void notifyPlotChange()
221 {
222 DatasetChangeEvent event = new DatasetChangeEvent(this, this);
223 for (DatasetChangeListener dcl : this.listeners)
224 {
225 dcl.datasetChanged(event);
226 }
227 }
228
229
230
231
232
233 public final JFreeChart getChart()
234 {
235 return this.chart;
236 }
237
238
239 @Override
240 public final String getId()
241 {
242 return this.id;
243 }
244
245
246 @Override
247 public final void addChangeListener(final DatasetChangeListener listener)
248 {
249 this.listeners.add(listener);
250 }
251
252
253 @Override
254 public final void removeChangeListener(final DatasetChangeListener listener)
255 {
256 this.listeners.remove(listener);
257 }
258
259
260
261
262
263 public OTSSimulatorInterface getSimulator()
264 {
265 return this.simulator;
266 }
267
268
269
270
271
272 public final void setUpdateInterval(final Duration interval)
273 {
274 if (this.updateEvent != null)
275 {
276 this.simulator.cancelEvent(this.updateEvent);
277 }
278 this.updates = (int) (this.simulator.getSimulatorTime().si / interval.si);
279 this.updateInterval = interval;
280 this.updateTime = Time.instantiateSI(this.updates * this.updateInterval.si);
281 scheduleNextUpdateEvent();
282 }
283
284
285
286
287
288 public final Time getUpdateTime()
289 {
290 return this.updateTime;
291 }
292
293
294
295
296 protected void update()
297 {
298
299 this.updateTime = this.simulator.getSimulatorTime();
300 increaseTime(this.updateTime.minus(this.delay));
301 notifyPlotChange();
302 scheduleNextUpdateEvent();
303 }
304
305
306
307
308 private void scheduleNextUpdateEvent()
309 {
310 try
311 {
312 this.updates++;
313
314 this.updateEvent = this.simulator.scheduleEventAbs(Time.instantiateSI(this.updateInterval.si * this.updates
315 + this.delay.si), this, this, "update", null);
316 }
317 catch (SimRuntimeException exception)
318 {
319 throw new RuntimeException("Unexpected exception while updating plot.", exception);
320 }
321 }
322
323
324
325
326
327 public String getCaption()
328 {
329 return this.caption;
330 }
331
332 }