# This is an example script for a JMRI "Automat" in Python
# It is based on the AutomatonExample.
#
# It listens to two sensors, running a locomotive back and
# forth between them by changing its direction when a sensor
# detects the engine. 
#
# Author:  Howard Watkins, January 2007.
# Part of the JMRI distribution
#
# Modified by Peter Ulvestad, June 2009
# Added Mid Station stop
#
# The next line is maintained by CVS, please don't change it
# $Revision: 1.1 $

import jarray
import jmri

class FireValleyTest(jmri.jmrit.automat.AbstractAutomaton) :

	def init(self):
		# init() is called exactly once at the beginning to do
		# any necessary configuration.
		print "Inside init(self)"

		# set up sensor numbers
		# fwdSensor is reached when loco is running forward
		self.fwdSensor = sensors.provideSensor("LS5")
		self.revSensor = sensors.provideSensor("LS8")
		self.fwdstnSensor = sensors.provideSensor("LS7")
		self.revstnSensor = sensors.provideSensor("LS6")

		# get loco address. For long address change "False" to "True" 
		self.throttle = self.getThrottle(11, False)  # short address 11

		return

	def handle(self):
		# handle() is called repeatedly until it returns false.
		print "Inside handle(self)"

		# turn on Headlight
		self.throttle.setF0(True)

		# set loco to forward
		print "Set Loco Forward"
		self.throttle.setIsForward(True)

		# wait 1 second for layout to catch up, then set speed
		self.waitMsec(1000)                 
		print "Set Speed"
		self.throttle.setSpeedSetting(0.40)

		# wait for middle sensor (Station Stop) in forward direction to trigger, then stop
		print "Wait for Middle Forward Sensor"
		self.waitSensorActive(self.fwdstnSensor)
		print "Set Speed Stop - Station Stop"
		self.throttle.setSpeedSetting(0)

		# wait for Station stop, then set speed
		self.waitMsec(20000)          # wait for 20 seconds
		print "Set Speed"
		self.throttle.setSpeedSetting(0.40)

		# wait for sensor in forward direction to trigger, then stop
		print "Wait for Forward Sensor"
		self.waitSensorActive(self.fwdSensor)
		print "Set Speed Stop"
		self.throttle.setSpeedSetting(0)

		# delay for a time (remember loco could still be moving
		# due to simulated or actual inertia). Time is in milliseconds
		print "wait 20 seconds"
		self.waitMsec(20000)          # wait for 20 seconds
		
		# turn on whistle, set direction to reverse, set speed
		self.throttle.setF3(True)     # turn on whistle
		self.waitMsec(1000)           # wait for 1 seconds
		self.throttle.setF3(False)    # turn off whistle
		self.waitMsec(1000)           # wait for 1 second

		print "Set Loco Reverse"
		self.throttle.setIsForward(False)
		self.waitMsec(1000)                 # wait 1 second for Xpressnet to catch up
		print "Set Speed"
		self.throttle.setSpeedSetting(0.40)

		# wait for middle sensor (Station Stop) in reverse direction to trigger, then stop
		print "Wait for Middle Reverse Sensor"
		self.waitSensorActive(self.revstnSensor)
		print "Set Speed Stop - Station Stop"
		self.throttle.setSpeedSetting(0)

		# wait for Station stop, then set speed
		self.waitMsec(20000)          # wait for 20 seconds
		print "Set Speed"
		self.throttle.setSpeedSetting(0.40)

		# wait for sensor in reverse direction to trigger
		print "Wait for Reverse Sensor"
		self.waitSensorActive(self.revSensor)
		print "Set Speed Stop"
		self.throttle.setSpeedSetting(0)

		# delay for a time (remember loco could still be moving
		# due to simulated or actual inertia). Time is in milliseconds
		print "wait 20 seconds"
		self.waitMsec(20000)          # wait for 20 seconds

		# turn on whistle, set direction to forward, set speed
		self.throttle.setF3(True)     # turn on whistle
		self.waitMsec(1000)           # wait for 1 seconds
		self.throttle.setF3(False)    # turn off whistle
		self.waitMsec(1000)           # wait for 1 second

		# and continue around again
		print "End of Loop"
		return 1	
		# (requires JMRI to be terminated to stop - caution
		# doing so could leave loco running if not careful)

# end of class definition

# start one of these up
FireValleyTest().start()