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
173
174
175
176
177
178
179
180
181
182
183
184
185
|
# (C) Copyright 2004-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!
"""
This demo illustrates use of TreeNodeRenderers for displaying more complex
contents inside the cells of a TreeEditor.
To run this demonstration successfully, you must have **NumPy**
(``numpy``) installed.
"""
from random import choice, uniform
import colorsys
import numpy as np
from pyface.qt import QtCore, QtGui, qt_api
from traits.api import Array, Float, HasTraits, Instance, Int, List, Str
from traitsui.api import TreeEditor, TreeNode, UItem, View, RGBColor
from traitsui.tree_node_renderer import AbstractTreeNodeRenderer
from traitsui.qt.tree_editor import WordWrapRenderer
class MyDataElement(HasTraits):
"""A node in a tree of data."""
#: Some text to display.
text = Str()
#: A random walk to plot.
data = Array
#: A color to show as an icon.
color = RGBColor
def _data_default(self):
return np.random.standard_normal((1000,)).cumsum()
def _color_default(self):
return colorsys.hsv_to_rgb(uniform(0.0, 1.0), 1.0, 1.0)
class MyData(HasTraits):
"""The root node for a tree of data."""
#: The name of the root node.
name = Str('Rooty McRootface')
#: The elements contained in the root node.
elements = List(Instance(MyDataElement))
def _elements_default(self):
DATA_ELEMENTS = (
'I live on\nmultiple\nlines!',
'Foo\nBar',
'Baz',
'Qux',
'z ' * 20,
__doc__,
)
return [MyDataElement(text=choice(DATA_ELEMENTS)) for _ in range(5)]
class SparklineRenderer(AbstractTreeNodeRenderer):
"""Renderer that draws sparklines into a cell."""
#: This renderer handles all rendering.
handles_all = True
#: This renderer handles text rendering (there is none).
handles_text = True
#: The scale for y-values.
y_scale = Float(1.0)
#: The extra border applied by Qt internally
# XXX get this dynamically from Qt? How?
extra_space = Int(8)
def paint(self, editor, node, column, object, paint_context):
painter, option, index = paint_context
data = self.get_data(object)
xs = (
np.linspace(0, option.rect.width(), len(data)) + option.rect.left()
)
ys = (data.max() - data) / self.y_scale + option.rect.top()
height = option.rect.height()
plot_height = ys.ptp()
extra = height - plot_height
if bool(option.displayAlignment & QtCore.Qt.AlignmentFlag.AlignVCenter):
ys += extra / 2
elif bool(option.displayAlignment & QtCore.Qt.Bottom):
ys += extra
if bool(option.state & QtGui.QStyle.StateFlag.State_Selected):
painter.fillRect(option.rect, option.palette.highlight())
points = [QtCore.QPointF(x, y) for x, y in zip(xs, ys)]
old_pen = painter.pen()
if bool(option.state & QtGui.QStyle.StateFlag.State_Selected):
painter.setPen(QtGui.QPen(option.palette.highlightedText(), 0))
try:
if qt_api.startswith('pyside'):
painter.drawPolyline(points)
else:
painter.drawPolyline(*points)
finally:
painter.setPen(old_pen)
return None
def get_data(self, object):
return object.data
def size(self, editor, node, column, object, size_context):
data = self.get_data(object)
return (100, data.ptp() / self.y_scale + self.extra_space)
class SparklineTreeNode(TreeNode):
"""A TreeNode that renders sparklines in column index 1"""
#: static instance of SparklineRenderer
#: (it has no state, so this is fine)
sparkline_renderer = SparklineRenderer()
#: static instance of WordWrapRenderer
#: (it has no state, so this is fine)
word_wrap_renderer = WordWrapRenderer()
def get_renderer(self, object, column=0):
if column == 1:
return self.sparkline_renderer
else:
return self.word_wrap_renderer
def get_icon(self, object, is_expanded):
return object.color
class SparklineTreeView(HasTraits):
"""Class that views the data with sparklines."""
#: The root of the tree.
root = Instance(MyData, args=())
traits_view = View(
UItem(
'root',
editor=TreeEditor(
nodes=[
TreeNode(
node_for=[MyData],
children='elements',
label='name',
),
SparklineTreeNode(
node_for=[MyDataElement],
auto_open=True,
label='text',
),
],
column_headers=["The Tree View", "The Sparklines"],
hide_root=False,
editable=False,
),
),
resizable=True,
width=400,
height=300,
)
if __name__ == '__main__':
SparklineTreeView().configure_traits()
|
