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 fetchGenData(self, gen_data_fetcher, key, cases):
plot_data = PlotData(key)
gen_data_observation_fetcher = ObservationGenDataFetcher(self.ert())
if gen_data_observation_fetcher.hasData(key):
self.addObservationData(plot_data, key, gen_data_observation_fetcher)
self.addEnsembleData(plot_data, key, cases, gen_data_fetcher)
self.addPcaData(plot_data, key, cases)
return plot_data
def fetchGenData(self, gen_data_fetcher, key, cases):
plot_data = PlotData(key)
gen_data_observation_fetcher = ObservationGenDataFetcher(self.ert())
if gen_data_observation_fetcher.hasData(key):
self.addObservationData(plot_data, key,
gen_data_observation_fetcher)
self.addEnsembleData(plot_data, key, cases, gen_data_fetcher)
self.addPcaData(plot_data, key, cases)
return plot_data
def getAllGenDataKeys(self):
""" @rtype: list of str """
if self.__gen_data_keys is None:
keys = ObservationGenDataFetcher(self.ert()).getSupportedKeys()
self.__gen_data_keys = sorted(keys, key=lambda k : k.lower())
return self.__gen_data_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 isGenDataKey(self, key):
obs_gen_data_fetcher = ObservationGenDataFetcher(self.ert())
return obs_gen_data_fetcher.supportsKey(key)
def isGenDataKey(self, key):
obs_gen_data_fetcher = ObservationGenDataFetcher(self.ert())
return obs_gen_data_fetcher.supportsKey(key)