utility-route-planner/multilayer_main.py at main · alliander-opensource/utility-route-planner · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
# SPDX-FileCopyrightText: Contributors to the utility-route-project and Alliander N.V.
#
# SPDX-License-Identifier: Apache-2.0
import pathlib

import shapely
import geopandas as gpd
import time

import structlog
import typer
from structlog.contextvars import bound_contextvars

from settings import Config
from utility_route_planner.models.benchmark_routes import BenchmarkRouteCollection
from utility_route_planner.models.mcda.mcda_engine import McdaCostSurfaceEngine
from utility_route_planner.models.multilayer_network.hexagon_graph.hexagon_edge_generator import HexagonEdgeGenerator
from utility_route_planner.models.multilayer_network.hexagon_graph.hexagon_graph_builder import HexagonGraphBuilder
from utility_route_planner.models.multilayer_network.hexagon_graph.hexagon_grid_builder import HexagonGridBuilder
from utility_route_planner.models.multilayer_network.multilayer_route_planner import MultilayerRouteEngine
from utility_route_planner.models.multilayer_network.osm_graph_downloader import OSMGraphDownloader
from utility_route_planner.models.multilayer_network.osm_graph_preprocessing import OSMGraphPreprocessor
from utility_route_planner.models.multilayer_network.pipe_ramming import GetPotentialPipeRammingCrossings
from utility_route_planner.util.write import reset_geopackage

logger = structlog.get_logger(__name__)
app = typer.Typer(pretty_exceptions_enable=False)


def run_multilayer_network(
    preset: str,
    path_geopackage_mcda_input: pathlib.Path,
    start_mid_end_points: shapely.LineString,
    project_area_geometry: shapely.Polygon,
    write_output: bool,
):
    reset_geopackage(Config.PATH_GEOPACKAGE_MULTILAYER_NETWORK_OUTPUT, truncate=False)
    reset_geopackage(Config.PATH_GEOPACKAGE_MCDA_OUTPUT, truncate=False)

    start_cpu_time = time.process_time_ns()

    raw_graph = OSMGraphDownloader(project_area_geometry).download_graph()
    osm_graph_preprocessed = OSMGraphPreprocessor(raw_graph).preprocess_graph()

    mcda_engine = McdaCostSurfaceEngine(preset, path_geopackage_mcda_input, project_area_geometry)
    mcda_engine.preprocess_vectors()

    raster_groups = {
        criteria_key: criteria.group for criteria_key, criteria in mcda_engine.raster_preset.criteria.items()
    }
    grid_constructor = HexagonGridBuilder(hexagon_size=Config.HEXAGON_SIZE, block_size=Config.HEXAGON_BLOCK_SIZE)
    hexagon_edge_generator = HexagonEdgeGenerator()
    hexagon_graph_builder = HexagonGraphBuilder(grid_builder=grid_constructor, edge_generator=hexagon_edge_generator)
    cost_surface_graph, cost_surface_nodes = hexagon_graph_builder.build_graph(
        mcda_engine.project_area_geometry,
        raster_groups,
        mcda_engine.processed_vectors,
    )
    pipe_ramming = GetPotentialPipeRammingCrossings(osm_graph_preprocessed, cost_surface_graph, cost_surface_nodes)
    pipe_ramming.get_crossings()

    multi_layer_route_engine = MultilayerRouteEngine(
        pipe_ramming.cost_surface_graph,
        pipe_ramming.osm_graph,
        pipe_ramming.cost_surface_nodes,
        Config.HEXAGON_SIZE,
        prefix="testing_",
        write_output=write_output,
    )
    multi_layer_route_engine.find_route(start_mid_end_points)

    logger.info(f"Multilayer route CPU time: {(time.process_time_ns() - start_cpu_time) / 1e9:.2f} seconds.")


@app.command()
def run_debug_case(write_routing_output: bool = False):
    with bound_contextvars(project_area="Componistenbuurt"):
        logger.info("Running multilayer routing engine for debug project area")
        run_multilayer_network(
            Config.RASTER_PRESET_NAME_BENCHMARK,
            Config.PYTEST_PATH_GEOPACKAGE_MCDA,
            shapely.LineString([(174847.18, 451178.43), (175746.347, 450435.534)]),
            gpd.read_file(Config.PYTEST_PATH_GEOPACKAGE_MCDA, layer=Config.PYTEST_LAYER_NAME_PROJECT_AREA)
            .iloc[0]
            .geometry,
            write_routing_output,
        )


@app.command()
def run_benchmark_case(benchmark_case_id: int, write_routing_output: bool = False):
    benchmark_routes = BenchmarkRouteCollection()
    with bound_contextvars(benchmark_id=benchmark_case_id):
        benchmark_route = benchmark_routes.get_case(benchmark_case_id)
        run_multilayer_network(
            Config.RASTER_PRESET_NAME_BENCHMARK,
            benchmark_route.path_geopackage,
            gpd.read_file(benchmark_route.path_geopackage, layer=benchmark_route.layer_name_human_designed_route)
            .iloc[0]
            .geometry,
            gpd.read_file(benchmark_route.path_geopackage, layer=benchmark_route.layer_name_project_area)
            .iloc[0]
            .geometry,
            write_routing_output,
        )


if __name__ == "__main__":
    app()