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

startDir = os.getcwd()

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


def doSelect():
    glSelectBuffer(1000)
    glRenderMode(GL_SELECT)
    glInitNames()
    glPushName(0)
    draw()
    hits = glRenderMode(GL_RENDER)
    print ""
    for hit in hits:
        minDepth, maxDepth, names = hit
        if names[0] == 1:
            print 'Square is visible',
        elif names[0] == 2:
            print 'Teapot is visible',
        elif names[0] == 3:
            print 'Sphere is visible',
        print '\t\thitdata:', hit

def draw():
    glClearColor(0, 0.3, 0.5, 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, 0.0, -viewDistance)
    glRotatef(viewRotX, 1.0, 0.0, 0.0)
    glRotatef(viewRotY, 0.0, 1.0, 0.0)

    drawSquare()
    drawTeapot()
    drawSphere()

    glutSwapBuffers()

def drawSquare():
    glLoadName(1)
    glPushMatrix()
    glTranslatef(-3, 2, 0)
    tex1.apply()
    glBegin(GL_QUADS)
    glTexCoord2f(0, 0)
    glVertex3f(-1, -1, 0)
    glTexCoord2f(2, 0)
    glVertex3f(1, -1, 0)
    glTexCoord2f(2, 2)
    glVertex3f(1, 1, 0)
    glTexCoord2f(0, 2)
    glVertex3f(-1, 1, 0)
    glEnd()
    tex1.disable()
    glPopMatrix()

def drawTeapot():
    glLoadName(2)
    glPushMatrix()
    glTranslatef(3, 0, 0)
    tex2.apply()
    glutSolidTeapot(2)
    tex2.disable()
    glPopMatrix()

def drawSphere():
    glLoadName(3)
    glPushMatrix()
    glTranslatef(0, -4, 0)
    tex3.apply()
    gluSphere(quadric,2,32,16)
    tex3.disable()
    glPopMatrix()

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

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
    global prevTime
    t = time.time()
    dt = t - prevTime
    prevTime = t
    viewRotX += cameraVelX * dt
    viewRotY += cameraVelY * dt
    glutPostRedisplay()

glutInit([])
glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE | GLUT_DEPTH)
glutInitWindowSize(400,400)
glutCreateWindow(sys.argv[0])
glutDisplayFunc(draw)
glutKeyboardFunc(keyboard)
glutSpecialFunc(specialkey)
glutSpecialUpFunc(specialkeyUp)
glutIgnoreKeyRepeat(1)
glutIdleFunc(update)
glEnable(GL_DEPTH_TEST)
os.chdir(startDir)
tex1 = Texture2D('carparts.tif')
tex2 = Texture2D('brick.jpg')
tex3 = Texture2D('earth.jpg')
quadric = gluNewQuadric()
gluQuadricTexture(quadric, GL_TRUE)
glutMainLoop()