Skip to content

Commit

Permalink
add: clear preview prompt polygon when loop prompt
Browse files Browse the repository at this point in the history
  • Loading branch information
Fanchengyan committed Dec 31, 2023
1 parent 4145d2a commit 847db2a
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions tools/widgetTool.py
Original file line number Diff line number Diff line change
Expand Up @@ -445,6 +445,12 @@ def clear_canvas_layers_safely(self, clear_extent: bool = False):
if hasattr(self, "polygon"):
self.polygon.clear_canvas_polygons()
self.canvas.refresh()

def clear_preview_prompt_polygon(self):
self.tool_click_fg.clear_hover_prompt()
self.tool_click_bg.clear_hover_prompt()
self.tool_click_rect.clear_hover_prompt()
self.polygon.canvas_preview_polygon.clear()

def _set_feature_related(self):
"""Init or reload feature related objects"""
Expand Down Expand Up @@ -508,6 +514,8 @@ def loop_prompt_type(self):
self.tool_click_fg.pressed = False
self.tool_click_bg.pressed = False
self.tool_click_rect.pressed = False

self.clear_preview_prompt_polygon()

if self.wdg_sel.pushButton_fg.isChecked():
self.draw_background_point()
Expand Down Expand Up @@ -593,10 +601,7 @@ def toggle_sam_hover_mode(self):
self.tool_click_bg.preview_mode = False
self.tool_click_rect.preview_mode = False
# clear hover prompts
self.tool_click_fg.clear_hover_prompt()
self.tool_click_bg.clear_hover_prompt()
self.tool_click_rect.clear_hover_prompt()
self.polygon.canvas_preview_polygon.clear()
self.clear_preview_prompt_polygon()

if self.need_execute_sam_toggle_mode:
self.execute_SAM.emit()
Expand Down

0 comments on commit 847db2a

Please sign in to comment.