def batch_create_trajectories(self):
for i in range(len(self.experiment_table)):
d = self.experiment_table.iloc[i]
traj = trajectory(self.hhm)
kwargs = dict(
edge_energy=d['Position'],
offsets=([
d['pre-edge start'], d['pre-edge stop'],
d['post_edge stop'],
xray.k2e(d['k-range'], d['Position']) - d['Position']
]),
trajectory_type='Double Sine',
dsine_preedge_duration=d['time 1'],
dsine_postedge_duration=d['time 2'],
)
traj.define(**kwargs)
# Add some more parameters manually:
traj.elem = d['Element']
traj.edge = d['Edge']
traj.e0 = d['Position']
traj.interpolate()
traj.revert()
self.trajectories.append(traj)
self.experiments.append(XASExperiment(definition=d))
def __init__(self, hhm=None, run_prep_traj=None, *args, **kwargs):
super().__init__(*args, **kwargs)
self.setupUi(self)
self.addCanvas()
self.element = 'Scandium (21)'
self.e0 = '4492'
self.edge = 'K'
self.widget_energy_selector = isstools.widgets.widget_energy_selector.UIEnergySelector(
)
self.layout_energy_selector_trajectory.addWidget(
self.widget_energy_selector)
#communication between the Energy Selector widget and Trajectory Manager
self.widget_energy_selector.edit_E0.textChanged.connect(self.update_E0)
self.widget_energy_selector.comboBox_edge.currentTextChanged.connect(
self.update_edge)
self.widget_energy_selector.comboBox_element.currentTextChanged.connect(
self.update_element)
self.run_prep_traj = run_prep_traj
self.hhm = hhm
self.hhm.angle_offset.subscribe(self.update_angle_offset)
self.traj_manager = trajectory_manager(hhm)
self.comboBox_slot_to_load_trajectory.addItems(
['1', '2', '3', '4', '5', '6', '7', '8'])
self.comboBox_slot_to_init_trajectory.addItems(
['1', '2', '3', '4', '5', '6', '7', '8'])
self.comboBox_slot_to_init_trajectory.setCurrentIndex(
self.traj_manager.current_lut() - 1)
try:
pass
# self.trajectories = self.traj_manager.read_info(silent=True)
# self.trajectories = collections.OrderedDict(sorted(self.trajectories.items()))
except OSError as err:
print('Error loading:', err)
self.traj_creator = trajectory(self.hhm)
self.trajectory_path = self.hhm.traj_filepath #'/GPFS/xf08id/trajectory/'
self.push_build_trajectory.clicked.connect(self.build_trajectory)
self.push_save_trajectory.clicked.connect(self.save_trajectory)
self.push_update_offset.clicked.connect(self.update_offset)
self.push_select_traj_file.clicked.connect(self.get_traj_names)
self.push_load_trajectory.clicked.connect(self.load_trajectory)
self.push_init_trajectory.clicked.connect(self.init_trajectory)
self.push_read_traj_info.clicked.connect(self.read_trajectory_info)
self.push_prepare_trajectory.clicked.connect(self.run_prep_traj)
self.push_plot_traj.clicked.connect(self.plot_traj_file)
self.push_plot_traj.setDisabled(True)
self.push_save_trajectory.setDisabled(True)
self.checkBox_traj_single_dir.stateChanged.connect(
self.update_repetitions_spinbox)
self.checkBox_traj_single_dir.stateChanged.connect(
self.checkBox_traj_revert.setEnabled)