# (C) Copyright 2004-2023 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!

# -------------------------------------------------------------------------
#  Imports:
# -------------------------------------------------------------------------

import wx

from traits.api import (
    Trait,
    HasTraits,
    Str,
    Int,
    Range,
    List,
    Event,
    File,
    Directory,
    Bool,
    Enum,
)

from traitsui.api import (
    View,
    Handler,
    Item,
    CheckListEditor,
    ButtonEditor,
    FileEditor,
    DirectoryEditor,
    ImageEnumEditor,
    Color,
    Font,
)

# -------------------------------------------------------------------------
#  Constants:
# -------------------------------------------------------------------------

origin_values = ['top left', 'top right', 'bottom left', 'bottom right']

# -------------------------------------------------------------------------
#  'Instance' class:
# -------------------------------------------------------------------------


class Instance(HasTraits):

    # -------------------------------------------------------------------------
    #  Trait definitions:
    # -------------------------------------------------------------------------

    integer_text = Int(1)
    enumeration = Enum('one', 'two', 'three', 'four', 'five', 'six', cols=3)
    float_range = Range(0.0, 10.0, 10.0)
    int_range = Range(1, 5)
    boolean = Bool(True)

    view = View(
        'integer_text', 'enumeration', 'float_range', 'int_range', 'boolean'
    )


# -------------------------------------------------------------------------
#  'TraitsTest' class
# -------------------------------------------------------------------------


class TraitsTest(HasTraits):

    # -------------------------------------------------------------------------
    #  Trait definitions:
    # -------------------------------------------------------------------------

    integer_text = Int(1)
    enumeration = Enum('one', 'two', 'three', 'four', 'five', 'six', cols=3)
    float_range = Range(0.0, 10.0, 10.0)
    int_range = Range(1, 6)
    int_range2 = Range(1, 50)
    compound = Trait(
        1, Range(1, 6), 'one', 'two', 'three', 'four', 'five', 'six'
    )
    boolean = Bool(True)
    instance = Trait(Instance())
    color = Color
    font = Font
    check_list = List(
        editor=CheckListEditor(values=['one', 'two', 'three', 'four'], cols=4)
    )
    list = List(
        Str, ['East of Eden', 'The Grapes of Wrath', 'Of Mice and Men']
    )
    button = Event(0, editor=ButtonEditor(label='Click'))
    file = File()
    directory = Directory()
    image_enum = Trait(
        editor=ImageEnumEditor(
            values=origin_values, suffix='_origin', cols=4, klass=Instance
        ),
        *origin_values,
    )

    # -------------------------------------------------------------------------
    #  View definitions:
    # -------------------------------------------------------------------------

    view = View(
        (
            '|{Enum}',
            (
                '|<[Enumeration]',
                'enumeration[Simple]',
                '_',
                'enumeration[Custom]@',
                '_',
                'enumeration[Text]*',
                '_',
                'enumeration[Readonly]~',
            ),
            (
                '|<[Check List]',
                'check_list[Simple]',
                '_',
                'check_list[Custom]@',
                '_',
                'check_list[Text]*',
                '_',
                'check_list[Readonly]~',
            ),
        ),
        (
            '|{Range}',
            (
                '|<[Float Range]',
                'float_range[Simple]',
                '_',
                'float_range[Custom]@',
                '_',
                'float_range[Text]*',
                '_',
                'float_range[Readonly]~',
            ),
            (
                '|<[Int Range]',
                'int_range[Simple]',
                '_',
                'int_range[Custom]@',
                '_',
                'int_range[Text]*',
                '_',
                'int_range[Readonly]~',
            ),
            (
                '|<[Int Range 2]',
                'int_range2[Simple]',
                '_',
                'int_range2[Custom]@',
                '_',
                'int_range2[Text]*',
                '_',
                'int_range2[Readonly]~',
            ),
        ),
        (
            '|{Misc}',
            (
                '|<[Integer Text]',
                'integer_text[Simple]',
                '_',
                'integer_text[Custom]@',
                '_',
                'integer_text[Text]*',
                '_',
                'integer_text[Readonly]~',
            ),
            (
                '|<[Compound]',
                'compound[Simple]',
                '_',
                'compound[Custom]@',
                '_',
                'compound[Text]*',
                '_',
                'compound[Readonly]~',
            ),
            (
                '|<[Boolean]',
                'boolean[Simple]',
                '_',
                'boolean[Custom]@',
                '_',
                'boolean[Text]*',
                '_',
                'boolean[Readonly]~',
            ),
        ),
        (
            '|{Color/Font}',
            (
                '|<[Color]',
                'color[Simple]',
                '_',
                'color[Custom]@',
                '_',
                'color[Text]*',
                '_',
                'color[Readonly]~',
            ),
            (
                '|<[Font]',
                'font[Simple]',
                '_',
                'font[Custom]@',
                '_',
                'font[Text]*',
                '_',
                'font[Readonly]~',
            ),
        ),
        (
            '|{List}',
            (
                '|<[List]',
                'list[Simple]',
                '_',
                'list[Custom]@',
                '_',
                'list[Text]*',
                '_',
                'list[Readonly]~',
            ),
        ),
        (
            '|{Button}',
            ('|<[Button]', 'button[Simple]', '_', 'button[Custom]@'),
            #                                        'button[Text]*',
            #                                        'button[Readonly]~' ),
            (
                '|<[Image Enum]',
                'image_enum[Simple]',
                '_',
                'image_enum[Custom]@',
                '_',
                'image_enum[Text]*',
                '_',
                'image_enum[Readonly]~',
            ),
            (
                '|<[Instance]',
                'instance[Simple]',
                '_',
                'instance[Custom]@',
                '_',
                'instance[Text]*',
                '_',
                'instance[Readonly]~',
            ),
        ),
        (
            '|{File}',
            (
                '|<[File]',
                'file[Simple]',
                '_',
                'file[Custom]@',
                '_',
                'file[Text]*',
                '_',
                'file[Readonly]~',
            ),
            (
                '|<[Directory]',
                'directory[Simple]',
                '_',
                'directory[Custom]@',
                '_',
                'directory[Text]*',
                '_',
                'directory[Readonly]~',
            ),
        ),
        buttons=['Apply', 'Revert', 'Undo', 'OK'],
    )


# -------------------------------------------------------------------------
#  'TraitSheetApp' class:
# -------------------------------------------------------------------------


class TraitSheetApp(wx.App):

    # -------------------------------------------------------------------------
    #  Initialize the object:
    # -------------------------------------------------------------------------

    def __init__(self, object):
        self.object = object
        wx.InitAllImageHandlers()
        wx.App.__init__(self, 1, 'debug.log')
        self.MainLoop()

    # -------------------------------------------------------------------------
    #  Handle application initialization:
    # -------------------------------------------------------------------------

    def OnInit(self):
        ui = self.object.edit_traits(kind='modal')
        ui = self.object.edit_traits(kind='wizard')
        ui = self.object.edit_traits(kind='nonmodal')
        ui = self.object.edit_traits(kind='live')
        self.SetTopWindow(ui.control)
        return True


# -------------------------------------------------------------------------
#  Main program:
# -------------------------------------------------------------------------

if __name__ == '__main__':
    TraitSheetApp(TraitsTest())