Source code for shasta.preprocessing.utils

from pathlib import Path

import numpy as np
import osmnx as ox


def extract_building_info(osm_path, save_fig=False):
    """Perfrom initial building setup."""
    read_path = osm_path
    G = ox.graph_from_xml(read_path, retain_all=True)

    # TODO: This method doesn't work if the building info is not there in OSM
    nodes, streets = ox.graph_to_gdfs(G)

    west, north, east, south = nodes.geometry.total_bounds
    polygon = ox.utils_geo.bbox_to_poly(north, south, east, west)
    gdf = ox.geometries.geometries_from_polygon(polygon, tags={'building': True})
    buildings_proj = ox.project_gdf(gdf, to_crs="EPSG:4326").to_crs(4328)

    # Building Info
    building_info = buildings_proj.copy()

    # Add more information
    # Save the dataframe representing buildings
    building_info['lon'] = buildings_proj['geometry'].centroid.x
    building_info['lat'] = buildings_proj['geometry'].centroid.y
    building_info['area'] = buildings_proj.area
    building_info['perimeter'] = buildings_proj.length
    try:
        building_info.loc[:, 'height'] = buildings_proj['height']
    except KeyError:
        building_info.loc[:, 'height'] = 10  # assumption
    building_info['id'] = np.arange(len(buildings_proj))

    if save_fig:
        save_path = Path(osm_path)
        ox.plot_graph(G)
        ox.plot.plot_footprints(
            buildings_proj, save=True, filepath=save_path.with_suffix('.jpg')
        )

    return building_info


def save_buildings_map(osm_path):
    raise NotImplementedError


def extract_path_info(osm_path):
    raise NotImplementedError