PyOpenGL-PyGame-texture mapping to simulate rain on a rotating cylinder.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'> OpenGL for a rainy day on a cylinder in PyGame.</span>\n",
    "<br><b>Hint:</b> We use PyOpenGL to have a cylinder covered with a modifying texture in PyGame.\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": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pygame as pg\n",
    "from OpenGL.GL import *\n",
    "from OpenGL.GLU import *\n",
    "import random"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def loadTexture():\n",
    "    textureSurface = pg.image.load('images/ezhdeha2.jpg')\n",
    "    textureData = pg.image.tostring(textureSurface, \"RGBA\", 1)\n",
    "    Twidth,Theight = textureSurface.get_size()\n",
    "    glEnable(GL_TEXTURE_2D)\n",
    "    texid = glGenTextures(1)\n",
    "    glBindTexture(GL_TEXTURE_2D, texid)\n",
    "    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, Twidth, Theight,\n",
    "                 0, GL_RGBA, GL_UNSIGNED_BYTE, textureData)\n",
    "    glGenerateMipmap(GL_TEXTURE_2D);\n",
    "    glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT)\n",
    "    glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT)\n",
    "    glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST)\n",
    "    glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST)\n",
    "\n",
    "    return texid,textureSurface"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "pg.init()\n",
    "width,height=800,600\n",
    "screen=pg.display.set_mode((width,height),pg.DOUBLEBUF| pg.OPENGL)\n",
    "pg.display.set_caption('OpenGL in PyGame for texture mapping on cylinder'+\\\n",
    "                      ', by Hamed Shah-Hosseini, Twitter:@chekaad, Instagram: @ostad.ai')\n",
    "glcyn=gluNewQuadric()\n",
    "def setupRC():\n",
    "    glMatrixMode(GL_PROJECTION)\n",
    "    glLoadIdentity()\n",
    "    gluPerspective(45.0, float(width)/float(height), 0.1, 100.0)\n",
    "    glMatrixMode(GL_MODELVIEW)\n",
    "    glLoadIdentity()\n",
    "    glClearColor(.3,.2,.1,1.)\n",
    "    gluQuadricNormals(glcyn, GLU_SMOOTH) \n",
    "    gluQuadricTexture(glcyn, GL_TRUE)\n",
    "    #-----------\n",
    "    glEnable(GL_DEPTH_TEST)\n",
    "    glLightfv(GL_LIGHT0, GL_AMBIENT, (0.3, 0.3, 0.3, 1.0)) \n",
    "    glLightfv(GL_LIGHT0, GL_DIFFUSE, (1.0, 1.0, 1.0, 1.0)) \n",
    "    glLightfv(GL_LIGHT0, GL_POSITION, (0.0, 0.0, 3.0, 1.0)) \n",
    "    glLightfv(GL_LIGHT0,GL_SPECULAR,(1.0,1.0,1.0,1.0))\n",
    "    glEnable(GL_LIGHT0)\n",
    "    glEnable(GL_LIGHTING)\n",
    "    glEnable(GL_COLOR_MATERIAL)\n",
    "    glColorMaterial(GL_FRONT,GL_AMBIENT_AND_DIFFUSE)\n",
    "    glMaterialfv(GL_FRONT,GL_SPECULAR,(1.0,1.0,1.0,1.0))\n",
    "    glMateriali(GL_FRONT,GL_SHININESS,100)\n",
    "    texid,texSurface=loadTexture()\n",
    "    return texid, texSurface\n",
    "#---------------------------\n",
    "speed=.6; dx=9.; dy=13.\n",
    "N=500; xys=[];colors=[]\n",
    "def randXY():\n",
    "    return [1.7*width*random.random(),random.random()*-height/2]\n",
    "for i in range(N):\n",
    "    xys.append(randXY())\n",
    "    colors.append(random.randint(0,100))\n",
    "    \n",
    "def drawLines(surface):\n",
    "    for i in range(N):\n",
    "        x1,y1=xys[i]\n",
    "        x2,y2=x1-dx,y1+dy\n",
    "        pg.draw.line(surface,(colors[i],colors[i],255),\n",
    "                                [x1,y1],[x2,y2],width=1)\n",
    "        x1-=speed*dx\n",
    "        y1+=speed*dy\n",
    "        if x1<0 or y1>height:\n",
    "            xys[i]=randXY()\n",
    "        else:\n",
    "            xys[i]=[x1,y1]\n",
    "#----------------\n",
    "def RenderScene(angle):\n",
    "    glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT)\n",
    "    cynH=5.\n",
    "    glColor3f(.8,.4,.4)\n",
    "    glPushMatrix()\n",
    "    glTranslatef(0.,0.,-10.)\n",
    "    glTranslate(0.,0.,cynH/2.)\n",
    "    glRotatef(angle*2,1.0,0.0,0.0)\n",
    "    glRotatef(angle*4,0.0,1.0,0.0)\n",
    "    glRotatef(angle*8,0.0,0.0,1.0)\n",
    "    glTranslate(0.,0.,-cynH/2.)\n",
    "    gluCylinder(glcyn,1.25,1.25,cynH,32,32)\n",
    "    glPopMatrix()\n",
    "_,texSurface=setupRC()\n",
    "wFlag=True\n",
    "angle=0.\n",
    "while wFlag:\n",
    "    for event in pg.event.get():\n",
    "        if event.type==pg.QUIT:\n",
    "            wFlag=False\n",
    "        elif event.type==pg.KEYDOWN and event.key==pg.K_ESCAPE:\n",
    "            wFlag=False\n",
    "    RenderScene(angle)\n",
    "    angle+=.1\n",
    "    if angle>=360.:\n",
    "        angle=-360.\n",
    "    pg.display.flip()\n",
    "    pg.time.wait(10)\n",
    "    #---------------\n",
    "    texImage=texSurface.copy()\n",
    "    drawLines(texImage)\n",
    "    Tw,Th=texImage.get_size()\n",
    "    texData=pg.image.tostring(texImage, \"RGBA\", 1)\n",
    "    glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, Tw, Th, GL_RGBA, GL_UNSIGNED_BYTE, texData)   \n",
    "    #------------\n",
    "pg.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.8.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}