def _compute_geometry(self): return [ SumoRoadNetwork.buffered_lane_or_edge( edge, width=sum([lane.getWidth() for lane in edge.getLanes()]) ) for edge in self.edges ]
def to_geometry(self, road_network: SumoRoadNetwork) -> Polygon: def resolve_offset(offset, geometry_length, lane_length, buffer_from_ends): if offset == "base" or offset == 0: return buffer_from_ends # push off of end of lane elif offset == "max": return lane_length - geometry_length - buffer_from_ends elif offset == "random": return random.uniform( 0, lane_length - geometry_length - buffer_from_ends) else: return float(offset) lane_shapes = [] edge_id, lane_idx, offset = self.start edge = road_network.edge_by_id(edge_id) for lane_idx in range(lane_idx, lane_idx + self.n_lanes): lane = edge.getLanes()[lane_idx] lane_length = lane.getLength() geom_length = max(self.length - 1e-6, 1e-6) assert lane_length > geom_length # Geom is too long for lane assert geom_length > 0 # Geom length is negative lane_shape = SumoRoadNetwork.buffered_lane_or_edge( lane, width=lane.getWidth() + 0.3) min_cut = resolve_offset(offset, geom_length, lane_length, 1e-6) # Second cut takes into account shortening of geometry by `min_cut`. max_cut = min(min_cut + geom_length, lane_length - min_cut - 1e-6) lane_shape = road_network.split_lane_shape_at_offset( Polygon(lane_shape), lane, min_cut) if isinstance(lane_shape, GeometryCollection): if len(lane_shape) < 2: break lane_shape = lane_shape[1] lane_shape = road_network.split_lane_shape_at_offset( lane_shape, lane, max_cut, )[0] lane_shapes.append(lane_shape) geom = unary_union(MultiPolygon(lane_shapes)) return geom