feat: Determines text color based in luminance

Author: 101zh

Random Color Generator/RandColorGen.py

@@ -1,14 +1,50 @@
 import pygame
 import random
+import math
 
 
 def GenerateRandomColorValue():
-    r = random.randint(0, 255)
-    g = random.randint(0, 255)
-    b = random.randint(0, 255)
+    # r = random.randint(0, 255)
+    # g = random.randint(0, 255)
+    # b = random.randint(0, 255)
+    r = 255
+    g = 255
+    b = 255
+
     return (r, g, b)
 
 
+def ReturnTextColor(colorValue):
+    r = colorValue[0]
+    g = colorValue[1]
+    b = colorValue[2]
+    colors = [r, g, b]
+
+    i = 0
+    for c in colors:
+        c = c / 255.0
+        if c <= 0.04045:
+            c = c / 12.92
+        else:
+            c = math.pow(((c + 0.055) / 1.055), 2.4)
+        colors[i] = c
+        i += 1
+
+    r = colors[0]
+    g = colors[1]
+    b = colors[2]
+
+    L = 0.2126 * r + 0.7152 * g + 0.0722 * b
+
+    shouldBeWhite = L > 0.179
+
+    # shouldBeWhite = (r * 0.299 + g * 0.587 + b * 0.114) > 186
+    if shouldBeWhite:
+        return (0, 0, 0)
+    else:
+        return (255, 255, 255)
+
+
 pygame.init()
 
 height = 750
@@ -22,7 +58,7 @@ def GenerateRandomColorValue():
 font = pygame.font.Font(pygame.font.get_default_font(), 32)
 
 # RGB Value
-RGBText = font.render("RGB Value: (255 ,255 ,255)", True, (0, 0, 0))
+RGBText = font.render("RGB Value: (255, 255, 255)", True, (0, 0, 0))
 RGBTextRect = RGBText.get_rect()
 RGBTextRect.center = (width // 2, height // 2 - 20)
 
@@ -44,8 +80,9 @@ def GenerateRandomColorValue():
             color = GenerateRandomColorValue()
             RGBString = "RGB Value: " + str(color)
             hexString = "Hex Value: " + str("#%02x%02x%02x" % color)
-            RGBText = font.render(RGBString, True, (0, 0, 0))
-            hexText = font.render(hexString, True, (0, 0, 0))
+            TextColor = ReturnTextColor(color)
+            RGBText = font.render(RGBString, True, TextColor)
+            hexText = font.render(hexString, True, TextColor)
             print(RGBString + "; " + hexString)
 
             canvas.fill(color)