From e8a0eee94111226706c220d734581f1c4edd6378 Mon Sep 17 00:00:00 2001
From: En Yi <randomsity1@gmail.com>
Date: Sat, 21 Jul 2018 11:47:53 +0800
Subject: [PATCH] Draw a scribble in a cell

---
 gameplay/sudoku_gameplay.py           |  2 ++
 graphic_components/sudoku_graphics.py | 11 +++++++++++
 2 files changed, 13 insertions(+)

diff --git a/gameplay/sudoku_gameplay.py b/gameplay/sudoku_gameplay.py
index d2c0140..d6b7868 100644
--- a/gameplay/sudoku_gameplay.py
+++ b/gameplay/sudoku_gameplay.py
@@ -18,6 +18,8 @@ class SudokuSystem:
     def __init__(self):
         self.number_grid = np.zeros((9, 9), dtype=np.uint8)
         self.cell_status = np.zeros((9, 9), dtype=np.uint8)
+        self.scribbles = np.zeros((9, 9), dtype='<U9')
+        self.scribbles[0, 0] = '234579'
         self.offending_cells = []
         for i in range(9):
             row = []
diff --git a/graphic_components/sudoku_graphics.py b/graphic_components/sudoku_graphics.py
index d5948dc..2fb8f66 100644
--- a/graphic_components/sudoku_graphics.py
+++ b/graphic_components/sudoku_graphics.py
@@ -43,6 +43,8 @@ class NumberPainter(BaseSudokuItem):
         self.fixed_pen.setColor(Qt.white)
         self.fixed_font = QFont("Helvetica", pointSize=14, weight=QFont.Bold)
 
+        self.scribble_font = QFont("Helvetica", pointSize=5)
+
     def paint(self, painter, style, widget=None):
         for i in range(9):
             for j in range(9):
@@ -70,6 +72,15 @@ class NumberPainter(BaseSudokuItem):
         painter.drawText(QRectF(w*self.parent.cell_width, h*self.parent.cell_height,
                                 self.parent.cell_width, self.parent.cell_height), Qt.AlignCenter, str(val))
 
+        painter.setPen(self.default_pen)
+        painter.setFont(self.scribble_font)
+        radius = 15
+        for scrib in self.sudoku_grid.scribbles[h, w]:
+            num = int(scrib)
+            num_x = radius * np.sin(np.deg2rad(360/10*num)) + w * self.parent.cell_width
+            num_y = - radius * np.cos(np.deg2rad(360 / 10 * num)) + h * self.parent.cell_height
+            painter.drawText(QRectF(num_x, num_y, self.parent.cell_width, self.parent.cell_height),
+                             Qt.AlignCenter, scrib)
 
 class SudokuGrid(BaseSudokuItem):
     # TODO: Add functions to animated the grid lines