DFlee move rules enforcement of location weights. #81

djgroen · Aug 27, 2023 · fe276e5 · fe276e5
1 parent 218eb88
commit fe276e5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/flee/moving.py b/flee/moving.py
@@ -48,6 +48,13 @@ def getEndPointScore(agent, link) -> float:
         base *= 1.0/(max(1.0,link.endpoint.calculateDistance(agent.location))**power_factor)
 
 
+    # DFlee Flood Location Weight implementation
+    if SimulationSettings.move_rules["FloodRulesEnabled"] is True:
+        flood_level = link.endpoint.attributes.get("flood_level",0)
+        if flood_level > 0:
+            base *= float(SimulationSettings.move_rules["FloodLocWeights"][flood_level])
+
+
     if link.endpoint.camp is True:
         if SimulationSettings.move_rules["MatchCampEthnicity"]:
             base *= (spawning.getAttributeRatio(link.endpoint, "ethnicity") * 10.0)

diff --git a/tests/test_dflee.py b/tests/test_dflee.py
@@ -42,6 +42,8 @@ def test_flood_level_location_attribute():
 
     flee.SimulationSettings.ReadFromYML("empty.yml")
     flee.SimulationSettings.move_rules["FloodRulesEnabled"] = True
+    flee.SimulationSettings.move_rules["FloodLocWeights"] = [0.0,1.0,1.0,1.0,1.0]
+    flee.SimulationSettings.move_rules["FloodMoveChances"] = [0.0,1.0,1.0,1.0,1.0]
 
     e = flee.Ecosystem()