Particle quantities

 1from idefix2python import RunContext, Pipeline, PartQuantity
 2import utilities
 3from pathlib import Path
 4
 5projectPath = Path(__file__).parent / "data_test"
 6task = "particles_test"
 7# By default the vtks are expected to be in {projetPath}/{task}/outputs/vtks/
 8
 9
10def analytical_drift(t_array):
11    Stokes0 = 1
12    fluid = utilities.Fluid(0.05, -0.5, 0.125, -0.5, Stokes0=Stokes0)
13    r0 = 2
14    return utilities.integrate(fluid.vrDrift, r0, t_array)
15
16
17custom_partQuantities = [
18    PartQuantity(
19        "PART_X1",
20        r"$r$",
21        plot_coords=[0, 0],
22        ref_function=analytical_drift,
23    )
24]
25PartQuantity.suptitle = "A particle radial evolution, with an analytical trajectory"
26
27runContext = RunContext(
28    task,
29    projectPath,
30    active_directions=[0],  # currently necessary for lagrangian particles.)
31)
32pipeline = Pipeline(runContext, partQuantities=custom_partQuantities)
33
34pipeline.run()
../_images/particles_test_timeseries.png

particles_test_timeseries.png