diff options
author | 2018-12-11 10:37:21 -0800 | |
---|---|---|
committer | 2018-12-11 10:37:21 -0800 | |
commit | b53d4a7ea5c3963e16727aafdbda01771af04c0d (patch) | |
tree | 33ddccf2ceb4ae6670baa4ed31b9b97cb7035473 /Python | |
parent | 4e4a84aba36b2d2168cf41077f596549180f0dd9 (diff) | |
download | WarpX-b53d4a7ea5c3963e16727aafdbda01771af04c0d.tar.gz WarpX-b53d4a7ea5c3963e16727aafdbda01771af04c0d.tar.zst WarpX-b53d4a7ea5c3963e16727aafdbda01771af04c0d.zip |
fix conflicts for merge revert
Diffstat (limited to 'Python')
-rw-r--r-- | Python/pywarpx/Bucket.py | 4 | ||||
-rw-r--r-- | Python/pywarpx/WarpX.py | 2 | ||||
-rw-r--r-- | Python/pywarpx/picmi.py | 90 |
3 files changed, 1 insertions, 95 deletions
diff --git a/Python/pywarpx/Bucket.py b/Python/pywarpx/Bucket.py index 66494a700..c73b3dac9 100644 --- a/Python/pywarpx/Bucket.py +++ b/Python/pywarpx/Bucket.py @@ -20,10 +20,6 @@ class Bucket(object): except KeyError: return object.__getattr__(self, name) - def check_consistency(self, vname, value, errmsg): - if vname in self.argvattrs: - assert (self.argvattrs[vname] is None) or (self.argvattrs[vname] == value), Exception(errmsg) - def attrlist(self): "Concatenate the attributes into a string" result = [] diff --git a/Python/pywarpx/WarpX.py b/Python/pywarpx/WarpX.py index f58d4f111..9c4a4b380 100644 --- a/Python/pywarpx/WarpX.py +++ b/Python/pywarpx/WarpX.py @@ -70,7 +70,7 @@ class WarpX(Bucket): argv = self.create_argv_list() with open(filename, 'w') as ff: - for k, v in kw.items(): + for k, v in kw.iteritems(): ff.write('{0} = {1}\n'.format(k, v)) for arg in argv: diff --git a/Python/pywarpx/picmi.py b/Python/pywarpx/picmi.py index 81b1543bb..1d02f8b29 100644 --- a/Python/pywarpx/picmi.py +++ b/Python/pywarpx/picmi.py @@ -462,9 +462,6 @@ class Simulation(picmistandard.PICMI_Simulation): self.lasers[i].initialize_inputs() self.laser_injection_methods[i].initialize_inputs(self.lasers[i]) - for diagnostic in self.diagnostics: - diagnostic.initialize_inputs() - def initialize_warpx(self): if self.warpx_initialized: return @@ -495,90 +492,3 @@ class Simulation(picmistandard.PICMI_Simulation): if self.warpx_initialized: self.warpx_initialized = False pywarpx.warpx.finalize() - - -# ---------------------------- -# Simulation frame diagnostics -# ---------------------------- - - -class FieldDiagnostic(picmistandard.PICMI_FieldDiagnostic): - def init(self, kw): - - self.plot_raw_fields = kw.pop('warpx_plot_raw_fields', None) - self.plot_raw_fields_guards = kw.pop('warpx_plot_raw_fields_guards', None) - self.plot_finepatch = kw.pop('warpx_plot_finepatch', None) - self.plot_crsepatch = kw.pop('warpx_plot_crsepatch', None) - - def initialize_inputs(self): - # --- For now, the period must be the same as plot_int if set - pywarpx.amr.check_consistency('plot_int', self.period, 'The period must be the same for all simulation frame diagnostics') - pywarpx.amr.plot_int = self.period - - if 'rho' in self.data_list: - pywarpx.warpx.plot_rho = 1 - if 'dive' in self.data_list: - pywarpx.warpx.plot_dive = 1 - if 'divb' in self.data_list: - pywarpx.warpx.plot_divb = 1 - if 'F' in self.data_list: - pywarpx.warpx.plot_F = 1 - if 'proc_number' in self.data_list: - pywarpx.warpx.plot_proc_number = 1 - - pywarpx.warpx.plot_raw_fields = self.plot_raw_fields - pywarpx.warpx.plot_raw_fields_guards = self.plot_raw_fields_guards - - pywarpx.amr.check_consistency('plot_finepatch', self.plot_finepatch, 'The fine patch flag must be the same for all simulation frame field diagnostics') - pywarpx.amr.check_consistency('plot_crsepatch', self.plot_crsepatch, 'The coarse patch flag must be the same for all simulation frame field diagnostics') - pywarpx.warpx.plot_finepatch = self.plot_finepatch - pywarpx.warpx.plot_crsepatch = self.plot_crsepatch - -class ElectrostaticFieldDiagnostic(picmistandard.PICMI_ElectrostaticFieldDiagnostic): - def initialize_inputs(self): - # --- For now, the period must be the same as plot_int if set - pywarpx.amr.check_consistency('plot_int', self.period, 'The period must be the same for all simulation frame diagnostics') - pywarpx.amr.plot_int = self.period - - -class ParticleDiagnostic(picmistandard.PICMI_ParticleDiagnostic): - def initialize_inputs(self): - # --- For now, the period must be the same as plot_int if set - pywarpx.amr.check_consistency('plot_int', self.period, 'The period must be the same for all simulation frame diagnostics') - pywarpx.amr.plot_int = self.period - - if 'part_per_cell' in self.data_list: - pywarpx.warpx.plot_part_per_cell = 1 - if 'part_per_grid' in self.data_list: - pywarpx.warpx.plot_part_per_grid = 1 - if 'part_per_proc' in self.data_list: - pywarpx.warpx.plot_part_per_proc = 1 - - -# ---------------------------- -# Lab frame diagnostics -# ---------------------------- - - -class LabFrameFieldDiagnostic(picmistandard.PICMI_LabFrameFieldDiagnostic): - def initialize_inputs(self): - - pywarpx.warpx.check_consistency('num_snapshots_lab', self.num_snapshots, 'The number of snapshots must be the same in all lab frame diagnostics') - pywarpx.warpx.check_consistency('dt_snapshots_lab', self.dt_snapshots, 'The time between snapshots must be the same in all lab frame diagnostics') - - pywarpx.warpx.do_boosted_frame_diagnostic = 1 - pywarpx.warpx.num_snapshots_lab = self.num_snapshots - pywarpx.warpx.dt_snapshots_lab = self.dt_snapshots - pywarpx.warpx.do_boosted_frame_fields = 1 - - -class LabFrameParticleDiagnostic(picmistandard.PICMI_LabFrameParticleDiagnostic): - def initialize_inputs(self): - - pywarpx.warpx.check_consistency('num_snapshots_lab', self.num_snapshots, 'The number of snapshots must be the same in all lab frame diagnostics') - pywarpx.warpx.check_consistency('dt_snapshots_lab', self.dt_snapshots, 'The time between snapshots must be the same in all lab frame diagnostics') - - pywarpx.warpx.do_boosted_frame_diagnostic = 1 - pywarpx.warpx.num_snapshots_lab = self.num_snapshots - pywarpx.warpx.dt_snapshots_lab = self.dt_snapshots - pywarpx.warpx.do_boosted_frame_particles = 1 |