def __init__(self):
QtGui.QMainWindow.__init__(self)
# This is always the same
self.ui=Ui_MainWindow()
self.ui.setupUi(self)
# models used in the respective listView
self.model_available = QtGui.QStandardItemModel()
self.model_selected = QtGui.QStandardItemModel()
# connect the buttons press with the correct function
self.ui.button_add.clicked.connect(self.addConf)
self.ui.button_remove.clicked.connect(self.rmConf)
self.ui.button_execute.clicked.connect(self.executeCastalia)
# connect change button to openFile method
self.ui.button_open_conf_file.clicked.connect(self.openFile)
# set multiselection in the lists
self.ui.list_conf_available.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
self.ui.list_conf_selected.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
self.conf_filename = "omnetpp.ini"
# check if the file omnetpp.ini exist
# if not show a message where say to choose a configuration file
# cause you don't find omnetpp.ini
try:
in_file = open(self.conf_filename,"r")
except IOError:
self.ui.label_show_name.setText("-----------")
else:
self.ui.label_show_name.setText("omnetpp.ini")
self.loadConfigFile()
in_file.close()
class Main(QtGui.QMainWindow):
def __init__(self):
QtGui.QMainWindow.__init__(self)
# This is always the same
self.ui=Ui_MainWindow()
self.ui.setupUi(self)
# models used in the respective listView
self.model_available = QtGui.QStandardItemModel()
self.model_selected = QtGui.QStandardItemModel()
# connect the buttons press with the correct function
self.ui.button_add.clicked.connect(self.addConf)
self.ui.button_remove.clicked.connect(self.rmConf)
self.ui.button_execute.clicked.connect(self.executeCastalia)
# connect change button to openFile method
self.ui.button_open_conf_file.clicked.connect(self.openFile)
# set multiselection in the lists
self.ui.list_conf_available.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
self.ui.list_conf_selected.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
self.conf_filename = "omnetpp.ini"
# check if the file omnetpp.ini exist
# if not show a message where say to choose a configuration file
# cause you don't find omnetpp.ini
try:
in_file = open(self.conf_filename,"r")
except IOError:
self.ui.label_show_name.setText("-----------")
else:
self.ui.label_show_name.setText("omnetpp.ini")
self.loadConfigFile()
in_file.close()
def loadConfigFile(self):
""" read the omnet.ini file and find the configurations name. """
try:
in_file = open(self.conf_filename,"r")
file_content = in_file.read()
except IOError:
QtGui.QMessageBox.warning(self, "Error", "Can\'t find file"+str(self.conf_filename)+" or read data")
in_file.close()
return
in_file.close()
names = []
# there is always an entry called "General"
matches = re.findall(r"\[General\]", file_content)
if matches == []:
QtGui.QMessageBox.warning(self, "Error", "This file is not a configuration file, please choose another one")
return
names.append("General")
# find all [Config <something>] match in the
matches = re.findall(r"Config [a-zA-Z0-9]+", file_content)
# for each match we clean it in order to have only the name of the configuration
for match in matches:
name = match.replace("Config", "")
name = name.replace(" ","")
names.append(name)
# fill the listView with all the configuration names
self.fillConfigAvailable(names)
# set the current configuration filename
self.ui.label_show_name.setText(str(self.conf_filename))
def openFile(self):
fName = QtGui.QFileDialog.getOpenFileName(self, "Open file", "Open file", self.tr("*"))
if fName.isEmpty():
return
self.conf_filename = fName
self.loadConfigFile()
def fillConfigAvailable(self, configs):
# clear the model
if self.model_available.rowCount() != 0:
self.model_available.clear()
if self.model_selected.rowCount() != 0:
self.model_selected.clear()
for conf in configs:
# Create an item with a caption
item = QtGui.QStandardItem(conf)
self.model_available.appendRow(item)
self.ui.list_conf_available.setModel(self.model_available)
def addConf(self):
""" add selected item to list_conf_selected """
selected = []
already_present = []
# retrieve all the configuration in list_conf_available
indexes = self.ui.list_conf_available.selectedIndexes()
for index in indexes:
selected.append (self.model_available.itemFromIndex(index).text() )
# retrieve all the configuration in list_conf_selected
# look in its model
row_count = self.model_selected.rowCount()
for row in range(row_count):
already_present.append(self.model_selected.item(row).text())
# compute the new names not already presents in list_conf_selected
new_names = list(set(selected) - set(already_present))
# add to the list_conf_selected the new names
for name in new_names:
# Create an item with a caption
item = QtGui.QStandardItem(name)
self.model_selected.appendRow(item)
self.ui.list_conf_selected.setModel(self.model_selected)
# un-select the list_conf_available
def rmConf(self):
""" delete the selected names from list_conf_selected """
indexes = self.ui.list_conf_selected.selectedIndexes()
for index in indexes:
self.model_selected.removeRow(index.row())
self.ui.list_conf_selected.setModel(self.model_selected)
def executeCastalia(self):
""" it's time to execute Castalia to start the simulation/s """
error = False
# retrieve the number of the runs
# print a error if is something different from a number
try:
runs = int( self.ui.run_number.displayText() )
except ValueError:
# show a dialog
QtGui.QMessageBox.warning(self, "Warning", "Must use only integer number.")
error = True
# if there is not any configuration in "Selected" or no run number
# print a message
row_count = self.model_selected.rowCount()
configs = []
if row_count == 0:
QtGui.QMessageBox.warning(self, "Warning","You must selected almost one configuration.")
error = True
else:
for row in range(row_count):
configs.append( str(self.model_selected.item(row).text()) )
# ok everythings seems fine, time to execute Castalia and wait it
if not error:
command = "Castalia "
command += "-i "+str(self.conf_filename)
command += " -c "
for i in range(len(configs)):
if i == 0:
command += configs[i]
else:
command += ","+configs[i]
command += " -r "+str(runs)
# disable the execute button until the simulation/s is/are not ended
self.ui.button_execute.setDisabled(True)
# call and wait
subprocess.call(command, shell=True, stderr=None, stdout=None )
QtGui.QMessageBox.information(self, "End", "Simulations ended")
# ok simulations ended
self.ui.button_execute.setDisabled(False)
