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

""" Traits Test case
"""

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

import wx

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

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 = Trait('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')

#-------------------------------------------------------------------------
#  'TraitsTestHandler' class:
#-------------------------------------------------------------------------


class TraitsTestHandler(Handler):

    def object_enabled_changed(self, info):
        enabled = info.object.enabled
        for i in range(1, 63):
            getattr(info, 'f%d' % i).enabled = enabled

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


class TraitsTest(HasTraits):

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

    enabled = Bool(True)
    integer_text = Int(1)
    enumeration = Trait('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('cyan')
    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}',
         ('enabled', ),
         ('|<[Enumeration]', 'f1:enumeration[Simple]', '_',
          'f2:enumeration[Custom]@', '_',
          'f3:enumeration[Text]*', '_',
          'f4:enumeration[Readonly]~'),
         ('|<[Check List]', 'f5:check_list[Simple]', '_',
          'f6:check_list[Custom]@', '_',
          'f7:check_list[Text]*', '_',
          'f8:check_list[Readonly]~')
         ),
        ('|{Range}',
         ('|<[Float Range]', 'f9:float_range[Simple]', '_',
          'f10:float_range[Custom]@', '_',
          'f11:float_range[Text]*', '_',
          'f12:float_range[Readonly]~'),
         ('|<[Int Range]', 'f13:int_range[Simple]', '_',
          'f14:int_range[Custom]@', '_',
          'f15:int_range[Text]*', '_',
          'f16:int_range[Readonly]~'),
         ('|<[Int Range 2]', 'f17:int_range2[Simple]', '_',
          'f18:int_range2[Custom]@', '_',
          'f19:int_range2[Text]*', '_',
          'f20:int_range2[Readonly]~')
         ),
        ('|{Misc}',
         ('|<[Integer Text]', 'f21:integer_text[Simple]', '_',
          'f22:integer_text[Custom]@', '_',
          'f23:integer_text[Text]*', '_',
          'f24:integer_text[Readonly]~'),
         ('|<[Compound]', 'f25:compound[Simple]', '_',
          'f26:compound[Custom]@', '_',
          'f27:compound[Text]*', '_',
          'f28:compound[Readonly]~'),
         ('|<[Boolean]', 'f29:boolean[Simple]', '_',
          'f30:boolean[Custom]@', '_',
          'f31:boolean[Text]*', '_',
          'f32:boolean[Readonly]~')
         ),
        ('|{Color/Font}',
         ('|<[Color]', 'f33:color[Simple]', '_',
          'f34:color[Custom]@', '_',
          'f35:color[Text]*', '_',
          'f36:color[Readonly]~'),
         ('|<[Font]', 'f37:font[Simple]', '_',
          'f38:font[Custom]@', '_',
          'f39:font[Text]*', '_',
          'f40:font[Readonly]~')
         ),
        ('|{List}',
         ('|<[List]', 'f41:list[Simple]', '_',
          'f42:list[Custom]@', '_',
          'f43:list[Text]*', '_',
          'f44:list[Readonly]~')
         ),
        ('|{Button}',
         ('|<[Button]', 'f45:button[Simple]', '_',
          'f46:button[Custom]@'),
         #                                        'button[Text]*',
         #                                        'button[Readonly]~' ),
         ('|<[Image Enum]', 'f47:image_enum[Simple]', '_',
          'f48:image_enum[Custom]@', '_',
          'f49:image_enum[Text]*', '_',
          'f50:image_enum[Readonly]~'),
         ('|<[Instance]', 'f51:instance[Simple]', '_',
          'f52:instance[Custom]@', '_',
          'f53:instance[Text]*', '_',
          'f54:instance[Readonly]~'),
         ),
        ('|{File}',

         ('|<[File]', 'f55:file[Simple]', '_',
          'f56:file[Custom]@', '_',
          'f57:file[Text]*', '_',
          'f58:file[Readonly]~', ),
         ('|<[Directory]', 'f59:directory[Simple]', '_',
          'f60:directory[Custom]@', '_',
          'f61:directory[Text]*', '_',
          'f62:directory[Readonly]~')
         ),
        buttons=['Apply', 'Revert', 'Undo', 'OK'],
        handler=TraitsTestHandler()
    )

#-------------------------------------------------------------------------
#  '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='live')
        ui = self.object.edit_traits(kind='modal')
        ui = self.object.edit_traits(kind='nonmodal')
        ui = self.object.edit_traits(kind='wizard')
        self.SetTopWindow(ui.control)
        return True

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

TraitSheetApp(TraitsTest())