Serge NOEL
215 lines
8.8 KiB
Python
215 lines
8.8 KiB
Python
# Use a USB device as a throttle
|
|
#
|
|
# Author: Bob Jacobsen, copyright 2008
|
|
# Part of the JMRI distribution
|
|
|
|
import jmri
|
|
import java
|
|
import java.beans
|
|
|
|
try:
|
|
#
|
|
# Set the name of the controller you're using
|
|
#
|
|
desiredControllerName = "WingMan Cordless Gamepad"
|
|
|
|
#
|
|
# Use the following if you have an absolute, analog control
|
|
# (as "slider") you'd like to use to control speed
|
|
#
|
|
componentASlider = "" # absolute proportional device for throttle
|
|
sliderAMin = 0.0 # value of componentASlider for zero speed
|
|
sliderAMax = 0.99 # value of componentASlider for max speed
|
|
|
|
#
|
|
# Use the following if you have a relative analog or digital (on/off)
|
|
# control you'd like ot use to control speed
|
|
#
|
|
componentWheel = "Y Axis" # relative device for throttle
|
|
componentWheelInverted = True # negative values is more positive speed
|
|
componentWheelReverses = True # will go through zero to change direction
|
|
componentWheelIncrement = 1 # speed slider increment per click
|
|
componentWheelUpperEdge = 0.2 # counts as a positive click if more than this
|
|
componentWheelLowerEdge = -0.2 # counts as a negative click if less than this
|
|
|
|
#
|
|
# Set the follow to buttons you want to control speed
|
|
# and direction of the locomotive
|
|
#
|
|
componentUp = "" # button, click to raise speed
|
|
componentDown = "" # button, click to raise speed
|
|
componentStop = "" # button, sets stop when clicked
|
|
componentReverse = "" # button, reverse when on, forward when off
|
|
|
|
#
|
|
# Set the following to the buttons you want
|
|
# to use for function control
|
|
#
|
|
componentF0 = "" # F0 follows this
|
|
componentF1 = "1" # F1 follows this
|
|
componentF2 = "2" # F2 follows this
|
|
componentF3 = "0" # F3 follows this
|
|
componentF4 = "5" # F4 follows this
|
|
componentF5 = "" # F5 follows this
|
|
componentF6 = "" # F6 follows this
|
|
componentF7 = "" # F7 follows this
|
|
componentF8 = "" # F8 follows this
|
|
|
|
# from here down is the code for the throttle
|
|
# Generally, you shouldn't touch it unless you're debugging a problem
|
|
|
|
# uncomment the following two lines to run from a keyboard for debug
|
|
#desiredControllerName = "Apple Internal Keyboard / Trackpad"
|
|
#componentWheel = "a"
|
|
|
|
# open a throttle window and get components
|
|
tf = jmri.jmrit.throttle.ThrottleFrameManager.instance().createThrottleFrame()
|
|
tf.toFront()
|
|
controlPanel = tf.getControlPanel()
|
|
functionPanel = tf.getFunctionPanel()
|
|
|
|
# connect to USB device
|
|
model = jmri.jmrix.jinput.TreeModel.instance()
|
|
|
|
# add listener for USB events
|
|
class TreeListener(java.beans.PropertyChangeListener):
|
|
def propertyChange(self, event):
|
|
if (event.propertyName == "Value") :
|
|
# event.oldValue is the UsbNode
|
|
#
|
|
# uncomment the following line to see controller names
|
|
# print "|"+event.oldValue.getController().toString()+"|"
|
|
#
|
|
# Select just the device (controller) we want
|
|
if (event.oldValue.getController().toString() == desiredControllerName) :
|
|
# event.newValue is the value, e.g. 1.0
|
|
# Check for desired component and act
|
|
component = event.oldValue.getComponent().toString()
|
|
value = event.newValue
|
|
#
|
|
# uncomment the following to see the entries
|
|
#print component, value
|
|
#
|
|
# uncomment the following to debug wheel (relative throttle) support
|
|
#if (component == "A" ) :
|
|
# component = componentWheel
|
|
# value = -1
|
|
#if (component == "S" ) :
|
|
# component = componentWheel
|
|
# value = 0
|
|
#if (component == "D" ) :
|
|
# component = componentWheel
|
|
# value = 1
|
|
#
|
|
# basic throttle keys
|
|
if (component == componentUp and value > 0.0) :
|
|
controlPanel.accelerate1()
|
|
if (component == componentDown and value > 0.0) :
|
|
controlPanel.decelerate1()
|
|
if (component == componentStop and value > 0.0) :
|
|
controlPanel.stop()
|
|
if (component == componentReverse) :
|
|
if (value > 0.0) :
|
|
# reverse
|
|
controlPanel.setForwardDirection(False)
|
|
print "sFD False"
|
|
else :
|
|
# forward
|
|
controlPanel.setForwardDirection(True)
|
|
print "SFD True"
|
|
|
|
# absolute throttle component
|
|
if (component == componentASlider) :
|
|
# handle speed setting input
|
|
# limit range
|
|
if (value < sliderAMin) :
|
|
value = sliderAMin
|
|
if (value > sliderAMax) :
|
|
value = sliderAMax
|
|
# convert fraction of input to speed step
|
|
fraction = (value-sliderAMin)/(sliderAMax-sliderAMin)
|
|
slider = controlPanel.getSpeedSlider()
|
|
setting = int(round(fraction*(slider.getMaximum()-slider.getMinimum()), 0))
|
|
slider.setValue(setting)
|
|
|
|
# relative throttle component
|
|
if (component == componentWheel) :
|
|
# calculate new speed
|
|
slider = controlPanel.getSpeedSlider()
|
|
isfwd = controlPanel.getIsForward()
|
|
nowsetting = slider.getValue()
|
|
if (componentWheelReverses and not isfwd) :
|
|
# code direction into value as its sign
|
|
nowsetting = -1 * nowsetting
|
|
# update to new fraction
|
|
if (not componentWheelInverted and value > componentWheelUpperEdge) :
|
|
newsetting = nowsetting+componentWheelIncrement
|
|
elif (componentWheelInverted and value > componentWheelUpperEdge) :
|
|
newsetting = nowsetting-componentWheelIncrement
|
|
elif (not componentWheelInverted and value < componentWheelLowerEdge) :
|
|
newsetting = nowsetting-componentWheelIncrement
|
|
elif (componentWheelInverted and value < componentWheelLowerEdge) :
|
|
newsetting = nowsetting+componentWheelIncrement
|
|
else :
|
|
newsetting = nowsetting # don't do anything
|
|
print "At A: ", value, nowsetting, newsetting, controlPanel.getIsForward()
|
|
self.setdirection = False
|
|
if (componentWheelReverses) :
|
|
# negative values mean go in reverse
|
|
if (newsetting > 0 and not isfwd ) :
|
|
self.setdirection = True
|
|
self.newdirection = True
|
|
print "Set FWD"
|
|
elif (newsetting < 0 ) :
|
|
if (isfwd) :
|
|
self.setdirection = True
|
|
self.newdirection = False
|
|
print "Set RVRS"
|
|
newsetting = -1 * newsetting
|
|
else :
|
|
# do not drop down through zero
|
|
if (newsetting < 0 ) :
|
|
newsetting = 0;
|
|
# truncate
|
|
if (newsetting < slider.getMinimum()) :
|
|
newsetting = slider.getMinimum()
|
|
if (newsetting > slider.getMaximum()) :
|
|
newsetting = slider.getMaximum()
|
|
# store the result as a scaled speed
|
|
slider.setValue(newsetting)
|
|
if (self.setdirection) :
|
|
controlPanel.setForwardDirection(self.newdirection)
|
|
print "At Z:", value, nowsetting, newsetting, controlPanel.getIsForward()
|
|
|
|
# function keys
|
|
if (component == componentF0) :
|
|
functionPanel.getFunctionButtons()[0].setSelected(value>0.5)
|
|
if (component == componentF1) :
|
|
functionPanel.getFunctionButtons()[1].setSelected(value>0.5)
|
|
if (component == componentF2) :
|
|
functionPanel.getFunctionButtons()[2].setSelected(value>0.5)
|
|
if (component == componentF3) :
|
|
functionPanel.getFunctionButtons()[3].setSelected(value>0.5)
|
|
if (component == componentF4) :
|
|
functionPanel.getFunctionButtons()[4].setSelected(value>0.5)
|
|
if (component == componentF5) :
|
|
functionPanel.getFunctionButtons()[5].setSelected(value>0.5)
|
|
if (component == componentF6) :
|
|
functionPanel.getFunctionButtons()[6].setSelected(value>0.5)
|
|
if (component == componentF7) :
|
|
functionPanel.getFunctionButtons()[7].setSelected(value>0.5)
|
|
if (component == componentF8) :
|
|
functionPanel.getFunctionButtons()[8].setSelected(value>0.5)
|
|
return
|
|
|
|
model.addPropertyChangeListener(TreeListener())
|
|
except:
|
|
import javax.swing.JOptionPane as JOptionPane
|
|
import javax.swing.JFrame as JFrame
|
|
JOptionPane.showMessageDialog(JFrame(),
|
|
"""This code is no longer maintained.
|
|
|
|
Please use Jynstrument jython/Jynstrument/ThrottleWindowToolBar/USBThrottle.jyn instead
|
|
Or jython/USBThrottleAsJynstrument.py script""", "Outdated code", JOptionPane.WARNING_MESSAGE)
|
|
|