SwissFEL-Simulation/OpticsTools
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Basic Tracking done

Browse Source
This commit is contained in:
reiche
2025-05-05 17:52:45 +02:00
parent 93527fc7e5
commit b4c25511d1
10 changed files with 100017 additions and 278 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
OpticsPlot.py
View File

@@ -28,6 +28,7 @@ class OpticsPlot(QtWidgets.QMainWindow, Ui_OpticsPlotGUI):
self.initmpl(self.mplvl, self.mplwindow)
self.twiss=None
self.energy=None
# all action for optics plotting
self.PBetax.toggled.connect(self.doplot)
@@ -37,12 +38,14 @@ class OpticsPlot(QtWidgets.QMainWindow, Ui_OpticsPlotGUI):
self.PEtax.toggled.connect(self.doplot)
self.PEtay.toggled.connect(self.doplot)
self.PR56.toggled.connect(self.doplot)
self.PEnergy.toggled.connect(self.doplot)
self.PStart.editingFinished.connect(self.doplot)
self.PEnd.editingFinished.connect(self.doplot)
def newData(self,twiss):
def newData(self,twiss,energy):
self.twiss=twiss
self.energy=energy
self.updateOpticsTable()
self.doplot()
@@ -52,10 +55,10 @@ class OpticsPlot(QtWidgets.QMainWindow, Ui_OpticsPlotGUI):
return
nrow = len(self.twiss.s)
ncol = 9
ncol = 10
self.UITwissValues.setColumnCount(ncol)
self.UITwissValues.setRowCount(nrow)
self.UITwissValues.setHorizontalHeaderLabels(['Name','s','betax','betay','alphax','alphay','etax','etay','r56'])
self.UITwissValues.setHorizontalHeaderLabels(['Name','s','betax','betay','alphax','alphay','etax','etay','r56','energy'])
for i in range(nrow):
self.UITwissValues.setItem(i, 0, QtWidgets.QTableWidgetItem(self.twiss.name[i].split(':')[0]))
self.UITwissValues.setItem(i, 1, QtWidgets.QTableWidgetItem('%10.6f' % self.twiss.s[i]))
@@ -66,6 +69,7 @@ class OpticsPlot(QtWidgets.QMainWindow, Ui_OpticsPlotGUI):
self.UITwissValues.setItem(i, 6, QtWidgets.QTableWidgetItem('%10.6f' % self.twiss.dx[i]))
self.UITwissValues.setItem(i, 7, QtWidgets.QTableWidgetItem('%10.6f' % self.twiss.dy[i]))
self.UITwissValues.setItem(i, 8, QtWidgets.QTableWidgetItem('%10.6f' % self.twiss.re56[i]))
self.UITwissValues.setItem(i, 9, QtWidgets.QTableWidgetItem('%10.6f' % self.energy[i]))
self.UITwissValues.resizeColumnsToContents()
self.UITwissValues.verticalHeader().hide()
self.UITwissValues.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)
@@ -91,7 +95,7 @@ class OpticsPlot(QtWidgets.QMainWindow, Ui_OpticsPlotGUI):
filt['DX']=self.PEtax.isChecked()
filt['DY']=self.PEtay.isChecked()
filt['RE56']=self.PR56.isChecked()
# filt['Energy']=self.PEnergy.isChecked()
filt['Energy']=self.PEnergy.isChecked()
s = self.twiss.s
i1 = np.argmin(np.abs(s - z0))
@@ -122,9 +126,9 @@ class OpticsPlot(QtWidgets.QMainWindow, Ui_OpticsPlotGUI):
if filt['RE56']:
self.plotSingle(s[i1:i2], self.twiss.re56[i1:i2], (0, 0, 0, 1), r'$R_{56}$')
ylabel = ylabel + r'$R_{56}$ (m), '
# if filt['Energy']:
# self.plotSingle(s[i1:i2], data['Energy'][i1:i2], (0, 1, 0, 1), r'$E$')
# ylabel = ylabel + r'$E$ (MeV), '
if filt['Energy']:
self.plotSingle(s[i1:i2], self.energy[i1:i2], (0, 1, 0, 1), r'$E$')
ylabel = ylabel + r'$E$ (MeV), '
if len(ylabel) < 3:
self.canvas.draw()