From 1d23aac7d1617daa1b2230d655510f11df8fd8fe Mon Sep 17 00:00:00 2001
From: mmcwilliams <hi@mmcwilliams.com>
Date: Tue, 24 Oct 2023 16:46:08 -0400
Subject: [PATCH] Perform comparison in different color spaces. Move common
 utilty functions to common lib

---
 py/common.py | 53 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 py/common.py

diff --git a/py/common.py b/py/common.py
new file mode 100644
index 0000000..d8e68ce
--- /dev/null
+++ b/py/common.py
@@ -0,0 +1,53 @@
+import cv2
+import numpy as np
+
+def convert_color (color, color_space_a, color_space_b) :
+	pixel = np.zeros([1, 1, 3], dtype=np.uint8)
+	if color_space_a == 'RGB' :
+		pixel = cv2.cvtColor(pixel, cv2.COLOR_BGR2RGB)
+	elif color_space_a == 'LAB' :
+		pixel = cv2.cvtColor(pixel, cv2.COLOR_BGR2LAB)
+	elif color_space_a == 'HSV' :
+		pixel = cv2.cvtColor(pixel, cv2.COLOR_BGR2HSV)
+	#default is BGR
+	pixel[:] = color
+	if color_space_a == 'RGB' and color_space_b == 'BGR' :
+		b = cv2.COLOR_RGB2BGR
+	elif color_space_a == 'BGR' and color_space_b == 'RGB' :
+		b = cv2.COLOR_BGR2RGB
+	elif color_space_a == 'RGB' and color_space_b == 'LAB' :
+		b = cv2.COLOR_RGB2LAB
+	elif color_space_a == 'LAB' and color_space_b == 'RGB' :
+		b = cv2.COLOR_LAB2RGB
+	elif color_space_a == 'BGR' and color_space_b == 'LAB' :
+		b = cv2.COLOR_BGR2LAB
+	elif color_space_a == 'LAB' and color_space_b == 'BGR' :
+		b = cv2.COLOR_LAB2BGR
+	elif color_space_a == 'HSV' and color_space_b == 'LAB' :
+		b = cv2.COLOR_HSV2LAB
+	elif color_space_a == 'LAB' and color_space_b == 'HSV' :
+		b = cv2.COLOR_LAB2HSV
+	elif color_space_a == 'RGB' and color_space_b == 'HSV' :
+		b = cv2.COLOR_RGB2HSV
+	elif color_space_a == 'HSV' and color_space_b == 'RGB' :
+		b = cv2.COLOR_HSV2RGB
+	elif color_space_a == 'BGR' and color_space_b == 'HSV' :
+		b = cv2.COLOR_BGR2HSV
+	elif color_space_a == 'HSV' and color_space_b == 'BGR' :
+		b = cv2.COLOR_HSV2BGR
+
+	cvt = cv2.cvtColor(pixel, b)
+	return cvt[0, 0]
+
+def closest_color (colors, color):
+	colors = np.array(colors)
+	color = np.array(color)
+	distances = np.sqrt(np.sum((colors - color) ** 2, axis=1))
+	index_of_smallest = np.where(distances == np.amin(distances))
+	smallest_distance = colors[index_of_smallest]
+	return smallest_distance
+
+def create_colored_image (width, height, bgr_color):
+	image = np.zeros((height, width, 3), np.uint8)
+	image[:] = bgr_color
+	return image
\ No newline at end of file