use of org.onebusaway.transit_data_federation.impl.transit_graph.TransitGraphImpl in project onebusaway-application-modules by camsys.
the class TransitGraphTask method run.
@Transactional
public void run() {
TransitGraphImpl graph = new TransitGraphImpl();
_agencyEntriesFactory.processAgencies(graph);
_stopEntriesFactory.processStops(graph);
_routeEntriesFactory.processRoutes(graph);
_routeCollectionEntriesFactory.processRouteCollections(graph);
_tripEntriesFactory.processTrips(graph);
_blockEntriesFactory.processBlocks(graph);
_frequencyEntriesFactory.processFrequencies(graph);
/**
* Make sure the graph is initialized as result of the graph building
* process, as it will be used by subsequent tasks
*/
graph.initialize();
try {
ObjectSerializationLibrary.writeObject(_bundle.getTransitGraphPath(), graph);
} catch (Exception ex) {
throw new IllegalStateException("error writing graph to file", ex);
}
_refreshService.refresh(RefreshableResources.ROUTE_COLLECTIONS_DATA);
_refreshService.refresh(RefreshableResources.TRANSIT_GRAPH);
}
use of org.onebusaway.transit_data_federation.impl.transit_graph.TransitGraphImpl in project onebusaway-application-modules by camsys.
the class FrequenciesEntriesFactoryTest method before.
@Before
public void before() {
_factory = new FrequencyEntriesFactory();
_dao = Mockito.mock(GtfsRelationalDao.class);
_factory.setGtfsDao(_dao);
_graph = new TransitGraphImpl();
_routeEntry = route("route");
_lsid = lsid("serviceId");
_stopA = stop("stopA");
_stopB = stop("stopB");
_stopC = stop("stopC");
}
use of org.onebusaway.transit_data_federation.impl.transit_graph.TransitGraphImpl in project onebusaway-application-modules by camsys.
the class RouteCollectionEntriesFactoryTest method testProcessRouteCollections.
@Test
public void testProcessRouteCollections() {
TransitGraphImpl graph = new TransitGraphImpl();
AgencyEntryImpl agency = new AgencyEntryImpl();
agency.setId("A");
graph.putAgencyEntry(agency);
graph.refreshAgencyMapping();
RouteEntryImpl routeA = new RouteEntryImpl();
routeA.setId(new AgencyAndId("A", "routeA"));
graph.putRouteEntry(routeA);
RouteEntryImpl routeB = new RouteEntryImpl();
routeB.setId(new AgencyAndId("A", "routeB"));
graph.putRouteEntry(routeB);
RouteCollectionEntriesFactory factory = new RouteCollectionEntriesFactory();
GtfsRelationalDao gtfsDao = Mockito.mock(GtfsRelationalDao.class);
factory.setGtfsDao(gtfsDao);
factory.setUniqueService(new UniqueServiceImpl());
factory.processRouteCollections(graph);
RouteCollectionEntry routeEntryA = graph.getRouteCollectionForId(routeA.getId());
assertEquals(routeA.getId(), routeEntryA.getId());
List<RouteEntry> routes = routeEntryA.getChildren();
assertEquals(1, routes.size());
assertTrue(routes.contains(routeA));
RouteCollectionEntry routeEntryB = graph.getRouteCollectionForId(routeB.getId());
assertEquals(routeB.getId(), routeEntryB.getId());
routes = routeEntryB.getChildren();
assertEquals(1, routes.size());
assertTrue(routes.contains(routeB));
List<RouteCollectionEntry> routeCollections = graph.getAllRouteCollections();
assertEquals(2, routeCollections.size());
assertTrue(routeCollections.contains(routeEntryA));
assertTrue(routeCollections.contains(routeEntryB));
routeCollections = agency.getRouteCollections();
assertEquals(2, routeCollections.size());
assertTrue(routeCollections.contains(routeEntryA));
assertTrue(routeCollections.contains(routeEntryB));
}
use of org.onebusaway.transit_data_federation.impl.transit_graph.TransitGraphImpl in project onebusaway-application-modules by camsys.
the class RouteCollectionEntriesFactoryTest method testGroupRoutesByShortName.
@Test
public void testGroupRoutesByShortName() {
TransitGraphImpl graph = new TransitGraphImpl();
AgencyEntryImpl agency = new AgencyEntryImpl();
agency.setId("A");
graph.putAgencyEntry(agency);
graph.refreshAgencyMapping();
RouteEntryImpl routeA = new RouteEntryImpl();
routeA.setId(new AgencyAndId("A", "routeA"));
graph.putRouteEntry(routeA);
RouteEntryImpl routeB = new RouteEntryImpl();
routeB.setId(new AgencyAndId("A", "routeB"));
graph.putRouteEntry(routeB);
GtfsRelationalDao gtfsDao = Mockito.mock(GtfsRelationalDao.class);
Route rA = new Route();
rA.setId(routeA.getId());
rA.setShortName("10");
Mockito.when(gtfsDao.getRouteForId(routeA.getId())).thenReturn(rA);
Route rB = new Route();
rB.setId(routeB.getId());
rB.setShortName("10");
Mockito.when(gtfsDao.getRouteForId(routeB.getId())).thenReturn(rB);
RouteCollectionEntriesFactory factory = new RouteCollectionEntriesFactory();
factory.setGroupRoutesByShortName(true);
factory.setGtfsDao(gtfsDao);
factory.setUniqueService(new UniqueServiceImpl());
factory.processRouteCollections(graph);
AgencyAndId id = new AgencyAndId("A", "10");
RouteCollectionEntry routeCollectionEntry = graph.getRouteCollectionForId(id);
assertEquals(id, routeCollectionEntry.getId());
List<RouteEntry> routes = routeCollectionEntry.getChildren();
assertEquals(2, routes.size());
assertTrue(routes.contains(routeA));
assertTrue(routes.contains(routeB));
List<RouteCollectionEntry> routeCollections = graph.getAllRouteCollections();
assertEquals(1, routeCollections.size());
assertTrue(routeCollections.contains(routeCollectionEntry));
routeCollections = agency.getRouteCollections();
assertEquals(1, routeCollections.size());
assertTrue(routeCollections.contains(routeCollectionEntry));
}
use of org.onebusaway.transit_data_federation.impl.transit_graph.TransitGraphImpl in project onebusaway-application-modules by camsys.
the class StopTimeEntriesFactoryTest method testThreeInARowDuplicateRemoval.
@Test
public void testThreeInARowDuplicateRemoval() {
TransitGraphImpl graph = new TransitGraphImpl();
Stop stopA = new Stop();
stopA.setId(aid("stopA"));
graph.putStopEntry(stop("stopA", 47.672207391799056, -122.387855896286));
Stop stopB = new Stop();
stopB.setId(aid("stopB"));
graph.putStopEntry(stop("stopB", 47.66852277218285, -122.3853882639923));
Stop stopC = new Stop();
stopC.setId(aid("stopC"));
graph.putStopEntry(stop("stopC", 47.66847942216854, -122.37545336180114));
Stop stopD = new Stop();
stopD.setId(aid("stopD"));
graph.putStopEntry(stop("stopD", 47.66947942216854, -122.37545336180114));
graph.refreshStopMapping();
Agency agency = new Agency();
agency.setId("1");
agency.setTimezone("America/Los_Angeles");
Route route = new Route();
route.setAgency(agency);
Trip trip = new Trip();
trip.setRoute(route);
trip.setServiceId(aid("serviceId"));
TripEntryImpl tripEntry = trip("trip");
StopTime stA = new StopTime();
stA.setId(100);
stA.setArrivalTime(time(9, 00));
stA.setDepartureTime(time(9, 05));
stA.setStopSequence(100);
stA.setStop(stopA);
stA.setTrip(trip);
StopTime stB = new StopTime();
stB.setId(101);
stB.setArrivalTime(time(9, 00));
stB.setDepartureTime(time(9, 05));
stB.setStopSequence(101);
stB.setStop(stopB);
stB.setTrip(trip);
StopTime stC = new StopTime();
stC.setId(102);
stC.setArrivalTime(time(9, 00));
stC.setDepartureTime(time(9, 05));
stC.setStopSequence(102);
stC.setStop(stopC);
stC.setTrip(trip);
StopTime stD = new StopTime();
stD.setId(103);
stD.setArrivalTime(time(11, 00));
stD.setDepartureTime(time(11, 05));
stD.setStopSequence(103);
stD.setStop(stopD);
stD.setTrip(trip);
StopTimeEntriesFactory factory = new StopTimeEntriesFactory();
factory.setDistanceAlongShapeLibrary(new DistanceAlongShapeLibrary());
List<StopTime> stopTimes = Arrays.asList(stA, stB, stC, stD);
ShapePointsFactory shapePointsFactory = new ShapePointsFactory();
shapePointsFactory.addPoint(47.673840100841396, -122.38756621771239);
shapePointsFactory.addPoint(47.668667271970484, -122.38756621771239);
shapePointsFactory.addPoint(47.66868172192725, -122.3661729186096);
shapePointsFactory.addPoint(47.66947942216854, -122.37545336180114);
ShapePoints shapePoints = shapePointsFactory.create();
List<StopTimeEntryImpl> entries = factory.processStopTimes(graph, stopTimes, tripEntry, shapePoints);
assertEquals(stopTimes.size(), entries.size());
StopTimeEntryImpl entry = entries.get(0);
assertEquals(0, entry.getAccumulatedSlackTime());
assertEquals(time(9, 00), entry.getArrivalTime());
assertEquals(time(9, 05), entry.getDepartureTime());
assertEquals(0, entry.getSequence());
assertEquals(181.5, entry.getShapeDistTraveled(), 0.1);
assertEquals(5 * 60, entry.getSlackTime());
entry = entries.get(1);
assertEquals(5 * 60, entry.getAccumulatedSlackTime());
assertEquals(time(9, 28, 45), entry.getArrivalTime());
assertEquals(time(9, 28, 45), entry.getDepartureTime());
assertEquals(1, entry.getSequence());
assertEquals(738.7, entry.getShapeDistTraveled(), 0.1);
assertEquals(0, entry.getSlackTime());
entry = entries.get(2);
assertEquals(5 * 60, entry.getAccumulatedSlackTime());
assertEquals(time(10, 00, 34), entry.getArrivalTime());
assertEquals(time(10, 00, 34), entry.getDepartureTime());
assertEquals(2, entry.getSequence());
assertEquals(1484.5, entry.getShapeDistTraveled(), 0.1);
assertEquals(0, entry.getSlackTime());
entry = entries.get(3);
assertEquals(5 * 60, entry.getAccumulatedSlackTime());
assertEquals(time(11, 00), entry.getArrivalTime());
assertEquals(time(11, 05), entry.getDepartureTime());
assertEquals(3, entry.getSequence());
assertEquals(2877.69, entry.getShapeDistTraveled(), 0.1);
assertEquals(60 * 5, entry.getSlackTime());
}
Aggregations