Issue #1. Added special node classes for arrival and departure.

Author: highsource

core/src/main/java/org/hisrc/gtfs/graph/builder/jgrapht/GtfsDirectedGraphBuilder.java

@@ -3,6 +3,8 @@
 import java.util.HashSet;
 import java.util.Set;
 
+import org.hisrc.gtfs.graph.model.TemporalStopArrivalNode;
+import org.hisrc.gtfs.graph.model.TemporalStopDepartureNode;
 import org.hisrc.gtfs.graph.model.TemporalStopNode;
 import org.hisrc.gtfs.graph.model.TransitionEdge;
 import org.hisrc.gtfs.graph.model.TransitionType;
@@ -100,10 +102,10 @@ public void addStopTime(StopTime stopTime) {
 		final int arrivalTime = stopTime.getArrivalTime();
 		final int departureTime = stopTime.getDepartureTime();
 
-		final TemporalStopNode arrivalNode = new TemporalStopNode(stop,
-				arrivalTime, false);
-		final TemporalStopNode departureNode = new TemporalStopNode(stop,
-				departureTime, true);
+		final TemporalStopNode arrivalNode = new TemporalStopArrivalNode(stop,
+				arrivalTime);
+		final TemporalStopNode departureNode = new TemporalStopDepartureNode(
+				stop, departureTime);
 		graph.addVertex(arrivalNode);
 		graph.addVertex(departureNode);
 

core/src/main/java/org/hisrc/gtfs/graph/model/TemporalStopArrivalNode.java

@@ -0,0 +1,11 @@
+package org.hisrc.gtfs.graph.model;
+
+import org.onebusaway.gtfs.model.Stop;
+
+public class TemporalStopArrivalNode extends TemporalStopNode {
+
+	public TemporalStopArrivalNode(Stop stop, int time) {
+		super(stop, time);
+	}
+
+}

core/src/main/java/org/hisrc/gtfs/graph/model/TemporalStopDepartureNode.java

@@ -0,0 +1,11 @@
+package org.hisrc.gtfs.graph.model;
+
+import org.onebusaway.gtfs.model.Stop;
+
+public class TemporalStopDepartureNode extends TemporalStopNode {
+
+	public TemporalStopDepartureNode(Stop stop, int time) {
+		super(stop, time);
+	}
+
+}

core/src/main/java/org/hisrc/gtfs/graph/model/TemporalStopNode.java

@@ -2,20 +2,17 @@
 
 import org.onebusaway.gtfs.model.Stop;
 
-public class TemporalStopNode {
+public abstract class TemporalStopNode {
 
 	// Reference to the stop
 	private final Stop stop;
 	// Time point
 	private final int time;
-	// Whether this is a starting point for the routing
-	private final boolean departure;
 
-	public TemporalStopNode(Stop stop, int time, boolean departure) {
+	public TemporalStopNode(Stop stop, int time) {
 		super();
 		this.stop = stop;
 		this.time = time;
-		this.departure = departure;
 	}
 
 	public Stop getStop() {
@@ -28,6 +25,6 @@ public int getTime() {
 
 	@Override
 	public String toString() {
-		return stop + "@" + time;
+		return "(Arrival" + stop + "@" + time + ")";
 	}
 }