jgrewe/tracking_tools
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[image_marker] markere customization

Browse Source
This commit is contained in:
Jan Grewe 2020-12-19 18:34:58 +01:00
parent ea02bb7ee3
commit 185e348a0b
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
image_marker.py
View File

@ -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])