[Thermo] add support for concentration string in findIsomers

Cantera · Sep 24, 2019 · ff5a863 · ff5a863
1 parent 84a0981
commit ff5a863
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 2 deletions.
diff --git a/include/cantera/thermo/Phase.h b/include/cantera/thermo/Phase.h
@@ -717,6 +717,11 @@ class Phase
     //!     @return A vector of species names for matching species.
     virtual std::vector<std::string> findIsomers(const compositionMap& compMap) const;
 
+    //! Return a vector with isomers names matching a given composition string
+    //!     @param comp String containing a composition map
+    //!     @return A vector of species names for matching species.
+    virtual std::vector<std::string> findIsomers(const std::string& comp) const;
+
     //! Return the Species object for the named species. Changes to this object
     //! do not affect the ThermoPhase object until the #modifySpecies function
     //! is called.

diff --git a/interfaces/cython/cantera/_cantera.pxd b/interfaces/cython/cantera/_cantera.pxd
@@ -175,6 +175,7 @@ cdef extern from "cantera/thermo/ThermoPhase.h" namespace "Cantera":
         void setCaseSensitiveSpecies(cbool)
         cbool addSpeciesAlias(string, string) except +translate_exception
         vector[string] findIsomers(Composition&) except +translate_exception
+        vector[string] findIsomers(string) except +translate_exception
 
         double molecularWeight(size_t) except +translate_exception
         double meanMolecularWeight()

diff --git a/interfaces/cython/cantera/test/test_thermo.py b/interfaces/cython/cantera/test/test_thermo.py
@@ -1116,6 +1116,8 @@ def test_isomers(self):
         gas = ct.Solution('nDodecane_Reitz.yaml')
         iso = gas.find_isomers({'C':4, 'H':9, 'O':2})
         self.assertTrue(len(iso) == 2)
+        iso = gas.find_isomers('C:4, H:9, O:2')
+        self.assertTrue(len(iso) == 2)
         iso = gas.find_isomers({'C':7, 'H':15})
         self.assertTrue(len(iso) == 1)
         iso = gas.find_isomers({'C':7, 'H':16})

diff --git a/interfaces/cython/cantera/thermo.pyx b/interfaces/cython/cantera/thermo.pyx
@@ -531,8 +531,12 @@ cdef class ThermoPhase(_SolutionBase):
         Find species/isomers matching a composition specified by *comp*.
         """
 
-        assert isinstance(comp, dict), 'Composition needs to be specified as dictionary'
-        iso = self.thermo.findIsomers(comp_map(comp))
+        if isinstance(comp, dict):
+            iso = self.thermo.findIsomers(comp_map(comp))
+        elif isinstance(comp, (str, bytes)):
+            iso = self.thermo.findIsomers(stringify(comp))
+        else:
+            raise CanteraError('Invalid composition')
 
         return [b.decode('utf-8') for b in iso]
 

diff --git a/src/thermo/Phase.cpp b/src/thermo/Phase.cpp
@@ -867,6 +867,11 @@ vector<std::string> Phase::findIsomers(const compositionMap& compMap) const
     return isomersVector;
 }
 
+vector<std::string> Phase::findIsomers(const std::string& comp) const
+{
+    return findIsomers(parseCompString(comp));
+}
+
 shared_ptr<Species> Phase::species(const std::string& name) const
 {
     size_t k = speciesIndex(name);