import Sofa
import sys
import os

############################################################################################
# this is a PythonScriptController example script
############################################################################################


print("CURRENT PATH IS: "+os.getcwd())

############################################################################################
# following defs are used later in the script
############################################################################################


def testNodes(node):
	node.findData('name').value = "ROOT"

	# Node creation
	adam = node.createChild('Adam')
	eve = node.createChild('Eve')
	abel = eve.createChild('Abel')

	#you can animate simulation directly by uncommenting the following line:
	#node.animate=true

	return 0


# Python version of the "oneParticleSample" in cpp located in applications/tutorials/oneParticle
def oneParticleSample(node):
	node.findData('name').value='oneParticleSample'
	node.findData('gravity').value=$GRAVITYPY
	solver = node.createObject('EulerExplicitSolver',printLog='false')
	particule_node = node.createChild('particle_node')
	particle = particule_node.createObject('MechanicalObject')
	particle.resize(1)
	mass = particule_node.createObject('UniformMass',totalMass=1)

	return 0



############################################################################################
# following defs are optionnal entry points, called by the PythonScriptController component;
############################################################################################


class MyController(Sofa.PythonScriptController):
	# called once the script is loaded
	def onLoaded(self,node):
		self.counter = 0
		print 'Controller script loaded from node %s'%node.findData('name').value
		return 0

	# optionnally, script can create a graph...
	def createGraph(self,node):
		print 'createGraph called (python side)'

		#uncomment to create nodes
		#testNodes(node)

		#uncomment to create the "oneParticle" sample
		oneParticleSample(node)

		return 0



	# called once graph is created, to init some stuff...
	def initGraph(self,node):
		print 'initGraph called (python side)'
		return 0

	def bwdInitGraph(self,node):
		print 'bwdInitGraph called (python side)'
		sys.stdout.flush()
		return 0



	# called on each animation step
	total_time = 0
	def onBeginAnimationStep(self,dt):
		self.total_time += dt
		#print 'onBeginAnimatinStep (python) dt=%f total time=%f'%(dt,self.total_time)
		return 0

	def onEndAnimationStep(self,dt):
		sys.stdout.flush()
		return 0

	# called when necessary by Sofa framework... 
	def storeResetState(self):
		print 'storeResetState called (python side)'
		sys.stdout.flush()
		return 0

	def reset(self):
		print 'reset called (python side)'
		sys.stdout.flush()
		return 0

	def cleanup(self):
		print 'cleanup called (python side)'
		sys.stdout.flush()
		print("Matrix: ")
		
		saveMatrix("toto1.mtx", matrix)
		return 0


	# called when a GUIEvent is received
	def onGUIEvent(self,controlID,valueName,value):
		print 'GUIEvent received: controldID='+controlID+' valueName='+valueName+' value='+value
		sys.stdout.flush()
		return 0 



	# key and mouse events; use this to add some user interaction to your scripts 
	def onKeyPressed(self,k):
		print 'onKeyPressed '+k
		sys.stdout.flush()
		return 0 

	def onKeyReleased(self,k):
		print 'onKeyReleased '+k
		sys.stdout.flush()
		return 0 

	def onMouseButtonLeft(self,x,y,pressed):
		print 'onMouseButtonLeft x='+str(x)+' y='+str(y)+' pressed='+str(pressed)
		sys.stdout.flush()
		return 0

	def onMouseButtonRight(self,x,y,pressed):
		print 'onMouseButtonRight x='+str(x)+' y='+str(y)+' pressed='+str(pressed)
		sys.stdout.flush()
		return 0

	def onMouseButtonMiddle(self,x,y,pressed):
		print 'onMouseButtonMiddle x='+str(x)+' y='+str(y)+' pressed='+str(pressed)
		sys.stdout.flush()
		return 0

	def onMouseWheel(self,x,y,delta):
		print 'onMouseButtonWheel x='+str(x)+' y='+str(y)+' delta='+str(delta)
		sys.stdout.flush()
		return 0


	# called at each draw (possibility to use PyOpenGL)
	def draw(self):
		if self.counter < 10:
			print 'draw ('+str(self.counter+1)+' / 10)'
			self.counter += 1
		sys.stdout.flush()

 
