Fix CursorTool issue with LogMapper

chaco/tools/cursor_tool.py

@@ -200,7 +200,7 @@ def draw(self, gc, view_bounds=None):
         if plot is None:
             return
 
-        sx, sy = plot.map_screen(self.current_position)
+        sx, sy = plot.map_screen(self.current_position)[0]
         orientation = plot.orientation
 
         if orientation == "h" and sx is not None:
@@ -221,7 +221,7 @@ def is_draggable(self, x, y):
         plot = self.component
         if plot is not None:
             orientation = plot.orientation
-            sx, sy = plot.map_screen(self.current_position)
+            sx, sy = plot.map_screen(self.current_position)[0]
             if orientation == "h" and numpy.abs(sx - x) <= self.threshold:
                 return True
             elif orientation == "v" and numpy.abs(sy - y) <= self.threshold:

chaco/tools/tests/test_cursor_tool.py

@@ -0,0 +1,53 @@
+import unittest
+
+import numpy as np
+
+from enable.api import ComponentEditor
+from enable.testing import EnableTestAssistant
+from traits.api import HasTraits, Instance
+from traits.etsconfig.api import ETSConfig
+from traitsui.api import Item, View
+from traitsui.testing.api import UITester
+
+from chaco.api import ArrayPlotData, Plot
+from chaco.tools.cursor_tool import CursorTool
+
+
+class TestCursorTool(unittest.TestCase, EnableTestAssistant):
+
+    # regression test for enthought/chaco#289
+    @unittest.skipIf(ETSConfig.toolkit == "null", "Skip on 'null' toolkit")
+    def test_use_with_log_mappers(self):
+        class TestCursor(HasTraits):
+            plot = Instance(Plot)
+
+            traits_view = View(
+                Item('plot', editor=ComponentEditor(), show_label=False),
+                width=500,
+                height=500,
+                resizable=True
+            )
+
+            def _plot_default(self):
+                arr = np.logspace(0, 10, num=10)
+                data = ArrayPlotData(x=arr, y=arr)
+                plot = Plot(data)
+                renderer = plot.plot(
+                    ("x", "y"),
+                    index_scale="log",
+                    value_scale="log"
+                )[0]
+
+                cursor = CursorTool(renderer)
+
+                renderer.overlays.append(cursor)
+
+                return plot
+
+        tester = UITester()
+        test_cursor = TestCursor()
+        with tester.create_ui(test_cursor):
+            # should not fail
+            self.mouse_move(
+                test_cursor.plot, 10, 10
+            )