kristoferssolo/Tetris
1
0
Fork 0
mirror of https://github.com/kristoferssolo/Tetris.git synced 2025-10-21 20:00:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(ai): add holes

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-09 20:28:53 +02:00
parent 9daa4f6509
commit 00da10e482
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/ai/heuristics/holes.py
View File

@ -4,4 +4,19 @@ import numpy as np
def holes( def holes(
field: np.ndarray[int, np.dtype[np.uint8]], field: np.ndarray[int, np.dtype[np.uint8]],
) -> int: ) -> int:
return 0 """
Calculate the number of holes in each column of the given field.
Args:
field: The signal field.
peaks: Array containing peak indices. If not provided, it will be computed from the field.
Returns:
The total number of holes in the field.
"""
first_nonzero_indices = np.argmax(field != 0, axis=0)
mask = (field == 0) & (np.arange(field.shape[0])[:, np.newaxis] > first_nonzero_indices)
return int(np.sum(mask))