Shot
One discharge. Data is read from disk the first time it is requested and then
cached.
Shot(shot, imp_path, rest_path, lbo_paths, regime, species, heating,
time_interval=None, folder=None)
| Member | Description |
|---|---|
imp(time_interval=None), rest(...) |
traces as a DataFrame (scaled; see the sign note in io) |
lbo(idx) |
ne(rho) profile from the -lbo{idx+1}.txt file |
t_lbos, n_lbo |
times and number of -lboN.txt files |
signals, default_signal |
-imp signals; the species' core line |
emission(signal=None) |
(t, I) with the physical emission (ysc sign corrected) |
find_injections(**kw) |
listed + detected injections → DataFrame |
analyze_injections(**kw) |
full pipeline → InjectionSet (pipeline) |
fit_tau_imp(lbo_idx, double=False) |
|
decay(...) |
post-LBO curve with the baseline subtracted (rise included; for plotting) |
compare_decay_models(...) |
legacy (v0.1) |
plot_* |
shortcuts to aug.plot.shot |
GroupShots
A collection of Shot objects with metadata and flags.
group = GroupShots.from_directory(DATA_PATH, name="AUG", flag_bad_files=True)
| Member | Description |
|---|---|
group[shot], len, iteration |
access to the Shot objects |
all, good |
every shot / without the flagged ones |
flagged |
{shot: reason}; filled by the file QC, and you can add to it by hand |
qc_table |
table from aug.io.qc.check_files |
filter(regime, species, heating, exclude_flagged=True) |
subgroup |
apply(func) |
func(shot) on each shot; exceptions are returned and do not stop the batch |
summary() |
one row per shot |
analyze_injections(**kw) |
analyze_shot on each shot → a single InjectionSet |
compare_decay_models_summary, plot_* |
legacy (v0.1) |