def __init__(self, parent):
super(GenKWKeys, self).__init__("gen_kw", parent)
self.addShellFunction(name="list", function=GenKWKeys.list, help_message="Shows a list of all available GenKW keys.")
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "GenKW")
ShellPlot.addHistogramPlotSupport(self, "GenKW")
ShellPlot.addGaussianKDEPlotSupport(self, "GenKW")
ShellPlot.addDistributionPlotSupport(self, "GenKW")
ShellPlot.addCrossCaseStatisticsPlotSupport(self, "GenKW")
def __init__(self, parent):
super(CustomKWKeys, self).__init__("custom_kw", parent)
self.addShellFunction(**{"name": "list",
"function": CustomKWKeys.list,
"help_message": "List all CustomKW keys."})
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "CustomKW")
ShellPlot.addHistogramPlotSupport(self, "CustomKW")
ShellPlot.addGaussianKDEPlotSupport(self, "CustomKW")
ShellPlot.addDistributionPlotSupport(self, "CustomKW")
ShellPlot.addCrossCaseStatisticsPlotSupport(self, "CustomKW")
def __init__(self, parent):
super(GenKWKeys, self).__init__("gen_kw", parent)
self.addShellFunction(
name="list",
function=GenKWKeys.list,
help_message="Shows a list of all available GenKW keys.")
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "GenKW")
ShellPlot.addHistogramPlotSupport(self, "GenKW")
ShellPlot.addGaussianKDEPlotSupport(self, "GenKW")
ShellPlot.addDistributionPlotSupport(self, "GenKW")
ShellPlot.addCrossCaseStatisticsPlotSupport(self, "GenKW")
def do_histogram(self, line):
keys = matchItems(line, self.fetchSupportedKeys())
if len(keys) == 0:
print("Error: Must have at least one GenKW key")
return False
case_list = self.shellContext()["plot_settings"].getCurrentPlotCases()
for key in keys:
for case_name in case_list:
data = GenKwCollector.loadAllGenKwData(self.ert(), case_name, [key])
plot = ShellPlot(key)
plot.histogram(data, key, log_on_x=key.startswith("LOG10_"))
def do_plot(self, line):
keys = matchItems(line, self.summaryKeys())
if len(keys) == 0:
self.lastCommandFailed("Must have at least one Summary key")
return False
case_list = self.shellContext()["plot_settings"].getCurrentPlotCases()
for key in keys:
plot = ShellPlot(key)
for case_name in case_list:
data = SummaryCollector.loadAllSummaryData(
self.ert(), case_name, [key])
if not data.empty:
plot.plot(data, value_column=key, legend_label=case_name)
if len(
case_list
) > 0 and SummaryObservationCollector.summaryKeyHasObservations(
self.ert(), key):
observation_data = SummaryObservationCollector.loadObservationData(
self.ert(), case_list[0], [key])
if not observation_data.empty:
plot.plotObservations(observation_data, value_column=key)
plot.showLegend()
def do_plot(self, line):
keys = matchItems(line, self.fetchSupportedKeys())
if len(keys) == 0:
self.lastCommandFailed("Must have at least one GenData key")
return False
case_list = self.shellContext()["plot_settings"].getCurrentPlotCases()
for key in keys:
key, report_step = key.split("@", 1)
report_step = int(report_step)
plot = ShellPlot("%s at report step: %d" %(key, report_step))
for case_name in case_list:
data = GenDataCollector.loadGenData(self.ert(), case_name, key, report_step)
if not data.empty:
plot.plotGenData(data, legend_label=case_name)
obs_key = GenDataObservationCollector.getObservationKeyForDataKey(self.ert(), key, report_step)
if obs_key is not None:
obs_data = GenDataObservationCollector.loadGenDataObservations(self.ert(), case_name, [obs_key])
if not obs_data.empty:
plot.plotObservations(obs_data, obs_key)
plot.showLegend()
def do_histogram(self, line):
keys = matchItems(line, self.fetchSupportedKeys())
if len(keys) == 0:
self.lastCommandFailed("Must have at least one GenKW key")
return False
case_list = self.shellContext()["plot_settings"].getCurrentPlotCases()
for key in keys:
for case_name in case_list:
data = GenKwCollector.loadAllGenKwData(self.ert(), case_name, [key])
if not data.empty:
plot = ShellPlot(key)
plot.histogram(data, key, log_on_x=key.startswith("LOG10_"))
def do_density(self, line):
keys = matchItems(line, self.fetchSupportedKeys())
if len(keys) == 0:
print("Error: Must have at least one GenKW key")
return False
case_list = self.shellContext()["plot_settings"].getCurrentPlotCases()
for key in keys:
plot = ShellPlot(key)
for case_name in case_list:
data = GenKwCollector.loadAllGenKwData(self.ert(), case_name, [key])
plot.density(data, key, legend_label=case_name)
plot.showLegend()
def __init__(self, parent):
super(CustomKWKeys, self).__init__("custom_kw", parent)
self.addShellFunction(
**{
"name": "list",
"function": CustomKWKeys.list,
"help_message": "List all CustomKW keys."
})
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "CustomKW")
ShellPlot.addHistogramPlotSupport(self, "CustomKW")
ShellPlot.addGaussianKDEPlotSupport(self, "CustomKW")
ShellPlot.addDistributionPlotSupport(self, "CustomKW")
ShellPlot.addCrossCaseStatisticsPlotSupport(self, "CustomKW")
def __init__(self, parent):
super(SummaryKeys, self).__init__("summary", parent)
self.addShellFunction(
name="list",
function=SummaryKeys.list,
help_message=
"Shows a list of all available Summary keys. (* = with observations)"
)
self.addShellFunction(
name="observations",
function=SummaryKeys.observations,
help_message=
"Shows a list of all available Summary key observations.")
self.addShellFunction(
name="matchers",
function=SummaryKeys.matchers,
help_message=
"Shows a list of all Summary keys that the ensemble will match "
"against during simulations and manual load.")
self.addShellFunction(
name="add_matcher",
function=SummaryKeys.add_matcher,
help_arguments="<summary_key>",
help_message="Add a matcher to the Summary key matcher set.")
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "Summary")
ShellPlot.addEnsemblePlotSupport(self, "Summary")
ShellPlot.addQuantilesPlotSupport(self, "Summary")
def __init__(self, parent):
super(SummaryKeys, self).__init__("summary", parent)
self.addShellFunction(name="list",
function=SummaryKeys.list,
help_message="Shows a list of all available Summary keys. (* = with observations)")
self.addShellFunction(name="observations",
function=SummaryKeys.observations,
help_message= "Shows a list of all available Summary key observations.")
self.addShellFunction(name="matchers",
function=SummaryKeys.matchers,
help_message="Shows a list of all Summary keys that the ensemble will match "
"against during simulations and manual load.")
self.addShellFunction(name="add_matcher",
function=SummaryKeys.add_matcher,
help_arguments="<summary_key>",
help_message="Add a matcher to the Summary key matcher set.")
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "Summary")
ShellPlot.addEnsemblePlotSupport(self, "Summary")
ShellPlot.addQuantilesPlotSupport(self, "Summary")
def __init__(self, parent):
super(GenDataKeys, self).__init__("gen_data", parent)
self.addShellFunction(name="list", function=GenDataKeys.list, help_message="Shows a list of all available GenData keys.")
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "GenData")
ShellPlot.addEnsemblePlotSupport(self, "GenData")
ShellPlot.addQuantilesPlotSupport(self, "GenData")
def __init__(self, parent):
super(GenDataKeys, self).__init__("gen_data", parent)
self.addShellFunction(
name="list",
function=GenDataKeys.list,
help_message="Shows a list of all available GenData keys.")
self.__plot_data_gatherer = None
ShellPlot.addPrintSupport(self, "GenData")
ShellPlot.addEnsemblePlotSupport(self, "GenData")
ShellPlot.addQuantilesPlotSupport(self, "GenData")
def do_plot_area(self, line):
keys = matchItems(line, self.summaryKeys())
if len(keys) == 0:
print("Error: Must have at least one Summary key")
return False
case_list = self.shellContext()["plot_settings"].getCurrentPlotCases()
for key in keys:
plot = ShellPlot(key)
for case_name in case_list:
data = SummaryCollector.loadAllSummaryData(self.ert(), case_name, [key])
plot.plotArea(data, value_column=key, legend_label=case_name)
if len(case_list) > 0 and SummaryObservationCollector.summaryKeyHasObservations(self.ert(), key):
observation_data = SummaryObservationCollector.loadObservationData(self.ert(), case_list[0], [key])
plot.plotObservations(observation_data, value_column=key)
plot.showLegend()
def do_density(self, line):
keys = matchItems(line, self.fetchSupportedKeys())
if len(keys) == 0:
self.lastCommandFailed("Must have at least one GenKW key")
return False
case_list = self.shellContext()["plot_settings"].getCurrentPlotCases()
for key in keys:
plot = ShellPlot(key)
for case_name in case_list:
data = GenKwCollector.loadAllGenKwData(self.ert(), case_name, [key])
if not data.empty:
plot.density(data, key, legend_label=case_name)
plot.showLegend()
