Draw cities on SPC outlook maps

This commit is contained in:
XANTRONIX 2025-03-31 15:19:16 -04:00
parent 8c4b93d50e
commit ed7c5bf317

View file

@ -3,6 +3,7 @@
import argparse
import cairo
from xmet.db import Database
from xmet.spc import SPCOutlookParser, SPCOutlook, SPCOutlookMap
ASSETS = {
@ -20,6 +21,8 @@ ASSETS = {
def render_categorical(conus: SPCOutlookMap,
outlook: SPCOutlook,
args):
db = Database.connect(args.db)
assets = ASSETS['dark'] if args.dark else ASSETS['light']
with cairo.SVGSurface(args.categorical, conus.width, conus.height) as surface:
@ -27,6 +30,7 @@ def render_categorical(conus: SPCOutlookMap,
conus.draw_base_map_from_file(cr, assets['map'])
conus.draw_categories(cr, outlook)
conus.draw_cities(cr, db)
conus.draw_logo(cr, assets['logo'])
if args.dark:
@ -39,6 +43,8 @@ def render_probabilistic(conus: SPCOutlookMap,
hazard: str,
path: str,
args):
db = Database.connect(args.db)
assets = ASSETS['dark'] if args.dark else ASSETS['light']
with cairo.SVGSurface(path, conus.width, conus.height) as surface:
@ -46,6 +52,7 @@ def render_probabilistic(conus: SPCOutlookMap,
conus.draw_base_map_from_file(cr, assets['map'])
conus.draw_probabilities(cr, outlook, hazard.upper())
conus.draw_cities(cr, db)
conus.draw_logo(cr, assets['logo'])
if args.dark:
@ -54,6 +61,7 @@ def render_probabilistic(conus: SPCOutlookMap,
conus.draw_annotation(cr, f"Day {outlook.day} Probabilistic {hazard.lower().capitalize()} Risk")
argparser = argparse.ArgumentParser(description='Render graphical SPC outlooks from text file')
argparser.add_argument('db', help='Spatialite database file')
argparser.add_argument('--dark', action='store_true', help='Output dark mode graphics')
argparser.add_argument('--categorical', type=str, help='Output categorical risk graphic file')
argparser.add_argument('--any-severe', type=str, help='Output probabilistic severe risk graphic file')