1 package org.opentrafficsim.road.gtu.lane.perception;
2
3 import java.util.Iterator;
4 import java.util.NoSuchElementException;
5 import java.util.function.Supplier;
6
7 import org.djunits.value.vdouble.scalar.Length;
8 import org.djutils.exceptions.Try;
9 import org.opentrafficsim.base.parameters.ParameterException;
10 import org.opentrafficsim.core.gtu.GTUException;
11 import org.opentrafficsim.road.gtu.lane.LaneBasedGTU;
12 import org.opentrafficsim.road.gtu.lane.perception.headway.Headway;
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30 public abstract class AbstractPerceptionReiterable<H extends Headway, U> implements PerceptionCollectable<H, U>
31 {
32
33
34 private SecondaryIteratorEntry first;
35
36
37 private SecondaryIteratorEntry last;
38
39
40 private Iterator<PrimaryIteratorEntry> primaryIterator;
41
42
43 private final LaneBasedGTU gtu;
44
45
46
47
48
49 protected AbstractPerceptionReiterable(final LaneBasedGTU perceivingGtu)
50 {
51 this.gtu = perceivingGtu;
52 }
53
54
55
56
57
58 protected LaneBasedGTU getGtu()
59 {
60 return this.gtu;
61 }
62
63
64
65
66
67 final Iterator<PrimaryIteratorEntry> getPrimaryIterator()
68 {
69 if (this.primaryIterator == null)
70 {
71 this.primaryIterator = primaryIterator();
72 }
73 return this.primaryIterator;
74 }
75
76
77
78
79
80 protected abstract Iterator<PrimaryIteratorEntry> primaryIterator();
81
82
83
84
85
86
87
88
89
90
91 protected abstract H perceive(LaneBasedGTU perceivingGtu, U object, Length distance)
92 throws GTUException, ParameterException;
93
94
95 @Override
96 public final synchronized H first()
97 {
98 assureFirst();
99 if (this.first == null)
100 {
101 return null;
102 }
103 return this.first.getValue();
104 }
105
106
107
108
109 private synchronized void assureFirst()
110 {
111 if (this.first == null && getPrimaryIterator().hasNext())
112 {
113 addNext(getPrimaryIterator().next());
114 }
115 }
116
117
118
119
120
121 @SuppressWarnings("synthetic-access")
122 final void addNext(final PrimaryIteratorEntry next)
123 {
124 SecondaryIteratorEntry entry = new SecondaryIteratorEntry(next.object, next.distance);
125 if (AbstractPerceptionReiterable.this.last == null)
126 {
127 AbstractPerceptionReiterable.this.first = entry;
128 AbstractPerceptionReiterable.this.last = entry;
129 }
130 else
131 {
132 AbstractPerceptionReiterable.this.last.next = entry;
133 AbstractPerceptionReiterable.this.last = entry;
134 }
135 }
136
137
138 @Override
139 public final boolean isEmpty()
140 {
141 return first() == null;
142 }
143
144
145 @Override
146 public final Iterator<H> iterator()
147 {
148 return new PerceptionIterator();
149 }
150
151
152 @SuppressWarnings("synthetic-access")
153 @Override
154 public final <C, I> C collect(final Supplier<I> identity, final PerceptionAccumulator<? super U, I> accumulator,
155 final PerceptionFinalizer<C, I> finalizer)
156 {
157 Intermediate<I> intermediate = new Intermediate<>(identity.get());
158 assureFirst();
159 if (this.first != null)
160 {
161 SecondaryIteratorEntry lastReturned = null;
162 SecondaryIteratorEntry next = this.first;
163 next = assureNext(next, lastReturned);
164 while (next != null && !intermediate.isStop())
165 {
166 intermediate = accumulator.accumulate(intermediate, next.object, next.distance);
167 intermediate.step();
168 lastReturned = next;
169 next = lastReturned.next;
170 next = assureNext(next, lastReturned);
171 }
172 }
173 return finalizer.collect(intermediate.getObject());
174 }
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190 public class PerceptionIterator implements Iterator<H>
191 {
192
193
194 private SecondaryIteratorEntry lastReturned;
195
196
197 private SecondaryIteratorEntry next;
198
199
200 @SuppressWarnings("synthetic-access")
201 PerceptionIterator()
202 {
203 this.next = AbstractPerceptionReiterable.this.first;
204 }
205
206
207 @Override
208 public boolean hasNext()
209 {
210 this.next = assureNext(this.next, this.lastReturned);
211 return this.next != null;
212 }
213
214
215 @SuppressWarnings("synthetic-access")
216 @Override
217 public H next()
218 {
219 this.next = assureNext(this.next, this.lastReturned);
220 if (this.next == null)
221 {
222 throw new NoSuchElementException();
223 }
224 this.lastReturned = this.next;
225 this.next = this.lastReturned.next;
226 return this.lastReturned.getValue();
227 }
228
229 }
230
231
232
233
234
235
236
237
238 @SuppressWarnings("synthetic-access")
239 synchronized SecondaryIteratorEntry assureNext(final SecondaryIteratorEntry next, final SecondaryIteratorEntry lastReturned)
240 {
241 if (next == null && getPrimaryIterator().hasNext())
242 {
243 addNext(getPrimaryIterator().next());
244 if (lastReturned == null)
245 {
246 return AbstractPerceptionReiterable.this.first;
247 }
248 else
249 {
250 return lastReturned.next;
251 }
252 }
253 return next;
254 }
255
256
257
258
259
260
261
262
263
264
265
266
267
268 protected class PrimaryIteratorEntry implements Comparable<PrimaryIteratorEntry>
269 {
270
271 private final U object;
272
273
274 private final Length distance;
275
276
277
278
279
280
281 public PrimaryIteratorEntry(final U object, final Length distance)
282 {
283 this.object = object;
284 this.distance = distance;
285 }
286
287
288 @Override
289 public int compareTo(final PrimaryIteratorEntry o)
290 {
291 return this.distance.compareTo(o.distance);
292 }
293
294
295
296
297
298 protected U getObject()
299 {
300 return this.object;
301 }
302 }
303
304
305
306
307
308
309
310
311
312
313
314
315
316 private class SecondaryIteratorEntry
317 {
318
319 private final U object;
320
321
322 private final Length distance;
323
324
325 private H value;
326
327
328 private SecondaryIteratorEntry next;
329
330
331
332
333
334
335 SecondaryIteratorEntry(final U object, final Length distance)
336 {
337 this.object = object;
338 this.distance = distance;
339 }
340
341
342
343
344
345 H getValue()
346 {
347 if (this.value == null)
348 {
349 this.value = Try.assign(() -> perceive(AbstractPerceptionReiterable.this.getGtu(), this.object, this.distance),
350 "Exception during perception of object.");
351 }
352 return this.value;
353 }
354 }
355
356 }