def getHierarchy(self,
interp_order,
refinement_boxes,
qty,
diag_outputs,
nbr_part_per_cell=100,
density=_density,
beam=False,
time_step_nbr=1,
smallest_patch_size=None,
largest_patch_size=10,
cells=120,
dl=0.1,
ndim=1):
diag_outputs = f"phare_outputs/init/{diag_outputs}"
from pyphare.pharein import global_vars
global_vars.sim = None
if smallest_patch_size is None:
from pyphare.pharein.simulation import check_patch_size
_, smallest_patch_size = check_patch_size(
ndim, interp_order=interp_order, cells=cells)
Simulation(
smallest_patch_size=smallest_patch_size,
largest_patch_size=largest_patch_size,
time_step_nbr=time_step_nbr,
final_time=30.,
boundary_types=["periodic"] * ndim,
cells=[cells] * ndim,
dl=[dl] * ndim,
interp_order=interp_order,
refinement_boxes=refinement_boxes,
diag_options={
"format": "phareh5",
"options": {
"dir": diag_outputs,
"mode": "overwrite"
}
},
strict=True,
)
def beam_density(*xyz):
return np.zeros_like(xyz[0]) + 0.3
def bx(*xyz):
return 1.
def by(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.cos(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def bz(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.sin(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def vx(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.cos(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def vy(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.cos(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def vz(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.sin(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def vth(*xyz):
return 0.01 + np.zeros_like(xyz[0])
def vthx(*xyz):
return vth(*xyz)
def vthy(*xyz):
return vth(*xyz)
def vthz(*xyz):
return vth(*xyz)
if beam:
MaxwellianFluidModel(bx=bx,
by=by,
bz=bz,
protons={
"charge": 1,
"density": density,
"vbulkx": vx,
"vbulky": vy,
"vbulkz": vz,
"vthx": vthx,
"vthy": vthy,
"vthz": vthz,
"nbr_part_per_cell": nbr_part_per_cell,
"init": {
"seed": 1337
}
},
beam={
"charge": 1,
"density": beam_density,
"vbulkx": vx,
"vbulky": vy,
"vbulkz": vz,
"vthx": vthx,
"vthy": vthy,
"vthz": vthz,
"nbr_part_per_cell": nbr_part_per_cell,
"init": {
"seed": 1337
}
})
else:
MaxwellianFluidModel(bx=bx,
by=by,
bz=bz,
protons={
"charge": 1,
"density": density,
"vbulkx": vx,
"vbulky": vy,
"vbulkz": vz,
"vthx": vthx,
"vthy": vthy,
"vthz": vthz,
"nbr_part_per_cell": nbr_part_per_cell,
"init": {
"seed": 1337
}
})
ElectronModel(closure="isothermal", Te=0.12)
for quantity in ["E", "B"]:
ElectromagDiagnostics(quantity=quantity,
write_timestamps=np.zeros(time_step_nbr),
compute_timestamps=np.zeros(time_step_nbr))
for quantity in ["density", "bulkVelocity"]:
FluidDiagnostics(quantity=quantity,
write_timestamps=np.zeros(time_step_nbr),
compute_timestamps=np.zeros(time_step_nbr))
poplist = ["protons", "beam"] if beam else ["protons"]
for pop in poplist:
for quantity in ["density", "flux"]:
FluidDiagnostics(quantity=quantity,
write_timestamps=np.zeros(time_step_nbr),
compute_timestamps=np.zeros(time_step_nbr),
population_name=pop)
for quantity in ['domain', 'levelGhost', 'patchGhost']:
ParticleDiagnostics(quantity=quantity,
compute_timestamps=np.zeros(time_step_nbr),
write_timestamps=np.zeros(time_step_nbr),
population_name=pop)
Simulator(global_vars.sim).initialize().reset()
eb_hier = None
if qty in ["e", "eb"]:
eb_hier = hierarchy_from(h5_filename=diag_outputs + "/EM_E.h5",
hier=eb_hier)
if qty in ["b", "eb"]:
eb_hier = hierarchy_from(h5_filename=diag_outputs + "/EM_B.h5",
hier=eb_hier)
if qty in ["e", "b", "eb"]:
return eb_hier
is_particle_type = qty == "particles" or qty == "particles_patch_ghost"
if is_particle_type:
particle_hier = None
if qty == "particles":
particle_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_pop_protons_domain.h5")
particle_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_pop_protons_levelGhost.h5",
hier=particle_hier)
if is_particle_type:
particle_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_pop_protons_patchGhost.h5",
hier=particle_hier)
if qty == "particles":
merge_particles(particle_hier)
if is_particle_type:
return particle_hier
if qty == "moments":
mom_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_density.h5")
mom_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_bulkVelocity.h5",
hier=mom_hier)
mom_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_pop_protons_density.h5",
hier=mom_hier)
mom_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_pop_protons_flux.h5",
hier=mom_hier)
if beam:
mom_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_pop_beam_density.h5",
hier=mom_hier)
mom_hier = hierarchy_from(h5_filename=diag_outputs +
"/ions_pop_beam_flux.h5",
hier=mom_hier)
return mom_hier
def setup_model(ppc=100):
def density(*xyz):
return 1.
def by(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.sin(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def bz(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.sin(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def bx(*xyz):
return 1.
def vx(*xyz):
return 0.
def vy(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.cos(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def vz(*xyz):
from pyphare.pharein.global_vars import sim
L = sim.simulation_domain()
_ = lambda i: 0.1 * np.cos(2 * np.pi * xyz[i] / L[i])
return np.asarray([_(i) for i, v in enumerate(xyz)]).prod(axis=0)
def vthx(*xyz):
return 0.01
def vthy(*xyz):
return 0.01
def vthz(*xyz):
return 0.01
vvv = {
"vbulkx": vx,
"vbulky": vy,
"vbulkz": vz,
"vthx": vthx,
"vthy": vthy,
"vthz": vthz
}
model = ph.MaxwellianFluidModel(
bx=bx,
by=by,
bz=bz,
protons={
"mass": 1,
"charge": 1,
"density": density,
**vvv, "nbr_part_per_cell": ppc,
"init": {
"seed": 1337
}
},
alpha={
"mass": 4,
"charge": 1,
"density": density,
**vvv, "nbr_part_per_cell": ppc,
"init": {
"seed": 2334
}
},
)
ElectronModel(closure="isothermal", Te=0.12)
return model
