Issue #1.

Author: highsource

core/src/main/java/org/hisrc/gtfs/graph/builder/GraphBuilder.java

@@ -0,0 +1,29 @@
+package org.hisrc.gtfs.graph.builder;
+
+import org.hisrc.gtfs.graph.model.edge.TransitionEdge;
+import org.hisrc.gtfs.graph.model.vertex.TemporalVertex;
+import org.onebusaway.gtfs.model.Stop;
+
+public interface GraphBuilder {
+
+	public TemporalVertex addParentStationVertex(Stop stop, int time);
+
+	public TemporalVertex addArrivalVertex(final Stop stop,
+			final int arrivalTime);
+
+	public TemporalVertex addDepartureVertex(final Stop stop,
+			final int departureTime);
+
+	public TransitionEdge addArrivalDepartureEdge(TemporalVertex arrivalVertex,
+			TemporalVertex departureVertex, int cost);
+
+	public TransitionEdge addDepartureArrivalEdge(TemporalVertex arrivalVertex,
+			TemporalVertex departureVertex, int cost);
+
+	public TransitionEdge addParentChildEdge(TemporalVertex parentVertex,
+			TemporalVertex childVertex);
+
+	public TransitionEdge addChildParentEdge(TemporalVertex childVertex,
+			TemporalVertex parentVertex);
+
+}

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

@@ -0,0 +1,87 @@
+package org.hisrc.gtfs.graph.builder.jgrapht;
+
+import org.hisrc.gtfs.graph.builder.GraphBuilder;
+import org.hisrc.gtfs.graph.model.edge.ArrivalDepartureEdge;
+import org.hisrc.gtfs.graph.model.edge.ChildParentEdge;
+import org.hisrc.gtfs.graph.model.edge.DepartureArrivalEdge;
+import org.hisrc.gtfs.graph.model.edge.ParentChildEdge;
+import org.hisrc.gtfs.graph.model.edge.TransitionEdge;
+import org.hisrc.gtfs.graph.model.vertex.ArrivalVertex;
+import org.hisrc.gtfs.graph.model.vertex.DepartureVertex;
+import org.hisrc.gtfs.graph.model.vertex.ParentStationVertex;
+import org.hisrc.gtfs.graph.model.vertex.TemporalVertex;
+import org.jgrapht.DirectedGraph;
+import org.jgrapht.EdgeFactory;
+import org.jgrapht.graph.DirectedMultigraph;
+import org.onebusaway.gtfs.model.Stop;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class JGraphTGraphBuilder implements GraphBuilder {
+
+	private DirectedGraph<TemporalVertex, TransitionEdge> graph = new DirectedMultigraph<TemporalVertex, TransitionEdge>(
+			new EdgeFactory<TemporalVertex, TransitionEdge>() {
+				@Override
+				public TransitionEdge createEdge(TemporalVertex start,
+						TemporalVertex stop) {
+					throw new UnsupportedOperationException();
+				}
+			});
+
+	private Logger logger = LoggerFactory.getLogger(JGraphTGraphBuilder.class);
+
+	public TemporalVertex addParentStationVertex(Stop stop, int time) {
+		final ParentStationVertex vertex = new ParentStationVertex(stop, time);
+		graph.addVertex(vertex);
+		return vertex;
+	}
+
+	public TemporalVertex addArrivalVertex(final Stop stop,
+			final int arrivalTime) {
+		final TemporalVertex arrivalNode = new ArrivalVertex(stop, arrivalTime);
+		graph.addVertex(arrivalNode);
+		return arrivalNode;
+	}
+
+	public TemporalVertex addDepartureVertex(final Stop stop,
+			final int departureTime) {
+		final TemporalVertex departureNode = new DepartureVertex(stop,
+				departureTime);
+		graph.addVertex(departureNode);
+		return departureNode;
+	}
+
+	public TransitionEdge addParentChildEdge(final TemporalVertex childVertex,
+			final TemporalVertex parentVertex) {
+		// logger.info("Adding [" + parentVertex + "-pc->" + childVertex + "]");
+		final ParentChildEdge edge = new ParentChildEdge();
+		graph.addEdge(parentVertex, childVertex, edge);
+
+		return edge;
+	}
+
+	public TransitionEdge addChildParentEdge(final TemporalVertex childVertex,
+			final TemporalVertex parentVertex) {
+		// logger.info("Adding [" + childVertex + "-cp->" + parentVertex + "]");
+		ChildParentEdge edge = new ChildParentEdge();
+		graph.addEdge(childVertex, parentVertex, edge);
+		return edge;
+	}
+
+	public TransitionEdge addArrivalDepartureEdge(
+			final TemporalVertex arrivalVertex,
+			final TemporalVertex departureVertex, int cost) {
+		// logger.info("Adding [" + arrivalNode + "-" + departureNode + "]");
+		final TransitionEdge edge = new ArrivalDepartureEdge(cost);
+		graph.addEdge(arrivalVertex, departureVertex, edge);
+		return edge;
+	}
+
+	public TransitionEdge addDepartureArrivalEdge(
+			final TemporalVertex departureVertex,
+			final TemporalVertex arrivalVertex, final int cost) {
+		final TransitionEdge edge = new DepartureArrivalEdge(cost);
+		graph.addEdge(departureVertex, arrivalVertex, edge);
+		return edge;
+	}
+}

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

@@ -0,0 +1,10 @@
+package org.hisrc.gtfs.graph.model.edge;
+
+
+public class ArrivalDepartureEdge extends TransitionEdge {
+
+	public ArrivalDepartureEdge(int cost) {
+		super(cost);
+	}
+
+}

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

@@ -0,0 +1,10 @@
+package org.hisrc.gtfs.graph.model.edge;
+
+
+public class ChildParentEdge extends TransitionEdge {
+
+	public ChildParentEdge() {
+		super(0);
+	}
+
+}

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

@@ -0,0 +1,10 @@
+package org.hisrc.gtfs.graph.model.edge;
+
+
+public class DepartureArrivalEdge extends TransitionEdge {
+
+	public DepartureArrivalEdge(int cost) {
+		super(cost);
+	}
+
+}

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

@@ -0,0 +1,10 @@
+package org.hisrc.gtfs.graph.model.edge;
+
+
+public class ParentChildEdge extends TransitionEdge {
+
+	public ParentChildEdge() {
+		super(0);
+	}
+
+}