import atexit
import logging
import os
import sys
import json
from contextlib import ExitStack
from select import select
import time
import numpy as np
import numpy.lib.recfunctions as rfn
import pandas as pd
import numba as nb
from numba.typed import Dict, List
from numba.types import Tuple as nb_Tuple
from numba.types import int32 as nb_int32
from numba.types import int64 as nb_int64
from oasislmf.utils.data import analysis_settings_loader
from oasis_data_manager.filestore.config import get_storage_from_config_path
from oasislmf.pytools.common.data import nb_areaperil_int, nb_oasis_float, oasis_float, nb_oasis_int, oasis_int, correlations_dtype, items_dtype
from oasislmf.pytools.common.event_stream import PIPE_CAPACITY
from oasislmf.pytools.common.input_files import read_coverages, read_correlations
from oasislmf.pytools.data_layer.footprint_layer import FootprintLayerClient
from oasislmf.pytools.getmodel.footprint import Footprint
from oasislmf.pytools.getmodel.manager import get_damage_bins, get_vulns, get_intensity_bin_dict, encode_peril_id
from oasislmf.pytools.gul.common import MAX_LOSS_IDX, CHANCE_OF_LOSS_IDX, TIV_IDX, STD_DEV_IDX, MEAN_IDX, NUM_IDX
from oasislmf.pytools.gul.core import compute_mean_loss, get_gul
from oasislmf.pytools.gul.manager import write_losses, adjust_byte_mv_size
from oasislmf.pytools.gul.random import (compute_norm_cdf_lookup, compute_norm_inv_cdf_lookup,
generate_correlated_hash_vector, generate_hash,
generate_hash_hazard, get_corr_rval_float, get_random_generator)
from oasislmf.pytools.gul.utils import binary_search
from oasislmf.pytools.gulmc.aggregate import (
process_aggregate_vulnerability, process_vulnerability_weights, read_aggregate_vulnerability,
read_vulnerability_weights, )
from oasislmf.pytools.gulmc.common import (DAMAGE_TYPE_ABSOLUTE,
DAMAGE_TYPE_DURATION,
DAMAGE_TYPE_RELATIVE,
NP_BASE_ARRAY_SIZE, Keys,
ItemAdjustment,
NormInversionParameters,
coverage_type, gul_header,
gulSampleslevelHeader_size,
gulSampleslevelRec_size,
haz_arr_type, items_MC_data_type,
gulmc_compute_info_type)
from oasislmf.pytools.gulmc.items import read_items, generate_item_map
from oasislmf.pytools.utils import redirect_logging
from oasislmf.utils.ping import oasis_ping
from oasislmf.utils.defaults import SERVER_UPDATE_TIME
[docs]
logger = logging.getLogger(__name__)
[docs]
VULN_LOOKUP_KEY_TYPE = nb_int64
[docs]
VULN_LOOKUP_VALUE_TYPE = nb_Tuple((nb_int32, nb_int32))
# Sentinel value to distinguish eff_damage_cdf keys from per-bin keys
# eff_damage_cdf key: int64(eff_cdf_id) << 32 | 0xFFFFFFFF
# per-bin cdf key: int64(eff_cdf_id) << 32 | int64(haz_bin_id)
[docs]
EFF_CDF_KEY_SENTINEL = nb_int64(0xFFFFFFFF)
# parameter for get_corr_rval in a normal cdf
[docs]
inv_factor = (norm_inv_N - 1) / (x_max - x_min)
[docs]
norm_factor = (norm_inv_N - 1) / (cdf_max - cdf_min)
@nb.njit(cache=True)
[docs]
def gen_empty_vuln_cdf_lookup(list_size, compute_info):
"""Generate structures needed to store and retrieve vulnerability cdfs in the cache.
Initializes a Numba typed Dict and a keys list used for circular (LRU-like) eviction.
The Dict maps composite int64 keys to (slot_index, cdf_length) tuples. The keys list
tracks which key occupies each slot so that evicted entries can be removed from the Dict.
The composite int64 key encodes:
- eff_cdf_id (upper 32 bits): sequential id assigned per unique (areaperil, vuln_id)
group during reconstruct_coverages.
- discriminator (lower 32 bits): 0xFFFFFFFF for the effective damage cdf,
or haz_bin_id for per-intensity-bin vulnerability cdfs.
Args:
list_size (int): maximum number of cdfs that can be stored in the cache (i.e., the
number of rows in cached_vuln_cdfs).
compute_info (gulmc_compute_info_type): computation state; its 'next_cached_vuln_cdf_i'
field is reset to 0.
Returns:
cached_vuln_cdf_lookup (Dict[int64, Tuple(int32, int32)]): empty dict mapping
composite int64 cache key to (slot_index, cdf_length).
cached_vuln_cdf_lookup_keys (List[int64]): list of length `list_size`, initialized
with dummy keys (-1), used for eviction tracking.
"""
cached_vuln_cdf_lookup = Dict.empty(VULN_LOOKUP_KEY_TYPE, VULN_LOOKUP_VALUE_TYPE)
cached_vuln_cdf_lookup_keys = List.empty_list(VULN_LOOKUP_KEY_TYPE)
dummy = nb_int64(-1)
for _ in range(list_size):
cached_vuln_cdf_lookup_keys.append(dummy)
compute_info['next_cached_vuln_cdf_i'] = 0
return cached_vuln_cdf_lookup, cached_vuln_cdf_lookup_keys
[docs]
def get_peril_id(input_path):
"""
Get peril_id associated with item_id
Args:
input_path (str): The directory path where the 'gul_summary_map.csv' file is located.
Returns:
np.ndarray: A structured NumPy array with the following fields:
- 'item_id' (oasis_int): The item ID as an integer.
- 'peril_id' (oasis_int): The encoded peril ID as an integer.
"""
dtype = np.dtype([
('item_id', oasis_int),
('peril_id', oasis_int)
])
item_peril = pd.read_csv(
os.path.join(input_path, 'gul_summary_map.csv'),
usecols=['item_id', 'peril_id']
)[['item_id', 'peril_id']]
item_peril['peril_id'] = item_peril['peril_id'].apply(encode_peril_id)
item_peril = np.array(
list(item_peril.itertuples(index=False, name=None)),
dtype=dtype)
return item_peril
[docs]
def get_vuln_rngadj(run_dir, vuln_dict):
"""
Loads vulnerability adjustments from the analysis settings file.
Args:
run_dir (str): path to the run directory (used to load the analysis settings)
Returns: (Dict[nb_int32, nb_float64]) vulnerability adjustments dictionary
"""
settings_path = os.path.join(run_dir, "analysis_settings.json")
vuln_adj = np.ones(len(vuln_dict), dtype=oasis_float)
if not os.path.exists(settings_path):
logger.debug(f"analysis_settings.json not found in {run_dir}.")
return vuln_adj
vulnerability_adjustments_field = analysis_settings_loader(settings_path).get('vulnerability_adjustments', None)
if vulnerability_adjustments_field is not None:
adjustments = vulnerability_adjustments_field.get('adjustments', None)
else:
adjustments = None
if adjustments is None:
logger.debug(f"vulnerability_adjustments not found in {settings_path}.")
return vuln_adj
for key, value in adjustments.items():
if nb_int32(key) in vuln_dict.keys():
vuln_adj[vuln_dict[nb_int32(key)]] = nb_oasis_float(value)
return vuln_adj
@redirect_logging(exec_name='gulmc')
[docs]
def run(run_dir,
ignore_file_type,
sample_size,
loss_threshold,
alloc_rule,
debug,
random_generator,
peril_filter=[],
file_in=None,
file_out=None,
data_server=None,
ignore_correlation=False,
ignore_haz_correlation=False,
effective_damageability=False,
max_cached_vuln_cdf_size_MB=200,
model_df_engine="oasis_data_manager.df_reader.reader.OasisPandasReader",
dynamic_footprint=False,
**kwargs):
"""Execute the main gulmc workflow.
Args:
run_dir (str): the directory of where the process is running
ignore_file_type set(str): file extension to ignore when loading
sample_size (int): number of random samples to draw.
loss_threshold (float): threshold above which losses are printed to the output stream.
alloc_rule (int): back-allocation rule.
debug (int): for each random sample, print to the output stream the random loss (if 0), the random value used to draw
the hazard intensity sample (if 1), the random value used to draw the damage sample (if 2). Defaults to 0.
random_generator (int): random generator function id.
peril_filter (list[int], optional): list of perils to include in the computation (if None, all perils will be included). Defaults to [].
file_in (str, optional): filename of input stream. Defaults to None.
file_out (str, optional): filename of output stream. Defaults to None.
data_server (bool, optional): if True, run the data server. Defaults to None.
ignore_correlation (bool, optional): if True, do not compute correlated random samples. Defaults to False.
effective_damageability (bool, optional): if True, it uses effective damageability to draw damage samples instead of
using the full monte carlo approach (i.e., to draw hazard intensity first, then damage).
max_cached_vuln_cdf_size_MB (int, optional): size in MB of the in-memory cache to store and reuse vulnerability cdf. Defaults to 200.
model_df_engine: (str) The engine to use when loading model dataframes
Raises:
ValueError: if alloc_rule is not 0, 1, 2, or 3.
ValueError: if alloc_rule is 1, 2, or 3 when debug is 1 or 2.
Returns:
int: 0 if no errors occurred.
"""
logger.info("starting gulmc")
model_storage = get_storage_from_config_path(
os.path.join(run_dir, 'model_storage.json'),
os.path.join(run_dir, 'static'),
)
input_path = os.path.join(run_dir, 'input')
ignore_file_type = set(ignore_file_type)
if alloc_rule not in [0, 1, 2, 3]:
raise ValueError(f"Expect alloc_rule to be 0, 1, 2, or 3, got {alloc_rule}")
if debug > 0 and alloc_rule != 0:
raise ValueError(f"Expect alloc_rule to be 0 if debug is 1 or 2, got {alloc_rule}")
if data_server:
logger.debug("data server active")
FootprintLayerClient.register()
logger.debug("registered with data server")
atexit.register(FootprintLayerClient.unregister)
else:
logger.debug("data server not active")
with ExitStack() as stack:
if file_in is None:
streams_in = sys.stdin.buffer
else:
streams_in = stack.enter_context(open(file_in, 'rb'))
event_id_mv = memoryview(bytearray(4))
event_ids = np.ndarray(1, buffer=event_id_mv, dtype='i4')
# load keys.csv to determine included AreaPerilID from peril_filter
if os.path.exists(os.path.join(input_path, 'keys.csv')):
keys_df = pd.read_csv(os.path.join(input_path, 'keys.csv'), dtype=Keys)
if peril_filter:
valid_areaperil_id = np.unique(keys_df.loc[keys_df['PerilID'].isin(peril_filter), 'AreaPerilID'])
logger.debug(
f'Peril specific run: ({peril_filter}), {len(valid_areaperil_id)} AreaPerilID included out of {len(keys_df)}')
else:
valid_areaperil_id = np.unique(keys_df['AreaPerilID'])
else:
valid_areaperil_id = None
logger.debug('import damage bins')
damage_bins = get_damage_bins(model_storage, ignore_file_type)
logger.debug('import coverages')
# coverages are numbered from 1, therefore we skip element 0 in `coverages`
coverages_tb = read_coverages(input_path, ignore_file_type)
coverages = np.zeros(coverages_tb.shape[0] + 1, coverage_type)
coverages[1:]['tiv'] = coverages_tb
# prepare for stochastic disaggregation
logger.debug('import aggregate vulnerability definitions and vulnerability weights')
aggregate_vulnerability = read_aggregate_vulnerability(model_storage, ignore_file_type)
aggregate_weights = read_vulnerability_weights(model_storage, ignore_file_type)
agg_vuln_to_vuln_ids = process_aggregate_vulnerability(aggregate_vulnerability)
if aggregate_vulnerability is not None and aggregate_weights is None:
raise FileNotFoundError(
f"Vulnerability weights file not found at {model_storage.get_storage_url('', print_safe=True)[1]}"
)
logger.debug('import items and correlations tables')
# since items and correlations have the same granularity (one row per item_id) we merge them on `item_id`.
correlations_tb = read_correlations(input_path, ignore_file_type)
items_tb = read_items(input_path, ignore_file_type)
if len(correlations_tb) != len(items_tb):
logger.info(
f"The items table has length {len(items_tb)} while the correlations table has length {len(correlations_tb)}.\n"
"It is possible that the correlations are not set up properly in the model settings file."
)
# merge the tables, using defaults for missing values, and sort the resulting table
items = rfn.join_by(
'item_id', items_tb, correlations_tb,
jointype='leftouter', usemask=False,
defaults={'peril_correlation_group': 0,
'damage_correlation_value': 0.,
'hazard_group_id': 0,
'hazard_correlation_value': 0.}
)
items = rfn.merge_arrays((items,
np.empty(items.shape,
dtype=nb.from_dtype(np.dtype([("vulnerability_idx", np.int32)])))),
flatten=True)
# intensity adjustment
if dynamic_footprint:
logger.debug('get dynamic footprint adjustments')
adjustments_tb = get_dynamic_footprint_adjustments(input_path)
items = rfn.join_by(
'item_id', items, adjustments_tb,
jointype='leftouter', usemask=False,
defaults={'intensity_adjustment': 0, 'return_period': 0}
)
# include peril_id
if dynamic_footprint:
logger.debug('get peril_id')
item_peril = get_peril_id(input_path)
items = rfn.join_by(
'item_id', items, item_peril,
jointype='leftouter', usemask=False,
defaults={'peril_id': 0}
)
# Pre-compute sequential indices for group_id and hazard_group_id
# to enable array-based lookups instead of Numba Dict lookups in reconstruct_coverages
unique_group_ids_arr, group_seq_ids = np.unique(items['group_id'], return_inverse=True)
unique_haz_group_ids_arr, haz_group_seq_ids = np.unique(items['hazard_group_id'], return_inverse=True)
n_unique_groups = len(unique_group_ids_arr)
n_unique_haz_groups = len(unique_haz_group_ids_arr)
items = rfn.merge_arrays((items,
np.empty(items.shape,
dtype=nb.from_dtype(np.dtype([("group_seq_id", np.int32),
("hazard_group_seq_id", np.int32)])))),
flatten=True)
items['group_seq_id'] = group_seq_ids
items['hazard_group_seq_id'] = haz_group_seq_ids
items.sort(order=['areaperil_id', 'vulnerability_id'])
# build item map
item_map, areaperil_ids_map, vuln_dict, agg_vuln_to_vuln_idxs, areaperil_vuln_idx_to_weight = generate_item_map(
items,
coverages,
valid_areaperil_id,
agg_vuln_to_vuln_ids)
if aggregate_weights is not None:
logger.debug('reconstruct aggregate vulnerability definitions and weights')
process_vulnerability_weights(areaperil_vuln_idx_to_weight, vuln_dict, aggregate_weights)
# import array to store the coverages to be computed
# coverages are numebered from 1, therefore skip element 0.
compute = np.zeros(coverages.shape[0] + 1, items_dtype['coverage_id'])
logger.debug('import peril correlation groups')
unique_peril_correlation_groups = np.unique(items['peril_correlation_group'])
Nperil_correlation_groups = unique_peril_correlation_groups.shape[0]
logger.info(f"Detected {Nperil_correlation_groups} peril correlation groups.")
logger.debug('import footprint')
footprint_obj = stack.enter_context(Footprint.load(model_storage, ignore_file_type,
df_engine=model_df_engine, areaperil_ids=list(areaperil_ids_map.keys())))
if data_server:
num_intensity_bins: int = FootprintLayerClient.get_number_of_intensity_bins()
logger.info(f"got {num_intensity_bins} intensity bins from server")
else:
num_intensity_bins: int = footprint_obj.num_intensity_bins
logger.debug('import vulnerabilities')
vuln_adj = get_vuln_rngadj(run_dir, vuln_dict)
vuln_array, _, _ = get_vulns(model_storage, run_dir, vuln_dict, num_intensity_bins, ignore_file_type, df_engine=model_df_engine)
Nvulnerability, Ndamage_bins_max, Nintensity_bins = vuln_array.shape
# set up streams
if file_out is None or file_out == '-':
stream_out = sys.stdout.buffer
else:
stream_out = stack.enter_context(open(file_out, 'wb'))
select_stream_list = [stream_out]
# prepare output stream
stream_out.write(gul_header)
stream_out.write(np.int32(sample_size).tobytes())
# set the random generator function
generate_rndm = get_random_generator(random_generator)
# create the array to store the seeds
haz_seeds = np.zeros(n_unique_haz_groups, dtype=correlations_dtype['hazard_group_id'])
vuln_seeds = np.zeros(n_unique_groups, dtype=items_dtype['group_id'])
# Pre-allocated arrays for group_id -> rng_index mapping (replaces per-event Numba Dicts)
group_seq_rng_index = np.empty(n_unique_groups, dtype=np.int64)
hazard_group_seq_rng_index = np.empty(n_unique_haz_groups, dtype=np.int64)
# haz correlation
if not ignore_haz_correlation and Nperil_correlation_groups > 0 and any(items['hazard_correlation_value'] > 0):
# there will be some hazard correlation
do_haz_correlation = True
haz_peril_correlation_groups = unique_peril_correlation_groups
haz_corr_seeds = np.zeros(np.max(haz_peril_correlation_groups) + 1, dtype='int64')
else:
do_haz_correlation = False
haz_peril_correlation_groups = unique_peril_correlation_groups[:0]
haz_corr_seeds = np.zeros(1, dtype='int64')
if ignore_haz_correlation:
logger.info(
"correlated random number generation for hazard intensity sampling: switched OFF because --ignore-haz-correlation is True.")
else:
logger.info("correlated random number generation for hazard intensity sampling: switched OFF because 0 peril correlation groups were detected or "
"the hazard correlation value is zero for all peril correlation groups.")
# damage correlation
if not ignore_correlation and Nperil_correlation_groups > 0 and any(items['damage_correlation_value'] > 0):
do_correlation = True
damage_peril_correlation_groups = unique_peril_correlation_groups
damage_corr_seeds = np.zeros(np.max(damage_peril_correlation_groups) + 1, dtype='int64')
else:
do_correlation = False
damage_peril_correlation_groups = unique_peril_correlation_groups[:0]
damage_corr_seeds = np.zeros(1, dtype='int64')
if ignore_correlation:
logger.info(
"correlated random number generation for damage sampling: switched OFF because --ignore-correlation is True.")
else:
logger.info("correlated random number generation for damage sampling: switched OFF because 0 peril correlation groups were detected or "
"the damage correlation value is zero for all peril correlation groups.")
if do_correlation or do_haz_correlation:
logger.info(f"correlated random number generation for hazard intensity sampling: switched {'ON' if do_haz_correlation else 'OFF'}.")
logger.info(f"Correlated random number generation for damage sampling: switched {'ON' if do_correlation else 'OFF'}.")
logger.info(f"Correlation values for {Nperil_correlation_groups} peril correlation groups have been imported.")
# pre-compute lookup tables for the Gaussian cdf and inverse cdf
# Notes:
# - the size `N` can be increased to achieve better resolution in the Gaussian cdf and inv cdf.
# - the function `get_corr_rval` to compute the correlated numbers is not affected by N.
norm_inv_parameters = np.array((x_min, x_max, norm_inv_N, cdf_min, cdf_max, inv_factor, norm_factor), dtype=NormInversionParameters)
norm_inv_cdf = compute_norm_inv_cdf_lookup(norm_inv_parameters['x_min'], norm_inv_parameters['x_max'], norm_inv_parameters['N'])
norm_cdf = compute_norm_cdf_lookup(norm_inv_parameters['cdf_min'], norm_inv_parameters['cdf_max'], norm_inv_parameters['N'])
else:
# create dummy data structures with proper dtypes to allow correct numba compilation
norm_inv_parameters = np.array((0., 0., 0, 0., 0., 0., 0.), dtype=NormInversionParameters)
norm_inv_cdf, norm_cdf = np.zeros(1, dtype='float64'), np.zeros(1, dtype='float64')
# buffer to be re-used to store all the correlated random values
vuln_z_unif = np.zeros(sample_size, dtype='float64')
haz_z_unif = np.zeros(sample_size, dtype='float64')
if effective_damageability is True:
logger.info("effective_damageability is True: gulmc will draw the damage samples from the effective damageability distribution.")
else:
logger.info("effective_damageability is False: gulmc will perform the full Monte Carlo sampling: "
"sample the hazard intensity first, then sample the damage from the corresponding vulnerability function.")
# create buffers to be reused when computing losses
byte_mv = np.empty(PIPE_CAPACITY * 2, dtype='b')
losses = np.zeros((sample_size + NUM_IDX + 1, np.max(coverages[1:]['max_items'])), dtype=oasis_float)
# maximum bytes to be written in the output stream for 1 item
max_bytes_per_item = gulSampleslevelHeader_size + (sample_size + NUM_IDX + 1) * gulSampleslevelRec_size
# define vulnerability cdf cache size
max_cached_vuln_cdf_size_bytes = max_cached_vuln_cdf_size_MB * 1024 * 1024 # cahce size in bytes
max_Nnumbers_cached_vuln_cdf = max_cached_vuln_cdf_size_bytes // oasis_float.itemsize # total numbers that can fit in the cache
max_Nvulnerability_cached_vuln_cdf = max_Nnumbers_cached_vuln_cdf // Ndamage_bins_max # max number of vulnerability funcions that can be stored in cache
# number of vulnerability functions to be cached
Nvulns_cached = min(Nvulnerability * Nintensity_bins, max_Nvulnerability_cached_vuln_cdf)
logger.info(f"max vulnerability cdf cache size is {max_cached_vuln_cdf_size_MB}MB")
logger.info(
f"generating a cache of shape ({Nvulns_cached}, {Ndamage_bins_max}) and size {Nvulns_cached * Ndamage_bins_max * oasis_float.itemsize / 1024 / 1024:8.3f}MB")
# maximum bytes to be written in the output stream for 1 item
event_footprint_obj = FootprintLayerClient if data_server else footprint_obj
if dynamic_footprint:
intensity_bin_dict = get_intensity_bin_dict(os.path.join(run_dir, 'static'))
else:
intensity_bin_dict = Dict.empty(nb_Tuple((nb_int32, nb_int32)), nb_int32)
dynamic_footprint = None
compute_info = np.zeros(1, dtype=gulmc_compute_info_type)[0]
compute_info['max_bytes_per_item'] = max_bytes_per_item
compute_info['Ndamage_bins_max'] = Ndamage_bins_max
compute_info['loss_threshold'] = loss_threshold
compute_info['alloc_rule'] = alloc_rule
compute_info['do_correlation'] = do_correlation
compute_info['do_haz_correlation'] = do_haz_correlation
compute_info['effective_damageability'] = effective_damageability
compute_info['debug'] = debug
# default random values array for sample_size==0 case
haz_rndms_base = np.empty((1, sample_size), dtype='float64')
vuln_rndms_base = np.empty((1, sample_size), dtype='float64')
haz_eps_ij = np.empty((1, sample_size), dtype='float64')
damage_eps_ij = np.empty((1, sample_size), dtype='float64')
# Pre-allocate CDF cache once (reused across events, no need to zero)
cached_vuln_cdfs = np.empty((Nvulns_cached, Ndamage_bins_max), dtype=oasis_float)
counter = 0
timer = time.time()
ping = kwargs.get('socket_server', 'False') != 'False'
while True:
if not streams_in.readinto(event_id_mv):
if ping:
oasis_ping({"events_complete": counter, "analysis_pk": kwargs.get("analysis_pk", None)})
break
# get the next event_id from the input stream
compute_info['event_id'] = event_ids[0]
event_footprint = event_footprint_obj.get_event(event_ids[0])
if event_footprint is not None:
areaperil_ids, Nhaz_arr_this_event, areaperil_to_haz_arr_i, haz_pdf, haz_arr_ptr = process_areaperils_in_footprint(
event_footprint,
areaperil_ids_map,
dynamic_footprint)
if Nhaz_arr_this_event == 0:
# no items to be computed for this event
counter += 1
continue
items_event_data, rng_index, hazard_rng_index, byte_mv = reconstruct_coverages(
compute_info,
areaperil_ids,
areaperil_ids_map,
areaperil_to_haz_arr_i,
item_map,
items,
coverages,
compute,
haz_seeds,
haz_peril_correlation_groups,
haz_corr_seeds,
vuln_seeds,
damage_peril_correlation_groups,
damage_corr_seeds,
dynamic_footprint,
byte_mv,
group_seq_rng_index,
hazard_group_seq_rng_index
)
# since these are never used outside of a sample > 0 branch we can remove the need to
# generate (and potentially allocate) the random values. As at 2.3.5 the sampling method
# for random values accounts for 25% of the runtime of the losses step not including
# the get_event despite having a sample size of 0.
if sample_size > 0:
# generation of "base" random values for hazard intensity and vulnerability sampling
haz_rndms_base = generate_rndm(haz_seeds[:hazard_rng_index], sample_size)
vuln_rndms_base = generate_rndm(vuln_seeds[:rng_index], sample_size)
haz_eps_ij = generate_rndm(haz_corr_seeds, sample_size, skip_seeds=1)
damage_eps_ij = generate_rndm(damage_corr_seeds, sample_size, skip_seeds=1)
# Reset CDF cache lookup per event (cached_vuln_cdfs array is reused, no reallocation)
cached_vuln_cdf_lookup, lookup_keys = gen_empty_vuln_cdf_lookup(Nvulns_cached, compute_info)
processing_done = False
while not processing_done:
try:
processing_done = compute_event_losses(
compute_info,
coverages,
compute,
items_event_data,
items,
sample_size,
haz_pdf,
haz_arr_ptr,
vuln_array,
damage_bins,
cached_vuln_cdf_lookup,
lookup_keys,
cached_vuln_cdfs,
agg_vuln_to_vuln_idxs,
areaperil_vuln_idx_to_weight,
losses,
haz_rndms_base,
vuln_rndms_base,
vuln_adj,
haz_eps_ij,
damage_eps_ij,
norm_inv_parameters,
norm_inv_cdf,
norm_cdf,
vuln_z_unif,
haz_z_unif,
byte_mv,
dynamic_footprint,
intensity_bin_dict
)
except Exception:
data = {
"event_id": event_ids[0]
}
with open("event_error.json", "w") as f:
json.dump(data, f, default=str)
logger.error(f"event id={event_ids[0]} failed in summary")
raise
# write the losses to the output stream
write_start = 0
while write_start < compute_info['cursor']:
select([], select_stream_list, select_stream_list)
write_start += stream_out.write(memoryview(byte_mv[write_start: compute_info['cursor']]))
logger.info(f"event {event_ids[0]} DONE")
counter += 1
if ping and time.time() - timer > SERVER_UPDATE_TIME:
timer = time.time()
oasis_ping({"events_complete": counter, "analysis_pk": kwargs.get("analysis_pk", None)})
counter = 0
return 0
@nb.njit(cache=True, fastmath=True)
[docs]
def get_haz_cdf(item_event_data, haz_cdf, haz_cdf_ptr, dynamic_footprint, intensity_adjustment, intensity_bin_dict):
# get the right hazard cdf from the array containing all hazard cdfs
hazcdf_i = item_event_data['hazcdf_i']
haz_cdf_record = haz_cdf[haz_cdf_ptr[hazcdf_i]:haz_cdf_ptr[hazcdf_i + 1]]
haz_cdf_prob = haz_cdf_record['probability']
if dynamic_footprint:
# adjust intensity in dynamic footprint
haz_cdf_intensity = haz_cdf_record['intensity']
haz_cdf_intensity = haz_cdf_intensity - intensity_adjustment
haz_cdf_intensity = np.where(haz_cdf_intensity < 0, nb_int32(0), haz_cdf_intensity)
haz_cdf_bin_id = np.zeros_like(haz_cdf_record['intensity_bin_id'])
for haz_bin_idx in range(haz_cdf_bin_id.shape[0]):
if haz_cdf_intensity[haz_bin_idx] <= 0:
haz_cdf_bin_id[haz_bin_idx] = intensity_bin_dict[0]
else:
haz_cdf_bin_id[haz_bin_idx] = intensity_bin_dict[haz_cdf_intensity[haz_bin_idx]]
else:
haz_cdf_bin_id = haz_cdf_record['intensity_bin_id']
return haz_cdf_prob, haz_cdf_bin_id
@nb.njit(fastmath=True)
[docs]
def get_last_non_empty(cdf, bin_i):
"""
remove empty bucket from the end
Args:
cdf: cumulative distribution
bin_i: last valid bin index
Returns:
last bin index with an increased in the cdf
"""
last_prob = cdf[bin_i]
while bin_i > 0 and cdf[bin_i - 1] == last_prob:
bin_i -= 1
return bin_i
@nb.njit(fastmath=True)
[docs]
def pdf_to_cdf(pdf, empty_cdf):
"""
return the cumulative distribution from the probality distribution
Args:
pdf (np.array[float]): probality distribution
empty_cdf (np.array[float]): cumulative distribution buffer for output
Returns:
cdf (np.array[float]): here we return only the valid part if needed
"""
cumsum = 0
i = 0
while i < pdf.shape[0]:
cumsum += pdf[i]
empty_cdf[i] = cumsum
i += 1
if cumsum >= 0.999999940:
break
i = get_last_non_empty(empty_cdf, i - 1)
return empty_cdf[: i + 1]
@nb.njit(fastmath=True)
[docs]
def calc_eff_damage_cdf(vuln_pdf, haz_pdf, eff_damage_cdf_empty):
"""
calculate the covoluted cumulative distribution between vulnerability damage and hazard probability distribution
Args:
vuln_pdf (np.array[float]) : vulnerability damage probability distribution
haz_pdf (np.array[float]): hazard probability distribution
eff_damage_cdf_empty (np.array[float]): output buffer
Returns:
eff_damage_cdf (np.array[float]): cdf is stored in eff_damage_cdf_empty, here we return only the valid part if needed
"""
eff_damage_cdf_cumsum = 0.
damage_bin_i = 0
while damage_bin_i < vuln_pdf.shape[1]:
for haz_i in range(vuln_pdf.shape[0]):
eff_damage_cdf_cumsum += vuln_pdf[haz_i, damage_bin_i] * haz_pdf[haz_i]
eff_damage_cdf_empty[damage_bin_i] = eff_damage_cdf_cumsum
damage_bin_i += 1
if eff_damage_cdf_cumsum >= 0.999999940:
break
damage_bin_i = get_last_non_empty(eff_damage_cdf_empty, damage_bin_i - 1)
return eff_damage_cdf_empty[:damage_bin_i + 1]
@nb.njit()
[docs]
def cache_cdf(next_cached_vuln_cdf_i, cached_vuln_cdfs, cached_vuln_cdf_lookup, cached_vuln_cdf_lookup_keys, cdf, cdf_key):
"""Store a cdf in the circular cache, evicting the oldest entry if the cache is full.
Uses a circular buffer strategy: `next_cached_vuln_cdf_i` is the write cursor that
wraps around when it reaches the end of the cache. When a slot is reused, the previous
key occupying that slot is removed from the lookup Dict.
Args:
next_cached_vuln_cdf_i (int): current write cursor position in the circular buffer.
cached_vuln_cdfs (np.array[oasis_float]): 2d cache array of shape (Nvulns_cached, Ndamage_bins_max).
Pre-allocated once and reused across events.
cached_vuln_cdf_lookup (Dict[int64, Tuple(int32, int32)]): maps composite int64 key
to (slot_index, cdf_length).
cached_vuln_cdf_lookup_keys (List[int64]): reverse mapping from slot to key, used to
remove evicted entries from the Dict.
cdf (np.array[oasis_float]): the cdf values to cache.
cdf_key (int64): composite cache key (eff_cdf_id << 32 | discriminator).
Returns:
int: updated write cursor position.
"""
if cdf_key not in cached_vuln_cdf_lookup: # already cached
# cache the cdf
if cached_vuln_cdf_lookup_keys[next_cached_vuln_cdf_i] in cached_vuln_cdf_lookup:
# overwrite cache
cached_vuln_cdf_lookup.pop(cached_vuln_cdf_lookup_keys[next_cached_vuln_cdf_i])
cached_vuln_cdfs[next_cached_vuln_cdf_i, :cdf.shape[0]] = cdf
cached_vuln_cdf_lookup[cdf_key] = tuple((nb_int32(next_cached_vuln_cdf_i), nb_int32(cdf.shape[0])))
cached_vuln_cdf_lookup_keys[next_cached_vuln_cdf_i] = cdf_key
next_cached_vuln_cdf_i += 1
next_cached_vuln_cdf_i %= cached_vuln_cdfs.shape[0]
return next_cached_vuln_cdf_i
@nb.njit(fastmath=True, cache=True)
[docs]
def get_gul_from_vuln_cdf(vuln_rval, vuln_cdf, Ndamage_bins, damage_bins, bin_scaling):
# find the damage cdf bin in which the random value `vuln_rval` falls into
vuln_bin_idx = binary_search(vuln_rval, vuln_cdf, Ndamage_bins - 1)
# compute ground-up losses
return get_gul(
damage_bins['bin_from'][vuln_bin_idx],
damage_bins['bin_to'][vuln_bin_idx],
damage_bins['interpolation'][vuln_bin_idx],
vuln_cdf[vuln_bin_idx - 1] * (vuln_bin_idx > 0),
vuln_cdf[vuln_bin_idx],
vuln_rval,
bin_scaling,
)
@nb.njit(cache=True, fastmath=True)
[docs]
def compute_event_losses(compute_info,
coverages,
coverage_ids,
items_event_data,
items,
sample_size,
haz_pdf,
haz_arr_ptr,
vuln_array,
damage_bins,
cached_vuln_cdf_lookup,
cached_vuln_cdf_lookup_keys,
cached_vuln_cdfs,
agg_vuln_to_vuln_idxs,
areaperil_vuln_idx_to_weight,
losses,
haz_rndms_base,
vuln_rndms_base,
vuln_adj,
haz_eps_ij,
damage_eps_ij,
norm_inv_parameters,
norm_inv_cdf,
norm_cdf,
vuln_z_unif,
haz_z_unif,
byte_mv,
dynamic_footprint,
intensity_bin_dict
):
"""Compute ground-up losses for all coverages in a single event.
Iterates over coverages and their items, looking up or computing the vulnerability cdf
for each item, then sampling losses using the pre-generated random numbers. Results are
written into a byte buffer for streaming output.
For each item, the function:
1. Retrieves the hazard intensity pdf for the item's areaperil (via haz_arr_i).
2. Looks up or computes the effective damage cdf (combining vulnerability and hazard).
When effective_damageability is False, also caches per-intensity-bin vulnerability cdfs.
3. Computes mean loss, standard deviation, chance of loss, and max loss.
4. For each random sample, draws the ground-up loss from the cdf.
5. Optionally applies hazard and damage correlation.
6. Writes results to the output byte buffer.
CDF caching uses composite int64 keys built from `eff_cdf_id` (assigned per unique
(areaperil, vulnerability) group in reconstruct_coverages). The upper 32 bits encode the
eff_cdf_id, the lower 32 bits encode a discriminator (0xFFFFFFFF for effective damage cdfs,
or the intensity_bin_id for per-bin vulnerability cdfs). Circular eviction is used when
the cache is full.
If the output buffer cannot fit the next coverage, returns False so the caller can flush
the buffer and call again to continue processing.
Args:
compute_info (gulmc_compute_info_type): computation state (event_id, cursor position,
coverage range, cache pointer, thresholds, flags).
coverages (numpy.array[coverage_type]): coverage data indexed by coverage_id.
coverage_ids (numpy.array[int]): ordered list of coverage_ids to process in this event.
items_event_data (numpy.array[items_MC_data_type]): per-item event data populated by
reconstruct_coverages, containing item_idx, haz_arr_i, rng_index, hazard_rng_index,
and eff_cdf_id.
items (np.ndarray): items table merged with correlation parameters.
sample_size (int): number of random samples to draw.
haz_pdf (np.array[haz_arr_type]): hazard intensity pdf records for this event.
haz_arr_ptr (List[int]): indices where each areaperil's hazard records start in haz_pdf.
vuln_array (np.array[float]): 3d vulnerability array of shape
(Nvulnerability, Ndamage_bins_max, Nintensity_bins).
damage_bins (np.array): damage bin dictionary with bin_from, bin_to, interpolation, damage_type.
cached_vuln_cdf_lookup (Dict[int64, Tuple(int32, int32)]): cdf cache lookup mapping
composite int64 key to (slot_index, cdf_length).
cached_vuln_cdf_lookup_keys (List[int64]): reverse mapping from cache slot to key,
for circular eviction.
cached_vuln_cdfs (np.array[oasis_float]): 2d cdf cache of shape (Nvulns_cached, Ndamage_bins_max).
agg_vuln_to_vuln_idxs (Dict[int, List[int]]): map from aggregate vulnerability_id to
the list of individual vulnerability indices in vuln_array.
areaperil_vuln_idx_to_weight (Dict[Tuple, float]): map from (areaperil_id, vuln_idx) to
the weight for aggregate vulnerability composition.
losses (numpy.array[oasis_float]): reusable 2d buffer of shape
(sample_size + NUM_IDX + 1, max_items_per_coverage) for loss values.
haz_rndms_base (numpy.array[float64]): 2d array of shape (n_seeds, sample_size) with
base random values for hazard intensity sampling.
vuln_rndms_base (numpy.array[float64]): 2d array of shape (n_seeds, sample_size) with
base random values for damage sampling.
vuln_adj (np.array[float]): per-vulnerability adjustment factors applied to random samples
for non-aggregate vulnerabilities.
haz_eps_ij (np.array[float]): correlated random values for hazard sampling.
damage_eps_ij (np.array[float]): correlated random values for damage sampling.
norm_inv_parameters (NormInversionParameters): parameters for Gaussian inversion
(x_min, x_max, N, cdf_min, cdf_max, inv_factor, norm_factor).
norm_inv_cdf (np.array[float]): inverse Gaussian cdf lookup table.
norm_cdf (np.array[float]): Gaussian cdf lookup table.
vuln_z_unif (np.array[float]): reusable buffer for correlated vulnerability random values.
haz_z_unif (np.array[float]): reusable buffer for correlated hazard random values.
byte_mv (numpy.array[byte]): output byte buffer for the binary stream.
dynamic_footprint (None or object): None if no dynamic footprint, otherwise truthy.
intensity_bin_dict (Dict[Tuple(int32, int32), int32]): map from (peril_id, intensity)
to intensity_bin_id, used for dynamic footprint intensity adjustment.
Returns:
bool: True if all coverages have been processed, False if the buffer is full and
the caller should flush and call again.
"""
haz_cdf_empty = np.empty(vuln_array.shape[2], dtype=oasis_float)
vuln_pdf_empty = np.empty((vuln_array.shape[2], compute_info['Ndamage_bins_max']), dtype=vuln_array.dtype)
eff_damage_cdf_empty = np.empty(compute_info['Ndamage_bins_max'], dtype=oasis_float)
haz_i_to_Ndamage_bins_empty = np.empty(vuln_array.shape[2], dtype=oasis_int)
haz_i_to_vuln_cdf_empty = np.empty((vuln_array.shape[2], compute_info['Ndamage_bins_max']), dtype=vuln_array.dtype)
# we process at least one full coverage at a time, so when we write to stream, we write the whole buffer
compute_info['cursor'] = 0
# loop through all the coverages that remain to be computed
for coverage_i in range(compute_info['coverage_i'], compute_info['coverage_n']):
coverage = coverages[coverage_ids[coverage_i]]
tiv = coverage['tiv']
Nitems = coverage['cur_items']
exposureValue = tiv / Nitems
# estimate max number of bytes needed to output this coverage
# conservatively assume all random samples are printed (losses>loss_threshold)
est_cursor_bytes = Nitems * compute_info['max_bytes_per_item']
# return before processing this coverage if the number of free bytes left in the buffer
# is not sufficient to write out the full coverage
if compute_info['cursor'] + est_cursor_bytes > byte_mv.shape[0]:
return False
# compute losses for each item
for item_j in range(Nitems):
item_event_data = items_event_data[coverage['start_items'] + item_j]
rng_index = item_event_data['rng_index']
hazard_rng_index = item_event_data['hazard_rng_index']
item = items[item_event_data['item_idx']]
if dynamic_footprint is not None:
intensity_adjustment = item['intensity_adjustment']
else:
intensity_adjustment = nb_oasis_int(0)
haz_arr_i = item_event_data['haz_arr_i']
haz_pdf_record = haz_pdf[haz_arr_ptr[haz_arr_i]:haz_arr_ptr[haz_arr_i + 1]]
# we calculate this adjusted hazard pdf
# get the right hazard pdf from the array containing all hazard cdfs
if dynamic_footprint is not None:
# adjust intensity in dynamic footprint
haz_intensity = haz_pdf_record['intensity']
haz_intensity = haz_intensity - intensity_adjustment
haz_bin_id = np.zeros_like(haz_pdf_record['intensity_bin_id'])
peril_id = item['peril_id']
for haz_bin_idx in range(haz_bin_id.shape[0]):
try:
haz_bin_id[haz_bin_idx] = intensity_bin_dict[peril_id, haz_intensity[haz_bin_idx]]
except Exception:
haz_bin_id[haz_bin_idx] = intensity_bin_dict[peril_id, 0]
else:
haz_bin_id = haz_pdf_record['intensity_bin_id']
haz_pdf_prob = haz_pdf_record['probability']
# Build int64 composite keys from eff_cdf_id
eff_cdf_id_shifted = nb_int64(item_event_data['eff_cdf_id']) << nb_int64(32)
eff_damage_cdf_key = eff_cdf_id_shifted | EFF_CDF_KEY_SENTINEL
# determine if all the needed cdf are cached
do_calc_vuln_ptf = eff_damage_cdf_key not in cached_vuln_cdf_lookup
haz_cdf_prob = pdf_to_cdf(haz_pdf_prob, haz_cdf_empty)
Nhaz_bins = haz_cdf_prob.shape[0]
if not compute_info['effective_damageability']:
for haz_i in range(Nhaz_bins):
haz_lookup_key = eff_cdf_id_shifted | nb_int64(haz_bin_id[haz_i])
do_calc_vuln_ptf = do_calc_vuln_ptf or (haz_lookup_key not in cached_vuln_cdf_lookup)
if do_calc_vuln_ptf: # some cdf are not cached
# we get the vuln_pdf, needed for effcdf and each cdf
vuln_pdf = vuln_pdf_empty[:Nhaz_bins]
vuln_pdf[:] = 0
if item['vulnerability_id'] in agg_vuln_to_vuln_idxs: # we calculate the custom vuln_array for this aggregate
tot_weights = 0.
agg_vulns_idx = agg_vuln_to_vuln_idxs[item['vulnerability_id']]
for j, vuln_i in enumerate(agg_vulns_idx):
if (item['areaperil_id'], vuln_i) in areaperil_vuln_idx_to_weight:
weight = np.float64(areaperil_vuln_idx_to_weight[(item['areaperil_id'], vuln_i)])
if weight > 0:
tot_weights += weight
for haz_i in range(Nhaz_bins):
has_prob = False
for damage_bin_i in range(compute_info['Ndamage_bins_max']):
if vuln_array[vuln_i, damage_bin_i, haz_bin_id[haz_i] - 1] > 0:
has_prob = True
vuln_pdf[haz_i, damage_bin_i] += vuln_array[vuln_i, damage_bin_i, haz_bin_id[
haz_i] - 1] * weight
if not has_prob:
# the pdf is all zeros, i.e. probability of no loss is 100%
# store it as 100% * weight in the first damage bin
vuln_pdf[haz_i, 0] += weight
if tot_weights > 0:
vuln_pdf /= tot_weights
else:
for j, vuln_i in enumerate(agg_vulns_idx):
for haz_i in range(Nhaz_bins):
vuln_pdf[haz_i] += vuln_array[vuln_i, :, haz_bin_id[haz_i] - 1]
vuln_pdf /= len(agg_vulns_idx)
else:
for haz_i in range(Nhaz_bins):
vuln_pdf[haz_i] = vuln_array[item['vulnerability_idx'], :, haz_bin_id[haz_i] - 1]
# calculate and cache all cdf
eff_damage_cdf = calc_eff_damage_cdf(vuln_pdf, haz_pdf_prob, eff_damage_cdf_empty)
compute_info['next_cached_vuln_cdf_i'] = cache_cdf(
compute_info['next_cached_vuln_cdf_i'], cached_vuln_cdfs, cached_vuln_cdf_lookup,
cached_vuln_cdf_lookup_keys, eff_damage_cdf, eff_damage_cdf_key)
if not compute_info['effective_damageability']: # we cache all the vuln_cdf needed
haz_i_to_Ndamage_bins = haz_i_to_Ndamage_bins_empty[:Nhaz_bins]
haz_i_to_vuln_cdf = haz_i_to_vuln_cdf_empty[:Nhaz_bins]
for haz_i in range(Nhaz_bins):
haz_i_to_Ndamage_bins[haz_i] = pdf_to_cdf(vuln_pdf[haz_i], haz_i_to_vuln_cdf[haz_i]).shape[0]
lookup_key = eff_cdf_id_shifted | nb_int64(haz_bin_id[haz_i])
compute_info['next_cached_vuln_cdf_i'] = cache_cdf(
compute_info['next_cached_vuln_cdf_i'], cached_vuln_cdfs, cached_vuln_cdf_lookup,
cached_vuln_cdf_lookup_keys, haz_i_to_vuln_cdf[haz_i][:haz_i_to_Ndamage_bins[haz_i]],
lookup_key)
else: # cdf are cached
start, Ndamage_bins = cached_vuln_cdf_lookup[eff_damage_cdf_key]
eff_damage_cdf = cached_vuln_cdfs[start, :Ndamage_bins]
if not compute_info['effective_damageability']:
haz_i_to_Ndamage_bins = haz_i_to_Ndamage_bins_empty[:Nhaz_bins]
haz_i_to_vuln_cdf = haz_i_to_vuln_cdf_empty[:Nhaz_bins]
for haz_i in range(Nhaz_bins):
lookup_key = eff_cdf_id_shifted | nb_int64(haz_bin_id[haz_i])
start, Ndamage_bins = cached_vuln_cdf_lookup[lookup_key]
haz_i_to_Ndamage_bins[haz_i] = Ndamage_bins
haz_i_to_vuln_cdf[haz_i][:Ndamage_bins] = cached_vuln_cdfs[start, :Ndamage_bins]
Neff_damage_bins = eff_damage_cdf.shape[0]
damage_type = damage_bins[Neff_damage_bins - 1]['damage_type']
if damage_type == DAMAGE_TYPE_RELATIVE:
damage_bin_scaling = tiv
elif damage_type == DAMAGE_TYPE_ABSOLUTE:
damage_bin_scaling = 1
elif damage_type == DAMAGE_TYPE_DURATION:
# convert annual tiv to daily
damage_bin_scaling = tiv / 365
else: # default behaviour
# for relative vulnerability functions, gul are fraction of the tiv
# for absolute vulnerability functions, gul are absolute values
damage_bin_scaling = tiv if damage_bins[Neff_damage_bins - 1]['bin_to'] <= 1 else 1.0
# compute mean loss values
gul_mean, std_dev, chance_of_loss, max_loss = compute_mean_loss(
damage_bin_scaling,
eff_damage_cdf,
damage_bins['interpolation'],
Neff_damage_bins,
damage_bins[Neff_damage_bins - 1]['bin_to'],
)
losses[MAX_LOSS_IDX, item_j] = max_loss
losses[CHANCE_OF_LOSS_IDX, item_j] = chance_of_loss
losses[TIV_IDX, item_j] = exposureValue
losses[STD_DEV_IDX, item_j] = std_dev
losses[MEAN_IDX, item_j] = gul_mean
if sample_size > 0: # compute random losses
if compute_info['do_haz_correlation'] and item['hazard_correlation_value'] > 0:
# use correlation definitions to draw correlated random values into haz_z_unif
get_corr_rval_float(
haz_eps_ij[item['peril_correlation_group']], haz_rndms_base[hazard_rng_index], item['hazard_correlation_value'],
norm_inv_parameters['x_min'], norm_inv_cdf, norm_inv_parameters['inv_factor'],
norm_inv_parameters['cdf_min'], norm_cdf, norm_inv_parameters['norm_factor'],
sample_size, haz_z_unif
)
else:
haz_z_unif[:] = haz_rndms_base[hazard_rng_index]
if compute_info['do_correlation'] and item['damage_correlation_value'] > 0:
# use correlation definitions to draw correlated random values into vuln_z_unif
get_corr_rval_float(
damage_eps_ij[item['peril_correlation_group']], vuln_rndms_base[rng_index], item['damage_correlation_value'],
norm_inv_parameters['x_min'], norm_inv_cdf, norm_inv_parameters['inv_factor'],
norm_inv_parameters['cdf_min'], norm_cdf, norm_inv_parameters['norm_factor'],
sample_size, vuln_z_unif
)
else:
# do not use correlation
vuln_z_unif[:] = vuln_rndms_base[rng_index]
if item['vulnerability_id'] not in agg_vuln_to_vuln_idxs: # single vuln id (non-aggregate)
vuln_z_unif *= vuln_adj[item['vulnerability_idx']]
if compute_info['debug'] == 1: # store the random value used for the hazard sampling instead of the loss
losses[1:, item_j] = haz_z_unif[:]
elif compute_info['debug'] == 2: # store the random value used for the damage sampling instead of the loss
losses[1:, item_j] = vuln_z_unif[:]
else: # calculate gul
if compute_info['effective_damageability']:
for sample_idx in range(1, sample_size + 1):
losses[sample_idx, item_j] = get_gul_from_vuln_cdf(vuln_z_unif[sample_idx - 1], eff_damage_cdf,
Neff_damage_bins, damage_bins, damage_bin_scaling)
elif Nhaz_bins == 1: # only one hazard possible
Ndamage_bins = haz_i_to_Ndamage_bins[0]
vuln_cdf = haz_i_to_vuln_cdf[0][:Ndamage_bins]
for sample_idx in range(1, sample_size + 1):
losses[sample_idx, item_j] = get_gul_from_vuln_cdf(vuln_z_unif[sample_idx - 1], vuln_cdf,
Ndamage_bins, damage_bins, damage_bin_scaling)
else:
for sample_idx in range(1, sample_size + 1):
# find the hazard intensity cdf bin in which the random value `haz_z_unif[sample_idx - 1]` falls into
# we don't need to use last haz_cdf_prob value because if for rounding reason haz_rval
# is bigger, we want the index Nhaz_bins-1 anyway. if we were using Nhaz_bins,
# bigger than haz_cdf_prob[-1] haz_rval would have index Nhaz_bins, outside haz_i_to_Ndamage_bins
haz_bin_idx = binary_search(haz_z_unif[sample_idx - 1], haz_cdf_prob, Nhaz_bins - 1)
# get the individual vulnerability cdf
Ndamage_bins = haz_i_to_Ndamage_bins[haz_bin_idx]
vuln_cdf = haz_i_to_vuln_cdf[haz_bin_idx][:Ndamage_bins]
losses[sample_idx, item_j] = get_gul_from_vuln_cdf(vuln_z_unif[sample_idx - 1], vuln_cdf,
Ndamage_bins, damage_bins, damage_bin_scaling)
# write the losses to the output memoryview
compute_info['cursor'] = write_losses(
compute_info['event_id'],
sample_size,
compute_info['loss_threshold'],
losses[:, :Nitems],
items_event_data[coverage['start_items']: coverage['start_items'] + Nitems]['item_id'],
compute_info['alloc_rule'],
tiv,
byte_mv,
compute_info['cursor'])
# register that another `coverage_id` has been processed
compute_info['coverage_i'] += 1
return True
@nb.njit(cache=True, fastmath=True)
@nb.njit(cache=True, fastmath=True)
[docs]
def reconstruct_coverages(compute_info,
areaperil_ids,
areaperil_ids_map,
areaperil_to_haz_arr_i,
item_map,
items,
coverages,
compute,
haz_seeds,
haz_peril_correlation_groups,
haz_corr_seeds,
vuln_seeds,
damage_peril_correlation_groups,
damage_corr_seeds,
dynamic_footprint,
byte_mv,
group_seq_rng_index,
hazard_group_seq_rng_index):
"""Register each item to its coverage and prepare per-item event data for loss computation.
For each (areaperil_id, vulnerability_id) pair present in the event footprint, iterates
over all mapped items and:
1. Computes deterministic hash-based random seeds for hazard and damage sampling,
using group_id and hazard_group_id respectively. Seeds are deduplicated via
pre-allocated arrays indexed by sequential group ids.
2. Maps each item to its coverage structure, tracking the start offset and count.
3. Stores per-item event data (haz_arr_i, rng_index, hazard_rng_index, eff_cdf_id)
in the items_event_data array.
4. Assigns a sequential eff_cdf_id to each unique CDF group. For non-dynamic footprints,
each (areaperil_id, vulnerability_id) pair gets one eff_cdf_id. For dynamic footprints,
groups are further subdivided by intensity_adjustment since different adjustments
produce different CDFs. The eff_cdf_id is later used in compute_event_losses to build
composite int64 cache keys.
Args:
compute_info (gulmc_compute_info_type): computation state; coverage_i, coverage_n,
and event_id fields are read/written.
areaperil_ids (List[int]): areaperil_ids present in the event footprint (from
process_areaperils_in_footprint).
areaperil_ids_map (Dict[int, Dict[int, int]]): mapping from areaperil_id to the set
of vulnerability_ids associated with it.
areaperil_to_haz_arr_i (Dict[int, int]): mapping from areaperil_id to its sequential
index in haz_arr_ptr (assigned per event in process_areaperils_in_footprint).
item_map (Dict[Tuple(areaperil_int, int32), List[int64]]): mapping from
(areaperil_id, vulnerability_id) to the list of item indices in the items array.
items (np.ndarray): items table merged with correlation parameters, containing
group_id, hazard_group_id, coverage_id, group_seq_id, hazard_group_seq_id, etc.
coverages (numpy.array[coverage_type]): coverage data indexed by coverage_id.
compute (numpy.array[int]): output buffer for the list of coverage_ids to be computed.
haz_seeds (numpy.array[int]): output buffer for hazard intensity random seeds.
haz_peril_correlation_groups (numpy.array[int]): unique peril correlation groups for hazard.
haz_corr_seeds (numpy.array[int]): output buffer for hazard correlation seeds.
vuln_seeds (numpy.array[int]): output buffer for damage random seeds.
damage_peril_correlation_groups (numpy.array[int]): unique peril correlation groups for damage.
damage_corr_seeds (numpy.array[int]): output buffer for damage correlation seeds.
dynamic_footprint (None or object): None if no dynamic footprint, otherwise truthy.
byte_mv (numpy.array[byte]): output byte buffer, may be resized if needed.
group_seq_rng_index (numpy.array[int64]): pre-allocated array of size n_unique_groups,
used for O(1) group_id to rng_index mapping (reset to -1 each event).
hazard_group_seq_rng_index (numpy.array[int64]): pre-allocated array of size
n_unique_haz_groups, for hazard_group_id to rng_index mapping.
Returns:
tuple: (items_event_data, rng_index, hazard_rng_index, byte_mv)
- items_event_data (numpy.array[items_MC_data_type]): per-item data including
item_idx, haz_arr_i, rng_index, hazard_rng_index, eff_cdf_id.
- rng_index (int): number of unique damage random seeds generated.
- hazard_rng_index (int): number of unique hazard random seeds generated.
- byte_mv (numpy.array[byte]): output buffer, possibly resized.
"""
# init data structures
# Reset pre-allocated arrays instead of creating new Numba Dicts
group_seq_rng_index[:] = -1
hazard_group_seq_rng_index[:] = -1
rng_index = 0
hazard_rng_index = 0
compute_i = 0
items_data_i = 0
# Sequential counter for unique CDF groups
eff_cdf_id = nb_oasis_int(0)
coverages['cur_items'].fill(0)
items_event_data = np.empty(2 ** NP_BASE_ARRAY_SIZE, dtype=items_MC_data_type)
# for each areaperil_id, loop over all vulnerability functions used in that areaperil_id and,
# for each item:
# - compute the seeds for the hazard intensity sampling and for the damage sampling
# - store data for later processing (hazard cdf index, etc.)
for areaperil_id in areaperil_ids:
for vuln_id in areaperil_ids_map[areaperil_id]:
# register the items to their coverage
item_key = tuple((areaperil_id, vuln_id))
# Each (areaperil_id, vuln_id) pair gets a unique eff_cdf_id
# For dynamic footprints, items within the same pair may have different
# intensity_adjustment, so we sub-divide using a local Dict.
this_group_eff_cdf_id = eff_cdf_id
if dynamic_footprint is not None:
adj_to_cdf_id = Dict.empty(nb_oasis_int, nb_oasis_int)
for item_idx in item_map[item_key]:
# if this group_id was not seen yet, process it.
# it assumes that hash only depends on event_id and group_id
# and that only 1 event_id is processed at a time.
# Use sequential index for array-based lookup instead of Dict
group_seq_id = items[item_idx]['group_seq_id']
if group_seq_rng_index[group_seq_id] == -1:
group_seq_rng_index[group_seq_id] = rng_index
vuln_seeds[rng_index] = generate_hash(items[item_idx]['group_id'], compute_info['event_id'])
this_rng_index = rng_index
rng_index += 1
else:
this_rng_index = group_seq_rng_index[group_seq_id]
hazard_group_seq_id = items[item_idx]['hazard_group_seq_id']
if hazard_group_seq_rng_index[hazard_group_seq_id] == -1:
hazard_group_seq_rng_index[hazard_group_seq_id] = hazard_rng_index
haz_seeds[hazard_rng_index] = generate_hash_hazard(items[item_idx]['hazard_group_id'], compute_info['event_id'])
this_hazard_rng_index = hazard_rng_index
hazard_rng_index += 1
else:
this_hazard_rng_index = hazard_group_seq_rng_index[hazard_group_seq_id]
coverage_id = items[item_idx]['coverage_id']
coverage = coverages[coverage_id]
if coverage['cur_items'] == 0:
# no items were collected for this coverage yet: set up the structure
compute[compute_i], compute_i = coverage_id, compute_i + 1
while items_event_data.shape[0] < items_data_i + coverage['max_items']:
# if items_data needs to be larger to store all the items, double it in size
temp_items_data = np.empty(items_event_data.shape[0] * 2, dtype=items_event_data.dtype)
temp_items_data[:items_data_i] = items_event_data[:items_data_i]
items_event_data = temp_items_data
coverage['start_items'], items_data_i = items_data_i, items_data_i + coverage['max_items']
# append the data of this item
item_i = coverage['start_items'] + coverage['cur_items']
items_event_data[item_i]['item_idx'] = item_idx
items_event_data[item_i]['item_id'] = items[item_idx]['item_id']
items_event_data[item_i]['haz_arr_i'] = areaperil_to_haz_arr_i[areaperil_id]
items_event_data[item_i]['rng_index'] = this_rng_index
items_event_data[item_i]['hazard_rng_index'] = this_hazard_rng_index
if dynamic_footprint is not None:
items_event_data[item_i]['intensity_adjustment'] = items[item_idx]['intensity_adjustment']
items_event_data[item_i]['return_period'] = items[item_idx]['return_period']
# For dynamic footprints, sub-divide by intensity_adjustment
adj = items[item_idx]['intensity_adjustment']
if adj not in adj_to_cdf_id:
adj_to_cdf_id[adj] = eff_cdf_id
eff_cdf_id += 1
items_event_data[item_i]['eff_cdf_id'] = adj_to_cdf_id[adj]
else:
items_event_data[item_i]['eff_cdf_id'] = this_group_eff_cdf_id
coverage['cur_items'] += 1
# Increment eff_cdf_id for non-dynamic footprints (one id per (areaperil, vuln) pair)
if dynamic_footprint is None:
eff_cdf_id += 1
compute_info['coverage_i'] = 0
compute_info['coverage_n'] = compute_i
byte_mv = adjust_byte_mv_size(byte_mv, np.max(coverages['cur_items']) * compute_info['max_bytes_per_item'])
generate_correlated_hash_vector(haz_peril_correlation_groups, compute_info['event_id'], haz_corr_seeds)
generate_correlated_hash_vector(damage_peril_correlation_groups, compute_info['event_id'], damage_corr_seeds)
return (items_event_data,
rng_index,
hazard_rng_index,
byte_mv)
if __name__ == '__main__':
[docs]
kwargs = {
'alloc_rule': 1,
'debug': 0,
'file_in': './static/events_p.bin',
'file_out': '/dev/null',
'loss_threshold': 0.0,
'sample_size': 10,
'effective_damageability': False,
'ignore_correlation': False,
'ignore_haz_correlation': False,
'ignore_file_type': set(),
'data_server': False,
'max_cached_vuln_cdf_size_MB': 200,
'peril_filter': None,
'random_generator': 1,
'run_dir': '.',
'model_df_engine': 'oasis_data_manager.df_reader.reader.OasisPandasReader',
'dynamic_footprint': False}
run(**kwargs)