class RunTemplatePanel(HelpedWidget):
"""
Widget for adding, removing and editing run templates.
"""
def __init__(self, parent=None):
HelpedWidget.__init__(self, parent, "",
"config/simulation/run_template")
self.run_template = RunTemplate("undefined", "", "", "")
self.createWidgets(parent)
self.emptyPanel = ert_gui.widgets.util.createEmptyPanel()
self.pagesWidget = QtGui.QStackedWidget()
self.pagesWidget.addWidget(self.emptyPanel)
self.pagesWidget.addWidget(self.run_template_panel)
self.addWidget(self.pagesWidget)
self.addHelpButton()
def createWidgets(self, parent):
self.searchableList = SearchableList(parent,
list_height=150,
list_width=150,
ignore_case=True)
self.addWidget(self.searchableList)
self.connect(
self.searchableList,
QtCore.SIGNAL(
'currentItemChanged(QListWidgetItem, QListWidgetItem)'),
self.changeParameter)
self.connect(self.searchableList, QtCore.SIGNAL('addItem(list)'),
self.addItem)
self.connect(self.searchableList, QtCore.SIGNAL('removeItem(list)'),
self.removeItem)
self.run_template_panel = ert_gui.widgets.util.createEmptyPanel()
layout = QtGui.QFormLayout()
layout.setLabelAlignment(QtCore.Qt.AlignRight)
self.run_template_file = PathChooser(
self, "", "config/simulation/run_template_file", show_files=True)
self.run_template_file.initialize = ContentModel.emptyInitializer
self.run_template_file.setter = self.setTemplateFile
self.run_template_file.getter = lambda model: self.run_template.template_file
self.run_template_target = PathChooser(
self, "", "config/simulation/run_template_target", show_files=True)
self.run_template_target.initialize = ContentModel.emptyInitializer
self.run_template_target.setter = self.setTargetFile
self.run_template_target.getter = lambda model: self.run_template.target_file
self.run_template_args = StringBox(
self, "", "config/simulation/run_template_arguments")
self.run_template_args.initialize = ContentModel.emptyInitializer
self.run_template_args.setter = self.setArguments
self.run_template_args.getter = lambda model: self.run_template.arguments
layout.addRow("Template file:", self.run_template_file)
layout.addRow("Target file:", self.run_template_target)
layout.addRow("Arguments:", self.run_template_args)
layout.addRow(ert_gui.widgets.util.createSpace(20))
self.run_template_panel.setLayout(layout)
def setArguments(self, model, arguments):
self.run_template.setArguments(arguments)
self.runTemplateChanged()
def setTemplateFile(self, model, template_file):
self.run_template.setTemplateFile(template_file)
self.runTemplateChanged()
def setTargetFile(self, model, target_file):
self.run_template.setTargetFile(target_file)
self.runTemplateChanged()
def fetchContent(self):
"""
Retrieves data from the model and inserts it into the widget.
List of tuples: (name, template_file, target_file, arguments)
"""
data = self.getFromModel()
self.searchableList.list.clear()
for item in data:
jobitem = QtGui.QListWidgetItem()
jobitem.setText(item[0])
run_template = RunTemplate(item[0], item[1], item[2], item[3])
jobitem.setData(QtCore.Qt.UserRole, run_template)
jobitem.setToolTip(item[0])
self.searchableList.list.addItem(jobitem)
def setRunTemplate(self, run_template):
"""Set the current visible run template"""
self.run_template = run_template
self.run_template_args.fetchContent()
self.run_template_file.fetchContent()
self.run_template_target.fetchContent()
def changeParameter(self, current, previous):
"""Switch between run templates. Selection from the list"""
if current is None:
self.pagesWidget.setCurrentWidget(self.emptyPanel)
else:
self.pagesWidget.setCurrentWidget(self.run_template_panel)
self.setRunTemplate(current.data(QtCore.Qt.UserRole).toPyObject())
def runTemplateChanged(self):
"""
Called whenever the run template is changed. (adding, removing)
The data submitted to the updateContent() (from ContentModel) is a list of tuples (name, template_file, target_file, arguments)
"""
items = self.searchableList.getItems()
currentRow = self.searchableList.list.currentRow()
run_templates = []
for item in items:
r_t = item.data(QtCore.Qt.UserRole).toPyObject()
run_template_tuple = (r_t.name, r_t.template_file, r_t.target_file,
r_t.arguments)
run_templates.append(run_template_tuple)
self.updateContent(run_templates)
self.fetchContent()
self.searchableList.list.setCurrentRow(currentRow)
def addToList(self, list, name):
"""Adds a new run template to the list"""
param = QtGui.QListWidgetItem()
param.setText(name)
new_job = RunTemplate(name)
param.setData(QtCore.Qt.UserRole, new_job)
list.addItem(param)
list.setCurrentItem(param)
return param
def addItem(self, list):
"""Called by the add button to insert a new run template"""
uniqueNames = []
for index in range(list.count()):
uniqueNames.append(str(list.item(index).text()))
pd = ValidatedDialog(self, "New run template",
"Enter name of new run template:", uniqueNames)
if pd.exec_():
self.addToList(list, pd.getName())
self.runTemplateChanged()
def removeItem(self, list):
"""Called by the remove button to remove a selected job"""
currentRow = list.currentRow()
if currentRow >= 0:
title = "Delete run template?"
msg = "Are you sure you want to delete the run template?"
btns = QtGui.QMessageBox.Yes | QtGui.QMessageBox.No
doDelete = QtGui.QMessageBox.question(self, title, msg, btns)
if doDelete == QtGui.QMessageBox.Yes:
list.takeItem(currentRow)
self.runTemplateChanged()
class RunTemplatePanel(HelpedWidget):
"""
Widget for adding, removing and editing run templates.
"""
def __init__(self, parent=None):
HelpedWidget.__init__(self, parent, "", "config/simulation/run_template")
self.run_template = RunTemplate("undefined", "", "", "")
self.createWidgets(parent)
self.emptyPanel = ert_gui.widgets.util.createEmptyPanel()
self.pagesWidget = QtGui.QStackedWidget()
self.pagesWidget.addWidget(self.emptyPanel)
self.pagesWidget.addWidget(self.run_template_panel)
self.addWidget(self.pagesWidget)
self.addHelpButton()
def createWidgets(self, parent):
self.searchableList = SearchableList(parent, list_height=150, list_width=150, ignore_case=True)
self.addWidget(self.searchableList)
self.connect(self.searchableList, QtCore.SIGNAL('currentItemChanged(QListWidgetItem, QListWidgetItem)'),
self.changeParameter)
self.connect(self.searchableList, QtCore.SIGNAL('addItem(list)'), self.addItem)
self.connect(self.searchableList, QtCore.SIGNAL('removeItem(list)'), self.removeItem)
self.run_template_panel = ert_gui.widgets.util.createEmptyPanel()
layout = QtGui.QFormLayout()
layout.setLabelAlignment(QtCore.Qt.AlignRight)
self.run_template_file = PathChooser(self, "", "config/simulation/run_template_file", show_files=True)
self.run_template_file.initialize = ContentModel.emptyInitializer
self.run_template_file.setter = self.setTemplateFile
self.run_template_file.getter = lambda model: self.run_template.template_file
self.run_template_target = PathChooser(self, "", "config/simulation/run_template_target", show_files=True)
self.run_template_target.initialize = ContentModel.emptyInitializer
self.run_template_target.setter = self.setTargetFile
self.run_template_target.getter = lambda model: self.run_template.target_file
self.run_template_args = StringBox(self, "", "config/simulation/run_template_arguments")
self.run_template_args.initialize = ContentModel.emptyInitializer
self.run_template_args.setter = self.setArguments
self.run_template_args.getter = lambda model: self.run_template.arguments
layout.addRow("Template file:", self.run_template_file)
layout.addRow("Target file:", self.run_template_target)
layout.addRow("Arguments:", self.run_template_args)
layout.addRow(ert_gui.widgets.util.createSpace(20))
self.run_template_panel.setLayout(layout)
def setArguments(self, model, arguments):
self.run_template.setArguments(arguments)
self.runTemplateChanged()
def setTemplateFile(self, model, template_file):
self.run_template.setTemplateFile(template_file)
self.runTemplateChanged()
def setTargetFile(self, model, target_file):
self.run_template.setTargetFile(target_file)
self.runTemplateChanged()
def fetchContent(self):
"""
Retrieves data from the model and inserts it into the widget.
List of tuples: (name, template_file, target_file, arguments)
"""
data = self.getFromModel()
self.searchableList.list.clear()
for item in data:
jobitem = QtGui.QListWidgetItem()
jobitem.setText(item[0])
run_template = RunTemplate(item[0], item[1], item[2], item[3])
jobitem.setData(QtCore.Qt.UserRole, run_template)
jobitem.setToolTip(item[0])
self.searchableList.list.addItem(jobitem)
def setRunTemplate(self, run_template):
"""Set the current visible run template"""
self.run_template = run_template
self.run_template_args.fetchContent()
self.run_template_file.fetchContent()
self.run_template_target.fetchContent()
def changeParameter(self, current, previous):
"""Switch between run templates. Selection from the list"""
if current is None:
self.pagesWidget.setCurrentWidget(self.emptyPanel)
else:
self.pagesWidget.setCurrentWidget(self.run_template_panel)
self.setRunTemplate(current.data(QtCore.Qt.UserRole).toPyObject())
def runTemplateChanged(self):
"""
Called whenever the run template is changed. (adding, removing)
The data submitted to the updateContent() (from ContentModel) is a list of tuples (name, template_file, target_file, arguments)
"""
items = self.searchableList.getItems()
currentRow = self.searchableList.list.currentRow()
run_templates = []
for item in items:
r_t = item.data(QtCore.Qt.UserRole).toPyObject()
run_template_tuple = (r_t.name, r_t.template_file, r_t.target_file, r_t.arguments)
run_templates.append(run_template_tuple)
self.updateContent(run_templates)
self.fetchContent()
self.searchableList.list.setCurrentRow(currentRow)
def addToList(self, list, name):
"""Adds a new run template to the list"""
param = QtGui.QListWidgetItem()
param.setText(name)
new_job = RunTemplate(name)
param.setData(QtCore.Qt.UserRole, new_job)
list.addItem(param)
list.setCurrentItem(param)
return param
def addItem(self, list):
"""Called by the add button to insert a new run template"""
uniqueNames = []
for index in range(list.count()):
uniqueNames.append(str(list.item(index).text()))
pd = ValidatedDialog(self, "New run template", "Enter name of new run template:", uniqueNames)
if pd.exec_():
self.addToList(list, pd.getName())
self.runTemplateChanged()
def removeItem(self, list):
"""Called by the remove button to remove a selected job"""
currentRow = list.currentRow()
if currentRow >= 0:
title = "Delete run template?"
msg = "Are you sure you want to delete the run template?"
btns = QtGui.QMessageBox.Yes | QtGui.QMessageBox.No
doDelete = QtGui.QMessageBox.question(self, title, msg, btns)
if doDelete == QtGui.QMessageBox.Yes:
list.takeItem(currentRow)
self.runTemplateChanged()
class ForwardModelPanel(HelpedWidget):
"""
Widget for adding, removing and editing forward models.
Only uses the setter and getter hooks.
The panel expects remove to return True or False based on the success of the removal.
"""
def __init__(self, parent=None):
HelpedWidget.__init__(self, parent, "Forward Model", "config/simulation/forward_model")
self.forward_model_job = ForwardModelJob("undefined")
self.createWidgets(parent)
self.emptyPanel = ert_gui.widgets.util.createEmptyPanel()
self.pagesWidget = QtGui.QStackedWidget()
self.pagesWidget.addWidget(self.emptyPanel)
self.pagesWidget.addWidget(self.forward_model_panel)
self.addWidget(self.pagesWidget)
self.addHelpButton()
def createWidgets(self, parent):
self.searchableList = SearchableList(parent, list_height=150, list_width=150, ignore_case=True, order_editable=True)
self.addWidget(self.searchableList)
self.connect(self.searchableList, QtCore.SIGNAL('currentItemChanged(QListWidgetItem, QListWidgetItem)'),
self.changeParameter)
self.connect(self.searchableList, QtCore.SIGNAL('addItem(list)'), self.addItem)
self.connect(self.searchableList, QtCore.SIGNAL('removeItem(list)'), self.removeItem)
self.connect(self.searchableList, QtCore.SIGNAL('orderChanged(list)'), self.forwardModelChanged)
self.forward_model_panel = ert_gui.widgets.util.createEmptyPanel()
layout = QtGui.QFormLayout()
layout.setLabelAlignment(QtCore.Qt.AlignRight)
self.forward_model_args = StringBox(self, "", "config/simulation/forward_model_arguments")
self.forward_model_args.initialize = ContentModel.emptyInitializer
self.forward_model_args.setter = self.setArguments
self.forward_model_args.getter = lambda model: self.forward_model_job.arguments
layout.addRow("Arguments:", self.forward_model_args)
layout.addRow(ert_gui.widgets.util.createSpace(20))
self.help_text = QtGui.QLabel()
self.help_text.setText("")
layout.addRow(ert_gui.widgets.util.centeredWidget(self.help_text))
self.forward_model_panel.setLayout(layout)
self.modelConnect('jobListChanged()', self.fetchContent)
def setArguments(self, model, arguments):
"""Set the arguments of the current forward model job."""
self.forward_model_job.setArguments(arguments)
self.forwardModelChanged()
def fetchContent(self):
"""
Retrieves data from the model and inserts it into the widget.
Expects a hash containing these two keys: available_jobs and forward_model.
available_jobs=list of names
forward_model)list of tuples containing(name, arguments, help_text)
"""
data = self.getFromModel()
self.available_jobs = data['available_jobs']
forward_model = data['forward_model']
self.searchableList.list.clear()
for job in forward_model:
jobitem = QtGui.QListWidgetItem()
jobitem.setText(job[0])
forward_model_job = ForwardModelJob(job[0])
forward_model_job.setArguments(job[1])
forward_model_job.setHelpText(job[2])
jobitem.setData(QtCore.Qt.UserRole, forward_model_job)
jobitem.setToolTip(job[0])
self.searchableList.list.addItem(jobitem)
def setForwardModelJob(self, forward_model_job):
"""Set the current visible forward model job"""
self.forward_model_job = forward_model_job
self.help_text.setText(forward_model_job.help_text)
self.forward_model_args.fetchContent()
def changeParameter(self, current, previous):
"""Switch between forward models. Selection from the list"""
if current is None:
self.pagesWidget.setCurrentWidget(self.emptyPanel)
else:
self.pagesWidget.setCurrentWidget(self.forward_model_panel)
self.setForwardModelJob(current.data(QtCore.Qt.UserRole).toPyObject())
def forwardModelChanged(self):
"""
Called whenever the forward model is changed. (reordering, adding, removing)
The data submitted to the updateContent() (from ContentModel) is a list of tuples (name, arguments)
"""
items = self.searchableList.getItems()
currentRow = self.searchableList.list.currentRow()
forward_model = []
for item in items:
forward_model_job = item.data(QtCore.Qt.UserRole).toPyObject()
forward_model.append((forward_model_job.name, forward_model_job.arguments))
self.updateContent(forward_model)
self.fetchContent()
self.searchableList.list.setCurrentRow(currentRow)
def addToList(self, list, name):
"""Adds a new job to the list"""
param = QtGui.QListWidgetItem()
param.setText(name)
new_job = ForwardModelJob(name)
param.setData(QtCore.Qt.UserRole, new_job)
list.addItem(param)
list.setCurrentItem(param)
return param
def addItem(self, list):
"""Called by the add button to insert a new job"""
pd = ValidatedDialog(self, "New forward model job", "Select a job:", self.available_jobs, True)
if pd.exec_():
self.addToList(list, pd.getName())
self.forwardModelChanged()
def removeItem(self, list):
"""Called by the remove button to remove a selected job"""
currentRow = list.currentRow()
if currentRow >= 0:
title = "Delete forward model job?"
msg = "Are you sure you want to delete the job from the forward model?"
btns = QtGui.QMessageBox.Yes | QtGui.QMessageBox.No
doDelete = QtGui.QMessageBox.question(self, title, msg, btns)
if doDelete == QtGui.QMessageBox.Yes:
list.takeItem(currentRow)
self.forwardModelChanged()
class ForwardModelPanel(HelpedWidget):
"""
Widget for adding, removing and editing forward models.
Only uses the setter and getter hooks.
The panel expects remove to return True or False based on the success of the removal.
"""
def __init__(self, parent=None):
HelpedWidget.__init__(self, parent, "Forward Model",
"config/simulation/forward_model")
self.forward_model_job = ForwardModelJob("undefined")
self.createWidgets(parent)
self.emptyPanel = ert_gui.widgets.util.createEmptyPanel()
self.pagesWidget = QtGui.QStackedWidget()
self.pagesWidget.addWidget(self.emptyPanel)
self.pagesWidget.addWidget(self.forward_model_panel)
self.addWidget(self.pagesWidget)
self.addHelpButton()
def createWidgets(self, parent):
self.searchableList = SearchableList(parent,
list_height=150,
list_width=150,
ignore_case=True,
order_editable=True)
self.addWidget(self.searchableList)
self.connect(
self.searchableList,
QtCore.SIGNAL(
'currentItemChanged(QListWidgetItem, QListWidgetItem)'),
self.changeParameter)
self.connect(self.searchableList, QtCore.SIGNAL('addItem(list)'),
self.addItem)
self.connect(self.searchableList, QtCore.SIGNAL('removeItem(list)'),
self.removeItem)
self.connect(self.searchableList, QtCore.SIGNAL('orderChanged(list)'),
self.forwardModelChanged)
self.forward_model_panel = ert_gui.widgets.util.createEmptyPanel()
layout = QtGui.QFormLayout()
layout.setLabelAlignment(QtCore.Qt.AlignRight)
self.forward_model_args = StringBox(
self, "", "config/simulation/forward_model_arguments")
self.forward_model_args.initialize = ContentModel.emptyInitializer
self.forward_model_args.setter = self.setArguments
self.forward_model_args.getter = lambda model: self.forward_model_job.arguments
layout.addRow("Arguments:", self.forward_model_args)
layout.addRow(ert_gui.widgets.util.createSpace(20))
self.help_text = QtGui.QLabel()
self.help_text.setText("")
layout.addRow(ert_gui.widgets.util.centeredWidget(self.help_text))
self.forward_model_panel.setLayout(layout)
self.modelConnect('jobListChanged()', self.fetchContent)
def setArguments(self, model, arguments):
"""Set the arguments of the current forward model job."""
self.forward_model_job.setArguments(arguments)
self.forwardModelChanged()
def fetchContent(self):
"""
Retrieves data from the model and inserts it into the widget.
Expects a hash containing these two keys: available_jobs and forward_model.
available_jobs=list of names
forward_model)list of tuples containing(name, arguments, help_text)
"""
data = self.getFromModel()
self.available_jobs = data['available_jobs']
forward_model = data['forward_model']
self.searchableList.list.clear()
for job in forward_model:
jobitem = QtGui.QListWidgetItem()
jobitem.setText(job[0])
forward_model_job = ForwardModelJob(job[0])
forward_model_job.setArguments(job[1])
forward_model_job.setHelpText(job[2])
jobitem.setData(QtCore.Qt.UserRole, forward_model_job)
jobitem.setToolTip(job[0])
self.searchableList.list.addItem(jobitem)
def setForwardModelJob(self, forward_model_job):
"""Set the current visible forward model job"""
self.forward_model_job = forward_model_job
self.help_text.setText(forward_model_job.help_text)
self.forward_model_args.fetchContent()
def changeParameter(self, current, previous):
"""Switch between forward models. Selection from the list"""
if current is None:
self.pagesWidget.setCurrentWidget(self.emptyPanel)
else:
self.pagesWidget.setCurrentWidget(self.forward_model_panel)
self.setForwardModelJob(
current.data(QtCore.Qt.UserRole).toPyObject())
def forwardModelChanged(self):
"""
Called whenever the forward model is changed. (reordering, adding, removing)
The data submitted to the updateContent() (from ContentModel) is a list of tuples (name, arguments)
"""
items = self.searchableList.getItems()
currentRow = self.searchableList.list.currentRow()
forward_model = []
for item in items:
forward_model_job = item.data(QtCore.Qt.UserRole).toPyObject()
forward_model.append(
(forward_model_job.name, forward_model_job.arguments))
self.updateContent(forward_model)
self.fetchContent()
self.searchableList.list.setCurrentRow(currentRow)
def addToList(self, list, name):
"""Adds a new job to the list"""
param = QtGui.QListWidgetItem()
param.setText(name)
new_job = ForwardModelJob(name)
param.setData(QtCore.Qt.UserRole, new_job)
list.addItem(param)
list.setCurrentItem(param)
return param
def addItem(self, list):
"""Called by the add button to insert a new job"""
pd = ValidatedDialog(self, "New forward model job", "Select a job:",
self.available_jobs, True)
if pd.exec_():
self.addToList(list, pd.getName())
self.forwardModelChanged()
def removeItem(self, list):
"""Called by the remove button to remove a selected job"""
currentRow = list.currentRow()
if currentRow >= 0:
title = "Delete forward model job?"
msg = "Are you sure you want to delete the job from the forward model?"
btns = QtGui.QMessageBox.Yes | QtGui.QMessageBox.No
doDelete = QtGui.QMessageBox.question(self, title, msg, btns)
if doDelete == QtGui.QMessageBox.Yes:
list.takeItem(currentRow)
self.forwardModelChanged()
