from hazards.flood import generate_flood_raster

class HazardSource:
    def __init__(self, name, rp_options, generate_fn, data_layers):
        self.name = name
        self.rp_options = rp_options
        self.generate_fn = generate_fn
        self.data_layers = data_layers

hazard_registry = {
    "Flood": HazardSource(
        name="Flood",
        rp_options=[10, 20, 50, 75, 100, 200, 500],
        generate_fn=generate_flood_raster,
        data_layers=["populated_places", "roads", "buildings"]
    ),
    # Future: "Drought": HazardSource(...)
}