Plugins for parameter response multiple regression and parallel coordinates

.github/workflows/webviz-subsurface.yml

@@ -77,10 +77,10 @@ jobs:
       env:
         # If you want the CI to (temporarily) run against your fork of the testdada,
         # change the value her from "equinor" to your username.
-        TESTDATA_REPO_OWNER: equinor
+        TESTDATA_REPO_OWNER: asnyv
         # If you want the CI to (temporarily) run against another branch than master,
         # change the value her from "master" to the relevant branch name.
-        TESTDATA_REPO_BRANCH: master
+        TESTDATA_REPO_BRANCH: group_b_testdata
       run: |
         webviz certificate
         git clone --depth 1 --branch $TESTDATA_REPO_BRANCH https://github.com/$TESTDATA_REPO_OWNER/webviz-subsurface-testdata.git

setup.py

@@ -45,6 +45,10 @@
             "ReservoirSimulationTimeSeriesRegional = "
             + "webviz_subsurface.plugins:ReservoirSimulationTimeSeriesRegional",
             "RftPlotter =  webviz_subsurface.plugins:RftPlotter",
+            "ParameterResponseMultipleRegression = "
+            + "webviz_subsurface.plugins:ParameterResponseMultipleRegression",
+            "ParameterResponseParallelCoordinates = "
+            + "webviz_subsurface.plugins:ParameterResponseParallelCoordinates",
         ]
     },
     install_requires=[
@@ -57,6 +61,7 @@
         "webviz-config>=0.0.55",
         "webviz-subsurface-components>=0.0.23",
         "xtgeo>=2.8",
+        "statsmodels>=0.11.1",
     ],
     tests_require=TESTS_REQUIRE,
     extras_require={"tests": TESTS_REQUIRE},

webviz_subsurface/_utils/response_aggregation.py

@@ -0,0 +1,49 @@
+from webviz_config.common_cache import CACHE
+import numpy as np
+
+
+@CACHE.memoize(timeout=CACHE.TIMEOUT)
+def filter_and_sum_responses(
+    dframe, ensemble, response, filteroptions=None, aggregation="sum"
+):
+    """Cached wrapper for _filter_and_sum_responses"""
+    return _filter_and_sum_responses(
+        dframe=dframe,
+        ensemble=ensemble,
+        response=response,
+        filteroptions=filteroptions,
+        aggregation=aggregation,
+    )
+
+
+def _filter_and_sum_responses(
+    dframe, ensemble, response, filteroptions=None, aggregation="sum",
+):
+    """Filter response dataframe for the given ensemble
+    and optional filter columns. Returns dataframe grouped and
+    aggregated per realization."""
+    df = dframe.copy()
+    df = df.loc[df["ENSEMBLE"] == ensemble]
+    if filteroptions:
+        for opt in filteroptions:
+            if opt["type"] == "multi" or opt["type"] == "single":
+                if isinstance(opt["values"], list):
+                    df = df.loc[df[opt["name"]].isin(opt["values"])]
+                else:
+                    if opt["name"] == "DATE" and isinstance(opt["values"], str):
+                        df = df.loc[df["DATE"].astype(str) == opt["values"]]
+                    else:
+                        df = df.loc[df[opt["name"]] == opt["values"]]
+
+            elif opt["type"] == "range":
+                df = df.loc[
+                    (df[opt["name"]] >= np.min(opt["values"]))
+                    & (df[opt["name"]] <= np.max(opt["values"]))
+                ]
+    if aggregation == "sum":
+        return df.groupby("REAL").sum().reset_index()[["REAL", response]]
+    if aggregation == "mean":
+        return df.groupby("REAL").mean().reset_index()[["REAL", response]]
+    raise ValueError(
+        f"Aggregation of response file specified as '{aggregation}'' is invalid. "
+    )

webviz_subsurface/plugins/__init__.py

@@ -49,7 +49,10 @@
     ReservoirSimulationTimeSeriesRegional,
 )
 from ._rft_plotter.rft_plotter import RftPlotter
-
+from ._parameter_response_multiple_regression import ParameterResponseMultipleRegression
+from ._parameter_response_parallel_coordinates import (
+    ParameterResponseParallelCoordinates,
+)
 
 __all__ = [
     "ParameterDistribution",
@@ -72,4 +75,6 @@
     "RelativePermeability",
     "ReservoirSimulationTimeSeriesRegional",
     "RftPlotter",
+    "ParameterResponseMultipleRegression",
+    "ParameterResponseParallelCoordinates",
 ]