add __call__ method to with_units

Cantera · Dec 11, 2023 · 605844a · 605844a
1 parent 8563be1
commit 605844a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/site_scons/site_tools/UnitsInterfaceBuilder.py b/site_scons/site_tools/UnitsInterfaceBuilder.py
@@ -188,6 +188,12 @@ def __setattr__(self, name, value):
             else:
                 setattr(self._phase, name, value)
 
+        def __call__(self, *species):
+            out = SolutionArray(self._phase[species], init=False)
+            self._share(out, range(self.size))
+            out.shape = self.shape
+            return out
+
         @property
         def basis_units(self):
             \"\"\"The units associated with the mass/molar basis of this phase.\"\"\"