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

startdir = os.getcwd() + os.sep

camera = PerspCamera()
camera.moveBackward(15)

light1 = Light()
light2 = Light()
light2.position = [-1, 1, -1, 0]
material = Material(diffuse=Color.White)
xform = SimpleTransform(rotateAxis=[1,1,1])
tex = Texture2D(startdir + 'grass.jpg')


texCamera = OrthoCamera(left=-1.5,right=1.5,bottom=-1.5,top=1.5)
circleTex = Texture2D(startdir + 'circle.tiff')
blend = Transparency(blendSource=GL_ONE, blendDest=GL_ONE)
copyTex = EmptyTexture2D()

circles = []
circles.append([Color([1,0,0,0.4]), 1, 0, 1, 7])
circles.append([Color([0,1,0,0.4]), 1, 1, 1.1, 8])
circles.append([Color([0,0,1,0.4]), 1, 2, 1.2, 9])
circles.append([Color([1,1,0,0.4]), 1, 3, 1.3, 10])
circles.append([Color([1,0,1,0.4]), 1.4, 4, 1, 11])
circles.append([Color([0,1,1,0.4]), 1.5, 5, 1, 12])
circles.append([Color([1,0,0,0.4]), -0.5, 0, 1, 5])
circles.append([Color([0,1,0,0.4]), -0.5, 1, 1.1, 4])
circles.append([Color([0,0,1,0.4]), -0.5, 2, 1.2, 3])
circles.append([Color([1,1,0,0.4]), 0.6, -5, 1.3, 10])
circles.append([Color([1,0,1,0.4]), 0.4, -3, 1, 11])
circles.append([Color([0,1,1,0.4]), 0.7, -4, 1, 12])
circles.append([Color([1,1,0,0.4]), 2.6, -5, -1.3, 1])
circles.append([Color([1,0,1,0.4]), 2.4, -3, -1, -1])
circles.append([Color([0,1,1,0.4]), 2.7, -4, -1, 0])


startTime = time.time()


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

    Color.X11['DeepSkyBlue'].clear()
    light1.apply()
    light2.apply()
    material.apply()

    copyTex.apply()
    xform.pushApply()
    glutSolidTeapot(3)
    xform.pop()
    copyTex.disable()

    tex.apply()
    glBegin(GL_QUADS)
    glNormal3f(0, 0, 1)
    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()
    tex.disable()
    material.disable()


def drawTexture():
    glViewport(0, 0, 256, 256)
    glScissor(0, 0, 256, 256)
    glDisable(GL_DEPTH_TEST)
    texCamera.apply()
    t = time.time() - startTime
    Color.Grey20.clear()
    texCamera.apply()
    blend.apply()
    circleTex.apply()
    for c in circles:
        c[0].apply()
        glPushMatrix()
        glTranslatef(math.sin(t*c[1]+c[2]), math.cos(t*c[3]+c[4]), 0)
        glBegin(GL_QUADS)
        glTexCoord2i(0,0)
        glVertex2f(-1.0, -1.0)
        glTexCoord2i(1,0)
        glVertex2f(1.0, -1.0)
        glTexCoord2i(1,1)
        glVertex2f(1.0, 1.0)
        glTexCoord2i(0,1)
        glVertex2f(-1.0, 1.0)
        glEnd()
        glPopMatrix()
    circleTex.disable()
    blend.disable()
    glEnable(GL_DEPTH_TEST)

    copyTex.apply()
    glCopyTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, 0, 0, 256, 256, 0)
    copyTex.disable()


def draw():
    drawTexture()
    drawWorld()
    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, xform
    global prevTime
    t = time.time()
    dt = t - prevTime
    prevTime = t
    camera.pitch(cameraVelPitch * dt)
    camera.turn(cameraVelTurn * dt)
    camera.moveForward(cameraVelDist * dt)
    xform.rotateAngle += 10 * dt
    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()