def make_padlock(dst, colour, size=qubesimgconverter.ICON_MAXSIZE, disp=False): cs = cairo.ImageSurface(cairo.FORMAT_ARGB32, size, size) cr = cairo.Context(cs) cr.set_source_rgb( *[c / 256.0 for c in qubesimgconverter.hex_to_int(colour)]) cr.set_line_width(.125 * size) cr.rectangle(.125 * size, .5 * size, .75 * size, .4375 * size) cr.fill() cr.move_to(.25 * size, .5 * size) cr.line_to(.25 * size, .375 * size) cr.arc(.5 * size, .375 * size, .25 * size, math.pi, 2 * math.pi) cr.move_to(.75 * size, .375 * size) # this is unneccessary, but helps readability cr.line_to(.75 * size, .5 * size) cr.stroke() if disp: # Careful with those. I have run into severe # floating point errors when adjusting. arrows = 2 gap = 45 * math.pi / 180 offset = 0 radius = .1875 * size width = 0.05 * size cx = .5 * size # cy = .6875 * size cy = .625 * size arrow = 2 * math.pi / arrows for i in range(arrows): cr.move_to(cx, cy) cr.rel_move_to(*polar(radius - width, offset + i * arrow)) cr.arc(cx, cy, radius - width, offset + i * arrow, offset + (i + 1) * arrow - gap) cr.rel_line_to(*polar(width, offset + (i + 1) * arrow - gap + math.pi)) cr.rel_line_to(*polar(width * math.sqrt(8), offset + (i + 1) * arrow - gap + math.pi / 4)) cr.rel_line_to(*polar(width * math.sqrt(8), offset + (i + 1) * arrow - gap - math.pi / 4)) cr.rel_line_to(*polar(width, offset + (i + 1) * arrow - gap + math.pi)) cr.arc_negative(cx, cy, radius + width, offset + (i + 1) * arrow - gap, offset + i * arrow) cr.close_path() cr.set_source_rgb(0xcc / 256.0, 0, 0) # tango's red cr.set_line_width(.0500 * size) cr.set_line_join(cairo.LINE_JOIN_ROUND) cr.stroke_preserve() cr.set_source_rgb(1.0, 1.0, 1.0) cr.fill() cs.write_to_png(dst)
def assertIconColor(self, path, expected_color): image_color_float = self.get_image_color(path, expected_color) expected_color_float = [c / 256.0 for c in qubesimgconverter.hex_to_int( expected_color)] if not all(map(lambda a, b: abs(a - b) <= 0.25, image_color_float, expected_color_float)): self.fail( "Icon {} is not colored as {}".format(path, expected_color))
def get_image_color(self, path, expected_color): """Return mean color of the image as (r, g, b) in float""" image = qubesimgconverter.Image.load_from_file(path) _, l, _ = colorsys.rgb_to_hls( *[c / 256.0 for c in qubesimgconverter.hex_to_int(expected_color)]) def get_hls(pixels, l): for i in range(0, len(pixels), 4): r, g, b, a = tuple(c / 255. for c in pixels[i:i + 4]) if a == 0.0: continue h, _, s = colorsys.rgb_to_hls(r, g, b) yield h, l, s mean_hls = reduce(lambda x, y: (x[0] + y[0], x[1] + y[1], x[2] + y[2]), get_hls(image.data, l), (0, 0, 0)) mean_hls = [x / (mean_hls[1] / l) for x in mean_hls] image_color = colorsys.hls_to_rgb(*mean_hls) return image_color
def test_12_hex_to_float_depth_3_not_implemented(self): with self.assertRaises(ValueError): qubesimgconverter.hex_to_int('123456', depth=3)
def test_11_hex_to_float_result_ff(self): self.assertEqual(qubesimgconverter.hex_to_int('0xffffff'), (0xff, 0xff, 0xff))
def test_10_hex_to_float_result_00(self): self.assertEqual(qubesimgconverter.hex_to_int('#000000'), (0, 0, 0))