1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#****************************************************************************
# unitlistview.py, provides a list view of available units
#
# ConvertAll, a units conversion program
# Copyright (C) 2014, Douglas W. Bell
#
# This is free software; you can redistribute it and/or modify it under the
# terms of the GNU General Public License, either Version 2 or any later
# version. This program is distributed in the hope that it will be useful,
# but WITTHOUT ANY WARRANTY. See the included LICENSE file for details.
#*****************************************************************************
from PyQt4 import QtCore, QtGui
import convertdlg
class UnitListView(QtGui.QTreeWidget):
"""ListView of units available.
"""
unitChanged = QtCore.pyqtSignal()
def __init__(self, unitGroup, unitRefNum, parent=None):
QtGui.QTreeWidget.__init__(self, parent)
self.unitGroup = unitGroup
self.unitRefNum = unitRefNum
self.buttonList = []
self.setRootIsDecorated(False)
self.setColumnCount(3)
self.setHeaderLabels([_('Unit Name'), _('Unit Type'), _('Comments')])
self.header().setStretchLastSection(False)
self.itemSelectionChanged.connect(self.replaceUnit)
self.loadUnits()
def loadUnits(self):
"""Load unit items.
"""
self.clear()
for name in convertdlg.ConvertDlg.unitData.sortedKeys:
UnitListViewItem(convertdlg.ConvertDlg.unitData[name],
self.unitRefNum, self)
for col in range(3):
self.resizeColumnToContents(col)
def relayChange(self):
"""Update list after buttons changed the unit group.
"""
self.updateSelection()
self.setFocus()
self.unitChanged.emit() # update unitEdit
def updateSelection(self):
"""Update list after change to line editor.
"""
self.blockSignals(True)
self.enableButtons(True)
self.clearSelection()
unit = self.unitGroup.currentUnit()
if unit and unit.equiv:
self.setCurrentItem(unit.viewLink[self.unitRefNum])
self.setItemSelected(unit.viewLink[self.unitRefNum], True)
else:
unit = self.unitGroup.currentPartialUnit()
if unit:
self.setCurrentItem(unit.viewLink[self.unitRefNum])
self.setItemSelected(unit.viewLink[self.unitRefNum], False)
else:
unit = self.unitGroup.currentSortPos()
self.enableButtons(False)
self.scrollToCenter(unit)
self.blockSignals(False)
def replaceUnit(self):
"""Replace current unit in response to a selection change.
"""
selectList = self.selectedItems()
if selectList:
selection = selectList[-1]
self.unitGroup.replaceCurrent(selection.unit)
self.unitChanged.emit() # update unitEdit
self.enableButtons(True)
def enableButtons(self, enable=True):
"""Enable unit modification buttons for valid unit.
"""
for button in self.buttonList:
button.setEnabled(enable)
def scrollToCenter(self, unit):
"""Scroll so given unit is in the center of the viewport.
"""
unitItem = unit.viewLink[self.unitRefNum]
index = self.indexOfTopLevelItem(unitItem)
itemHeight = self.visualItemRect(unitItem).height()
viewHeight = self.viewport().height()
bottomIndex = index + viewHeight / (2 * itemHeight)
bottomItem = self.topLevelItem(bottomIndex)
if not bottomItem:
bottomItem = self.topLevelItem(self.topLevelItemCount() - 1)
self.scrollToItem(bottomItem, QtGui.QAbstractItemView.PositionAtBottom)
def sizeHint(self):
"""Adjust width smaller.
"""
size = QtGui.QTreeWidget.sizeHint(self)
size.setWidth(self.columnWidth(0) + self.columnWidth(1) +
self.verticalScrollBar().sizeHint().width())
return size
class UnitListViewItem(QtGui.QTreeWidgetItem):
"""Item in list view, references unit.
"""
def __init__(self, unit, unitRefNum, parent=None):
QtGui.QTreeWidgetItem.__init__(self, parent)
self.unit = unit
unit.viewLink[unitRefNum] = self
self.setText(0, unit.description())
self.setText(1, unit.typeName)
self.setText(2, unit.comments[1])
|
