diff --git a/image_marker.py b/image_marker.py index a462720..df1a066 100644 --- a/image_marker.py +++ b/image_marker.py @@ -74,7 +74,7 @@ class ImageMarker: self._fig.canvas.draw() def _on_click_event(self, event): - self._fig.gca().scatter(event.xdata, event.ydata, s=20) + self._fig.gca().scatter(event.xdata, event.ydata, marker=self._current_task.marker_symbol, color=self._current_task.marker_color, s=20) event.canvas.draw() self._current_task.set_position(self._current_task.current_marker, event.xdata, event.ydata) self._next_task() @@ -83,7 +83,7 @@ class ImageMarker: self._interrupt = True class MarkerTask(): - def __init__(self, name:str, marker_names=[], message="") -> None: + def __init__(self, name:str, marker_names=[], message="", marker="o", color="tab:blue") -> None: super().__init__() self._positions = {} self._marker_names = marker_names @@ -91,6 +91,8 @@ class MarkerTask(): self._message = message self._current_marker = marker_names[0] if len(marker_names) > 0 else None self._current_index = 0 + self._marker = marker + self._marker_color = color @property def positions(self): @@ -122,6 +124,14 @@ class MarkerTask(): def current_marker(self): return self._current_marker + @property + def marker_symbol(self): + return self._marker + + @property + def marker_color(self): + return self._marker_color + def __str__(self) -> str: return "MarkerTask %s with markers: %s" % (self.name, [mn for mn in self._marker_names])