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

fix: rotation always counter-clockwise

Browse Source
This commit is contained in:
wyzula-jan
2023-09-08 14:33:35 +02:00
parent 66da042882
commit 00385abbf9
1 changed files with 3 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -75,18 +75,13 @@ class EigerPlot(QWidget):
self.hist_levels[1] + 0.1 * self.hist_levels[1],
)
def rotate_data(self, data, k: int = 0, direction: int = 0) -> np.ndarray:
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.
direction(int): 0 for clockwise, 1 for counter-clockwise
"""
if direction == 0:
data = np.rot90(self.image, k=k, axes=(1, 0))
else:
data = np.rot90(self.image, k=k, axes=(0, 1))
data = np.rot90(data, k=k, axes=(0, 1))
return data
@ -98,11 +93,7 @@ class EigerPlot(QWidget):
if self.comboBox_rotation.currentIndex() == 0: # non rotated image
self.imageItem.setImage(self.image, autoLevels=False)
else: # rotated image
self.image = self.rotate_data(
data=self.image,
k=self.comboBox_rotation.currentIndex(),
direction=self.comboBox_direction.currentIndex(),
)
self.image = self.rotate_data(data=self.image, k=self.comboBox_rotation.currentIndex())
self.imageItem.setImage(self.image, autoLevels=False)