private void readDao(GtfsReader reader) throws IOException { dao = new GtfsRelationalDaoImpl(); reader.setEntityStore(dao); reader.setDefaultAgencyId(getFeedId().getId()); reader.run(); }
public GtfsImport(File path) throws IOException { GtfsReader reader = new GtfsReader(); reader.setInputLocation(path); readFeedId(reader); readDao(reader); }
GtfsReader reader = new GtfsReader(); reader.setInputSource(gtfsBundle.getCsvInputSource()); reader.setEntityStore(store); reader.setInternStrings(true); reader.setDefaultAgencyId(gtfsFeedId.getId()); reader.addEntityHandler(counter); reader.addEntityHandler(new EntityBikeability(true)); for (Class<?> entityClass : reader.getEntityClasses()) { LOG.info("reading entities: " + entityClass.getName()); reader.readEntities(entityClass); store.flush(); for (Agency agency : reader.getAgencies()) { String agencyId = agency.getId(); LOG.info("This Agency has the ID {}", agencyId); reader.addAgencyIdMapping(agencyId, generatedAgencyId); // NULL key should work agency.setId(generatedAgencyId); agencyId = generatedAgencyId; shapePoint.getShapeId().setAgencyId(reader.getDefaultAgencyId()); route.getId().setAgencyId(reader.getDefaultAgencyId()); generateRouteColor(route); stop.getId().setAgencyId(reader.getDefaultAgencyId());
@SuppressWarnings("unchecked") public static void main(String[] args) throws IOException { if (args.length != 2) { System.err.println("usage: gtfs_path polygon_path"); System.exit(-1); } Geometry points = readPoints(args[1]); GtfsReader reader = new GtfsReader(); reader.setDefaultAgencyId("1"); reader.getEntityClasses().retainAll(Arrays.asList(Stop.class)); reader.setInputLocation(new File(args[0])); reader.addEntityHandler(new EntityHandlerImpl(points)); reader.run(); }
public GtfsMutableRelationalDao read(GtfsReader reader) throws IOException { reader.setInputLocation(_path); GtfsRelationalDaoImpl dao = new GtfsRelationalDaoImpl(); reader.setEntityStore(dao); try { reader.run(); } finally { reader.close(); } return dao; }
GtfsReader reader = new GtfsReader(); reader.addEntityHandler(handler); reader.setInputLocation(bundle.getPath()); if (bundle.getDefaultAgencyId() != null) reader.setDefaultAgencyId(bundle.getDefaultAgencyId()); reader.readEntities(Stop.class);
private static Collection<Stop> readStopsFromGtfsPath(String path, String defaultAgencyId) throws IOException { GtfsReader reader = new GtfsReader(); reader.setInputLocation(new File(path)); if (defaultAgencyId != null) reader.setDefaultAgencyId(defaultAgencyId); reader.readEntities(Agency.class); reader.readEntities(Stop.class); return reader.getEntityStore().getAllEntitiesForType(Stop.class); }
reader.setEntityStore(store); reader.addEntityHandler(new EntityCounter()); List<Class<?>> entityClasses = _readers.get(0).getEntityClasses(); reader.setAgencies(agencies); reader.readEntities(entityClass); agencies = new ArrayList<Agency>(reader.getAgencies());
GtfsReader reader = new GtfsReader(); reader.setEntitySchemaFactory(factory); reader.setInputLocation(gtfsBundle.getPath()); reader.setDefaultAgencyId(gtfsBundle.getDefaultAgencyId()); reader.addAgencyIdMapping(entry.getKey(), entry.getValue());
public GtfsReader() { _entityClasses.add(Agency.class); _entityClasses.add(ShapePoint.class); _entityClasses.add(Route.class); _entityClasses.add(Stop.class); _entityClasses.add(Trip.class); _entityClasses.add(StopTime.class); _entityClasses.add(ServiceCalendar.class); _entityClasses.add(ServiceCalendarDate.class); _entityClasses.add(FareAttribute.class); _entityClasses.add(FareRule.class); _entityClasses.add(Frequency.class); _entityClasses.add(Pathway.class); _entityClasses.add(Transfer.class); _entityClasses.add(FeedInfo.class); CsvTokenizerStrategy tokenizerStrategy = new CsvTokenizerStrategy(); tokenizerStrategy.getCsvParser().setTrimInitialWhitespace(true); setTokenizerStrategy(tokenizerStrategy); setTrimValues(true); /** * Prep the Entity Schema Factories */ DefaultEntitySchemaFactory schemaFactory = createEntitySchemaFactory(); setEntitySchemaFactory(schemaFactory); CsvEntityContext ctx = getContext(); ctx.put(KEY_CONTEXT, _context); addEntityHandler(new EntityHandlerImpl()); }
public void run() throws IOException { run(getInputSource()); }
public void run(CsvInputSource source) throws IOException { List<Class<?>> classes = getEntityClasses(); _entityStore.open(); for (Class<?> entityClass : classes) { _log.info("reading entities: " + entityClass.getName()); readEntities(entityClass, source); _entityStore.flush(); } _entityStore.close(); }
public GtfsMutableRelationalDao read() throws IOException { GtfsReader reader = new GtfsReader(); return read(reader); }
public void addEntitySchemasFromGtfsReader(GtfsReader reader) { EntitySchemaFactory factory = reader.getEntitySchemaFactory(); for (Class<?> entityType : reader.getEntityClasses()) { EntitySchema schema = factory.getSchema(entityType); addEntitySchema(schema); } }
public List<Agency> getAgencies() { return GtfsReader.this.getAgencies(); }
@SuppressWarnings("unchecked") public static void main(String[] args) throws IOException { if (args.length != 2) { System.err.println("usage: gtfs_path polygon_path"); System.exit(-1); } Geometry points = readPoints(args[1]); GtfsReader reader = new GtfsReader(); reader.setDefaultAgencyId("1"); reader.getEntityClasses().retainAll(Arrays.asList(Stop.class)); reader.setInputLocation(new File(args[0])); reader.addEntityHandler(new EntityHandlerImpl(points)); reader.run(); }
public GtfsMutableRelationalDao read(GtfsReader reader) throws IOException { reader.setInputLocation(_path); GtfsRelationalDaoImpl dao = new GtfsRelationalDaoImpl(); reader.setEntityStore(dao); try { reader.run(); } finally { reader.close(); } return dao; }
GtfsReader reader = new GtfsReader(); reader.addEntityHandler(handler); reader.setInputLocation(bundle.getPath()); if (bundle.getDefaultAgencyId() != null) reader.setDefaultAgencyId(bundle.getDefaultAgencyId()); reader.readEntities(Stop.class);
private static Collection<Stop> readStopsFromGtfsPath(String path, String defaultAgencyId) throws IOException { GtfsReader reader = new GtfsReader(); reader.setInputLocation(new File(path)); if (defaultAgencyId != null) reader.setDefaultAgencyId(defaultAgencyId); reader.readEntities(Agency.class); reader.readEntities(Stop.class); return reader.getEntityStore().getAllEntitiesForType(Stop.class); }
reader.setEntityStore(store); reader.addEntityHandler(new EntityCounter()); List<Class<?>> entityClasses = _readers.get(0).getEntityClasses(); reader.setAgencies(agencies); reader.readEntities(entityClass); agencies.addAll(reader.getAgencies());