From 6e29949ca3bdb6adb492fdbf2bce56ef7d1fcae5 Mon Sep 17 00:00:00 2001 From: Don Allingham Date: Mon, 29 Jan 2007 19:13:15 +0000 Subject: [PATCH] * src/DisplayModels/_BaseModel.py: Refactor out node/path maps * various: remove sets import svn: r8011 --- gramps2/ChangeLog | 2 + gramps2/src/DataViews/_PersonView.py | 5 - gramps2/src/DataViews/_RelationView.py | 5 - gramps2/src/DisplayModels/_BaseModel.py | 191 ++++++++++++------ gramps2/src/DisplayModels/_EventModel.py | 5 - gramps2/src/DisplayModels/_FamilyModel.py | 5 - gramps2/src/DisplayModels/_MediaModel.py | 5 - gramps2/src/DisplayModels/_PlaceModel.py | 5 - gramps2/src/DisplayModels/_RepositoryModel.py | 5 - gramps2/src/DisplayModels/_SourceModel.py | 5 - gramps2/src/Editors/_EditFamily.py | 11 - gramps2/src/Editors/_EditPerson.py | 5 - gramps2/src/GrampsDb/_GrampsBSDDB.py | 7 - gramps2/src/GrampsDb/_GrampsDbBase.py | 5 - gramps2/src/GrampsDb/_ReadGedcom.py | 6 - gramps2/src/GrampsDb/_WriteGedcom.py | 4 - gramps2/src/Selectors/_BaseSelector.py | 10 - gramps2/src/Utils.py | 5 - gramps2/src/plugins/Check.py | 4 - gramps2/src/plugins/CountAncestors.py | 4 - gramps2/src/plugins/NarrativeWeb.py | 5 - gramps2/src/plugins/RelCalc.py | 4 - gramps2/src/plugins/Verify.py | 4 - 23 files changed, 128 insertions(+), 179 deletions(-) diff --git a/gramps2/ChangeLog b/gramps2/ChangeLog index 25a987e6d..3631ae537 100644 --- a/gramps2/ChangeLog +++ b/gramps2/ChangeLog @@ -1,4 +1,6 @@ 2007-01-29 Don Allingham + * src/DisplayModels/_BaseModel.py: Refactor out node/path maps + * various: remove sets import * NEWS: update * configure.in: update to 2.2.6 * src/ReportBase/_ReportOptions.py: fix landscape saving. diff --git a/gramps2/src/DataViews/_PersonView.py b/gramps2/src/DataViews/_PersonView.py index 62f1570c2..a2548ebaa 100644 --- a/gramps2/src/DataViews/_PersonView.py +++ b/gramps2/src/DataViews/_PersonView.py @@ -36,11 +36,6 @@ __revision__ = "$Revision$" from gettext import gettext as _ import cPickle as pickle -try: - set() -except NameError: - from sets import Set as set - #------------------------------------------------------------------------- # # gtk diff --git a/gramps2/src/DataViews/_RelationView.py b/gramps2/src/DataViews/_RelationView.py index eddc41f96..4a631495c 100644 --- a/gramps2/src/DataViews/_RelationView.py +++ b/gramps2/src/DataViews/_RelationView.py @@ -34,11 +34,6 @@ __revision__ = "$Revision$" from gettext import gettext as _ import cgi -try: - set() -except NameError: - from sets import Set as set - #------------------------------------------------------------------------- # # GTK/Gnome modules diff --git a/gramps2/src/DisplayModels/_BaseModel.py b/gramps2/src/DisplayModels/_BaseModel.py index 43fe4f974..eb5c45425 100644 --- a/gramps2/src/DisplayModels/_BaseModel.py +++ b/gramps2/src/DisplayModels/_BaseModel.py @@ -25,10 +25,6 @@ # #------------------------------------------------------------------------- import locale -try: - set() -except: - from sets import Set as set #------------------------------------------------------------------------- # @@ -44,6 +40,94 @@ import gtk #------------------------------------------------------------------------- from Filters import SearchFilter +#------------------------------------------------------------------------- +# +# NodeMap +# +#------------------------------------------------------------------------- +class NodeMap: + """ + Provides the Path to Iter mappings for a TreeView model. The implementation + provides a list of nodes and a dictionary of handles. The datalist provides + the path (index) to iter (handle) mapping, while the the indexmap provides + the handle to index mappings + """ + + def __init__(self): + """ + Create a new instance, clearing the datalist and indexmap + """ + self.data_list = [] + self.index_map = {} + + def set_path_map(self, dlist): + """ + Takes a list of handles and builds the index map from it. + """ + self.data_list = dlist + i = 0 + self.index_map = {} + for key in self.data_list: + self.index_map[key] = i + i +=1 + + def clear_map(self): + """ + Clears out the data_list and the index_map + """ + self.data_list = [] + self.index_map = {} + + def get_path(self, handle): + """ + Returns the path from the passed handle. This is accomplished by + indexing into the index_map to get the index (path) + """ + return self.index_map.get(handle) + + def get_handle(self, path): + """ + Returns the handle from the path. The path is assumed to be an integer. + This is accomplished by indexing into the data_list + """ + return self.data_list[path] + + def delete_by_index(self, index): + """ + Deletes the item at the specified path, then rebuilds the index_map, + subtracting one from each item greater than the deleted index. + """ + handle = self.data_list[index] + del self.data_list[index] + del self.index_map[handle] + + for key in self.index_map: + if self.index_map[key] > index: + self.index_map[key] -= 1 + + def find_next_handle(self, handle): + """ + Finds the next handle based off the passed handle. This is accomplished + by finding the index of associated with the handle, adding one to find + the next index, then finding the handle associated with the next index. + """ + try: + return self.data_list[self.index_map.get(handle)+1] + except IndexError: + return None + + def __len__(self): + """ + Returns the number of entries in the map. + """ + return len(self.data_list) + + def get_first_handle(self): + """ + Returns the first handle in the map. + """ + return self.data_list[0] + #------------------------------------------------------------------------- # # BaseModel @@ -68,6 +152,8 @@ class BaseModel(gtk.GenericTreeModel): self.sort_col = scol self.skip = skip + self.node_map = NodeMap() + if search: if search[0]: self.search = search[1] @@ -95,98 +181,76 @@ class BaseModel(gtk.GenericTreeModel): def sort_keys(self): cursor = self.gen_cursor() - self.sarray = [] + self.sort_data = [] data = cursor.next() while data: key = locale.strxfrm(self.sort_func(data[1])) - self.sarray.append((key,data[0])) + self.sort_data.append((key,data[0])) data = cursor.next() cursor.close() - self.sarray.sort(reverse=self.reverse) + self.sort_data.sort(reverse=self.reverse) - return [ x[1] for x in self.sarray ] + return [ x[1] for x in self.sort_data ] def _rebuild_search(self,ignore=None): if self.db.is_open(): if self.search and self.search.text: - self.datalist = [h for h in self.sort_keys()\ - if self.search.match(h) and \ - h not in self.skip and h != ignore] + dlist = [h for h in self.sort_keys()\ + if self.search.match(h) and \ + h not in self.skip and h != ignore] else: - self.datalist = [h for h in self.sort_keys() \ - if h not in self.skip and h != ignore] - i = 0 - self.indexlist = {} - for key in self.datalist: - self.indexlist[key] = i - i += 1 + dlist = [h for h in self.sort_keys() \ + if h not in self.skip and h != ignore] + self.node_map.set_path_map(dlist) else: - self.datalist = [] - self.indexlist = {} + self.node_map.clear_map() def _rebuild_filter(self, ignore=None): if self.db.is_open(): if self.search: - self.datalist = self.search.apply(self.db, - [ k for k in self.sort_keys()\ - if k != ignore]) + dlist = self.search.apply(self.db, + [ k for k in self.sort_keys()\ + if k != ignore]) else: - self.datalist = [ k for k in self.sort_keys() \ - if k != ignore ] + dlist = [ k for k in self.sort_keys() \ + if k != ignore ] - i = 0 - self.indexlist = {} - for key in self.datalist: - if key not in self.skip: - self.indexlist[key] = i - i += 1 + self.node_map.set_path_map(dlist) else: - self.datalist = [] - self.indexlist = {} + self.node_map.clear_map() def add_row_by_handle(self,handle): if self.search and self.search.match(handle): data = self.map(handle) key = locale.strxfrm(self.sort_func(data)) - self.sarray.append((key,handle)) - self.sarray.sort(reverse=self.reverse) - self.datalist = [ x[1] for x in self.sarray ] + self.sort_data.append((key,handle)) + self.sort_data.sort(reverse=self.reverse) + self.node_map.set_path_map([ x[1] for x in self.sort_data ]) - i = 0 - self.indexlist = {} - for key in self.datalist: - self.indexlist[key] = i - i += 1 - - index = self.indexlist.get(handle) + index = self.node_map.get_path(handle) if index != None: node = self.get_iter(index) self.row_inserted(index, node) def delete_row_by_handle(self,handle): - index = self.indexlist[handle] + index = self.node_map.get_path(handle) # remove from sort array i = 0 - for (key, node) in self.sarray: + for (key, node) in self.sort_data: if handle == node: - del self.sarray[i] + del self.sort_data[i] break i += 1 - del self.datalist[index] - del self.indexlist[handle] - - for key in self.indexlist: - if self.indexlist[key] > index: - self.indexlist[key] -= 1 + self.node_map.delete_by_index(index) self.row_deleted(index) def update_row_by_handle(self,handle): - index = self.indexlist[handle] + index = self.node_map.get_path(handle) node = self.get_iter(index) self.row_changed(index,node) @@ -200,7 +264,7 @@ class BaseModel(gtk.GenericTreeModel): def on_get_path(self, node): '''returns the tree path (a tuple of indices at the various levels) for a particular node.''' - return self.indexlist[node] + return self.node_map.get_path(node) def on_get_column_type(self,index): if index == self.tooltip_column: @@ -209,8 +273,8 @@ class BaseModel(gtk.GenericTreeModel): def on_get_iter(self, path): try: - return self.datalist[path[0]] - except IndexError: + return self.node_map.get_handle(path[0]) + except: return None def on_get_value(self,node,col): @@ -224,31 +288,28 @@ class BaseModel(gtk.GenericTreeModel): def on_iter_next(self, node): '''returns the next node at this level of the tree''' - try: - return self.datalist[self.indexlist[node]+1] - except IndexError: - return None + return self.node_map.find_next_handle(node) def on_iter_children(self,node): """Return the first child of the node""" - if node == None and self.datalist: - return self.datalist[0] + if node == None and len(self.node_map): + return self.node_map.get_first_handle() return None def on_iter_has_child(self, node): '''returns true if this node has children''' if node == None: - return len(self.datalist) > 0 + return len(self.node_map) > 0 return False def on_iter_n_children(self,node): if node == None: - return len(self.datalist) + return len(self.node_map) return 0 def on_iter_nth_child(self,node,n): if node == None: - return self.datalist[n] + return self.node_map.get_handle(n) return None def on_iter_parent(self, node): diff --git a/gramps2/src/DisplayModels/_EventModel.py b/gramps2/src/DisplayModels/_EventModel.py index b937ec5bd..a6897b47d 100644 --- a/gramps2/src/DisplayModels/_EventModel.py +++ b/gramps2/src/DisplayModels/_EventModel.py @@ -28,11 +28,6 @@ import time import logging log = logging.getLogger(".") -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GNOME/GTK modules diff --git a/gramps2/src/DisplayModels/_FamilyModel.py b/gramps2/src/DisplayModels/_FamilyModel.py index 9cc3c484b..043db32fc 100644 --- a/gramps2/src/DisplayModels/_FamilyModel.py +++ b/gramps2/src/DisplayModels/_FamilyModel.py @@ -28,11 +28,6 @@ import time import logging log = logging.getLogger(".") -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GNOME/GTK modules diff --git a/gramps2/src/DisplayModels/_MediaModel.py b/gramps2/src/DisplayModels/_MediaModel.py index f2304587e..4cfe7ef0e 100644 --- a/gramps2/src/DisplayModels/_MediaModel.py +++ b/gramps2/src/DisplayModels/_MediaModel.py @@ -29,11 +29,6 @@ from gettext import gettext as _ import logging log = logging.getLogger(".") -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GNOME/GTK modules diff --git a/gramps2/src/DisplayModels/_PlaceModel.py b/gramps2/src/DisplayModels/_PlaceModel.py index 82e2ee6e1..889a9fc94 100644 --- a/gramps2/src/DisplayModels/_PlaceModel.py +++ b/gramps2/src/DisplayModels/_PlaceModel.py @@ -28,11 +28,6 @@ import time import logging log = logging.getLogger(".") -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GNOME/GTK modules diff --git a/gramps2/src/DisplayModels/_RepositoryModel.py b/gramps2/src/DisplayModels/_RepositoryModel.py index a1f249f28..11c7f6ce5 100644 --- a/gramps2/src/DisplayModels/_RepositoryModel.py +++ b/gramps2/src/DisplayModels/_RepositoryModel.py @@ -27,11 +27,6 @@ import logging log = logging.getLogger(".") -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GNOME/GTK modules diff --git a/gramps2/src/DisplayModels/_SourceModel.py b/gramps2/src/DisplayModels/_SourceModel.py index 2a06d682f..7fbff78b2 100644 --- a/gramps2/src/DisplayModels/_SourceModel.py +++ b/gramps2/src/DisplayModels/_SourceModel.py @@ -28,11 +28,6 @@ import time import logging log = logging.getLogger(".") -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GNOME/GTK modules diff --git a/gramps2/src/Editors/_EditFamily.py b/gramps2/src/Editors/_EditFamily.py index 643b5db8a..5d192d306 100644 --- a/gramps2/src/Editors/_EditFamily.py +++ b/gramps2/src/Editors/_EditFamily.py @@ -28,17 +28,6 @@ from gettext import gettext as _ import cPickle as pickle -#------------------------------------------------------------------------- -# -# 2.4 provides a built in set. We want to use this, but need to handle -# older versions of python as well -# -#------------------------------------------------------------------------- -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # enable logging for error handling diff --git a/gramps2/src/Editors/_EditPerson.py b/gramps2/src/Editors/_EditPerson.py index 1b9329120..d815fb5d3 100644 --- a/gramps2/src/Editors/_EditPerson.py +++ b/gramps2/src/Editors/_EditPerson.py @@ -43,11 +43,6 @@ from gettext import gettext as _ #------------------------------------------------------------------------- import gtk -try: - set() -except NameError: - from sets import Set as set - #------------------------------------------------------------------------- # # gramps modules diff --git a/gramps2/src/GrampsDb/_GrampsBSDDB.py b/gramps2/src/GrampsDb/_GrampsBSDDB.py index 801fc58e5..fd6b496ce 100644 --- a/gramps2/src/GrampsDb/_GrampsBSDDB.py +++ b/gramps2/src/GrampsDb/_GrampsBSDDB.py @@ -39,13 +39,6 @@ from bsddb import dbshelve, db import logging log = logging.getLogger(".GrampsDb") -# hack to use native set for python2.4 -# and module sets for earlier pythons -try: - set() -except NameError: - from sets import Set as set - #------------------------------------------------------------------------- # # Gramps modules diff --git a/gramps2/src/GrampsDb/_GrampsDbBase.py b/gramps2/src/GrampsDb/_GrampsDbBase.py index 9755f1b64..7cefcd3ea 100644 --- a/gramps2/src/GrampsDb/_GrampsDbBase.py +++ b/gramps2/src/GrampsDb/_GrampsDbBase.py @@ -42,11 +42,6 @@ from sys import maxint from bsddb import db from gettext import gettext as _ -try: - set() -except NameError: - from sets import Set as set - import logging log = logging.getLogger(".GrampsDb") diff --git a/gramps2/src/GrampsDb/_ReadGedcom.py b/gramps2/src/GrampsDb/_ReadGedcom.py index 2c29cf430..db94d911b 100644 --- a/gramps2/src/GrampsDb/_ReadGedcom.py +++ b/gramps2/src/GrampsDb/_ReadGedcom.py @@ -34,12 +34,6 @@ import time from bsddb import db from gettext import gettext as _ -# and module sets for earlier pythons -try: - set() -except NameError: - from sets import Set as set - #------------------------------------------------------------------------ # # Set up logging diff --git a/gramps2/src/GrampsDb/_WriteGedcom.py b/gramps2/src/GrampsDb/_WriteGedcom.py index 943aa9bcc..13a0f08c0 100644 --- a/gramps2/src/GrampsDb/_WriteGedcom.py +++ b/gramps2/src/GrampsDb/_WriteGedcom.py @@ -31,10 +31,6 @@ import os import time import re import shutil -try: - set() -except: - from sets import Set as set import logging log = logging.getLogger(".WriteGedcom") diff --git a/gramps2/src/Selectors/_BaseSelector.py b/gramps2/src/Selectors/_BaseSelector.py index 2c2acf63f..0f2199068 100644 --- a/gramps2/src/Selectors/_BaseSelector.py +++ b/gramps2/src/Selectors/_BaseSelector.py @@ -20,16 +20,6 @@ # $Id$ -#------------------------------------------------------------------------- -# -# Python modules -# -#------------------------------------------------------------------------- -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GTK/Gnome modules diff --git a/gramps2/src/Utils.py b/gramps2/src/Utils.py index 3c18fb32b..118ababe8 100644 --- a/gramps2/src/Utils.py +++ b/gramps2/src/Utils.py @@ -33,11 +33,6 @@ import time from gettext import gettext as _ -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------- # # GNOME/GTK diff --git a/gramps2/src/plugins/Check.py b/gramps2/src/plugins/Check.py index afe26a37e..6ca0e9736 100644 --- a/gramps2/src/plugins/Check.py +++ b/gramps2/src/plugins/Check.py @@ -30,10 +30,6 @@ import os import sys import cStringIO -try: - set() -except: - from sets import Set as set from gettext import gettext as _ diff --git a/gramps2/src/plugins/CountAncestors.py b/gramps2/src/plugins/CountAncestors.py index d91633c92..35c1a311f 100644 --- a/gramps2/src/plugins/CountAncestors.py +++ b/gramps2/src/plugins/CountAncestors.py @@ -31,10 +31,6 @@ import os from gettext import gettext as _ from math import pow -try: - set() -except NameError: - from sets import Set as set #------------------------------------------------------------------------ # diff --git a/gramps2/src/plugins/NarrativeWeb.py b/gramps2/src/plugins/NarrativeWeb.py index 593db57bc..d6529f5cd 100644 --- a/gramps2/src/plugins/NarrativeWeb.py +++ b/gramps2/src/plugins/NarrativeWeb.py @@ -40,11 +40,6 @@ from gettext import gettext as _ from cStringIO import StringIO from textwrap import TextWrapper -try: - set() -except: - from sets import Set as set - #------------------------------------------------------------------------ # # Set up logging diff --git a/gramps2/src/plugins/RelCalc.py b/gramps2/src/plugins/RelCalc.py index a6f25e297..f56f6b7e4 100644 --- a/gramps2/src/plugins/RelCalc.py +++ b/gramps2/src/plugins/RelCalc.py @@ -29,10 +29,6 @@ #------------------------------------------------------------------------- import os from gettext import gettext as _ -try: - set() -except NameError: - from sets import Set as set #------------------------------------------------------------------------- # diff --git a/gramps2/src/plugins/Verify.py b/gramps2/src/plugins/Verify.py index 6c8c3826a..ee753f23c 100644 --- a/gramps2/src/plugins/Verify.py +++ b/gramps2/src/plugins/Verify.py @@ -34,10 +34,6 @@ import os from gettext import gettext as _ import cPickle import md5 -try: - set() -except NameError: - from sets import Set as set import Errors #------------------------------------------------------------------------