""" Defines the base class for mappings.
"""
# Major library imports
from numpy import array

# Enthought library imports
from traits.api import Event, HasTraits, Tuple


class AbstractMapper(HasTraits):
    """ Defines an abstract mapping from a region in input space to a
    region in output space.
    """

    # A generic "update" event that generally means that anything that relies
    # on this mapper for visual output should do a redraw or repaint.
    updated = Event

    # A tuple representing the minimum and maximum values of the domain (data
    # space).  The dimensionality of each value varies depending on the
    # dimensions of the mapper, so for 1D mappers these will be scalars, for
    # image and 2D mappers these will be tuples.
    domain_limits = Tuple(None, None)

    def map_screen(self, data_array):
        """ map_screen(data_array) -> screen_array

        Maps values from data space into screen space.
        """
        return

    def map_data(self, screen_val):
        """ map_data(screen_val) -> data_val

        Maps values from screen space into data space.
        """
        return

    def map_data_array(self, screen_vals):
        """ map_data_array(screen_vals) -> data_vals

        Maps an array of values from screen space into data space.
        By default, this method just loops over the points, calling map_data()
        on each one.  For vectorizable mapping functions, override this
        implmentation with a faster one.
        """
        return array([self.map_data(v) for v in screen_vals])


    #------------------------------------------------------------------------
    # Persistence-related methods
    #------------------------------------------------------------------------
    def __getstate__(self):
        state = super(AbstractMapper,self).__getstate__()
        for key in ['_cache_valid']:
            if state.has_key(key):
                del state[key]

        return state

    def _post_load(self):
        self._cache_valid = False
        self._range_changed(None, self.range)
        return

# EOF
