refactor digit.py (now scan_field.py)

add new data (alex_notebook monitor size)

Author: alexnurin

AI_main.py

@@ -3,7 +3,7 @@
 from direct_keys import *
 import time
 from figures import piece_weight, find_figure
-from digit import get_field
+from scan_field import get_field
 import keyboard
 
 FIELD_SIZE = [20, 10]

data.py

@@ -34,6 +34,8 @@ def get_next_3(self, img):
         next3 = n3xt[150:195]
         return next1, next2, next3
 
+
 misha = Consts(370, 970, 155, 455, 340, 540, 485, 560)
 alex = Consts(339, 895, 142, 422, 315, 505, 450, 475)
+alex_notebook = Consts(225, 888, 115, 441, 315, 505, 450, 475)
 maxx = Consts(370, 970, 155, 455, 340, 540, 485, 560)

main.py

@@ -1,5 +1,5 @@
 import time
-from digit import get_field
+from scan_field import get_field
 from figures import type_figure_ext
 from AI_main import AI
 import numpy as np

digit.py scan_field.pydigit.py renamed to scan_field.py

@@ -3,8 +3,8 @@
 from mss import mss
 import data
 
-consts = data.alex  # CUSTOM
-dbug = 0
+consts = data.alex_notebook  # CUSTOM
+debug_status = 0
 
 d = dict()
 d[0] = (230, 228, 180)
@@ -17,10 +17,10 @@
 d[7] = (96, 45, 36)
 
 
-def pD(arr, fsize=10):
-    if dbug == 1:
+def print_image(arr, figure_size=10):
+    if debug_status == 1:
         n = len(arr)
-        fig = plt.figure(figsize=(fsize, fsize))
+        fig = plt.figure(figsize=(figure_size, figure_size))
         for i in range(n):
             fig.add_subplot(1, n, i + 1)
             plt.imshow(arr[i])
@@ -52,13 +52,13 @@ def get_field():
 
         next1, next2, next3 = map(get_figure, consts.get_next_3(img))
 
-        sizeCell = field.shape[0] // 20
+        size_cell = field.shape[0] // 20
         arr = np.zeros((20, 10))
-        for i in range(sizeCell):
+        for i in range(size_cell):
             r1 = np.array(np.linspace(0, field.shape[0], 21)[:-1], int) + i
             r2 = np.array(np.linspace(0, field.shape[1], 11)[:-1], int) + i
             arr += field[r1][:, r2]
 
-        kek = np.array(arr / sizeCell + 0.5, int)
-        pD((img, field, kek))
+        kek = np.array(arr / size_cell + 0.5, int)
+        print_image((img, field, kek))
         return kek, next1