def step_pt_pathfinder(
self,
broken_routes=True,
broken_modes=True,
route_column='route_id',
mode_column='route_type',
speedup=False,
walk_on_road=False,
keep_graph=False,
keep_pathfinder=False,
force=False,
**kwargs):
"""
* requires: zones, links, footpaths, zone_to_road, zone_to_transit
* builds: pt_los
"""
sets = ['nodes', 'links', 'zones']
if walk_on_road:
sets += ['road_nodes', 'road_links']
if not force:
self.integrity_test_collision(sets)
self.links = engine.graph_links(self.links)
publicpathfinder = PublicPathFinder(self, walk_on_road=walk_on_road)
publicpathfinder.find_best_paths(
broken_routes=broken_routes,
broken_modes=broken_modes,
route_column=route_column,
mode_column=mode_column,
speedup=speedup,
**kwargs
)
self.pt_los = publicpathfinder.paths.copy()
if keep_graph:
self.nx_graph=publicpathfinder.nx_graph
if keep_pathfinder:
self.publicpathfinder = publicpathfinder
analysis_nodes = pd.concat([self.nodes, self.road_nodes]) if walk_on_road else self.nodes
self.pt_los = analysis.path_analysis_od_matrix(
od_matrix=self.pt_los,
links=self.links,
nodes=analysis_nodes,
centroids=self.centroids,
)
def step_pathfinder(
self,
walk_on_road=False,
complete=True,
**kwargs
):
"""
* requires: links, footpaths, zone_to_transit, zone_to_road
* builds: pt_los
"""
assert self.links['time'].isnull().sum() == 0
self.links = engine.graph_links(self.links)
self.walk_on_road = walk_on_road
if walk_on_road:
footpaths = self.road_links.copy()
footpaths['time'] = footpaths['walk_time']
ntlegs = self.zone_to_road
nodes = self.road_nodes
else:
footpaths = self.footpaths
ntlegs = self.zone_to_transit
nodes = self.nodes
#TODO even with walk on road, transit nodes may not belong to road_nodes
self.pt_los, self.graph = engine.path_and_duration_from_links_and_ntlegs(
self.links,
ntlegs=ntlegs,
pole_set=set(self.zones.index),
footpaths=footpaths,
**kwargs
)
if complete:
self.pt_los = analysis.path_analysis_od_matrix(
od_matrix=self.pt_los,
links=self.links,
nodes=nodes,
centroids=self.centroids,
)
def __init__(self, model, walk_on_road=False):
self.zones = model.zones.copy()
self.links = engine.graph_links(model.links.copy())
if walk_on_road:
road_links = model.road_links.copy()
road_links['time'] = road_links['walk_time']
self.footpaths = pd.concat(
[model.footpaths, road_links, model.road_to_transit])
self.ntlegs = pd.concat(
[model.zone_to_road, model.zone_to_transit])
else:
self.footpaths = model.footpaths.copy()
self.ntlegs = model.zone_to_transit.copy()
try:
self.centroids = model.centroids.copy()
except AttributeError:
self.centroids = self.zones.copy()
self.centroids['geometry'] = self.centroids['geometry'].apply(
lambda g: g.centroid)