-
Notifications
You must be signed in to change notification settings - Fork 59
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Use new simplified hillshading for surface plugins (#541)
- Loading branch information
Hans Kallekleiv
authored
Jan 26, 2021
1 parent
651e89a
commit a205358
Showing
14 changed files
with
181 additions
and
446 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,181 +1,13 @@ | ||
| from typing import Optional, Union, List, Dict, Any | ||
| import io | ||
| import base64 | ||
|
|
||
| import numpy as np | ||
| from xtgeo import RegularSurface | ||
| from webviz_config.common_cache import CACHE | ||
| from PIL import Image | ||
| from .image_processing import array_to_png | ||
|
|
||
|
|
||
| @CACHE.memoize(timeout=CACHE.TIMEOUT) | ||
| def load_surface(surface_path: str) -> RegularSurface: | ||
| return RegularSurface(surface_path) | ||
|
|
||
|
|
||
| def get_surface_arr( | ||
| surface: RegularSurface, | ||
| unrotate: bool = True, | ||
| flip: bool = True, | ||
| clip_min: Union[float, np.ndarray, None] = None, | ||
| clip_max: Union[float, np.ndarray, None] = None, | ||
| ) -> List[np.ndarray]: | ||
| if clip_min or clip_max: | ||
| np.ma.clip(surface.values, clip_min, clip_max, out=surface.values) | ||
| if unrotate: | ||
| surface.unrotate() | ||
|
|
||
| x, y, z = surface.get_xyz_values() | ||
| if flip: | ||
| x = np.flip(x.transpose(), axis=0) | ||
| y = np.flip(y.transpose(), axis=0) | ||
| z = np.flip(z.transpose(), axis=0) | ||
| z.filled(np.nan) | ||
| return [x, y, z] | ||
|
|
||
|
|
||
| @CACHE.memoize(timeout=CACHE.TIMEOUT) | ||
| def get_surface_fence(fence: np.ndarray, surface: RegularSurface) -> np.ndarray: | ||
| return surface.get_fence(fence) | ||
|
|
||
|
|
||
| # pylint: disable=too-many-arguments | ||
| def make_surface_layer( | ||
| surface: RegularSurface, | ||
| name: str = "surface", | ||
| updatemode: str = "update", | ||
| min_val: Optional[float] = None, | ||
| max_val: Optional[float] = None, | ||
| color: Optional[List[str]] = None, | ||
| shader_type: Optional[str] = "soft-hillshading", | ||
| shadows: bool = False, | ||
| unit: str = " ", | ||
| ) -> Dict[str, Any]: | ||
| """Make NewLayeredMap surface image base layer | ||
| Args: | ||
| surface: an xtgeo surface object | ||
| name: name of the surface object | ||
| min_val: minimum value to be plotted in map | ||
| max_val: maximum value to be plotted in map | ||
| color: an array with colors as strings | ||
| shader_type: determines shader in map | ||
| unit: determines unit on the map axes | ||
| Returns: | ||
| A surface layer that can be plotted in NewLayeredMap | ||
| """ | ||
|
|
||
| zvalues = get_surface_arr(surface, clip_min=min_val, clip_max=max_val)[2] | ||
| min_val = min_val if min_val is not None else np.nanmin(zvalues) | ||
| max_val = max_val if max_val is not None else np.nanmax(zvalues) | ||
| if shader_type == "hillshading_shadows": | ||
| shader_type = "hillshading" | ||
| shadows = True | ||
| img = Image.open(io.BytesIO(base64.b64decode(array_to_png(zvalues.copy())[22:]))) | ||
| width, height = img.size | ||
| if width * height >= 300 * 300: | ||
| scale = 1.0 | ||
| else: | ||
| ratio = (1000 ** 2) / (width * height) | ||
| scale = np.sqrt(ratio).round(2) | ||
| color = ( | ||
| [ | ||
| "#fde725", | ||
| "#b5de2b", | ||
| "#6ece58", | ||
| "#35b779", | ||
| "#1f9e89", | ||
| "#26828e", | ||
| "#31688e", | ||
| "#3e4989", | ||
| "#482878", | ||
| "#440154", | ||
| ] | ||
| if color is None | ||
| else color | ||
| ) | ||
| return { | ||
| "name": name, | ||
| "checked": True, | ||
| "id": name, | ||
| "action": updatemode, | ||
| "baseLayer": True, | ||
| "data": [ | ||
| { | ||
| "type": "image", | ||
| "url": array_to_png(zvalues.copy()), | ||
| "colorScale": { | ||
| "colors": color, | ||
| "prefixZeroAlpha": False, | ||
| "scaleType": "linear", | ||
| }, | ||
| "shader": { | ||
| "type": shader_type, | ||
| "shadows": shadows, | ||
| "shadowIterations": 128, | ||
| "elevationScale": -1.0, | ||
| "pixelScale": 11000, | ||
| "setBlackToAlpha": True, | ||
| }, | ||
| "bounds": [[surface.xmin, surface.ymin], [surface.xmax, surface.ymax]], | ||
| "minvalue": round(np.nanmin(zvalues), 4), | ||
| "maxvalue": round(np.nanmax(zvalues), 4), | ||
| "unit": str(unit), | ||
| "imageScale": scale, | ||
| } | ||
| ], | ||
| } | ||
|
|
||
|
|
||
| def get_surface_layers( | ||
| switch: Dict[str, bool], | ||
| surface_name: str, | ||
| surfaces: List[RegularSurface], | ||
| min_val: Optional[float] = None, | ||
| max_val: Optional[float] = None, | ||
| ) -> List[Dict[str, Any]]: | ||
| """Creates layers in map view from all surfaces | ||
| Args: | ||
| switch: Toggle hillshading on/off | ||
| surface_name: Name of surface | ||
| surfaces: List containing a single surface with corresponding depth error, depth trend etc. | ||
| min_val: Minimum z-value of surface | ||
| max_val: Maximum z-value of surface | ||
| Returns: | ||
| layers: List of all surface layers | ||
| """ | ||
| shader_type = "hillshading" if switch["value"] is True else None | ||
| depth_list = [ | ||
| "Depth", | ||
| "Depth uncertainty", | ||
| "Depth residual", | ||
| "Depth residual uncertainty", | ||
| "Depth trend", | ||
| "Depth trend uncertainty", | ||
| ] | ||
| layers = [] | ||
| for i, surface in enumerate(surfaces): | ||
| if surface is not None: | ||
| s_layer = make_surface_layer( | ||
| surface, | ||
| name=depth_list[i], | ||
| min_val=min_val, | ||
| max_val=max_val, | ||
| color=[ | ||
| "#440154", | ||
| "#482878", | ||
| "#3e4989", | ||
| "#31688e", | ||
| "#26828e", | ||
| "#1f9e89", | ||
| "#35b779", | ||
| "#6ece58", | ||
| "#b5de2b", | ||
| "#fde725", | ||
| ], | ||
| shader_type=shader_type, | ||
| ) | ||
| s_layer["id"] = surface_name + " " + depth_list[i] + "-id" | ||
| s_layer["action"] = "add" | ||
| layers.append(s_layer) | ||
| return layers |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,2 +1,3 @@ | ||
| from .ensemble_model import EnsembleModel | ||
| from .ensemble_set_model import EnsembleSetModel | ||
| from .surface_leaflet_model import SurfaceLeafletModel |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.