/usr/lib/python2.7/dist-packages/chaco/function_image_data.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 | from numpy import array
from traits.api import Instance, Callable, on_trait_change
from chaco.api import DataRange2D, ImageData
# Adapted (ie. copied and modified) from function_data_source.
# Given the time frequently required for image manipulation,
# it would be awesome if there was a mechanism for returning
# partial results as they become available.
class FunctionImageData(ImageData):
""" A class that provides data for a 2-D image based upon the range
supplied. This class can be used as the data source for an image plot
or contour plot.
Computation should be fairly swift for acceptable interactive performance.
"""
# The function to call with the low and high values of the range
# in the x and y dimensions. It should return either a 2-D array
# of numerical values, or an array of RGB or RGBA values (shape should
# be (n, m), (n, m, 3) or (n, m, 4)).
func = Callable
# the 2D data_range required for the data shown
data_range = Instance(DataRange2D)
def __init__(self, **kw):
super(FunctionImageData, self).__init__(**kw)
# Explicitly construct the initial data set for ImageData
self.recalculate()
@on_trait_change('data_range.updated')
def recalculate(self):
if self.func is not None and self.data_range is not None:
newarray = self.func(
self.data_range.x_range.low,
self.data_range.x_range.high,
self.data_range.y_range.low,
self.data_range.y_range.high
)
ImageData.set_data(self, newarray)
else:
self._data = array([], dtype=float)
def set_data(self, *args, **kw):
raise RuntimeError("Cannot set numerical data on a FunctionImageData")
def set_mask(self, mask):
# This would be REALLY FREAKING SLICK, but it's currently unimplemented
raise NotImplementedError
def remove_mask(self):
raise NotImplementedError
|