-def light_fitting_image(size, base_image_name, colours):
- """ Render a light fitting image. """
- size = str(size)
- fitting_colours = [COLOURS[c] for c in colours]
- ncolour = len(fitting_colours)
- if ncolour > 3:
- print "Multicoloured light should not have more than 3 colours"
- ncolour = 3
-
- if ncolour == 1:
- return loader.load_image(
- size, base_image_name,
- transform=Multiply(colour=fitting_colours[0]))
-
- colour_mult_image = pygame.surface.Surface((48, 48))
- for i in range(ncolour):
- sector = loader.load_image(
- size, "light_mask_%d_%d.png" % (ncolour, i + 1),
- transform=Multiply(colour=fitting_colours[i]))
- colour_mult_image.blit(sector, (0, 0), None, 0)
-
- fitting_image = loader.load_image(size, base_image_name)
- fitting_image.blit(colour_mult_image, (0, 0), None, pgl.BLEND_RGBA_MULT)
- return fitting_image
+def light_fitting_by_type(light_type):
+ """ Render a light fitting image for a light type. """
+ return BaseLight.find_cls(light_type).FITTING_IMG