WSTrack

This package provides parallelized weak-strong beam-beam simulation solvers.

Installation

The package can be installed using Julia's REPL

julia> import Pkg
julia> Pkg.add(url="https://github.com/ykkan/WSTrack.jl.git")

or with Pkg mode (hitting ] in the command prompt)

pkg> add https://github.com/ykkan/WSTrack.jl.git

Example - EIC CDR Simulation (GPU simulation)

using WSTrack
using StaticArrays

# create weak beam (proton) on GPU 
nmpar = 1024000
beam_gpu = BeamGPU(
            sp=PROTON, 
            num_particle=0.6881e11, 
            energy=275.0e9, 
            num_macro_particle=nmpar,
            dist=Gaussian(emmx=11.3e-9, emmy=1.0e-9, betx=0.8, bety=0.072, sigz=6e-2, sigpz=6.6e-4)
           )

# crate strong beam element (electron)
sb = StrongBeam(npar=1.7203e11, sp=ELECTRON,
                emmx=20.0e-9, emmy=1.3e-9, sigz=0.7e-2, betx=0.45, bety=0.056, 
                nslice=7, cross_angle=25e-3, f_crab=200.0e6)

# define linear one-turn map element
oneturn = OneTurn(betx=0.8, bety=0.072, alx=0.0, aly=0.0, 
                  Qx=0.228, Qy=0.210, Qs=0.01,
                  sigz=6e-2, sigpz=6.6e-4)


# define crab cavity elements
ccstrength = -0.0003876287347981857
cc1 = CrabCavity(f=200.0e6, kick_strength= ccstrength, strength_factors=SVector(1.0), phi=0.0)
cc2 = CrabCavity(f=200.0e6, kick_strength= ccstrength, strength_factors=SVector(1.0), phi=0.0)


# two transfermap elements between ip and crab cavity
backward = @SMatrix [0 -32.2490309931942 0 0 0 0;
              0.031008683647302117 0 0 0 0 0;
                0 0 1 0 0 0;
                0 0 0 1 0 0;
                0 0 0 0 1 0;
                0 0 0 0 0 1;]

forward = @SMatrix [0 32.2490309931942 0 0 0 0;
              -0.031008683647302117 0 0 0 0 0;
                0 0 1 0 0 0;
                0 0 0 1 0 0;
                0 0 0 0 1 0;
                0 0 0 0 0 1;]

b = @SVector [0.0, 0.0, 0.0, 0.0, 0.0, 0.0]

ccip_backward = LinearMap(backward, b)
ccip_forward = LinearMap(forward, b)

# create two diagnosis elements
gdiag = GeneralDiagnosis()
ldiag = LuminosityDiagnosis()

# tracking through the previously defined elements
interact!(beam_gpu, gdiag)
for i in 1:1000000
  interact!(beam_gpu, ccip_backward)
  interact!(beam_gpu, cc1)
  interact!(beam_gpu, ccip_forward)
  interact!(beam_gpu, sb, ldiag)
  interact!(beam_gpu, ccip_forward)
  interact!(beam_gpu, cc2)
  interact!(beam_gpu, ccip_backward)
  interact!(beam_gpu, oneturn)
  interact!(beam_gpu, gdiag)
end