/usr/lib/python2.7/dist-packages/chaco/contour_line_plot.py is in python-chaco 4.4.1-1.2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 198 199 200 201 202 203 204 205 206 207 208 | """ Defines the ContourLinePlot class.
"""
from __future__ import with_statement
# Major library imports
from numpy import array, linspace, meshgrid, transpose
# Enthought library imports
from enable.api import LineStyle
from kiva import constants
from traits.api import Bool, Dict, Float, List, Str, Trait
# Local relative imports
from base_contour_plot import BaseContourPlot
from contour.contour import Cntr
class ContourLinePlot(BaseContourPlot):
""" Takes a value data object whose elements are scalars, and renders them
as a contour plot.
"""
# TODO: Modify ImageData to explicitly support scalar value arrays
#------------------------------------------------------------------------
# Data-related traits
#------------------------------------------------------------------------
# The thickness(es) of the contour lines.
# It can be either a scalar value, valid for all contour lines, or a list
# of widths. If the list is too short with respect to then number of
# contour lines, the values are repeated from the beginning of the list.
# Widths are associated with levels of increasing value.
widths = Trait(1.0, Float, List)
# The line dash style(s).
styles = Trait("signed", Str, List)
# Line style for positive levels.
positive_style = LineStyle("solid")
# Line style for negative levels.
negative_style = LineStyle("dash")
#------------------------------------------------------------------------
# Private traits
#------------------------------------------------------------------------
# Are the cached contours valid? If False, new ones need to be computed.
_contour_cache_valid = Bool(False)
# Cached collection of traces.
_cached_contours = Dict
# Is the cached width data valid?
_widths_cache_valid = Bool(False)
# Is the cached style data valid?
_styles_cache_valid = Bool(False)
# Cached list of line widths
_widths = List
# Cached list of line styles
_styles = List
# Mapped trait used to convert user-supplied line style values to
# AGG-acceptable ones. (Mapped traits in lists are not supported, must be
# converted one at a time.)
_style_map_trait = LineStyle
#------------------------------------------------------------------------
# Private methods
#------------------------------------------------------------------------
def _render(self, gc):
""" Actually draws the plot.
Implements the Base2DPlot interface.
"""
if not self._level_cache_valid:
self._update_levels()
if not self._contour_cache_valid:
self._update_contours()
if not self._widths_cache_valid:
self._update_widths()
if not self._styles_cache_valid:
self._update_styles()
if not self._colors_cache_valid:
self._update_colors()
with gc:
gc.set_antialias(True)
gc.clip_to_rect(self.x, self.y, self.width, self.height)
gc.set_alpha(self.alpha)
gc.set_line_join(constants.JOIN_BEVEL)
gc.set_line_cap(constants.CAP_ROUND)
for i in range(len(self._levels)):
gc.set_stroke_color(self._colors[i])
gc.set_line_width(self._widths[i])
gc.set_line_dash(self._styles[i])
for trace in self._cached_contours[self._levels[i]]:
if self.orientation == "h":
strace = self.index_mapper.map_screen(trace)
else:
strace = array(self.index_mapper.map_screen(trace))[:,::-1]
gc.begin_path()
gc.lines(strace)
gc.stroke_path()
def _update_contours(self):
""" Updates the cache of contour lines """
# x and y data are "fenceposts" so ignore the last value
# XXX: this truncation is causing errors in Cntr() as of r13735
xdata = self.index._xdata.get_data()
ydata = self.index._ydata.get_data()
xs = linspace(xdata[0], xdata[-1], len(xdata)-1)
ys = linspace(ydata[0], ydata[-1], len(ydata)-1)
xg, yg = meshgrid(xs, ys)
if self.orientation == "h":
c = Cntr(xg, yg, self.value.raw_value)
else:
c = Cntr(xg, yg, self.value.raw_value.T)
self._cached_contours = {}
for level in self._levels:
self._cached_contours[level] = []
traces = c.trace(level)
for trace in traces:
self._cached_contours[level].append(transpose(trace))
self._contour_cache_valid = True
def _update_levels(self):
""" Extends the parent method to also invalidate some other things """
super(ContourLinePlot, self)._update_levels()
self._contour_cache_valid = False
self._widths_cache_valid = False
self._styles_cache_valid = False
def _update_widths(self):
""" Updates the widths cache.
"""
# If we are given a single width, apply it to all levels
if isinstance(self.widths, float):
self._widths = [self.widths] * len(self._levels)
# If the list of widths is shorter than the list of levels,
# simply repeat widths from the beginning of the list as needed
else:
self._widths = []
for i in range(len(self._levels)):
self._widths.append(self.widths[i%len(self.widths)])
self._widths_cache_valid = True
def _update_styles(self):
""" Updates the styles cache.
"""
# If the style type is "signed" then assign styles to levels based
# on their sign
if self.styles == "signed":
self._styles = []
for level in self._levels:
if level < 0:
self._styles.append(self.negative_style_)
else:
self._styles.append(self.positive_style_)
# If we not given a list, apply the one style to all levels
elif not isinstance(self.styles, list):
self._style_map_trait = self.styles
self._styles = [self._style_map_trait_] * len(self._levels)
# If the list of styles is shorter than the list of levels,
# simply repeat styles from the beginning of the list as needed
else:
self._styles = []
for i in range(len(self._levels)):
self._style_map_trait = self.styles[i%len(self.styles)]
self._styles.append(self._style_map_trait_)
self._styles_cache_valid = True
#------------------------------------------------------------------------
# Event handlers
#------------------------------------------------------------------------
def _widths_changed(self):
if self._level_cache_valid:
self._update_widths()
self.invalidate_draw()
def _styles_changed(self):
if self._level_cache_valid:
self._update_styles()
self.invalidate_draw()
def _negative_style_changed(self):
if self._level_cache_valid:
self._update_styles()
self.invalidate_draw()
def _positive_style_changed(self):
if self._level_cache_valid:
self._update_styles()
self.invalidate_draw()
|