1 package org.opentrafficsim.demo.geometry.shape;
2
3 import java.awt.Color;
4 import java.io.File;
5 import java.io.IOException;
6 import java.net.URL;
7 import java.rmi.RemoteException;
8 import java.util.Collection;
9 import java.util.HashMap;
10 import java.util.Map;
11
12 import javax.naming.NamingException;
13
14 import org.djunits.unit.FrequencyUnit;
15 import org.djunits.unit.UNITS;
16 import org.djunits.value.vdouble.scalar.DoubleScalar;
17 import org.djunits.value.vdouble.scalar.Frequency;
18 import org.djunits.value.vdouble.scalar.Length;
19 import org.djunits.value.vdouble.scalar.Speed;
20 import org.geotools.data.FileDataStoreFinder;
21 import org.geotools.data.shapefile.ShapefileDataStore;
22 import org.geotools.data.simple.SimpleFeatureCollection;
23 import org.geotools.data.simple.SimpleFeatureIterator;
24 import org.geotools.data.simple.SimpleFeatureSource;
25 import org.geotools.geometry.jts.JTSFactoryFinder;
26 import org.opengis.feature.Property;
27 import org.opengis.feature.simple.SimpleFeature;
28 import org.opentrafficsim.core.dsol.OTSSimulatorInterface;
29 import org.opentrafficsim.core.geometry.OTSGeometryException;
30 import org.opentrafficsim.core.geometry.OTSLine3D;
31 import org.opentrafficsim.core.geometry.OTSPoint3D;
32 import org.opentrafficsim.core.network.Link;
33 import org.opentrafficsim.core.network.LinkType;
34 import org.opentrafficsim.core.network.LongitudinalDirectionality;
35 import org.opentrafficsim.core.network.Network;
36 import org.opentrafficsim.core.network.NetworkException;
37 import org.opentrafficsim.core.network.OTSNode;
38 import org.opentrafficsim.core.network.animation.LinkAnimation;
39 import org.opentrafficsim.road.network.animation.LaneAnimation;
40 import org.opentrafficsim.road.network.animation.ShoulderAnimation;
41 import org.opentrafficsim.road.network.lane.CrossSectionLink;
42 import org.opentrafficsim.road.network.lane.Lane;
43 import org.opentrafficsim.road.network.lane.NoTrafficLane;
44 import org.opentrafficsim.road.network.lane.Shoulder;
45 import org.opentrafficsim.road.network.lane.changing.LaneKeepingPolicy;
46 import org.opentrafficsim.road.network.lane.changing.OvertakingConditions;
47
48 import com.vividsolutions.jts.geom.Coordinate;
49 import com.vividsolutions.jts.geom.Geometry;
50 import com.vividsolutions.jts.geom.GeometryFactory;
51 import com.vividsolutions.jts.geom.LineString;
52 import com.vividsolutions.jts.geom.Point;
53
54
55
56
57
58
59
60
61
62
63
64 public final class ShapeFileReader implements UNITS
65 {
66
67 private ShapeFileReader()
68 {
69
70 }
71
72
73
74
75
76
77
78
79
80
81 public static Map<String, OTSNode> readNodes(final Network network, final String shapeFileName, final String numberType,
82 final boolean returnCentroid, final boolean allCentroids) throws IOException
83 {
84
85
86
87
88
89
90
91
92
93 URL url;
94 if (new File(shapeFileName).canRead())
95 {
96 url = new File(shapeFileName).toURI().toURL();
97 }
98 else
99 {
100 url = ShapeFileReader.class.getResource(shapeFileName);
101 }
102 ShapefileDataStore storeNodes = (ShapefileDataStore) FileDataStoreFinder.getDataStore(url);
103
104 Map<String, OTSNode> nodes = new HashMap<>();
105
106 SimpleFeatureSource featureSourceNodes = storeNodes.getFeatureSource();
107 SimpleFeatureCollection featureCollectionNodes = featureSourceNodes.getFeatures();
108 SimpleFeatureIterator iterator = featureCollectionNodes.features();
109 try
110 {
111 while (iterator.hasNext())
112 {
113 SimpleFeature feature = iterator.next();
114 Coordinate coordinate = ((Point) feature.getAttribute("the_geom")).getCoordinate();
115 String nr = removeQuotes(String.valueOf(feature.getAttribute(numberType)));
116 boolean addThisNode = false;
117 if (returnCentroid)
118 {
119 if (nr.substring(0, 1).equals("C") || allCentroids)
120 {
121 addThisNode = true;
122 }
123 }
124 else
125 {
126 if (nr == null)
127 {
128 System.out.println("null found");
129 }
130 if (!nr.substring(0, 1).equals("C"))
131 {
132 addThisNode = true;
133 }
134 }
135 if (addThisNode)
136 {
137 OTSNode node = new OTSNode(network, nr, new OTSPoint3D(coordinate));
138 nodes.put(nr, node);
139 }
140 }
141 }
142 catch (Exception problem)
143 {
144 problem.printStackTrace();
145 }
146 finally
147 {
148 iterator.close();
149 storeNodes.dispose();
150 }
151 System.out.println("aantal knopen (353): geteld " + nodes.size());
152 return nodes;
153 }
154
155
156
157
158
159 public static boolean inspectNodeCentroid(final String number)
160 {
161 boolean isCentroid = false;
162 String[] names = removeQuotes(number).split(":");
163 String name = names[0];
164 if (name.charAt(0) == 'C')
165 {
166 isCentroid = true;
167 }
168 return isCentroid;
169 }
170
171
172
173
174
175
176
177
178
179 public static void readLinks(final Network network, final String shapeFileName, final Map<String, Link> links,
180 final Map<String, OTSNode> nodes, final OTSSimulatorInterface simulator) throws IOException
181 {
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199 URL url;
200 if (new File(shapeFileName).canRead())
201 {
202 url = new File(shapeFileName).toURI().toURL();
203 }
204 else
205 {
206 url = ShapeFileReader.class.getResource(shapeFileName);
207 }
208
209 ShapefileDataStore storeLinks = (ShapefileDataStore) FileDataStoreFinder.getDataStore(url);
210 SimpleFeatureSource featureSourceLinks = storeLinks.getFeatureSource();
211 SimpleFeatureCollection featureCollectionLinks = featureSourceLinks.getFeatures();
212 SimpleFeatureIterator iterator = featureCollectionLinks.features();
213
214 try
215 {
216 while (iterator.hasNext())
217 {
218 SimpleFeature feature = iterator.next();
219 GeometryFactory geometryFactory = JTSFactoryFinder.getGeometryFactory();
220 Geometry geometry = (Geometry) feature.getAttribute("the_geom");
221 Coordinate[] coords = geometry.getCoordinates();
222 LineString line = geometryFactory.createLineString(coords);
223 String nr = String.valueOf(feature.getAttribute("LINKNR"));
224 String nrBA = nr + "_BA";
225 String name = String.valueOf(feature.getAttribute("NAME"));
226
227
228 double lengthIn = Double.parseDouble(String.valueOf(feature.getAttribute("LENGTH")));
229 Length length = new Length(lengthIn, KILOMETER);
230 short direction = (short) Long.parseLong(String.valueOf(feature.getAttribute("DIRECTION")));
231 String lNodeA = String.valueOf(feature.getAttribute("ANODE"));
232 String lNodeB = String.valueOf(feature.getAttribute("BNODE"));
233
234 String linkTag = (String) feature.getAttribute("LINKTAG");
235 String wegtype = (String) feature.getAttribute("WEGTYPEAB");
236 String typeWegVak = (String) feature.getAttribute("TYPEWEGVAB");
237 String typeWeg = (String) feature.getAttribute("TYPEWEG_AB");
238 Double speedIn = Double.parseDouble(String.valueOf(feature.getAttribute("SPEEDAB")));
239 Speed speed = new Speed(speedIn, KM_PER_HOUR);
240 double capacityIn = Double.parseDouble(String.valueOf(feature.getAttribute("CAPACITYAB")));
241 Frequency capacity = new Frequency(capacityIn, PER_HOUR);
242
243
244 OTSNode nodeA = nodes.get(lNodeA);
245 OTSNode nodeB = nodes.get(lNodeB);
246
247 if (nodeA != null && nodeB != null)
248 {
249 CrossSectionLink linkAB = null;
250 CrossSectionLink linkBA = null;
251 linkAB = new CrossSectionLink(network, nr, nodeA, nodeB, LinkType.ALL,
252 new OTSLine3D(new OTSPoint3D[] { nodeA.getPoint(), nodeB.getPoint() }), simulator,
253 LongitudinalDirectionality.DIR_BOTH, LaneKeepingPolicy.KEEP_RIGHT);
254 animate(linkAB, typeWegVak, simulator);
255 linkBA = new CrossSectionLink(network, nrBA, nodeB, nodeA, LinkType.ALL,
256 new OTSLine3D(new OTSPoint3D[] { nodeB.getPoint(), nodeA.getPoint() }), simulator,
257 LongitudinalDirectionality.DIR_BOTH, LaneKeepingPolicy.KEEP_RIGHT);
258 animate(linkBA, typeWegVak, simulator);
259 if (direction == 1)
260 {
261 links.put(nr, linkAB);
262 }
263 else if (direction == 2)
264 {
265 links.put(nrBA, linkBA);
266 }
267 else if (direction == 3)
268 {
269 links.put(nr, linkAB);
270 links.put(nrBA, linkBA);
271 }
272
273 }
274 else
275 {
276 System.out.println("Node lNodeA=" + lNodeA + " or lNodeB=" + lNodeB + " not found for linknr=" + nr
277 + ", name=" + name);
278 }
279 }
280
281 }
282 catch (Exception problem)
283 {
284 problem.printStackTrace();
285 }
286 finally
287 {
288 iterator.close();
289 storeLinks.dispose();
290 }
291
292 }
293
294
295
296
297
298 public static void shapeFileInfo(final String shapeFileName) throws IOException
299 {
300 URL url;
301 if (new File(shapeFileName).canRead())
302 {
303 url = new File(shapeFileName).toURI().toURL();
304 }
305 else
306 {
307 url = ShapeFileReader.class.getResource(shapeFileName);
308 }
309 ShapefileDataStore store = (ShapefileDataStore) FileDataStoreFinder.getDataStore(url);
310
311 SimpleFeatureSource featureSource = store.getFeatureSource();
312 SimpleFeatureCollection featureCollection = featureSource.getFeatures();
313 SimpleFeatureIterator iterator = featureCollection.features();
314 try
315 {
316 while (iterator.hasNext())
317 {
318 SimpleFeature feature = iterator.next();
319 Collection<Property> areaProperties = feature.getProperties();
320 for (Property p : areaProperties)
321 {
322 System.out.println(p.getName() + " " + p.getValue().getClass() + " " + p.getValue().toString());
323 }
324 return;
325 }
326 }
327 catch (Exception problem)
328 {
329 problem.printStackTrace();
330 }
331 finally
332 {
333 iterator.close();
334 store.dispose();
335 }
336 }
337
338
339
340
341
342 public static String removeQuotes(final String name)
343 {
344 String newName = name;
345 if (newName.length() >= 2 && newName.charAt(0) == '"' && newName.charAt(newName.length() - 1) == '"')
346 {
347 newName = newName.substring(1, newName.length() - 1);
348 }
349 return newName;
350 }
351
352
353
354
355
356
357
358
359
360 private static void animate(final CrossSectionLink link, final String wegType, final OTSSimulatorInterface simulator)
361 throws NamingException, NetworkException, RemoteException
362 {
363
364 new LinkAnimation(link, simulator, 0.1f);
365 if (wegType.startsWith("asw") || wegType.startsWith("80"))
366 {
367 int spits = 0;
368 int n = 1;
369 if (wegType.contains("2x2"))
370 {
371 n = 2;
372 }
373 if (wegType.contains("2x3"))
374 {
375 n = 3;
376 }
377 if (wegType.contains("2x4"))
378 {
379 n = 4;
380 }
381 if (wegType.contains("2x5"))
382 {
383 n = 5;
384 }
385 if (wegType.contains("+ 1") || wegType.contains("+1"))
386 {
387 spits = 1;
388 }
389 if (wegType.contains("+ 2") || wegType.contains("+2"))
390 {
391 spits = 2;
392 }
393 addNLanes(n, spits, link, simulator);
394 }
395 if (wegType.startsWith("stads"))
396 {
397 int n = 1;
398 if (wegType.contains("2x2"))
399 {
400 n = 2;
401 }
402 if (wegType.contains("2x3"))
403 {
404 n = 3;
405 }
406 boolean middenberm = wegType.contains("met middenberm");
407 addCityStreetLanes(n, middenberm, link, simulator);
408 }
409 else
410 {
411 addCityStreet(link, simulator);
412 }
413 }
414
415
416
417
418
419
420
421
422 private static void addNLanes(final int n, final int spits, final CrossSectionLink link,
423 final OTSSimulatorInterface simulator) throws NetworkException
424 {
425
426
427 Length m05 = new Length(0.5, METER);
428 Length m10 = new Length(1.0, METER);
429 Length m35 = new Length(3.5, METER);
430 Speed speedLimit = new Speed(100, KM_PER_HOUR);
431
432 try
433 {
434
435 Shoulder sM = new Shoulder(link, "sM", new Length(0.0, METER), m10);
436 new ShoulderAnimation(sM, simulator, Color.GREEN);
437 for (int i = -1; i <= 1; i += 2)
438 {
439 LongitudinalDirectionality dir =
440 (i < 0) ? LongitudinalDirectionality.DIR_PLUS : LongitudinalDirectionality.DIR_MINUS;
441 String lr = i < 0 ? "L" : "R";
442
443 Lane laneEM = new NoTrafficLane(link, lr + "." + "EM", new Length(i * 0.75, METER), new Length(i * 0.75, METER),
444 m05, m05);
445 new LaneAnimation(laneEM, simulator, Color.LIGHT_GRAY, false);
446 double lat = 1;
447 for (int j = 0; j < n; j++)
448 {
449 lat += i * 1.75;
450 Lane lane = new Lane(link, "lane." + lr + "." + j, new Length(lat, METER), new Length(lat, METER), m35, m35,
451 null, dir, speedLimit, new OvertakingConditions.LeftAndRight());
452 new LaneAnimation(lane, simulator, Color.GRAY, false);
453 lat += i * 1.75;
454 }
455
456 for (int j = 0; j < spits; j++)
457 {
458 lat += i * 1.75;
459 Lane lane = new NoTrafficLane(link, "extra." + lr + "." + j, new Length(lat, METER), new Length(lat, METER),
460 m35, m35);
461 new LaneAnimation(lane, simulator, Color.LIGHT_GRAY, false);
462 lat += i * 1.75;
463 }
464 Lane laneEO = new NoTrafficLane(link, lr + "." + "EO", new Length(lat + i * 0.25, METER),
465 new Length(lat + i * 0.25, METER), m05, m05);
466 new LaneAnimation(laneEO, simulator, Color.LIGHT_GRAY, false);
467 lat += i * 0.5;
468 Shoulder sO = new Shoulder(link, lr + "." + "sO", new Length(lat, METER), m10);
469 new ShoulderAnimation(sO, simulator, Color.GREEN);
470 }
471 }
472 catch (NamingException | RemoteException | OTSGeometryException ne)
473 {
474
475 }
476 }
477
478
479
480
481
482
483
484
485 private static void addCityStreetLanes(final int n, final boolean middenberm, final CrossSectionLink link,
486 final OTSSimulatorInterface simulator) throws NetworkException
487 {
488
489
490 Length m10 = new Length(1.0, METER);
491 Length m30 = new Length(3.0, METER);
492 Speed speedLimit = new Speed(100, KM_PER_HOUR);
493
494 try
495 {
496 if (middenberm)
497 {
498 Shoulder sM = new Shoulder(link, "sM", new Length(0.0, METER), m10);
499 new ShoulderAnimation(sM, simulator, Color.GREEN);
500 }
501 for (int i = -1; i <= 1; i += 2)
502 {
503 LongitudinalDirectionality dir =
504 (i < 0) ? LongitudinalDirectionality.DIR_PLUS : LongitudinalDirectionality.DIR_MINUS;
505 double lat = middenberm ? 0.5 : 0.0;
506 for (int j = 0; j < n; j++)
507 {
508 lat += i * 1.5;
509 String lr = i < 0 ? "L" : "R";
510 Lane lane = new Lane(link, "lane." + lr + "." + j, new Length(lat, METER), new Length(lat, METER), m30, m30,
511 null, dir, speedLimit, new OvertakingConditions.LeftAndRight());
512 new LaneAnimation(lane, simulator, Color.DARK_GRAY, false);
513 lat += i * 1.5;
514 }
515 }
516 }
517 catch (NamingException | RemoteException | ArrayIndexOutOfBoundsException | OTSGeometryException ne)
518 {
519 ne.printStackTrace();
520 }
521 }
522
523
524
525
526
527
528 private static void addCityStreet(final CrossSectionLink link, final OTSSimulatorInterface simulator)
529 throws NetworkException
530 {
531 Length m60 = new Length(6.0, METER);
532 Speed speedLimit = new Speed(100, KM_PER_HOUR);
533
534 try
535 {
536 Lane lane = new Lane(link, "lane", new Length(0.0, METER), new Length(0.0, METER), m60, m60, null,
537 LongitudinalDirectionality.DIR_PLUS, speedLimit, new OvertakingConditions.LeftAndRight());
538 new LaneAnimation(lane, simulator, Color.DARK_GRAY, false);
539 }
540 catch (NamingException | RemoteException | OTSGeometryException ne)
541 {
542
543 }
544 }
545
546 }