class Ui_MainWindow(object):
def setupUi(self, MainWindow):
MainWindow.setObjectName(_fromUtf8("MainWindow"))
MainWindow.resize(1155, 853)
self.centralwidget = QWidget(MainWindow)
self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
self.verticalLayout = QVBoxLayout(self.centralwidget)
self.verticalLayout.setObjectName(_fromUtf8("verticalLayout"))
self.tabWidget = QTabWidget(self.centralwidget)
self.tabWidget.setObjectName(_fromUtf8("tabWidget"))
self.tab = QWidget()
self.tab.setObjectName(_fromUtf8("tab"))
self.tabWidget.addTab(self.tab, _fromUtf8(""))
self.tab_2 = QWidget()
self.tab_2.setObjectName(_fromUtf8("tab_2"))
self.tabWidget.addTab(self.tab_2, _fromUtf8(""))
self.verticalLayout.addWidget(self.tabWidget)
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QMenuBar(MainWindow)
self.menubar.setGeometry(QRect(0, 0, 1155, 20))
self.menubar.setObjectName(_fromUtf8("menubar"))
MainWindow.setMenuBar(self.menubar)
self.statusbar = QStatusBar(MainWindow)
self.statusbar.setObjectName(_fromUtf8("statusbar"))
MainWindow.setStatusBar(self.statusbar)
self.retranslateUi(MainWindow)
self.tabWidget.setCurrentIndex(0)
QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow):
MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow", None))
self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab),
_translate("MainWindow", "Tab 1", None))
self.tabWidget.setTabText(self.tabWidget.indexOf(self.tab_2),
_translate("MainWindow", "Tab 2", None))
class PlotWindow(QMainWindow):
def __init__(self, config_file, parent):
QMainWindow.__init__(self, parent)
self._api = PlotApi(ERT.enkf_facade)
self.setMinimumWidth(850)
self.setMinimumHeight(650)
self.setWindowTitle("Plotting - {}".format(config_file))
self.activateWindow()
self._key_definitions = self._api.all_data_type_keys()
self._plot_customizer = PlotCustomizer(self, self._key_definitions)
self._plot_customizer.settingsChanged.connect(self.keySelected)
self._central_tab = QTabWidget()
central_widget = QWidget()
central_layout = QVBoxLayout()
central_layout.setContentsMargins(0, 0, 0, 0)
central_widget.setLayout(central_layout)
central_layout.addWidget(self._central_tab)
self.setCentralWidget(central_widget)
self._plot_widgets = []
""":type: list of PlotWidget"""
self.addPlotWidget(ENSEMBLE, EnsemblePlot())
self.addPlotWidget(STATISTICS, StatisticsPlot())
self.addPlotWidget(HISTOGRAM, HistogramPlot())
self.addPlotWidget(GAUSSIAN_KDE, GaussianKDEPlot())
self.addPlotWidget(DISTRIBUTION, DistributionPlot())
self.addPlotWidget(CROSS_CASE_STATISTICS, CrossCaseStatisticsPlot())
self._central_tab.currentChanged.connect(self.currentPlotChanged)
cases = self._api.get_all_cases_not_running()
case_names = [case["name"] for case in cases if not case["hidden"]]
self._data_type_keys_widget = DataTypeKeysWidget(self._key_definitions)
self._data_type_keys_widget.dataTypeKeySelected.connect(self.keySelected)
self.addDock("Data types", self._data_type_keys_widget)
self._case_selection_widget = CaseSelectionWidget(case_names)
self._case_selection_widget.caseSelectionChanged.connect(self.keySelected)
self.addDock("Plot case", self._case_selection_widget)
current_plot_widget = self._plot_widgets[self._central_tab.currentIndex()]
self._data_type_keys_widget.selectDefault()
self._updateCustomizer(current_plot_widget)
def currentPlotChanged(self):
key_def = self.getSelectedKey()
key = key_def["key"]
for plot_widget in self._plot_widgets:
index = self._central_tab.indexOf(plot_widget)
if index == self._central_tab.currentIndex() \
and plot_widget._plotter.dimensionality == key_def["dimensionality"]:
self._updateCustomizer(plot_widget)
cases = self._case_selection_widget.getPlotCaseNames()
case_to_data_map = {case: self._api.data_for_key(case, key)[key] for case in cases}
if len(key_def["observations"]) > 0:
observations = self._api.observations_for_obs_keys(cases[0], key_def["observations"])
else:
observations = None
plot_config = PlotConfig.createCopy(self._plot_customizer.getPlotConfig())
plot_config.setTitle(key)
plot_context = PlotContext(plot_config, cases, key)
if key_def["has_refcase"]:
plot_context.refcase_data = self._api.refcase_data(key)
plot_widget.updatePlot(plot_context, case_to_data_map, observations)
class PlotWindow(QMainWindow):
def __init__(self, config_file, parent):
QMainWindow.__init__(self, parent)
self._ert = ERT.ert
""":type: res.enkf.enkf_main.EnKFMain"""
key_manager = self._ert.getKeyManager()
""":type: res.enkf.key_manager.KeyManager """
self.setMinimumWidth(850)
self.setMinimumHeight(650)
self.setWindowTitle("Plotting - {}".format(config_file))
self.activateWindow()
self._plot_customizer = PlotCustomizer(self)
def plotConfigCreator(key):
return PlotConfigFactory.createPlotConfigForKey(self._ert, key)
self._plot_customizer.setPlotConfigCreator(plotConfigCreator)
self._plot_customizer.settingsChanged.connect(self.keySelected)
self._central_tab = QTabWidget()
self._central_tab.currentChanged.connect(self.currentPlotChanged)
central_widget = QWidget()
central_layout = QVBoxLayout()
central_layout.setContentsMargins(0, 0, 0, 0)
central_widget.setLayout(central_layout)
central_layout.addWidget(self._central_tab)
self.setCentralWidget(central_widget)
self._plot_widgets = []
""":type: list of PlotWidget"""
self._data_gatherers = []
""":type: list of PlotDataGatherer """
summary_gatherer = self.createDataGatherer(PDG.gatherSummaryData, key_manager.isSummaryKey, refcaseGatherFunc=PDG.gatherSummaryRefcaseData, observationGatherFunc=PDG.gatherSummaryObservationData, historyGatherFunc=PDG.gatherSummaryHistoryData)
gen_data_gatherer = self.createDataGatherer(PDG.gatherGenDataData, key_manager.isGenDataKey, observationGatherFunc=PDG.gatherGenDataObservationData)
gen_kw_gatherer = self.createDataGatherer(PDG.gatherGenKwData, key_manager.isGenKwKey)
custom_kw_gatherer = self.createDataGatherer(PDG.gatherCustomKwData, key_manager.isCustomKwKey)
self.addPlotWidget(ENSEMBLE, plots.plotEnsemble, [summary_gatherer, gen_data_gatherer])
self.addPlotWidget(STATISTICS, plots.plotStatistics, [summary_gatherer, gen_data_gatherer])
self.addPlotWidget(HISTOGRAM, plots.plotHistogram, [gen_kw_gatherer, custom_kw_gatherer])
self.addPlotWidget(GAUSSIAN_KDE, plots.plotGaussianKDE, [gen_kw_gatherer, custom_kw_gatherer])
self.addPlotWidget(DISTRIBUTION, plots.plotDistribution, [gen_kw_gatherer, custom_kw_gatherer])
self.addPlotWidget(CROSS_CASE_STATISTICS, plots.plotCrossCaseStatistics, [gen_kw_gatherer, custom_kw_gatherer])
data_types_key_model = DataTypeKeysListModel(self._ert)
self._data_type_keys_widget = DataTypeKeysWidget(data_types_key_model)
self._data_type_keys_widget.dataTypeKeySelected.connect(self.keySelected)
self.addDock("Data types", self._data_type_keys_widget)
current_case = getCurrentCaseName()
self._case_selection_widget = CaseSelectionWidget(current_case)
self._case_selection_widget.caseSelectionChanged.connect(self.keySelected)
self.addDock("Plot case", self._case_selection_widget)
current_plot_widget = self._plot_widgets[self._central_tab.currentIndex()]
current_plot_widget.setActive()
self._data_type_keys_widget.selectDefault()
self._updateCustomizer(current_plot_widget)
def createDataGatherer(self, dataGatherFunc, gatherConditionFunc, refcaseGatherFunc=None, observationGatherFunc=None, historyGatherFunc=None):
data_gatherer = PDG(dataGatherFunc, gatherConditionFunc, refcaseGatherFunc=refcaseGatherFunc, observationGatherFunc=observationGatherFunc, historyGatherFunc=historyGatherFunc)
self._data_gatherers.append(data_gatherer)
return data_gatherer
def currentPlotChanged(self):
for plot_widget in self._plot_widgets:
plot_widget.setActive(False)
index = self._central_tab.indexOf(plot_widget)
if index == self._central_tab.currentIndex() and plot_widget.canPlotKey(self.getSelectedKey()):
plot_widget.setActive()
self._updateCustomizer(plot_widget)
plot_widget.updatePlot()
def _updateCustomizer(self, plot_widget):
""" @type plot_widget: PlotWidget """
key = self.getSelectedKey()
key_manager = self._ert.getKeyManager()
index_type = PlotContext.UNKNOWN_AXIS
if key_manager.isGenDataKey(key):
index_type = PlotContext.INDEX_AXIS
elif key_manager.isSummaryKey(key):
index_type = PlotContext.DATE_AXIS
x_axis_type = PlotContext.UNKNOWN_AXIS
y_axis_type = PlotContext.UNKNOWN_AXIS
if plot_widget.name == ENSEMBLE:
x_axis_type = index_type
y_axis_type = PlotContext.VALUE_AXIS
elif plot_widget.name == STATISTICS:
x_axis_type = index_type
y_axis_type = PlotContext.VALUE_AXIS
elif plot_widget.name == DISTRIBUTION:
y_axis_type = PlotContext.VALUE_AXIS
elif plot_widget.name == CROSS_CASE_STATISTICS:
y_axis_type = PlotContext.VALUE_AXIS
elif plot_widget.name == HISTOGRAM:
x_axis_type = PlotContext.VALUE_AXIS
y_axis_type = PlotContext.COUNT_AXIS
elif plot_widget.name == GAUSSIAN_KDE:
x_axis_type = PlotContext.VALUE_AXIS
y_axis_type = PlotContext.DENSITY_AXIS
self._plot_customizer.setAxisTypes(x_axis_type, y_axis_type)
def createPlotContext(self, figure):
key = self.getSelectedKey()
cases = self._case_selection_widget.getPlotCaseNames()
data_gatherer = self.getDataGathererForKey(key)
plot_config = PlotConfig.createCopy(self._plot_customizer.getPlotConfig())
plot_config.setTitle(key)
return PlotContext(self._ert, figure, plot_config, cases, key, data_gatherer)
def getDataGathererForKey(self, key):
""" @rtype: PlotDataGatherer """
return next((data_gatherer for data_gatherer in self._data_gatherers if data_gatherer.canGatherDataForKey(key)), None)
def getSelectedKey(self):
return str(self._data_type_keys_widget.getSelectedItem())
def addPlotWidget(self, name, plotFunction, data_gatherers, enabled=True):
plot_condition_function_list = [data_gatherer.canGatherDataForKey for data_gatherer in data_gatherers]
plot_widget = PlotWidget(name, plotFunction, plot_condition_function_list, self.createPlotContext)
plot_widget.customizationTriggered.connect(self.toggleCustomizeDialog)
index = self._central_tab.addTab(plot_widget, name)
self._plot_widgets.append(plot_widget)
self._central_tab.setTabEnabled(index, enabled)
def addDock(self, name, widget, area=Qt.LeftDockWidgetArea, allowed_areas=Qt.AllDockWidgetAreas):
dock_widget = QDockWidget(name)
dock_widget.setObjectName("%sDock" % name)
dock_widget.setWidget(widget)
dock_widget.setAllowedAreas(allowed_areas)
dock_widget.setFeatures(QDockWidget.DockWidgetFloatable | QDockWidget.DockWidgetMovable)
self.addDockWidget(area, dock_widget)
return dock_widget
@showWaitCursorWhileWaiting
def keySelected(self):
key = self.getSelectedKey()
self._plot_customizer.switchPlotConfigHistory(key)
for plot_widget in self._plot_widgets:
plot_widget.setDirty()
index = self._central_tab.indexOf(plot_widget)
self._central_tab.setTabEnabled(index, plot_widget.canPlotKey(key))
for plot_widget in self._plot_widgets:
if plot_widget.canPlotKey(key):
plot_widget.updatePlot()
def toggleCustomizeDialog(self):
self._plot_customizer.toggleCustomizationDialog()
