def writeDataToXML(self, root_da):
cav_long_twiss_bucket_da = root_da.createChild(
"Cavity_Long_Twiss_Bucket")
cav_long_twiss_bucket_da.setValue("cav", self.cav_wrapper.alias)
cav_long_twiss_bucket_da.setValue("isReady", self.isReady)
cav_long_twiss_bucket_da.setValue("amp_avg_fit_err",
self.fit_bpm_amp_avg_err)
twiss_arr_da = cav_long_twiss_bucket_da.createChild("Twiss_arr")
twiss_arr = self.twiss_arr
[alphaX, betaX, emittX] = [
twiss_arr[0].getAlpha(), twiss_arr[0].getBeta(),
twiss_arr[0].getEmittance()
]
[alphaY, betaY, emittY] = [
twiss_arr[1].getAlpha(), twiss_arr[1].getBeta(),
twiss_arr[1].getEmittance()
]
[alphaZ, betaZ, emittZ] = [
twiss_arr[2].getAlpha(), twiss_arr[2].getBeta(),
twiss_arr[2].getEmittance()
]
twiss_arr_da.setValue("alphaX", alphaX)
twiss_arr_da.setValue("betaX", betaX)
twiss_arr_da.setValue("emittX", emittX)
twiss_arr_da.setValue("alphaY", alphaY)
twiss_arr_da.setValue("betaY", betaY)
twiss_arr_da.setValue("emittY", emittY)
twiss_arr_da.setValue("alphaZ", alphaZ)
twiss_arr_da.setValue("betaZ", betaZ)
twiss_arr_da.setValue("emittZ", emittZ)
twiss_arr_da.setValue("alphaZ_err", self.long_Twiss_arr_err[0])
twiss_arr_da.setValue("betaZ_err", self.long_Twiss_arr_err[1])
twiss_arr_da.setValue("emittZ_err", self.long_Twiss_arr_err[2])
twiss_arr_da.setValue("alphaZ_steps", self.long_Twiss_arr_steps[0])
twiss_arr_da.setValue("betaZ_steps", self.long_Twiss_arr_steps[1])
twiss_arr_da.setValue("emittZ_steps", self.long_Twiss_arr_steps[2])
twiss_fit_da = cav_long_twiss_bucket_da.createChild("Twiss_fit")
twiss_fit_da.setValue("alphaZ", self.long_Twiss_fit.getAlpha())
twiss_fit_da.setValue("betaZ", self.long_Twiss_fit.getBeta())
twiss_fit_da.setValue("emittZ", self.long_Twiss_fit.getEmittance())
twiss_matrix_da = cav_long_twiss_bucket_da.createChild("Twiss_matrix")
twiss_matrix_da.setValue("alphaZ", self.long_Twiss_matrix.getAlpha())
twiss_matrix_da.setValue("betaZ", self.long_Twiss_matrix.getBeta())
twiss_matrix_da.setValue("emittZ",
self.long_Twiss_matrix.getEmittance())
twiss_matrix_da.setValue("z2_err", self.long_Twiss_z2_zzp_zp2_err[0])
twiss_matrix_da.setValue("z_zp_err", self.long_Twiss_z2_zzp_zp2_err[1])
twiss_matrix_da.setValue("z2_err", self.long_Twiss_z2_zzp_zp2_err[2])
bpm_amps_da = cav_long_twiss_bucket_da.createChild("BPM_Amplitudes")
for [bpm_wrapper, amp_plotTh] in self.bpm_amp_plotTh_arr:
if (amp_plotTh.getNumbOfPoints() > 0):
bpm_amp_da = bpm_amps_da.createChild("BPM")
bpm_amp_da.setValue("BPM", bpm_wrapper.alias)
dumpGraphDataToDA(amp_plotTh, bpm_amp_da, "BPM_AMPL_GD",
"%10.5f", "%10.5f")
def writeDataToXML(self,root_da):
cav_long_twiss_bucket_da = root_da.createChild("Cavity_Long_Twiss_Bucket")
cav_long_twiss_bucket_da.setValue("cav",self.cav_wrapper.alias)
cav_long_twiss_bucket_da.setValue("isReady",self.isReady)
cav_long_twiss_bucket_da.setValue("amp_avg_fit_err",self.fit_bpm_amp_avg_err)
twiss_arr_da = cav_long_twiss_bucket_da.createChild("Twiss_arr")
twiss_arr = self.twiss_arr
[alphaX,betaX,emittX] = [twiss_arr[0].getAlpha(),twiss_arr[0].getBeta(),twiss_arr[0].getEmittance()]
[alphaY,betaY,emittY] = [twiss_arr[1].getAlpha(),twiss_arr[1].getBeta(),twiss_arr[1].getEmittance()]
[alphaZ,betaZ,emittZ] = [twiss_arr[2].getAlpha(),twiss_arr[2].getBeta(),twiss_arr[2].getEmittance()]
twiss_arr_da.setValue("alphaX",alphaX)
twiss_arr_da.setValue("betaX",betaX)
twiss_arr_da.setValue("emittX",emittX)
twiss_arr_da.setValue("alphaY",alphaY)
twiss_arr_da.setValue("betaY",betaY)
twiss_arr_da.setValue("emittY",emittY)
twiss_arr_da.setValue("alphaZ",alphaZ)
twiss_arr_da.setValue("betaZ",betaZ)
twiss_arr_da.setValue("emittZ",emittZ)
twiss_arr_da.setValue("alphaZ_err",self.long_Twiss_arr_err[0])
twiss_arr_da.setValue("betaZ_err",self.long_Twiss_arr_err[1])
twiss_arr_da.setValue("emittZ_err",self.long_Twiss_arr_err[2])
twiss_arr_da.setValue("alphaZ_steps",self.long_Twiss_arr_steps[0])
twiss_arr_da.setValue("betaZ_steps",self.long_Twiss_arr_steps[1])
twiss_arr_da.setValue("emittZ_steps",self.long_Twiss_arr_steps[2])
twiss_fit_da = cav_long_twiss_bucket_da.createChild("Twiss_fit")
twiss_fit_da.setValue("alphaZ",self.long_Twiss_fit.getAlpha())
twiss_fit_da.setValue("betaZ",self.long_Twiss_fit.getBeta())
twiss_fit_da.setValue("emittZ",self.long_Twiss_fit.getEmittance())
twiss_matrix_da = cav_long_twiss_bucket_da.createChild("Twiss_matrix")
twiss_matrix_da.setValue("alphaZ",self.long_Twiss_matrix.getAlpha())
twiss_matrix_da.setValue("betaZ",self.long_Twiss_matrix.getBeta())
twiss_matrix_da.setValue("emittZ",self.long_Twiss_matrix.getEmittance())
twiss_matrix_da.setValue("z2_err",self.long_Twiss_z2_zzp_zp2_err[0])
twiss_matrix_da.setValue("z_zp_err",self.long_Twiss_z2_zzp_zp2_err[1])
twiss_matrix_da.setValue("z2_err",self.long_Twiss_z2_zzp_zp2_err[2])
bpm_amps_da = cav_long_twiss_bucket_da.createChild("BPM_Amplitudes")
for [bpm_wrapper,amp_plotTh] in self.bpm_amp_plotTh_arr:
if(amp_plotTh.getNumbOfPoints() > 0):
bpm_amp_da = bpm_amps_da.createChild("BPM")
bpm_amp_da.setValue("BPM",bpm_wrapper.alias)
dumpGraphDataToDA(amp_plotTh,bpm_amp_da,"BPM_AMPL_GD","%10.5f","%10.5f")
def writeDataToXML(self,root_da):
accStates_Arr_da = root_da.createChild("Accelerator_States_Arr_Analysis_Panel")
for accState in self.accState_arr:
accState_da = accStates_Arr_da.createChild("Accelerator_State")
accState_da.setValue("isOn", accState.isOn)
tr_twiss_analysis_controller = self.linac_wizard_document.tr_twiss_analysis_controller
dict_panel = tr_twiss_analysis_controller.dict_panel
button_ind = -1
if(dict_panel.gaussButton.isSelected()): button_ind = 0
if(dict_panel.custom_gaussButton.isSelected()): button_ind = 1
if(dict_panel.custom_rmsButton.isSelected()): button_ind = 2
accState_da.setValue("fit_button_ind", button_ind)
dumpGraphDataToDA(accState.gd_model_hor,accState_da,"gd_model_hor","%12.5f","%12.5f")
dumpGraphDataToDA(accState.gd_model_ver,accState_da,"gd_model_ver","%12.5f","%12.5f")
dumpGraphDataToDA(accState.gd_model_lon,accState_da,"gd_model_lon","%12.5f","%12.5f")
size_record_da = accState_da.createChild("HOR_RECORDS")
for ws_lw_size_record in accState.size_hor_record_arr:
ws_lw_size_record_da = size_record_da.createChild(ws_lw_size_record.ws_node.getId())
ws_lw_size_record_da.setValue("isOn", ws_lw_size_record.isOn)
ws_lw_size_record_da.setValue("index", ws_lw_size_record.index)
size_record_da = accState_da.createChild("VER_RECORDS")
for ws_lw_size_record in accState.size_ver_record_arr:
ws_lw_size_record_da = size_record_da.createChild(ws_lw_size_record.ws_node.getId())
ws_lw_size_record_da.setValue("isOn", ws_lw_size_record.isOn)
ws_lw_size_record_da.setValue("index", ws_lw_size_record.index)
def writeDataToXML(self, root_da):
cav_wrapper_da = root_da.createChild(self.alias)
cav_wrapper_da.setValue("cav", self.cav.getId())
bpm0_name = "null"
bpm1_name = "null"
if (self.bpm_wrapper0 != null): bpm0_name = self.bpm_wrapper0.alias
if (self.bpm_wrapper1 != null): bpm1_name = self.bpm_wrapper1.alias
cav_wrapper_da.setValue("bpm0", bpm0_name)
cav_wrapper_da.setValue("bpm1", bpm1_name)
cav_wrapper_da.setValue("isGood", self.isGood)
cav_wrapper_da.setValue("isMeasured", self.isMeasured)
cav_wrapper_da.setValue("isAnalyzed", self.isAnalyzed)
#--------------------------------
cav_wrapper_params_da = cav_wrapper_da.createChild("Params")
cav_wrapper_params_da.setValue("initDesignAmp", self.initDesignAmp)
cav_wrapper_params_da.setValue("initDesignPhase", self.initDesignPhase)
cav_wrapper_params_da.setValue("designAmp", self.designAmp)
cav_wrapper_params_da.setValue("designPhase", self.designPhase)
cav_wrapper_params_da.setValue("avg_gap_phase", self.avg_gap_phase)
cav_wrapper_params_da.setValue("initLiveAmp", self.initLiveAmp)
cav_wrapper_params_da.setValue("initLivePhase", self.initLivePhase)
cav_wrapper_params_da.setValue("livePhase", self.livePhase)
cav_wrapper_params_da.setValue("scanPhaseShift", self.scanPhaseShift)
cav_wrapper_params_da.setValue("phase_scan_harm_err",
self.phase_scan_harm_err)
cav_wrapper_params_da.setValue("phase_scan_harm_amp",
self.phase_scan_harm_amp)
cav_wrapper_params_da.setValue("eKin_in_guess", self.eKin_in_guess)
cav_wrapper_params_da.setValue("eKin_out_guess", self.eKin_out_guess)
cav_wrapper_params_da.setValue("eKin_in", self.eKin_in)
cav_wrapper_params_da.setValue("eKin_out", self.eKin_out)
cav_wrapper_params_da.setValue("eKin_err", self.eKin_err)
cav_wrapper_params_da.setValue("bpm_eKin_out", self.bpm_eKin_out)
cav_wrapper_params_da.setValue("model_eKin_out", self.model_eKin_out)
cav_wrapper_params_da.setValue("real_scanPhaseShift",
self.real_scanPhaseShift)
#--------- write the cavity rescale data
self.rescaleBacket.writeDataToXML(cav_wrapper_da)
#---------------------------------
cav_wrapper_phase_harm_da = cav_wrapper_da.createChild(
"Phase_Harm_Func")
cav_wrapper_phase_harm_da.setValue(
"params_arr", self.phase_scan_harm_funcion.getTxtParamArr())
cav_wrapper_energy_harm_da = cav_wrapper_da.createChild(
"Eenergy_guess_Harm_Func")
cav_wrapper_energy_harm_da.setValue(
"params_arr", self.energy_guess_harm_funcion.getTxtParamArr())
#----------------------------------
dumpGraphDataToDA(self.phaseDiffPlot, cav_wrapper_da, "Phase_Diff_GD",
"%8.3f", "%8.3f")
dumpGraphDataToDA(self.phaseDiffPlotTh, cav_wrapper_da,
"Phase_Diff_Fit_GD", "%8.3f", "%8.3f")
dumpGraphDataToDA(self.eKinOutPlot, cav_wrapper_da, "Ekin_Out_GD",
"%8.3f", "%10.4f")
dumpGraphDataToDA(self.eKinOutPlotTh, cav_wrapper_da,
"Ekin_Out_Fit_GD", "%8.3f", "%10.4f")
#------------------------------------
cav_wrapper_bpm_list_da = cav_wrapper_da.createChild("bpm_arr")
txt = ""
for bpm_wrapper in self.bpm_wrappers:
txt = txt + " " + bpm_wrapper.alias
cav_wrapper_bpm_list_da.setValue("bpm_wrappers", txt)
cav_wrapper_bpm_list_da = cav_wrapper_da.createChild(
"bpm_use_in_phase_analysis_arr")
txt = ""
for bool_val in self.bpm_wrappers_useInPhaseAnalysis:
val = "0"
if (bool_val): val = "1"
txt = txt + " " + val
cav_wrapper_bpm_list_da.setValue("use_arr", txt)
cav_wrapper_bpm_list_da = cav_wrapper_da.createChild(
"bpm_use_in_amp_analysis_arr")
txt = ""
for bool_val in self.bpm_wrappers_useInAmpBPMs:
val = "0"
if (bool_val): val = "1"
txt = txt + " " + val
cav_wrapper_bpm_list_da.setValue("use_arr", txt)
#---------------------------------------------
cav_wrapper_scan_data_da = cav_wrapper_da.createChild("scan_data")
for bpm_wrapper in self.bpm_wrappers:
if (self.bpm_amp_phase_dict.has_key(bpm_wrapper)):
(graphDataAmp,
graphDataPhase) = self.bpm_amp_phase_dict[bpm_wrapper]
cav_wrapper_scan_data_bpm_da = cav_wrapper_scan_data_da.createChild(
bpm_wrapper.alias)
dumpGraphDataToDA(graphDataPhase, cav_wrapper_scan_data_bpm_da,
"phase", "%8.3f", "%8.3f")
dumpGraphDataToDA(graphDataAmp, cav_wrapper_scan_data_bpm_da,
"amplitude", "%8.3f", "%10.3g")
def writeDataToXML(self,root_da):
cav_wrapper_da = root_da.createChild(self.alias)
cav_wrapper_da.setValue("cav",self.cav.getId())
bpm0_name = "null"
bpm1_name = "null"
if(self.bpm_wrapper0 != null): bpm0_name = self.bpm_wrapper0.alias
if(self.bpm_wrapper1 != null): bpm1_name = self.bpm_wrapper1.alias
cav_wrapper_da.setValue("bpm0",bpm0_name)
cav_wrapper_da.setValue("bpm1",bpm1_name)
cav_wrapper_da.setValue("isGood",self.isGood )
cav_wrapper_da.setValue("isMeasured",self.isMeasured)
cav_wrapper_da.setValue("isAnalyzed",self.isAnalyzed)
#--------------------------------
cav_wrapper_params_da = cav_wrapper_da.createChild("Params")
cav_wrapper_params_da.setValue("initDesignAmp",self.initDesignAmp)
cav_wrapper_params_da.setValue("initDesignPhase",self.initDesignPhase)
cav_wrapper_params_da.setValue("designAmp",self.designAmp)
cav_wrapper_params_da.setValue("designPhase",self.designPhase)
cav_wrapper_params_da.setValue("avg_gap_phase",self.avg_gap_phase)
cav_wrapper_params_da.setValue("initLiveAmp",self.initLiveAmp)
cav_wrapper_params_da.setValue("initLivePhase",self.initLivePhase)
cav_wrapper_params_da.setValue("livePhase",self.livePhase)
cav_wrapper_params_da.setValue("scanPhaseShift",self.scanPhaseShift)
cav_wrapper_params_da.setValue("phase_scan_harm_err",self.phase_scan_harm_err)
cav_wrapper_params_da.setValue("phase_scan_harm_amp",self.phase_scan_harm_amp)
cav_wrapper_params_da.setValue("eKin_in_guess",self.eKin_in_guess)
cav_wrapper_params_da.setValue("eKin_out_guess",self.eKin_out_guess)
cav_wrapper_params_da.setValue("eKin_in",self.eKin_in)
cav_wrapper_params_da.setValue("eKin_out",self.eKin_out)
cav_wrapper_params_da.setValue("eKin_err",self.eKin_err)
cav_wrapper_params_da.setValue("bpm_eKin_out",self.bpm_eKin_out)
cav_wrapper_params_da.setValue("model_eKin_out",self.model_eKin_out)
cav_wrapper_params_da.setValue("real_scanPhaseShift",self.real_scanPhaseShift)
#--------- write the cavity rescale data
self.rescaleBacket.writeDataToXML(cav_wrapper_da)
#---------------------------------
cav_wrapper_phase_harm_da = cav_wrapper_da.createChild("Phase_Harm_Func")
cav_wrapper_phase_harm_da.setValue("params_arr",self.phase_scan_harm_funcion.getTxtParamArr())
cav_wrapper_energy_harm_da = cav_wrapper_da.createChild("Eenergy_guess_Harm_Func")
cav_wrapper_energy_harm_da.setValue("params_arr",self.energy_guess_harm_funcion.getTxtParamArr())
#----------------------------------
dumpGraphDataToDA(self.phaseDiffPlot,cav_wrapper_da,"Phase_Diff_GD","%8.3f","%8.3f")
dumpGraphDataToDA(self.phaseDiffPlotTh,cav_wrapper_da,"Phase_Diff_Fit_GD","%8.3f","%8.3f")
dumpGraphDataToDA(self.eKinOutPlot,cav_wrapper_da,"Ekin_Out_GD","%8.3f","%10.4f")
dumpGraphDataToDA(self.eKinOutPlotTh,cav_wrapper_da,"Ekin_Out_Fit_GD","%8.3f","%10.4f")
#------------------------------------
cav_wrapper_bpm_list_da = cav_wrapper_da.createChild("bpm_arr")
txt = ""
for bpm_wrapper in self.bpm_wrappers:
txt = txt +" "+ bpm_wrapper.alias
cav_wrapper_bpm_list_da.setValue("bpm_wrappers",txt)
cav_wrapper_bpm_list_da = cav_wrapper_da.createChild("bpm_use_in_phase_analysis_arr")
txt = ""
for bool_val in self.bpm_wrappers_useInPhaseAnalysis:
val = "0"
if(bool_val): val = "1"
txt = txt +" "+ val
cav_wrapper_bpm_list_da.setValue("use_arr",txt)
cav_wrapper_bpm_list_da = cav_wrapper_da.createChild("bpm_use_in_amp_analysis_arr")
txt = ""
for bool_val in self.bpm_wrappers_useInAmpBPMs:
val = "0"
if(bool_val): val = "1"
txt = txt +" "+ val
cav_wrapper_bpm_list_da.setValue("use_arr",txt)
#---------------------------------------------
cav_wrapper_scan_data_da = cav_wrapper_da.createChild("scan_data")
for bpm_wrapper in self.bpm_wrappers:
if(self.bpm_amp_phase_dict.has_key(bpm_wrapper)):
(graphDataAmp,graphDataPhase) = self.bpm_amp_phase_dict[bpm_wrapper]
cav_wrapper_scan_data_bpm_da = cav_wrapper_scan_data_da.createChild(bpm_wrapper.alias)
dumpGraphDataToDA(graphDataPhase,cav_wrapper_scan_data_bpm_da,"phase","%8.3f","%8.3f")
dumpGraphDataToDA(graphDataAmp,cav_wrapper_scan_data_bpm_da,"amplitude","%8.3f","%10.3g")
