12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139 |
- /**************************************************************************/
- /* color_picker.cpp */
- /**************************************************************************/
- /* This file is part of: */
- /* GODOT ENGINE */
- /* https://godotengine.org */
- /**************************************************************************/
- /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
- /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
- /* */
- /* Permission is hereby granted, free of charge, to any person obtaining */
- /* a copy of this software and associated documentation files (the */
- /* "Software"), to deal in the Software without restriction, including */
- /* without limitation the rights to use, copy, modify, merge, publish, */
- /* distribute, sublicense, and/or sell copies of the Software, and to */
- /* permit persons to whom the Software is furnished to do so, subject to */
- /* the following conditions: */
- /* */
- /* The above copyright notice and this permission notice shall be */
- /* included in all copies or substantial portions of the Software. */
- /* */
- /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
- /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
- /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
- /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
- /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
- /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
- /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
- /**************************************************************************/
- #include "color_picker.h"
- #include "core/os/input.h"
- #include "core/os/keyboard.h"
- #include "core/os/os.h"
- #ifdef TOOLS_ENABLED
- #include "editor/editor_scale.h"
- #include "editor/editor_settings.h"
- #endif
- #include "scene/main/viewport.h"
- List<Color> ColorPicker::preset_cache;
- void ColorPicker::_notification(int p_what) {
- switch (p_what) {
- case NOTIFICATION_THEME_CHANGED: {
- btn_pick->set_icon(get_icon("screen_picker", "ColorPicker"));
- btn_add_preset->set_icon(get_icon("add_preset"));
- _update_presets();
- _update_controls();
- } break;
- case NOTIFICATION_ENTER_TREE: {
- btn_pick->set_icon(get_icon("screen_picker", "ColorPicker"));
- btn_add_preset->set_icon(get_icon("add_preset"));
- _update_controls();
- _update_color();
- #ifdef TOOLS_ENABLED
- if (Engine::get_singleton()->is_editor_hint()) {
- if (preset_cache.empty()) {
- PoolColorArray saved_presets = EditorSettings::get_singleton()->get_project_metadata("color_picker", "presets", PoolColorArray());
- for (int i = 0; i < saved_presets.size(); i++) {
- preset_cache.push_back(saved_presets[i]);
- }
- }
- for (int i = 0; i < preset_cache.size(); i++) {
- presets.push_back(preset_cache[i]);
- }
- }
- #endif
- } break;
- case NOTIFICATION_PARENTED: {
- for (int i = 0; i < 4; i++) {
- set_margin((Margin)i, get_margin((Margin)i) + get_constant("margin"));
- }
- } break;
- case NOTIFICATION_VISIBILITY_CHANGED: {
- Popup *p = Object::cast_to<Popup>(get_parent());
- if (p) {
- p->set_size(Size2(get_combined_minimum_size().width + get_constant("margin") * 2, get_combined_minimum_size().height + get_constant("margin") * 2));
- }
- } break;
- case MainLoop::NOTIFICATION_WM_QUIT_REQUEST: {
- if (screen != nullptr && screen->is_visible()) {
- screen->hide();
- }
- } break;
- }
- }
- void ColorPicker::set_focus_on_line_edit() {
- c_text->call_deferred("grab_focus");
- }
- void ColorPicker::_update_controls() {
- const char *rgb[3] = { "R", "G", "B" };
- const char *hsv[3] = { "H", "S", "V" };
- if (hsv_mode_enabled) {
- for (int i = 0; i < 3; i++) {
- labels[i]->set_text(hsv[i]);
- }
- } else {
- for (int i = 0; i < 3; i++) {
- labels[i]->set_text(rgb[i]);
- }
- }
- if (hsv_mode_enabled) {
- set_raw_mode(false);
- btn_raw->set_disabled(true);
- } else if (raw_mode_enabled) {
- set_hsv_mode(false);
- btn_hsv->set_disabled(true);
- } else {
- btn_raw->set_disabled(false);
- btn_hsv->set_disabled(false);
- }
- if (edit_alpha) {
- values[3]->show();
- scroll[3]->show();
- labels[3]->show();
- } else {
- values[3]->hide();
- scroll[3]->hide();
- labels[3]->hide();
- }
- }
- void ColorPicker::_set_pick_color(const Color &p_color, bool p_update_sliders) {
- color = p_color;
- if (color != last_hsv) {
- h = color.get_h();
- s = color.get_s();
- v = color.get_v();
- last_hsv = color;
- }
- if (!is_inside_tree()) {
- return;
- }
- _update_color(p_update_sliders);
- }
- void ColorPicker::set_pick_color(const Color &p_color) {
- _set_pick_color(p_color, true); //because setters can't have more arguments
- }
- void ColorPicker::set_edit_alpha(bool p_show) {
- edit_alpha = p_show;
- _update_controls();
- if (!is_inside_tree()) {
- return;
- }
- _update_color();
- sample->update();
- }
- bool ColorPicker::is_editing_alpha() const {
- return edit_alpha;
- }
- void ColorPicker::_value_changed(double) {
- if (updating) {
- return;
- }
- if (hsv_mode_enabled) {
- h = scroll[0]->get_value() / 360.0;
- s = scroll[1]->get_value() / 100.0;
- v = scroll[2]->get_value() / 100.0;
- color.set_hsv(h, s, v, scroll[3]->get_value() / 255.0);
- last_hsv = color;
- } else {
- for (int i = 0; i < 4; i++) {
- color.components[i] = scroll[i]->get_value() / (raw_mode_enabled ? 1.0 : 255.0);
- }
- }
- _set_pick_color(color, false);
- emit_signal("color_changed", color);
- }
- void ColorPicker::_html_entered(const String &p_html) {
- if (updating || text_is_constructor || !c_text->is_visible()) {
- return;
- }
- Color previous_color = color;
- color = Color::html(p_html);
- if (!is_editing_alpha()) {
- color.a = previous_color.a;
- }
- if (color == previous_color) {
- return;
- }
- if (!is_inside_tree()) {
- return;
- }
- set_pick_color(color);
- emit_signal("color_changed", color);
- }
- void ColorPicker::_update_color(bool p_update_sliders) {
- updating = true;
- if (p_update_sliders) {
- if (hsv_mode_enabled) {
- for (int i = 0; i < 4; i++) {
- scroll[i]->set_step(1.0);
- }
- scroll[0]->set_max(359);
- scroll[0]->set_value(h * 360.0);
- scroll[1]->set_max(100);
- scroll[1]->set_value(s * 100.0);
- scroll[2]->set_max(100);
- scroll[2]->set_value(v * 100.0);
- scroll[3]->set_max(255);
- scroll[3]->set_value(color.components[3] * 255.0);
- } else {
- for (int i = 0; i < 4; i++) {
- if (raw_mode_enabled) {
- scroll[i]->set_step(0.01);
- scroll[i]->set_max(100);
- if (i == 3) {
- scroll[i]->set_max(1);
- }
- scroll[i]->set_value(color.components[i]);
- } else {
- scroll[i]->set_step(1);
- const float byte_value = color.components[i] * 255.0;
- scroll[i]->set_max(next_power_of_2(MAX(255, byte_value)) - 1);
- scroll[i]->set_value(byte_value);
- }
- }
- }
- }
- _update_text_value();
- sample->update();
- uv_edit->update();
- w_edit->update();
- updating = false;
- }
- void ColorPicker::_update_presets() {
- // Only load preset buttons when the only child is the add-preset button.
- int preset_size = _get_preset_size();
- btn_add_preset->set_custom_minimum_size(Size2(preset_size, preset_size));
- if (preset_container->get_child_count() == 1) {
- for (int i = 0; i < preset_cache.size(); i++) {
- _add_preset_button(preset_size, preset_cache[i]);
- }
- _notification(NOTIFICATION_VISIBILITY_CHANGED);
- }
- }
- void ColorPicker::_text_type_toggled() {
- text_is_constructor = !text_is_constructor;
- if (text_is_constructor) {
- text_type->set_text("");
- text_type->set_icon(get_icon("Script", "EditorIcons"));
- c_text->set_editable(false);
- } else {
- text_type->set_text("#");
- text_type->set_icon(nullptr);
- c_text->set_editable(true);
- }
- _update_color();
- }
- Color ColorPicker::get_pick_color() const {
- return color;
- }
- inline int ColorPicker::_get_preset_size() {
- return (int(get_size().width) - (preset_container->get_constant("hseparation") * (preset_column_count - 1))) / preset_column_count;
- }
- void ColorPicker::_add_preset_button(int p_size, const Color &p_color) {
- ColorPresetButton *btn_preset = memnew(ColorPresetButton(p_color));
- btn_preset->set_preset_color(p_color);
- btn_preset->set_custom_minimum_size(Size2(p_size, p_size));
- btn_preset->connect("gui_input", this, "_preset_input", varray(p_color));
- btn_preset->set_tooltip(vformat(RTR("Color: #%s\nLMB: Apply color\nRMB: Remove preset"), p_color.to_html(p_color.a < 1)));
- preset_container->add_child(btn_preset);
- }
- void ColorPicker::add_preset(const Color &p_color) {
- if (presets.find(p_color)) {
- presets.move_to_back(presets.find(p_color));
- // Find button to move to the end.
- for (int i = 1; i < preset_container->get_child_count(); i++) {
- ColorPresetButton *current_btn = Object::cast_to<ColorPresetButton>(preset_container->get_child(i));
- if (current_btn && p_color == current_btn->get_preset_color()) {
- preset_container->move_child(current_btn, preset_container->get_child_count() - 1);
- break;
- }
- }
- } else {
- presets.push_back(p_color);
- preset_cache.push_back(p_color);
- _add_preset_button(_get_preset_size(), p_color);
- _notification(NOTIFICATION_VISIBILITY_CHANGED);
- }
- #ifdef TOOLS_ENABLED
- if (Engine::get_singleton()->is_editor_hint()) {
- PoolColorArray arr_to_save = get_presets();
- EditorSettings::get_singleton()->set_project_metadata("color_picker", "presets", arr_to_save);
- }
- #endif
- }
- void ColorPicker::erase_preset(const Color &p_color) {
- if (presets.find(p_color)) {
- presets.erase(presets.find(p_color));
- preset_cache.erase(preset_cache.find(p_color));
- // Find preset button to remove.
- for (int i = 1; i < preset_container->get_child_count(); i++) {
- ColorPresetButton *current_btn = Object::cast_to<ColorPresetButton>(preset_container->get_child(i));
- if (current_btn && p_color == current_btn->get_preset_color()) {
- current_btn->queue_delete();
- break;
- }
- }
- #ifdef TOOLS_ENABLED
- if (Engine::get_singleton()->is_editor_hint()) {
- PoolColorArray arr_to_save = get_presets();
- EditorSettings::get_singleton()->set_project_metadata("color_picker", "presets", arr_to_save);
- }
- #endif
- }
- }
- PoolColorArray ColorPicker::get_presets() const {
- PoolColorArray arr;
- arr.resize(presets.size());
- for (int i = 0; i < presets.size(); i++) {
- arr.set(i, presets[i]);
- }
- return arr;
- }
- void ColorPicker::set_hsv_mode(bool p_enabled) {
- if (hsv_mode_enabled == p_enabled || raw_mode_enabled) {
- return;
- }
- hsv_mode_enabled = p_enabled;
- if (btn_hsv->is_pressed() != p_enabled) {
- btn_hsv->set_pressed(p_enabled);
- }
- if (!is_inside_tree()) {
- return;
- }
- _update_controls();
- _update_color();
- }
- bool ColorPicker::is_hsv_mode() const {
- return hsv_mode_enabled;
- }
- void ColorPicker::set_raw_mode(bool p_enabled) {
- if (raw_mode_enabled == p_enabled || hsv_mode_enabled) {
- return;
- }
- raw_mode_enabled = p_enabled;
- if (btn_raw->is_pressed() != p_enabled) {
- btn_raw->set_pressed(p_enabled);
- }
- if (!is_inside_tree()) {
- return;
- }
- _update_controls();
- _update_color();
- }
- bool ColorPicker::is_raw_mode() const {
- return raw_mode_enabled;
- }
- void ColorPicker::set_deferred_mode(bool p_enabled) {
- deferred_mode_enabled = p_enabled;
- }
- bool ColorPicker::is_deferred_mode() const {
- return deferred_mode_enabled;
- }
- void ColorPicker::_update_text_value() {
- bool visible = true;
- if (text_is_constructor) {
- String t = "Color(" + String::num(color.r) + ", " + String::num(color.g) + ", " + String::num(color.b);
- if (edit_alpha && color.a < 1) {
- t += ", " + String::num(color.a) + ")";
- } else {
- t += ")";
- }
- c_text->set_text(t);
- }
- if (color.r > 1 || color.g > 1 || color.b > 1 || color.r < 0 || color.g < 0 || color.b < 0) {
- visible = false;
- } else if (!text_is_constructor) {
- c_text->set_text(color.to_html(edit_alpha && color.a < 1));
- }
- text_type->set_visible(visible);
- c_text->set_visible(visible);
- }
- void ColorPicker::_sample_input(const Ref<InputEvent> &p_event) {
- const Ref<InputEventMouseButton> mb = p_event;
- if (mb.is_valid() && mb->is_pressed() && mb->get_button_index() == BUTTON_LEFT) {
- if (display_old_color) {
- const Rect2 rect_old = Rect2(Point2(), Size2(uv_edit->get_size().width * 0.5, sample->get_size().height * 0.95));
- if (rect_old.has_point(mb->get_position())) {
- // Revert to the old color when left-clicking the old color sample.
- set_pick_color(old_color);
- //_update_color();
- emit_signal("color_changed", color);
- }
- }
- }
- }
- void ColorPicker::_sample_draw() {
- // Covers the right half of the sample if the old color is being displayed,
- // or the whole sample if it's not being displayed.
- Rect2 rect_new;
- if (display_old_color) {
- rect_new = Rect2(Point2(uv_edit->get_size().width * 0.5, 0), Size2(uv_edit->get_size().width * 0.5, sample->get_size().height * 0.95));
- // Draw both old and new colors for easier comparison (only if spawned from a ColorPickerButton).
- const Rect2 rect_old = Rect2(Point2(), Size2(uv_edit->get_size().width * 0.5, sample->get_size().height * 0.95));
- if (display_old_color && old_color.a < 1.0) {
- sample->draw_texture_rect(get_icon("preset_bg", "ColorPicker"), rect_old, true);
- }
- sample->draw_rect(rect_old, old_color);
- if (old_color.r > 1 || old_color.g > 1 || old_color.b > 1) {
- // Draw an indicator to denote that the old color is "overbright" and can't be displayed accurately in the preview.
- sample->draw_texture(get_icon("overbright_indicator", "ColorPicker"), Point2());
- }
- } else {
- rect_new = Rect2(Point2(), Size2(uv_edit->get_size().width, sample->get_size().height * 0.95));
- }
- if (color.a < 1.0) {
- sample->draw_texture_rect(get_icon("preset_bg", "ColorPicker"), rect_new, true);
- }
- sample->draw_rect(rect_new, color);
- if (color.r > 1 || color.g > 1 || color.b > 1) {
- // Draw an indicator to denote that the new color is "overbright" and can't be displayed accurately in the preview
- sample->draw_texture(get_icon("overbright_indicator", "ColorPicker"), Point2(uv_edit->get_size().width * 0.5, 0));
- }
- }
- void ColorPicker::_hsv_draw(int p_which, Control *c) {
- if (!c) {
- return;
- }
- if (p_which == 0) {
- Vector<Point2> points;
- points.push_back(Vector2());
- points.push_back(Vector2(c->get_size().x, 0));
- points.push_back(c->get_size());
- points.push_back(Vector2(0, c->get_size().y));
- Vector<Color> colors;
- colors.push_back(Color(1, 1, 1, 1));
- colors.push_back(Color(1, 1, 1, 1));
- colors.push_back(Color(0, 0, 0, 1));
- colors.push_back(Color(0, 0, 0, 1));
- c->draw_polygon(points, colors);
- Vector<Color> colors2;
- Color col = color;
- col.set_hsv(h, 1, 1);
- col.a = 0;
- colors2.push_back(col);
- col.a = 1;
- colors2.push_back(col);
- col.set_hsv(h, 1, 0);
- colors2.push_back(col);
- col.a = 0;
- colors2.push_back(col);
- c->draw_polygon(points, colors2);
- int x = CLAMP(c->get_size().x * s, 0, c->get_size().x);
- int y = CLAMP(c->get_size().y - c->get_size().y * v, 0, c->get_size().y);
- col = color;
- col.a = 1;
- c->draw_line(Point2(x, 0), Point2(x, c->get_size().y), col.inverted());
- c->draw_line(Point2(0, y), Point2(c->get_size().x, y), col.inverted());
- c->draw_line(Point2(x, y), Point2(x, y), Color(1, 1, 1), 2);
- } else if (p_which == 1) {
- Ref<Texture> hue = get_icon("color_hue", "ColorPicker");
- c->draw_texture_rect(hue, Rect2(Point2(), c->get_size()));
- int y = c->get_size().y - c->get_size().y * (1.0 - h);
- Color col = Color();
- col.set_hsv(h, 1, 1);
- c->draw_line(Point2(0, y), Point2(c->get_size().x, y), col.inverted());
- }
- }
- void ColorPicker::_uv_input(const Ref<InputEvent> &p_event) {
- Ref<InputEventMouseButton> bev = p_event;
- if (bev.is_valid()) {
- if (bev->is_pressed() && bev->get_button_index() == BUTTON_LEFT) {
- changing_color = true;
- float x = CLAMP((float)bev->get_position().x, 0, uv_edit->get_size().width);
- float y = CLAMP((float)bev->get_position().y, 0, uv_edit->get_size().height);
- s = x / uv_edit->get_size().width;
- v = 1.0 - y / uv_edit->get_size().height;
- color.set_hsv(h, s, v, color.a);
- last_hsv = color;
- set_pick_color(color);
- _update_color();
- if (!deferred_mode_enabled) {
- emit_signal("color_changed", color);
- }
- } else if (deferred_mode_enabled && !bev->is_pressed() && bev->get_button_index() == BUTTON_LEFT) {
- emit_signal("color_changed", color);
- changing_color = false;
- } else {
- changing_color = false;
- }
- }
- Ref<InputEventMouseMotion> mev = p_event;
- if (mev.is_valid()) {
- if (!changing_color) {
- return;
- }
- float x = CLAMP((float)mev->get_position().x, 0, uv_edit->get_size().width);
- float y = CLAMP((float)mev->get_position().y, 0, uv_edit->get_size().height);
- s = x / uv_edit->get_size().width;
- v = 1.0 - y / uv_edit->get_size().height;
- color.set_hsv(h, s, v, color.a);
- last_hsv = color;
- set_pick_color(color);
- _update_color();
- if (!deferred_mode_enabled) {
- emit_signal("color_changed", color);
- }
- }
- }
- void ColorPicker::_w_input(const Ref<InputEvent> &p_event) {
- Ref<InputEventMouseButton> bev = p_event;
- if (bev.is_valid()) {
- if (bev->is_pressed() && bev->get_button_index() == BUTTON_LEFT) {
- changing_color = true;
- float y = CLAMP((float)bev->get_position().y, 0, w_edit->get_size().height);
- h = y / w_edit->get_size().height;
- } else {
- changing_color = false;
- }
- color.set_hsv(h, s, v, color.a);
- last_hsv = color;
- set_pick_color(color);
- _update_color();
- if (!deferred_mode_enabled) {
- emit_signal("color_changed", color);
- } else if (!bev->is_pressed() && bev->get_button_index() == BUTTON_LEFT) {
- emit_signal("color_changed", color);
- }
- }
- Ref<InputEventMouseMotion> mev = p_event;
- if (mev.is_valid()) {
- if (!changing_color) {
- return;
- }
- float y = CLAMP((float)mev->get_position().y, 0, w_edit->get_size().height);
- h = y / w_edit->get_size().height;
- color.set_hsv(h, s, v, color.a);
- last_hsv = color;
- set_pick_color(color);
- _update_color();
- if (!deferred_mode_enabled) {
- emit_signal("color_changed", color);
- }
- }
- }
- void ColorPicker::_preset_input(const Ref<InputEvent> &p_event, const Color &p_color) {
- Ref<InputEventMouseButton> bev = p_event;
- if (bev.is_valid()) {
- if (bev->is_pressed() && bev->get_button_index() == BUTTON_LEFT) {
- set_pick_color(p_color);
- _update_color();
- emit_signal("color_changed", p_color);
- } else if (bev->is_pressed() && bev->get_button_index() == BUTTON_RIGHT && presets_enabled) {
- erase_preset(p_color);
- emit_signal("preset_removed", p_color);
- }
- }
- }
- void ColorPicker::_screen_input(const Ref<InputEvent> &p_event) {
- if (!is_inside_tree()) {
- return;
- }
- Ref<InputEventMouseButton> bev = p_event;
- if (bev.is_valid() && bev->get_button_index() == BUTTON_LEFT && !bev->is_pressed()) {
- emit_signal("color_changed", color);
- screen->hide();
- }
- Ref<InputEventMouseMotion> mev = p_event;
- if (mev.is_valid()) {
- Viewport *r = get_tree()->get_root();
- if (!r->get_visible_rect().has_point(Point2(mev->get_global_position().x, mev->get_global_position().y))) {
- return;
- }
- Ref<Image> img = r->get_texture()->get_data();
- if (img.is_valid() && !img->empty()) {
- img->lock();
- Vector2 ofs = mev->get_global_position() - r->get_visible_rect().get_position();
- Color c = img->get_pixel(ofs.x, r->get_visible_rect().size.height - ofs.y);
- img->unlock();
- set_pick_color(c);
- }
- }
- }
- void ColorPicker::_add_preset_pressed() {
- add_preset(color);
- emit_signal("preset_added", color);
- }
- void ColorPicker::_screen_pick_pressed() {
- if (!is_inside_tree()) {
- return;
- }
- Viewport *r = get_tree()->get_root();
- if (!screen) {
- screen = memnew(Control);
- r->add_child(screen);
- screen->set_as_toplevel(true);
- screen->set_anchors_and_margins_preset(Control::PRESET_WIDE);
- screen->set_default_cursor_shape(CURSOR_POINTING_HAND);
- screen->connect("gui_input", this, "_screen_input");
- // It immediately toggles off in the first press otherwise.
- screen->call_deferred("connect", "hide", btn_pick, "set_pressed", varray(false));
- }
- screen->raise();
- screen->show_modal();
- }
- void ColorPicker::_focus_enter() {
- bool has_ctext_focus = c_text->has_focus();
- if (has_ctext_focus) {
- c_text->select_all();
- } else {
- c_text->select(0, 0);
- }
- for (int i = 0; i < 4; i++) {
- if (values[i]->get_line_edit()->has_focus() && !has_ctext_focus) {
- values[i]->get_line_edit()->select_all();
- } else {
- values[i]->get_line_edit()->select(0, 0);
- }
- }
- }
- void ColorPicker::_focus_exit() {
- for (int i = 0; i < 4; i++) {
- if (!values[i]->get_line_edit()->get_menu()->is_visible()) {
- values[i]->get_line_edit()->select(0, 0);
- }
- }
- c_text->select(0, 0);
- }
- void ColorPicker::_html_focus_exit() {
- if (c_text->get_menu()->is_visible()) {
- return;
- }
- _html_entered(c_text->get_text());
- _focus_exit();
- }
- void ColorPicker::set_presets_enabled(bool p_enabled) {
- presets_enabled = p_enabled;
- if (!p_enabled) {
- btn_add_preset->set_disabled(true);
- btn_add_preset->set_focus_mode(FOCUS_NONE);
- } else {
- btn_add_preset->set_disabled(false);
- btn_add_preset->set_focus_mode(FOCUS_ALL);
- }
- }
- bool ColorPicker::are_presets_enabled() const {
- return presets_enabled;
- }
- void ColorPicker::set_presets_visible(bool p_visible) {
- presets_visible = p_visible;
- preset_separator->set_visible(p_visible);
- preset_container->set_visible(p_visible);
- }
- bool ColorPicker::are_presets_visible() const {
- return presets_visible;
- }
- void ColorPicker::_bind_methods() {
- ClassDB::bind_method(D_METHOD("set_pick_color", "color"), &ColorPicker::set_pick_color);
- ClassDB::bind_method(D_METHOD("get_pick_color"), &ColorPicker::get_pick_color);
- ClassDB::bind_method(D_METHOD("set_hsv_mode", "mode"), &ColorPicker::set_hsv_mode);
- ClassDB::bind_method(D_METHOD("is_hsv_mode"), &ColorPicker::is_hsv_mode);
- ClassDB::bind_method(D_METHOD("set_raw_mode", "mode"), &ColorPicker::set_raw_mode);
- ClassDB::bind_method(D_METHOD("is_raw_mode"), &ColorPicker::is_raw_mode);
- ClassDB::bind_method(D_METHOD("set_deferred_mode", "mode"), &ColorPicker::set_deferred_mode);
- ClassDB::bind_method(D_METHOD("is_deferred_mode"), &ColorPicker::is_deferred_mode);
- ClassDB::bind_method(D_METHOD("set_edit_alpha", "show"), &ColorPicker::set_edit_alpha);
- ClassDB::bind_method(D_METHOD("is_editing_alpha"), &ColorPicker::is_editing_alpha);
- ClassDB::bind_method(D_METHOD("set_presets_enabled", "enabled"), &ColorPicker::set_presets_enabled);
- ClassDB::bind_method(D_METHOD("are_presets_enabled"), &ColorPicker::are_presets_enabled);
- ClassDB::bind_method(D_METHOD("set_presets_visible", "visible"), &ColorPicker::set_presets_visible);
- ClassDB::bind_method(D_METHOD("are_presets_visible"), &ColorPicker::are_presets_visible);
- ClassDB::bind_method(D_METHOD("add_preset", "color"), &ColorPicker::add_preset);
- ClassDB::bind_method(D_METHOD("erase_preset", "color"), &ColorPicker::erase_preset);
- ClassDB::bind_method(D_METHOD("get_presets"), &ColorPicker::get_presets);
- ClassDB::bind_method(D_METHOD("_value_changed"), &ColorPicker::_value_changed);
- ClassDB::bind_method(D_METHOD("_html_entered"), &ColorPicker::_html_entered);
- ClassDB::bind_method(D_METHOD("_text_type_toggled"), &ColorPicker::_text_type_toggled);
- ClassDB::bind_method(D_METHOD("_add_preset_pressed"), &ColorPicker::_add_preset_pressed);
- ClassDB::bind_method(D_METHOD("_screen_pick_pressed"), &ColorPicker::_screen_pick_pressed);
- ClassDB::bind_method(D_METHOD("_sample_input"), &ColorPicker::_sample_input);
- ClassDB::bind_method(D_METHOD("_sample_draw"), &ColorPicker::_sample_draw);
- ClassDB::bind_method(D_METHOD("_update_presets"), &ColorPicker::_update_presets);
- ClassDB::bind_method(D_METHOD("_hsv_draw"), &ColorPicker::_hsv_draw);
- ClassDB::bind_method(D_METHOD("_uv_input"), &ColorPicker::_uv_input);
- ClassDB::bind_method(D_METHOD("_w_input"), &ColorPicker::_w_input);
- ClassDB::bind_method(D_METHOD("_preset_input"), &ColorPicker::_preset_input);
- ClassDB::bind_method(D_METHOD("_screen_input"), &ColorPicker::_screen_input);
- ClassDB::bind_method(D_METHOD("_focus_enter"), &ColorPicker::_focus_enter);
- ClassDB::bind_method(D_METHOD("_focus_exit"), &ColorPicker::_focus_exit);
- ClassDB::bind_method(D_METHOD("_html_focus_exit"), &ColorPicker::_html_focus_exit);
- ADD_PROPERTY(PropertyInfo(Variant::COLOR, "color"), "set_pick_color", "get_pick_color");
- ADD_PROPERTY(PropertyInfo(Variant::BOOL, "edit_alpha"), "set_edit_alpha", "is_editing_alpha");
- ADD_PROPERTY(PropertyInfo(Variant::BOOL, "hsv_mode"), "set_hsv_mode", "is_hsv_mode");
- ADD_PROPERTY(PropertyInfo(Variant::BOOL, "raw_mode"), "set_raw_mode", "is_raw_mode");
- ADD_PROPERTY(PropertyInfo(Variant::BOOL, "deferred_mode"), "set_deferred_mode", "is_deferred_mode");
- ADD_PROPERTY(PropertyInfo(Variant::BOOL, "presets_enabled"), "set_presets_enabled", "are_presets_enabled");
- ADD_PROPERTY(PropertyInfo(Variant::BOOL, "presets_visible"), "set_presets_visible", "are_presets_visible");
- ADD_SIGNAL(MethodInfo("color_changed", PropertyInfo(Variant::COLOR, "color")));
- ADD_SIGNAL(MethodInfo("preset_added", PropertyInfo(Variant::COLOR, "color")));
- ADD_SIGNAL(MethodInfo("preset_removed", PropertyInfo(Variant::COLOR, "color")));
- }
- ColorPicker::ColorPicker() :
- BoxContainer(true) {
- updating = true;
- edit_alpha = true;
- text_is_constructor = false;
- hsv_mode_enabled = false;
- raw_mode_enabled = false;
- deferred_mode_enabled = false;
- changing_color = false;
- presets_enabled = true;
- presets_visible = true;
- screen = nullptr;
- HBoxContainer *hb_edit = memnew(HBoxContainer);
- add_child(hb_edit);
- hb_edit->set_v_size_flags(SIZE_EXPAND_FILL);
- uv_edit = memnew(Control);
- hb_edit->add_child(uv_edit);
- uv_edit->connect("gui_input", this, "_uv_input");
- uv_edit->set_mouse_filter(MOUSE_FILTER_PASS);
- uv_edit->set_h_size_flags(SIZE_EXPAND_FILL);
- uv_edit->set_v_size_flags(SIZE_EXPAND_FILL);
- uv_edit->set_custom_minimum_size(Size2(get_constant("sv_width"), get_constant("sv_height")));
- uv_edit->connect("draw", this, "_hsv_draw", make_binds(0, uv_edit));
- w_edit = memnew(Control);
- hb_edit->add_child(w_edit);
- w_edit->set_custom_minimum_size(Size2(get_constant("h_width"), 0));
- w_edit->set_h_size_flags(SIZE_FILL);
- w_edit->set_v_size_flags(SIZE_EXPAND_FILL);
- w_edit->connect("gui_input", this, "_w_input");
- w_edit->connect("draw", this, "_hsv_draw", make_binds(1, w_edit));
- HBoxContainer *hb_smpl = memnew(HBoxContainer);
- add_child(hb_smpl);
- sample = memnew(TextureRect);
- hb_smpl->add_child(sample);
- sample->set_h_size_flags(SIZE_EXPAND_FILL);
- sample->connect("gui_input", this, "_sample_input");
- sample->connect("draw", this, "_sample_draw");
- btn_pick = memnew(ToolButton);
- hb_smpl->add_child(btn_pick);
- btn_pick->set_toggle_mode(true);
- btn_pick->set_tooltip(RTR("Pick a color from the editor window."));
- btn_pick->connect("pressed", this, "_screen_pick_pressed");
- VBoxContainer *vbl = memnew(VBoxContainer);
- add_child(vbl);
- add_child(memnew(HSeparator));
- VBoxContainer *vbr = memnew(VBoxContainer);
- add_child(vbr);
- vbr->set_h_size_flags(SIZE_EXPAND_FILL);
- for (int i = 0; i < 4; i++) {
- HBoxContainer *hbc = memnew(HBoxContainer);
- labels[i] = memnew(Label());
- labels[i]->set_custom_minimum_size(Size2(get_constant("label_width"), 0));
- labels[i]->set_v_size_flags(SIZE_SHRINK_CENTER);
- hbc->add_child(labels[i]);
- scroll[i] = memnew(HSlider);
- scroll[i]->set_v_size_flags(SIZE_SHRINK_CENTER);
- scroll[i]->set_focus_mode(FOCUS_NONE);
- hbc->add_child(scroll[i]);
- values[i] = memnew(SpinBox);
- scroll[i]->share(values[i]);
- hbc->add_child(values[i]);
- values[i]->get_line_edit()->connect("focus_entered", this, "_focus_enter");
- values[i]->get_line_edit()->connect("focus_exited", this, "_focus_exit");
- scroll[i]->set_min(0);
- scroll[i]->set_page(0);
- scroll[i]->set_h_size_flags(SIZE_EXPAND_FILL);
- scroll[i]->connect("value_changed", this, "_value_changed");
- vbr->add_child(hbc);
- }
- labels[3]->set_text("A");
- HBoxContainer *hhb = memnew(HBoxContainer);
- vbr->add_child(hhb);
- btn_hsv = memnew(CheckButton);
- hhb->add_child(btn_hsv);
- btn_hsv->set_text(RTR("HSV"));
- btn_hsv->connect("toggled", this, "set_hsv_mode");
- btn_raw = memnew(CheckButton);
- hhb->add_child(btn_raw);
- btn_raw->set_text(RTR("Raw"));
- btn_raw->connect("toggled", this, "set_raw_mode");
- text_type = memnew(Button);
- hhb->add_child(text_type);
- text_type->set_text("#");
- text_type->set_tooltip(TTR("Switch between hexadecimal and code values."));
- if (Engine::get_singleton()->is_editor_hint()) {
- #ifdef TOOLS_ENABLED
- text_type->set_custom_minimum_size(Size2(28 * EDSCALE, 0)); // Adjust for the width of the "Script" icon.
- #endif
- text_type->connect("pressed", this, "_text_type_toggled");
- } else {
- text_type->set_flat(true);
- text_type->set_mouse_filter(MOUSE_FILTER_IGNORE);
- }
- c_text = memnew(LineEdit);
- hhb->add_child(c_text);
- c_text->set_h_size_flags(SIZE_EXPAND_FILL);
- c_text->connect("text_entered", this, "_html_entered");
- c_text->connect("focus_entered", this, "_focus_enter");
- c_text->connect("focus_exited", this, "_html_focus_exit");
- _update_controls();
- updating = false;
- set_pick_color(Color(1, 1, 1));
- preset_separator = memnew(HSeparator);
- add_child(preset_separator);
- preset_container->set_h_size_flags(SIZE_EXPAND_FILL);
- preset_container->set_columns(preset_column_count);
- add_child(preset_container);
- btn_add_preset = memnew(Button);
- btn_add_preset->connect("pressed", this, "_add_preset_pressed");
- btn_add_preset->set_tooltip(RTR("Add current color as a preset."));
- preset_container->add_child(btn_add_preset);
- }
- /////////////////
- void ColorPickerButton::_about_to_show() {
- set_pressed(true);
- if (picker) {
- picker->set_old_color(color);
- }
- }
- void ColorPickerButton::_color_changed(const Color &p_color) {
- color = p_color;
- update();
- emit_signal("color_changed", color);
- }
- void ColorPickerButton::_modal_closed() {
- emit_signal("popup_closed");
- }
- void ColorPickerButton::pressed() {
- _update_picker();
- picker->_update_presets();
- popup->set_position(get_global_position() - picker->get_combined_minimum_size() * get_global_transform().get_scale());
- popup->set_scale(get_global_transform().get_scale());
- popup->popup();
- picker->set_focus_on_line_edit();
- }
- void ColorPickerButton::_notification(int p_what) {
- switch (p_what) {
- case NOTIFICATION_DRAW: {
- const Ref<StyleBox> normal = get_stylebox("normal");
- const Rect2 r = Rect2(normal->get_offset(), get_size() - normal->get_minimum_size());
- draw_texture_rect(Control::get_icon("bg", "ColorPickerButton"), r, true);
- draw_rect(r, color);
- if (color.r > 1 || color.g > 1 || color.b > 1) {
- // Draw an indicator to denote that the color is "overbright" and can't be displayed accurately in the preview
- draw_texture(Control::get_icon("overbright_indicator", "ColorPicker"), normal->get_offset());
- }
- } break;
- case MainLoop::NOTIFICATION_WM_QUIT_REQUEST: {
- if (popup) {
- popup->hide();
- }
- } break;
- }
- if (p_what == NOTIFICATION_VISIBILITY_CHANGED) {
- if (popup && !is_visible_in_tree()) {
- popup->hide();
- }
- }
- }
- void ColorPickerButton::set_pick_color(const Color &p_color) {
- color = p_color;
- if (picker) {
- picker->set_pick_color(p_color);
- }
- update();
- }
- Color ColorPickerButton::get_pick_color() const {
- return color;
- }
- void ColorPicker::set_old_color(const Color &p_color) {
- old_color = p_color;
- }
- void ColorPicker::set_display_old_color(bool p_enabled) {
- display_old_color = p_enabled;
- }
- bool ColorPicker::is_displaying_old_color() const {
- return display_old_color;
- }
- void ColorPickerButton::set_edit_alpha(bool p_show) {
- edit_alpha = p_show;
- if (picker) {
- picker->set_edit_alpha(p_show);
- }
- }
- bool ColorPickerButton::is_editing_alpha() const {
- return edit_alpha;
- }
- ColorPicker *ColorPickerButton::get_picker() {
- _update_picker();
- return picker;
- }
- PopupPanel *ColorPickerButton::get_popup() {
- _update_picker();
- return popup;
- }
- void ColorPickerButton::_update_picker() {
- if (!picker) {
- popup = memnew(PopupPanel);
- picker = memnew(ColorPicker);
- popup->add_child(picker);
- add_child(popup);
- picker->connect("color_changed", this, "_color_changed");
- popup->connect("modal_closed", this, "_modal_closed");
- popup->connect("about_to_show", this, "_about_to_show");
- popup->connect("popup_hide", this, "set_pressed", varray(false));
- picker->set_pick_color(color);
- picker->set_edit_alpha(edit_alpha);
- picker->set_display_old_color(true);
- emit_signal("picker_created");
- }
- }
- void ColorPickerButton::_bind_methods() {
- ClassDB::bind_method(D_METHOD("set_pick_color", "color"), &ColorPickerButton::set_pick_color);
- ClassDB::bind_method(D_METHOD("get_pick_color"), &ColorPickerButton::get_pick_color);
- ClassDB::bind_method(D_METHOD("get_picker"), &ColorPickerButton::get_picker);
- ClassDB::bind_method(D_METHOD("get_popup"), &ColorPickerButton::get_popup);
- ClassDB::bind_method(D_METHOD("set_edit_alpha", "show"), &ColorPickerButton::set_edit_alpha);
- ClassDB::bind_method(D_METHOD("is_editing_alpha"), &ColorPickerButton::is_editing_alpha);
- ClassDB::bind_method(D_METHOD("_about_to_show"), &ColorPickerButton::_about_to_show);
- ClassDB::bind_method(D_METHOD("_color_changed"), &ColorPickerButton::_color_changed);
- ClassDB::bind_method(D_METHOD("_modal_closed"), &ColorPickerButton::_modal_closed);
- ADD_SIGNAL(MethodInfo("color_changed", PropertyInfo(Variant::COLOR, "color")));
- ADD_SIGNAL(MethodInfo("popup_closed"));
- ADD_SIGNAL(MethodInfo("picker_created"));
- ADD_PROPERTY(PropertyInfo(Variant::COLOR, "color"), "set_pick_color", "get_pick_color");
- ADD_PROPERTY(PropertyInfo(Variant::BOOL, "edit_alpha"), "set_edit_alpha", "is_editing_alpha");
- }
- ColorPickerButton::ColorPickerButton() {
- // Initialization is now done deferred,
- // this improves performance in the inspector as the color picker
- // can be expensive to initialize.
- picker = nullptr;
- popup = nullptr;
- edit_alpha = true;
- set_toggle_mode(true);
- }
- /////////////////
- void ColorPresetButton::_notification(int p_what) {
- switch (p_what) {
- case NOTIFICATION_DRAW: {
- const Rect2 r = Rect2(Point2(0, 0), get_size());
- Ref<StyleBox> sb_raw = get_stylebox("preset_fg", "ColorPresetButton")->duplicate();
- Ref<StyleBoxFlat> sb_flat = sb_raw;
- Ref<StyleBoxTexture> sb_texture = sb_raw;
- if (sb_raw->get_class_name() == "StyleBoxFlat") {
- if (preset_color.a < 1) {
- // Draw a background pattern when the color is transparent.
- sb_flat->set_bg_color(Color(1, 1, 1));
- sb_flat->draw(get_canvas_item(), r);
- Rect2 bg_texture_rect = r.grow_margin(MARGIN_LEFT, -sb_flat->get_margin(MARGIN_LEFT));
- bg_texture_rect = bg_texture_rect.grow_margin(MARGIN_RIGHT, -sb_flat->get_margin(MARGIN_RIGHT));
- bg_texture_rect = bg_texture_rect.grow_margin(MARGIN_TOP, -sb_flat->get_margin(MARGIN_TOP));
- bg_texture_rect = bg_texture_rect.grow_margin(MARGIN_BOTTOM, -sb_flat->get_margin(MARGIN_BOTTOM));
- draw_texture_rect(get_icon("preset_bg_icon", "ColorPresetButton"), bg_texture_rect, true);
- sb_flat->set_bg_color(preset_color);
- }
- sb_flat->set_bg_color(preset_color);
- sb_flat->draw(get_canvas_item(), r);
- } else if (sb_raw->get_class_name() == "StyleBoxTexture") {
- if (preset_color.a < 1) {
- // Draw a background pattern when the color is transparent.
- bool use_tile_texture = (sb_texture->get_h_axis_stretch_mode() == StyleBoxTexture::AxisStretchMode::AXIS_STRETCH_MODE_TILE) || (sb_texture->get_h_axis_stretch_mode() == StyleBoxTexture::AxisStretchMode::AXIS_STRETCH_MODE_TILE_FIT);
- draw_texture_rect(get_icon("preset_bg_icon", "ColorPresetButton"), r, use_tile_texture);
- }
- sb_texture->set_modulate(preset_color);
- sb_texture->draw(get_canvas_item(), r);
- } else {
- WARN_PRINT("Unsupported StyleBox used for ColorPresetButton. Use StyleBoxFlat and StyleBoxTexture instead.");
- }
- if (preset_color.r > 1 || preset_color.g > 1 || preset_color.b > 1) {
- // Draw an indicator to denote that the color is "overbright" and can't be displayed accurately in the preview.
- draw_texture(Control::get_icon("overbright_indicator", "ColorPresetButton"), Vector2(0, 0));
- }
- } break;
- }
- }
- void ColorPresetButton::set_preset_color(const Color &p_color) {
- preset_color = p_color;
- }
- Color ColorPresetButton::get_preset_color() const {
- return preset_color;
- }
- ColorPresetButton::ColorPresetButton(Color p_color) {
- preset_color = p_color;
- }
- ColorPresetButton::~ColorPresetButton() {
- }
|