Sudoku Solver with GUI.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "pygame 1.9.6\n",
      "Hello from the pygame community. https://www.pygame.org/contribute.html\n"
     ]
    }
   ],
   "source": [
    "import pygame\n",
    "import time\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "class Grid:\n",
    "    board = [\n",
    "        [0, 0, 0, 0, 0, 0, 0, 7, 0],\n",
    "        [0, 0, 4, 8, 0, 0, 0, 0, 0],\n",
    "        [1, 0, 9, 4, 6, 0, 0, 3, 0],\n",
    "        [4, 0, 0, 0, 1, 0, 0, 0, 0],\n",
    "        [0, 5, 0, 0, 2, 0, 0, 0, 0],\n",
    "        [0, 1, 0, 3, 0, 6, 4, 0, 0],\n",
    "        [0, 0, 0, 0, 0, 0, 0, 9, 8],\n",
    "        [5, 0, 0, 0, 0, 0, 6, 0, 7],\n",
    "        [2, 0, 0, 1, 0, 0, 0, 0, 0],\n",
    "        \n",
    "    ]\n",
    "\n",
    "    def __init__(self, rows, cols, width, height, win):\n",
    "        self.rows = rows\n",
    "        self.cols = cols\n",
    "        self.cubes = [[Cube(self.board[i][j], i, j, width, height) for j in range(cols)] for i in range(rows)]\n",
    "        self.width = width\n",
    "        self.height = height\n",
    "        self.model = None\n",
    "        self.update_model()\n",
    "        self.selected = None\n",
    "        self.win = win\n",
    "\n",
    "    def update_model(self):\n",
    "        self.model = [[self.cubes[i][j].value for j in range(self.cols)] for i in range(self.rows)]\n",
    "\n",
    "    def place(self, val):\n",
    "        row, col = self.selected\n",
    "        if self.cubes[row][col].value == 0:\n",
    "            self.cubes[row][col].set(val)\n",
    "            self.update_model()\n",
    "\n",
    "            if valid(self.model, val, (row,col)) and self.solve():\n",
    "                return True\n",
    "            else:\n",
    "                self.cubes[row][col].set(0)\n",
    "                self.cubes[row][col].set_temp(0)\n",
    "                self.update_model()\n",
    "                return False\n",
    "\n",
    "    def sketch(self, val):\n",
    "        row, col = self.selected\n",
    "        self.cubes[row][col].set_temp(val)\n",
    "\n",
    "    def draw(self):\n",
    "        # Draw Grid Lines\n",
    "        gap = self.width / 9\n",
    "        for i in range(self.rows+1):\n",
    "            if i % 3 == 0 and i != 0:\n",
    "                thick = 4\n",
    "            else:\n",
    "                thick = 1\n",
    "            pygame.draw.line(self.win, (0,0,0), (0, i*gap), (self.width, i*gap), thick)\n",
    "            pygame.draw.line(self.win, (0, 0, 0), (i * gap, 0), (i * gap, self.height), thick)\n",
    "\n",
    "        # Draw Cubes\n",
    "        for i in range(self.rows):\n",
    "            for j in range(self.cols):\n",
    "                self.cubes[i][j].draw(self.win)\n",
    "\n",
    "    def select(self, row, col):\n",
    "        # Reset all other\n",
    "        for i in range(self.rows):\n",
    "            for j in range(self.cols):\n",
    "                self.cubes[i][j].selected = False\n",
    "\n",
    "        self.cubes[row][col].selected = True\n",
    "        self.selected = (row, col)\n",
    "\n",
    "    def clear(self):\n",
    "        row, col = self.selected\n",
    "        if self.cubes[row][col].value == 0:\n",
    "            self.cubes[row][col].set_temp(0)\n",
    "\n",
    "    def click(self, pos):\n",
    "        \"\"\"\n",
    "        :param: pos\n",
    "        :return: (row, col)\n",
    "        \"\"\"\n",
    "        if pos[0] < self.width and pos[1] < self.height:\n",
    "            gap = self.width / 9\n",
    "            x = pos[0] // gap\n",
    "            y = pos[1] // gap\n",
    "            return (int(y),int(x))\n",
    "        else:\n",
    "            return None\n",
    "\n",
    "    def is_finished(self):\n",
    "        for i in range(self.rows):\n",
    "            for j in range(self.cols):\n",
    "                if self.cubes[i][j].value == 0:\n",
    "                    return False\n",
    "        return True\n",
    "\n",
    "    def solve(self):\n",
    "        find = find_empty(self.model)\n",
    "        if not find:\n",
    "            return True\n",
    "        else:\n",
    "            row, col = find\n",
    "\n",
    "        for i in range(1, 10):\n",
    "            if valid(self.model, i, (row, col)):\n",
    "                self.model[row][col] = i\n",
    "\n",
    "                if self.solve():\n",
    "                    return True\n",
    "\n",
    "                self.model[row][col] = 0\n",
    "\n",
    "        return False\n",
    "\n",
    "    def solve_gui(self):\n",
    "        find = find_empty(self.model)\n",
    "        if not find:\n",
    "            return True\n",
    "        else:\n",
    "            row, col = find\n",
    "\n",
    "        for i in range(1, 10):\n",
    "            if valid(self.model, i, (row, col)):\n",
    "                self.model[row][col] = i\n",
    "                self.cubes[row][col].set(i)\n",
    "                self.cubes[row][col].draw_change(self.win, True)\n",
    "                self.update_model()\n",
    "                pygame.display.update()\n",
    "                pygame.time.delay(100)\n",
    "\n",
    "                if self.solve_gui():\n",
    "                    return True\n",
    "\n",
    "                self.model[row][col] = 0\n",
    "                self.cubes[row][col].set(0)\n",
    "                self.update_model()\n",
    "                self.cubes[row][col].draw_change(self.win, False)\n",
    "                pygame.display.update()\n",
    "                pygame.time.delay(100)\n",
    "\n",
    "        return False\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "class Cube:\n",
    "    rows = 9\n",
    "    cols = 9\n",
    "\n",
    "    def __init__(self, value, row, col, width, height):\n",
    "        self.value = value\n",
    "        self.temp = 0\n",
    "        self.row = row\n",
    "        self.col = col\n",
    "        self.width = width\n",
    "        self.height = height\n",
    "        self.selected = False\n",
    "\n",
    "    def draw(self, win):\n",
    "        fnt = pygame.font.SysFont(\"comicsans\", 40)\n",
    "\n",
    "        gap = self.width / 9\n",
    "        x = self.col * gap\n",
    "        y = self.row * gap\n",
    "\n",
    "        if self.temp != 0 and self.value == 0:\n",
    "            text = fnt.render(str(self.temp), 1, (128,128,128))\n",
    "            win.blit(text, (x+5, y+5))\n",
    "        elif not(self.value == 0):\n",
    "            text = fnt.render(str(self.value), 1, (0, 0, 0))\n",
    "            win.blit(text, (x + (gap/2 - text.get_width()/2), y + (gap/2 - text.get_height()/2)))\n",
    "\n",
    "        if self.selected:\n",
    "            pygame.draw.rect(win, (255,0,0), (x,y, gap ,gap), 3)\n",
    "\n",
    "    def draw_change(self, win, g=True):\n",
    "        fnt = pygame.font.SysFont(\"comicsans\", 40)\n",
    "\n",
    "        gap = self.width / 9\n",
    "        x = self.col * gap\n",
    "        y = self.row * gap\n",
    "\n",
    "        pygame.draw.rect(win, (255, 255, 255), (x, y, gap, gap), 0)\n",
    "\n",
    "        text = fnt.render(str(self.value), 1, (0, 0, 0))\n",
    "        win.blit(text, (x + (gap / 2 - text.get_width() / 2), y + (gap / 2 - text.get_height() / 2)))\n",
    "        if g:\n",
    "            pygame.draw.rect(win, (0, 255, 0), (x, y, gap, gap), 3)\n",
    "        else:\n",
    "            pygame.draw.rect(win, (255, 0, 0), (x, y, gap, gap), 3)\n",
    "\n",
    "    def set(self, val):\n",
    "        self.value = val\n",
    "\n",
    "    def set_temp(self, val):\n",
    "        self.temp = val\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def find_empty(bo):\n",
    "    for i in range(len(bo)):\n",
    "        for j in range(len(bo[0])):\n",
    "            if bo[i][j] == 0:\n",
    "                return (i, j)  # row, col\n",
    "\n",
    "    return None"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "def valid(bo, num, pos):\n",
    "    # Check row\n",
    "    for i in range(len(bo[0])):\n",
    "        if bo[pos[0]][i] == num and pos[1] != i:\n",
    "            return False\n",
    "\n",
    "    # Check column\n",
    "    for i in range(len(bo)):\n",
    "        if bo[i][pos[1]] == num and pos[0] != i:\n",
    "            return False\n",
    "\n",
    "    # Check box\n",
    "    box_x = pos[1] // 3\n",
    "    box_y = pos[0] // 3\n",
    "\n",
    "    for i in range(box_y*3, box_y*3 + 3):\n",
    "        for j in range(box_x * 3, box_x*3 + 3):\n",
    "            if bo[i][j] == num and (i,j) != pos:\n",
    "                return False\n",
    "\n",
    "    return True\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def redraw_window(win, board, time, strikes):\n",
    "    win.fill((255,255,255))\n",
    "    # Draw time\n",
    "    fnt = pygame.font.SysFont(\"comicsans\", 40)\n",
    "    text = fnt.render(\"Time: \" + format_time(time), 1, (0,0,0))\n",
    "    win.blit(text, (540 - 160, 560))\n",
    "    # Draw Strikes\n",
    "    text = fnt.render(\"X \" * strikes, 1, (255, 0, 0))\n",
    "    win.blit(text, (20, 560))\n",
    "    # Draw grid and board\n",
    "    board.draw()\n",
    "\n",
    "\n",
    "def format_time(secs):\n",
    "    sec = secs%60\n",
    "    minute = secs//60\n",
    "    hour = minute//60\n",
    "\n",
    "    mat = \" \" + str(minute) + \":\" + str(sec)\n",
    "    return mat\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Wrong\n",
      "Wrong\n",
      "Wrong\n"
     ]
    }
   ],
   "source": [
    "pygame.font.init()\n",
    "def main():\n",
    "    win = pygame.display.set_mode((540,600))\n",
    "    pygame.display.set_caption(\"Sudoku\")\n",
    "    board = Grid(9, 9, 540, 540, win)\n",
    "    key = None\n",
    "    run = True\n",
    "    start = time.time()\n",
    "    strikes = 0\n",
    "    while run:\n",
    "\n",
    "        play_time = round(time.time() - start)\n",
    "\n",
    "        for event in pygame.event.get():\n",
    "            if event.type == pygame.QUIT:\n",
    "                run = False\n",
    "            if event.type == pygame.KEYDOWN:\n",
    "                if event.key == pygame.K_1:\n",
    "                    key = 1\n",
    "                if event.key == pygame.K_2:\n",
    "                    key = 2\n",
    "                if event.key == pygame.K_3:\n",
    "                    key = 3\n",
    "                if event.key == pygame.K_4:\n",
    "                    key = 4\n",
    "                if event.key == pygame.K_5:\n",
    "                    key = 5\n",
    "                if event.key == pygame.K_6:\n",
    "                    key = 6\n",
    "                if event.key == pygame.K_7:\n",
    "                    key = 7\n",
    "                if event.key == pygame.K_8:\n",
    "                    key = 8\n",
    "                if event.key == pygame.K_9:\n",
    "                    key = 9\n",
    "                if event.key == pygame.K_DELETE:\n",
    "                    board.clear()\n",
    "                    key = None\n",
    "\n",
    "                if event.key == pygame.K_SPACE:\n",
    "                    board.solve_gui()\n",
    "\n",
    "                if event.key == pygame.K_RETURN:\n",
    "                    i, j = board.selected\n",
    "                    if board.cubes[i][j].temp != 0:\n",
    "                        if board.place(board.cubes[i][j].temp):\n",
    "                            print(\"Success\")\n",
    "                        else:\n",
    "                            print(\"Wrong\")\n",
    "                            strikes += 1\n",
    "                        key = None\n",
    "\n",
    "                        if board.is_finished():\n",
    "                            print(\"Game over\")\n",
    "\n",
    "            if event.type == pygame.MOUSEBUTTONDOWN:\n",
    "                pos = pygame.mouse.get_pos()\n",
    "                clicked = board.click(pos)\n",
    "                if clicked:\n",
    "                    board.select(clicked[0], clicked[1])\n",
    "                    key = None\n",
    "\n",
    "        if board.selected and key != None:\n",
    "            board.sketch(key)\n",
    "\n",
    "        redraw_window(win, board, play_time, strikes)\n",
    "        pygame.display.update()\n",
    "\n",
    "\n",
    "main()\n",
    "pygame.quit()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}