def test_composite_merge(self):
# http://stackoverflow.com/questions/3374878
img1 = Image.new("RGBA", size=(100, 100), color=(255, 0, 0, 255))
draw = ImageDraw.Draw(img1)
draw.rectangle((33, 0, 66, 100), fill=(255, 0, 0, 128))
draw.rectangle((67, 0, 100, 100), fill=(255, 0, 0, 0))
img1 = ImageSource(img1)
img2 = Image.new("RGBA", size=(100, 100), color=(0, 255, 0, 255))
draw = ImageDraw.Draw(img2)
draw.rectangle((0, 33, 100, 66), fill=(0, 255, 0, 128))
draw.rectangle((0, 67, 100, 100), fill=(0, 255, 0, 0))
img2 = ImageSource(img2)
result = merge_images([img2, img1], ImageOptions(transparent=True))
img = result.as_image()
assert img.mode == "RGBA"
assert_img_colors_eq(
img,
[
(1089, (0, 255, 0, 255)),
(1089, (255, 255, 255, 0)),
(1122, (0, 255, 0, 128)),
(1122, (128, 126, 0, 255)),
(1122, (255, 0, 0, 128)),
(1156, (170, 84, 0, 191)),
(3300, (255, 0, 0, 255)),
],
)
def test_mask_partial_image_transparent(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(transparent=True))
result = mask_image_source_from_coverage(img, [0, 0, 10, 10], SRS(4326), coverage([5, 5, 30, 30]))
assert_img_colors_eq(result.as_image().getcolors(),
[(7500, (255, 255, 255, 0)), (2500, (100, 0, 200, 255))])
def test_composite_merge(self):
# http://stackoverflow.com/questions/3374878
if not hasattr(Image, 'alpha_composite'):
raise SkipTest()
img1 = Image.new('RGBA', size=(100, 100), color=(255, 0, 0, 255))
draw = ImageDraw.Draw(img1)
draw.rectangle((33, 0, 66, 100), fill=(255, 0, 0, 128))
draw.rectangle((67, 0, 100, 100), fill=(255, 0, 0, 0))
img1 = ImageSource(img1)
img2 = Image.new('RGBA', size =(100, 100), color=(0, 255, 0, 255))
draw = ImageDraw.Draw(img2)
draw.rectangle((0, 33, 100, 66), fill=(0, 255, 0, 128))
draw.rectangle((0, 67, 100, 100), fill=(0, 255, 0, 0))
img2 = ImageSource(img2)
result = merge_images([img2, img1], ImageOptions(transparent=True))
img = result.as_image()
eq_(img.mode, 'RGBA')
assert_img_colors_eq(img, [
(1089, (0, 255, 0, 255)),
(1089, (255, 255, 255, 0)),
(1122, (0, 255, 0, 128)),
(1122, (128, 126, 0, 255)),
(1122, (255, 0, 0, 128)),
(1156, (170, 84, 0, 191)),
(3300, (255, 0, 0, 255))])
def test_mask_partial_image_transparent(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(transparent=True))
result = mask_image_source_from_coverage(img, [0, 0, 10, 10], SRS(4326), coverage([5, 5, 30, 30]))
assert_img_colors_eq(result.as_image().getcolors(),
[(7500, (255, 255, 255, 0)), (2500, (100, 0, 200, 255))])
def test_merge_L(self):
img1 = ImageSource(Image.new("RGBA", (10, 10), (255, 0, 255, 255)))
img2 = ImageSource(Image.new("L", (10, 10), 100))
# img2 overlays img1
result = merge_images([img1, img2], ImageOptions(transparent=True))
img = result.as_image()
assert_img_colors_eq(img, [(10 * 10, (100, 100, 100, 255))])
def test_mask_outside_of_image_bgcolor(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(bgcolor=(200, 30, 120)))
result = mask_image_source_from_coverage(img, [0, 0, 10, 10],
SRS(4326),
coverage([20, 20, 30, 30]))
assert_img_colors_eq(result.as_image().getcolors(),
[((100 * 100), (200, 30, 120))])
def test_opacity_merge_mixed_modes(self):
img1 = ImageSource(Image.new('RGBA', (10, 10), (255, 0, 255, 255)))
img2 = ImageSource(Image.new('RGB', (10, 10), (0, 255, 255)).convert('P'),
image_opts=ImageOptions(opacity=0.5))
result = merge_images([img1, img2], ImageOptions(transparent=True))
img = result.as_image()
assert_img_colors_eq(img, [
(10*10, (127, 127, 255, 255)),
])
def test_opacity_merge_mixed_modes(self):
img1 = ImageSource(Image.new("RGBA", (10, 10), (255, 0, 255, 255)))
img2 = ImageSource(
Image.new("RGB", (10, 10), (0, 255, 255)).convert("P"),
image_opts=ImageOptions(opacity=0.5),
)
result = merge_images([img1, img2], ImageOptions(transparent=True))
img = result.as_image()
assert_img_colors_eq(img, [(10 * 10, (127, 127, 255, 255))])
def test_shapely_mask_with_transform_partial_image_transparent(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(transparent=True))
p = Polygon([(0, 0), (222000, 0), (222000, 222000), (0, 222000)]) # ~ 2x2 degres
result = mask_image_source_from_coverage(img, [0, 0, 10, 10], SRS(4326), coverage(p, 'EPSG:3857'))
# 20*20 = 400
assert_img_colors_eq(result.as_image().getcolors(),
[(10000-400, (255, 255, 255, 0)), (400, (100, 0, 200, 255))])
def test_merge_L(self):
img1 = ImageSource(Image.new('RGBA', (10, 10), (255, 0, 255, 255)))
img2 = ImageSource(Image.new('L', (10, 10), 100))
# img2 overlays img1
result = merge_images([img1, img2], ImageOptions(transparent=True))
img = result.as_image()
assert_img_colors_eq(img, [
(10*10, (100, 100, 100, 255)),
])
def test_shapely_mask_with_transform_partial_image_transparent(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(transparent=True))
p = Polygon([(0, 0), (222000, 0), (222000, 222000), (0, 222000)]) # ~ 2x2 degres
result = mask_image_source_from_coverage(img, [0, 0, 10, 10], SRS(4326), coverage(p, 'EPSG:3857'))
# 20*20 = 400
assert_img_colors_eq(result.as_image().getcolors(),
[(10000-400, (255, 255, 255, 0)), (400, (100, 0, 200, 255))])
def test_mask_outside_of_image_transparent(self):
img = ImageSource(
Image.new("RGB", (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(transparent=True),
)
result = mask_image_source_from_coverage(
img, [0, 0, 10, 10], SRS(4326), coverage([20, 20, 30, 30])
)
assert_img_colors_eq(
result.as_image().getcolors(), [((100 * 100), (255, 255, 255, 0))]
)
def test_wkt_mask_partial_image_transparent(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(transparent=True))
# polygon with hole
geom = 'POLYGON((2 2, 2 8, 8 8, 8 2, 2 2), (4 4, 4 6, 6 6, 6 4, 4 4))'
result = mask_image_source_from_coverage(img, [0, 0, 10, 10], SRS(4326), coverage(geom))
# 60*61 - 20*21 = 3240
assert_img_colors_eq(result.as_image().getcolors(),
[(10000-3240, (255, 255, 255, 0)), (3240, (100, 0, 200, 255))])
def test_wkt_mask_partial_image_transparent(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(transparent=True))
# polygon with hole
geom = 'POLYGON((2 2, 2 8, 8 8, 8 2, 2 2), (4 4, 4 6, 6 6, 6 4, 4 4))'
result = mask_image_source_from_coverage(img, [0, 0, 10, 10], SRS(4326), coverage(geom))
# 60*60 - 20*20 = 3200
assert_img_colors_eq(result.as_image().getcolors(),
[(10000-3200, (255, 255, 255, 0)), (3200, (100, 0, 200, 255))])
def test_mask_partial_image_bgcolor(self):
img = ImageSource(
Image.new("RGB", (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(bgcolor=(200, 30, 120)),
)
result = mask_image_source_from_coverage(
img, [0, 0, 10, 10], SRS(4326), coverage([5, 5, 30, 30])
)
assert_img_colors_eq(
result.as_image().getcolors(),
[(7500, (200, 30, 120)), (2500, (100, 0, 200))],
)
def test_composite_merge_opacity(self):
bg = Image.new("RGBA", size=(100, 100), color=(255, 0, 255, 255))
bg = ImageSource(bg)
fg = Image.new("RGBA", size=(100, 100), color=(0, 0, 0, 0))
draw = ImageDraw.Draw(fg)
draw.rectangle((10, 10, 89, 89), fill=(0, 255, 255, 255))
fg = ImageSource(fg, image_opts=ImageOptions(opacity=0.5))
result = merge_images([bg, fg], ImageOptions(transparent=True))
img = result.as_image()
assert img.mode == "RGBA"
assert_img_colors_eq(img, [(3600, (255, 0, 255, 255)),
(6400, (128, 127, 255, 255))])
def test_composite_merge_opacity(self):
if not hasattr(Image, 'alpha_composite'):
raise SkipTest()
bg = Image.new('RGBA', size=(100, 100), color=(255, 0, 255, 255))
bg = ImageSource(bg)
fg = Image.new('RGBA', size=(100, 100), color=(0, 0, 0, 0))
draw = ImageDraw.Draw(fg)
draw.rectangle((10, 10, 89, 89), fill=(0, 255, 255, 255))
fg = ImageSource(fg, image_opts=ImageOptions(opacity=0.5))
result = merge_images([bg, fg], ImageOptions(transparent=True))
img = result.as_image()
eq_(img.mode, 'RGBA')
assert_img_colors_eq(img, [(3600, (255, 0, 255, 255)),
(6400, (128, 127, 255, 255))])
def test_composite_merge_opacity(self):
if not hasattr(Image, 'alpha_composite'):
raise SkipTest()
bg = Image.new('RGBA', size=(100, 100), color=(255, 0, 255, 255))
bg = ImageSource(bg)
fg = Image.new('RGBA', size =(100, 100), color=(0, 0, 0, 0))
draw = ImageDraw.Draw(fg)
draw.rectangle((10, 10, 89, 89), fill=(0, 255, 255, 255))
fg = ImageSource(fg, image_opts=ImageOptions(opacity=0.5))
result = merge_images([bg, fg], ImageOptions(transparent=True))
img = result.as_image()
eq_(img.mode, 'RGBA')
assert_img_colors_eq(img, [
(3600, (255, 0, 255, 255)),
(6400, (128, 127, 255, 255))])
def test_mask_outside_of_image_bgcolor(self):
img = ImageSource(Image.new('RGB', (100, 100), color=(100, 0, 200)),
image_opts=ImageOptions(bgcolor=(200, 30, 120)))
result = mask_image_source_from_coverage(img, [0, 0, 10, 10], SRS(4326), coverage([20, 20, 30, 30]))
assert_img_colors_eq(result.as_image().getcolors(), [((100*100), (200, 30, 120))])
