Serge NOEL
109 lines
4.1 KiB
Python
109 lines
4.1 KiB
Python
# This script does a simple Digitrax SIM for Decoder Ops Mode Read and Writes.
|
|
# It can be useful when testing e.g. DecoderPro using a LocoNet simulator connection.
|
|
#
|
|
# Author: Robin Becker, 2023
|
|
|
|
|
|
import java
|
|
import javax.swing
|
|
import jmri
|
|
import jmri.jmrit.automat.AbstractAutomaton as AbstractAutomaton
|
|
import jmri.jmrix.loconet.LnConstants as LnConstants
|
|
import jmri.jmrix.loconet.LocoNetMessage as LocoNetMessage
|
|
import jmri.jmrix.loconet.LocoNetListener as LocoNetListener
|
|
import time
|
|
|
|
# set the intended LocoNet connection by its index; when you have just 1 connection index = 0
|
|
connectionIndex = 0
|
|
# decoder CV value buffer
|
|
CVBuffer = [0]*256;
|
|
|
|
class MessageBuffer(object):
|
|
def __init__ (self):
|
|
self.size = 32
|
|
self.buffer = [LocoNetMessage(4)] * self.size
|
|
self.delay = [int] * self.size
|
|
self.addIndex = 0
|
|
self.getIndex = 0
|
|
self.messages = 0
|
|
|
|
def addMessage (self, lmsg):
|
|
self.buffer[self.addIndex] = lmsg
|
|
self.delay[self.addIndex] = time.time()
|
|
self.messages +=1
|
|
self.addIndex +=1
|
|
if (self.addIndex >= self.size):
|
|
self.addIndex = 0
|
|
|
|
def getMessage (self):
|
|
lmsg = self.buffer[self.getIndex]
|
|
self.messages -=1
|
|
self.getIndex +=1
|
|
if (self.getIndex >= self.size):
|
|
self.getIndex = 0
|
|
return lmsg
|
|
|
|
|
|
|
|
class LoconetOpsModeSIM(AbstractAutomaton, LocoNetListener):
|
|
|
|
def init (self):
|
|
# create loconet message buffer
|
|
self.loconetMessageBuffer = MessageBuffer()
|
|
# get LnTrafficController
|
|
self.tc = jmri.InstanceManager.getList(jmri.jmrix.loconet.LocoNetSystemConnectionMemo).get(connectionIndex).getLnTrafficController()
|
|
# register LoconetListner for Programming messages only
|
|
self.tc.addLocoNetListener(jmri.jmrix.loconet.LocoNetInterface.ALL, self)
|
|
# add sensor for clean exit
|
|
self.exitSensor = sensors.provideSensor("ISOpsModeSimExit")
|
|
self.exitSensor.setKnownState(INACTIVE)
|
|
|
|
def handle (self):
|
|
self.waitSensorActive(self.exitSensor);
|
|
self.tc.removeLocoNetListener(jmri.jmrix.loconet.LocoNetInterface.TURNOUTS, self)
|
|
return False
|
|
|
|
# LoconetListener handler
|
|
def message (self, msg):
|
|
# check if Loconet Message is an Programmer Read or Write
|
|
if (msg.getOpCode() == LnConstants.OPC_WR_SL_DATA) and (msg.getElement(2) == LnConstants.PRG_SLOT) :
|
|
# check for Ops Mode Byte Read or Write
|
|
pcmd = msg.getElement(3)
|
|
if (pcmd & LnConstants.PCMD_OPS_MODE) != 0 :
|
|
hopsa = msg.getElement(5);
|
|
lopsa = msg.getElement(6);
|
|
# accept any device address for now
|
|
cvNumHi = msg.getElement(8);
|
|
cvNumLo = msg.getElement(9);
|
|
data = msg.getElement(10);
|
|
# extract CV7 and D7 from cvNumHi
|
|
if (cvNumHi & 0x01) != 0 :
|
|
cnNumLo |= 0x80;
|
|
if (cvNumHi & 0x02) != 0 :
|
|
data |= 0x80;
|
|
# adjust bits CV9 and CV8 into proper position
|
|
temp = cvNumHi / 16
|
|
cvNumHi = (cvNumHi / 8) | (cvNumHi & 0x01);
|
|
# just set cvNumHi to 0 for now so we do CV number mod 256
|
|
cvNumHi = 0;
|
|
|
|
#send LACK with Accepted Blind code 0x40
|
|
lmsg = LocoNetMessage(4)
|
|
lmsg.setOpCode(LnConstants.OPC_LONG_ACK)
|
|
lmsg.setElement(1, LnConstants.OPC_WR_SL_DATA & 0x7f)
|
|
lmsg.setElement(2, 0x40)
|
|
self.tc.sendLocoNetMessage(lmsg)
|
|
|
|
# check if CV Write or CV Read
|
|
if (pcmd & LnConstants.PCMD_RW) != 0 :
|
|
# CV Write - update buffer
|
|
CVBuffer[cvNumLo] = data;
|
|
else :
|
|
#CV Read - send response
|
|
lmsg = LocoNetMessage(msg)
|
|
lmsg.setElement(0,LnConstants.OPC_SL_RD_DATA)
|
|
lmsg.setElement(10, CVBuffer[cvNumLo]);
|
|
self.tc.sendLocoNetMessage(lmsg)
|
|
|
|
LoconetOpsModeSIM("Loconet Ops Mode SIM").start()
|