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")