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

dir = os.getcwd() + os.sep

camera = PerspCamera()
camera.moveBackward(10)

tex = Texture2D(dir+'spot.tif', wrap=GL_CLAMP)
material1 = Material()
material2 = Material(diffuse=Color.Green)
light = Light()
light.position=[1,3,2]

startTime = time.time()

def projectTexture():
	glEnable(GL_TEXTURE_GEN_S)
	glEnable(GL_TEXTURE_GEN_T)
	glEnable(GL_TEXTURE_GEN_R)
	glEnable(GL_TEXTURE_GEN_Q)
	glTexGeni(GL_S, GL_TEXTURE_GEN_MODE, GL_EYE_LINEAR)
	glTexGeni(GL_T, GL_TEXTURE_GEN_MODE, GL_EYE_LINEAR)
	glTexGeni(GL_R, GL_TEXTURE_GEN_MODE, GL_EYE_LINEAR)
	glTexGeni(GL_Q, GL_TEXTURE_GEN_MODE, GL_EYE_LINEAR)
	glTexGenfv(GL_S, GL_EYE_PLANE, [1, 0, 0, 0])
	glTexGenfv(GL_T, GL_EYE_PLANE, [0, 1, 0, 0])
	glTexGenfv(GL_R, GL_EYE_PLANE, [0, 0, 1, 0])
	glTexGenfv(GL_Q, GL_EYE_PLANE, [0, 0, 0, 1])

	glMatrixMode(GL_TEXTURE)
	glLoadIdentity()
	glTranslatef(0.5, 0.5, 0)
	glScalef(0.5, 0.5, 1)
	gluPerspective(30.0, 1.0, 0.1, 100.0)
	glTranslatef(0, 0, -15)
	t = time.time() - startTime
	glRotatef(10*math.sin(t/4.0), 1, 0, 0)
	glRotatef(25*(math.sin(t/3.0)-1), 0, 1, 0)
	glMatrixMode(GL_MODELVIEW)

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

    projectTexture()
    tex.apply()
    material1.apply()
    glutSolidTeapot(2)
    material2.apply()
    glBegin(GL_QUADS)
    glNormal3f(0, 0, 1)
    glVertex3f(-5, -3, -5)
    glVertex3f(5, -3, -5)
    glVertex3f(5, 7, -5)
    glVertex3f(-5, 7, -5)
    glNormal3f(1, 0, 0)
    glVertex3f(-5, -3, 5)
    glVertex3f(-5, -3, -5)
    glVertex3f(-5, 7, -5)
    glVertex3f(-5, 7, 5)
    glNormal3f(0, 1, 0)
    glVertex3f(-5, -3, 5)
    glVertex3f(5, -3, 5)
    glVertex3f(5, -3, -5)
    glVertex3f(-5, -3, -5)
    glEnd()
    tex.disable()
    
    glutSwapBuffers()


def keyboard(key, x, y):
    global viewDistance
    if key == chr(27):
        sys.exit(0)
    elif key == '.':
        camera.moveForward(1)
    elif key == ',':
        camera.moveBackward(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


cameraVelX = 0.0
cameraVelY = 0.0
prevTime = time.time()

def update():
    global viewRotX, viewRotY, cameraVelX, cameraVelY
    global prevTime
    t = time.time()
    dt = t - prevTime
    prevTime = t
    camera.pitch(cameraVelX * dt)
    camera.turn(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)
glutMainLoop()