Added functions to python interface

g-bauer · g-bauer · commit c05d72067307 · 2022-03-18T16:44:00.000+01:00
diff --git a/src/python/state.rs b/src/python/state.rs
@@ -565,6 +565,29 @@ macro_rules! impl_state {
                 self.0.ln_phi().view().to_pyarray(py)
             }
 
+            /// Return logarithmic pure substance fugacity coefficient.
+            /// 
+            /// For each component, the hypothetical liquid fugacity coefficient
+            /// at mixture temperature and pressure is computed.
+            /// 
+            /// Returns
+            /// -------
+            /// numpy.ndarray
+            #[pyo3(text_signature = "($self)")]
+            fn ln_phi_pure<'py>(&self, py: Python<'py>) -> PyResult<&'py PyArray1<f64>> {
+                Ok(self.0.ln_phi_pure()?.view().to_pyarray(py))
+            }
+
+            /// Return logarithmic symmetric activity coefficient.
+            /// 
+            /// Returns
+            /// -------
+            /// numpy.ndarray
+            #[pyo3(text_signature = "($self)")]
+            fn ln_symmetric_activity_coefficient<'py>(&self, py: Python<'py>) -> PyResult<&'py PyArray1<f64>> {
+                Ok(self.0.ln_symmetric_activity_coefficient()?.view().to_pyarray(py))
+            }
+
             /// Return derivative of logarithmic fugacity coefficient w.r.t. temperature.
             ///
             /// Returns
diff --git a/src/state/properties.rs b/src/state/properties.rs
@@ -313,11 +313,19 @@ impl<U: EosUnit, E: EquationOfState> State<U, E> {
                     &(arr1(&[1.0]) * U::reference_moles()),
                     crate::DensityInitialization::Liquid,
                 )?;
-                state.ln_phi()[0]
+                Ok(state.ln_phi()[0])
             })
             .collect()
     }
 
+    /// Activity coefficient $\ln \gamma_i = \ln \varphi_i(T, p, \mathbf{N}) - \ln \varphi_i(T, p)$
+    pub fn ln_symmetric_activity_coefficient(&self) -> EosResult<Array1<f64>> {
+        match self.eos.components() {
+            1 => Ok(arr1(&[0.0])),
+            _ => Ok(self.ln_phi() - &self.ln_phi_pure()?),
+        }
+    }
+
     /// Partial derivative of the logarithm of the fugacity coefficient w.r.t. temperature: $\left(\frac{\partial\ln\varphi_i}{\partial T}\right)_{p,N_i}$
     pub fn dln_phi_dt(&self) -> QuantityArray1<U> {
         let func = |s: &Self, evaluate: Evaluate| {
@@ -489,11 +497,6 @@ impl<U: EosUnit, E: EquationOfState> State<U, E> {
             .unwrap()
     }
 
-    /// Activity coefficient $\ln \gamma_i = \ln \varphi_i(T, p, \mathbf{N}) - \ln \varphi_i(T, p)$
-    pub fn ln_symmetric_activity_coefficient(&self) -> Array1<f64> {
-        self.ln_phi() - self.ln_phi_pure()
-    }
-
     /// Helmholtz energy $A$ evaluated for each contribution of the equation of state.
     pub fn helmholtz_energy_contributions(&self) -> Vec<(String, QuantityScalar<U>)> {
         let new_state = self.derive0();
