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
186
187
188
189
190
191
192
193
194
195
196
197
|
#-------------------------------------------------------------------------------
#
# A simple dual scatter-plot template defined as a test for the template
# package.
#
# Written by: David C. Morrill
# (based on the original cp.plot geo_scatter_plot.py file)
#
# Date: 08/01/2007
#
# (c) Copy 2007 by Enthought, Inc.
#
#-------------------------------------------------------------------------------
""" A simple dual scatter-plot template defined as a test for the template
package.
"""
#-------------------------------------------------------------------------------
# Imports:
#-------------------------------------------------------------------------------
from traits.api \
import Undefined
from traitsui.api \
import View, VGroup, Item, Label, Theme, TextEditor
from traitsui.wx.themed_slider_editor \
import ThemedSliderEditor
from traitsui.wx.themed_text_editor \
import ThemedTextEditor
from enable.api \
import ColorTrait
from chaco.api \
import HPlotContainer
from chaco.scatter_markers \
import marker_trait
from apptools.template.api \
import Template, TRange, TStr, TInstance, TDerived
from enable_editor \
import EnableEditor
from scatter_plot \
import ScatterPlot
#-------------------------------------------------------------------------------
# Trait definitions:
#-------------------------------------------------------------------------------
# Template color trait:
TColor = ColorTrait( template = 'copy' )
#-------------------------------------------------------------------------------
# 'ScatterPlot2' class:
#-------------------------------------------------------------------------------
class ScatterPlot2 ( Template ):
#-- Template Traits --------------------------------------------------------
# The title of the plot:
title = TStr( 'Dual Scatter Plots' )
# The type of marker to use. This is a mapped trait using strings as the
# keys:
marker = marker_trait( template = 'copy', event = 'update' )
# The pixel size of the marker (doesn't include the thickness of the
# outline):
marker_size = TRange( 1, 5, 1, event = 'update' )
# The thickness, in pixels, of the outline to draw around the marker. If
# this is 0, no outline will be drawn.
line_width = TRange( 0.0, 5.0, 1.0 )
# The fill color of the marker:
color = TColor( 'red', event = 'update' )
# The color of the outline to draw around the marker
outline_color = TColor( 'black', event = 'update' )
# The amount of space between plots:
spacing = TRange( 0.0, 20.0, 0.0 )
# The contained scatter plots:
scatter_plot_1 = TInstance( ScatterPlot, () )
scatter_plot_2 = TInstance( ScatterPlot, () )
#-- Derived Traits ---------------------------------------------------------
plot = TDerived
#-- Traits UI Views --------------------------------------------------------
# The scatter plot view:
template_view = View(
VGroup(
Item( 'title',
show_label = False,
style = 'readonly',
editor = ThemedTextEditor(
theme = Theme( '@GBB', alignment = 'center' ) )
),
Item( 'plot',
show_label = False,
resizable = True,
editor = EnableEditor(),
item_theme = Theme( '@GF5', margins = 0 )
)
),
resizable = True
)
# The scatter plot options view:
options_view = View(
VGroup(
VGroup(
Label( 'Scatter Plot Options',
item_theme = Theme( '@GBB', alignment = 'center' ) ),
show_labels = False
),
VGroup(
Item( 'title', editor = TextEditor() ),
Item( 'marker' ),
Item( 'marker_size', editor = ThemedSliderEditor() ),
Item( 'line_width',
label = 'Line Width',
editor = ThemedSliderEditor() ),
Item( 'spacing', editor = ThemedSliderEditor() ),
Item( 'color', label = 'Fill Color' ),
Item( 'outline_color', label = 'Outline Color' ),
group_theme = Theme( '@GF5', margins = ( -5, -1 ) ),
item_theme = Theme( '@G0B', margins = 0 )
)
)
)
#-- ITemplate Interface Implementation -------------------------------------
def activate_template ( self ):
""" Converts all contained 'TDerived' objects to real objects using the
template traits of the object. This method must be overridden in
subclasses.
Returns
-------
None
"""
plots = [ p for p in [ self.scatter_plot_1.plot,
self.scatter_plot_2.plot ] if p is not None ]
if len( plots ) == 2:
self.plot = HPlotContainer( spacing = self.spacing )
self.plot.add( *plots )
elif len( plots ) == 1:
self.plot = plots[0]
#-- Default Values ---------------------------------------------------------
def _scatter_plot_1_default ( self ):
""" Returns the default value for the first scatter plot.
"""
result = ScatterPlot()
result.index.description = 'Shared Plot Index'
result.value.description += ' 1'
return result
def _scatter_plot_2_default ( self ):
""" Returns the default value for the second scatter plot.
"""
result = ScatterPlot( index = self.scatter_plot_1.index )
result.value.description += ' 2'
result.value.optional = True
return result
#-- Trait Event Handlers ---------------------------------------------------
def _update_changed ( self, name, old, new ):
""" Handles a plot option being changed.
"""
setattr( self.scatter_plot_1, name, new )
setattr( self.scatter_plot_2, name, new )
self.plot = Undefined
def _spacing_changed ( self, spacing ):
""" Handles the spacing between plots being changed.
"""
self.plot = Undefined
|
