positions of RF gaps and energies after each gap.
Keep in mind that phases in pyORBIT linac RF cavities are shifted by 180
because we track the negative ions for SNS.
"""
import sys
import math
from orbit.sns_linac import SimplifiedLinacParser
from orbit.sns_linac import LinacLatticeFactory, LinacAccLattice
from bunch import Bunch
from orbit.lattice import AccLattice, AccNode, AccActionsContainer
parser = SimplifiedLinacParser("../SNS_Linac_XML/sns_linac.xml")
linacTree = parser.getLinacStructureTree()
print "======================================="
print "Total length=", linacTree.getLength()
print "======================================="
sequences = linacTree.getSeqs()
totalLength = 0.
for seq in sequences:
totalLength += seq.getLength()
print "seq=", seq.getName(), " L=", seq.getLength(
), " total length=", totalLength
lattFactory = LinacLatticeFactory(linacTree)
accLattice = lattFactory.getLinacAccLattice([
"MEBT", "DTL1", "DTL2", "DTL3", "DTL4", "DTL5", "DTL6", "CCL1", "CCL2",
"CCL3", "CCL4", "SCLMed", "SCLHigh"
from orbit.bunch_generators import TwissContainer
from orbit.bunch_generators import WaterBagDist3D, GaussDist3D, KVDist3D
from bunch import Bunch
from bunch import BunchTwissAnalysis
from orbit.lattice import AccLattice, AccNode, AccActionsContainer
from sns_linac_bunch_generator import SNS_Linac_BunchGenerator
random.seed(100)
parser = SimplifiedLinacParser("../SNS_Linac_XML/sns_linac_sts.xml")
linacTree = parser.getLinacStructureTree()
print "======================================="
print "Total length=",linacTree.getLength()
print "======================================="
sequences = linacTree.getSeqs()
totalLength = 0.
for seq in sequences:
totalLength += seq.getLength()
print "seq=",seq.getName()," L=",seq.getLength()," total length=",totalLength
lattFactory = LinacLatticeFactory(linacTree)
lattFactory.setMaxDriftLength(0.01)
#accLattice = lattFactory.getLinacAccLattice(["MEBT","DTL1","DTL2","DTL3","DTL4","DTL5","DTL6","CCL1","CCL2","CCL3","CCL4","SCLMed","SCLHigh"])
#accLattice = lattFactory.getLinacAccLattice(["SCLMed","SCLHigh"])
accLattice = lattFactory.getLinacAccLattice(["MEBT","DTL1","DTL2","DTL3","DTL4","DTL5","DTL6","CCL1","CCL2","CCL3","CCL4"])