diff options
author | 2019-05-07 19:44:54 -0700 | |
---|---|---|
committer | 2019-05-07 19:44:54 -0700 | |
commit | 5f2cfab237461821ee79c0ca17c506b07be95fbd (patch) | |
tree | 302396db649e9bf599db9e687912cb8a53a4bb79 /Python/pywarpx/picmi.py | |
parent | 2b2286fa3221d14feed9961ddeaed70f54f7a4e8 (diff) | |
parent | 8ae480604c740fff0c28838ae84fdc31dc0c8254 (diff) | |
download | WarpX-5f2cfab237461821ee79c0ca17c506b07be95fbd.tar.gz WarpX-5f2cfab237461821ee79c0ca17c506b07be95fbd.tar.zst WarpX-5f2cfab237461821ee79c0ca17c506b07be95fbd.zip |
Merge pull request #119 from ECP-WarpX/inject_laser
Continuous injection for laser particles
Diffstat (limited to 'Python/pywarpx/picmi.py')
-rw-r--r-- | Python/pywarpx/picmi.py | 18 |
1 files changed, 4 insertions, 14 deletions
diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py index a5d9fba87..a439205a7 100644 --- a/Python/pywarpx/picmi.py +++ b/Python/pywarpx/picmi.py @@ -113,6 +113,9 @@ class GaussianBunchDistribution(picmistandard.PICMI_GaussianBunchDistribution): species.uy = self.centroid_velocity[1] species.uz = self.centroid_velocity[2] + if self.fill_in: + species.do_continuous_injection = 1 + class UniformDistribution(picmistandard.PICMI_UniformDistribution): def initialize_inputs(self, species_number, layout, species): @@ -153,14 +156,6 @@ class UniformDistribution(picmistandard.PICMI_UniformDistribution): species.uy = self.directed_velocity[1] species.uz = self.directed_velocity[2] - if self.fill_in: - pywarpx.warpx.do_plasma_injection = 1 - if not hasattr(pywarpx.warpx, 'injected_plasma_species'): - pywarpx.warpx.injected_plasma_species = [] - - pywarpx.warpx.injected_plasma_species.append(species_number) - pywarpx.warpx.num_injected_species = len(pywarpx.warpx.injected_plasma_species) - class AnalyticDistribution(picmistandard.PICMI_AnalyticDistribution): def initialize_inputs(self, species_number, layout, species): @@ -205,12 +200,7 @@ class AnalyticDistribution(picmistandard.PICMI_AnalyticDistribution): species.uz = self.directed_velocity[2] if self.fill_in: - pywarpx.warpx.do_plasma_injection = 1 - if not hasattr(pywarpx.warpx, 'injected_plasma_species'): - pywarpx.warpx.injected_plasma_species = [] - - pywarpx.warpx.injected_plasma_species.append(species_number) - pywarpx.warpx.num_injected_species = len(pywarpx.warpx.injected_plasma_species) + species.do_continuous_injection = 1 class ParticleListDistribution(picmistandard.PICMI_ParticleListDistribution): |