org.onebusaway.gtfs_transformer.updates.DeduplicateRoutesStrategy Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of onebusaway-gtfs-transformer Show documentation
Show all versions of onebusaway-gtfs-transformer Show documentation
A Java library for transforming Google Transit Feed Spec feeds
The newest version!
/**
* Copyright (C) 2011 Brian Ferris
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.onebusaway.gtfs_transformer.updates;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import org.onebusaway.collections.FactoryMap;
import org.onebusaway.gtfs.model.AgencyAndId;
import org.onebusaway.gtfs.model.Route;
import org.onebusaway.gtfs.model.Trip;
import org.onebusaway.gtfs.services.GtfsMutableRelationalDao;
import org.onebusaway.gtfs_transformer.services.EntityTransformStrategy;
import org.onebusaway.gtfs_transformer.services.GtfsTransformStrategy;
import org.onebusaway.gtfs_transformer.services.TransformContext;
import org.onebusaway.gtfs_transformer.updates.UpdateLibrary;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class DeduplicateRoutesStrategy implements GtfsTransformStrategy {
private static Logger _log = LoggerFactory.getLogger(org.onebusaway.gtfs_transformer.updates.DeduplicateServiceIdsStrategy.class);
@Override
public String getName() {
return this.getClass().getSimpleName();
}
@Override
public void run(TransformContext context, GtfsMutableRelationalDao dao) {
Map> routesById = new FactoryMap>(
new ArrayList());
for (Route route : dao.getAllRoutes()) {
AgencyAndId aid = route.getId();
String id = aid.getId();
int index = id.indexOf('_');
if (index == -1)
continue;
String commonId = id.substring(index + 1);
AgencyAndId commonIdFull = new AgencyAndId(aid.getAgencyId(), commonId);
routesById.get(commonIdFull).add(route);
}
for (Map.Entry> entry : routesById.entrySet()) {
AgencyAndId routeId = entry.getKey();
List routes = entry.getValue();
if (routes.size() == 1)
continue;
// Remove the route with the old id
Route route = routes.get(0);
dao.removeEntity(route);
// Add the route with the new id
route.setId(routeId);
dao.saveEntity(route);
for (int i = 1; i < routes.size(); i++) {
Route duplicateRoute = routes.get(i);
dao.removeEntity(duplicateRoute);
List trips = dao.getTripsForRoute(duplicateRoute);
for (Trip trip : trips)
trip.setRoute(route);
}
}
UpdateLibrary.clearDaoCache(dao);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy