Merge pull request #41 from ChristopherMayes/fix_genesis_skip

Fix importing Genesis4 with zero current slices
ChristopherMayes · Jun 28, 2023 · 3607576 · 3607576
2 parents 7d3a7a6 + 0066bc6
commit 3607576
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pmd_beamphysics/interfaces/genesis.py b/pmd_beamphysics/interfaces/genesis.py
@@ -520,6 +520,7 @@ def genesis4_par_to_data(h5, species='electron', smear=True):
         assert len(current) == 1
         # Skip zero current slices. These usually have nans in the particle data.
         if current == 0:
+            i0 += sample
             continue        
 
         x.append( g['x'][:])