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

cubeFaces = [ GL_TEXTURE_CUBE_MAP_POSITIVE_X_ARB, GL_TEXTURE_CUBE_MAP_POSITIVE_Y_ARB, GL_TEXTURE_CUBE_MAP_POSITIVE_Z_ARB, GL_TEXTURE_CUBE_MAP_NEGATIVE_X_ARB, GL_TEXTURE_CUBE_MAP_NEGATIVE_Y_ARB, GL_TEXTURE_CUBE_MAP_NEGATIVE_Z_ARB ]
cubeCamera = PerspCamera(fovy=90)

cubeTex = TextureCube(256, 256)


camera = PerspCamera()
camera.moveBackward(15)
light = Light()
material1 = Material(diffuse=[1, 0.8, 0.1, 1])
material2 = Material(diffuse=[0, 0.5, 0.6, 1])
material3 = Material(diffuse=[1, 1, 0.2, 0.2])
xform1 = SimpleTransform(translate=[5, 0, 0], rotateAxis=[0.2, 1, 0.5])
xform2 = SimpleTransform(translate=[-5, 0, 3])
xform3 = SimpleTransform(rotateAxis=[1, -1, 0.1])
tex = Texture2D(startdir + 'grass.jpg')
blend = Transparency(blendSource=GL_SRC_ALPHA, blendDest = GL_ONE_MINUS_SRC_ALPHA)


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

    tex.apply()
    glBegin(GL_QUADS)
    glTexCoord2f(0, 0)
    glVertex3f(-100, -3, -100)
    glTexCoord2f(1, 0)
    glVertex3f(100, -3, -100)
    glTexCoord2f(1, 1)
    glVertex3f(100, -3, 100)
    glTexCoord2f(0, 1)
    glVertex3f(-100, -3, 100)
    glEnd()
    tex.disable()

    material1.apply()
    xform1.pushApply()
    glutSolidCube(2.0)
    xform1.pop()

    material2.apply()
    xform2.pushApply()
    glutSolidSphere(2.0, 16, 16)
    xform2.pop()
    material2.disable()



def draw():
    firstFrame = False
    glViewport(0, 0, 256, 256)
    glScissor(0, 0, 256, 256)
    for c in cubeFaces:
        cubeCamera.apply()
        if c == GL_TEXTURE_CUBE_MAP_POSITIVE_X_ARB:
            glRotatef(180, 0, 0, 1)
            glRotatef(90, 0, 1, 0)
        elif c == GL_TEXTURE_CUBE_MAP_POSITIVE_Y_ARB:
            glRotatef(-90, 1, 0, 0)
        elif c == GL_TEXTURE_CUBE_MAP_POSITIVE_Z_ARB:
            glRotatef(180, 0, 1, 0)
            glRotatef(180, 0, 0, 1)
        elif c == GL_TEXTURE_CUBE_MAP_NEGATIVE_X_ARB:
            glRotatef(180, 0, 0, 1)
            glRotatef(-90, 0, 1, 0)
        elif c == GL_TEXTURE_CUBE_MAP_NEGATIVE_Y_ARB:
            glRotatef(90, 1, 0, 0)
        elif c == GL_TEXTURE_CUBE_MAP_NEGATIVE_Z_ARB:
            glRotatef(180, 0, 0, 1)
        drawWorld()
        cubeTex.copyImage(c)

    w = glutGet(GLUT_WINDOW_WIDTH)
    h = glutGet(GLUT_WINDOW_HEIGHT)
    glViewport(0, 0, w, h)   
    glScissor(0, 0, w, h)
    camera.apply()
    drawWorld()

    xform3.pushApply()
    glEnable(GL_TEXTURE_GEN_S)
    glEnable(GL_TEXTURE_GEN_T)
    glEnable(GL_TEXTURE_GEN_R)
    glTexGeni(GL_S, GL_TEXTURE_GEN_MODE, GL_REFLECTION_MAP_ARB)
    glTexGeni(GL_T, GL_TEXTURE_GEN_MODE, GL_REFLECTION_MAP_ARB)
    glTexGeni(GL_R, GL_TEXTURE_GEN_MODE, GL_REFLECTION_MAP_ARB)
    cubeTex.apply()
#    glutSolidSphere(1.0, 32, 16)
    glutSolidTeapot(2.0)
    cubeTex.disable()
    glDisable(GL_TEXTURE_GEN_S)
    glDisable(GL_TEXTURE_GEN_T)
    glDisable(GL_TEXTURE_GEN_R)

    glDepthFunc(GL_EQUAL)
    material3.apply()
    blend.apply()
    glutSolidTeapot(2.0)
    blend.disable()
    material3.disable()
    glDepthFunc(GL_LESS)
    xform3.pop()

    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()
startTime = 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)
    xform1.rotateAngle += dt * 80
    t = time.time() - startTime
    xform1.translate[1] = abs(math.sin(t*2))*10 - 3
    xform2.translate[1] = abs(math.sin(t/1.5))*7 - 3
    xform3.rotateAngle += dt * 30
    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)
glEnable(GL_SCISSOR_TEST)
glutMainLoop()