[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

examples/wind_driven_fire/README.md

@@ -15,29 +15,29 @@ Fire spreads probabilistically between neighboring trees, with ignition probabil
 
 ### Fire Spread Rule
 
-At each time step, trees that are *On Fire* attempt to ignite their neighbors (Moore neighborhood).  
+At each time step, trees that are *On Fire* attempt to ignite their neighbors (Moore neighborhood).
 The ignition probability is modified by wind alignment:
 
 $$
 p = p_{\text{spread}} \cdot \left( 1 + s \cdot \cos(\theta) \right)
 $$
 where:
-- $ p_{\text{spread}} $ is the baseline spread probability  
-- $  s $  is wind strength  
-- $  \theta $  is the angle between the wind direction and the neighbor direction  
+- $ p_{\text{spread}} $ is the baseline spread probability
+- $  s $  is wind strength
+- $  \theta $  is the angle between the wind direction and the neighbor direction
 After spreading fire, burning trees transition to *Burned Out*.
 
 
 ### Wind Representation
 
 Wind is represented as a **global vector field** defined by:
 
-- **Wind direction (degrees)** using the meteorological convention (wind coming *from*)  
-- **Wind strength**, controlling the magnitude of directional bias  
+- **Wind direction (degrees)** using the meteorological convention (wind coming *from*)
+- **Wind strength**, controlling the magnitude of directional bias
 
 ### Fire-Head–Based Rate of Spread
 
-- **Head Distance (Downwind)**: The **fire head** is defined as the furthest burned cell projected onto the wind axis.  
+- **Head Distance (Downwind)**: The **fire head** is defined as the furthest burned cell projected onto the wind axis.
 Fire-head distance is measured relative to the ignition reference point:
 
 $$

examples/wind_driven_fire/app.py

@@ -1,7 +1,5 @@
-from mesa.visualization import SolaraViz, make_space_component, make_plot_component
+from mesa.visualization import SolaraViz, make_plot_component, make_space_component
 from mesa.visualization.user_param import Slider
-import matplotlib.cm as cm
-from matplotlib.colors import to_hex
 from wind_driven_fire.agent import Tree
 from wind_driven_fire.model import ForestFire
 
@@ -10,17 +8,13 @@
 COLORS = {"Fine": "#00AA00", "On Fire": "#880000", "Burned Out": "#000000"}
 
 cx, cy = WIDTH // 2, HEIGHT // 2
-center_9_points = [
-    (x, y) 
-    for x in range(cx - 1, cx + 2)  
-    for y in range(cy - 1, cy + 2)  
-]
+center_9_points = [(x, y) for x in range(cx - 1, cx + 2) for y in range(cy - 1, cy + 2)]
 
 
 def forest_fire_portrayal(agent):
     if agent is None:
         return
-    
+
     class_name = type(agent).__name__
 
     # Tree
@@ -30,26 +24,34 @@ def forest_fire_portrayal(agent):
             "size": 20,
             "marker": "s",
         }
-    
+
+
 ros_plot = make_plot_component(
-    {"Rate of spread at the fire head": "blue", "Rate of spread at the fire Flank": "orange"}
+    {
+        "Rate of spread at the fire head": "blue",
+        "Rate of spread at the fire Flank": "orange",
+    }
 )
 
 
-model = ForestFire(width=100, height=100, p_spread=0.25, wind_dir=0.0, wind_strength=0.8,ignite_pos=center_9_points)
+model = ForestFire(
+    width=100,
+    height=100,
+    p_spread=0.25,
+    wind_dir=0.0,
+    wind_strength=0.8,
+    ignite_pos=center_9_points,
+)
 model_params = {
     "height": HEIGHT,
     "width": WIDTH,
     "p_spread": Slider("Spread probability", 0.25, 0.0, 1.0, 0.05),
     "wind_dir": Slider("Wind direction (deg)", 0.0, 0.0, 360.0, 5.0),
     "wind_strength": Slider("Wind strength", 0.8, 0.0, 1.0, 0.1),
-    "ignite_pos": center_9_points #central ignite
+    "ignite_pos": center_9_points,  # central ignite
 }
 
-space_component = make_space_component(
-    forest_fire_portrayal,
-    backend="matplotlib"
-)
+space_component = make_space_component(forest_fire_portrayal, backend="matplotlib")
 
 page = SolaraViz(
     model,
@@ -61,10 +63,10 @@ def forest_fire_portrayal(agent):
     name="Wind-Driven Forest Fire ",
     description="""
     **Anisotropic Fire Spread Demonstration**
-    
+
     This example highlights how wind introduces **anisotropy** into the system.
-    
+
     * **Visual Proof**: The blue arrow indicates wind direction. Note how the fire scar (black agents) elongates along the arrow's direction.
     * **Statistical Proof**: Observe the chart below. "Span Y" vs "Span X" shows the divergence in spread width vs height, confirming the fire is elliptical, not circular.
-    """
-)
+    """,
+)

examples/wind_driven_fire/wind_driven_fire/agent.py

@@ -18,12 +18,10 @@ def step(self):
 
         # Spread fire (8-neighborhood)
         neighbors = self.model.grid.get_neighbors(
-            self.pos,
-            moore=True,
-            include_center=False)
-        
-        for n in neighbors:
+            self.pos, moore=True, include_center=False
+        )
 
+        for n in neighbors:
             if n.condition != "Fine":
                 continue
             # Calculate wind bias
@@ -38,4 +36,3 @@ def step(self):
 
         # After spreading fire, this tree burns out
         self.condition = "Burned Out"
-

examples/wind_driven_fire/wind_driven_fire/model.py

@@ -1,12 +1,14 @@
-"""Forest fire with global wind """
+"""Forest fire with global wind"""
 
-import mesa
 import math
 
-from mesa.space import MultiGrid
+import mesa
 from mesa.datacollection import DataCollector
+from mesa.space import MultiGrid
+
 from .agent import Tree
 
+
 class ForestFire(mesa.Model):
     """
     Wind-driven Forest Fire Model
