def _add_color_layer(self, data, name=None, visible=False): ''' adds a color layer to the layerstack :param data: numpy array (2D, c) containing the data (c is color) :param name: name of layer :param visible: bool determining whether this layer should be set to visible :return: ''' assert len(data.shape) == 3 data_sources = [] for i in range(data.shape[2]): a, data_shape = createDataSource(data[:,:,i], True) data_sources.append(a) self.editor.dataShape = list(data_shape) if data.shape[2] == 2: new_layer = RGBALayer(data_sources[0], data_sources[1]) elif data.shape[2] == 3: new_layer = RGBALayer(data_sources[0], data_sources[1], data_sources[2]) elif data.shape[2] == 4: new_layer = RGBALayer(data_sources[0], data_sources[1], data_sources[2], data_sources[3]) else: raise Exception("Unexpected number of colors") new_layer.visible = visible if name is not None: new_layer.name = name self.layerstack.append(new_layer)
def _add_color_layer(self, data, name=None, visible=False): ''' adds a color layer to the layerstack :param data: numpy array (2D, c) containing the data (c is color) :param name: name of layer :param visible: bool determining whether this layer should be set to visible :return: ''' assert len(data.shape) == 3 data_sources = [] for i in range(data.shape[2]): a, data_shape = createDataSource(data[:,:,i], True) data_sources.append(a) self.editor.dataShape = list(data_shape) if data.shape[2] == 2: new_layer = RGBALayer(data_sources[0], data_sources[1]) elif data.shape[2] == 3: new_layer = RGBALayer(data_sources[0], data_sources[1], data_sources[2]) elif data.shape[2] == 4: new_layer = RGBALayer(data_sources[0], data_sources[1], data_sources[2], data_sources[3]) else: raise Exception("Unexpected number of colors") new_layer.visible = visible if name is not None: new_layer.name = name self.layerstack.append(new_layer)
def testOpaqueness(self): ims_opaque = RGBAImageSource(self.red, self.green, self.blue, ConstantSource(), RGBALayer(self.red, self.green, self.blue, alpha_missing_value=255), guarantees_opaqueness=True) self.assertTrue(ims_opaque.isOpaque()) ims_notopaque = RGBAImageSource( self.red, self.green, self.blue, ConstantSource(), RGBALayer(self.red, self.green, self.blue, alpha_missing_value=100)) self.assertFalse(ims_notopaque.isOpaque())
def setUp(self): super(RGBAImageSourceTest, self).setUp() basedir = os.path.dirname(volumina._testing.__file__) self.data = numpy.load(os.path.join(basedir, 'rgba129x104.npy')) self.red = _ArraySource2d(self.data[:, :, 0]) self.green = _ArraySource2d(self.data[:, :, 1]) self.blue = _ArraySource2d(self.data[:, :, 2]) self.alpha = _ArraySource2d(self.data[:, :, 3]) self.ims_rgba = RGBAImageSource( self.red, self.green, self.blue, self.alpha, RGBALayer(self.red, self.green, self.blue, self.alpha)) self.ims_rgb = RGBAImageSource( self.red, self.green, self.blue, ConstantSource(), RGBALayer(self.red, self.green, self.blue)) self.ims_rg = RGBAImageSource(self.red, self.green, ConstantSource(), ConstantSource(), RGBALayer(self.red, self.green)) self.ims_ba = RGBAImageSource(red=ConstantSource(), green=ConstantSource(), blue=self.blue, alpha=self.alpha, layer=RGBALayer(blue=self.blue, alpha=self.alpha)) self.ims_a = RGBAImageSource(red=ConstantSource(), green=ConstantSource(), blue=ConstantSource(), alpha=self.alpha, layer=RGBALayer(alpha=self.alpha)) self.ims_none = RGBAImageSource(ConstantSource(), ConstantSource(), ConstantSource(), ConstantSource(), RGBALayer())