"""Toolbar Plot

A toolbar plot is the same as a regular Plot, but also provides a 
usually hidden drop-down toolbar with configurable buttons.

The toolbar appears when the mouse hovers over the top of the plot area.

This toolbar provides buttons for X log scale, Y log scale, Save as,
Copy image, Copy data, and Zoom reset.

"""

import numpy

from chaco.plot import Plot, ArrayPlotData
from chaco.api import ToolbarPlot
from enable.api import ComponentEditor
from traits.api import Instance, HasTraits
from traitsui.api import View, Item


class ExamplePlotApp(HasTraits):

    plot = Instance(Plot)

    def _plot_default(self):
        index = numpy.arange(1.0, 10.0, 0.01)
        series1 = (100.0 + index) / (100.0 - 20 * index ** 2 + 5.0 * index**4)
        series2 = (100.0 + index) / (100.0 - 20 * index ** 2 + 5.0 * index**3)

        plot_data = ArrayPlotData(index=index)
        plot_data.set_data("series1", series1)
        plot_data.set_data("series2", series2)

        plot = ToolbarPlot(plot_data)
        plot.plot(("index", "series1"), color="auto")
        plot.plot(("index", "series2"), color="auto")

        return plot

    traits_view = View(
        Item(
            "plot",
            editor=ComponentEditor(),
            width=600,
            height=600,
            show_label=False,
        ),
        resizable=True,
    )


demo = ExamplePlotApp()

if __name__ == "__main__":
    demo.configure_traits()