def __init__(self, config_file, storage_client, parent):
QMainWindow.__init__(self, parent)
if storage_client:
self._api = storage_client
else:
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 __init__(self, 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")
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 __init__(self, parent):
QMainWindow.__init__(self, parent)
self.setMinimumWidth(750)
self.setMinimumHeight(500)
self.setWindowTitle("Plotting")
self.activateWindow()
self.__plot_data = None
self.__plot_metrics_tracker = PlotMetricsTracker()
self.__plot_metrics_tracker.addScaleType("value", float)
self.__plot_metrics_tracker.addScaleType("depth", float)
self.__plot_metrics_tracker.addScaleType("pca", float)
self.__plot_metrics_tracker.addScaleType("index", int)
self.__plot_metrics_tracker.addScaleType("count", int)
self.__plot_metrics_tracker.addScaleType("time", CTime)
self.__central_tab = QTabWidget()
self.__central_tab.currentChanged.connect(self.currentPlotChanged)
self.__plot_panel_tracker = PlotPanelTracker(self.__central_tab)
self.__plot_panel_tracker.addKeyTypeTester(
"summary", PlotDataFetcher.isSummaryKey)
self.__plot_panel_tracker.addKeyTypeTester(
"block", PlotDataFetcher.isBlockObservationKey)
self.__plot_panel_tracker.addKeyTypeTester("gen_kw",
PlotDataFetcher.isGenKWKey)
self.__plot_panel_tracker.addKeyTypeTester(
"gen_data", PlotDataFetcher.isGenDataKey)
self.__plot_panel_tracker.addKeyTypeTester(
"custom_pca", PlotDataFetcher.isCustomPcaDataKey)
central_widget = QWidget()
central_layout = QVBoxLayout()
central_layout.setContentsMargins(0, 0, 0, 0)
central_widget.setLayout(central_layout)
self.__toolbar = PlotToolBar()
self.__toolbar.resetScalesClicked.connect(self.resetCurrentScales)
central_layout.addWidget(self.__toolbar)
central_layout.addWidget(self.__central_tab)
self.setCentralWidget(central_widget)
self.__plot_panels = []
""":type: list of PlotPanel"""
self.addPlotPanel("Ensemble plot",
"gui/plots/simple_plot.html",
short_name="EP")
self.addPlotPanel("Ensemble overview plot",
"gui/plots/simple_overview_plot.html",
short_name="EOP")
self.addPlotPanel("Ensemble statistics",
"gui/plots/ensemble_statistics_plot.html",
short_name="ES")
self.addPlotPanel("Histogram",
"gui/plots/histogram.html",
short_name="Histogram")
self.addPlotPanel("Distribution",
"gui/plots/gen_kw.html",
short_name="Distribution")
self.addPlotPanel("RFT plot", "gui/plots/rft.html", short_name="RFT")
self.addPlotPanel("RFT overview plot",
"gui/plots/rft_overview.html",
short_name="oRFT")
self.addPlotPanel("Ensemble plot",
"gui/plots/gen_data.html",
short_name="epGenData")
self.addPlotPanel("Ensemble overview plot",
"gui/plots/gen_data_overview.html",
short_name="eopGenData")
self.addPlotPanel("Ensemble statistics",
"gui/plots/gen_data_statistics_plot.html",
short_name="esGenData")
self.addPlotPanel("PCA plot", "gui/plots/pca.html", short_name="PCA")
self.__data_type_keys_widget = DataTypeKeysWidget()
self.__data_type_keys_widget.dataTypeKeySelected.connect(
self.keySelected)
self.addDock("Data types", self.__data_type_keys_widget)
current_case = CaseSelectorModel().getCurrentChoice()
self.__case_selection_widget = CaseSelectionWidget(current_case)
self.__case_selection_widget.caseSelectionChanged.connect(
self.caseSelectionChanged)
plot_case_dock = self.addDock("Plot case",
self.__case_selection_widget)
self.__customize_plot_widget = CustomizePlotWidget()
self.__customize_plot_widget.customPlotSettingsChanged.connect(
self.plotSettingsChanged)
customize_plot_dock = self.addDock("Customize",
self.__customize_plot_widget)
self.__toolbar.exportClicked.connect(self.exportActivePlot)
self.__toolbar.plotScalesChanged.connect(self.plotSettingsChanged)
self.__toolbar.reportStepChanged.connect(self.plotSettingsChanged)
self.__exporter = None
self.tabifyDockWidget(plot_case_dock, customize_plot_dock)
plot_case_dock.show()
plot_case_dock.raise_()
self.__plot_cases = self.__case_selection_widget.getPlotCaseNames()
def __init__(self, ert, parent):
QMainWindow.__init__(self, parent)
self.__ert = ert
""":type: ert.enkf.enkf_main.EnKFMain"""
self.setMinimumWidth(750)
self.setMinimumHeight(500)
self.setWindowTitle("Plotting")
self.activateWindow()
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)
key_manager = ert.getKeyManager()
""":type: ert.enkf.key_manager.KeyManager """
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)
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.__data_types_key_model = DataTypeKeysListModel(ert)
self.__data_type_keys_widget = DataTypeKeysWidget(
self.__data_types_key_model)
self.__data_type_keys_widget.dataTypeKeySelected.connect(
self.keySelected)
self.addDock("Data types", self.__data_type_keys_widget)
current_case = CaseSelectorModel().getCurrentChoice()
self.__case_selection_widget = CaseSelectionWidget(current_case)
self.__case_selection_widget.caseSelectionChanged.connect(
self.keySelected)
plot_case_dock = self.addDock("Plot case",
self.__case_selection_widget)
self.__customize_plot_widget = CustomizePlotWidget()
self.__customize_plot_widget.customPlotSettingsChanged.connect(
self.keySelected)
customize_plot_dock = self.addDock("Customize",
self.__customize_plot_widget)
self.tabifyDockWidget(plot_case_dock, customize_plot_dock)
plot_case_dock.show()
plot_case_dock.raise_()
self.__plot_widgets[self.__central_tab.currentIndex()].setActive()
self.__data_type_keys_widget.selectDefault()