From b42c1f82fc31cae95dc35b97591da31a0f760c85 Mon Sep 17 00:00:00 2001
From: kobewi <kobewi4e@gmail.com>
Date: Mon, 17 Oct 2022 13:59:25 +0200
Subject: [PATCH] Improve ColorPicker sliders in OKHSL mode

---
 scene/gui/color_mode.cpp   | 84 ++++++++++++++++++++++++--------------
 scene/gui/color_picker.cpp | 22 ++++++----
 2 files changed, 68 insertions(+), 38 deletions(-)

diff --git a/scene/gui/color_mode.cpp b/scene/gui/color_mode.cpp
index 3a5013dabe0..a063cd344a7 100644
--- a/scene/gui/color_mode.cpp
+++ b/scene/gui/color_mode.cpp
@@ -284,46 +284,68 @@ Color ColorModeOKHSL::get_color() const {
 }
 
 void ColorModeOKHSL::slider_draw(int p_which) {
-	Vector<Vector2> pos;
-	pos.resize(4);
-	Vector<Color> col;
-	col.resize(4);
 	HSlider *slider = color_picker->get_slider(p_which);
 	Size2 size = slider->get_size();
-	Color left_color;
-	Color right_color;
-	Color color = color_picker->get_pick_color();
 	const real_t margin = 16 * color_picker->get_theme_default_base_scale();
 
-	if (p_which == ColorPicker::SLIDER_COUNT) {
-		slider->draw_texture_rect(color_picker->get_theme_icon(SNAME("sample_bg"), SNAME("ColorPicker")), Rect2(Point2(0, 0), Size2(size.x, margin)), true);
-
-		left_color = color;
-		left_color.a = 0;
-		right_color = color;
-		right_color.a = 1;
-	} else if (p_which == 0) {
+	if (p_which == 0) { // H
 		Ref<Texture2D> hue = color_picker->get_theme_icon(SNAME("color_hue"), SNAME("ColorPicker"));
 		slider->draw_set_transform(Point2(), -Math_PI / 2, Size2(1.0, 1.0));
 		slider->draw_texture_rect(hue, Rect2(Vector2(margin * -1, 0), Vector2(margin, size.x)), false);
 		return;
-	} else {
-		Color s_col;
-		Color v_col;
-		s_col.set_ok_hsl(color.get_h(), 0, color.get_v());
-		left_color = (p_which == 1) ? s_col : Color(0, 0, 0);
-		s_col.set_ok_hsl(color.get_h(), 1, color.get_v());
-		v_col.set_ok_hsl(color.get_h(), color.get_s(), 1);
-		right_color = (p_which == 1) ? s_col : v_col;
 	}
-	col.set(0, left_color);
-	col.set(1, right_color);
-	col.set(2, right_color);
-	col.set(3, left_color);
-	pos.set(0, Vector2(0, 0));
-	pos.set(1, Vector2(size.x, 0));
-	pos.set(2, Vector2(size.x, margin));
-	pos.set(3, Vector2(0, margin));
+
+	Vector<Vector2> pos;
+	Vector<Color> col;
+	Color left_color;
+	Color right_color;
+	Color color = color_picker->get_pick_color();
+
+	if (p_which == 2) { // L
+		pos.resize(6);
+		col.resize(6);
+		left_color = Color(0, 0, 0);
+		Color middle_color;
+		middle_color.set_ok_hsl(color.get_ok_hsl_h(), color.get_ok_hsl_s(), 0.5);
+		right_color.set_ok_hsl(color.get_ok_hsl_h(), color.get_ok_hsl_s(), 1);
+
+		col.set(0, left_color);
+		col.set(1, middle_color);
+		col.set(2, right_color);
+		col.set(3, right_color);
+		col.set(4, middle_color);
+		col.set(5, left_color);
+		pos.set(0, Vector2(0, 0));
+		pos.set(1, Vector2(size.x * 0.5, 0));
+		pos.set(2, Vector2(size.x, 0));
+		pos.set(3, Vector2(size.x, margin));
+		pos.set(4, Vector2(size.x * 0.5, margin));
+		pos.set(5, Vector2(0, margin));
+	} else { // A / S
+		pos.resize(4);
+		col.resize(4);
+
+		if (p_which == ColorPicker::SLIDER_COUNT) {
+			slider->draw_texture_rect(color_picker->get_theme_icon(SNAME("sample_bg"), SNAME("ColorPicker")), Rect2(Point2(0, 0), Size2(size.x, margin)), true);
+
+			left_color = color;
+			left_color.a = 0;
+			right_color = color;
+			right_color.a = 1;
+		} else {
+			left_color.set_ok_hsl(color.get_ok_hsl_h(), 0, color.get_ok_hsl_l());
+			right_color.set_ok_hsl(color.get_ok_hsl_h(), 1, color.get_ok_hsl_l());
+		}
+
+		col.set(0, left_color);
+		col.set(1, right_color);
+		col.set(2, right_color);
+		col.set(3, left_color);
+		pos.set(0, Vector2(0, 0));
+		pos.set(1, Vector2(size.x, 0));
+		pos.set(2, Vector2(size.x, margin));
+		pos.set(3, Vector2(0, margin));
+	}
 
 	slider->draw_polygon(pos, col);
 }
diff --git a/scene/gui/color_picker.cpp b/scene/gui/color_picker.cpp
index a2c138c0b29..e5cb25da782 100644
--- a/scene/gui/color_picker.cpp
+++ b/scene/gui/color_picker.cpp
@@ -1087,16 +1087,24 @@ void ColorPicker::_hsv_draw(int p_which, Control *c) {
 			Vector<Color> colors;
 			Color col;
 			col.set_ok_hsl(h, s, 1);
-			points.resize(4);
-			colors.resize(4);
-			points.set(0, Vector2());
-			points.set(1, Vector2(c->get_size().x, 0));
+			Color col2;
+			col2.set_ok_hsl(h, s, 0.5);
+			Color col3;
+			col3.set_ok_hsl(h, s, 0);
+			points.resize(6);
+			colors.resize(6);
+			points.set(0, Vector2(c->get_size().x, 0));
+			points.set(1, Vector2(c->get_size().x, c->get_size().y * 0.5));
 			points.set(2, c->get_size());
 			points.set(3, Vector2(0, c->get_size().y));
+			points.set(4, Vector2(0, c->get_size().y * 0.5));
+			points.set(5, Vector2());
 			colors.set(0, col);
-			colors.set(1, col);
-			colors.set(2, Color(0, 0, 0));
-			colors.set(3, Color(0, 0, 0));
+			colors.set(1, col2);
+			colors.set(2, col3);
+			colors.set(3, col3);
+			colors.set(4, col2);
+			colors.set(5, col);
 			c->draw_polygon(points, colors);
 			int y = c->get_size().y - c->get_size().y * CLAMP(v, 0, 1);
 			col.set_ok_hsl(h, 1, v);