def _step(self): print "[DEBUG] ----- start step" # grab image container from port using traits in_img_c = self.grab_input_using_trait('image') # Get image from conatiner in_img = in_img_c.get_image() # convert generic image to PIL image pil_image = in_img.get_pil_image() # draw on the image to prove we can do it num = 37 import ImageDraw draw = ImageDraw.Draw(pil_image) draw.line((0, 0) + pil_image.size, fill=128, width=5) draw.line((0, pil_image.size[1], pil_image.size[0], 0), fill=32768, width=5) # x0 y0 x1 y1 draw.rectangle( [num, num, num+100, num+100], outline=125 ) del draw new_image = Image.from_pil( pil_image ) # get new image handle new_ic = ImageContainer( new_image ) # push object to output port self.push_to_port_using_trait( 'out_image', new_ic ) self._base_step()
def _step(self): print "[DEBUG] ----- start step" # grab image container from port using traits in_img_c = self.grab_input_using_trait('image') # Get image from conatiner in_img = in_img_c.get_image() # convert generic image to PIL image pil_image = in_img.get_pil_image() # draw on the image to prove we can do it num = 37 import ImageDraw draw = ImageDraw.Draw(pil_image) draw.line((0, 0) + pil_image.size, fill=128, width=5) draw.line((0, pil_image.size[1], pil_image.size[0], 0), fill=32768, width=5) # x0 y0 x1 y1 draw.rectangle([num, num, num + 100, num + 100], outline=125) del draw new_image = Image.from_pil(pil_image) # get new image handle new_ic = ImageContainer(new_image) # push object to output port self.push_to_port_using_trait('out_image', new_ic) self._base_step()
def test_pil_I(self): # test int image img = Image(720, 480, 1, True, Image.PIXEL_SIGNED, 4) pil_img = img.get_pil_image() img2 = Image.from_pil(pil_img) nose.tools.assert_equal(img.equal_content(img2), True)
def test_pil_1(self): # test bool image img = Image(720, 480, 1, True, Image.PIXEL_BOOL, 1) pil_img = img.get_pil_image() img2 = Image.from_pil(pil_img) nose.tools.assert_equal(img.equal_content(img2), True)
def test_pil_RGBA(self): # test RGBA image img = Image(720, 480, 4, True) pil_img = img.get_pil_image() img2 = Image.from_pil(pil_img) nose.tools.assert_equal(img.equal_content(img2), True)