def plot(self):
if self.plotDialog is None:
from PyMca5.PyMcaGui.plotting.PlotWindow import PlotWindow
dialog = qt.QDialog(self)
dialog.mainLayout = qt.QVBoxLayout(dialog)
dialog.mainLayout.setContentsMargins(0, 0, 0, 0)
dialog.mainLayout.setSpacing(0)
dialog.plotWidget = PlotWindow(dialog,
newplot=False,
fit=False,
plugins=False,
control=True,
position=True)
dialog.plotWidget.setDefaultPlotLines(True)
dialog.plotWidget.setDefaultPlotPoints(True)
dialog.plotWidget.setDataMargins(0.05, 0.05, 0.05, 0.05)
dialog.mainLayout.addWidget(dialog.plotWidget)
self.plotDialog = dialog
legend = self._transmissionTable["name"]
if legend == "":
legend = None
x = self._transmissionTable["energy"]
y = self._transmissionTable["transmission"]
comment = self._transmissionTable["comment"]
self.plotDialog.plotWidget.addCurve(x,
y,
legend=legend,
xlabel="Energy (keV)",
ylabel="Transmission",
replot=True,
replace=True)
self.plotDialog.plotWidget.setGraphTitle(comment)
def __init__(self, parent=None, initdict = None):
qt.QWidget.__init__(self, parent)
self.l = qt.QVBoxLayout(self)
self.l.setContentsMargins(0, 0, 0, 0)
self.l.setSpacing(0)
self.tubeWidget = TubeWidget(self, initdict = initdict)
self.setParameters = self.tubeWidget.setParameters
self.getParameters = self.tubeWidget.getParameters
label = qt.QLabel(self)
hbox = qt.QWidget(self)
hboxl = qt.QHBoxLayout(hbox)
hboxl.setContentsMargins(0, 0, 0, 0)
hboxl.setSpacing(0)
self.plotButton = qt.QPushButton(hbox)
self.plotButton.setText("Plot Continuum")
self.exportButton = qt.QPushButton(hbox)
self.exportButton.setText("Export to Fit")
#grid.addWidget(self.plotButton, 7, 1)
#grid.addWidget(self.exportButton, 7, 3)
hboxl.addWidget(self.plotButton)
hboxl.addWidget(self.exportButton)
self.l.addWidget(self.tubeWidget)
f = label.font()
f.setItalic(1)
label.setFont(f)
label.setAlignment(qt.Qt.AlignRight)
label.setText("H. Ebel, X-Ray Spectrometry 28 (1999) 255-266 ")
self.l.addWidget(label)
self.l.addWidget(hbox)
self.graph = PlotWindow(self, backend=None)
self.l.addWidget(self.graph)
self.graph.setGraphXLabel("Energy (keV)")
self.graph.setGraphYLabel("photons/sr/mA/keV/s")
self.plotButton.clicked.connect(self.plot)
self.exportButton.clicked.connect(self._export)
def __init__(self, parent=None, initdict = None):
qt.QWidget.__init__(self, parent)
self.l = qt.QVBoxLayout(self)
self.l.setContentsMargins(0, 0, 0, 0)
self.l.setSpacing(0)
self.tubeWidget = TubeWidget(self, initdict = initdict)
self.setParameters = self.tubeWidget.setParameters
self.getParameters = self.tubeWidget.getParameters
label = qt.QLabel(self)
hbox = qt.QWidget(self)
hboxl = qt.QHBoxLayout(hbox)
hboxl.setContentsMargins(0, 0, 0, 0)
hboxl.setSpacing(0)
self.plotButton = qt.QPushButton(hbox)
self.plotButton.setText("Plot Continuum")
self.exportButton = qt.QPushButton(hbox)
self.exportButton.setText("Export to Fit")
#grid.addWidget(self.plotButton, 7, 1)
#grid.addWidget(self.exportButton, 7, 3)
hboxl.addWidget(self.plotButton)
hboxl.addWidget(self.exportButton)
self.l.addWidget(self.tubeWidget)
f = label.font()
f.setItalic(1)
label.setFont(f)
label.setAlignment(qt.Qt.AlignRight)
label.setText("H. Ebel, X-Ray Spectrometry 28 (1999) 255-266 ")
self.l.addWidget(label)
self.l.addWidget(hbox)
self.graph = PlotWindow(self, backend=None)
self.l.addWidget(self.graph)
self.graph.setGraphXLabel("Energy (keV)")
self.graph.setGraphYLabel("photons/sr/mA/keV/s")
self.plotButton.clicked.connect(self.plot)
self.exportButton.clicked.connect(self._export)
Options: plugin, demoServer demoClient, auto
- plugin: Start a 1D plot window with JSPON-RPC plugin available
- demoServer: HTTP server, to load JSON-RPC from URL: http://localhost:8000
- demoClient: TCP client that sends JSPN-RPC to localhost:8000
- auto: Starts demo of polling and server that runs alone
""")
sys.exit()
if sys.argv[1] in ('plugin', 'auto'):
# Create Qt main application
app = qt.QApplication([])
# Create plot window
plot = PlotWindow(roi=True)
plot.show()
# Load plugin
pluginDir = os.path.dirname(os.path.abspath(__file__))
pluginName = os.path.splitext(os.path.basename(__file__))[0]
plot.setPluginDirectoryList([pluginDir])
nbPlugins = plot.getPlugins() # Update plug-in list
assert nbPlugins >= 1
plugin = plot.pluginInstanceDict[pluginName]
if sys.argv[1] == 'auto':
# Run automated demos
serverDemo = _DemoServerModeAuto(plugin, onFinish=app.quit)
clientDemo = _DemoClientModeAuto(plugin, onFinish=serverDemo.start)
Options: plugin, demoServer demoClient, auto
- plugin: Start a 1D plot window with JSPON-RPC plugin available
- demoServer: HTTP server, to load JSON-RPC from URL: http://localhost:8000
- demoClient: TCP client that sends JSPN-RPC to localhost:8000
- auto: Starts demo of polling and server that runs alone
""")
sys.exit()
if sys.argv[1] in ('plugin', 'auto'):
# Create Qt main application
app = qt.QApplication([])
# Create plot window
plot = PlotWindow(roi=True)
plot.show()
# Load plugin
pluginDir = os.path.dirname(os.path.abspath(__file__))
pluginName = os.path.splitext(os.path.basename(__file__))[0]
plot.setPluginDirectoryList([pluginDir])
nbPlugins = plot.getPlugins() # Update plug-in list
assert nbPlugins >= 1
plugin = plot.pluginInstanceDict[pluginName]
if sys.argv[1] == 'auto':
# Run automated demos
serverDemo = _DemoServerModeAuto(plugin, onFinish=app.quit)
clientDemo = _DemoClientModeAuto(plugin, onFinish=serverDemo.start)
def calculate(self):
if ShadowCongruence.checkEmptyBeam(self.input_beam):
if ShadowCongruence.checkGoodBeam(self.input_beam):
beam_to_analize = self.input_beam._beam
if self.image_plane == 1:
new_shadow_beam = self.input_beam.duplicate(history=False)
dist = 0.0
if self.image_plane_rel_abs_position == 1: # relative
dist = self.image_plane_new_position
else: # absolute
historyItem = self.input_beam.getOEHistory(oe_number=self.input_beam._oe_number)
if historyItem is None: image_plane = 0.0
elif self.input_beam._oe_number == 0: image_plane = 0.0
else: image_plane = historyItem._shadow_oe_end._oe.T_IMAGE
dist = self.image_plane_new_position - image_plane
new_shadow_beam._beam.retrace(dist)
beam_to_analize = new_shadow_beam._beam
if self.mode==2:
center=[self.center_x, self.center_z]
else:
center=[0.0, 0.0]
if self.y_range == 1:
if self.y_range_min >= self.y_range_max:
raise Exception("Y range min cannot be greater or than Y range max")
ticket = ST.focnew(beam_to_analize, mode=self.mode, center=center)
self.focnewInfo.setText(ticket["text"])
if self.plot_canvas_x is None:
self.plot_canvas_x = PlotWindow(roi=False, control=False, position=True, plugins=False)
self.plot_canvas_x.setDefaultPlotLines(True)
self.plot_canvas_x.setActiveCurveColor(color='blue')
self.plot_canvas_x.setDrawModeEnabled(False)
self.plot_canvas_x.setZoomModeEnabled(False)
self.plot_canvas_x.toolBar.setVisible(False)
self.plot_canvas_z = PlotWindow(roi=False, control=False, position=True, plugins=False)
self.plot_canvas_z.setDefaultPlotLines(True)
self.plot_canvas_z.setActiveCurveColor(color='red')
self.plot_canvas_z.setDrawModeEnabled(False)
self.plot_canvas_z.setZoomModeEnabled(False)
self.plot_canvas_z.toolBar.setVisible(False)
self.plot_canvas_t = PlotWindow(roi=False, control=False, position=True, plugins=False)
self.plot_canvas_t.setDefaultPlotLines(True)
self.plot_canvas_t.setActiveCurveColor(color='green')
self.plot_canvas_t.setDrawModeEnabled(False)
self.plot_canvas_t.setZoomModeEnabled(False)
self.plot_canvas_t.toolBar.setVisible(False)
gridLayout = QtGui.QGridLayout()
gridLayout.addWidget(self.plot_canvas_x, 0, 0)
gridLayout.addWidget(self.plot_canvas_z, 0, 1)
gridLayout.addWidget(self.plot_canvas_t, 1, 0)
widget = QtGui.QWidget()
widget.setLayout(gridLayout)
self.image_box.layout().addWidget(widget)
if self.y_range == 0:
y = numpy.linspace(-10.0, 10.0, 1001)
else:
y = numpy.linspace(self.y_range_min, self.y_range_max, self.y_npoints)
pos = [0.25, 0.15, 0.7, 0.75]
self.plot_canvas_x.addCurve(y, 2.35*ST.focnew_scan(ticket["AX"], y)*ShadowPlot.get_factor(1, self.workspace_units_to_cm), "x (tangential)", symbol='', color="blue", replace=True) #'+', '^', ','
self.plot_canvas_x._plot.graph.ax.get_yaxis().get_major_formatter().set_useOffset(True)
self.plot_canvas_x._plot.graph.ax.get_yaxis().get_major_formatter().set_scientific(True)
self.plot_canvas_x._plot.graph.ax.set_position(pos)
self.plot_canvas_x._plot.graph.ax2.set_position(pos)
self.plot_canvas_x.setGraphXLabel("Y [" + self.workspace_units_label + "]")
self.plot_canvas_x.setGraphYLabel("2.35*<X> [$\mu$m]")
self.plot_canvas_x._plot.graph.ax.set_title("X", horizontalalignment='left')
self.plot_canvas_x.replot()
self.plot_canvas_z.addCurve(y, 2.35*ST.focnew_scan(ticket["AZ"], y)*ShadowPlot.get_factor(3, self.workspace_units_to_cm), "z (sagittal)", symbol='', color="red", replace=False) #'+', '^', ','
self.plot_canvas_z._plot.graph.ax.get_yaxis().get_major_formatter().set_useOffset(True)
self.plot_canvas_z._plot.graph.ax.get_yaxis().get_major_formatter().set_scientific(True)
self.plot_canvas_z._plot.graph.ax.set_position(pos)
self.plot_canvas_z._plot.graph.ax2.set_position(pos)
self.plot_canvas_z.setGraphXLabel("Y [" + self.workspace_units_label + "]")
self.plot_canvas_z.setGraphYLabel("2.35*<Z> [$\mu$m]")
self.plot_canvas_z._plot.graph.ax.set_title("Z", horizontalalignment='left')
self.plot_canvas_z.replot()
self.plot_canvas_t.addCurve(y, 2.35*ST.focnew_scan(ticket["AT"], y)*ShadowPlot.get_factor(1, self.workspace_units_to_cm), "combined x,z", symbol='', color="green", replace=True) #'+', '^', ','
self.plot_canvas_t._plot.graph.ax.get_yaxis().get_major_formatter().set_useOffset(True)
self.plot_canvas_t._plot.graph.ax.get_yaxis().get_major_formatter().set_scientific(True)
self.plot_canvas_t._plot.graph.ax.set_position(pos)
self.plot_canvas_t._plot.graph.ax2.set_position(pos)
self.plot_canvas_t.setGraphXLabel("Y [" + self.workspace_units_label + "]")
self.plot_canvas_t.setGraphYLabel("2.35*<X,Z> [$\mu$m]")
self.plot_canvas_t._plot.graph.ax.set_title("X,Z (Combined)", horizontalalignment='left')
self.plot_canvas_t.replot()
class FocNew(ow_automatic_element.AutomaticElement):
name = "FocNew"
description = "Shadow: FocNew"
icon = "icons/focnew.png"
maintainer = "Luca Rebuffi"
maintainer_email = "luca.rebuffi(@at@)elettra.eu"
priority = 4
category = "Data Display Tools"
keywords = ["data", "file", "load", "read"]
inputs = [("Input Beam", ShadowBeam, "setBeam")]
IMAGE_WIDTH = 860
IMAGE_HEIGHT = 675
want_main_area=1
want_control_area = 1
input_beam=None
image_plane=Setting(0)
image_plane_new_position=Setting(10.0)
image_plane_rel_abs_position=Setting(0)
mode = Setting(0)
center_x = Setting(0.0)
center_z = Setting(0.0)
y_range=Setting(0)
y_range_min=Setting(-10.0)
y_range_max=Setting(10.0)
y_npoints=Setting(1001)
plot_canvas_x=None
plot_canvas_z=None
plot_canvas_t=None
def __init__(self, show_automatic_box=True):
super().__init__()
gui.button(self.controlArea, self, "Calculate", callback=self.calculate, height=45)
general_box = oasysgui.widgetBox(self.controlArea, "General Settings", addSpace=True, orientation="vertical", width=self.CONTROL_AREA_WIDTH-8, height=220)
gui.comboBox(general_box, self, "mode", label="Mode", labelWidth=250,
items=["Center at Origin",
"Center at Baricenter",
"Define Center..."],
callback=self.set_Center, sendSelectedValue=False, orientation="horizontal")
self.center_box = oasysgui.widgetBox(general_box, "", addSpace=False, orientation="vertical", height=50)
self.center_box_empty = oasysgui.widgetBox(general_box, "", addSpace=False, orientation="vertical", height=50)
self.le_center_x = oasysgui.lineEdit(self.center_box, self, "center_x", "Center X", labelWidth=260, valueType=float, orientation="horizontal")
self.le_center_z = oasysgui.lineEdit(self.center_box, self, "center_z", "Center Z", labelWidth=260, valueType=float, orientation="horizontal")
self.set_Center()
gui.comboBox(general_box, self, "y_range", label="Y Range",labelWidth=250,
items=["<Default>",
"Set.."],
callback=self.set_YRange, sendSelectedValue=False, orientation="horizontal")
self.yrange_box = oasysgui.widgetBox(general_box, "", addSpace=False, orientation="vertical", height=100)
self.yrange_box_empty = oasysgui.widgetBox(general_box, "", addSpace=False, orientation="vertical", height=100)
self.le_y_range_min = oasysgui.lineEdit(self.yrange_box, self, "y_range_min", "Y min", labelWidth=260, valueType=float, orientation="horizontal")
self.le_y_range_max = oasysgui.lineEdit(self.yrange_box, self, "y_range_max", "Y max", labelWidth=260, valueType=float, orientation="horizontal")
oasysgui.lineEdit(self.yrange_box, self, "y_npoints", "Points", labelWidth=260, valueType=float, orientation="horizontal")
self.set_YRange()
screen_box = oasysgui.widgetBox(self.controlArea, "Screen Position Settings", addSpace=True, orientation="vertical", height=110)
self.image_plane_combo = gui.comboBox(screen_box, self, "image_plane", label="Position of the Image",
items=["On Image Plane", "Retraced"], labelWidth=260,
callback=self.set_ImagePlane, sendSelectedValue=False, orientation="horizontal")
self.image_plane_box = oasysgui.widgetBox(screen_box, "", addSpace=True, orientation="vertical", height=110)
self.image_plane_box_empty = oasysgui.widgetBox(screen_box, "", addSpace=True, orientation="vertical", height=110)
oasysgui.lineEdit(self.image_plane_box, self, "image_plane_new_position", "Image Plane new Position", labelWidth=220, valueType=float, orientation="horizontal")
gui.comboBox(self.image_plane_box, self, "image_plane_rel_abs_position", label="Position Type", labelWidth=250,
items=["Absolute", "Relative"], sendSelectedValue=False, orientation="horizontal")
self.set_ImagePlane()
gui.separator(self.controlArea, height=200)
tabs_setting = gui.tabWidget(self.mainArea)
tabs_setting.setFixedHeight(self.IMAGE_HEIGHT+5)
tabs_setting.setFixedWidth(self.IMAGE_WIDTH)
tab_info = oasysgui.createTabPage(tabs_setting, "Focnew Info")
tab_scan = oasysgui.createTabPage(tabs_setting, "Focnew Scan")
self.focnewInfo = QtGui.QTextEdit()
self.focnewInfo.setReadOnly(True)
self.focnewInfo.setMaximumHeight(self.IMAGE_HEIGHT-35)
info_box = oasysgui.widgetBox(tab_info, "", addSpace=True, orientation="horizontal", height = self.IMAGE_HEIGHT-20, width = self.IMAGE_WIDTH-20)
info_box.layout().addWidget(self.focnewInfo)
self.image_box = gui.widgetBox(tab_scan, "Scan", addSpace=True, orientation="vertical")
self.image_box.setFixedHeight(self.IMAGE_HEIGHT-30)
self.image_box.setFixedWidth(self.IMAGE_WIDTH-20)
def after_change_workspace_units(self):
label = self.le_center_x.parent().layout().itemAt(0).widget()
label.setText(label.text() + " [" + self.workspace_units_label + "]")
label = self.le_center_z.parent().layout().itemAt(0).widget()
label.setText(label.text() + " [" + self.workspace_units_label + "]")
label = self.le_y_range_min.parent().layout().itemAt(0).widget()
label.setText(label.text() + " [" + self.workspace_units_label + "]")
label = self.le_y_range_max.parent().layout().itemAt(0).widget()
label.setText(label.text() + " [" + self.workspace_units_label + "]")
def set_Center(self):
self.center_box.setVisible(self.mode == 2)
self.center_box_empty.setVisible(self.mode < 2)
def set_YRange(self):
self.yrange_box.setVisible(self.y_range == 1)
self.yrange_box_empty.setVisible(self.y_range == 0)
def set_ImagePlane(self):
self.image_plane_box.setVisible(self.image_plane==1)
self.image_plane_box_empty.setVisible(self.image_plane==0)
def setBeam(self, beam):
if ShadowCongruence.checkEmptyBeam(beam):
if ShadowCongruence.checkGoodBeam(beam):
self.input_beam = beam
if self.is_automatic_run:
self.calculate()
else:
QtGui.QMessageBox.critical(self, "Error",
"Data not displayable: No good rays or bad content",
QtGui.QMessageBox.Ok)
def calculate(self):
if ShadowCongruence.checkEmptyBeam(self.input_beam):
if ShadowCongruence.checkGoodBeam(self.input_beam):
beam_to_analize = self.input_beam._beam
if self.image_plane == 1:
new_shadow_beam = self.input_beam.duplicate(history=False)
dist = 0.0
if self.image_plane_rel_abs_position == 1: # relative
dist = self.image_plane_new_position
else: # absolute
historyItem = self.input_beam.getOEHistory(oe_number=self.input_beam._oe_number)
if historyItem is None: image_plane = 0.0
elif self.input_beam._oe_number == 0: image_plane = 0.0
else: image_plane = historyItem._shadow_oe_end._oe.T_IMAGE
dist = self.image_plane_new_position - image_plane
new_shadow_beam._beam.retrace(dist)
beam_to_analize = new_shadow_beam._beam
if self.mode==2:
center=[self.center_x, self.center_z]
else:
center=[0.0, 0.0]
if self.y_range == 1:
if self.y_range_min >= self.y_range_max:
raise Exception("Y range min cannot be greater or than Y range max")
ticket = ST.focnew(beam_to_analize, mode=self.mode, center=center)
self.focnewInfo.setText(ticket["text"])
if self.plot_canvas_x is None:
self.plot_canvas_x = PlotWindow(roi=False, control=False, position=True, plugins=False)
self.plot_canvas_x.setDefaultPlotLines(True)
self.plot_canvas_x.setActiveCurveColor(color='blue')
self.plot_canvas_x.setDrawModeEnabled(False)
self.plot_canvas_x.setZoomModeEnabled(False)
self.plot_canvas_x.toolBar.setVisible(False)
self.plot_canvas_z = PlotWindow(roi=False, control=False, position=True, plugins=False)
self.plot_canvas_z.setDefaultPlotLines(True)
self.plot_canvas_z.setActiveCurveColor(color='red')
self.plot_canvas_z.setDrawModeEnabled(False)
self.plot_canvas_z.setZoomModeEnabled(False)
self.plot_canvas_z.toolBar.setVisible(False)
self.plot_canvas_t = PlotWindow(roi=False, control=False, position=True, plugins=False)
self.plot_canvas_t.setDefaultPlotLines(True)
self.plot_canvas_t.setActiveCurveColor(color='green')
self.plot_canvas_t.setDrawModeEnabled(False)
self.plot_canvas_t.setZoomModeEnabled(False)
self.plot_canvas_t.toolBar.setVisible(False)
gridLayout = QtGui.QGridLayout()
gridLayout.addWidget(self.plot_canvas_x, 0, 0)
gridLayout.addWidget(self.plot_canvas_z, 0, 1)
gridLayout.addWidget(self.plot_canvas_t, 1, 0)
widget = QtGui.QWidget()
widget.setLayout(gridLayout)
self.image_box.layout().addWidget(widget)
if self.y_range == 0:
y = numpy.linspace(-10.0, 10.0, 1001)
else:
y = numpy.linspace(self.y_range_min, self.y_range_max, self.y_npoints)
pos = [0.25, 0.15, 0.7, 0.75]
self.plot_canvas_x.addCurve(y, 2.35*ST.focnew_scan(ticket["AX"], y)*ShadowPlot.get_factor(1, self.workspace_units_to_cm), "x (tangential)", symbol='', color="blue", replace=True) #'+', '^', ','
self.plot_canvas_x._plot.graph.ax.get_yaxis().get_major_formatter().set_useOffset(True)
self.plot_canvas_x._plot.graph.ax.get_yaxis().get_major_formatter().set_scientific(True)
self.plot_canvas_x._plot.graph.ax.set_position(pos)
self.plot_canvas_x._plot.graph.ax2.set_position(pos)
self.plot_canvas_x.setGraphXLabel("Y [" + self.workspace_units_label + "]")
self.plot_canvas_x.setGraphYLabel("2.35*<X> [$\mu$m]")
self.plot_canvas_x._plot.graph.ax.set_title("X", horizontalalignment='left')
self.plot_canvas_x.replot()
self.plot_canvas_z.addCurve(y, 2.35*ST.focnew_scan(ticket["AZ"], y)*ShadowPlot.get_factor(3, self.workspace_units_to_cm), "z (sagittal)", symbol='', color="red", replace=False) #'+', '^', ','
self.plot_canvas_z._plot.graph.ax.get_yaxis().get_major_formatter().set_useOffset(True)
self.plot_canvas_z._plot.graph.ax.get_yaxis().get_major_formatter().set_scientific(True)
self.plot_canvas_z._plot.graph.ax.set_position(pos)
self.plot_canvas_z._plot.graph.ax2.set_position(pos)
self.plot_canvas_z.setGraphXLabel("Y [" + self.workspace_units_label + "]")
self.plot_canvas_z.setGraphYLabel("2.35*<Z> [$\mu$m]")
self.plot_canvas_z._plot.graph.ax.set_title("Z", horizontalalignment='left')
self.plot_canvas_z.replot()
self.plot_canvas_t.addCurve(y, 2.35*ST.focnew_scan(ticket["AT"], y)*ShadowPlot.get_factor(1, self.workspace_units_to_cm), "combined x,z", symbol='', color="green", replace=True) #'+', '^', ','
self.plot_canvas_t._plot.graph.ax.get_yaxis().get_major_formatter().set_useOffset(True)
self.plot_canvas_t._plot.graph.ax.get_yaxis().get_major_formatter().set_scientific(True)
self.plot_canvas_t._plot.graph.ax.set_position(pos)
self.plot_canvas_t._plot.graph.ax2.set_position(pos)
self.plot_canvas_t.setGraphXLabel("Y [" + self.workspace_units_label + "]")
self.plot_canvas_t.setGraphYLabel("2.35*<X,Z> [$\mu$m]")
self.plot_canvas_t._plot.graph.ax.set_title("X,Z (Combined)", horizontalalignment='left')
self.plot_canvas_t.replot()
def writeStdOut(self, text):
cursor = self.shadow_output.textCursor()
cursor.movePosition(QtGui.QTextCursor.End)
cursor.insertText(text)
self.shadow_output.setTextCursor(cursor)
self.shadow_output.ensureCursorVisible()
class QXTube(qt.QWidget):
sigQXTubeSignal = qt.pyqtSignal(object)
def __init__(self, parent=None, initdict = None):
qt.QWidget.__init__(self, parent)
self.l = qt.QVBoxLayout(self)
self.l.setContentsMargins(0, 0, 0, 0)
self.l.setSpacing(0)
self.tubeWidget = TubeWidget(self, initdict = initdict)
self.setParameters = self.tubeWidget.setParameters
self.getParameters = self.tubeWidget.getParameters
label = qt.QLabel(self)
hbox = qt.QWidget(self)
hboxl = qt.QHBoxLayout(hbox)
hboxl.setContentsMargins(0, 0, 0, 0)
hboxl.setSpacing(0)
self.plotButton = qt.QPushButton(hbox)
self.plotButton.setText("Plot Continuum")
self.exportButton = qt.QPushButton(hbox)
self.exportButton.setText("Export to Fit")
#grid.addWidget(self.plotButton, 7, 1)
#grid.addWidget(self.exportButton, 7, 3)
hboxl.addWidget(self.plotButton)
hboxl.addWidget(self.exportButton)
self.l.addWidget(self.tubeWidget)
f = label.font()
f.setItalic(1)
label.setFont(f)
label.setAlignment(qt.Qt.AlignRight)
label.setText("H. Ebel, X-Ray Spectrometry 28 (1999) 255-266 ")
self.l.addWidget(label)
self.l.addWidget(hbox)
self.graph = PlotWindow(self, backend=None)
self.l.addWidget(self.graph)
self.graph.setGraphXLabel("Energy (keV)")
self.graph.setGraphYLabel("photons/sr/mA/keV/s")
self.plotButton.clicked.connect(self.plot)
self.exportButton.clicked.connect(self._export)
def plot(self):
d = self.tubeWidget.getParameters()
transmission = d["transmission"]
anode = d["anode"]
anodedensity = d["anodedensity"]
anodethickness = d["anodethickness"]
voltage = d["voltage"]
wele = d["window"]
wdensity = d["windowdensity"]
wthickness = d["windowthickness"]
fele = d["filter1"]
fdensity = d["filter1density"]
fthickness = d["filter1thickness"]
filterlist =[[fele, fdensity, fthickness]]
alphae = d["alphae"]
alphax = d["alphax"]
delta = d["deltaplotting"]
e = numpy.arange(1, voltage, delta)
if __name__ == "__main__":
continuumR = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=0,
targetthickness=anodethickness,
filterlist=filterlist)
continuumT = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=1,
targetthickness=anodethickness,
filterlist=filterlist)
self.graph.addCurve(e, continuumR, "continuumR", replot=False)
self.graph.addCurve(e, continuumT, "continuumT", replot=False)
else:
continuum = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
self.graph.addCurve(e, continuum, "continuum", replot=False)
self.graph.setActiveCurve("continuum")
self.graph.resetZoom()
self.graph.replot()
def _export(self):
d = self.tubeWidget.getParameters()
transmission = d["transmission"]
anode = d["anode"]
anodedensity = d["anodedensity"]
anodethickness = d["anodethickness"]
voltage = d["voltage"]
wele = d["window"]
wdensity = d["windowdensity"]
wthickness = d["windowthickness"]
fele = d["filter1"]
fdensity = d["filter1density"]
fthickness = d["filter1thickness"]
filterlist =[[fele, fdensity, fthickness]]
alphae = d["alphae"]
alphax = d["alphax"]
delta = d["deltaplotting"]
e = numpy.arange(1, voltage, delta)
d["event"] = "TubeUpdated"
d["energyplot"] = e
d["continuum"] = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
fllines = XRayTubeEbel.characteristicEbel([anode, anodedensity, anodethickness],
voltage,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
d["characteristic"] = fllines
energy, energyweight, energyscatter = XRayTubeEbel.generateLists(
[anode, anodedensity, anodethickness],
voltage,
window=[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
d["energylist"] = energy
d["weightlist"] = energyweight
d["scatterlist"] = energyscatter
d["flaglist"] = numpy.ones(len(energy))
self.sigQXTubeSignal.emit(d)
class QXTube(qt.QWidget):
sigQXTubeSignal = qt.pyqtSignal(object)
def __init__(self, parent=None, initdict = None):
qt.QWidget.__init__(self, parent)
self.l = qt.QVBoxLayout(self)
self.l.setContentsMargins(0, 0, 0, 0)
self.l.setSpacing(0)
self.tubeWidget = TubeWidget(self, initdict = initdict)
self.setParameters = self.tubeWidget.setParameters
self.getParameters = self.tubeWidget.getParameters
label = qt.QLabel(self)
hbox = qt.QWidget(self)
hboxl = qt.QHBoxLayout(hbox)
hboxl.setContentsMargins(0, 0, 0, 0)
hboxl.setSpacing(0)
self.plotButton = qt.QPushButton(hbox)
self.plotButton.setText("Plot Continuum")
self.exportButton = qt.QPushButton(hbox)
self.exportButton.setText("Export to Fit")
#grid.addWidget(self.plotButton, 7, 1)
#grid.addWidget(self.exportButton, 7, 3)
hboxl.addWidget(self.plotButton)
hboxl.addWidget(self.exportButton)
self.l.addWidget(self.tubeWidget)
f = label.font()
f.setItalic(1)
label.setFont(f)
label.setAlignment(qt.Qt.AlignRight)
label.setText("H. Ebel, X-Ray Spectrometry 28 (1999) 255-266 ")
self.l.addWidget(label)
self.l.addWidget(hbox)
self.graph = PlotWindow(self, backend=None)
self.l.addWidget(self.graph)
self.graph.setGraphXLabel("Energy (keV)")
self.graph.setGraphYLabel("photons/sr/mA/keV/s")
self.plotButton.clicked.connect(self.plot)
self.exportButton.clicked.connect(self._export)
def plot(self):
d = self.tubeWidget.getParameters()
transmission = d["transmission"]
anode = d["anode"]
anodedensity = d["anodedensity"]
anodethickness = d["anodethickness"]
voltage = d["voltage"]
wele = d["window"]
wdensity = d["windowdensity"]
wthickness = d["windowthickness"]
fele = d["filter1"]
fdensity = d["filter1density"]
fthickness = d["filter1thickness"]
filterlist =[[fele, fdensity, fthickness]]
alphae = d["alphae"]
alphax = d["alphax"]
delta = d["deltaplotting"]
e = numpy.arange(1, voltage, delta)
if __name__ == "__main__":
continuumR = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=0,
targetthickness=anodethickness,
filterlist=filterlist)
continuumT = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=1,
targetthickness=anodethickness,
filterlist=filterlist)
self.graph.addCurve(e, continuumR, "continuumR", replot=False)
self.graph.addCurve(e, continuumT, "continuumT", replot=False)
else:
continuum = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
self.graph.addCurve(e, continuum, "continuum", replot=False)
self.graph.setActiveCurve("continuum")
self.graph.resetZoom()
self.graph.replot()
def _export(self):
d = self.tubeWidget.getParameters()
transmission = d["transmission"]
anode = d["anode"]
anodedensity = d["anodedensity"]
anodethickness = d["anodethickness"]
voltage = d["voltage"]
wele = d["window"]
wdensity = d["windowdensity"]
wthickness = d["windowthickness"]
fele = d["filter1"]
fdensity = d["filter1density"]
fthickness = d["filter1thickness"]
filterlist =[[fele, fdensity, fthickness]]
alphae = d["alphae"]
alphax = d["alphax"]
delta = d["deltaplotting"]
e = numpy.arange(1, voltage, delta)
d["event"] = "TubeUpdated"
d["energyplot"] = e
d["continuum"] = XRayTubeEbel.continuumEbel([anode, anodedensity, anodethickness],
voltage, e,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
fllines = XRayTubeEbel.characteristicEbel([anode, anodedensity, anodethickness],
voltage,
[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
d["characteristic"] = fllines
energy, energyweight, energyscatter = XRayTubeEbel.generateLists(
[anode, anodedensity, anodethickness],
voltage,
window=[wele, wdensity, wthickness],
alphae=alphae, alphax=alphax,
transmission=transmission,
targetthickness=anodethickness,
filterlist=filterlist)
d["energylist"] = energy
d["weightlist"] = energyweight
d["scatterlist"] = energyscatter
d["flaglist"] = numpy.ones(len(energy), dtype=numpy.int32)
self.sigQXTubeSignal.emit(d)
def do_plot(self, custom_data):
x = custom_data[0, :]
y = custom_data[-1, :]
x.shape = -1
y.shape = -1
title = "top"
xtitle = "X"
ytitle = "Y"
print(x, y)
plot = PlotWindow(roi=True, control=True, position=True)
plot.setDefaultPlotLines(False)
plot.setActiveCurveColor(color='darkblue')
plot.addCurve(x, y, title, symbol='o', color='blue') #'+', '^',
plot.setGraphXLabel(xtitle)
plot.setGraphYLabel(ytitle)
plot.setDrawModeEnabled(True, 'rectangle')
plot.setZoomModeEnabled(True)
if self.figure_canvas is not None:
self.mainArea.layout().removeWidget(self.figure_canvas)
self.figure_canvas = plot
self.mainArea.layout().addWidget(self.figure_canvas)
