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
|
""" A Plot which uses ScaleSystems for its ticks.
"""
from enthought.traits.api import Any
from enthought.chaco.api import (DataRange2D, LinearMapper, LogMapper,
PlotGrid, Plot, PlotAxis)
from enthought.chaco.scales_tick_generator import ScalesTickGenerator
from enthought.chaco.scales.api import DefaultScale, LogScale, ScaleSystem
def add_default_axes(plot, orientation="normal", vtitle="", htitle=""):
"""
Creates left and bottom axes for a plot. Assumes that the index is
horizontal and value is vertical by default; set orientation to
something other than "normal" if they are flipped.
"""
if orientation in ("normal", "h"):
v_mapper = plot.value_mapper
h_mapper = plot.index_mapper
else:
v_mapper = plot.index_mapper
h_mapper = plot.value_mapper
yticks = ScalesTickGenerator()
left = PlotAxis(
orientation='left',
title=vtitle,
mapper=v_mapper,
component=plot,
tick_generator=yticks,
)
xticks = ScalesTickGenerator()
bottom = PlotAxis(
orientation='bottom',
title=htitle,
mapper=h_mapper,
component=plot,
tick_generator=xticks,
)
plot.underlays.append(left)
plot.underlays.append(bottom)
return left, bottom
class ScalyPlot(Plot):
x_axis = Any()
y_axis = Any()
x_ticks = Any()
y_ticks = Any()
linear_scale_factory = Any()
log_scale_factory = Any()
def _linear_scale_default(self):
return self._make_scale("linear")
def _log_scale_default(self):
return self._make_scale("log")
def _make_scale(self, scale_type="linear"):
""" Returns a new linear or log scale """
if scale_type == "linear":
if self.linear_scale_factory is not None:
return self.linear_scale_factory()
else:
return ScaleSystem(DefaultScale())
else:
if self.log_scale_factory is not None:
return self.log_scale_factory()
else:
return ScaleSystem(LogScale())
def _init_components(self):
# Since this is called after the HasTraits constructor, we have to make
# sure that we don't blow away any components that the caller may have
# already set.
if self.range2d is None:
self.range2d = DataRange2D()
if self.index_mapper is None:
if self.index_scale == "linear":
imap = LinearMapper(range=self.range2d.x_range)
else:
imap = LogMapper(range=self.range2d.x_range)
self.index_mapper = imap
if self.value_mapper is None:
if self.value_scale == "linear":
vmap = LinearMapper(range=self.range2d.y_range)
else:
vmap = LogMapper(range=self.range2d.y_range)
self.value_mapper = vmap
if self.x_ticks is None:
self.x_ticks = ScalesTickGenerator(scale=self._make_scale(self.index_scale))
if self.y_ticks is None:
self.y_ticks = ScalesTickGenerator(scale=self._make_scale(self.value_scale))
if self.x_grid is None:
self.x_grid = PlotGrid(mapper=self.x_mapper, orientation="vertical",
line_color="lightgray", line_style="dot",
component=self, tick_generator=self.x_ticks)
if self.y_grid is None:
self.y_grid = PlotGrid(mapper=self.y_mapper, orientation="horizontal",
line_color="lightgray", line_style="dot",
component=self, tick_generator=self.y_ticks)
if self.x_axis is None:
self.x_axis = PlotAxis(mapper=self.x_mapper, orientation="bottom",
component=self, tick_generator=self.x_ticks)
if self.y_axis is None:
self.y_axis = PlotAxis(mapper=self.y_mapper, orientation="left",
component=self, tick_generator=self.y_ticks)
def _index_scale_changed(self, old, new):
Plot._index_scale_changed(self, old, new)
# Now adjust the ScaleSystems.
self.x_ticks.scale = self._make_scale(self.index_scale)
def _value_scale_changed(self, old, new):
Plot._value_scale_changed(self, old, new)
# Now adjust the ScaleSystems.
self.y_ticks.scale = self._make_scale(self.value_scale)
|