# This is an example script for a JMRI "Automat" in Python
#
# It listens to two sensors, running a locomotive back and 
# forth between them by changing its direction when a sensor
# detects the engine. You need to set the speed of the engine
# using a throttle.
#
# Author: Bob Jacobsen, copyright 2004, 2005
# Part of the JMRI distribution
#
# The next line is maintained by CVS, please don't change it
# $Revision: 1.8 $

import jarray
import jmri

class AutomatExample(jmri.jmrit.automat.AbstractAutomaton) :
	
	# init() is called exactly once at the beginning to do
	# any necessary configuration.
	def init(self):
		
		# get the sensor and throttle objects
		self.fwdSensor = sensors.provideSensor("12")
		self.revSensor = sensors.provideSensor("13")
		self.throttle = self.getThrottle(1234, True)  # long address 1234
		if (self.throttle == None) :
			print "Couldn't assign throttle!"
		 		
		return

	# handle() is called repeatedly until it returns false.
	#
	# Modify this to do your calculation.
	def handle(self):

		# set loco to forward
		self.throttle.setIsForward(True)
		
		# wait for sensor in forward direction to trigger
		self.waitSensorActive(self.fwdSensor)
		
		# set loco to reverse
		self.throttle.setIsForward(False)
		
		# wait for sensor inactive, meaning loco has reversed out
		# (prevent infinite loop if both sensors go active in the overlap)
		self.waitSensorInactive(self.fwdSensor)

		# wait for sensor in reverse direction to trigger
		self.waitSensorActive(self.revSensor)
		
		# and continue around again
		return 1	# to continue
	
# end of class definition

# create one of these
a = AutomatExample()

# set the name, as a example of configuring it
a.setName("Automat example script")

# and start it running
a.start()