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() self._fig.canvas.draw()
def _on_click_event(self, event): 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() event.canvas.draw()
self._current_task.set_position(self._current_task.current_marker, event.xdata, event.ydata) self._current_task.set_position(self._current_task.current_marker, event.xdata, event.ydata)
self._next_task() self._next_task()
@ -83,7 +83,7 @@ class ImageMarker:
self._interrupt = True self._interrupt = True
class MarkerTask(): 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__() super().__init__()
self._positions = {} self._positions = {}
self._marker_names = marker_names self._marker_names = marker_names
@ -91,6 +91,8 @@ class MarkerTask():
self._message = message self._message = message
self._current_marker = marker_names[0] if len(marker_names) > 0 else None self._current_marker = marker_names[0] if len(marker_names) > 0 else None
self._current_index = 0 self._current_index = 0
self._marker = marker
self._marker_color = color
@property @property
def positions(self): def positions(self):
@ -122,6 +124,14 @@ class MarkerTask():
def current_marker(self): def current_marker(self):
return self._current_marker 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: def __str__(self) -> str:
return "MarkerTask %s with markers: %s" % (self.name, [mn for mn in self._marker_names]) return "MarkerTask %s with markers: %s" % (self.name, [mn for mn in self._marker_names])