Source code for pyface.data_view.index_manager

# (C) Copyright 2005-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!
"""
Index Managers
==============

This module provides a number of classes for efficiently managing the
mapping between different ways of representing indices.  To do so, each
index manager provides an intermediate, opaque index object that is
suitable for use in these situations and is guaranteed to have a long
enough life that it will not change or be garbage collected while a C++
object has a reference to it.

The wx DataView classes expect to be given an integer id value that is
stable and can be used to return the parent reference id.

And Qt's ModelView system expects to be given a pointer to an object
that is long-lived (in particular, it will not be garbage-collected
during the lifetime of a QModelIndex) and which can be used to find
the parent object of the current object.

The default representation of an index from the point of view of the
data view infrastructure is a sequence of integers, giving the index at
each level of the hierarchy.  DataViewModel classes can then use these
indices to identify objects in the underlying data model.

There are three main classes defined in the module: AbstractIndexManager,
IntIndexManager, and TupleIndexManager.

AbstractIndexManager
    An ABC that defines the API

IntIndexManager
    An efficient index manager for non-hierarchical data, such as
    lists, tables and 2D arrays.

TupleIndexManager
    An index manager that handles non-hierarchical data while trying
    to be fast and memory efficient.

The two concrete subclasses should be sufficient for most cases, but advanced
users may create their own if for some reason the provided managers do not
work well for a particular situation.  Developers who implement this API
need to be mindful of the requirements on the lifetime and identity
constraints required by the various toolkit APIs.

"""

from abc import abstractmethod

from traits.api import ABCHasStrictTraits, Dict, Int, Tuple


#: The singular root object for all index managers.
Root = ()


