import sys, time, math
from OpenGL.GL import *
from OpenGL.GLU import *
from OpenGL.GLUT import *
from dmsgl import *
import os
startdir = os.getcwd() + os.sep

camera = PerspCamera()
camera.moveBackward(15)
camera.moveUp(2)
camera.pitch(-20)
light = Light()
light.position = [1, 3, 2, 0]
material1 = Material(diffuse=Color.White)
material2 = Material(diffuse=[1, 1, 1, 0.2])
tex1 = Texture2D(startdir + 'grass.jpg', minFilter=GL_LINEAR_MIPMAP_LINEAR)
tex2 = Texture2D(startdir + 'slashes.tiff')
tex3 = Texture2D(startdir + 'carpaint.jpg')
tex4 = Texture2D(startdir + 'flowers.tiff')
blend1 = Transparency(blendSource=GL_SRC_ALPHA, blendDest = GL_ONE_MINUS_SRC_ALPHA)
blend2 = Transparency(blendSource=GL_SRC_ALPHA, blendDest = GL_ONE)
xform = SimpleTransform(rotateAxis=[1, 0.1, 0.5])

startTime = time.time()


def drawCylinder():
    xform.pushApply()
    glRotatef(90, 1, 0, 0)
    glTranslatef(0, 0, -2)
    gluCylinder(quadric, 1.5, 1.0, 6.0, 16, 1)
    xform.pop()

def drawFloor():
    material1.apply()
    tex1.apply()
    glBegin(GL_QUADS)
    glNormal3f(0, 1, 0)
    glTexCoord2f(0, 0)
    glVertex3f(-10, -4, -10)
    glTexCoord2f(16, 0)
    glVertex3f(10, -4, -10)
    glTexCoord2f(16, 16)
    glVertex3f(10, -4, 10)
    glTexCoord2f(0, 16)
    glVertex3f(-10, -4, 10)
    glEnd()
    tex1.disable()

    glDepthFunc(GL_EQUAL)
    blend1.apply()
    tex2.apply()
    glBegin(GL_QUADS)
    glNormal3f(0, 1, 0)
    glTexCoord2f(0, 0)
    glVertex3f(-10, -4, -10)
    glTexCoord2f(1, 0)
    glVertex3f(10, -4, -10)
    glTexCoord2f(1, 1)
    glVertex3f(10, -4, 10)
    glTexCoord2f(0, 1)
    glVertex3f(-10, -4, 10)
    glEnd()
    tex2.disable()
    blend1.disable()
    glDepthFunc(GL_LESS)

def draw():
    Color.X11['DeepSkyBlue'].clear()
    camera.apply()
    light.apply()

    drawFloor()

    material2.apply()
    tex3.apply()
    drawCylinder()
    tex3.disable()

    glDepthFunc(GL_EQUAL)
    blend2.apply()
    tex4.apply()
    glEnable(GL_TEXTURE_GEN_S)
    glEnable(GL_TEXTURE_GEN_T)
    glTexGeni(GL_S, GL_TEXTURE_GEN_MODE, GL_SPHERE_MAP)
    glTexGeni(GL_T, GL_TEXTURE_GEN_MODE, GL_SPHERE_MAP)
    drawCylinder()
    glDisable(GL_TEXTURE_GEN_S)
    glDisable(GL_TEXTURE_GEN_T)
    tex4.disable()
    blend2.disable()
    glDepthFunc(GL_LESS)

    glutSwapBuffers()


def keyboard(key, x, y):
    global cameraVelDist
    if key == chr(27):
        sys.exit(0)
    elif key == '.':
        cameraVelDist += 4
    elif key == ',':
        cameraVelDist += -4

def keyboardUp(key, x, y):
    global cameraVelDist
    if key == chr(27):
        sys.exit(0)
    elif key == '.':
        cameraVelDist -= 4
    elif key == ',':
        cameraVelDist -= -4

def specialkey(key,x,y):
    global cameraVelPitch, cameraVelTurn
    if key == GLUT_KEY_LEFT:
        cameraVelTurn += 90
    elif key == GLUT_KEY_RIGHT:
        cameraVelTurn -= 90
    elif key == GLUT_KEY_UP:
        cameraVelPitch += 90
    elif key == GLUT_KEY_DOWN:
        cameraVelPitch -= 90

def specialkeyUp(key,x,y):
    global cameraVelPitch, cameraVelTurn
    if key == GLUT_KEY_LEFT:
        cameraVelTurn -= 90
    elif key == GLUT_KEY_RIGHT:
        cameraVelTurn += 90
    elif key == GLUT_KEY_UP:
        cameraVelPitch -= 90
    elif key == GLUT_KEY_DOWN:
        cameraVelPitch += 90


cameraVelPitch = 0.0
cameraVelTurn = 0.0
cameraVelDist = 0.0
prevTime = time.time()

def update():
    global cameraVelPitch, cameraVelTurn, cameraVelDist
    global prevTime
    t = time.time()
    dt = t - prevTime
    prevTime = t
    camera.pitch(cameraVelPitch * dt)
    camera.turn(cameraVelTurn * dt)
    camera.moveForward(cameraVelDist * dt)
    xform.rotateAngle = (time.time()-startTime)*20
    glutPostRedisplay()

glutInit([])
glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE | GLUT_DEPTH)
glutInitWindowSize(400,400)
glutCreateWindow(sys.argv[0])
glutDisplayFunc(draw)
glutKeyboardFunc(keyboard)
glutKeyboardUpFunc(keyboardUp)
glutSpecialFunc(specialkey)
glutSpecialUpFunc(specialkeyUp)
glutIgnoreKeyRepeat(1)
glutIdleFunc(update)
glEnable(GL_DEPTH_TEST)
quadric = gluNewQuadric()
gluQuadricTexture(quadric, GL_TRUE)
glutMainLoop()