diff options
Diffstat (limited to 'Examples/Modules')
24 files changed, 93 insertions, 49 deletions
diff --git a/Examples/Modules/ParticleBoundaryScrape/analysis_scrape.py b/Examples/Modules/ParticleBoundaryScrape/analysis_scrape.py index 652fd3be3..8082b1133 100755 --- a/Examples/Modules/ParticleBoundaryScrape/analysis_scrape.py +++ b/Examples/Modules/ParticleBoundaryScrape/analysis_scrape.py @@ -1,8 +1,9 @@ #!/usr/bin/env python3 -import yt from pathlib import Path +import yt + # This test shoots a beam of electrons at cubic embedded boundary geometry # At time step 40, none of the particles have hit the boundary yet. At time # step 60, all of them should have been absorbed by the boundary. In the diff --git a/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py b/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py index 5e5a5bf74..ba04580af 100755 --- a/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py +++ b/Examples/Modules/RigidInjection/analysis_rigid_injection_BoostedFrame.py @@ -20,9 +20,10 @@ The simulation runs in a boosted frame, and the analysis is done in the lab frame, i.e., on the back-transformed diagnostics. ''' -import yt import numpy as np import read_raw_data +import yt + yt.funcs.mylog.setLevel(0) # Read data from back-transformed diagnostics diff --git a/Examples/Modules/RigidInjection/analysis_rigid_injection_LabFrame.py b/Examples/Modules/RigidInjection/analysis_rigid_injection_LabFrame.py index c764ef357..624fd8f7c 100755 --- a/Examples/Modules/RigidInjection/analysis_rigid_injection_LabFrame.py +++ b/Examples/Modules/RigidInjection/analysis_rigid_injection_LabFrame.py @@ -20,10 +20,12 @@ case rigid injection is OFF (i.e., the beam starts expanding from -5 microns), in which case a warning is raised. ''' +import os import sys -import yt + import numpy as np -import os +import yt + yt.funcs.mylog.setLevel(0) sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/dive_cleaning/analysis.py b/Examples/Modules/dive_cleaning/analysis.py index c4c09353e..504368fa8 100755 --- a/Examples/Modules/dive_cleaning/analysis.py +++ b/Examples/Modules/dive_cleaning/analysis.py @@ -15,13 +15,16 @@ This script verifies that the field at the end of the simulation corresponds to the theoretical field of a Gaussian beam. """ import sys + import matplotlib + matplotlib.use('Agg') import matplotlib.pyplot as plt -import yt import numpy as np import scipy.constants as scc from scipy.special import gammainc +import yt + yt.funcs.mylog.setLevel(0) # Parameters from the Simulation diff --git a/Examples/Modules/embedded_boundary_cube/analysis_fields.py b/Examples/Modules/embedded_boundary_cube/analysis_fields.py index 05f43ce1c..b81f72d9d 100755 --- a/Examples/Modules/embedded_boundary_cube/analysis_fields.py +++ b/Examples/Modules/embedded_boundary_cube/analysis_fields.py @@ -1,9 +1,12 @@ #!/usr/bin/env python3 -import yt -import os, sys -from scipy.constants import mu_0, pi, c +import os +import sys + import numpy as np +from scipy.constants import c, mu_0, pi +import yt + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/embedded_boundary_cube/analysis_fields_2d.py b/Examples/Modules/embedded_boundary_cube/analysis_fields_2d.py index 8127cc9e5..227048be9 100755 --- a/Examples/Modules/embedded_boundary_cube/analysis_fields_2d.py +++ b/Examples/Modules/embedded_boundary_cube/analysis_fields_2d.py @@ -1,10 +1,11 @@ #!/usr/bin/env python3 -import yt import os import sys -from scipy.constants import mu_0, pi, c + import numpy as np +from scipy.constants import c, mu_0, pi +import yt sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/embedded_boundary_python_API/PICMI_inputs_EB_API.py b/Examples/Modules/embedded_boundary_python_API/PICMI_inputs_EB_API.py index b0653ae01..6dea197d3 100755 --- a/Examples/Modules/embedded_boundary_python_API/PICMI_inputs_EB_API.py +++ b/Examples/Modules/embedded_boundary_python_API/PICMI_inputs_EB_API.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 import numpy as np import pywarpx -from pywarpx import picmi, fields +from pywarpx import fields, picmi max_steps = 1 unit = 1e-3 diff --git a/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields.py b/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields.py index 655322b5b..02f4c8012 100755 --- a/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields.py +++ b/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields.py @@ -7,10 +7,13 @@ # License: BSD-3-Clause-LBNL -import yt -import os, sys -from scipy.constants import mu_0, pi, c +import os +import sys + import numpy as np +from scipy.constants import c, mu_0, pi +import yt + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields_2d.py b/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields_2d.py index 2b48d1a36..5c8c544d0 100755 --- a/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields_2d.py +++ b/Examples/Modules/embedded_boundary_rotated_cube/analysis_fields_2d.py @@ -1,10 +1,11 @@ #!/usr/bin/env python3 -import yt import os import sys -from scipy.constants import mu_0, pi, c + import numpy as np +from scipy.constants import c, mu_0, pi +import yt sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py b/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py index a20f18a68..fc965787f 100755 --- a/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py +++ b/Examples/Modules/gaussian_beam/PICMI_inputs_gaussian_beam.py @@ -1,9 +1,9 @@ #!/usr/bin/env python3 -from pywarpx import picmi #from warp import picmi import argparse +from pywarpx import picmi parser = argparse.ArgumentParser(description="Gaussian beam PICMI example") diff --git a/Examples/Modules/ionization/analysis_ionization.py b/Examples/Modules/ionization/analysis_ionization.py index f11662b0a..d81e04345 100755 --- a/Examples/Modules/ionization/analysis_ionization.py +++ b/Examples/Modules/ionization/analysis_ionization.py @@ -18,10 +18,12 @@ checks that, after the laser went through the plasma, ~32% of Nitrogen ions are N5+, in agreement with theory from Chen's article. """ +import os import sys -import yt + import numpy as np -import os +import yt + yt.funcs.mylog.setLevel(0) sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/laser_injection/analysis_1d.py b/Examples/Modules/laser_injection/analysis_1d.py index 7c8e985f6..4f4bd505d 100755 --- a/Examples/Modules/laser_injection/analysis_1d.py +++ b/Examples/Modules/laser_injection/analysis_1d.py @@ -12,13 +12,16 @@ # the simulation and it compares it with theory. It also checks that the # central frequency of the Fourier transform is the expected one. -import yt import sys + import matplotlib +import yt + matplotlib.use('Agg') import matplotlib.pyplot as plt import numpy as np from scipy.signal import hilbert + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/laser_injection/analysis_2d.py b/Examples/Modules/laser_injection/analysis_2d.py index e26ae4478..377931718 100755 --- a/Examples/Modules/laser_injection/analysis_2d.py +++ b/Examples/Modules/laser_injection/analysis_2d.py @@ -17,15 +17,18 @@ # the simulation and it compares it with theory. It also checks that the # central frequency of the Fourier transform is the expected one. -import yt +import os import sys + import matplotlib + matplotlib.use('Agg') import matplotlib.pyplot as plt +from mpl_toolkits.axes_grid1 import make_axes_locatable import numpy as np -import os from scipy.signal import hilbert -from mpl_toolkits.axes_grid1 import make_axes_locatable +import yt + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/laser_injection/analysis_laser.py b/Examples/Modules/laser_injection/analysis_laser.py index 5e2234c30..8dde8d6e9 100755 --- a/Examples/Modules/laser_injection/analysis_laser.py +++ b/Examples/Modules/laser_injection/analysis_laser.py @@ -8,12 +8,15 @@ # License: BSD-3-Clause-LBNL +import os import sys + import matplotlib + matplotlib.use('Agg') import matplotlib.pyplot as plt import numpy as np -import os + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/laser_injection_from_file/analysis.py b/Examples/Modules/laser_injection_from_file/analysis.py index 05a245e52..03c2a3c5c 100755 --- a/Examples/Modules/laser_injection_from_file/analysis.py +++ b/Examples/Modules/laser_injection_from_file/analysis.py @@ -16,14 +16,17 @@ # - Compute the theory for laser envelope at time T # - Compare theory and simulation, for both envelope and central frequency -import yt ; yt.funcs.mylog.setLevel(50) -import numpy as np +import glob +import os + import matplotlib + matplotlib.use('Agg') import matplotlib.pyplot as plt +import numpy as np from scipy.signal import hilbert -import glob -import os + +import yt ; yt.funcs.mylog.setLevel(50) #Maximum acceptable error for this test relative_error_threshold = 0.065 diff --git a/Examples/Modules/nci_corrector/analysis_ncicorr.py b/Examples/Modules/nci_corrector/analysis_ncicorr.py index 409c4ba2f..d3849e707 100755 --- a/Examples/Modules/nci_corrector/analysis_ncicorr.py +++ b/Examples/Modules/nci_corrector/analysis_ncicorr.py @@ -8,11 +8,13 @@ # License: BSD-3-Clause-LBNL -import sys -import yt import re +import sys + import numpy as np import scipy.constants as scc +import yt + yt.funcs.mylog.setLevel(0) fn = sys.argv[1] diff --git a/Examples/Modules/qed/breit_wheeler/analysis_core.py b/Examples/Modules/qed/breit_wheeler/analysis_core.py index cb65fccb9..61d928bf2 100755 --- a/Examples/Modules/qed/breit_wheeler/analysis_core.py +++ b/Examples/Modules/qed/breit_wheeler/analysis_core.py @@ -7,13 +7,12 @@ # # License: BSD-3-Clause-LBNL +import matplotlib.pyplot as plt import numpy as np -import scipy.special as spe import scipy.integrate as integ +import scipy.special as spe import scipy.stats as st -import matplotlib.pyplot as plt - # This script performs detailed checks of the Breit-Wheeler pair production process. # Four populations of photons are initialized with different momenta in different # directions in a background EM field (with non-zero components along each direction). diff --git a/Examples/Modules/qed/breit_wheeler/analysis_opmd.py b/Examples/Modules/qed/breit_wheeler/analysis_opmd.py index e06bfe127..2b1fbc703 100755 --- a/Examples/Modules/qed/breit_wheeler/analysis_opmd.py +++ b/Examples/Modules/qed/breit_wheeler/analysis_opmd.py @@ -8,11 +8,13 @@ # -*- coding: utf-8 -*- import sys + +import analysis_core as ac import openpmd_api as io + #sys.path.insert(1, '../../../../warpx/Regression/Checksum/') #import checksumAPI -import analysis_core as ac # This script is a frontend for the analysis routines # in analysis_core.py (please refer to this file for diff --git a/Examples/Modules/qed/breit_wheeler/analysis_yt.py b/Examples/Modules/qed/breit_wheeler/analysis_yt.py index ef377d55a..e8950419f 100755 --- a/Examples/Modules/qed/breit_wheeler/analysis_yt.py +++ b/Examples/Modules/qed/breit_wheeler/analysis_yt.py @@ -9,11 +9,12 @@ import os import sys + import yt -sys.path.insert(1, '../../../../warpx/Regression/Checksum/') -import checksumAPI +sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import analysis_core as ac +import checksumAPI # This script is a frontend for the analysis routines # in analysis_core.py (please refer to this file for diff --git a/Examples/Modules/qed/quantum_synchrotron/analysis.py b/Examples/Modules/qed/quantum_synchrotron/analysis.py index 2abafa24a..ee6139f75 100755 --- a/Examples/Modules/qed/quantum_synchrotron/analysis.py +++ b/Examples/Modules/qed/quantum_synchrotron/analysis.py @@ -8,16 +8,17 @@ # -*- coding: utf-8 -*- -import yt -import numpy as np import os import sys -import scipy.special as spe + +import numpy as np import scipy.integrate as integ +import scipy.special as spe import scipy.stats as st +import yt + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI - import matplotlib.pyplot as plt # This script performs detailed checks of the Quantum Synchrotron photon emission process. diff --git a/Examples/Modules/qed/schwinger/analysis_schwinger.py b/Examples/Modules/qed/schwinger/analysis_schwinger.py index 8af14a425..c93c92021 100755 --- a/Examples/Modules/qed/schwinger/analysis_schwinger.py +++ b/Examples/Modules/qed/schwinger/analysis_schwinger.py @@ -11,11 +11,13 @@ ## The pair production rate is calculated using the formula described in ## Bulanov, S. S., et al. Physical review letters 104.22 (2010): 220404. -import yt -import numpy as np import os -import sys import re +import sys + +import numpy as np +import yt + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/relativistic_space_charge_initialization/analysis.py b/Examples/Modules/relativistic_space_charge_initialization/analysis.py index 19dc84ef2..569a7d267 100755 --- a/Examples/Modules/relativistic_space_charge_initialization/analysis.py +++ b/Examples/Modules/relativistic_space_charge_initialization/analysis.py @@ -11,14 +11,17 @@ This script checks the space-charge initialization routine, by verifying that the space-charge field of a Gaussian beam corresponds to the expected theoretical field. """ +import os import sys + import matplotlib + matplotlib.use('Agg') import matplotlib.pyplot as plt -import yt import numpy as np -import os import scipy.constants as scc +import yt + yt.funcs.mylog.setLevel(0) sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/resampling/analysis_leveling_thinning.py b/Examples/Modules/resampling/analysis_leveling_thinning.py index ffa2a4333..50fec6282 100755 --- a/Examples/Modules/resampling/analysis_leveling_thinning.py +++ b/Examples/Modules/resampling/analysis_leveling_thinning.py @@ -9,11 +9,13 @@ ## In this test, we check that leveling thinning works as expected on two simple cases. Each case ## corresponds to a different particle species. -import yt -import numpy as np import os import sys + +import numpy as np from scipy.special import erf +import yt + sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI diff --git a/Examples/Modules/space_charge_initialization/analysis.py b/Examples/Modules/space_charge_initialization/analysis.py index 130a54207..0df21e6d9 100755 --- a/Examples/Modules/space_charge_initialization/analysis.py +++ b/Examples/Modules/space_charge_initialization/analysis.py @@ -11,15 +11,18 @@ This script checks the space-charge initialization routine, by verifying that the space-charge field of a Gaussian beam corresponds to the expected theoretical field. """ +import os import sys + import matplotlib + matplotlib.use('Agg') import matplotlib.pyplot as plt -import yt import numpy as np -import os import scipy.constants as scc from scipy.special import gammainc +import yt + yt.funcs.mylog.setLevel(0) sys.path.insert(1, '../../../../warpx/Regression/Checksum/') import checksumAPI |