In [2]:
from jpype import *
from numpy import *
In [3]:
JAVA_JVM='/usr/lib/jvm/java-1.7.0-oracle-1.7.0.17.x86_64/jre/lib/amd64/server/libjvm.so'
JAVA_OpenXAL='/home/shared/OpenXAL/lib/openxal/openxal.library-1.0.2.jar'
JAVA_ClassPath='-Djava.class.path='+JAVA_OpenXAL
JAVA_Flags='-ea'
In [4]:
startJVM(JAVA_JVM, JAVA_Flags, JAVA_ClassPath)
In [5]:
_XMLDataManager = JClass("xal.smf.data.XMLDataManager")
In [6]:
Accelerator = _XMLDataManager.acceleratorWithPath('/home/shared/OpenXAL/etc/main.xal')
Accelerator Methods:
http://xaldev.sourceforge.net/javadoc/xal/smf/Accelerator.html
In [7]:
print Accelerator.getAllSeqs()
In [8]:
print Accelerator.getComboSequences()
Sequence Methods:
http://xaldev.sourceforge.net/javadoc/xal/smf/AcceleratorSeq.html
In [9]:
#Sequence = Accelerator.getSequence('mebt')
Sequence = Accelerator.getComboSequence('from-mebt')
In [10]:
Sequence.getNodeCount()
Out[10]:
In [11]:
print Sequence.nodes
In [12]:
print Sequence.getAllNodesOfType('Quad')
In [13]:
print Sequence.getAllNodesOfType('BPM')
In [14]:
Sequence.getLength()
Out[14]:
Node Methods:
http://xaldev.sourceforge.net/javadoc/xal/smf/AcceleratorNode.html
In [15]:
MyQuad = Sequence.getNodeWithId('QP1')
In [16]:
MyQuad.getField()
In []:
MyQuad.field
In []:
MyQuad.isMagnet()
In []:
Sequence.getNodeWithId('BPM1').isMagnet()
In []:
print MyQuad.handles
In []:
MyQuad.handles.toString()
In []:
MyQuad.getChannel('fieldSet').getId()
In []:
MyQuad.getChannel('fieldRB').getId()
It is time to run a simulation
Setup generic parameters for an algorithm of simulation
http://xaldev.sourceforge.net/javadoc/xal/model/alg/EnvTrackerAdapt.html
In [17]:
_AlgorithFactory = JClass("xal.sim.scenario.AlgorithmFactory")
In [18]:
EnvelopeTracker = _AlgorithFactory.createEnvTrackerAdapt(Sequence)
In [19]:
EnvelopeTracker.setMaxIterations(1000)
EnvelopeTracker.setAccuracyOrder(1)
EnvelopeTracker.setErrorTolerance(0.001)
Setup a probe for the simulation (Beam)
http://xaldev.sourceforge.net/javadoc/xal/model/alg/EnvTrackerAdapt.html
In [20]:
_ProbeFactory = JClass("xal.sim.scenario.ProbeFactory")
In [21]:
Probe = _ProbeFactory.getEnvelopeProbe(Sequence, EnvelopeTracker)
Setup a model (Scenario) for the simulation
http://xaldev.sourceforge.net/javadoc/xal/sim/scenario/Scenario.html
In [22]:
_Scenario = JClass("xal.sim.scenario.Scenario")
In [23]:
Model = _Scenario.newScenarioFor(Sequence)
In [24]:
Model.setProbe(Probe)
In [25]:
Model.setSynchronizationMode('LIVE')
#Model.setSynchronizationMode('DESIGN')
In [26]:
Model.resync()
Run the simulation
In [27]:
Model.run()
Get back the probe (Beam)
In [28]:
Probe = Model.getProbe()
Get the trajectory
http://xaldev.sourceforge.net/javadoc/xal/model/probe/traj/Trajectory.html
In [29]:
Trajectory = Probe.getTrajectory()
In [30]:
Positions=[]
for State in Trajectory.stateIterator():
Positions.append(State.getPosition())
In [31]:
CovarianceMatrices=[]
for State in Trajectory.stateIterator():
CovarianceMatrices.append(State.getCovarianceMatrix())
In [32]:
SigmaX=[]
SigmaY=[]
SigmaZ=[]
for CovarianceMatrix in CovarianceMatrices:
SigmaX.append(CovarianceMatrix.getSigmaX())
SigmaY.append(CovarianceMatrix.getSigmaY())
SigmaZ.append(CovarianceMatrix.getSigmaZ())
In [33]:
fig, (ax1, ax2, ax3) = subplots(ncols=3, figsize=(18,8))
ax1.set_xlabel('Position [m]')
ax1.set_ylabel('$\sigma_x$ [m]')
ax1.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
ax1.plot(Positions, SigmaX, color='r')
ax2.plot(Positions, SigmaY, color='g')
ax2.set_xlabel('Position [m]')
ax2.set_ylabel('$\sigma_y$ [m]')
ax2.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
ax3.plot(Positions, SigmaZ, color='b')
ax3.set_xlabel('Position [m]')
ax3.set_ylabel('$\sigma_z$ [m]')
ax3.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
show()
A little break
What if we want to modify one variable?
We can use EPICS
In []:
from epics import caget, caput
In []:
caget('QP1:B')
In []:
caput('QP1:FldSet', 20)
In []:
caget('QP1:B')
So why not to do it interactively?
In []:
from IPython.html.widgets import interact, interactive
In []:
def PlayWithQuad(QP1Field=-16.1578):
caput('QP1:FldSet', QP1Field) # This is not part of OpenXAL, with this command we are changing the "Hardware".
# MyQuad.setDfltField(QP1Field) # This modifies a variable in OpenXAL.
Probe = _ProbeFactory.getEnvelopeProbe(Sequence, EnvelopeTracker)
Model.setProbe(Probe)
Model.resync()
Model.run()
Probe = Model.getProbe()
Trajectory = Probe.getTrajectory()
Positions=[]
CovarianceMatrices=[]
for State in Trajectory.stateIterator():
Positions.append(State.getPosition())
CovarianceMatrices.append(State.getCovarianceMatrix())
SigmaX=[]
SigmaY=[]
SigmaZ=[]
for CovarianceMatrix in CovarianceMatrices:
SigmaX.append(CovarianceMatrix.getSigmaX())
SigmaY.append(CovarianceMatrix.getSigmaY())
SigmaZ.append(CovarianceMatrix.getSigmaZ())
fig, (ax1, ax2, ax3) = subplots(ncols=3, figsize=(18,8))
ax1.set_xlabel('Position [m]')
ax1.set_ylabel('$\sigma_x$ [m]')
ax1.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
ax1.plot(Positions, SigmaX, color='r')
ax2.plot(Positions, SigmaY, color='g')
ax2.set_xlabel('Position [m]')
ax2.set_ylabel('$\sigma_y$ [m]')
ax2.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
ax3.plot(Positions, SigmaZ, color='b')
ax3.set_xlabel('Position [m]')
ax3.set_ylabel('$\sigma_z$ [m]')
ax3.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
show()
In []:
interactive(PlayWithQuad, QP1Field=(-20.0, 20.0))
In []:
def PlayWithCav(CAV1Amp=0.125):
caput('GAP1:AmpCtl', CAV1Amp)
Probe = _ProbeFactory.getEnvelopeProbe(Sequence, EnvelopeTracker)
Model.setProbe(Probe)
Model.resync()
Model.run()
Probe = Model.getProbe()
Trajectory = Probe.getTrajectory()
Positions=[]
CovarianceMatrices=[]
for State in Trajectory.stateIterator():
Positions.append(State.getPosition())
CovarianceMatrices.append(State.getCovarianceMatrix())
SigmaX=[]
SigmaY=[]
SigmaZ=[]
for CovarianceMatrix in CovarianceMatrices:
SigmaX.append(CovarianceMatrix.getSigmaX())
SigmaY.append(CovarianceMatrix.getSigmaY())
SigmaZ.append(CovarianceMatrix.getSigmaZ())
fig, (ax1, ax2, ax3) = subplots(ncols=3, figsize=(18,8))
ax1.set_xlabel('Position [m]')
ax1.set_ylabel('$\sigma_x$ [m]')
ax1.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
ax1.plot(Positions, SigmaX, color='r')
ax2.plot(Positions, SigmaY, color='g')
ax2.set_xlabel('Position [m]')
ax2.set_ylabel('$\sigma_y$ [m]')
ax2.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
ax3.plot(Positions, SigmaZ, color='b')
ax3.set_xlabel('Position [m]')
ax3.set_ylabel('$\sigma_z$ [m]')
ax3.ticklabel_format(style='sci', axis='y', scilimits=(0,0))
show()
In []:
interactive(PlayWithCav, CAV1Amp=(0, 10.0))
In []:
#shutdownJVM()