Parsing Genesis2 Field data¶

In [ ]:

from genesis import Genesis2, parsers

import numpy as np
import os

from genesis import Genesis2, parsers import numpy as np import os

In [ ]:

# Make some dfl, fld data
G = Genesis2("data/basic/genesis.in")
# Turn on field output
G["idmpfld"] = 1
# Turn on particle output
# G['idmppar'] = 1
G["npart"] = 2048

# Turn on history
# G['ippart'] = 10
G["ipradi"] = 10

G.run()

# Make some dfl, fld data G = Genesis2("data/basic/genesis.in") # Turn on field output G["idmpfld"] = 1 # Turn on particle output # G['idmppar'] = 1 G["npart"] = 2048 # Turn on history # G['ippart'] = 10 G["ipradi"] = 10 G.run()

In [ ]:

# These are the files written
!ls {G.path}

# These are the files written !ls {G.path}

In [ ]:

# Change this
# test_dir = 'path/to/your/output'
test_dir = G.path


out_fname = os.path.join(test_dir, "genesis.out")
dfl_fname = os.path.join(test_dir, "genesis.out.dfl")
fld_fname = os.path.join(test_dir, "genesis.out.fld")

# Get parameters from .out file
odat = parsers.parse_genesis_out(out_fname)
params = odat["param"]


my_ncar = params["ncar"]
my_dgrid = params["dgrid"]

my_nz = 1

# Change this # test_dir = 'path/to/your/output' test_dir = G.path out_fname = os.path.join(test_dir, "genesis.out") dfl_fname = os.path.join(test_dir, "genesis.out.dfl") fld_fname = os.path.join(test_dir, "genesis.out.fld") # Get parameters from .out file odat = parsers.parse_genesis_out(out_fname) params = odat["param"] my_ncar = params["ncar"] my_dgrid = params["dgrid"] my_nz = 1

In [ ]:

my_dfl = parsers.parse_genesis_dfl(dfl_fname, nx=my_ncar)
my_dfl

my_dfl = parsers.parse_genesis_dfl(dfl_fname, nx=my_ncar) my_dfl

In [ ]:

my_fld = parsers.parse_genesis_fld(fld_fname, my_ncar, my_nz)
my_fld[-1]

my_fld = parsers.parse_genesis_fld(fld_fname, my_ncar, my_nz) my_fld[-1]

Plot¶

In [ ]:

import matplotlib.pyplot as plt

import matplotlib.pyplot as plt

In [ ]:

# Field phase at end, slice 0
def plot_field(dat, dgrid):
    ndat = np.angle(dat)
    plt.imshow(ndat, extent=[1000 * dgrid * i for i in [-1, 1, -1, 1]])
    plt.xlabel("x (mm)")
    plt.ylabel("y (mm)")
    plt.show()


plot_field(my_dfl[:, :, 0], my_dgrid)

# Field phase at end, slice 0 def plot_field(dat, dgrid): ndat = np.angle(dat) plt.imshow(ndat, extent=[1000 * dgrid * i for i in [-1, 1, -1, 1]]) plt.xlabel("x (mm)") plt.ylabel("y (mm)") plt.show() plot_field(my_dfl[:, :, 0], my_dgrid)

In [ ]:

# Field phrase from history file, slice 0
plot_field(my_fld[:, :, 0, -1], my_dgrid)

# Field phrase from history file, slice 0 plot_field(my_fld[:, :, 0, -1], my_dgrid)