j40-cejst-2/data/data-pipeline/data_pipeline/tile/generate.py

import os
from pathlib import Path
from subprocess import call

from data_pipeline.utils import get_module_logger, remove_all_from_dir

logger = get_module_logger(__name__)


def generate_tiles(data_path: Path, generate_tribal_layer: bool) -> None:
    """Generates map tiles from geojson files

    Args:
        data_path (Path):  Path to data folder
        generate_tribal_layer (bool): If true, generate the tribal layer of the map

    Returns:
        None
    """

    def _generate_score_tiles() -> None:
        """Generates score map tiles"""
        score_tiles_path = data_path / "score" / "tiles"
        high_tile_path = score_tiles_path / "high"
        low_tile_path = score_tiles_path / "low"
        score_geojson_dir = data_path / "score" / "geojson"

        USA_HIGH_MIN_ZOOM = 5
        USA_HIGH_MAX_ZOOM = 11
        USA_LOW_MIN_ZOOM = 0
        USA_LOW_MAX_ZOOM = 7

        # remove existing mbtiles file
        remove_all_from_dir(score_tiles_path)

        # create dirs
        os.mkdir(high_tile_path)
        os.mkdir(low_tile_path)

        # generate high mbtiles file
        logger.info("Generating USA High mbtiles file")
        cmd = "tippecanoe "
        cmd += f"--minimum-zoom={USA_HIGH_MIN_ZOOM} --maximum-zoom={USA_HIGH_MAX_ZOOM} --layer=blocks "
        cmd += f"--output={high_tile_path}/usa_high.mbtiles "
        cmd += str(score_geojson_dir / "usa-high.json")
        call(cmd, shell=True)

        # generate high mvts
        logger.info("Generating USA High mvt folders and files")
        cmd = "tippecanoe "
        cmd += f"--minimum-zoom={USA_HIGH_MIN_ZOOM} --maximum-zoom={USA_HIGH_MAX_ZOOM} --no-tile-compression "
        cmd += "--drop-densest-as-needed "
        cmd += f"--output-to-directory={high_tile_path} --layer=blocks "
        cmd += str(score_geojson_dir / "usa-high.json")
        call(cmd, shell=True)

        # generate low mbtiles file
        logger.info("Generating USA Low mbtiles file")
        cmd = "tippecanoe "
        cmd += f"--minimum-zoom={USA_LOW_MIN_ZOOM} --maximum-zoom={USA_LOW_MAX_ZOOM} --layer=blocks "
        cmd += f"--output={low_tile_path}/usa_low.mbtiles "
        cmd += str(score_geojson_dir / "usa-low.json")
        call(cmd, shell=True)

        # generate low mvts
        logger.info("Generating USA Low mvt folders and files")
        cmd = "tippecanoe "
        cmd += f"--minimum-zoom={USA_LOW_MIN_ZOOM} --maximum-zoom={USA_LOW_MAX_ZOOM} --no-tile-compression "
        cmd += "--drop-densest-as-needed "
        cmd += f"--output-to-directory={low_tile_path} --layer=blocks "
        cmd += str(score_geojson_dir / "usa-low.json")
        call(cmd, shell=True)

    def _generate_tribal_tiles() -> None:
        """Generates tribal layer tiles"""

        USA_TRIBAL_MIN_ZOOM = 0
        USA_TRIBAL_MAX_ZOOM = 11

        tribal_tiles_path = data_path / "tribal" / "tiles"
        tribal_geojson_dir = data_path / "tribal" / "geojson"

        # remove existing mbtiles file
        remove_all_from_dir(tribal_tiles_path)

        # generate mbtiles file
        logger.info("Generating Tribal mbtiles file")
        cmd = "tippecanoe "
        cmd += "--layer=blocks "
        cmd += "--base-zoom=3 "
        cmd += f"--minimum-zoom={USA_TRIBAL_MIN_ZOOM} --maximum-zoom={USA_TRIBAL_MAX_ZOOM} "
        cmd += f"--output={tribal_tiles_path}/usa.mbtiles "
        cmd += str(tribal_geojson_dir / "usa.json")
        call(cmd, shell=True)

        # generate mvts
        logger.info("Generating Tribal mvt folders and files")
        cmd = "tippecanoe "
        cmd += "--layer=blocks "
        cmd += "--base-zoom=3 "
        cmd += "--no-tile-compression "
        cmd += "--drop-densest-as-needed "
        cmd += f"--minimum-zoom={USA_TRIBAL_MIN_ZOOM} --maximum-zoom={USA_TRIBAL_MAX_ZOOM} "
        cmd += f"--output-to-directory={tribal_tiles_path} "
        cmd += str(tribal_geojson_dir / "usa.json")
        call(cmd, shell=True)

    if generate_tribal_layer:
        _generate_tribal_tiles()
    else:
        _generate_score_tiles()