feat: add cost now accepts initial_default_weights argument ( known…

… use case is for air_vs_ground grid )

Closes #81

MapAnalyzer/MapData.py

@@ -292,7 +292,7 @@ def pathfind(self, start: Union[Tuple[int, int], Point2], goal: Union[Tuple[int,
                                     sensitivity=sensitivity)
 
     def add_cost(self, position: Tuple[int, int], radius: int, grid: ndarray, weight: int = 100, safe: bool = True,
-                 ) -> ndarray:
+                 initial_default_weights: int = 0) -> ndarray:
         """
         :rtype: numpy.ndarray
 
@@ -304,7 +304,8 @@ def add_cost(self, position: Tuple[int, int], radius: int, grid: ndarray, weight
             When ``safe=False`` the Pather will not adjust illegal values below 1 which could result in a crash`
 
         """
-        return self.pather.add_cost(position=position, radius=radius, arr=grid, weight=weight, safe=safe)
+        return self.pather.add_cost(position=position, radius=radius, arr=grid, weight=weight, safe=safe,
+                                    initial_default_weights=initial_default_weights)
 
     """Utility methods"""
 

MapAnalyzer/Pather.py

@@ -172,7 +172,7 @@ def pathfind(self, start: Tuple[int, int], goal: Tuple[int, int], grid: Optional
             return None
 
     def add_cost(self, position: Tuple[int, int], radius: int, arr: ndarray, weight: int = 100,
-                 safe: bool = True) -> ndarray:
+                 safe: bool = True, initial_default_weights: int = 0) -> ndarray:
         ri, ci = skdraw.disk(center=(int(position[0]), int(position[1])), radius=radius, shape=arr.shape)
         if len(ri) == 0 or len(ci) == 0:
             # this happens when the center point is near map edge, and the radius added goes beyond the edge
@@ -196,6 +196,9 @@ def in_bounds_ri(y):
         ri_vec = np.vectorize(in_bounds_ri)
         ci = ci_vec(ci)
         ri = ri_vec(ri)
+        if initial_default_weights > 0:
+            arr[ri, ci] = np.where(arr[ri, ci] == 1, initial_default_weights, arr[ri, ci])
+
         arr[ri, ci] += weight
         if np.any(arr < 1) and safe:
             self.map_data.logger.warning(

run.py

@@ -47,7 +47,9 @@ def get_map_file_list() -> List[str]:
     reg_end = map_data.where_all(map_data.bot.enemy_start_locations[0].position)[0]
     p0 = Point2(reg_start.center)
     p1 = Point2(reg_end.center)
-    influence_grid = map_data.get_air_vs_ground_grid()
+    influence_grid = map_data.get_air_vs_ground_grid(default_weight=50)
+    p = (50, 130)
+    influence_grid = map_data.add_cost(grid=influence_grid, position=p, radius=10, initial_default_weights=50)
     map_data.plot_influenced_path(start=p0, goal=p1, weight_array=influence_grid, allow_diagonal=True)
     map_data.show()
     break