aboutsummaryrefslogtreecommitdiffstats
path: root/fg21sim/utils/rotate.py
diff options
context:
space:
mode:
Diffstat (limited to 'fg21sim/utils/rotate.py')
-rw-r--r--fg21sim/utils/rotate.py17
1 files changed, 13 insertions, 4 deletions
diff --git a/fg21sim/utils/rotate.py b/fg21sim/utils/rotate.py
index b89365c..84e4f9d 100644
--- a/fg21sim/utils/rotate.py
+++ b/fg21sim/utils/rotate.py
@@ -61,10 +61,10 @@ def rotate_center(imgin, angle, interp=True, reshape=True, fill_value=0.0):
if reshape:
dest = np.dot(corners.astype(np.float64), mrotate)
# XXX: ``numba`` does not support ``np.max()`` with arguments
- minc = np.min(dest[:, 0])
- minr = np.min(dest[:, 1])
- maxc = np.max(dest[:, 0])
- maxr = np.max(dest[:, 1])
+ minr = np.min(dest[:, 0])
+ minc = np.min(dest[:, 1])
+ maxr = np.max(dest[:, 0])
+ maxc = np.max(dest[:, 1])
nr = int(maxr - minr + 0.5)
nc = int(maxc - minc + 0.5)
else:
@@ -110,6 +110,15 @@ def rotate_center(imgin, angle, interp=True, reshape=True, fill_value=0.0):
# Use area mapping of the 4 closest input pixels
idx_rf, idx_cf = np.floor(p_in).astype(np.int64)
idx_rc, idx_cc = np.ceil(p_in).astype(np.int64)
+ # NOTE:
+ # It is possible that ``p_in[0]`` and/or ``p_in[1]``
+ # are just integers, which cause ``idx_rf == idx_rc``
+ # and/or ``idx_cf == idx_cc``, which further lead to
+ # the calculated pixel value ``p_val = 0``.
+ if idx_rf == idx_rc:
+ idx_rc += 1
+ if idx_cf == idx_cc:
+ idx_cc += 1
# Calculate the overlapping areas
p_r, p_c = p_in
p4_area = np.array([(idx_rc - p_r) * (idx_cc - p_c),