def __init__(
self,
app: Dash,
webviz_settings: WebvizSettings,
ensembles: list,
attributes: list = None,
attribute_settings: dict = None,
wellfolder: Path = None,
wellsuffix: str = ".w",
map_height: int = 600,
):
super().__init__()
self.ens_paths = {
ens: webviz_settings.shared_settings["scratch_ensembles"][ens]
for ens in ensembles
}
# Find surfaces
self._surface_table = find_surfaces(self.ens_paths)
# Extract realizations and sensitivity information
self.ens_df = get_realizations(
ensemble_paths=self.ens_paths, ensemble_set_name="EnsembleSet"
)
# Drop any ensembles that does not have surfaces
self.ens_df = self.ens_df.loc[
self.ens_df["ENSEMBLE"].isin(self._surface_table["ENSEMBLE"].unique())
]
if attributes is not None:
self._surface_table = self._surface_table[
self._surface_table["attribute"].isin(attributes)
]
if self._surface_table.empty:
raise ValueError("No surfaces found with the given attributes")
self._surface_ensemble_set_model = {
ens: SurfaceSetModel(surf_ens_df)
for ens, surf_ens_df in self._surface_table.groupby("ENSEMBLE")
}
self.attribute_settings: dict = attribute_settings if attribute_settings else {}
self.map_height = map_height
self.surfaceconfig = surfacedf_to_dict(self._surface_table)
self.wellfolder = wellfolder
self.wellsuffix = wellsuffix
self.wellfiles: Union[List[str], None] = (
json.load(find_files(wellfolder, wellsuffix))
if wellfolder is not None
else None
)
self.well_layer = (
make_well_layers([get_path(wellfile) for wellfile in self.wellfiles])
if self.wellfiles
else None
)
self.selector = SurfaceSelector(app, self.surfaceconfig, ensembles)
self.selector2 = SurfaceSelector(app, self.surfaceconfig, ensembles)
self.set_callbacks(app)
def __init__(
self,
app,
ensembles: list,
attributes: list = None,
attribute_settings: dict = None,
wellfolder: Path = None,
wellsuffix: str = ".w",
):
super().__init__()
self.ens_paths = {
ens: app.webviz_settings["shared_settings"]["scratch_ensembles"][ens]
for ens in ensembles
}
# Find surfaces
self.surfacedf = find_surfaces(self.ens_paths)
if attributes is not None:
self.surfacedf = self.surfacedf[
self.surfacedf["attribute"].isin(attributes)
]
if self.surfacedf.empty:
raise ValueError("No surfaces found with the given attributes")
self.attribute_settings = attribute_settings if attribute_settings else {}
self.surfaceconfig = surfacedf_to_dict(self.surfacedf)
self.wellfolder = wellfolder
self.wellsuffix = wellsuffix
self.wellfiles = (
json.load(find_files(wellfolder, wellsuffix))
if wellfolder is not None
else None
)
self.well_layer = (
make_well_layers([get_path(wellfile) for wellfile in self.wellfiles])
if self.wellfiles
else None
)
# Extract realizations and sensitivity information
self.ens_df = get_realizations(
ensemble_paths=self.ens_paths, ensemble_set_name="EnsembleSet"
)
self.selector = SurfaceSelector(app, self.surfaceconfig, ensembles)
self.selector2 = SurfaceSelector(app, self.surfaceconfig, ensembles)
self.set_callbacks(app)