pygame-generative art-squares-rotating-checkerboard.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<h2>Python: Generative Art</h2>\n",
    "<hr style='background-color:yellow;height:2px'>\n",
    "<b>Title:</b>\n",
    "<span style='color:blue; font-size:1.2em'> Rotating black and white squares in PyGame.</span>\n",
    "<br><b>Hint:</b> We use PyGame to generative a checkerboard with rotating blocks.\n",
    "<br><b>Author:</b> <span style='color:red; font-weight:bold;'>Hamed Shah-Hosseini</span>, Twitter:@chekaad, Instagram:@ostad.ai"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pygame"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "width,height=512,512\n",
    "nRow,nCol=8,8\n",
    "sizeX,sizeY=width//nCol,height//nRow\n",
    "pygame.init()\n",
    "screen=pygame.display.set_mode((width,height),pygame.DOUBLEBUF)\n",
    "pygame.display.set_caption('Squares with PyGame: by Hamed Shah-Hosseini, Twitter@chekaad')\n",
    "background=pygame.Surface(screen.get_size())\n",
    "background.fill((0,0,0))\n",
    "background=background.convert()\n",
    "screen.blit(background,(0,0))\n",
    "clock=pygame.time.Clock()\n",
    "wFlag=True\n",
    "rects=[]\n",
    "class Square(pygame.sprite.Sprite):\n",
    "    number=0\n",
    "    squares={}\n",
    "    def __init__(self,pos,size,color=(255,255,255)): #---pos and size are lists\n",
    "        pygame.sprite.Sprite.__init__(self,self.groups)\n",
    "        self.pos=pos.copy()\n",
    "        self.size=size.copy()\n",
    "        self.color=color\n",
    "        self.image=pygame.Surface(self.size)\n",
    "        self.image.set_colorkey((0,0,0), pygame.RLEACCEL)\n",
    "        self.image=self.image.convert_alpha()\n",
    "        self.image.fill(color)\n",
    "        self.rect=self.image.get_rect()\n",
    "        self.rect.topleft=self.pos.copy()\n",
    "        self.number=Square.number\n",
    "        Square.squares[self.number]=self\n",
    "        Square.number+=1\n",
    "        self.original=self.image\n",
    "        self.angle=0.\n",
    "        self.rFlag=True\n",
    "    def setRotFlag(self,rFlag):\n",
    "        self.rFlag=rFlag\n",
    "    def getRotFlag(self):\n",
    "        return self.rFlag\n",
    "    def update(self):\n",
    "        center=self.rect.center\n",
    "        if self.angle<180 and self.rFlag:\n",
    "            self.angle+=1\n",
    "        elif self.angle>=180:\n",
    "            self.angle=0\n",
    "            self.rFlag=False\n",
    "        rotate=pygame.transform.rotate\n",
    "        self.image=rotate(self.original,self.angle)\n",
    "        self.rect=self.image.get_rect(center=center)\n",
    "#--------------------\n",
    "class Square2(pygame.sprite.Sprite):\n",
    "    number=0\n",
    "    squares={}\n",
    "    def __init__(self,pos,size,color=(0,0,0)): #---pos and size are lists\n",
    "        pygame.sprite.Sprite.__init__(self,self.groups)\n",
    "        self.pos=pos.copy()\n",
    "        self.size=size.copy()\n",
    "        self.color=color\n",
    "        self.image=pygame.Surface(self.size)\n",
    "        self.image.set_colorkey((255,255,255), pygame.RLEACCEL)\n",
    "        self.image=self.image.convert_alpha()\n",
    "        self.image.fill(color)\n",
    "        self.rect=self.image.get_rect()\n",
    "        self.rect.topleft=self.pos.copy()\n",
    "        self.number=Square.number\n",
    "        Square.squares[self.number]=self\n",
    "        Square.number+=1\n",
    "        self.original=self.image\n",
    "        self.angle=0.\n",
    "        self.rFlag=False\n",
    "    def setRotFlag(self,rFlag):\n",
    "        self.rFlag=rFlag\n",
    "    def getRotFlag(self):\n",
    "        return self.rFlag\n",
    "    def update(self):\n",
    "        center=self.rect.center\n",
    "        if self.angle<180 and self.rFlag:\n",
    "            self.angle+=1\n",
    "        elif self.angle>=180:\n",
    "            self.angle=0\n",
    "            self.rFlag=False\n",
    "        rotate=pygame.transform.rotate\n",
    "        self.image=rotate(self.original,self.angle)\n",
    "        self.rect=self.image.get_rect(center=center)\n",
    "#--------------\n",
    "sGroup=pygame.sprite.Group()\n",
    "Square.groups=sGroup\n",
    "sGroup2=pygame.sprite.Group()\n",
    "Square2.groups=sGroup2\n",
    "\n",
    "for row in range(nRow):\n",
    "    for col in range(nCol):\n",
    "        if (row+col)%2==0:\n",
    "            sy,sx=row*sizeY,col*sizeX\n",
    "            Square([sx,sy],[sizeX,sizeY])\n",
    "        elif (row+col)%2==1:\n",
    "            sy,sx=row*sizeY,col*sizeX\n",
    "            Square2([sx,sy],[sizeX,sizeY])\n",
    "\n",
    "blackFlag=False\n",
    "while wFlag:  \n",
    "    clock.tick(30)  #----FPS=30\n",
    "    #-----------------------\n",
    "    if blackFlag==False:\n",
    "        for spr in sGroup:\n",
    "            if  not spr.getRotFlag():\n",
    "                blackFlag=True\n",
    "                for spr in sGroup2:\n",
    "                    spr.setRotFlag(True)\n",
    "                break\n",
    "        background.fill((0,0,0))\n",
    "        sGroup.update()\n",
    "        sGroup.draw(background)\n",
    "        screen.blit(background,(0,0))\n",
    "    #-------------------------------\n",
    "    else:\n",
    "        for spr in sGroup2:\n",
    "            if  not spr.getRotFlag():\n",
    "                blackFlag=False\n",
    "                for spr in sGroup:\n",
    "                    spr.setRotFlag(True)\n",
    "                break\n",
    "        background.fill((255,255,255))\n",
    "        sGroup2.update()\n",
    "        sGroup2.draw(background)\n",
    "        screen.blit(background,(0,0))\n",
    "    for event in pygame.event.get():\n",
    "        if event.type==pygame.QUIT:\n",
    "            wFlag=False\n",
    "        elif event.type==pygame.KEYDOWN:\n",
    "            if event.key==pygame.K_ESCAPE:\n",
    "                wFlag=False;\n",
    "    pygame.display.flip()\n",
    "    \n",
    "pygame.quit()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "newtor",
   "language": "python",
   "name": "newtor"
  },
  "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.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}