app.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### 모듈 선언과 기초 설정"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 28,
   "metadata": {},
   "outputs": [],
   "source": [
    "from random import shuffle\n",
    "import pygame\n",
    "\n",
    "pygame.init()\n",
    "\n",
    "size = [700, 600] # 윈도우 크기\n",
    "screen = pygame.display.set_mode(size) # 디스플레이\n",
    "clock = pygame.time.Clock() # 시간 관리 객체\n",
    "font = pygame.font.Font(\"fonts/NanumSquareR.ttf\", 16) # 폰트\n",
    "\n",
    "pygame.display.set_caption('Sort Visualization') # 윈도우 타이틀"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### 클래스 정의\n",
    "- 오브젝트(막대기)\n",
    "- 버블정렬\n",
    "- 선택정렬\n",
    "- 삽입정렬"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 29,
   "metadata": {},
   "outputs": [],
   "source": [
    "# 막대기들\n",
    "class Bars:\n",
    "    def __init__(self, screen: pygame.display, bar_count: int, set_pos: list, set_size: list, margin: float, label: str=''):\n",
    "        self.screen = screen\n",
    "        self.margin = margin\n",
    "        self.pos = set_pos\n",
    "        self.size = set_size\n",
    "        self.indexs = []\n",
    "        self.palette = ['#1C1C1C', '#F44341', '#F1C422', '#46A4EB'] # hex 사용\n",
    "        self.color = self.palette[0]\n",
    "        self.label = font.render(label, True, self.color)\n",
    "        self.bars = []\n",
    "        for i in range(bar_count):\n",
    "            h = self.size[1] // bar_count * (i + 1)\n",
    "            # bar (y, height)\n",
    "            self.bars.append((self.pos[1] + self.size[1] - h, h, i))\n",
    "\n",
    "    # 무작위 섞기\n",
    "    # sample이 주어지면 주어진 대로 섞기\n",
    "    def shuffle(self, sample: list = None):\n",
    "        if sample == None:\n",
    "            shuffle(self.bars)\n",
    "        else:\n",
    "            self.bars = [self.bars[i] for i in sample]\n",
    "\n",
    "    # 막대기 고유 번호 가져오기\n",
    "    def get(self, index):\n",
    "        return self.bars[index][2]\n",
    "\n",
    "    # 정렬되었는지 검사\n",
    "    def is_sorted(self):\n",
    "        for i in range(len(self.bars)):\n",
    "            if self.bars[i][2] != 0: return False\n",
    "        return True\n",
    "\n",
    "    # 스크린에 그리기\n",
    "    def draw(self):\n",
    "        for i, bar in enumerate(self.bars):\n",
    "            x = self.pos[0] + i * self.size[0] / len(self.bars)\n",
    "            width = self.size[0] / len(self.bars)\n",
    "            if i in self.indexs:\n",
    "                color = self.palette[self.indexs.index(i) + 1]\n",
    "            else:\n",
    "                color = self.color\n",
    "            pygame.draw.rect(self.screen, color, [x, bar[0], width, bar[1]])\n",
    "        self.screen.blit(self.label, (self.pos[0], self.pos[1] - 24))\n",
    "\n",
    "class BubbleSort:\n",
    "    def __init__(self, bars: Bars):\n",
    "        self.bars = bars\n",
    "        self.step = 0\n",
    "        self.i = 0\n",
    "        self.j = 0\n",
    "    \n",
    "    def switch(self, i, j):\n",
    "        temp = self.bars.bars[i]\n",
    "        self.bars.bars[i] = self.bars.bars[j]\n",
    "        self.bars.bars[j] = temp\n",
    "\n",
    "    def done(self):\n",
    "        if self.i == len(self.bars.bars) - 1: return True\n",
    "        else: return False\n",
    "            \n",
    "\n",
    "    def next(self):\n",
    "        if not self.done():\n",
    "            if self.bars.get(self.j) > self.bars.get(self.j + 1):\n",
    "                self.switch(self.j, self.j + 1)\n",
    "            \n",
    "            self.j += 1\n",
    "            if self.j == len(self.bars.bars) - self.i - 1:\n",
    "                self.j = 0\n",
    "                self.i += 1\n",
    "        \n",
    "            self.bars.indexs = [self.i, self.j]\n",
    "        else:\n",
    "            self.bars.indexs = []\n",
    "        self.bars.draw()\n",
    "\n",
    "class SelectionSort:\n",
    "    def __init__(self, bars: Bars):\n",
    "        self.bars = bars\n",
    "        self.step = 0\n",
    "        self.i = 0\n",
    "        self.j = 1\n",
    "        self.t = 0\n",
    "    \n",
    "    def switch(self, i, j):\n",
    "        temp = self.bars.bars[i]\n",
    "        self.bars.bars[i] = self.bars.bars[j]\n",
    "        self.bars.bars[j] = temp\n",
    "\n",
    "    def done(self):\n",
    "        if self.i == len(self.bars.bars) - 1: return True\n",
    "        else: return False\n",
    "            \n",
    "\n",
    "    def next(self):\n",
    "        if not self.done():\n",
    "            if self.bars.get(self.j) < self.bars.get(self.t):\n",
    "                self.t = self.j\n",
    "            \n",
    "            self.j += 1\n",
    "            if self.j == len(self.bars.bars):\n",
    "                self.switch(self.i, self.t)\n",
    "                self.i += 1\n",
    "                self.t = self.i\n",
    "                self.j = self.i + 1\n",
    "\n",
    "            self.bars.indexs = [self.i, self.j]\n",
    "        else:\n",
    "            self.bars.indexs = []\n",
    "        self.bars.draw()\n",
    "\n",
    "class InsertionSort:\n",
    "    def __init__(self, bars: Bars):\n",
    "        self.bars = bars\n",
    "        self.step = 0\n",
    "        self.i = 1\n",
    "        self.j = 1\n",
    "    \n",
    "    def switch(self, i, j):\n",
    "        temp = self.bars.bars[i]\n",
    "        self.bars.bars[i] = self.bars.bars[j]\n",
    "        self.bars.bars[j] = temp\n",
    "\n",
    "    def done(self):\n",
    "        if self.i == len(self.bars.bars): return True\n",
    "        else: return False\n",
    "            \n",
    "    def next(self):\n",
    "        if not self.done():\n",
    "            if self.bars.get(self.j - 1) > self.bars.get(self.j):\n",
    "                self.switch(self.j - 1, self.j)\n",
    "                self.j -= 1\n",
    "            else:\n",
    "                self.i += 1\n",
    "                self.j = self.i\n",
    "            \n",
    "            if self.j == 0:\n",
    "                self.i += 1\n",
    "                self.j = self.i\n",
    "        \n",
    "            self.bars.indexs = [self.i, self.j]\n",
    "        else:\n",
    "            self.bars.indexs = []\n",
    "        self.bars.draw()\n",
    "\n",
    "class MergeSort:\n",
    "    def __init__(self, bars: Bars):\n",
    "        self.bars = bars\n",
    "        self.step = 0\n",
    "        self.i = 1\n",
    "        self.j = 1\n",
    "    \n",
    "    def switch(self, i, j):\n",
    "        temp = self.bars.bars[i]\n",
    "        self.bars.bars[i] = self.bars.bars[j]\n",
    "        self.bars.bars[j] = temp\n",
    "\n",
    "    def done(self):\n",
    "        if self.i == len(self.bars.bars): return True\n",
    "        else: return False\n",
    "            \n",
    "    def next(self):\n",
    "        if not self.done():\n",
    "            if self.bars.get(self.j - 1) > self.bars.get(self.j):\n",
    "                self.switch(self.j - 1, self.j)\n",
    "                self.j -= 1\n",
    "            else:\n",
    "                self.i += 1\n",
    "                self.j = self.i\n",
    "            \n",
    "            if self.j == 0:\n",
    "                self.i += 1\n",
    "                self.j = self.i\n",
    "\n",
    "        self.bars.draw()\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### 루프"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [],
   "source": [
    "count = 40\n",
    "sample = list(range(count))\n",
    "shuffle(sample)\n",
    "\n",
    "bars = [\n",
    "    Bars(screen, count, [70, 80], [160, 120], 0, 'BubbleSort'),\n",
    "    Bars(screen, count, [270, 80], [160, 120], 0, 'SelectionSort'),\n",
    "    Bars(screen, count, [470, 80], [160, 120], 0, 'InsertionSort'),\n",
    "]\n",
    "Sorts = [ BubbleSort, SelectionSort, InsertionSort ]\n",
    "\n",
    "sorts = []\n",
    "for i in range(len(bars)):\n",
    "    bars[i].shuffle(sample)\n",
    "    sorts.append(Sorts[i](bars[i]))\n",
    "\n",
    "done = False\n",
    "while True:\n",
    "    clock.tick(120)\n",
    "\n",
    "    for event in pygame.event.get():\n",
    "        if event.type == pygame.QUIT:\n",
    "            pygame.quit()\n",
    "            done = True\n",
    "            break\n",
    "    if done: break\n",
    "    \n",
    "    screen.fill('#FAFAFA')\n",
    "    \n",
    "    for s in sorts: s.next()\n",
    "\n",
    "    pygame.display.flip()"
   ]
  }
 ],
 "metadata": {
  "interpreter": {
   "hash": "83005d820893e402838a3c613d92db160c049d19aee0fd864abf4e10b2628dbd"
  },
  "kernelspec": {
   "display_name": "Python 3.9.7 64-bit ('py9': conda)",
   "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.9.7"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}