From 63c92537b1956400b740723f881765f9642d5f35 Mon Sep 17 00:00:00 2001 From: Stefan van der Walt Date: Tue, 30 Sep 2014 06:18:03 +0200 Subject: [PATCH] Also add handle style to ThickLineTool --- skimage/viewer/canvastools/linetool.py | 8 ++++++-- skimage/viewer/tests/test_tools.py | 3 ++- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/skimage/viewer/canvastools/linetool.py b/skimage/viewer/canvastools/linetool.py index 39fcd4c1..f18a1915 100644 --- a/skimage/viewer/canvastools/linetool.py +++ b/skimage/viewer/canvastools/linetool.py @@ -151,6 +151,9 @@ class ThickLineTool(LineTool): Maximum pixel distance allowed when selecting control handle. line_props : dict Properties for :class:`matplotlib.lines.Line2D`. + handle_props : dict + Marker properties for the handles (also see + :class:`matplotlib.lines.Line2D`). Attributes ---------- @@ -159,13 +162,14 @@ class ThickLineTool(LineTool): """ def __init__(self, viewer, on_move=None, on_enter=None, on_release=None, - on_change=None, maxdist=10, line_props=None): + on_change=None, maxdist=10, line_props=None, handle_props=None): super(ThickLineTool, self).__init__(viewer, on_move=on_move, on_enter=on_enter, on_release=on_release, maxdist=maxdist, - line_props=line_props) + line_props=line_props, + handle_props=handle_props) if on_change is None: def on_change(*args): diff --git a/skimage/viewer/tests/test_tools.py b/skimage/viewer/tests/test_tools.py index b0d487a2..0805364c 100644 --- a/skimage/viewer/tests/test_tools.py +++ b/skimage/viewer/tests/test_tools.py @@ -104,7 +104,8 @@ def test_thick_line_tool(): img = data.camera() viewer = ImageViewer(img) - tool = ThickLineTool(viewer, maxdist=10) + tool = ThickLineTool(viewer, maxdist=10, line_props=dict(color='red'), + handle_props=dict(markersize=5)) tool.end_points = get_end_points(img) do_event(viewer, 'scroll', button='up')