Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Avoid qudratic lookup when serializing tag browser state

Browse Source
This commit is contained in:
Kovid Goyal 2017-11-30 13:45:02 +05:30
parent cc410ac6da
commit ce05d2cc3b
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/calibre/gui2/tag_browser/model.py
View File

@ -1046,6 +1046,9 @@ class TagsModel(QAbstractItemModel): # {{{
id(internal_pointer))
return idx
def category_row_map(self):
return {category.category_key:row for row, category in enumerate(self.root_item.children)}
def index_for_category(self, name):
for row, category in enumerate(self.root_item.children):
if category.category_key == name:

13
src/calibre/gui2/tag_browser/view.py
View File

@ -13,7 +13,7 @@ from itertools import izip
from PyQt5.Qt import (
QStyledItemDelegate, Qt, QTreeView, pyqtSignal, QSize, QIcon, QApplication,
QMenu, QPoint, QToolTip, QCursor, QDrag, QRect,
QMenu, QPoint, QToolTip, QCursor, QDrag, QRect, QModelIndex,
QLinearGradient, QPalette, QColor, QPen, QBrush, QFont
)
@ -213,16 +213,17 @@ class TagsView(QTreeView): # {{{
def get_state(self):
state_map = {}
expanded_categories = []
row = -1
hide_empty_categories = self.model().prefs['tag_browser_hide_empty_categories']
crmap = self._model.category_row_map()
for category in self._model.category_nodes:
if (category.category_key in self.hidden_categories or
(hide_empty_categories and len(category.child_tags()) == 0)):
continue
row += 1
index = self._model.index_for_category(category.category_key)
if index is not None and self.isExpanded(index):
expanded_categories.append(category.category_key)
row = crmap.get(category.category_key)
if row is not None:
index = self._model.index(row, 0, QModelIndex())
if self.isExpanded(index):
expanded_categories.append(category.category_key)
states = [c.tag.state for c in category.child_tags()]
names = [(c.tag.name, c.tag.category) for c in category.child_tags()]
state_map[category.category_key] = dict(izip(names, states))