Fix: The hidding brush after the watershed 3D expansion

Thiago Franco de Moraes
1 parent e532ca4e
Showing 1 changed file with 8 additions and 0 deletions Show diff stats
invesalius/data/styles.py
@@ -1341,6 +1341,14 @@ class WaterShedInteractorStyle(DefaultInteractorStyle):
             wp.Close()
             del wp
  
+            w_x, w_y = wx.GetMousePosition()
+            x, y = self.viewer.ScreenToClientXY(w_x, w_y)
+            flag = self.viewer.interactor.HitTest((x, y))
+
+            if flag == wx.HT_WINDOW_INSIDE:
+                self.OnEnterInteractor(None, None)
+
+
             if q.empty():
                 return
             #do_watershed(image, markers, tmp_mask, bstruct, self.config.algorithm,
...	...	@@ -1341,6 +1341,14 @@ class WaterShedInteractorStyle(DefaultInteractorStyle):
1341	1341	wp.Close()
1342	1342	del wp
1343	1343
	1344	+ w_x, w_y = wx.GetMousePosition()
	1345	+ x, y = self.viewer.ScreenToClientXY(w_x, w_y)
	1346	+ flag = self.viewer.interactor.HitTest((x, y))
	1347	+
	1348	+ if flag == wx.HT_WINDOW_INSIDE:
	1349	+ self.OnEnterInteractor(None, None)
	1350	+
	1351	+
1344	1352	if q.empty():
1345	1353	return
1346	1354	#do_watershed(image, markers, tmp_mask, bstruct, self.config.algorithm,
...	...