1

encoders/oneplane.py

@@ -0,0 +1,60 @@
+# tag::oneplane_imports[]
+import numpy as np
+
+from tugo.encoders.base import Encoder
+from tugo.goboard import Point
+# end::oneplane_imports[]
+
+
+# tag::oneplane_encoder[]
+class OnePlaneEncoder(Encoder):
+    def __init__(self, board_size):
+        self.board_width, self.board_height = board_size
+        self.num_planes = 1
+
+    def name(self):  # <1>
+        return 'oneplane'
+
+    def encode(self, game_state):  # <2>
+        board_matrix = np.zeros(self.shape())
+        next_player = game_state.next_player
+        for r in range(self.board_height):
+            for c in range(self.board_width):
+                p = Point(row=r + 1, col=c + 1)
+                go_string = game_state.board.get_go_string(p)
+                if go_string is None:
+                    continue
+                if go_string.color == next_player:
+                    board_matrix[0, r, c] = 1
+                else:
+                    board_matrix[0, r, c] = -1
+        return board_matrix
+
+# <1> We can reference this encoder by the name "oneplane".
+# <2> To encode, we fill a matrix with 1 if the point contains one of the current player's stones, -1 if the point contains the opponent's stones and 0 if the point is empty.
+# end::oneplane_encoder[]
+
+# tag::oneplane_encoder_2[]
+    def encode_point(self, point):  # <1>
+        return self.board_width * (point.row - 1) + (point.col - 1)
+
+    def decode_point_index(self, index):  # <2>
+        row = index // self.board_width
+        col = index % self.board_width
+        return Point(row=row + 1, col=col + 1)
+
+    def num_points(self):
+        return self.board_width * self.board_height
+
+    def shape(self):
+        return self.num_planes, self.board_height, self.board_width
+
+# <1> Turn a board point into an integer index.
+# <2> Turn an integer index into a board point.
+# end::oneplane_encoder_2[]
+
+
+# tag::oneplane_create[]
+def create(board_size):
+    return OnePlaneEncoder(board_size)
+# end::oneplane_create[]