Skip to content

Commit

Permalink
ENH: Use a common, backend-independent implementation of ColorColumn.
Browse files Browse the repository at this point in the history
  • Loading branch information
rkern committed May 9, 2012
1 parent 729c5d3 commit 55bed18
Show file tree
Hide file tree
Showing 3 changed files with 70 additions and 48 deletions.
2 changes: 1 addition & 1 deletion integrationtests/ui/table_editor_color_test.py
Original file line number Original file line Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
from traitsui.api \ from traitsui.api \
import View, Item, TableEditor import View, Item, TableEditor


from traitsui.wx.color_column \ from traitsui.color_column \
import ColorColumn import ColorColumn


from enable.api \ from enable.api \
Expand Down
65 changes: 65 additions & 0 deletions traitsui/color_column.py
Original file line number Original file line Diff line number Diff line change
@@ -0,0 +1,65 @@
#------------------------------------------------------------------------------
#
# Copyright (c) 2005, Enthought, Inc.
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in enthought/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!
#
# Author: David C. Morrill
#
#------------------------------------------------------------------------------

""" Table column object for Color traits.
"""

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

from traitsui.table_column \
import ObjectColumn

#-------------------------------------------------------------------------------
# 'ColorColumn' class:
#-------------------------------------------------------------------------------

class ColorColumn ( ObjectColumn ):
""" Table column object for Color traits. """

#-- ObjectColumn Overrides -----------------------------------------------------

def get_cell_color ( self, object ):
""" Returns the cell background color for the column for a specified
object.
"""
color_values = getattr( object, self.name + '_' )
if type( color_values ) is tuple:
tk_color = self._as_int_rgb_tuple( color_values )
else:
tk_color = super( ColorColumn, self ).get_cell_color( object )
return tk_color

def get_value ( self, object ):
""" Gets the value of the column for a specified object.
"""
value = getattr( self.get_object( object ), self.name )
if type( value ) is tuple:
value = "(%3d, %3d, %3d)" % self._as_int_rgb_tuple( value[:-1] )
elif type( value ) is not str:
value = str( value )

return value

#-- Private Methods ------------------------------------------------------------

def _as_int_rgb_tuple ( self, color_values ):
""" Returns object color as RGB integers. """
return ( int( 255 * color_values[0] ),
int( 255 * color_values[1] ),
int( 255 * color_values[2] ) )

51 changes: 4 additions & 47 deletions traitsui/wx/color_column.py
Original file line number Original file line Diff line number Diff line change
Expand Up @@ -17,52 +17,9 @@
""" Table column object for Color traits. """ Table column object for Color traits.
""" """


#------------------------------------------------------------------------------- import warnings
# Imports: warnings.warn("traitsui.wx.color_column is deprecated. Use the "
#------------------------------------------------------------------------------- "backend-independent implementation in traitsui.color_column")


from wx \ from traitsui.color_column import ColorColumn
import Colour as WxColour

from traitsui.table_column \
import ObjectColumn

#-------------------------------------------------------------------------------
# 'ColorColumn' class:
#-------------------------------------------------------------------------------

class ColorColumn ( ObjectColumn ):
""" Table column object for Color traits. """

#-- ObjectColumn Overrides -----------------------------------------------------

def get_cell_color ( self, object ):
""" Returns the cell background color for the column for a specified
object.
"""
color_values = getattr( object, self.name + '_' )
if type( color_values ) is tuple:
wxcolor = WxColour( *self._as_int_rgb_tuple( color_values ) )
else:
wxcolor = super( ColorColumn, self ).get_cell_color( object )
return wxcolor

def get_value ( self, object ):
""" Gets the value of the column for a specified object.
"""
value = getattr( self.get_object( object ), self.name )
if type( value ) is tuple:
value = "(%3d, %3d, %3d)" % self._as_int_rgb_tuple( value[:-1] )
elif type( value ) is not str:
value = str( value )

return value

#-- Private Methods ------------------------------------------------------------

def _as_int_rgb_tuple ( self, color_values ):
""" Returns object color as RGB integers. """
return ( int( 255 * color_values[0] ),
int( 255 * color_values[1] ),
int( 255 * color_values[2] ) )


0 comments on commit 55bed18

Please sign in to comment.