[docs]class AbstractIndexManager(ABCHasStrictTraits): """ Abstract base class for index managers. """
[docs] @abstractmethod def create_index(self, parent, row): """ Given a parent index and a row number, create an index. The internal structure of the index should not matter to consuming code. However obejcts returned from this method should persist until the reset method is called. Parameters ---------- parent : index object The parent index object. row : int The position of the resuling index in the parent's children. Returns ------- index : index object The resulting opaque index object. Raises ------ IndexError Negative row values raise an IndexError exception. RuntimeError If asked to create a persistent index for a parent and row where that is not possible, a RuntimeError will be raised. """ raise NotImplementedError()
[docs] @abstractmethod def get_parent_and_row(self, index): """ Given an index object, return the parent index and row. Parameters ---------- index : index object The opaque index object. Returns ------- parent : index object The parent index object. row : int The position of the resuling index in the parent's children. Raises ------ IndexError If the Root object is passed as the index, this method will raise an IndexError, as it has no parent. """ raise NotImplementedError()
[docs] def from_sequence(self, indices): """ Given a sequence of indices, return the index object. The default implementation starts at the root and repeatedly calls create_index() to find the index at each level, returning the final value. Parameters ---------- indices : sequence of int The row location at each level of the hierarchy. Returns ------- index : index object The persistent index object associated with this sequence. Raises ------ RuntimeError If asked to create a persistent index for a sequence of indices where that is not possible, a RuntimeError will be raised. """ index = Root for row in indices: index = self.create_index(index, row) return index
[docs] def to_sequence(self, index): """ Given an index, return the corresponding sequence of row values. The default implementation repeatedly calls get_parent_and_row() to walk up the hierarchy and push the row values into the start of the sequence. Parameters ---------- index : index object The opaque index object. Returns ------- sequence : tuple of int The row location at each level of the hierarchy. """ result = () while index != Root: index, row = self.get_parent_and_row(index) result = (row,) + result return result
[docs] @abstractmethod def from_id(self, id): """ Given an integer id, return the corresponding index. Parameters ---------- id : int An integer object id value. Returns ------- index : index object The persistent index object associated with this id. """ raise NotImplementedError()
[docs] @abstractmethod def id(self, index): """ Given an index, return the corresponding id. Parameters ---------- index : index object The persistent index object. Returns ------- id : int The associated integer object id value. """ raise NotImplementedError()
[docs] def reset(self): """ Reset any caches and other state. Resettable traits in subclasses are indicated by having ``can_reset=True`` metadata. This is provided to allow toolkit code to clear caches to prevent memory leaks when working with very large tables. Care should be taken when calling this method, as Qt may crash if a QModelIndex is referencing an index that no longer has a reference in a cache. For some IndexManagers, particularly for those which are flat or static, reset() may do nothing. """ resettable_traits = self.trait_names(can_reset=True) self.reset_traits(resettable_traits)
[docs]class IntIndexManager(AbstractIndexManager): """ Efficient IndexManager for non-hierarchical indexes. This is a simple index manager for flat data structures. The index values returned are either the Root, or simple integers that indicate the position of the index as a child of the root. While it cannot handle nested data, this index manager can operate without having to perform any caching, and so is very efficient. """
[docs] def create_index(self, parent, row): """ Given a parent index and a row number, create an index. This should only ever be called with Root as the parent. Parameters ---------- parent : index object The parent index object. row : non-negative int The position of the resulting index in the parent's children. Returns ------- index : index object The resulting opaque index object. Raises ------ IndexError Negative row values raise an IndexError exception. RuntimeError If the parent is not the Root, a RuntimeError will be raised """ if row < 0: raise IndexError("Row must be non-negative. Got {}".format(row)) if parent != Root: raise RuntimeError( "{} cannot create persistent index value for {}.".format( self.__class__.__name__, (parent, row) ) ) return row
[docs] def get_parent_and_row(self, index): """ Given an index object, return the parent index and row. Parameters ---------- index : index object The opaque index object. Returns ------- parent : index object The parent index object. row : int The position of the resuling index in the parent's children. Raises ------ IndexError If the Root object is passed as the index, this method will raise an IndexError, as it has no parent. """ if index == Root: raise IndexError("Root index has no parent.") return Root, int(index)
[docs] def from_id(self, id): """ Given an integer id, return the corresponding index. Parameters ---------- id : int An integer object id value. Returns ------- index : index object The persistent index object associated with this id. """ if id == 0: return Root return id - 1
[docs] def id(self, index): """ Given an index, return the corresponding id. Parameters ---------- index : index object The persistent index object. Returns ------- id : int The associated integer object id value. """ if index == Root: return 0 return index + 1
[docs]class TupleIndexManager(AbstractIndexManager): #: A dictionary that maps tuples to the canonical version of the tuple. _cache = Dict(Tuple, Tuple, {Root: Root}, can_reset=True) #: A dictionary that maps ids to the canonical version of the tuple. _id_cache = Dict(Int, Tuple, {0: Root}, can_reset=True)
[docs] def create_index(self, parent, row): """ Given a parent index and a row number, create an index. Parameters ---------- parent : index object The parent index object. row : non-negative int The position of the resulting index in the parent's children. Returns ------- index : index object The resulting opaque index object. Raises ------ IndexError Negative row values raise an IndexError exception. """ if row < 0: raise IndexError("Row must be non-negative. Got {}".format(row)) index = (parent, row) canonical_index = self._cache.setdefault(index, index) self._id_cache[self.id(canonical_index)] = canonical_index return canonical_index
[docs] def get_parent_and_row(self, index): """ Given an index object, return the parent index and row. Parameters ---------- index : index object The opaque index object. Returns ------- parent : index object The parent index object. row : int The position of the resuling index in the parent's children. Raises ------ IndexError If the Root object is passed as the index, this method will raise an IndexError, as it has no parent. """ if index == Root: raise IndexError("Root index has no parent.") return index
[docs] def from_id(self, id): """ Given an integer id, return the corresponding index. Parameters ---------- id : int An integer object id value. Returns ------- index : index object The persistent index object associated with this id. """ return self._id_cache[id]
[docs] def id(self, index): """ Given an index, return the corresponding id. Parameters ---------- index : index object The persistent index object. Returns ------- id : int The associated integer object id value. """ if index == Root: return 0 canonical_index = self._cache.setdefault(index, index) return id(canonical_index)