1 """ Tools for creating serum actors. """
4 SERUMS = ["smart", "fast", "strong"]
6 SERUM_OVERLAY_COLORS = {
7 "smart": (0, 0, 255, 255), # blue
8 "fast": (0, 255, 0, 255), # green
9 "strong": (255, 0, 255, 255), # purple
10 "none": (170, 68, 0, 255), # brown
14 def roach_serum_color(roach):
15 for serum_name in SERUMS:
16 if getattr(roach, serum_name):
17 return SERUM_OVERLAY_COLORS[serum_name]
18 return SERUM_OVERLAY_COLORS["none"]