Java AgencyAndId类代码示例

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testSyntheticGetTripAgencyIdsReferencingServiceId() {

  GtfsRelationalDaoImpl dao = new GtfsRelationalDaoImpl();

  AgencyAndId serviceId = new AgencyAndId("C", "serviceId");

  Trip tripA = new Trip();
  tripA.setId(new AgencyAndId("A", "tripId"));
  tripA.setServiceId(serviceId);
  dao.saveEntity(tripA);

  Trip tripB = new Trip();
  tripB.setId(new AgencyAndId("B", "tripId"));
  tripB.setServiceId(serviceId);
  dao.saveEntity(tripB);

  List<String> agencyIds = dao.getTripAgencyIdsReferencingServiceId(serviceId);
  assertEquals(2, agencyIds.size());
  assertTrue(agencyIds.contains("A"));
  assertTrue(agencyIds.contains("B"));
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testFrequency() throws CsvEntityIOException, IOException {

  _reader.setDefaultAgencyId("1");

  Trip trip = new Trip();
  trip.setId(new AgencyAndId("1", "trip"));
  _reader.injectEntity(trip);

  StringBuilder b = new StringBuilder();
  b.append("trip_id,start_time,end_time,headway_secs,exact_times\n");
  b.append("trip,08:30:00,09:45:00,300,1\n");

  _reader.readEntities(Frequency.class, new StringReader(b.toString()));

  Frequency frequency = _dao.getFrequencyForId(1);
  assertEquals(30600, frequency.getStartTime());
  assertEquals(35100, frequency.getEndTime());
  assertEquals(1, frequency.getExactTimes());
  assertEquals(300, frequency.getHeadwaySecs());
  assertSame(trip, frequency.getTrip());
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Override
public void translateFromCSVToObject(CsvEntityContext context,
    Map<String, Object> csvValues, BeanWrapper object) {

  String agencyId = (String) csvValues.get(_csvFieldName + "_agencyId");
  String id = (String) csvValues.get(_csvFieldName + "_id");
  boolean missing = (agencyId == null || agencyId.isEmpty())
      || (id == null || id.isEmpty());
  if (missing) {
    if (_required)
      throw new MissingRequiredFieldException(_entityType, _csvFieldName);
    return;
  }
  AgencyAndId agencyAndId = new AgencyAndId(agencyId, id);
  object.setPropertyValue(_objFieldName, agencyAndId);
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Override
public List<Stop> getStopsForStation(Stop station) {
  if (_stopsByStation == null) {
    _stopsByStation = new HashMap<Stop, List<Stop>>();
    for (Stop stop : getAllStops()) {
      if (stop.getLocationType() == 0 && stop.getParentStation() != null) {
        Stop parentStation = getStopForId(new AgencyAndId(
            stop.getId().getAgencyId(), stop.getParentStation()));
        List<Stop> subStops = _stopsByStation.get(parentStation);
        if (subStops == null) {
          subStops = new ArrayList<Stop>(2);
          _stopsByStation.put(parentStation, subStops);
        }
        subStops.add(stop);
      }
    }
  }
  return list(_stopsByStation.get(station));
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Override
public List<Trip> getTripsForBlockId(AgencyAndId blockId) {

  if (_tripsByBlockId == null) {
    _tripsByBlockId = new HashMap<AgencyAndId, List<Trip>>();
    for (Trip trip : getAllTrips()) {
      if (trip.getBlockId() != null) {
        AgencyAndId bid = new AgencyAndId(trip.getId().getAgencyId(),
            trip.getBlockId());
        List<Trip> trips = _tripsByBlockId.get(bid);
        if (trips == null) {
          trips = new ArrayList<Trip>();
          _tripsByBlockId.put(bid, trips);
        }
        trips.add(trip);
      }
    }
  }

  return list(_tripsByBlockId.get(blockId));
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testBart() throws IOException, ParseException {

  File resourcePath = GtfsTestData.getBartGtfs();
  String agencyId = "BART";
  GtfsDao entityStore = processFeed(resourcePath, agencyId, false);

  Collection<Frequency> frequencies = entityStore.getAllFrequencies();
  assertEquals(6, frequencies.size());

  List<Frequency> frequenciesForTrip = GtfsTestData.grep(frequencies,
      "trip.id", new AgencyAndId("AirBART", "M-FSAT1DN"));
  assertEquals(1, frequenciesForTrip.size());
  Frequency frequencyForTrip = frequenciesForTrip.get(0);
  assertEquals(18000, frequencyForTrip.getStartTime());
  assertEquals(21600, frequencyForTrip.getEndTime());
  assertEquals(1200, frequencyForTrip.getHeadwaySecs());

  Collection<Transfer> transfers = entityStore.getAllTransfers();
  assertEquals(4, transfers.size());
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testUtf8() throws IOException, ParseException,
    InterruptedException {

  MockGtfs mockGtfs = MockGtfs.create();
  mockGtfs.putDefaultStopTimes();
  mockGtfs.putFile("routes.txt", new File(
      "src/test/resources/org/onebusaway/gtfs/utf8-routes.txt"));

  String agencyId = "1";
  GtfsDao dao = processFeed(mockGtfs.getPath(), agencyId, false);

  Route route = dao.getRouteForId(new AgencyAndId(agencyId, "R10"));
  assertEquals("Enguera-Alcúdia de Crespins-Xàtiva", route.getLongName());

  route = dao.getRouteForId(new AgencyAndId(agencyId, "R11"));
  assertEquals("Tuéjar-Casinos", route.getLongName());
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testFrequency() throws CsvEntityIOException, IOException {

  GtfsReader reader = new GtfsReader();
  reader.setDefaultAgencyId("1");

  Trip trip = new Trip();
  trip.setId(new AgencyAndId("1", "trip"));
  reader.injectEntity(trip);

  StringBuilder b = new StringBuilder();
  b.append("trip_id,start_time,end_time,headway_secs,exact_times\n");
  b.append("trip,08:30:00,09:45:00,300,1\n");

  reader.readEntities(Frequency.class, new StringReader(b.toString()));

  Frequency frequency = reader.getEntityStore().getEntityForId(
      Frequency.class, 1);
  assertEquals(30600, frequency.getStartTime());
  assertEquals(35100, frequency.getEndTime());
  assertEquals(1, frequency.getExactTimes());
  assertEquals(300, frequency.getHeadwaySecs());
  assertSame(trip, frequency.getTrip());
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testCsvParser() throws CsvEntityIOException, IOException {
  GtfsReader reader = new GtfsReader();
  reader.setDefaultAgencyId("1");

  Agency agency = new Agency();
  agency.setId("1");
  reader.setAgencies(Arrays.asList(agency));

  StringBuilder b = new StringBuilder();
  b.append("agency_id,route_id,route_short_name,route_long_name,route_type\n");
  b.append("        1,    R-10,              10,   \"Ten, Ten\",3\n");
  reader.readEntities(Route.class, new StringReader(b.toString()));
  Route route = reader.getEntityStore().getEntityForId(Route.class,
      new AgencyAndId("1", "R-10"));
  assertEquals("Ten, Ten", route.getLongName());
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testExtensionRead() throws IOException {
  MockGtfs gtfs = MockGtfs.create();
  gtfs.putMinimal();
  gtfs.putStops(2, "label=a,b");

  DefaultEntitySchemaFactory factory = GtfsEntitySchemaFactory.createEntitySchemaFactory();
  factory.addExtension(Stop.class, StopExtension.class);
  
  GtfsReader reader = new GtfsReader();
  reader.setEntitySchemaFactory(factory);

  GtfsMutableRelationalDao dao = gtfs.read(reader);
  Stop stop = dao.getStopForId(new AgencyAndId("a0", "s0"));
  StopExtension extension = stop.getExtension(StopExtension.class);
  assertEquals("a", extension.getLabel());
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testPutCalendarDates() throws IOException {
  _gtfs.putMinimal();
  _gtfs.putCalendarDates("sid0=20120901,-20120902", "sid1=20120903");
  GtfsRelationalDao dao = _gtfs.read();
  List<ServiceCalendarDate> dates = dao.getCalendarDatesForServiceId(new AgencyAndId(
      "a0", "sid0"));
  assertEquals(2, dates.size());
  assertEquals(new ServiceDate(2012, 9, 1), dates.get(0).getDate());
  assertEquals(1, dates.get(0).getExceptionType());
  assertEquals(new ServiceDate(2012, 9, 2), dates.get(1).getDate());
  assertEquals(2, dates.get(1).getExceptionType());
  dates = dao.getCalendarDatesForServiceId(new AgencyAndId("a0", "sid1"));
  assertEquals(1, dates.size());
  assertEquals(new ServiceDate(2012, 9, 3), dates.get(0).getDate());
  assertEquals(1, dates.get(0).getExceptionType());
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testDaylightSavingTime() {

  CalendarServiceDataFactoryImpl factory = new CalendarServiceDataFactoryImpl();

  Agency agencyA = agency("A", "America/Los_Angeles");
  AgencyAndId serviceId = new AgencyAndId("A", "2");

  ServiceDate dStart = new ServiceDate(2012, 3, 1);
  ServiceDate dEnd = new ServiceDate(2012, 3, 31);

  ServiceCalendar c = calendar(serviceId, dStart, dEnd, "1111111");

  GtfsRelationalDaoImpl dao = new GtfsRelationalDaoImpl();
  factory.setGtfsDao(dao);

  saveEntities(dao, agencyA);
  saveEntities(dao, c);

  CalendarServiceData data = factory.createData();
  List<ServiceDate> serviceDates = data.getServiceDatesForServiceId(serviceId);
  assertTrue(serviceDates.contains(new ServiceDate(2012, 3, 11)));
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
/****
 * {@link GtfsRelationalDao} Interface
 ****/

@Override
public List<String> getTripAgencyIdsReferencingServiceId(AgencyAndId serviceId) {
  return _ops.findByNamedQueryAndNamedParam("agencyIdsReferencingServiceId",
      "serviceId", serviceId);
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Override
public List<AgencyAndId> getAllServiceIds() {
  List<AgencyAndId> calendarIds = _ops.findByNamedQuery("calendarServiceIds");
  List<AgencyAndId> calendarDateIds = _ops.findByNamedQuery("calendarDateServiceIds");
  Set<AgencyAndId> allIds = new HashSet<AgencyAndId>();
  allIds.addAll(calendarIds);
  allIds.addAll(calendarDateIds);
  return new ArrayList<AgencyAndId>(allIds);
}
 

import org.onebusaway.gtfs.model.AgencyAndId; //导入依赖的package包/类
@Test
public void testDaylightSavingTimeCalendarDatesOnly() throws IOException {

  CalendarServiceDataFactoryImpl factory = new CalendarServiceDataFactoryImpl();

  Agency agencyA = agency("A", "America/Los_Angeles");

  AgencyAndId serviceId = new AgencyAndId("A", "2");

  ServiceCalendarDate cd1 = calendarDate(serviceId, new ServiceDate(2012, 3,
      10), ServiceCalendarDate.EXCEPTION_TYPE_ADD);
  ServiceCalendarDate cd2 = calendarDate(serviceId, new ServiceDate(2012, 3,
      11), ServiceCalendarDate.EXCEPTION_TYPE_ADD);
  ServiceCalendarDate cd3 = calendarDate(serviceId, new ServiceDate(2012, 3,
      12), ServiceCalendarDate.EXCEPTION_TYPE_ADD);

  GtfsRelationalDaoImpl dao = new GtfsRelationalDaoImpl();
  factory.setGtfsDao(dao);

  saveEntities(dao, agencyA);
  saveEntities(dao, cd1, cd2, cd3);

  CalendarServiceData data = factory.createData();
  List<ServiceDate> serviceDates = data.getServiceDatesForServiceId(serviceId);
  assertEquals(serviceDates,
      Arrays.asList(cd1.getDate(), cd2.getDate(), cd3.getDate()));
}