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

camera = OrthoCamera(-25,25, 0,50, -25,25)

light = Light()

GravityAccel = Vector([0, -9.8, 0])
AirDrag = 1
BounceCoeff = 0.8

class Object:
	def __init__(self, position=Vector([0,0,0]), velocity=Vector([0,0,0]), mass=1.0, radius=1.0, material=Material()):
		self.point = PointMass(position,velocity,mass)
		self.radius = radius
		self.material = material
	def draw(self):
		self.material.apply()
		glPushMatrix()
		glTranslatef(self.point.position[0], self.point.position[1], self.point.position[2])
		gluSphere(quadric, self.radius, 16, 8)
		glPopMatrix()
		self.material.disable()
	def calculateForce(self):
		self.point.clearForce()
		self.point.addForce(self.point.mass * GravityAccel)
		self.point.addForce(self.point.velocity * -AirDrag)
	def update(self,dt):
		self.point.update(dt)
		if self.point.position[1] <= 0:
			self.point.velocity[1] = -self.point.velocity[1] * 0.8
			self.point.position[1] = -self.point.position[1]
		if self.point.position[0] < -25:
			self.point.velocity[0] = -self.point.velocity[0]
			self.point.position[0] = -25 - (self.point.position[0] + 25)
		if self.point.position[0] > 25:
			self.point.velocity[0] = -self.point.velocity[0]
			self.point.position[0] = 25 - (self.point.position[0] - 25)

num = 10

objects = []
for i in range(0,num):
	objects.append(Object(position=Vector([i*14.0/num, 30, 0]), mass=5.0, radius=0.1, material=Material(diffuse=Color.Green)))
objects[0].point.mobile = False
objects[9].radius = 3
objects[9].point.mass = 50

objects.append(Object(position=Vector([7,1,0]), mass=5.0, radius=5.0, material=Material(diffuse=Color.Red)))

springs = []
for i in range(0,num-1):
	springs.append(Spring(objects[i].point,objects[i+1].point,k=300))

collideSpring = Spring(objects[9].point, objects[10].point, k=100)

active = False

def draw():
    Color.Black.clear()
    camera.apply()

    light.apply()
    for o in objects: o.draw()
    for s in springs: s.draw()
    
    glutSwapBuffers()


def keyboard(key, x, y):
    global active
    if key == chr(27):
        sys.exit(0)
    elif key == ' ':
        active = not active
    elif key == 'r':
        objects[0].point.mobile = True

prevTime = time.time()

def checkCollide():
	if objects[9].point.position.distance(objects[10].point.position) < objects[9].radius + objects[10].radius:
#		objects[10].point.force = objects[9].point.force
#		objects[9].point.clearForce()
#		objects[9].point.velocity = -objects[9].point.velocity
		collideSpring.restLength = objects[9].radius + objects[10].radius
		collideSpring.calculateForce()

def update():
    global prevTime, active
    t = time.time()
    dt = t - prevTime
    prevTime = t
    if active:
            for o in objects: o.calculateForce()
            for s in springs: s.calculateForce()
            checkCollide()
            for o in objects: o.update(dt)
    glutPostRedisplay()

glutInit([])
glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE | GLUT_DEPTH)
glutInitWindowSize(400,400)
glutCreateWindow(sys.argv[0])
glutDisplayFunc(draw)
glutKeyboardFunc(keyboard)
glutIdleFunc(update)
glEnable(GL_DEPTH_TEST)
quadric = gluNewQuadric()
glutMainLoop()