ESS Beam Physics Python Tools

pip install git+https://gitlab.esss.lu.se/ess-bp/ess-python-tools.git -U --user

from ess import lib_tw

file_name_lat    ='lattice2.dat'
file_name_lat_new='lattice2.adjusted.dat'
file_name_fmap   =['Field_Maps/1D/'+f for f in ['Spoke_W_coupler.edz','MB_W_coupler.edz','HB_W_coupler.edz']]
file_name_st     ='Steerer_Values.txt'
file_name_adj    ='Adjusted_Values.txt'

lat=lib_tw.LATTICE(file_name_lat,file_name_fmap)
# Updates based on Steerer_Values:
lat.update_st(file_name_st)
# Updates based on Adjusted_Values:
lat.update_adj(file_name_adj)

lat.get_tw(file_name_lat_new)

#-- If we want to reset all steerers to 0 ...
#for i in range(len(lat.lst)):
#    if lat.lst[i].typ=='THIN_STEERING': lat.lst[i].BLx=0.0; lat.lst[i].BLy=0.0
#    if lat.lst[i].typ=='STEERER'      : lat.lst[i].Bx =0.0; lat.lst[i].By =0.0

# Note:
# - For a density file, be careful with the number of runs.
# - The instances for losses are "loss_pow", "loss_num", "loss_pow_ave", and etc for DENSITY class.
# - For PARTRAN class, loss_elem2den function is already implemented as a method. 
# - If file_name_dt isn't given, the half-cell lengths are used instead of drift-tube lnegths.
# - dlt_dt is a small number used to identify DTL_CEL elements. (Play with this number is there is an error message.)
#
# - An example to make a bar plot with an output file of this script can be found in
#   https://gitlab.esss.lu.se/ess-bp/pgfplots-examples/tree/master/loss.den

from ess import lib_tw

file_name_den ='density_file_name'
file_name_ptrn='partran_out_file_name'

file_name_dt  ='len.dt_dtl.v85.txt'  # File with (half cell length) vs (drift tube length) for DTL cells

#-- An example for a density file

den=DENSITY(file_name_den)
s_from_den   =den.s
loss_from_den=den.loss_pow         # For a file w/ single run
#loss_from_den=den.loss_pow[Nrun]  # For a file w/ multi runs (e.g., file from an error study)

loss_den_from_den=loss_elem2den(s_from_den,loss_from_den,file_name_dt,dtl_dt=5e-6)

# Note len(s_from_den) = len(loss_den_from_den) when writing to a file.

#-- An example for a partran out file

ptrn=PARTRAN(file_name_ptrn)

loss_den_from_ptrn=ptrn.loss_den(file_name_dt,dtl_dt=5e-6)

# Note len(ptrn.s) = len(loss_den_from_ptrn) when writing to a file.

# Example: read in plt file, write dst file for each location
from ess import TraceWin

data=TraceWin.plt('dtl1.plt')
for i in data.Nelp:
    data.save_dst(i,'distribution_{}.dst'.format(i))

# Example: read in many Density files from a folder, put data into one HDF5 file

from ess import TraceWin
import os

for f in os.listdir('calc'):
    if f[:7]=='Density' and f[-3:]=='dat':
        fpath=os.path.join('calc',f)
        group_name=f[8:-4]
        data=TraceWin.density_file(fpath)
        data.savetohdf('Density.h5',group_name)

usage: tracewin_errorstudy [-h] [-s SEED] [-c CALC_DIR] [-n NUM_JOBS]
                           [-t STUDY] [-m MULTI] [-l LATTICE] [-q SETTINGS]
                           [-p PRIORITY]

Simple setup script to run TraceWin on HT Condor.. Default values in square
brackets

optional arguments:
  -h, --help   show this help message and exit
  -s SEED      Define seed [None]
  -c CALC_DIR  Path to calculation folder [temp/]
  -n NUM_JOBS  Number of jobs [Count number of copied runs in TW]
  -t STUDY     Statistical study number [1]
  -m MULTI     Multi-dynamic study [1]
  -l LATTICE   The lattice file for the project [lattice.dat]
  -q SETTINGS  The settings for the project template (yaml) [settings.yml]
  -p PRIORITY  Job priority

tracewin_errorstudy -n 100 -c calc

tracewin_errorstudy -n 50 -c calc
# wait until simulations have finished (check condor_q, condor_status, condor_history)
tracewin_errorstudy -n 50 -m 40 -c calc