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)