Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
import tkinter as tk
import numpy as np
from PIL import Image, ImageTk
array_t = np.ndarray
class soma_validation_window_t:
def __init__(self, lmap: array_t, mip_axis: int = -1):
""""""
self.lmap = lmap
self.mip = np.max(lmap, axis=mip_axis)
mip_shape = self.mip.shape
offset = 50
scaling = (255.0 - offset) / np.max(self.mip)
self.scaled_mip = scaling * self.mip + offset
self.scaled_mip[self.mip == 0] = 0
self.root = tk.Tk()
self.canvas = tk.Canvas(
self.root, width=mip_shape[1], height=mip_shape[0]
)
self.tk_image = ImageTk.PhotoImage(
master=self.root, image=Image.fromarray(self.scaled_mip)
)
self.canvas_image = self.canvas.create_image(
0, 0, anchor="nw", image=self.tk_image
)
self.canvas.bind("<Button-1>", self._DeleteSoma)
self.canvas.pack()
def LaunchInteraction(self) -> None:
""""""
self.root.mainloop()
# scikit-image.relabel(self.lmap)
def _DeleteSoma(self, event) -> None:
""""""
row = event.y
col = event.x
value = self.mip[row, col]
if value > 0:
self.lmap[self.lmap == value] = 0
soma_bmap = self.mip == value
self.mip[soma_bmap] = 0
self.scaled_mip[soma_bmap] = 0
self.tk_image = ImageTk.PhotoImage(
master=self.root, image=Image.fromarray(self.scaled_mip)
)
self.canvas.itemconfig(self.canvas_image, image=self.tk_image)