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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
# coding: utf-8
# /*##########################################################################
#
# Copyright (c) 2016-2017 European Synchrotron Radiation Facility
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
# ###########################################################################*/
"""
This module define a hierarchical table view and model.
It allows to define many headers in the middle of a table.
The implementation hide the default header and allows to custom each cells
to became a header.
Row and column span is a concept of the view in a QTableView.
This implementation also provide a span property as part of the model of the
cell. A role is define to custom this information.
The view is updated everytime the model is reset to take care of the
changes of this information.
A default item delegate is used to redefine the paint of the cells.
"""
__authors__ = ["V. Valls"]
__license__ = "MIT"
__date__ = "07/04/2017"
from silx.gui import qt
class HierarchicalTableModel(qt.QAbstractTableModel):
"""
Abstract table model to provide more custom on row and column span and
headers.
Default headers are ignored and each cells can define IsHeaderRole and
SpanRole using the `data` function.
"""
SpanRole = qt.Qt.UserRole + 0
"""Role returning a tuple for number of row span then column span.
None and (1, 1) are neutral for the rendering.
"""
IsHeaderRole = qt.Qt.UserRole + 1
"""Role returning True is the identified cell is a header."""
UserRole = qt.Qt.UserRole + 2
"""First index of user defined roles"""
def headerData(self, section, orientation, role=qt.Qt.DisplayRole):
"""Returns the 0-based row or column index, for display in the
horizontal and vertical headers
In this case the headers are just ignored. Header information is part
of each cells.
"""
return None
class HierarchicalItemDelegate(qt.QStyledItemDelegate):
"""
Delegate item to take care of the rendering of the default table cells and
also the header cells.
"""
def __init__(self, parent=None):
"""
Constructor
:param qt.QObject parent: Parent of the widget
"""
qt.QStyledItemDelegate.__init__(self, parent)
def paint(self, painter, option, index):
"""Override the paint function to inject the style of the header.
:param qt.QPainter painter: Painter context used to displayed the cell
:param qt.QStyleOptionViewItem option: Control how the editor is shown
:param qt.QIndex index: Index of the data to display
"""
isHeader = index.data(role=HierarchicalTableModel.IsHeaderRole)
if isHeader:
span = index.data(role=HierarchicalTableModel.SpanRole)
span = 1 if span is None else span[1]
columnCount = index.model().columnCount()
if span == columnCount:
mainTitle = True
position = qt.QStyleOptionHeader.OnlyOneSection
else:
mainTitle = False
col = index.column()
if col == 0:
position = qt.QStyleOptionHeader.Beginning
elif col < columnCount - 1:
position = qt.QStyleOptionHeader.Middle
else:
position = qt.QStyleOptionHeader.End
opt = qt.QStyleOptionHeader()
opt.direction = option.direction
opt.text = index.data()
opt.textAlignment = qt.Qt.AlignCenter if mainTitle else qt.Qt.AlignVCenter
opt.direction = option.direction
opt.fontMetrics = option.fontMetrics
opt.palette = option.palette
opt.rect = option.rect
opt.state = option.state
opt.position = position
margin = -1
style = qt.QApplication.instance().style()
opt.rect = opt.rect.adjusted(margin, margin, -margin, -margin)
style.drawControl(qt.QStyle.CE_HeaderSection, opt, painter, None)
margin = 3
opt.rect = opt.rect.adjusted(margin, margin, -margin, -margin)
style.drawControl(qt.QStyle.CE_HeaderLabel, opt, painter, None)
else:
qt.QStyledItemDelegate.paint(self, painter, option, index)
class HierarchicalTableView(qt.QTableView):
"""A TableView which allow to display a `HierarchicalTableModel`."""
def __init__(self, parent=None):
"""
Constructor
:param qt.QWidget parent: Parent of the widget
"""
super(HierarchicalTableView, self).__init__(parent)
self.setItemDelegate(HierarchicalItemDelegate(self))
self.verticalHeader().setVisible(False)
self.horizontalHeader().setVisible(False)
def setModel(self, model):
"""Override the default function to connect the model to update
function"""
if self.model() is not None:
model.modelReset.disconnect(self.__modelReset)
super(HierarchicalTableView, self).setModel(model)
if self.model() is not None:
model.modelReset.connect(self.__modelReset)
self.__modelReset()
def __modelReset(self):
"""Update the model to take care of the changes of the span
information"""
self.clearSpans()
model = self.model()
for row in range(model.rowCount()):
for column in range(model.columnCount()):
index = model.index(row, column, qt.QModelIndex())
span = model.data(index, HierarchicalTableModel.SpanRole)
if span is not None and span != (1, 1):
self.setSpan(row, column, span[0], span[1])
|
