Fixing rotation angle in renderer.py (#277)

closes #276
pmelchior · Feb 15, 2023 · 5b0b5e0 · 5b0b5e0
1 parent 11d1363
commit 5b0b5e0
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/scarlet/renderer.py b/scarlet/renderer.py
@@ -322,29 +322,29 @@ def __init__(self, data_frame, model_frame, padding=10):
             Y_unrot = (
                 (coord_hr[:, 0] - center_y) * self.angle[0]
                 - (coord_hr[:, 1] - center_x) * self.angle[1]
-            ).reshape(lr_shape)
+            ).reshape(lr_shape[0])
             X_unrot = (
                 (coord_hr[:, 1] - center_x) * self.angle[0]
                 + (coord_hr[:, 0] - center_y) * self.angle[1]
-            ).reshape(lr_shape)
+            ).reshape(lr_shape[1])
 
             # Removing redundancy
-            self.Y_unrot = Y_unrot[:, 0]
-            self.X_unrot = X_unrot[0, :]
+            self.Y_unrot = Y_unrot
+            self.X_unrot = X_unrot
 
             if self.small_axis:
                 self.shifts = np.array(
-                    [self.Y_unrot * self.angle[0], self.Y_unrot * self.angle[1]]
+                    [self.Y_unrot * self.angle[0], -self.Y_unrot * self.angle[1],]
                 )
                 self.other_shifts = np.array(
-                    [-self.angle[1] * self.X_unrot, self.angle[0] * self.X_unrot,]
+                    [self.angle[1] * self.X_unrot, self.angle[0] * self.X_unrot,]
                 )
             else:
                 self.shifts = np.array(
-                    [-self.angle[1] * self.X_unrot, self.angle[0] * self.X_unrot,]
+                    [self.angle[1] * self.X_unrot, self.angle[0] * self.X_unrot,]
                 )
                 self.other_shifts = np.array(
-                    [self.Y_unrot * self.angle[0], self.Y_unrot * self.angle[1],]
+                    [self.Y_unrot * self.angle[0], -self.Y_unrot * self.angle[1],]
                 )
 
             axes = (1, 2)