def retrace(self):
try:
if not self.input_beam is None:
output_beam = self.input_beam.duplicate(history=True)
empty_element = ShadowOpticalElement.create_empty_oe()
empty_element._oe.DUMMY = 1.0 # self.workspace_units_to_cm
empty_element._oe.T_SOURCE = 0.0
empty_element._oe.T_IMAGE = self.retrace_distance
empty_element._oe.T_INCIDENCE = 0.0
empty_element._oe.T_REFLECTION = 180.0
empty_element._oe.ALPHA = 0.0
empty_element._oe.FWRITE = 3
empty_element._oe.F_ANGLE = 0
output_beam = ShadowBeam.traceFromOE(output_beam, empty_element, history=True)
self.setStatusMessage("Plotting Results")
self.plot_results(output_beam)
self.setStatusMessage("")
self.progressBarFinished()
self.send("Beam", output_beam)
self.send("Trigger", TriggerIn(new_object=True))
except Exception as exception:
QMessageBox.critical(self, "Error", str(exception), QMessageBox.Ok)
if self.IS_DEVELOP: raise exception
def get_zone_plate_beam(self): # WS Units
empty_element = ShadowOpticalElement.create_empty_oe()
empty_element._oe.DUMMY = self.workspace_units_to_cm
empty_element._oe.T_SOURCE = self.source_plane_distance
empty_element._oe.T_IMAGE = 0.0
empty_element._oe.T_INCIDENCE = 0.0
empty_element._oe.T_REFLECTION = 180.0
empty_element._oe.ALPHA = 0.0
empty_element._oe.FWRITE = 3
empty_element._oe.F_ANGLE = 0
n_screen = 1
i_screen = numpy.array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0])
i_abs = numpy.zeros(10)
i_slit = numpy.array([1, 0, 0, 0, 0, 0, 0, 0, 0, 0])
i_stop = numpy.zeros(10)
k_slit = numpy.array([1, 0, 0, 0, 0, 0, 0, 0, 0, 0])
thick = numpy.zeros(10)
file_abs = numpy.array(['', '', '', '', '', '', '', '', '', ''])
rx_slit = numpy.zeros(10)
rz_slit = numpy.zeros(10)
sl_dis = numpy.zeros(10)
file_scr_ext = numpy.array(['', '', '', '', '', '', '', '', '', ''])
cx_slit = numpy.zeros(10)
cz_slit = numpy.zeros(10)
sl_dis[0] = 0.0
rx_slit[0] = self.diameter*1e-6/self.workspace_units_to_m
rz_slit[0] = self.diameter*1e-6/self.workspace_units_to_m
cx_slit[0] = 0.0
cz_slit[0] = 0.0
empty_element._oe.set_screens(n_screen,
i_screen,
i_abs,
sl_dis,
i_slit,
i_stop,
k_slit,
thick,
file_abs,
rx_slit,
rz_slit,
cx_slit,
cz_slit,
file_scr_ext)
output_beam = ShadowBeam.traceFromOE(self.input_beam, empty_element, history=True)
go = numpy.where(output_beam._beam.rays[:, 9] == GOOD)
lo = numpy.where(output_beam._beam.rays[:, 9] != GOOD)
print("Zone Plate Beam: ", "GO", len(go[0]), "LO", len(lo[0]))
return output_beam
def create_dummy_oe():
empty_element = ShadowOpticalElement.create_empty_oe()
empty_element._oe.DUMMY = 1.0 # self.workspace_units_to_cm
empty_element._oe.T_SOURCE = 0.0
empty_element._oe.T_IMAGE = 0.0
empty_element._oe.T_INCIDENCE = 0.0
empty_element._oe.T_REFLECTION = 180.0
empty_element._oe.ALPHA = 0.0
empty_element._oe.FWRITE = 3
empty_element._oe.F_ANGLE = 0
return empty_element
def create_dummy_oe(self):
empty_element = ShadowOpticalElement.create_empty_oe()
empty_element._oe.DUMMY = self.workspace_units_to_cm
empty_element._oe.T_SOURCE = 0.0
empty_element._oe.T_IMAGE = 0.0
empty_element._oe.T_INCIDENCE = 0.0
empty_element._oe.T_REFLECTION = 180.0
empty_element._oe.ALPHA = 0.0
empty_element._oe.FWRITE = 3
empty_element._oe.F_ANGLE = 0
return empty_element
def get_output_beam(self, focused_beam):
empty_element = ShadowOpticalElement.create_empty_oe()
empty_element._oe.DUMMY = 1.0 # self.workspace_units_to_cm
empty_element._oe.T_SOURCE = 0.0
empty_element._oe.T_IMAGE = self.image_plane_distance
empty_element._oe.T_INCIDENCE = 0.0
empty_element._oe.T_REFLECTION = 180.0
empty_element._oe.ALPHA = 0.0
empty_element._oe.FWRITE = 3
empty_element._oe.F_ANGLE = 0
return ShadowBeam.traceFromOE(focused_beam, empty_element, history=True)
go, 7] = focused_beam._beam.rays[go, 7] * efficiency_weight_factor[:]
focused_beam._beam.rays[
go, 8] = focused_beam._beam.rays[go, 8] * efficiency_weight_factor[:]
focused_beam._beam.rays[
go, 15] = focused_beam._beam.rays[go, 15] * efficiency_weight_factor[:]
focused_beam._beam.rays[
go, 16] = focused_beam._beam.rays[go, 16] * efficiency_weight_factor[:]
focused_beam._beam.rays[
go, 17] = focused_beam._beam.rays[go, 17] * efficiency_weight_factor[:]
return focused_beam
#################################
empty_element = ShadowOpticalElement.create_empty_oe()
empty_element._oe.DUMMY = 1.0 # self.workspace_units_to_cm
empty_element._oe.T_SOURCE = source_plane_distance
empty_element._oe.T_IMAGE = 0.0
empty_element._oe.T_INCIDENCE = 0.0
empty_element._oe.T_REFLECTION = 180.0
empty_element._oe.ALPHA = 0.0
empty_element._oe.FWRITE = 3
empty_element._oe.F_ANGLE = 0
n_screen = 1
i_screen = numpy.array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0])
i_abs = numpy.zeros(10)
