From d52ee30259b2548391412926384fce56dd6ced08 Mon Sep 17 00:00:00 2001
From: Wenbo Hu <huwenbo.rambo@gmail.com>
Date: Tue, 12 Dec 2017 23:13:31 +0800
Subject: [PATCH] add nearby stones

---
 AlphaGo/game.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/AlphaGo/game.py b/AlphaGo/game.py
index 2a60a06..597d834 100644
--- a/AlphaGo/game.py
+++ b/AlphaGo/game.py
@@ -202,6 +202,24 @@ class Executor:
             elif color_estimate < 0:
                 return utils.WHITE
 
+   def _add_nearby_stones(self, neighbor_vertex_set, start_vertex_x, start_vertex_y, x_diff, y_diff, num_step):
+        '''
+        add the nearby stones around the input vertex
+        :param neighbor_vertex_set: input list
+        :param start_vertex_x: x axis of the input vertex
+        :param start_vertex_y: y axis of the input vertex
+        :param x_diff: add x axis
+        :param y_diff: add y axis
+        :param num_step: number of steps to be added
+        :return:
+        '''
+        for step in xrange(num_step):
+            new_neighbor_vertex = (start_vertex_x, start_vertex_y)
+            if self._in_board(new_neighbor_vertex):
+                neighbor_vertex_set.append((start_vertex_x, start_vertex_y))
+            start_vertex_x += x_diff
+            start_vertex_y += y_diff
+