Added modelbound variable extraction helper fun on context

Author: SMijin

RMK_support/remkit_context.py

@@ -848,3 +848,27 @@ def addTermDiagnostics(self, *args: Variable):
                         model + "_" + term, [pair], self.variables[model + "_" + term]
                     )
                 )
+
+    def extractMBVar(
+        self, model: mc.Model, mbVar: Variable, resultVarName: str, priority=4
+    ):
+        """Extract a modelbound variable from a model and add it to the context
+
+        Args:
+            model (mc.Model): Model to extract from
+            mbVar (Variable): Variable to extract
+            resultVarName (str): Variable name to extract to
+            priority (int, optional): Manipulator priority. Defaults to 4.
+        """
+        kwargs = mbVar.kwargs
+        if "derivation" in kwargs:
+            del kwargs["derivation"]
+        if "derivationArgs" in kwargs:
+            del kwargs["derivationArgs"]
+        kwargs["isDerived"] = True
+        newVar = Variable(resultVarName, mbVar.grid, **kwargs)
+
+        self.manipulators.add(
+            MBDataExtractor(resultVarName, model, mbVar, newVar, priority)
+        )
+        self.variables.add(newVar)

RMK_support/tests/test_remkit_context.py

@@ -397,7 +397,9 @@ def test_wrapper(grid: Grid):
 
     # Modelbound data extractor
 
-    mbExtract = MBDataExtractor("mbExtract", model, dModelbound, priority=1).dict()
+    rk.extractMBVar(model, dModelbound, dModelbound.name, priority=1)
+
+    mbExtract = rk.manipulators[dModelbound.name].dict()
 
     assert mbExtract == {
         "type": "modelboundDataExtractor",