Add Ground Zero water source

Scripts/setup_ground_zero_demo_map.py

@@ -179,6 +179,17 @@ BIOME_RESOURCE_ACTORS = [
 ]
 
 
+WATER_SOURCE_ACTORS = [
+    {
+        "label": "AGR_GZ_FreshWaterSource_01",
+        "class_path": "/Game/Agrarian/Blueprints/World/BP_FreshWaterSource",
+        "location_xy": unreal.Vector(-7200.0, 10400.0, 0.0),
+        "z_offset": 36.0,
+        "rotation": unreal.Rotator(0.0, 0.0, 0.0),
+    },
+]
+
+
 FOLIAGE_ZONES = {
     "trees": {
         "count": 42,
@@ -395,6 +406,7 @@ def main():
 
     labels = {spec["label"] for spec in DEMO_ACTORS}
     labels.update(spec["label"] for spec in BIOME_RESOURCE_ACTORS)
+    labels.update(spec["label"] for spec in WATER_SOURCE_ACTORS)
     labels.add(FOLIAGE_LABEL)
     remove_existing_demo_actors(labels)
 
@@ -402,6 +414,8 @@ def main():
     spawn_foliage_actor(height_values)
     for spec in BIOME_RESOURCE_ACTORS:
         spawn_demo_actor(spec, height_values)
+    for spec in WATER_SOURCE_ACTORS:
+        spawn_demo_actor(spec, height_values)
     for spec in DEMO_ACTORS:
         spawn_demo_actor(spec, height_values)
 

Scripts/setup_playable_blueprints.py

@@ -5,6 +5,7 @@ BLUEPRINT_ROOT = "/Game/Agrarian/Blueprints"
 RESOURCE_FOLDER = f"{BLUEPRINT_ROOT}/Resources"
 STRUCTURE_FOLDER = f"{BLUEPRINT_ROOT}/Structures"
 WILDLIFE_FOLDER = f"{BLUEPRINT_ROOT}/Wildlife"
+WORLD_FOLDER = f"{BLUEPRINT_ROOT}/World"
 
 WOOD_ITEM_PATH = "/Game/Agrarian/DataAssets/Items/DA_Item_Wood"
 FIBER_ITEM_PATH = "/Game/Agrarian/DataAssets/Items/DA_Item_Fiber"
@@ -62,6 +63,17 @@ BLUEPRINTS = [
         "mesh": MESH_CYLINDER_PATH,
         "scale": unreal.Vector(1.3, 1.3, 0.25),
     },
+    {
+        "asset": "BP_FreshWaterSource",
+        "folder": WORLD_FOLDER,
+        "parent": unreal.AgrarianWaterSource,
+        "defaults": {
+            "water_restore_amount": 45.0,
+            "display_name": "Fresh Water Spring",
+        },
+        "mesh": MESH_CYLINDER_PATH,
+        "scale": unreal.Vector(2.2, 2.2, 0.12),
+    },
     {
         "asset": "BP_PrimitiveShelter",
         "folder": STRUCTURE_FOLDER,
@@ -165,7 +177,7 @@ def apply_defaults(blueprint, config):
 
 
 def main():
-    for folder in (RESOURCE_FOLDER, STRUCTURE_FOLDER, WILDLIFE_FOLDER):
+    for folder in (RESOURCE_FOLDER, STRUCTURE_FOLDER, WILDLIFE_FOLDER, WORLD_FOLDER):
         unreal.EditorAssetLibrary.make_directory(folder)
 
     for config in BLUEPRINTS:

Scripts/verify_ground_zero_water_source.py

@@ -0,0 +1,43 @@
+import unreal
+
+
+MAP_PATH = "/Game/Agrarian/Maps/L_GroundZeroTerrain_Test"
+WATER_SOURCE_LABEL = "AGR_GZ_FreshWaterSource_01"
+EXPECTED_WATER_RESTORE = 45.0
+
+
+def nearly_equal(left, right, tolerance=0.001):
+    return abs(float(left) - float(right)) <= tolerance
+
+
+def get_actor_label(actor):
+    try:
+        return actor.get_actor_label()
+    except Exception:
+        return actor.get_name()
+
+
+def main():
+    if not unreal.EditorLevelLibrary.load_level(MAP_PATH):
+        raise RuntimeError(f"Could not load map: {MAP_PATH}")
+
+    actors = unreal.EditorLevelLibrary.get_all_level_actors()
+    water_sources = [actor for actor in actors if get_actor_label(actor) == WATER_SOURCE_LABEL]
+    if len(water_sources) != 1:
+        raise RuntimeError(f"Expected exactly one {WATER_SOURCE_LABEL}, found {len(water_sources)}")
+
+    water_source = water_sources[0]
+    actual_restore = water_source.get_editor_property("water_restore_amount")
+    if not nearly_equal(actual_restore, EXPECTED_WATER_RESTORE):
+        raise RuntimeError(f"Water restore expected {EXPECTED_WATER_RESTORE}, got {actual_restore}")
+
+    if not isinstance(water_source, unreal.AgrarianWaterSource):
+        raise RuntimeError(f"{WATER_SOURCE_LABEL} is not an AgrarianWaterSource")
+
+    unreal.log(
+        "Ground Zero water source verification complete: "
+        f"{WATER_SOURCE_LABEL}, restore={actual_restore}."
+    )
+
+
+main()

Scripts/verify_playable_blueprints.py

@@ -30,6 +30,11 @@ EXPECTED = {
             "warmth_per_second": 0.03,
         },
     },
+    "/Game/Agrarian/Blueprints/World/BP_FreshWaterSource": {
+        "properties": {
+            "water_restore_amount": 45.0,
+        },
+    },
     "/Game/Agrarian/Blueprints/Structures/BP_PrimitiveShelter": {
         "properties": {
             "weather_protection": 0.7,