from numpy import *
import matplotlib.pyplot as pl
from sixtracklib.ctrack import *
from sixtracklib.pysixtrack import *
from sixtracklib.sixdump import SixDump
from sixtracklib.sixinput import SixTrackInput
from sixtracklib.sixbin import SixBin
exampledir = 'med_80_mo_3_s1_4-6_30'
# exampledir='med_80_bb_2_s1_4-6_30'
# exampledir='med_80_mo_3_s1_0sigma'
s = SixTrackInput(exampledir)
convert = {
'drift': Drift,
'mult': Multipole,
'cav': Cavity,
'align': Align,
'block': Block
}
out, rest = s.expand_struct(convert=convert)
b = SixBin(exampledir)
p = Bunch(**b.get_particle(1, 0))
partiInt = ctypes.c_int * 5
partfFloat = ctypes.c_double * 14
parti = partiInt(1, 9, 0, 5, 5)
