import sys, math, string
from OpenGL.GLUT import *
from OpenGL.GL import *
from OpenGL.GLU import *

triangleData = [ ]

cameraDist = 10
cameraXrot = 0
cameraYrot = 0

def draw():
	global cameraDist, cameraXrot, cameraYrot
	glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
	glMatrixMode(GL_PROJECTION)
	glLoadIdentity()
	gluPerspective(60, 1, 0.1, 1000)
	glMatrixMode(GL_MODELVIEW)
	glLoadIdentity()
	glTranslatef(0, 0, -cameraDist)
	glRotatef(-cameraXrot, 1, 0, 0)
	glRotatef(-cameraYrot, 0, 1, 0)

	global triangleData
	drawTriangles(triangleData)

	glutSwapBuffers()


def drawTriangles(tri):
	glBegin(GL_TRIANGLES)
	for t in tri:
		glColor3fv(t["color"])
		glVertex3fv(t["vertex"])
	glEnd()

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

def specialkey(key,x,y):
    global cameraXrot, cameraYrot
    if key == GLUT_KEY_LEFT:
        cameraYrot = cameraYrot - 3
    elif key == GLUT_KEY_RIGHT:
        cameraYrot = cameraYrot + 3
    elif key == GLUT_KEY_UP:
        cameraXrot = cameraXrot - 3
    elif key == GLUT_KEY_DOWN:
        cameraXrot = cameraXrot + 3
    glutPostRedisplay()


def loadDataFile(filename):
	triangles = [ ]
	for line in open(filename, 'r').readlines():
		values = string.split(line)
		point = { }
		color = [ string.atof(values[0]),
				  string.atof(values[1]),
				  string.atof(values[2]) ]
		point["color"] = color
		vertex = [ string.atof(values[3]),
				   string.atof(values[4]),
				   string.atof(values[5]) ]
		point["vertex"] = vertex
		triangles.append(point)
	return triangles



triangleData = loadDataFile(sys.argv[1])

glutInit([])
glutInitDisplayMode(GLUT_DOUBLE | GLUT_RGB | GLUT_DEPTH)
glutInitWindowSize(400, 400)
glutInitWindowPosition(0,0)
glutCreateWindow(sys.argv[0])
glutDisplayFunc(draw)
glutKeyboardFunc(keyboard)
glutSpecialFunc(specialkey)
glEnable(GL_DEPTH_TEST)
glutMainLoop()