bec/bec_widgets
0
0
Fork 0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-13 19:21:50 +02:00
Code Activity

feat: added rotate and transpose logic

Browse Source
This commit is contained in:
wyzula-jan
2023-09-08 14:39:48 +02:00
parent 2398ee3e8c
commit acd7a3bc92
1 changed files with 7 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
bec_widgets/examples/eiger_plot/eiger_plot.py
View File

@ -75,23 +75,15 @@ class EigerPlot(QWidget):
self.hist_levels[1] + 0.1 * self.hist_levels[1],
)
def rotate_data(self, data, k: int = 0) -> np.ndarray:
"""Rotate image by 90 degrees k times.
Args:
k(int): Number of times to rotate image by 90 degrees.
"""
data = np.rot90(data, k=k, axes=(0, 1))
return data
def transpose_data(self):
self.image = np.transpose(self.image)
@pyqtSlot()
def on_image_update(self):
if self.comboBox_rotation.currentIndex() > 0: # rotated image
self.image = self.rotate_data(data=self.image, k=self.comboBox_rotation.currentIndex())
# TODO first rotate then transpose
if self.comboBox_rotation.currentIndex() > 0: # rotate
self.image = np.rot90(self.image, k=self.comboBox_rotation.currentIndex(), axes=(0, 1))
if self.checkBox_transpose.isChecked(): # transpose
self.image = np.transpose(self.image)
self.imageItem.setImage(self.image, autoLevels=False)