add EPOCH reader support for subsets

This commit also fixes that species will be recognized even if Px is not dumped. fix #8
skuschel · Dec 25, 2014 · 0fb9903 · 0fb9903
1 parent bd7c244
commit 0fb9903
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/postpic/analyzer/analyzer.py b/postpic/analyzer/analyzer.py
@@ -122,7 +122,7 @@ def identifyspecies(cls, species):
         ejected_c6b, tracer_proton, protonb
         """
         ret = {'tracer': False, 'ejected': False, 'name': species}
-        s = species.replace('/', '')
+        s = species.replace('/', '_')
 
         # Regex for parsing ion species name.
         # See docsting for valid examples

diff --git a/postpic/datareader/epochsdf.py b/postpic/datareader/epochsdf.py
@@ -92,11 +92,12 @@ def grid(self, axis):
         return self['Grid/Grid/' + axsuffix]
 
     def listSpecies(self):
-        ret = []
+        ret = set()
         for key in self.keys():
-            match = re.match('Particles/Px/(\w+)', key)
+            match = re.match('Particles/\w+/(\w+(/\w+)?)', key)
             if match:
-                ret = np.append(ret, match.group(1))
+                ret.add(match.group(1))
+        ret = list(ret)
         ret.sort()
         return ret