Tried to add PICMI class for ionization

1 files changed, 9 insertions, 0 deletions

diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py
index 6e9c9153b..b207ef0a5 100644
--- a/Python/pywarpx/picmi.py
+++ b/Python/pywarpx/picmi.py
@@ -56,6 +56,15 @@ class Species(picmistandard.PICMI_Species):
                 if self.mass is None:
                     self.mass = element.mass*periodictable.constants.atomic_mass_constant
 
+        if 'ionization' in self.interactions:
+            if self.interactions[1]=='ADK':
+                species.do_field_ionization=1
+                species.physical_element=self.particle_type
+                species.ionization_product_species = self.interactions[2]
+                species.ionization_initial_level = self.charge_state
+            else:
+                raise Exception('WarpX only supports the ionization model of ADK')
+
     def initialize_inputs(self, layout, initialize_self_fields=False):
         self.species_number = pywarpx.particles.nspecies
         pywarpx.particles.nspecies += 1