# (C) Copyright 2005-2021 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!

import unittest

from traits.api import HasTraits, Instance
from traits.etsconfig.api import ETSConfig
from traitsui.api import Item, View
from enable.api import ComponentEditor

from chaco.api import Plot, ArrayPlotData
from chaco.tools.highlight_tool import HighlightTool

import numpy as np


class PlotViewer(HasTraits):
    plot = Instance(Plot)
    traits_view = View(Item("plot", editor=ComponentEditor()))


class MockEvent(HasTraits):
    pass


@unittest.skipIf(ETSConfig.toolkit == "null", "Skip on 'null' toolkit")
class TestHighlightTool(unittest.TestCase):
    def test_highlight_on_log_plot(self):
        # test for bug: the highlight tool raises an exception when used on
        # a loglog plot

        x = np.linspace(1, 15, 200)

        plotdata = ArrayPlotData()
        plotdata.set_data("x", x)
        plotdata.set_data("y", x * x)

        plot = Plot(plotdata)
        plot.plot(("x", "y"), index_scale="log", value_scale="log")

        # necessary for the machinery involved in _find_curve
        plot.datasources["x"].sort_order = "ascending"
        plot.datasources["y"].sort_order = "ascending"

        # add the highlight tool
        htool = HighlightTool(plot, threshold=20.0)
        plot.tools.append(htool)

        # we create a view of the plot so that the screen bounds are set
        pv = PlotViewer(plot=plot)
        pv.edit_traits()

        # this should not raise an exception
        event = MockEvent(x=170.0, y=60.0)
        htool._find_curve(plot.plots["plot0"], event)