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

fix(ai): get_bumpiness

Browse Source
This commit is contained in:
Kristofers Solo 2024-01-13 19:49:39 +02:00
parent e84cacca1c
commit d458a56d27
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/ai/heuristics/bumpiness.py
View File

@ -1,11 +1,13 @@
import numpy as np
from .peaks import get_peaks
def get_bumpiness(
field: np.ndarray[int, np.dtype[np.uint8]],
) -> int:
"""
Calculate the bumpiness of a given signal based on peaks.
Calculate the bumpiness of a given field based on peaks.
Args:
field: The game field.
@ -13,4 +15,6 @@ def get_bumpiness(
Returns:
The bumpiness of the field.
"""
return int(np.sum(np.abs(np.diff(field.shape[0] - np.argmax(field, axis=0)))))
field = get_peaks(field)
diff = np.diff(field)
return int(np.sum(np.abs(diff)))

2
src/ai/heuristics/peaks.py
View File

@ -11,7 +11,7 @@ def get_peaks(field: np.ndarray[int, np.dtype[np.uint8]]) -> np.ndarray[int, np.
Returns:
2D array representing the peaks of the field.
"""
result = np.zeros(field.shape[1], dtype=np.uint8)
result = np.zeros(field.shape[1], dtype=int)
for col in range(field.shape[1]):
for row in range(field.shape[0]):
if field[row, col] != 0: