import sys, time, math, Image
from OpenGL.GL import *
from OpenGL.GLU import *
from OpenGL.GLUT import *

ballPos = [0, 0, -5]

viewDistance = 15.0
viewRotY = 0.0
viewRotX = 0.0
cameraVelX = 0.0
cameraVelY = 0.0
prevTime = time.time()


def draw():
    glClearColor(0.5, 0.7, 1, 0)
    glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
    
    glMatrixMode(GL_PROJECTION)
    glLoadIdentity()
    gluPerspective(50.0, 1.0, 1.0, 100.0)
    glMatrixMode(GL_MODELVIEW)

    glLoadIdentity()
    glTranslatef(0, 0, -viewDistance)
    glRotatef(viewRotX, 1, 0, 0)
    glRotatef(viewRotY, 0, 1, 0)
    glTranslatef(0, -4, 0)

    glEnable(GL_LIGHTING)
    glLightfv(GL_LIGHT0, GL_POSITION, [0, 20, 0, 1])
    
    drawFloor()

    glMaterialfv(GL_FRONT, GL_DIFFUSE, [0.1, 0.7, 0.2])
    glPushMatrix()
    glTranslatef(ballPos[0], ballPos[1], ballPos[2])
    gluSphere(quadric, 1.0, 32, 16)
    glPopMatrix()

    glDisable(GL_LIGHTING)
    glEnable(GL_BLEND)
    glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
    glColor4f(0, 0, 0, 0.6)
    glPushMatrix()
    glTranslatef(20*ballPos[0]/(20-ballPos[1]), 0.01, 20*ballPos[2]/(20-ballPos[1]))
    glRotatef(90.0, 1, 0, 0)
    gluDisk(quadric, 0, 1, 32, 1)
    glPopMatrix()
    glDisable(GL_BLEND)

    glutSwapBuffers()


textureID = 0

def drawFloor():
    glEnable(GL_TEXTURE_2D)
    glBindTexture(GL_TEXTURE_2D, textureID)
    glMaterialfv(GL_FRONT, GL_DIFFUSE, [1,1,1,1])
    glBegin(GL_QUADS)
    glNormal3f(0, 1, 0)
    glTexCoord2f(0,0)
    glVertex3f(-20, 0, -20)
    glTexCoord2f(1,0)
    glVertex3f(20, 0, -20)
    glTexCoord2f(1,1)
    glVertex3f(20, 0, 20)
    glTexCoord2f(0,1)
    glVertex3f(-20, 0, 20)
    glEnd()
    glBindTexture(GL_TEXTURE_2D, 0)
    glDisable(GL_TEXTURE_2D)


import os
startDir = os.getcwd() + os.sep

def initTexture(filename):
    img = Image.open(startDir + filename)
    img = img.transpose(Image.FLIP_TOP_BOTTOM)
    img = img.convert('RGB')
    global textureID
    textureID = glGenTextures(1)
    glBindTexture(GL_TEXTURE_2D, textureID)
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, img.size[0], img.size[1],
                 0, GL_RGB, GL_UNSIGNED_BYTE, img.tostring())
    glBindTexture(GL_TEXTURE_2D, 0)


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

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

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

def update():
    global viewRotX, viewRotY, cameraVelX, cameraVelY, ballPos
    global prevTime
    t = time.time()
    dt = t - prevTime
    prevTime = t
    viewRotX += cameraVelX * dt
    viewRotY += cameraVelY * dt
    ballPos[0] = math.sin(time.time()) * 5
    ballPos[2] = math.cos(time.time()) * 5
    ballPos[1] = abs(math.sin(time.time()*2)) * 8
    glutPostRedisplay()

glutInit([])
glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE | GLUT_DEPTH)
glutInitWindowSize(500,500)
glutCreateWindow(sys.argv[0])
glutDisplayFunc(draw)
glutKeyboardFunc(keyboard)
glutSpecialFunc(specialkey)
glutSpecialUpFunc(specialkeyUp)
glutIgnoreKeyRepeat(1)
glutIdleFunc(update)
glEnable(GL_DEPTH_TEST)
glEnable(GL_LIGHT0)
initTexture("floor.jpg")
quadric = gluNewQuadric()
glutMainLoop()