@@ -17,8 +19,8 @@ def __init__(
         width=100,
         height=100,
         p_spread=0.25,
-        wind_dir=0.0,          # degrees
-        wind_strength=0.8,     
+        wind_dir=0.0,  # degrees
+        wind_strength=0.8,
         seed=None,
         ignite_pos=None,
     ):
@@ -37,9 +39,8 @@ def __init__(
         # Place trees
         for x in range(width):
             for y in range(height):
-                    tree = Tree(self, condition="Fine")
-                    self.grid.place_agent(tree, (x, y))
-
+                tree = Tree(self, condition="Fine")
+                self.grid.place_agent(tree, (x, y))
 
         # Ignite Logic
         if ignite_pos:
@@ -59,47 +60,52 @@ def __init__(
             # if random ignition, fall back to grid center
             self.ignite_ref = (self.width / 2, self.height / 2)
 
-
         # Data collector
         self.datacollector = DataCollector(
             {
                 "Fine": lambda m: self.count_type(m, "Fine"),
                 "On Fire": lambda m: self.count_type(m, "On Fire"),
                 "Burned Out": lambda m: self.count_type(m, "Burned Out"),
                 "HeadDist Wind": lambda m: ForestFire.get_head_distance_wind(m),
-                "FlankHalfwidth Cross": lambda m: ForestFire.get_flank_halfwidth_crosswind(m),
-                "Rate of spread at the fire head": lambda m: ForestFire.get_head_distance_wind(m) / max(1, m.step_count),
-                "Rate of spread at the fire Flank": lambda m: ForestFire.get_flank_halfwidth_crosswind(m) / max(1, m.step_count),
-
+                "FlankHalfwidth Cross": lambda m: ForestFire.get_flank_halfwidth_crosswind(
+                    m
+                ),
+                "Rate of spread at the fire head": lambda m: ForestFire.get_head_distance_wind(
+                    m
+                )
+                / max(1, m.step_count),
+                "Rate of spread at the fire Flank": lambda m: ForestFire.get_flank_halfwidth_crosswind(
+                    m
+                )
+                / max(1, m.step_count),
             }
         )
 
         self.datacollector.collect(self)
 
     def ignite(self, pos=None):
-            """Ignite a specific tree or a random one."""
-            if pos is None:
-                # Random ignition
-                trees = [a for a in self.agents if a.condition == "Fine"]
-                if trees:
-                    self.random.choice(trees).condition = "On Fire"
-            else:
-                if not self.grid.out_of_bounds(pos):
-                    cell_agents = self.grid.get_cell_list_contents([pos])
-                    for a in cell_agents:
-                        if  a.condition == "Fine":
-                            a.condition = "On Fire"
-                            return
-
-    
+        """Ignite a specific tree or a random one."""
+        if pos is None:
+            # Random ignition
+            trees = [a for a in self.agents if a.condition == "Fine"]
+            if trees:
+                self.random.choice(trees).condition = "On Fire"
+        else:
+            if not self.grid.out_of_bounds(pos):
+                cell_agents = self.grid.get_cell_list_contents([pos])
+                for a in cell_agents:
+                    if a.condition == "Fine":
+                        a.condition = "On Fire"
+                        return
+
     def wind_unit_vector(self):
         """
         Convert meteorological wind direction to unit vector.
         0° = From North (Vector: 0, -1)
         """
         rad = math.radians(self.wind_dir)
         # Wind comes FROM rad, blows TOWARD opposite
-        # Grid (0,0) is bottom-left. 
+        # Grid (0,0) is bottom-left.
         wx = -math.sin(rad)
         wy = -math.cos(rad)
         return (wx, wy)
@@ -109,21 +115,20 @@ def wind_biased_probability(self, dx, dy):
         wx, wy = self.wind_unit_vector()
 
         norm = (dx**2 + dy**2) ** 0.5
-        if norm == 0: return 1.0
-        
+        if norm == 0:
+            return 1.0
+
         dxu, dyu = dx / norm, dy / norm
-        align = dxu * wx + dyu * wy 
-        
+        align = dxu * wx + dyu * wy
+
         # Simple linear bias: 1 + strength * alignment
         return max(0.0, 1.0 + (self.wind_strength * align))
 
-
     def step(self):
         self.step_count += 1
         self.agents.shuffle_do("step")
         self.datacollector.collect(self)
 
-
     @staticmethod
     def count_type(model, tree_condition):
         """Helper to count trees in a given condition in a fast way."""
@@ -136,8 +141,10 @@ def count_type(model, tree_condition):
     @staticmethod
     def _burnt_positions(model):
         return [
-            a.pos for a in model.agents
-            if a.__class__.__name__ == "Tree" and a.condition in ["On Fire", "Burned Out"]
+            a.pos
+            for a in model.agents
+            if a.__class__.__name__ == "Tree"
+            and a.condition in ["On Fire", "Burned Out"]
         ]
 
     @staticmethod
@@ -172,13 +179,3 @@ def get_flank_halfwidth_crosswind(model):
 
         dev = [abs(x * px + y * py - proj_ignite_c) for (x, y) in pts]
         return float(max(dev))
-
-
-    
-
-
-
-
-
-
-