def getObsKeys(self, data_key):
ensemble_data_fetcher = EnsembleDataFetcher(self.ert())
block_observation_data_fetcher = BlockObservationDataFetcher(self.ert())
gen_data_observation_data_fetcher = ObservationGenDataFetcher(self.ert())
if ensemble_data_fetcher.supportsKey(data_key):
return self.ert().ensembleConfig().getNode(data_key).getObservationKeys()
elif block_observation_data_fetcher.supportsKey(data_key):
return [data_key]
elif gen_data_observation_data_fetcher.supportsKey(data_key):
return gen_data_observation_data_fetcher.getAllObsKeysForKey(data_key)
def getPlotDataForKeyAndCases(self, key, cases):
observation_data_fetcher = ObservationDataFetcher(self.ert())
block_observation_data_fetcher = BlockObservationDataFetcher(
self.ert())
gen_kw_fetcher = EnsembleGenKWFetcher(self.ert())
gen_data_fetcher = EnsembleGenDataFetcher(self.ert())
if self.isBlockObservationKey(key):
return self.fetchBlockObservationData(
block_observation_data_fetcher, key, cases)
elif self.isSummaryKey(key):
return self.fetchSummaryData(observation_data_fetcher, key, cases)
elif self.isGenKWKey(key):
return self.fetchGenKWData(gen_kw_fetcher, key, cases)
elif self.isGenDataKey(key):
return self.fetchGenData(gen_data_fetcher, key, cases)
elif self.isPcaDataKey(key):
plot_data = PlotData(key)
pca_plot_data = self.fetchPcaData(key, cases)
plot_data.setUserData("PCA", pca_plot_data)
return plot_data
else:
raise NotImplementedError("Key %s not supported." % key)
def getAllBlockObservationKeys(self):
""" @rtype: list of str """
if self.__block_observation_keys is None:
keys = BlockObservationDataFetcher(self.ert()).getSupportedKeys()
self.__block_observation_keys = sorted(keys, key=lambda k : k.lower())
return self.__block_observation_keys
def fetchSupportedKeys(self):
summary_keys = EnsembleDataFetcher(self.ert()).getSupportedKeys()
keys = []
for key in summary_keys:
obs_keys = self.ert().ensembleConfig().getNode(key).getObservationKeys()
if len(obs_keys) > 0:
keys.append(key)
keys += BlockObservationDataFetcher(self.ert()).getSupportedKeys()
keys += ObservationGenDataFetcher(self.ert()).getSupportedKeys()
return keys
def isBlockObservationKey(self, key):
block_observation_data_fetcher = BlockObservationDataFetcher(self.ert())
return block_observation_data_fetcher.supportsKey(key)
def isBlockObservationKey(self, key):
block_observation_data_fetcher = BlockObservationDataFetcher(
self.ert())
return block_observation_data_fetcher.supportsKey(key)
