def actionPerformed(self,actionEvent):
fc = JFileChooser(constants_lib.const_path_dict["OPENXAL_XML_ACC_FILES_DIRS_PATH"])
fc.setDialogTitle("Read Main Accelerator Config File ...")
fc.setApproveButtonText("Open")
fl_filter = FileNameExtensionFilter("Accelerator Conf. File",["xal",])
fc.setFileFilter(fl_filter)
returnVal = fc.showOpenDialog(self.linac_setup_controller.linac_wizard_document.linac_wizard_window.frame)
if(returnVal == JFileChooser.APPROVE_OPTION):
fl_in = fc.getSelectedFile()
file_name = fl_in.getName()
accl_new = XMLDataManager.acceleratorWithPath(file_name)
cav_name_node_new_dict = self.linac_setup_controller.getCavNameNodeDict(accl_new)
accl = self.linac_setup_controller.linac_wizard_document.getAccl()
cav_name_node_dict = self.linac_setup_controller.getCavNameNodeDict(accl)
for cav_name in cav_name_node_new_dict.keys():
if(cav_name_node_dict.has_key(cav_name)):
cav_new = cav_name_node_new_dict[cav_name]
cav = cav_name_node_dict[cav_name]
cav.setDfltCavAmp(cav_new.getDfltCavAmp())
cav.setDfltCavPhase(cav_new.getDfltCavPhase())
def actionPerformed(self,actionEvent):
fc = JFileChooser(constants_lib.const_path_dict["OPENXAL_XML_ACC_FILES_DIRS_PATH"])
fc.setDialogTitle("Read Main Accelerator Config File ...")
fc.setApproveButtonText("Open")
fl_filter = FileNameExtensionFilter("Accelerator Conf. File",["xal",])
fc.setFileFilter(fl_filter)
returnVal = fc.showOpenDialog(self.linac_setup_controller.linac_wizard_document.linac_wizard_window.frame)
if(returnVal == JFileChooser.APPROVE_OPTION):
fl_in = fc.getSelectedFile()
file_name = fl_in.getName()
accl_new = XMLDataManager.acceleratorWithPath(file_name)
cav_name_node_new_dict = self.linac_setup_controller.getCavNameNodeDict(accl_new)
accl = self.linac_setup_controller.linac_wizard_document.getAccl()
cav_name_node_dict = self.linac_setup_controller.getCavNameNodeDict(accl)
for cav_name in cav_name_node_new_dict.keys():
if(cav_name_node_dict.has_key(cav_name)):
cav_new = cav_name_node_new_dict[cav_name]
cav = cav_name_node_dict[cav_name]
cav.setDfltCavAmp(cav_new.getDfltCavAmp())
cav.setDfltCavPhase(cav_new.getDfltCavPhase())
from xal.extension.solver import Variable
from xal.extension.solver import Stopper
from xal.extension.solver import SolveStopperFactory
from xal.extension.solver import ProblemFactory
from xal.extension.solver import Solver
from xal.extension.solver import Problem
from xal.extension.solver.algorithm import SimplexSearchAlgorithm
from xal.extension.solver.algorithm import RandomShrinkSearch
from xal.extension.solver.hint import Hint
from xal.extension.solver.hint import InitialDelta
#===============================================================
# MAIN PROGRAM
#===============================================================
# read the accelerator & make the sequence
accl = XMLDataManager.loadDefaultAccelerator()
"""
#====== Let's construct accelerator ===========
ccl1 = accl.getSequence("CCL1")
ccl2 = accl.getSequence("CCL2")
ccl3 = accl.getSequence("CCL3")
ccl4 = accl.getSequence("CCL4")
#+++++++++++++++++++++++++++++++++++++++++++++
accSeq = AcceleratorSeqCombo("SEQUENCE", [ccl1,ccl2,ccl3,ccl4])
"""
DTL1 = accl.getSequence("DTL1")
DTL2 = accl.getSequence("DTL2")
DTL3 = accl.getSequence("DTL3")
DTL4 = accl.getSequence("DTL4")
from xal.smf.data import XMLDataManager
from xal.sim.scenario import Scenario, AlgorithmFactory, ProbeFactory
accl = XMLDataManager.acceleratorWithPath(
'/home/student/lib/openxal/site/optics/production/main.xal')
seq = accl.findSequence("MEBT")
scenario = Scenario.newScenarioFor(seq)
tracker = AlgorithmFactory.createEnvTrackerAdapt(seq)
probe = ProbeFactory.getEnvelopeProbe(seq, tracker)
scenario.setProbe(probe)
scenario.run()
traj = scenario.getProbe().getTrajectory()
xsz = map(
lambda s: (s.getPosition(), s.twissParameters()[0].getEnvelopeRadius()),
traj.iterator())
print(list(xsz))