- #if diag:
- # x = int(math.floor(xf))
- # y = int(math.floor(yf))
-
- # diag0 = block0 + .5
- # diag0[
- # max(y - 21, 0):max(y + 22, 0),
- # max(x - 1, 0):max(x + 2, 0),
- # :
- # ] = 0.
- # diag0[
- # max(y - 1, 0):max(y + 2, 0),
- # max(x - 21, 0):max(x + 22, 0),
- # :
- # ] = 0.
- # gamma.write_image(f'diag_{xc:d}_{yc:d}_0.jpg', diag0)
-
- # diag1 = block1 + .5
- # diag1[
- # max(y - 21, 0):max(y + 22, 0),
- # max(x - 1, 0):max(x + 2, 0),
- # :
- # ] = 0.
- # diag1[
- # max(y - 1, 0):max(y + 2, 0),
- # max(x - 21, 0):max(x + 22, 0),
- # :
- # ] = 0.
- # gamma.write_image(f'diag_{xc:d}_{yc:d}_1.jpg', diag1)
+ if diag:
+ x = int(math.floor(xf))
+ y = int(math.floor(yf))
+
+ temp = match + .5
+ temp[
+ max(y - 21, 0):max(y + 22, 0),
+ max(x - 1, 0):max(x + 2, 0)
+ ] = 0.
+ temp[
+ max(y - 1, 0):max(y + 2, 0),
+ max(x - 21, 0):max(x + 22, 0)
+ ] = 0.
+ gamma.write_image(f'diag_{xc:d}_{yc:d}_match.jpg', temp)
+
+ diag0 = block0 + .5
+ diag0[
+ max(y - 21, 0):max(y + 22, 0),
+ max(x - 1, 0):max(x + 2, 0),
+ :
+ ] = 0.
+ diag0[
+ max(y - 1, 0):max(y + 2, 0),
+ max(x - 21, 0):max(x + 22, 0),
+ :
+ ] = 0.
+ gamma.write_image(f'diag_{xc:d}_{yc:d}_block0.jpg', diag0)
+
+ diag1 = block10 + .5
+ diag1[
+ max(y - 21, 0):max(y + 22, 0),
+ max(x - 1, 0):max(x + 2, 0),
+ :
+ ] = 0.
+ diag1[
+ max(y - 1, 0):max(y + 2, 0),
+ max(x - 21, 0):max(x + 22, 0),
+ :
+ ] = 0.
+ gamma.write_image(f'diag_{xc:d}_{yc:d}_block10.jpg', diag1)
+
+ x += xo
+ y += yo
+ diag1 = block1 + .5
+ diag1[
+ max(y - 21, 0):max(y + 22, 0),
+ max(x - 1, 0):max(x + 2, 0),
+ :
+ ] = 0.
+ diag1[
+ max(y - 1, 0):max(y + 2, 0),
+ max(x - 21, 0):max(x + 22, 0),
+ :
+ ] = 0.
+ gamma.write_image(f'diag_{xc:d}_{yc:d}_block11.jpg', diag1)