Update: uncertainty of totals

Author: heptaflar

kaleidoscope/config/config.random.json

@@ -143,6 +143,66 @@
         10.0
       ]
     },
+    "atot_412": {
+      "total": [
+        "aph_412",
+        "adg_412"
+      ],
+      "clip": [
+        0.0,
+        10.0
+      ]
+    },
+    "atot_443": {
+      "total": [
+        "aph_443",
+        "adg_443"
+      ],
+      "clip": [
+        0.0,
+        10.0
+      ]
+    },
+    "atot_490": {
+      "total": [
+        "aph_490",
+        "adg_490"
+      ],
+      "clip": [
+        0.0,
+        10.0
+      ]
+    },
+    "atot_510": {
+      "total": [
+        "aph_510",
+        "adg_510"
+      ],
+      "clip": [
+        0.0,
+        10.0
+      ]
+    },
+    "atot_560": {
+      "total": [
+        "aph_560",
+        "adg_560"
+      ],
+      "clip": [
+        0.0,
+        10.0
+      ]
+    },
+    "atot_665": {
+      "total": [
+        "aph_665",
+        "adg_665"
+      ],
+      "clip": [
+        0.0,
+        10.0
+      ]
+    },
     "kd_490": {
       "bias": "kd_490_bias",
       "rmsd": "kd_490_rmsd",

kaleidoscope/operators/randomizeop.py

@@ -91,7 +91,7 @@ def run(self, source: Dataset) -> Dataset:  # noqa: D102
             if v not in config or self._args.selector == 0:
                 continue
             get_logger().info(f"starting graph for variable: {v}")
-            self.randomize(target, v, x, config[v])
+            self.randomize(source, target, v, x, config[v])
             get_logger().info(f"finished graph for variable: {v}")
         target.attrs["monte_carlo_software"] = __name__
         target.attrs["monte_carlo_software_version"] = __version__
@@ -110,6 +110,7 @@ def config(self) -> dict[str : dict[str:Any]]:
 
     def randomize(
         self,
+        source: Dataset,
         target: Dataset,
         v: str,
         x: DataArray,
@@ -123,7 +124,18 @@ def randomize(
         :param x: The data of the variable.
         :param config: The randomization configuration.
         """
-        if "uncertainty" in config:
+        if "total" in config:
+            s = None
+            z = _decode(x.data, x.attrs)
+            for ref in config["total"]:
+                a = _decode(target[ref].data, target[ref].attrs)
+                b = _decode(source[ref].data, source[ref].attrs)
+                z = z + (a - b)
+            if "clip" in config:
+                z = da.clip(
+                    z, config["clip"][0], config["clip"][1]
+                )
+        elif "uncertainty" in config:
             s = self.seed(self.uuid(v))
             f = Randomize(m=x.ndim, dist=config["distribution"], seed=s)
             u = (
@@ -170,7 +182,8 @@ def randomize(
                 ],
                 dtype=z.dtype,
             )
-        target[v].attrs["seed"] = s
+        if s is not None:
+            target[v].attrs["seed"] = s
         if get_logger().is_enabled(Logging.DEBUG):
             get_logger().debug(f"seed: {s}")
             get_logger().debug(f"min:  {da.nanmin(z).compute() :.3f}")