from javax.swing import JTable, JScrollPane, JFrame, JPanel, JComboBox, BorderFactory, DefaultCellEditor, JLabel, UIManager, SwingConstants, JFileChooser
from javax.swing.table import TableCellRenderer, DefaultTableCellRenderer
from java.awt.event import MouseAdapter,MouseEvent, WindowListener, WindowEvent
from java.awt import GridLayout, Dimension, BorderLayout, Color
from javax.swing.table import AbstractTableModel, DefaultTableModel
from javax.swing.event import TableModelListener, TableModelEvent
from javax.swing.filechooser import FileNameExtensionFilter
from org.apache.commons.io import FilenameUtils
from java.io import File
import os
global g
class ResetButtonMaster(jmri.jmrit.automat.AbstractAutomaton):
# monitors station buttons and
# monitors sensors_requiring_use_of_station_buttons
# "setDispatchSensor", "setRouteSensor", "setStoppingDistanceSensor",
# "setStationWaitTimeSensor", "setStationDirectionSensor"
button_sensors_to_watch = []
count = 0
def __init__(self):
self.logLevel = 0
self.dm = DispatchMaster()
self.od = OptionDialog()
def init(self):
if self.logLevel > 0: print 'Create ResetButtonMaster Thread'
def setup(self):
if self.logLevel > 0: print "starting ResetButtonMaster setup"
#get dictionary of buttons self.button_dict
self.get_buttons()
self.get_sensors_requiring_use_of_station_buttons()
self.get_route_run_button()
self.get_set_stopping_length_button()
self.get_station_wait_time_button()
self.get_station_direction_button()
#set all move_to buttons inactive
for sensor in self.button_sensors:
if sensor != None:
sensor.setKnownState(INACTIVE)
for sensor in self.sensors_requiring_use_of_station_buttons:
if sensor == None: print "DispatcherSystem panel is out of date\nDelete DispatcherSystem panel and rerun Dispatcher System Script. "
sensor.setKnownState(INACTIVE)
for sensor in self.route_run_sensor:
sensor.setKnownState(INACTIVE)
for sensor in self.stopping_distance_sensor:
sensor.setKnownState(INACTIVE)
for sensor in self.station_wait_time_sensor:
sensor.setKnownState(INACTIVE)
for sensor in self.station_direction_sensor:
sensor.setKnownState(INACTIVE)
self.button_sensors_to_watch = self.route_run_sensor + self.button_sensors + self.sensors_requiring_use_of_station_buttons
if self.logLevel > 0: print "self.button_sensors_to_watch_init", [sensor.getUserName() for sensor in self.button_sensors_to_watch]
self.sensor_active = None
self.sensor_active_sensors_requiring_use_of_station_buttons = None
self.sensor_active_run_dispatch = None
self.sensor_active_old = None
self.sensor_active_sensors_requiring_use_of_station_buttons_old = None
if self.logLevel > 0: print "finished ResetButtonMaster setup"
return True
def handle(self):
if self.logLevel > 1: print "in resetbuttonmaster", [ str(sensor.getUserName()) for sensor in self.button_sensors_to_watch if sensor in self.sensors_requiring_use_of_station_buttons]
global setAllStoppingSensors
global stopping_sensor_choice
#wait for a sensor to go active
button_sensors_to_watch_JavaList = java.util.Arrays.asList(self.button_sensors_to_watch)
self.waitSensorState(button_sensors_to_watch_JavaList, ACTIVE)
#determine which one changed
if self.logLevel > 0: print "self.button_sensors_to_watch",self.button_sensors_to_watch
sensor_active_all_array = [sensor for sensor in self.button_sensors_to_watch if sensor.getKnownState() == ACTIVE]
if len(sensor_active_all_array) > 0:
sensor_changed = sensor_active_all_array[0] # there should be only one or zero items in this array, and that not in self.setup_route_or_run_dispatch_sensors
else:
sensor_changed = None
#reset button_sensors_to_watch
self.button_sensors_to_watch = self.route_run_sensor + self.button_sensors + self.sensors_requiring_use_of_station_buttons
# 1) modify button_sensors_to_watch so we don't keep triggering same sensor active
# 2) perform the correct action if a new button has been triggered
# note we have to see whether a new sensor has been triggered by looking at old values
# print statements
if self.logLevel > 0: print "sensor_active_all_array" , [s.getUserName() for s in sensor_active_all_array if sensor_active_all_array != None]
if self.logLevel > 0: print "self.sensor_active_route_dispatch_old" , self.sensor_active_sensors_requiring_use_of_station_buttons_old
#if self.logLevel > 0: print "self.sensor_active_route_dispatch" , self.sensor_active_sensors_requiring_use_of_station_buttons
#if self.logLevel > 0: print "self.sensor_active_route_dispatch", self.sensor_active_sensors_requiring_use_of_station_buttons.getUserName()
if self.logLevel > 0: print "self.sensor_active", self.sensor_active
if self.logLevel > 0: print "self.sensor_active_old", self.sensor_active_old
if self.logLevel > 0: print "stopping_sensor_choice", stopping_sensor_choice
# if self.sensor_active_setuproute_or_rundispatch_or_stoppinglength_old != None:
# self.button_sensors_to_watch.remove(self.sensor_active_setuproute_or_rundispatch_or_stoppinglength_old)
if sensor_changed in self.button_sensors:
#we only want to process if previous button != current buttom
self.sensor_active = sensor_changed
if self.sensor_active != self.sensor_active_old :
self.process_button_sensors(self.sensor_active)
self.sensor_active_old = self.sensor_active
#dont trigger this one again
self.button_sensors_to_watch.remove(self.sensor_active)
if self.sensor_active != self.sensor_active_old:
self.button_sensors_to_watch.append(self.sensor_active_old)
elif sensor_changed in self.route_run_sensor:
#we want to process in all circumstances
self.process_run_route()
elif sensor_changed in self.sensors_requiring_use_of_station_buttons:
#we want to process in all circumstances
if self.logLevel > 0: print "sensor in sensor_active_all_array", sensor.getUserName()
#self.sensor_active_sensor_requiring_use_of_station_buttons = sensor_changed
# if self.sensor_active_setuproute_or_rundispatch_or_stoppinglength != None and \
# self.sensor_active_setuproute_or_rundispatch_or_stoppinglength != \
# self.sensor_active_setuproute_or_rundispatch_or_stoppinglength_old:
#
# self.process_setuproute_or_rundispatch_or_setstoppinglength_sensors()
#
# self.sensor_active_setuproute_or_rundispatch_or_stoppinglength_old = \
# self.sensor_active_setuproute_or_rundispatch_or_stoppinglength
self.process_sensors_requiring_use_of_station_buttons(sensor_changed)
#print statements
if self.logLevel > 0: print "removing ", self.sensor_active_sensors_requiring_use_of_station_buttons_old
if self.sensor_active_sensors_requiring_use_of_station_buttons_old == None:
if self.logLevel > 0: print "self.sensor_active_setuproute_or_rundispatch_or_stoppinglength", "None"
else:
if self.logLevel > 0: print "self.sensor_active_setuproute_or_rundispatch_or_stoppinglength", \
self.sensor_active_sensors_requiring_use_of_station_buttons_old.getUserName()
if self.logLevel > 0: print "returning"
return True
def transit_restrictions(self, null_text):
folder = "restrictTransits"
filename = "restrictTransits.txt"
restricted_transits = DispatchMaster().read_list(folder,filename)
if restricted_transits == ["",""]:
restricted_transits = []
print "restricted_transits", restricted_transits
return restricted_transits
def switch_sensors_requiring_station_buttons(self, sensor, mode):
sensor_name = sensor.getUserName()
# if sensor remains on after being switched on, the sensor must be removed from the list of sensors we are waiting
# for.
# if the sensor is switched off, the sensor must be in the list of sensors we are waiting for
if mode == "sensor_on":
#reset self.sensors_requiring_use_of_station_buttons to full list
self.get_sensors_requiring_use_of_station_buttons()
[s.setKnownState(INACTIVE) for s in self.sensors_requiring_use_of_station_buttons]
#turn sensor on
sensor.setKnownState(ACTIVE)
# remove from list of sensors we are waiting for
if sensor in self.sensors_requiring_use_of_station_buttons:
self.sensors_requiring_use_of_station_buttons.remove(sensor)
# allow the button to be pressed again
self.sensor_active_sensors_requiring_use_of_station_buttons_old = None
elif mode == "sensor_off":
# turn sensor off
if self.logLevel > 1: print "turn sensor off", sensor.getUserName()
sensor.setKnownState(INACTIVE)
#reset self.sensors_requiring_use_of_station_buttons to full list
self.get_sensors_requiring_use_of_station_buttons()
# allow the button to be pressed again ##### inhibit the same sensor being pressed again
# self.sensor_active_sensors_requiring_use_of_station_buttons_old = None
self.get_sensors_requiring_use_of_station_buttons()
else:
#error
if self.logLevel > 1: print ("!!!!! Error in switch_sensors_requiring_station_buttons: contact developer!!!!!!")
def process_button_sensors(self, sensor_changed):
[sensor.setKnownState(INACTIVE) for sensor in self.button_sensors if sensor != sensor_changed]
def process_sensors_requiring_use_of_station_buttons(self, sensor_changed):
# all these sensors when active require the use of station buttons. Only one of these can be active at any one
# time, hence the grouping of these in a routine and the grouping of the buttons on the panel
global stopping_sensor_choice
self.count += 1
if self.logLevel > 0: print "sensor_active_setuproute_or_rundispatch_or_stoppinglength1", \
self.sensor_active_sensors_requiring_use_of_station_buttons
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
if sensor_changed == sensors.getSensor("setDispatchSensor"):
#stopping_sensor_choice = None
#sensors.getSensor("setRouteSensor").setKnownState(INACTIVE)
#sensors.getSensor("setStoppingDistanceSensor").setKnownState(INACTIVE)
msg = "Press section buttons to set dispatch \nA train needs to be set up in a section first"
self.od.displayMessage(msg)
if self.od.CLOSED_OPTION == True:
if self.logLevel > 0: print "closed option"
#make so can select DispatchSensor again
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
else:
# if self.sensor_active_sensors_requiring_use_of_station_buttons_old == None:
# # set old sensor non-null to make sure sensor setDispatchSensor remains active
# self.sensor_active_sensors_requiring_use_of_station_buttons_old = sensors.getSensor("setDispatchSensor")
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
elif sensor_changed == sensors.getSensor("setRouteSensor"):
#stopping_sensor_choice = None
#sensors.getSensor("setStoppingDistanceSensor").setKnownState(INACTIVE)
#sensors.getSensor("setDispatchSensor").setKnownState(INACTIVE)
msg = "Press station buttons to set route \n\nThe route may then be run directly or\nused to schedule a train"
self.od.displayMessage(msg)
if self.od.CLOSED_OPTION == True:
if self.logLevel > 0: print "closed option"
#make so can select RouteSensor again
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")#
self.get_sensors_requiring_use_of_station_buttons()
return
else:
# if self.sensor_active_sensors_requiring_use_of_station_buttons_old == None:
# # set old sensor non-null to make sure sensor setRouteSensor remains active
# self.sensor_active_sensors_requiring_use_of_station_buttons_old = sensors.getSensor("setRouteSensor")
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
elif sensor_changed == sensors.getSensor("setStoppingDistanceSensor"):
return_value = False
while(return_value == False):
stopping_distance_list = DispatchMaster().get_stopping_positions()
stopping_distance_list_desc = DispatchMaster().get_stopping_positions_desc()
message = ""
message += "overall stopping distance before stop point = " + str(int(self.get_overall_stopping_distance())) + " cm
"
message += "
"
message += "All stopping distances reduced by " + str(int(self.get_overall_stopping_distance())) + " cm unless indicated below:
"
message += ""
#optionbox
title = "Stopping distances?"
opt1 = "Change all stopping distances"
opt4 = "Change above selected stopping distance"
opt2 = "Set new stopping distance for particular transit"
opt3 = "Cancel"
if stopping_distance_list == []:
options = [opt1, opt2, opt3]
else:
options = [opt1, opt2, opt4, opt3]
[list_value, s, index] = self.od.ListOptions(stopping_distance_list_desc, title, options, message = message, preferred_size = "default")
if self.od.CLOSED_OPTION == True or s == opt3: #check of optionbox was closed prematurely
#stopping_sensor_choice = "setNoStoppingSensors"
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return_value = True
elif s == opt1:
#stopping_sensor_choice = "setAllStoppingSensors"
self.modify_all_stopping_distances()
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
elif s == opt2:
#opt2
stopping_sensor_choice = "setIndividualStoppingSensors"
msg = "Specify the transit whose stopping distance we will change\n\n" + \
"Press transit start station button\nthen the transit end station button\n" + \
"to select the transit in order to\nset stopping length"
self.od.displayMessage(msg)
if self.od.CLOSED_OPTION == True:
#stopping_sensor_choice = "setNoStoppingSensors"
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return_value = True
else:
if sensors.getSensor("stoppingDistanceSet") == None:
self.od.displayMessage("need to regenerate. virtual sensor required for this option is not created")
return_value = True
else:
sensors.getSensor("stoppingDistanceSet").setKnownState(INACTIVE)
self.waitForSensorSet("stoppingDistanceSet") # wait till we complete the action using the station buttons and the sensor is set to active again
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
elif s == opt4:
# print stopping_distance_list
found_edge_str = stopping_distance_list[index][0]
first_station, last_station = found_edge_str.strip("()").split(":")
first_station = first_station.strip()
last_station = last_station.strip()
# print "first_station", first_station, "last_station", last_station
for e in g.g_express.edgeSet():
from_station_name = g.g_stopping.getEdgeSource(e)
to_station_name = g.g_stopping.getEdgeTarget(e)
# print "from_station_name", from_station_name, "to_station_name", to_station_name
if from_station_name == first_station and to_station_name == last_station:
found_edge = e
break
# print "found_edge", found_edge
DispatchMaster().set_stopping_length_fraction(found_edge)
return
elif sensor_changed == sensors.getSensor("setStationWaitTimeSensor"):
#optionbox
title = "Station Wait Times"
msg = "modify all station wait times?"
opt1 = "All"
opt2 = "At particular station"
s = self.od.customQuestionMessage2str(msg,title,opt1,opt2)
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
#stopping_sensor_choice = "setNoStoppingSensors"
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
elif s == opt1:
#stopping_sensor_choice = "setAllStoppingSensors"
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
self.modify_all_station_wait_times()
else: #opt2
#stopping_sensor_choice = "setIndividualWaitTimes"
msg = "Press station buttons to select a transit (an approach to a station)\n\n" + \
"The wait time will be set for any train approaching in that direction (using any transit)\n\n" + \
"the wait time will be set at the section of the second station pressed"
self.od.displayMessage(msg)
if self.od.CLOSED_OPTION == True:
#stopping_sensor_choice = "setNoStoppingSensors"
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
else:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
elif sensor_changed == sensors.getSensor("setStationDirectionSensor"):
#optionbox
title = "Restrict Trains to run in only one direction in Stations or Blocks"
msg = "modify station directions?"
try:
folder = "blockDirections"
filename = "blockDirections.txt"
list_items1 = self.dm.read_list(folder, filename)
list_items = [ "from " + l[1] + " to " + l[0] for l in list_items1]
except:
list_items = []
if list_items == []:
list_items = ["no inhibited directions"]
opt1 = "Reset direction restrictions"
opt2 = "Set At particular station"
opt3 = "Set At block"
options = [opt1, opt2, opt3]
ss = self.od.ListOptions(list_items, title, options)
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
s = ss[1]
if s == opt1:
self.reset_direction_restrictions(sensor_changed)
elif s == opt2:
#stopping_sensor_choice = "Set At particular station
msg = "Press station buttons to select a section in order to\nset one way working at that station\n"
msg = msg + "secect the second station adjacent to the first to indicate the direction"
self.od.displayMessage(msg)
if self.od.CLOSED_OPTION == True:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
else:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
else: #opt3
#stopping_sensor_choice = "Set At block
#display list of blocks
msg = "Select block from list to\nset one way working from that block\n"
msg = msg + " then select a block adjacent to the first to indicate the direction"
self.od.displayMessage(msg)
if self.od.CLOSED_OPTION == True:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
else:
self.set_block_direction() # display list of blocks etc.
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
elif sensor_changed == sensors.getSensor("setStopSensor"):
#optionbox
title = "Station Sensors"
msg = "modify station directions?"
list_items = self.get_forward_stop_sensors()
# list_items = [ "from " + l[1] + " to " + l[0] for l in list_items1]
if list_items == []:
list_items = ["no stop sensors set up"]
opt1 = "Delete Selected Sensor"
opt2 = "Set At particular station"
opt3 = "Cancel"
options = [opt1, opt2, opt3]
ss = self.od.ListOptions(list_items, title, options)
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
[my_list,s, index] = ss
if s == opt1:
if my_list != "no stop sensors set up":
while(1):
if my_list != ["no stop sensors set up"]:
#delete the item
[section_text, stopping_sensor_text] = my_list
section_name = section_text.split(" ")[1]
stopping_sensor_name = stopping_sensor_text.split(" ")[1]
sections.getSection(section_name).setForwardStoppingSensorName(None)
# redisplay
opt1 = "Delete Selected Sensor"
opt2 = "Cancel"
options = [opt1, opt2]
title = "Select Stop Sensor to be removed"
list_items = self.get_forward_stop_sensors()
if self.logLevel > 1: print "list_items", list_items
if list_items == [] :
list_items = ["no stop sensors set up"]
if self.logLevel > 1: print "list_items", list_items
[my_list, option, index] = self.od.ListOptions(list_items, title, options)
if self.od.CLOSED_OPTION == True or option == "Cancel": #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
elif s == opt2:
#stopping_sensor_choice = "Set At particular station
msg = "To specify a stop sensor at a station you first need to specify the direction you will be travelling to the station\n" + \
"then specify the stop sensor\n\n" + \
"to specify the direction select two stations, the first en-route to the second station containing the stop sensor\n" + \
"then select the stop sensor\n\n" + \
"Select the first Station"
self.od.displayMessage(msg)
if self.od.CLOSED_OPTION == True:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
else:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
elif s == opt3:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
elif sensor_changed == sensors.getSensor("setTransitBlockRestrictionSensor"):
#optionbox
null_text = "no transit restrictions"
title = "Restrict Transit by specifying blocks which have to be free before transit can run"
msg = "Restrict Transit by specifying blocks which have to be free before transit can run?"
opt1 = "Remove Transit Restriction"
opt2 = "Set Transit Restriction"
opt3 = "Cancel"
list_items1 = self.transit_restrictions(null_text)
list_items = [ " transit '" + l[0] + "' waits for block '" + l[1] + "' to be clear " for l in list_items1]
if list_items == []:
list_items = [null_text]
options = [opt1, opt2, opt3]
[list_item, option, index] = self.od.ListOptions(list_items, title, options)
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
s = option
if option == opt1: # Remove Transit Restriction
self.reset_transit_restrictions(sensor_changed, null_text)
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
elif option == opt2: # Set Transit Restriction"
msg = "Press station buttons to select a section in order to\nselect transit, then specify the block that must be free"
self.od.displayMessage(msg)
#see class RunDispatch for code to do this
if self.od.CLOSED_OPTION == True:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
else:
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_on")
pass
else: # Cancel
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
elif sensor_changed == sensors.getSensor("DummyControlSensor"):
#used to reset the buttons
pass
else:
if sensor_changed != None:
msg = "error: " + sensor_changed.getUserName()
else:
msg = "error in checking buttons"
OptionDialog().displayMessage(msg)
def waitForSensorSet(self, sensor_name):
stopping_distance_sensor = sensors.getSensor(sensor_name)
stopping_distance_sensor.setKnownState(INACTIVE)
btn = [stopping_distance_sensor]
btn_to_watch = java.util.Arrays.asList(btn)
# print "waiting for button" + sensor_name + " which is now " , stopping_distance_sensor.getKnownState()
self.waitSensorState(btn_to_watch, ACTIVE)
def get_forward_stop_sensors(self):
forward_stop_sensors = \
[["section: " + str(section.getUserName()), "stop sensor: " + str(section.getForwardStoppingSensor().getUserName())] \
for section in sections.getNamedBeanSet() if section.getForwardStoppingSensor() != None]
return forward_stop_sensors
def reset_direction_restrictions(self, sensor_changed):
#stopping_sensor_choice = "Reset All"
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
title = "travel only in directions (when poss.)"
folder = "blockDirections"
filename = "blockDirections.txt"
list_items1 = self.dm.read_list(folder, filename)
list_items = [ "from " + l[1] + " to " + l[0] for l in list_items1]
if list_items == []:
list_items = ["no inhibited directions"]
msg = "choose"
opt1 = "Reset All"
opt2 = "Delete Selected Entry"
opt3 = "Cancel"
options = [opt1, opt2, opt3]
ss = self.od.ListOptions(list_items, title, options)
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
s = ss[1]
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
elif s == opt1: #Reset All
self.reset_all_station_directions()
return
elif s == opt2: #Delete Selected Entry
s_list = ss[0]
if s_list != "no inhibited directions":
index = list_items.index(s_list)
entry_to_delete = list_items1[index]
self.dm.delete_block_pair_from_list(entry_to_delete)
self.od.displayMessage("Deleted Item. Will update")
self.regenerate_traininfo_files("Regenerated TrainInfo Files")
# re-display to see that item has been deleted
self.reset_direction_restrictions(sensor_changed)
else: #cancel
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
def reset_transit_restrictions(self, sensor_changed, null_text):
title = "run transit only when block is unoccupied"
list_items1 = self.transit_restrictions(null_text)
list_items = [ "transit " + l[0] + " waits for block " + l[1] + " to be clear" for l in list_items1]
if list_items == []:
list_items = [null_text]
msg = "choose"
opt1 = "Reset All"
opt2 = "Delete Selected Entry"
opt3 = "Cancel"
options = [opt1, opt2, opt3]
[list_item, option, index] = self.od.ListOptions(list_items, title, options)
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
s = option
if self.od.CLOSED_OPTION == True: #check of optionbox was closed prematurely
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
elif s == opt1: #Reset All
if self.logLevel > 0: print "resetting all"
self.reset_all_transit_restrictions()
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
elif s == opt2: # Delete Selected Entry
if self.logLevel > 0: print "delete selected entry"
s_list = list_item
if s_list != null_text:
index = list_items.index(s_list)
[transit_name, transit_block_name] = list_items1[index]
item_to_remove = list_items1[index]
folder = "restrictTransits"
filename = "restrictTransits.txt"
restricted_transits = DispatchMaster().read_list(folder,filename)
filtered_restricted_transits = [ t for t in restricted_transits if t != item_to_remove]
DispatchMaster().write_list(filtered_restricted_transits, folder, filename)
self.od.displayMessage("Deleted block for transit: " + transit_name)
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
else: #cancel
self.switch_sensors_requiring_station_buttons(sensor_changed, "sensor_off")
return
def get_graph_edge(self, transit_name):
for edge in g.g_express.edgeSet():
filename_fwd = self.get_filename(edge, "fwd")
filename_rvs = self.get_filename(edge, "rvs")
trainInfo_fwd = jmri.jmrit.dispatcher.TrainInfoFile().readTrainInfo(filename_fwd)
transit_block_name = trainInfo_fwd.getBlockName()
transit_name_edge = trainInfo_fwd.getTransitName()
if transit_name_edge == transit_name:
break
return edge
def reset_all_transit_restrictions(self):
folder = "restrictTransits"
folder = "restrictTransits"
filename = "restrictTransits.txt"
file_path = DispatchMaster().directory(folder) + filename
# print "filepath", filepath
try:
os.remove(file_path)
# print(file_path + " has been deleted. The individual stopping distances have been zeroed")
except FileNotFoundError:
print(file_path + " not found, cannot delete the individual stopping distances")
except PermissionError:
print("Permission denied to delete" + file_path)
except Exception as e:
print("An error occurred: ", e)
# # check all edges and remove corresponding blocks
# for edge in g.g_express.edgeSet():
# transit_block_name = ""
# self.dm.write_to_TrainInfo(edge, transit_block_name)
def set_block_direction(self):
block_names = [block.getUserName() for block in blocks.getNamedBeanSet()]
block_names.sort()
if self.logLevel > 0: print "block_names", block_names
block_name = OptionDialog().List("choose first block", block_names)
if self.od.CLOSED_OPTION == True:
return
neighbors_block = self.get_neighbors(block_name)
block2_name = OptionDialog().List("choose neighbouring block", neighbors_block)
if self.od.CLOSED_OPTION == True:
return
self.store(block_name, block2_name)
#self.regenerate_traininfo_files("Regenerated TrainInfo Files")
return True
def delete_block_pair(self, entry):
DispatchMaster().delete_block_pair_from_list(entry)
return True
def store(self, block_name, block2_name):
msg = "selected block " + block_name + ". \nDo you wish to allow only one direction from this block ?"
title = "Continue selecting stations"
opt1 = "Allow only the direction from " + block_name + " towards " + str(block2_name)
opt2 = "Allow only the direction from " + block2_name + " towards " + block_name
opt3 = "Allow 2-way working "
s = self.od.customQuestionMessage3str(msg,title,opt1,opt2, opt3)
if self.od.CLOSED_OPTION == True :
# sensor_changed.setKnownState(INACTIVE)
# sensors.getSensor("setStationDirectionSensor").setKnownState(INACTIVE)
# self.button_sensors_to_watch = copy.copy(self.button_sensors)
return False
elif s == opt1:
# sensor_changed.setKnownState(INACTIVE)
first_two_blocks = [block2_name, block_name] # blocks in the inhibited direction
list_of_inhibited_blocks = self.dm.store_the_two_blocks(first_two_blocks)
if s == opt2:
# sensor_changed.setKnownState(INACTIVE)
first_two_blocks = [block_name, block2_name] # blocks in the inhibited direction
list_of_inhibited_blocks = self.dm.store_the_two_blocks(first_two_blocks)
if s == opt3:
# sensor_changed.setKnownState(INACTIVE)
first_two_blocks = [block_name, block2_name]
list_of_inhibited_blocks = self.dm.remove_the_two_blocks(first_two_blocks)
first_two_blocks = [block2_name, block_name]
list_of_inhibited_blocks = self.dm.remove_the_two_blocks(first_two_blocks) #remove from file
# The traininfo files for the express routes need to be regenerated
# so that the express routes are the shortest path allowed
self.regenerate_traininfo_files("Regenerated TrainInfo Files")
def get_neighbors(self, block_name):
layout_block = self.get_layout_block(block_name)
number_neighbors = layout_block.getNumberOfNeighbours()
neighbors = [layout_block.getNeighbourAtIndex(i).getDisplayName() for i in range(number_neighbors)]
return neighbors
def get_layout_block(self, block_name):
LayoutBlockManager=jmri.InstanceManager.getDefault(jmri.jmrit.display.layoutEditor.LayoutBlockManager)
layoutBlock = LayoutBlockManager.getLayoutBlock(block_name)
return layoutBlock
def select_block(self, title, list):
block_name = OptionDialog().List(title, list)
if self.od.CLOSED_OPTION == True:
return "cancel"
return block_name
def modify_all_stopping_distances(self):
title = "Modify all stopping distances"
msg = "Change all stopping distances?"
opt1 = "Increase/decrease all existing stopping distances"
opt2 = "Set all stopping distances to the same value"
s = self.od.customQuestionMessage2str(msg,title,opt1,opt2)
if self.od.CLOSED_OPTION == True:
return #if one has cancelled
#s is used in a little bit
# previous_overall_stopping_distance = self.get_overall_stopping_distance()
# print "previous_overall_stopping_distance", previous_overall_stopping_distance
# new_stopping_position = self.set_overall_stopping_distance(previous_overall_stopping_distance)
# print "new_stopping_position", new_stopping_position
# if new_stopping_position == None:
# return # if one has cancelled
if s == opt1:
# Increase/decrease all existing stopping distances
previous_overall_stopping_distance = self.get_overall_stopping_distance()
# print "previous_overall_stopping_distance", previous_overall_stopping_distance
new_stopping_position = self.get_new_stopping_position(previous_overall_stopping_distance)
self.set_overall_stopping_distance(new_stopping_position)
title = "Stopping train before end of section"
msg = "all stop distances reduced by " + str(new_stopping_position) + " in addition to individual stop distances"
self.od.displayMessage(msg,title)
elif s == opt2:
# Set all stopping distances to the same value
# delete the individual stopping distances
folder = "stopping_positions"
filename = "stopping_positions.txt"
file_path = DispatchMaster().directory(folder) + filename
# print "filepath", filepath
try:
os.remove(file_path)
# print(file_path + " has been deleted. The individual stopping distances have been zeroed")
except OSError:
print(file_path + " not found, cannot delete the individual stopping distances")
except PermissionError:
print("Permission denied to delete" + file_path)
except Exception as e:
print("An error occurred: ", e)
# set the overall stopping distance to the set value
previous_overall_stopping_distance = self.get_overall_stopping_distance()
# print "previous_overall_stopping_distance", previous_overall_stopping_distance
new_stopping_position = self.get_new_stopping_position(previous_overall_stopping_distance)
self.set_overall_stopping_distance(new_stopping_position)
title = "Stopping train before end of section"
msg = "all stop distances reduced by " + str(new_stopping_position) + " individual stop distances deleted"
self.od.displayMessage(msg,title)
def set_overall_stopping_distance(self, overall_stopping_distance):
# print "set_overall_stopping_distance"
folder = "stopping_positions"
filename = "overall_stopping_offset.txt"
file_path = DispatchMaster().directory(folder) + filename
self.write_float(overall_stopping_distance, file_path)
def get_overall_stopping_distance(self):
folder = "stopping_positions"
filename = "overall_stopping_offset.txt"
file_path = DispatchMaster().directory(folder) + filename
overall_stopping_distance = self.read_float(file_path) # use a tuple cos we are set up for that
return overall_stopping_distance
def write_float(self, value, filename):
with open(filename, "w") as f:
f.write("%s" % value) # or f.write(str(value))
def read_float(self, filename):
try:
with open(filename, "r") as f:
return float(f.read().strip())
except:
return 0
def directory(self):
path = jmri.util.FileUtil.getUserFilesPath() + "dispatcher" + java.io.File.separator + "blockDirections"
if not os.path.exists(path):
os.makedirs(path)
return path + java.io.File.separator
def reset_all_station_directions(self):
global g
self.od.displayMessage("About to set All stations set to 2-way working")
if self.od.CLOSED_OPTION == True:
self.od.displayMessage("Cancelled setting all stations to 2-way working")
else:
file = self.directory() + "blockDirections.txt"
with open(file ,'w') as f: # empty the file containing the direction information
pass
self.regenerate_traininfo_files("All stations set to 2-way working")
def regenerate_traininfo_files(self, msg):
global g
# The traininfo files for the express routes need to be regenerated
if self.logLevel > 0: print "Creating Transits"
CreateIcons = jmri.util.FileUtil.getExternalFilename('program:jython/DispatcherSystem/CreateIcons.py')
exec(open (CreateIcons).read())
global dpg
dpg = DisplayProgress()
g = StationGraph() # recalculate the weights on the edges
CreateTransits = jmri.util.FileUtil.getExternalFilename('program:jython/DispatcherSystem/CreateTransits.py')
exec(open (CreateTransits).read())
if self.logLevel > 0: print "about to run CreateTransits"
CreateTransits().process_panels()
self.od.displayMessage(msg)
def modify_all_station_wait_times(self):
title = "Modify all station wait times"
msg = "Change all station wait times?"
opt1 = "Increase/decrease all existing station wait times"
opt2 = "Set all station wait times to the same value"
s = self.od.customQuestionMessage2str(msg,title,opt1,opt2)
if self.od.CLOSED_OPTION == True:
return #if one has cancelled
#s is used in a little bit
new_station_wait_time = self.get_new_station_wait_time()
if new_station_wait_time == None:
return # if one has cancelled
for e in g.g_express.edgeSet():
from_station_name = g.g_stopping.getEdgeSource(e)
to_station_name = g.g_stopping.getEdgeTarget(e)
found_edge = e
old_station_wait_time = self.get_existing_station_wait_time(found_edge)
if s == opt2:
combined_station_wait_time = new_station_wait_time
else:
combined_station_wait_time = new_station_wait_time + old_station_wait_time
filename_fwd = self.get_filename(found_edge, "fwd")
self.modify_station_wait_time(found_edge, combined_station_wait_time, filename_fwd)
filename_rvs = self.get_filename(found_edge, "rvs")
self.modify_station_wait_time(found_edge, combined_station_wait_time, filename_rvs)
def get_existing_stopping_position(self, found_edge, length_of_last_section):
filename_fwd = self.get_filename(found_edge, "fwd")
trainInfo_fwd = jmri.jmrit.dispatcher.TrainInfoFile().readTrainInfo(filename_fwd)
old_stopping_fraction = trainInfo_fwd.getStopBySpeedProfileAdjust()
old_stopping_position = (1.0 - old_stopping_fraction) * float(length_of_last_section)
return old_stopping_position
def get_existing_station_wait_time(self, found_edge):
filename_fwd = self.get_filename(found_edge, "fwd")
trainInfo_fwd = jmri.jmrit.dispatcher.TrainInfoFile().readTrainInfo(filename_fwd)
old_station_wait_time = trainInfo_fwd.getWaitTime()
return old_station_wait_time
def get_filename(self, e, suffix):
# suffix is "fwd" or "rvs"
# e is edge
from_station_name = g.g_express.getEdgeSource(e)
to_station_name = g.g_express.getEdgeTarget(e)
neighbor_name = e.getItem("neighbor_name")
index = e.getItem("index")
filename = "From " + str(from_station_name) + " To " + str(to_station_name) + " Via " + str(neighbor_name) + " " + str(index)
filename = filename.replace(" ", "_")
filename = filename + "_" + suffix + ".xml"
return filename
def get_length_of_last_section(self, found_edge):
filename_fwd = self.get_filename(found_edge, "fwd")
trainInfo_fwd = jmri.jmrit.dispatcher.TrainInfoFile().readTrainInfo(filename_fwd)
last_section = DispatchMaster().last_section_of_transit(trainInfo_fwd) #last_section_of_transit is defined in DispatchMaster, and don't want to repeat code
if self.logLevel > 0: print "last_section",last_section.getUserName()
length_of_last_section = float(DispatchMaster().length_of_last_section(last_section))/10.0 #cm
if self.logLevel > 0: print "length_of_last_section", length_of_last_section
return length_of_last_section
def modify_stopping_distance(self, found_edge, new_stopping_fraction, filename):
trainInfo = jmri.jmrit.dispatcher.TrainInfoFile().readTrainInfo(filename)
trainInfo.setStopBySpeedProfileAdjust(float(new_stopping_fraction))
#write the new stopping fraction back to file
jmri.jmrit.dispatcher.TrainInfoFile().writeTrainInfo(trainInfo, filename)
def modify_station_wait_time(self, found_edge, wait_time, filename):
trainInfo = jmri.jmrit.dispatcher.TrainInfoFile().readTrainInfo(filename)
trainInfo.setWaitTime(float(wait_time))
#write the new wait time back to file
jmri.jmrit.dispatcher.TrainInfoFile().writeTrainInfo(trainInfo, filename)
def is_integer(self, n):
try:
if n == None: return False
float(n)
except ValueError:
return False
else:
return float(n).is_integer()
def get_new_stopping_position(self, previous_overall_stopping_distance):
s = "redo"
while s == "redo":
#modify stopping fraction in traininfo
title = "Stop train before end of section"
msg ="enter how many cm to reduce all stopping distances by (-ve increase), current value shown"
new_stopping_position = self.od.input(msg, title, previous_overall_stopping_distance)
if not self.is_integer(new_stopping_position): return
# print "new_stopping_position", new_stopping_position
# print "new_stopping_fraction", new_stopping_fraction
if float(new_stopping_position) > 0:
sp = float(new_stopping_position)
else:
sp = 0 - float(new_stopping_position)
msg = "new stopping position: " + str(round(float(sp),1)) + " cm (" + \
str(round(float(sp)/2.54,1)) + " inches) after calculated position"
opt1 = "OK"
opt2 = "redo"
s = self.od.customQuestionMessage2str(msg, title, opt1, opt2)
previous_overall_stopping_distance = str(float(new_stopping_position))
return float(new_stopping_position)
def get_new_station_wait_time(self):
s = "redo"
while s == "redo":
#modify station wait time in traininfo
title = "Pause train at beginning of section"
msg ="enter how many secs to wait at station"
default_value = 0
new_wait_time = self.od.input(msg, title, default_value)
if not self.is_integer(new_wait_time): return
# print "new_stopping_position", new_stopping_position
# print "new_stopping_fraction", new_stopping_fraction
if float(new_wait_time) > 0:
msg = "new wait time: " + str(round(float(new_wait_time),1)) + " secs "
else:
sp = 0 # set to zero as cannot be -ve
msg = "new wait time: " + str(round(float(sp),1)) + " secs "
opt1 = "OK"
opt2 = "redo"
s = self.od.customQuestionMessage2str(msg, title, opt1, opt2)
# msg = "wait time = " + str(new_wait_time)
# self.od.displayMessage(msg,title)
return float(new_wait_time)
def get_new_stopping_fraction(self, new_stopping_position, length_of_last_section):
if self.logLevel > 0: "new_stopping_position",new_stopping_position, "length_of_last_section", length_of_last_section
new_stopping_fraction = 1.0-(float(new_stopping_position)/float(length_of_last_section))
return new_stopping_fraction
def process_run_route(self):
title = ""
msg = "run one or more routes"
opt1 = "1 route"
opt2 = "several routes"
reply = self.od.customQuestionMessage2str(msg, title, opt1, opt2)
if reply == opt1:
self.run_route_1()
else:
self.run_routes()
sensors.getSensor("runRouteSensor").setKnownState(INACTIVE)
def get_buttons(self):
self.button_sensors = [self.get_button_sensor_given_block_name(station_block_name) for station_block_name in g.station_block_list]
self.button_sensor_states = [self.check_sensor_state(button_sensor) for button_sensor in self.button_sensors]
# for button_sensor in self.button_sensors:
# self.button_dict[button_sensor] = self.check_sensor_state(button_sensor)
def get_sensors_requiring_use_of_station_buttons(self):
sensor_list = ["setDispatchSensor", "setRouteSensor", "setStoppingDistanceSensor", \
"setStationWaitTimeSensor", "setStationDirectionSensor", "setTransitBlockRestrictionSensor", \
"setStopSensor", "DummyControlSensor"]
self.sensors_requiring_use_of_station_buttons = \
[sensors.getSensor(sensorName) for sensorName in sensor_list]
# print "sensors_requiring_use_of_station_buttons", self.sensors_requiring_use_of_station_buttons
self.route_dispatch_states = [self.check_sensor_state(rd_sensor) for rd_sensor in self.sensors_requiring_use_of_station_buttons]
def get_route_run_button(self):
self.route_run_sensor = [sensors.getSensor(sensorName) for sensorName in ["runRouteSensor"]]
def get_set_stopping_length_button(self):
self.stopping_distance_sensor = [sensors.getSensor(sensorName) for sensorName in ["setStoppingDistanceSensor"]]
def get_station_wait_time_button(self):
self.station_wait_time_sensor = [sensors.getSensor(sensorName) for sensorName in ["setStationWaitTimeSensor"]]
def get_station_direction_button(self):
self.station_direction_sensor = [sensors.getSensor(sensorName) for sensorName in ["setStationDirectionSensor"]]
def check_sensor_state(self, sensor):
#if self.logLevel > 0: print("check_sensor_state",sensor)
if sensor == None :
#if self.logLevel > 0: print('Sensor in check_sensor_state is none')
return None
#sensor = sensors.getSensor(sensor_name)
if sensor is None:
OptionDialog().displayMessage('Sensor {} not found'.format( sensor_name))
return
currentState = True if sensor.getKnownState() == ACTIVE else False
#if self.logLevel > 0: print("check_sensor_state {}".format(currentState))
return currentState
def store_button_states(self):
self.button_sensor_states_old = self.button_sensor_states
if self.logLevel > 0: print "self.button_sensor_states_old",self.button_sensor_states_old
#self.button_dict_old = dict(self.button_dict)
def get_button_sensor_given_block_name(self, block_name):
button_sensor_name = "MoveTo"+block_name.replace(" ","_") +"_stored"
button_sensor = sensors.getSensor(button_sensor_name)
if button_sensor == None:
if self.logLevel > 0: print button_sensor_name + "= None"
return button_sensor
def run_route_1(self):
global trains_dispatched
# list_items = ("Run Route", "Cancel")
# title = "choose option"
# result = self.od.List(title, list_items)
# if self.od.CLOSED_OPTION == True:
# return
# if result == "Run Route":
trains_to_choose = self.get_list_of_engines_to_move()
# msg = "trains_to_choose" + str(trains_to_choose)
if trains_to_choose == []:
# JOptionPane.showMessageDialog(None,msg)
str_trains_dispatched= (' '.join(trains_dispatched))
msg = "There are no trains available for dispatch\nTrains dispatched are:\n"+str_trains_dispatched+"\n"
title = "Cannot move train"
opt1 = "continue"
opt2 = "stop all dispatches"
result = self.od.customQuestionMessage2str(msg, title, opt1, opt2)
if result == "stop all dispatches":
delete_transits()
return
title = "what train do you want to move?"
engine = self.od.List(title, trains_to_choose)
if self.od.CLOSED_OPTION == True:
return
station_from = self.get_position_of_train(engine)
RouteManager=jmri.InstanceManager.getDefault(jmri.jmrit.operations.routes.RouteManager)
opt1a = "show routes starting at train position"
opt1b = "show all routes"
opt1 = opt1a
opt2 = "OK"
s = opt1
list_items = RouteManager.getRoutesByNameList()
while s == opt1:
title = "choose route"
[l,s, index] = self.od.ListOptions(list_items,title,[opt2,opt1])
print "5"
if self.od.CLOSED_OPTION == True:
return
if s == opt1:
# xx = [str(station_block_name) for station_block_name in g.station_block_list \
# if blocks.getBlock(station_block_name).getValue()==engine]
if opt1 == opt1a:
# print "6 ", engine, engine
# print "7 g.station_block_list", g.station_block_list
# for station_block_name in g.station_block_list:
# engine = blocks.getBlock(station_block_name).getValue()
# print "engine", engine
station_where_engine_is = [str(station_block_name) for station_block_name in g.station_block_list \
if blocks.getBlock(station_block_name).getValue()==engine]
# print "station_where_engine_is", station_where_engine_is
station_where_engine_is = station_where_engine_is[0]
# print "8 ", "station_where_engine_is", station_where_engine_is
list_items = [l1 for l1 in list_items if str(l1.getName()).startswith(str(station_where_engine_is))]
# print "list_items", list_items
s = opt1 = opt1b
else:
list_items = RouteManager.getRoutesByNameList()
s = opt1 = opt1a
routeName = str(l)
if self.logLevel > 0: print "routeName", routeName
route = RouteManager.getRouteByName(routeName)
list_items = ["stop at end of route", "return to start position", "return to start and repeat", "cancel"]
title = "What do you want to do"
option = self.od.List(title, list_items)
if self.od.CLOSED_OPTION == True:
return
repeat = False
dont_run_route = False
if option == "stop at end of route":
station_to = None
repeat = False
elif option == "return to start position":
station_to = station_from
repeat = False
elif option == "return to start and repeat":
station_to = station_from
repeat = True
else:
dont_run_route = True
if repeat:
title = "repeat how many times?"
default_value = 3
msg = "repeat how many times"
no_repetitions = self.od.input(msg, title, default_value)
else:
no_repetitions = 0
if dont_run_route == False:
if self.logLevel > -1: print "station_from", station_from, "station_to",station_to, "repeat",repeat
run_train = RunRoute(route, g.g_express, station_from, station_to, no_repetitions, engine)
run_train.setName("running_route_" + routeName)
instanceList.append(run_train)
run_train.start()
def delete_transits(self):
# need to avoid concurrency issues when deleting more that one transit
# use java.util.concurrent.CopyOnWriteArrayList so can iterate through the transits while deleting
DF = jmri.InstanceManager.getDefault(jmri.jmrit.dispatcher.DispatcherFrame)
#DF.setState(DF.ICONIFIED);
activeTrainList = java.util.concurrent.CopyOnWriteArrayList()
for activeTrain in DF.getActiveTrainsList():
activeTrainList.add(activeTrain)
for activeTrain in activeTrainList:
# print "i", i
# activeTrain = activeTrainsList.get(i)
if self.logLevel > 0: print ("active train", activeTrain)
DF.terminateActiveTrain(activeTrain, True, False)
DF = None
def get_list_of_engines_to_move(self):
global trains_allocated
global trains_dispatched
#find what train we want to move
#select only from available trains %%%%todo%%%%%
all_trains = self.get_all_roster_entries_with_speed_profile()
#trains to choose from are the allocated - dispatched
trains_to_choose = copy.copy(trains_allocated)
if self.logLevel > 0: print "trains_dispatchedx", trains_dispatched
if self.logLevel > 0: print "trains_allocated",trains_allocated
if self.logLevel > 0: print "trains_to_choose",trains_to_choose
if trains_dispatched != []:
for train in trains_dispatched:
if self.logLevel > 0: print "removing" ,train
trains_to_choose.remove(train)
if self.logLevel > 0: print "trains_to_choose",trains_to_choose
return trains_to_choose
def get_position_of_train(self, train_to_move):
## Check the pressed button
for station_block_name in g.station_block_list:
if self.logLevel > 0: print "station_block_name", station_block_name
#get a True if the block block_value has the train name in it
block_value_state = self.check_train_in_block(station_block_name, train_to_move)
if self.logLevel > 0: print "block_value_state= ",block_value_state
#get a True if the block is occupied
block_occupied_state = self.check_sensor_state_given_block_name(station_block_name)
if self.logLevel > 0: print "block_occupied_state= ",block_occupied_state
if self.logLevel > 0: print ("station block name {} : {}". format(station_block_name, str(block_occupied_state)))
# # do not attempt to move to where you are
# button_pressed_in_occupied_station = (button_station_name == station_block_name)
#check if the block is occupied and has the required train in it
if block_value_state == True and block_occupied_state == True:
# and button_pressed_in_occupied_station == False:
return station_block_name
return None
def get_blockcontents(self, block_name):
block = blocks.getBlock(block_name)
value = block.getValue()
return value
def check_train_in_block(self, block_name, train_name):
mem_val = self.get_blockcontents(block_name)
if train_name == mem_val:
return True
else:
return False
def check_sensor_state_given_block_name(self, station_block_name):
#if self.logLevel > 0: print("station block name {}".format(station_block_name))
layoutBlock = layoutblocks.getLayoutBlock(station_block_name)
station_sensor = layoutBlock.getOccupancySensor()
if station_sensor is None:
OptionDialog().displayMessage(' Sensor in block {} not found'.format(station_block_name))
return
currentState = True if station_sensor.getKnownState() == ACTIVE else False
return currentState
def get_all_roster_entries_with_speed_profile(self):
roster_entries_with_speed_profile = []
r = jmri.jmrit.roster.Roster.getDefault()
for roster_entry in jmri.jmrit.roster.Roster.getAllEntries(r):
if self.logLevel > 0: print "roster_entry.getSpeedProfile()",roster_entry,roster_entry.getSpeedProfile()
if roster_entry.getSpeedProfile() != None:
roster_entries_with_speed_profile.append(roster_entry.getId())
if self.logLevel > 0: print "roster_entry.getId()",roster_entry.getId()
return roster_entries_with_speed_profile
def run_routes(self):
createandshowGUI2(self)
class createandshowGUI2(TableModelListener):
def __init__(self, class_ResetButtonMaster):
self.logLevel = 0
self.class_ResetButtonMaster = class_ResetButtonMaster
#Create and set up the window.
self.initialise_model(class_ResetButtonMaster)
self.frame = JFrame("Allocate Routes")
# self.frame.setSize(600, 600);
self.completeTablePanel()
# print "about to populate"
self.populate_action(None)
self.cancel = False
def completeTablePanel(self):
self.topPanel= JPanel();
self.topPanel.setLayout(BoxLayout(self.topPanel, BoxLayout.X_AXIS))
self.self_table()
scrollPane = JScrollPane(self.table);
# scrollPane.setSize(600,600);
self.topPanel.add(scrollPane);
self.buttonPane = JPanel();
self.buttonPane.setLayout(BoxLayout(self.buttonPane, BoxLayout.LINE_AXIS))
self.buttonPane.setBorder(BorderFactory.createEmptyBorder(0, 10, 10, 10))
button_populate = JButton("Populate", actionPerformed = self.populate_action)
self.buttonPane.add(button_populate);
self.buttonPane.add(Box.createRigidArea(Dimension(10, 0)))
button_tidy = JButton("Tidy", actionPerformed = self.tidy_action)
self.buttonPane.add(button_tidy);
self.buttonPane.add(Box.createRigidArea(Dimension(10, 0)))
button_apply = JButton("Run Routes", actionPerformed = self.apply_action)
self.buttonPane.add(button_apply)
self.buttonPane.add(Box.createHorizontalGlue());
button_close = JButton("Close", actionPerformed = self.close_action)
self.buttonPane.add(button_close)
self.buttonPane.add(Box.createHorizontalGlue());
button_task = JButton("Task", actionPerformed = self.task_action)
self.buttonPane.add(button_task)
self.buttonPane.add(Box.createHorizontalGlue());
button_task = JButton("Delay", actionPerformed = self.delay_action)
self.buttonPane.add(button_task)
self.buttonPane.add(Box.createHorizontalGlue());
button_repetitions = JButton("No. repetitions", actionPerformed = self.repetitions_action)
self.buttonPane.add(button_repetitions)
self.buttonPane.add(Box.createHorizontalGlue());
button_savetofile = JButton("Save To File", actionPerformed = self.savetofile_action)
self.buttonPane.add(button_savetofile)
self.buttonPane.add(Box.createHorizontalGlue());
button_loadfromfile = JButton("Load From File", actionPerformed = self.loadfromfile_action)
self.buttonPane.add(button_loadfromfile)
self.buttonPane.add(Box.createHorizontalGlue());
contentPane = self.frame.getContentPane()
contentPane.removeAll()
contentPane.add(self.topPanel, BorderLayout.CENTER)
contentPane.add(self.buttonPane, BorderLayout.PAGE_END)
self.tidy()
self.frame.pack();
self.frame.setVisible(True)
return
def buttonPanel(self):
row1_1_button = JButton("Add Row", actionPerformed = self.add_row_action)
row1_2_button = JButton("Save", actionPerformed = self.save_action)
row1 = JPanel()
row1.setLayout(BoxLayout(row1, BoxLayout.X_AXIS))
row1.add(Box.createVerticalGlue())
row1.add(Box.createRigidArea(Dimension(20, 0)))
row1.add(row1_1_button)
row1.add(Box.createRigidArea(Dimension(20, 0)))
row1.add(row1_2_button)
layout = BorderLayout()
jPanel = JPanel()
jPanel.setLayout(layout);
jPanel.add(self.table,BorderLayout.NORTH)
jPanel.add(row1,BorderLayout.SOUTH)
#return jPanel
return topPanel
def initialise_model(self, class_ResetButtonMaster):
self.model = None
self.model = MyTableModel1()
self.table = JTable(self.model)
self.model.addTableModelListener(MyModelListener1(self, class_ResetButtonMaster));
self.class_ResetButtonMaster = class_ResetButtonMaster
pass
def self_table(self):
#table.setPreferredScrollableViewportSize(Dimension(500, 70));
#table.setFillsViewportHeight(True)
#self.table.getModel().addtableModelListener(self)
self.table.setFillsViewportHeight(True);
self.table.setRowHeight(30);
#table.setAutoResizeMode( JTable.AUTO_RESIZE_OFF );
# self.resizeColumnWidth(table)
columnModel = self.table.getColumnModel();
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
columnModel.getColumn(route_col).setPreferredWidth(200);
columnModel.getColumn(task_col).setPreferredWidth(150);
# first column is the trains
self.trainColumn = self.table.getColumnModel().getColumn(train_col);
self.combobox0 = JComboBox()
for train in self.class_ResetButtonMaster.get_list_of_engines_to_move():
self.combobox0.addItem(train)
self.trainColumn.setCellEditor(DefaultCellEditor(self.combobox0))
renderer0 = ComboBoxCellRenderer1()
self.trainColumn.setCellRenderer(renderer0);
# second column is the routes
self.routesColumn = self.table.getColumnModel().getColumn(route_col);
self.combobox1 = JComboBox()
RouteManager=jmri.InstanceManager.getDefault(jmri.jmrit.operations.routes.RouteManager)
routes = RouteManager.getRoutesByNameList()
for route in routes:
self.combobox1.addItem(route)
self.routesColumn.setCellEditor(DefaultCellEditor(self.combobox1));
renderer1 = ComboBoxCellRenderer1()
self.routesColumn.setCellRenderer(renderer1);
# first column is the trains
self.taskColumn = self.table.getColumnModel().getColumn(task_col);
self.combobox3 = JComboBox()
tasks = ["stop at end of route", "return to start position","return to start and repeat"]
for task in tasks:
self.combobox3.addItem(task)
self.taskColumn.setCellEditor(DefaultCellEditor(self.combobox3));
renderer3 = ComboBoxCellRenderer1()
self.taskColumn.setCellRenderer(renderer3);
jpane = JScrollPane(self.table)
panel = JPanel()
panel.add(jpane)
result = JScrollPane(panel)
return self.table
def add_row_action(self, e):
model = e.getSource()
data = self.model.getValueAt(0, 0)
count = self.model.getRowCount()
colcount = self.model.getColumnCount()
self.model.add_row()
self.completeTablePanel()
def populate_action(self, event):
trains_to_put_in_dropdown = self.class_ResetButtonMaster.get_list_of_engines_to_move()
self.model.populate(trains_to_put_in_dropdown)
self.completeTablePanel()
pass
def tidy_action(self,e):
self.model.remove_not_set_row()
self.completeTablePanel()
def tidy_action(self,e):
self.tidy()
self.completeTablePanel()
def tidy(self):
# self.model.remove_not_set_row() # can't do this as removes all entries with no routes
size_of_one_row = 30
height = 130
for row in reversed(range(len(self.model.data))):
height += size_of_one_row
self.frame.setPreferredSize(Dimension(800, height))
def savetofile_action(self, event):
#Tidy
self.model.remove_not_set_row()
self.completeTablePanel()
if self.model.getRowCount() == 0:
msg = "There are no valid rows"
result = OptionDialog().displayMessage(msg)
return
msg = "Saving Valid rows"
result = OptionDialog().displayMessage(msg)
dir = self.directory()
j = JFileChooser(dir);
j.setAcceptAllFileFilterUsed(False)
filter = FileNameExtensionFilter("text files txt", ["txt"])
j.addChoosableFileFilter(filter);
j.setDialogTitle("Select a .txt file");
ret = j.showSaveDialog(None);
if (ret == JFileChooser.APPROVE_OPTION) :
file = j.getSelectedFile()
if file == "" or file == None:
msg = "No file selected"
result = OptionDialog().displayMessage(msg)
return
if FilenameUtils.getExtension(file.getName()).lower() == "txt" :
#filename is OK as-is
pass
else:
#file = File(file.toString() + ".txt"); # append .txt if "foo.jpg.txt" is OK
file = File(file.getParentFile(), FilenameUtils.getBaseName(file.getName())+".txt") # ALTERNATIVELY: remove the extension (if any) and replace it with ".xml"
else:
return
if self.logLevel > 0: print "savetofile action", file
my_list = []
[train, route, task, delay, repetitions] = [0, 1, 3, 4, 5]
for row in range(len(self.model.data)):
train_name = str(self.model.data[row][train])
route_name = str(self.model.data[row][route])
task_name = str(self.model.data[row][task])
delay_name = str(self.model.data[row][delay])
repetitions_name = str(self.model.data[row][repetitions])
row_list = [train_name, route_name, task_name, delay_name, repetitions_name]
my_list.append(row_list)
self.write_list(my_list,file)
def loadfromfile_action(self, event):
# load the file
dir = self.directory()
j = JFileChooser(dir);
j.setAcceptAllFileFilterUsed(False)
filter = FileNameExtensionFilter("text files txt", ["txt"])
j.setDialogTitle("Select a .txt file");
j.addChoosableFileFilter(filter);
# Automatically select the first file in the directory
files = j.getCurrentDirectory().listFiles()
j.setSelectedFile(files[0])
ret = j.showOpenDialog(None);
if (ret == JFileChooser.APPROVE_OPTION) :
file = j.getSelectedFile()
if self.logLevel > 0: print "about to read list", file
my_list = self.read_list(file)
if self.logLevel > 0: print "my_list", my_list
for row in reversed(range(len(self.model.data))):
self.model.data.pop(row)
i = 0
[train, route, task, delay, repetitions] = [0, 1, 3, 4, 5]
for row in my_list:
[train_val, route_val, task_val, delay_val, repetitions_val] = row
self.model.add_row()
self.model.data[i][train] = train_val.replace('"','')
self.model.data[i][route] = route_val.replace('"','')
self.model.data[i][task] = task_val.replace('"','')
self.model.data[i][delay] = delay_val.replace('"','')
self.model.data[i][repetitions] = repetitions_val.replace('"','')
i += 1
self.completeTablePanel()
msg = "Deleting invalid rows"
result = OptionDialog().displayMessage(msg)
# check the loaded contents
# 1) check that the trains are valid
# 2) check that the blocks are occupied by valid trains
# if either of the above are not valid we blank the entries
# 3) Tidy
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
# check the trains are valid
trains_to_put_in_dropdown = [t for t in self.class_ResetButtonMaster.get_list_of_engines_to_move()]
for row in reversed(range(len(self.model.data))):
if self.model.data[row][train_col] not in trains_to_put_in_dropdown:
self.model.data.pop(row)
RouteManager=jmri.InstanceManager.getDefault(jmri.jmrit.operations.routes.RouteManager)
routes = [str(route) for route in RouteManager.getRoutesByNameList()]
for row in reversed(range(len(self.model.data))):
if self.model.data[row][route_col] not in routes:
self.model.data.pop(row)
self.completeTablePanel()
def close_action(self, event):
self.frame.dispatchEvent(WindowEvent(self.frame, WindowEvent.WINDOW_CLOSING));
def delay_action(self, event):
# print "delay"
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
for row in reversed(range(len(self.model.data))):
old_delay = int(self.model.data[0][delay_col])
if old_delay == None: old_delay = 0
new_delay = self.new_delay(old_delay)
self.model.data[row][delay_col] = new_delay
self.completeTablePanel()
def new_delay(self, old_val):
if old_val < 3:
new_val = 3
elif old_val < 5:
new_val = 5
elif old_val < 10:
new_val = 10
elif old_val < 15:
new_val = 15
else:
new_val = 0
return new_val
def repetitions_action(self, event):
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
for row in reversed(range(len(self.model.data))):
old_val = int(self.model.data[0][repetition_col])
if old_val == None: old_val = 0
new_val = self.new_val(old_val)
self.model.data[row][repetition_col] = new_val
self.completeTablePanel()
def new_val(self, old_val):
if old_val < 3:
new_val = 3
elif old_val < 10:
new_val = 10
elif old_val < 30:
new_val = 30
elif old_val < 100:
new_val = 100
else:
new_val = 1
return new_val
def task_action(self, event):
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
for row in reversed(range(len(self.model.data))):
old_val = str(self.model.data[0][task_col])
if old_val == None: old_val = 0
new_val = self.new_task(old_val)
self.model.data[row][task_col] = new_val
self.completeTablePanel()
def new_task(self, old_val):
tasks = ["stop at end of route", "return to start position","return to start and repeat"]
if old_val == "stop at end of route":
new_val = "return to start position"
elif old_val == "return to start position":
new_val = "return to start and repeat"
else:
return "stop at end of route"
return new_val
def apply_action(self, event):
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
# print "apply action"
for row in reversed(range(len(self.model.data))):
train_name = str(self.model.data[row][train_col])
route_name = str(self.model.data[row][route_col])
delay_val = str(self.model.data[row][delay_col])
if train_name != "" and route_name != "" and delay_val != "":
# print "running route", route_name
self.run_route_2(row, self.model, self, self.class_ResetButtonMaster)
else:
msg = "not running route, train, route or delay is not set"
OptionDialog().displayMessage(msg,"")
self.completeTablePanel()
if self.model.getRowCount() == 0:
self.frame.dispatchEvent(WindowEvent(self.frame, WindowEvent.WINDOW_CLOSING))
def run_route_2(self, row, model, class_createandshowGUI2, class_ResetButtonMaster):
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
route_name = str(model.getValueAt(row, route_col))
if route_name == None:
msg = "not running route is not set"
self.od.displayMessage(msg,"")
return
RouteManager=jmri.InstanceManager.getDefault(jmri.jmrit.operations.routes.RouteManager)
route = RouteManager.getRouteByName(route_name)
train_name = str(model.getValueAt(row, train_col))
if train_name == None or train_name == "":
msg = "not running route, train is not set"
self.od.displayMessage(msg,"")
return
station_from = class_ResetButtonMaster.get_position_of_train(train_name)
option = str(model.getValueAt(row, task_col))
repeat = False
dont_run_route = False
no_repetitions = 0
if option == "stop at end of route":
station_to = None
repeat = False
elif option == "return to start position":
station_to = station_from
repeat = False
elif option == "return to start and repeat":
station_to = station_from
repeat = True
else:
dont_run_route = True
if repeat:
no_repetitions = str(model.getValueAt(row, repetition_col))
else:
no_repetitions = 0
# delay by delay_val before starting route
delay_val = int(model.getValueAt(row, delay_col)) *1000
if dont_run_route == False:
if self.logLevel > 0: print "station_from", station_from, "station_to",station_to, \
"repeat",repeat, "delay", delay_val, "no_repetitions", no_repetitions
run_train = RunRoute(route, g.g_express, station_from, station_to, no_repetitions, train_name, delay_val)
run_train.setName("running_route_" + route_name)
instanceList.append(run_train)
run_train.start()
model.data.pop(row)
class_createandshowGUI2.completeTablePanel()
def directory(self):
path = jmri.util.FileUtil.getUserFilesPath() + "dispatcher" + java.io.File.separator + "routes"
if not os.path.exists(path):
os.makedirs(path)
return path + java.io.File.separator
def write_list(self, a_list, file):
# store list in binary file so 'wb' mode
#file = self.directory() + "blockDirections.txt"
if self.logLevel > 0: print "block_info" , a_list
if self.logLevel > 0: print "file" , file
file = str(file)
with open(file, 'wb') as fp:
pass
if self.logLevel > 0: print "V"
with open(file, 'wb') as fp:
if self.logLevel > 0: print "B"
for items in a_list:
if self.logLevel > 0: print "C", items
i = 0
for item in items:
if self.logLevel > 0: print "item", item
fp.write('"%s"' %item)
if i != 4: fp.write(",")
i+=1
fp.write('\n')
#fp.write('\n'.join(item))
#fp.write(items)
# Read list to memory
def read_list(self, file):
file = str(file)
if self.logLevel > 0: print "read list", file
# for reading also binary mode is important
#file = self.directory() + "blockDirections.txt"
n_list = []
# try:
with open(file, 'rb') as fp:
for line in fp:
if self.logLevel > 0: print "line" , line
x = line[:-1]
if self.logLevel > 0: print x
y = x.split(",")
#y = [item.replace('"','') for item in y]
if self.logLevel > 0: print "y" , y
n_list.append(y)
return n_list
# except:
# return ["",""]
class MyModelListener1(TableModelListener):
def __init__(self, class_createandshowGUI2, class_ResetButtonMaster):
self.class_createandshowGUI2 = class_createandshowGUI2
self.class_ResetButtonMaster = class_ResetButtonMaster
self.cancel = False
self.logLevel = 0
def tableChanged(self, e) :
global trains_allocated
row = e.getFirstRow()
column = e.getColumn()
model = e.getSource()
columnName = model.getColumnName(column)
data = model.getValueAt(row, column)
class_createandshowGUI2 = self.class_createandshowGUI2
class_ResetButtonMaster = self.class_ResetButtonMaster
tablemodel = class_createandshowGUI2.model
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
if column == train_col: #trains
pass
elif column == route_col: # sections
pass
elif column == run_route_col:
# print "running route"
class_createandshowGUI2.run_route_2(row, model, class_createandshowGUI2, class_ResetButtonMaster)
class ComboBoxCellRenderer1 (TableCellRenderer):
def getTableCellRendererComponent(self, jtable, value, isSelected, hasFocus, row, column) :
panel = self.createPanel(value)
return panel
def createPanel(self, s):
p = JPanel(BorderLayout())
p.add(JLabel(str(s), JLabel.LEFT), BorderLayout.WEST)
icon = UIManager.getIcon("Table.descendingSortIcon")
p.add(JLabel(icon, JLabel.RIGHT), BorderLayout.EAST)
p.setBorder(BorderFactory.createLineBorder(Color.blue))
return p
class MyTableModel1 (DefaultTableModel):
columnNames = ["Train", "Route", "Run Route", "Task", "Delay (secs)", "No. Repetitons"]
def __init__(self):
l1 = ["", "", False, "stop at end of route", 10, 0]
self.data = [l1]
def remove_not_set_row(self):
b = False
for row in reversed(range(len(self.data))):
# print "row", row
if self.data[row][1] == "":
self.data.pop(row)
def add_row(self):
# print "addidn row"
# if row < len(self.data):
# print "add"
self.data.append(["", "", False, "stop at end of route", 10, 0])
# print self.data
# print "added"
def populate(self, trains_to_put_in_dropdown):
# for row in reversed(range(len(self.data))):
# self.data.pop(row)
# self.data = []
# append all trains to put in dropdown
[train_col, route_col, run_route_col, task_col, delay_col, repetition_col] = [0, 1, 2, 3, 4, 5]
for train in trains_to_put_in_dropdown:
train_present = False
for row in reversed(range(len(self.data))):
if self.data[row][train_col] == train:
train_present = True
if train_present == False:
self.data.append([train, "", False, "stop at end of route", 10, 3])
# delete rows with no trains
for row in reversed(range(len(self.data))):
if self.data[row][train_col] == None or self.data[row][train_col] == "":
self.data.pop(row)
def getColumnCount(self) :
return len(self.columnNames)
def getRowCount(self) :
return len(self.data)
def getColumnName(self, col) :
return self.columnNames[col]
def getValueAt(self, row, col) :
return self.data[row][col]
def getColumnClass(self, c) :
if c <= 1:
return java.lang.Boolean.getClass(JComboBox)
return java.lang.Boolean.getClass(self.getValueAt(0,c))
#only include if table editable
def isCellEditable(self, row, col) :
# Note that the data/cell address is constant,
# no matter where the cell appears onscreen.
return True
# only include if data can change.
def setValueAt(self, value, row, col) :
self.data[row][col] = value
self.fireTableCellUpdated(row, col)
class MonitorTrackMaster(jmri.jmrit.automat.AbstractAutomaton):
# 1) ensure that the trains in the train list are being displayed on the panel if the block is occupied
# 2) ensure that a non allocated train and an allocated train cannot exist in the same edge
global trains_allocated
logLevel = 0
def setup(self):
self.old_train_edge = {}
self.moved_under_dispatch = {}
return True
def handle(self):
# trains is a dictionary, with keys of the train_name
# each value is itself a dictionary with 3 items
# edge
# penultimate_block_name
# direction
# ensure that the trains in the train list are being displayed on the panel if the block is occupied
#
# if the end block of the route is occupied
# set th end block memname to the train_name
if trains != {}:
if self.logLevel > 0: print "trains", trains
for train_name in trains: #defaults tothe key which is the train name
# if self.logLevel > 0: print "train" , train
# train_name = str(train[0])
# train_list = train[1]
mytrain = trains[train_name] #mytrain is a dictionary
if self.logLevel > 0: print "train_name", train_name, #, "train_list", train_list
#edge = train_list("edge")
#penultimate_block_name = train_list["penultimate_block_name"]
if "edge" in mytrain:
edge = str(mytrain["edge"])
if self.logLevel > 0: print "edge",edge,
#if the train is at the start of the dispatch or at the end we
#block_name = str(mytrain["edge"].getSource()) #get the destination edge- the edges are reversed
#block_occupancy = self.check_sensor_state_given_block_name(block_name)
#if self.logLevel > 0: print "block_name", block_name,"block_occupancy", block_occupancy,
active_train = self.get_active_train(train_name) #get the active train if one exists (if a dispatch is in progress)
if active_train != None:
#[block_list, start_block, end_block] = self.get_occupied_blocks(active_train)
if self.logLevel > 0: print "active train is not None", active_train,
else:
if self.logLevel > 0: print "active train is None", active_train,
if train_name in self.old_train_edge:
if self.logLevel > 0: print "old edge", self.old_train_edge[train_name]
#if the edge for the train has changed we initialise moved_under_dispatch to False
if train_name in self.old_train_edge:
if self.old_train_edge[train_name] == edge:
pass
else:
#initialise the moved_under_dispatch variable
self.moved_under_dispatch[train_name] = False
if train_name not in self.moved_under_dispatch:
self.moved_under_dispatch[train_name] = False
if active_train == None:
# train is at the start block or the end block as the dispatch has not started or finished
if train_name in self.moved_under_dispatch:
if self.moved_under_dispatch[train_name] == False:
#block is start block
current_block = str(mytrain["edge"].getTarget())
block_occupancy = self.check_sensor_state_given_block_name(current_block)
if self.logLevel > 0: print "current_block", current_block,"block_occupancy", block_occupancy,
self.set_mem_variable(current_block,train_name,block_occupancy)
else:
#block is end block
current_block = str(mytrain["edge"].getSource())
block_occupancy = self.check_sensor_state_given_block_name(current_block)
if self.logLevel > 0: print "current_block", current_block,"block_occupancy", block_occupancy,
self.set_mem_variable(current_block,train_name,block_occupancy)
else:
#trai
# n is moving under dispatch
#we note this
self.moved_under_dispatch[train_name] = True
if edge != None:
self.old_train_edge[train_name] = edge
self.waitMsec(2000)
return True
def set_mem_variable(self, block_name, train_name, block_occupancy):
if block_name != None:
#print "block_name", block_name
#print "self.check_train_in_block:",self.check_train_in_block(block_name, train_name) ,"xxxx"
if block_occupancy == True:
#check and set the mem_name
if self.check_train_in_block(block_name, train_name) == False:
#print "setting train", train_name, "in block", block_name
self.set_train_in_block(block_name, train_name)
def get_active_train(self, train_name):
DF = jmri.InstanceManager.getDefault(jmri.jmrit.dispatcher.DispatcherFrame)
#DF.setState(DF.ICONIFIED);
java_active_trains_list = DF.getActiveTrainsList()
java_active_trains_Arraylist= java.util.ArrayList(java_active_trains_list)
#print "java_active_trains_Arraylist",java_active_trains_Arraylist
#print "train_name", train_name
for t in java_active_trains_Arraylist:
#print "activetrainname=",t.getActiveTrainName()
#print "train_name", train_name
#print "t.getActiveTrainName().count(train_name)", t.getActiveTrainName().count(train_name)
if t.getActiveTrainName().count(train_name) >0: #check if train_name is contained in
return t
DF = None
return None
def get_occupied_blocks(self,active_train):
block_list = active_train.getBlockList()
section_list = active_train.getAllocatedSectionList()
start_block = active_train.getStartBlock()
end_block = active_train.getEndBlock()
seq_no = active_train.getStartBlockSectionSequenceNumber()
LastAllocatedSectionSeqNumber = active_train.getLastAllocatedSectionSeqNumber()
NextSectionToAllocate = active_train.getNextSectionToAllocate()
LastAllocatedSection = active_train.getLastAllocatedSection()
LastAllocatedSectionName = active_train.getLastAllocatedSectionName()
NextSectionToAllocateName = active_train.getNextSectionToAllocateName()
# print "block_list", block_list
# print "LastAllocatedSectionSeqNumber",LastAllocatedSectionSeqNumber
# print "section_list",section_list
# print "start block", start_block
# print "end_block", end_block
# print "getStartBlockSectionSequenceNumber", seq_no
# print "section_list", section_list
# print "NextSectionToAllocate", NextSectionToAllocate
# print "LastAllocatedSection", LastAllocatedSection
# print "LastAllocatedSectionName", LastAllocatedSectionName
# print "getNextSectionToAllocateName", NextSectionToAllocateName
return [block_list, start_block, end_block]
def check_train_in_block(self, block_name, train_name):
mem_val = self.get_blockcontents(block_name)
#print "mem_val", mem_val, "train_name", train_name
if train_name == mem_val:
#print "return true"
return True
else:
return False
def set_train_in_block(self, block_name, train_name):
self.set_blockcontents(block_name,train_name)
def get_block_position_of_train(self, train_name):
allocated_trains = self.get_allocated_trains()
def get_allocated_trains(self):
return trains_allocated
def get_non_allocated_trains(self):
all_trains = self.get_all_roster_entries_with_speed_profile()
non_allocated_trains = copy.copy(all_trains)
for train in trains_allocated:
if train in non_allocated_trains:
non_allocated_trains.remove(train)
return non_allocated_trains
def get_all_roster_entries_with_speed_profile(self):
roster_entries_with_speed_profile = []
r = jmri.jmrit.roster.Roster.getDefault()
for roster_entry in jmri.jmrit.roster.Roster.getAllEntries(r):
if self.logLevel > 0: print "roster_entry.getSpeedProfile()",roster_entry,roster_entry.getSpeedProfile()
if roster_entry.getSpeedProfile() != None:
roster_entries_with_speed_profile.append(roster_entry.getId())
if self.logLevel > 0: print "roster_entry.getId()",roster_entry.getId()
return roster_entries_with_speed_profile
def get_blockcontents(self, block_name):
block = blocks.getBlock(block_name)
value = block.getValue()
return value
def set_blockcontents(self, block_name, value):
block = blocks.getBlock(block_name)
value = block.setValue(value)
def get_station_and_occupancy_and_block_value_of_train(self, train_to_move):
## Check the pressed button
for station_block_name in g.station_block_list:
if self.logLevel > 0: print "station_block_name", station_block_name
#get a True if the block block_value has the train name in it
block_value_state = self.check_train_in_block(station_block_name, train_to_move)
block_occupancy_state = self.check_sensor_state_given_block_name(station_block_name)
if self.logLevel > 0: print "block_value_state1= ",block_value_state
# # do not attempt to move to where you are
# button_pressed_in_occupied_station = (button_station_name == station_block_name)
#check if the block is occupied and has the required train in it
if block_occupancy_state:
# and button_pressed_in_occupied_station == False:
return [station_block_name, block_value_state, block_occupancy_state]
return None
def get_position_of_train(self, train_to_move):
## Check the pressed button
for station_block_name in g.station_block_list:
if self.logLevel > 0: print "station_block_name", station_block_name
#get a True if the block block_value has the train name in it
block_value_state = self.check_train_in_block(station_block_name, train_to_move)
if self.logLevel > 0: print "block_value_state= ",block_value_state
#get a True if the block is occupied
block_occupied_state = self.check_sensor_state_given_block_name(station_block_name)
if self.logLevel > 0: print "block_occupied_state= ",block_occupied_state
if self.logLevel > 0: print ("station block name {} : {}". format(station_block_name, str(block_occupied_state)))
# # do not attempt to move to where you are
# button_pressed_in_occupied_station = (button_station_name == station_block_name)
#check if the block is occupied and has the required train in it
if block_value_state == True and block_occupied_state == True:
# and button_pressed_in_occupied_station == False:
return station_block_name
return None
def check_sensor_state_given_block_name(self, station_block_name):
#if self.logLevel > 0: print("station block name {}".format(station_block_name))
layoutBlock = layoutblocks.getLayoutBlock(station_block_name)
station_sensor = layoutBlock.getOccupancySensor()
if station_sensor is None:
OptionDialog().displayMessage(' Sensor in block {} not found'.format(station_block_name))
return
currentState = True if station_sensor.getKnownState() == ACTIVE else False
return currentState
# NewTrainMaster checksfor the new train in siding. Needs to inform what station we are in
#DispatchMaster checks all button sensors