color_picker.cpp 93 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709
  1. /**************************************************************************/
  2. /* color_picker.cpp */
  3. /**************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /**************************************************************************/
  8. /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
  9. /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /**************************************************************************/
  30. #include "color_picker.h"
  31. #include "core/io/image.h"
  32. #include "core/math/expression.h"
  33. #include "scene/gui/color_mode.h"
  34. #include "scene/gui/color_picker_shape.h"
  35. #include "scene/gui/file_dialog.h"
  36. #include "scene/gui/grid_container.h"
  37. #include "scene/gui/label.h"
  38. #include "scene/gui/line_edit.h"
  39. #include "scene/gui/link_button.h"
  40. #include "scene/gui/margin_container.h"
  41. #include "scene/gui/menu_button.h"
  42. #include "scene/gui/panel.h"
  43. #include "scene/gui/popup_menu.h"
  44. #include "scene/gui/slider.h"
  45. #include "scene/gui/spin_box.h"
  46. #include "scene/gui/texture_rect.h"
  47. #include "scene/resources/atlas_texture.h"
  48. #include "scene/resources/color_palette.h"
  49. #include "scene/resources/image_texture.h"
  50. #include "scene/resources/style_box_flat.h"
  51. #include "scene/resources/style_box_texture.h"
  52. #include "scene/theme/theme_db.h"
  53. #include "thirdparty/misc/ok_color_shader.h"
  54. static inline bool is_color_overbright(const Color &color) {
  55. return (color.r > 1.0) || (color.g > 1.0) || (color.b > 1.0);
  56. }
  57. static inline bool is_color_valid_hex(const Color &color) {
  58. return !is_color_overbright(color) && color.r >= 0 && color.g >= 0 && color.b >= 0;
  59. }
  60. static inline String color_to_string(const Color &color, bool show_alpha = true, bool force_value_format = false) {
  61. if (!force_value_format && !is_color_overbright(color)) {
  62. return "#" + color.to_html(show_alpha);
  63. }
  64. String t = "(" + String::num(color.r, 3) + ", " + String::num(color.g, 3) + ", " + String::num(color.b, 3);
  65. if (show_alpha) {
  66. t += ", " + String::num(color.a, 3) + ")";
  67. } else {
  68. t += ")";
  69. }
  70. return t;
  71. }
  72. void ColorPicker::_notification(int p_what) {
  73. switch (p_what) {
  74. case NOTIFICATION_ACCESSIBILITY_UPDATE: {
  75. RID ae = get_accessibility_element();
  76. ERR_FAIL_COND(ae.is_null());
  77. DisplayServer::get_singleton()->accessibility_update_set_role(ae, DisplayServer::AccessibilityRole::ROLE_COLOR_PICKER);
  78. DisplayServer::get_singleton()->accessibility_update_set_color_value(ae, color);
  79. } break;
  80. case NOTIFICATION_ENTER_TREE: {
  81. _update_color();
  82. } break;
  83. #ifdef MACOS_ENABLED
  84. case NOTIFICATION_VISIBILITY_CHANGED: {
  85. if (is_visible_in_tree()) {
  86. perm_hb->set_visible(!OS::get_singleton()->get_granted_permissions().has("macos.permission.RECORD_SCREEN"));
  87. }
  88. } break;
  89. #endif
  90. case NOTIFICATION_READY: {
  91. if (DisplayServer::get_singleton()->has_feature(DisplayServer::FEATURE_NATIVE_COLOR_PICKER)) {
  92. btn_pick->set_tooltip_text(ETR("Pick a color from the screen."));
  93. btn_pick->connect(SceneStringName(pressed), callable_mp(this, &ColorPicker::_pick_button_pressed_native));
  94. } else if (DisplayServer::get_singleton()->has_feature(DisplayServer::FEATURE_SCREEN_CAPTURE) && !get_tree()->get_root()->is_embedding_subwindows()) {
  95. // FIXME: The embedding check is needed to fix a bug in single-window mode (GH-93718).
  96. btn_pick->set_tooltip_text(ETR("Pick a color from the screen."));
  97. btn_pick->connect(SceneStringName(pressed), callable_mp(this, &ColorPicker::_pick_button_pressed));
  98. } else {
  99. // On unsupported platforms, use a legacy method for color picking.
  100. btn_pick->set_tooltip_text(ETR("Pick a color from the application window."));
  101. btn_pick->connect(SceneStringName(pressed), callable_mp(this, &ColorPicker::_pick_button_pressed_legacy));
  102. }
  103. } break;
  104. case NOTIFICATION_THEME_CHANGED: {
  105. btn_pick->set_button_icon(theme_cache.screen_picker);
  106. _update_drop_down_arrow(btn_preset->is_pressed(), btn_preset);
  107. _update_drop_down_arrow(btn_recent_preset->is_pressed(), btn_recent_preset);
  108. btn_add_preset->set_button_icon(theme_cache.add_preset);
  109. menu_btn->set_button_icon(theme_cache.menu_option);
  110. btn_mode->set_button_icon(theme_cache.menu_option);
  111. btn_pick->set_custom_minimum_size(Size2(28 * theme_cache.base_scale, 0));
  112. btn_shape->set_custom_minimum_size(Size2(28 * theme_cache.base_scale, 0));
  113. btn_mode->set_custom_minimum_size(Size2(28 * theme_cache.base_scale, 0));
  114. {
  115. int i = 0;
  116. for (ColorPickerShape *shape : shapes) {
  117. if (shape->is_initialized) {
  118. shape->update_theme();
  119. }
  120. shape_popup->set_item_icon(i, shape->get_icon());
  121. i++;
  122. }
  123. }
  124. if (current_shape != SHAPE_NONE) {
  125. btn_shape->set_button_icon(shape_popup->get_item_icon(get_current_shape_index()));
  126. }
  127. for (int i = 0; i < MODE_SLIDER_COUNT; i++) {
  128. labels[i]->set_custom_minimum_size(Size2(theme_cache.label_width, 0));
  129. sliders[i]->add_theme_constant_override(SNAME("center_grabber"), theme_cache.center_slider_grabbers);
  130. }
  131. alpha_label->set_custom_minimum_size(Size2(theme_cache.label_width, 0));
  132. alpha_slider->add_theme_constant_override(SNAME("center_grabber"), theme_cache.center_slider_grabbers);
  133. intensity_label->set_custom_minimum_size(Size2(theme_cache.label_width, 0));
  134. for (int i = 0; i < MODE_BUTTON_COUNT; i++) {
  135. mode_btns[i]->begin_bulk_theme_override();
  136. mode_btns[i]->add_theme_style_override(SceneStringName(pressed), theme_cache.mode_button_pressed);
  137. mode_btns[i]->add_theme_style_override(CoreStringName(normal), theme_cache.mode_button_normal);
  138. mode_btns[i]->add_theme_style_override(SceneStringName(hover), theme_cache.mode_button_hover);
  139. mode_btns[i]->end_bulk_theme_override();
  140. }
  141. internal_margin->begin_bulk_theme_override();
  142. internal_margin->add_theme_constant_override(SNAME("margin_bottom"), theme_cache.content_margin);
  143. internal_margin->add_theme_constant_override(SNAME("margin_left"), theme_cache.content_margin);
  144. internal_margin->add_theme_constant_override(SNAME("margin_right"), theme_cache.content_margin);
  145. internal_margin->add_theme_constant_override(SNAME("margin_top"), theme_cache.content_margin);
  146. internal_margin->end_bulk_theme_override();
  147. _reset_sliders_theme();
  148. hex_label->set_custom_minimum_size(Size2(38 * theme_cache.base_scale, 0));
  149. // Adjust for the width of the "script" icon.
  150. text_type->set_custom_minimum_size(Size2(28 * theme_cache.base_scale, 0));
  151. _update_presets();
  152. _update_recent_presets();
  153. _update_controls();
  154. } break;
  155. case NOTIFICATION_WM_CLOSE_REQUEST: {
  156. if (picker_window != nullptr && picker_window->is_visible()) {
  157. picker_window->hide();
  158. }
  159. } break;
  160. case NOTIFICATION_FOCUS_ENTER:
  161. case NOTIFICATION_FOCUS_EXIT: {
  162. if (current_shape != SHAPE_NONE) {
  163. shapes[get_current_shape_index()]->cursor_editing = false;
  164. }
  165. } break;
  166. case NOTIFICATION_INTERNAL_PROCESS: {
  167. if (!is_picking_color) {
  168. Input *input = Input::get_singleton();
  169. if (input->is_action_just_released("ui_left") ||
  170. input->is_action_just_released("ui_right") ||
  171. input->is_action_just_released("ui_up") ||
  172. input->is_action_just_released("ui_down")) {
  173. gamepad_event_delay_ms = DEFAULT_GAMEPAD_EVENT_DELAY_MS;
  174. if (current_shape == SHAPE_NONE) {
  175. shapes[get_current_shape_index()]->echo_multiplier = 1;
  176. }
  177. accept_event();
  178. set_process_internal(false);
  179. return;
  180. }
  181. if (current_shape == SHAPE_NONE) {
  182. return;
  183. }
  184. gamepad_event_delay_ms -= get_process_delta_time();
  185. if (gamepad_event_delay_ms <= 0) {
  186. gamepad_event_delay_ms = GAMEPAD_EVENT_REPEAT_RATE_MS + gamepad_event_delay_ms;
  187. // Treat any input from joypad axis as -1, 0, or 1, as the value is added to Vector2i and would be lost.
  188. Vector2 color_change_vector = Vector2(
  189. input->is_action_pressed("ui_right") - input->is_action_pressed("ui_left"),
  190. input->is_action_pressed("ui_down") - input->is_action_pressed("ui_up"));
  191. shapes[get_current_shape_index()]->update_cursor(color_change_vector, true);
  192. accept_event();
  193. }
  194. return;
  195. }
  196. DisplayServer *ds = DisplayServer::get_singleton();
  197. Vector2 ofs = ds->mouse_get_position();
  198. Color c = DisplayServer::get_singleton()->screen_get_pixel(ofs);
  199. picker_preview_style_box_color->set_bg_color(c);
  200. picker_preview_style_box->set_bg_color(c.get_luminance() < 0.5 ? Color(1.0f, 1.0f, 1.0f) : Color(0.0f, 0.0f, 0.0f));
  201. if (ds->has_feature(DisplayServer::FEATURE_SCREEN_EXCLUDE_FROM_CAPTURE)) {
  202. Ref<Image> zoom_preview_img = ds->screen_get_image_rect(Rect2i(ofs.x - 8, ofs.y - 8, 17, 17));
  203. picker_window->set_position(ofs - Vector2(28, 28));
  204. picker_texture_zoom->set_texture(ImageTexture::create_from_image(zoom_preview_img));
  205. } else {
  206. Size2i screen_size = ds->screen_get_size(DisplayServer::SCREEN_WITH_MOUSE_FOCUS);
  207. Vector2i screen_position = ds->screen_get_position(DisplayServer::SCREEN_WITH_MOUSE_FOCUS);
  208. float ofs_decal_x = (ofs.x < screen_position.x + screen_size.width - 51) ? 8 : -36;
  209. float ofs_decal_y = (ofs.y < screen_position.y + screen_size.height - 51) ? 8 : -36;
  210. picker_window->set_position(ofs + Vector2(ofs_decal_x, ofs_decal_y));
  211. }
  212. set_pick_color(c);
  213. } break;
  214. }
  215. }
  216. void ColorPicker::_update_theme_item_cache() {
  217. VBoxContainer::_update_theme_item_cache();
  218. theme_cache.base_scale = get_theme_default_base_scale();
  219. }
  220. void ColorPicker::init_shaders() {
  221. wheel_shader.instantiate();
  222. wheel_shader->set_code(R"(
  223. // ColorPicker wheel shader.
  224. shader_type canvas_item;
  225. uniform float wheel_radius = 0.42;
  226. void fragment() {
  227. float x = UV.x - 0.5;
  228. float y = UV.y - 0.5;
  229. float a = atan(y, x);
  230. x += 0.001;
  231. y += 0.001;
  232. float b = float(sqrt(x * x + y * y) < 0.5) * float(sqrt(x * x + y * y) > wheel_radius);
  233. x -= 0.002;
  234. float b2 = float(sqrt(x * x + y * y) < 0.5) * float(sqrt(x * x + y * y) > wheel_radius);
  235. y -= 0.002;
  236. float b3 = float(sqrt(x * x + y * y) < 0.5) * float(sqrt(x * x + y * y) > wheel_radius);
  237. x += 0.002;
  238. float b4 = float(sqrt(x * x + y * y) < 0.5) * float(sqrt(x * x + y * y) > wheel_radius);
  239. COLOR = vec4(clamp((abs(fract(((a - TAU) / TAU) + vec3(3.0, 2.0, 1.0) / 3.0) * 6.0 - 3.0) - 1.0), 0.0, 1.0), (b + b2 + b3 + b4) / 4.00);
  240. }
  241. )");
  242. circle_shader.instantiate();
  243. circle_shader->set_code(R"(
  244. // ColorPicker circle shader.
  245. shader_type canvas_item;
  246. uniform float v = 1.0;
  247. void fragment() {
  248. float x = UV.x - 0.5;
  249. float y = UV.y - 0.5;
  250. float a = atan(y, x);
  251. x += 0.001;
  252. y += 0.001;
  253. float b = float(sqrt(x * x + y * y) < 0.5);
  254. x -= 0.002;
  255. float b2 = float(sqrt(x * x + y * y) < 0.5);
  256. y -= 0.002;
  257. float b3 = float(sqrt(x * x + y * y) < 0.5);
  258. x += 0.002;
  259. float b4 = float(sqrt(x * x + y * y) < 0.5);
  260. COLOR = vec4(mix(vec3(1.0), clamp(abs(fract(vec3((a - TAU) / TAU) + vec3(1.0, 2.0 / 3.0, 1.0 / 3.0)) * 6.0 - vec3(3.0)) - vec3(1.0), 0.0, 1.0), ((float(sqrt(x * x + y * y)) * 2.0)) / 1.0) * vec3(v), (b + b2 + b3 + b4) / 4.00);
  261. })");
  262. circle_ok_color_shader.instantiate();
  263. circle_ok_color_shader->set_code(OK_COLOR_SHADER + R"(
  264. // ColorPicker ok color hsl circle shader.
  265. uniform float ok_hsl_l = 1.0;
  266. void fragment() {
  267. float x = UV.x - 0.5;
  268. float y = UV.y - 0.5;
  269. float h = atan(y, x) / (2.0 * M_PI);
  270. float s = sqrt(x * x + y * y) * 2.0;
  271. vec3 col = okhsl_to_srgb(vec3(h, s, ok_hsl_l));
  272. x += 0.001;
  273. y += 0.001;
  274. float b = float(sqrt(x * x + y * y) < 0.5);
  275. x -= 0.002;
  276. float b2 = float(sqrt(x * x + y * y) < 0.5);
  277. y -= 0.002;
  278. float b3 = float(sqrt(x * x + y * y) < 0.5);
  279. x += 0.002;
  280. float b4 = float(sqrt(x * x + y * y) < 0.5);
  281. COLOR = vec4(col, (b + b2 + b3 + b4) / 4.00);
  282. })");
  283. rectangle_ok_color_hs_shader.instantiate();
  284. rectangle_ok_color_hs_shader->set_code(OK_COLOR_SHADER + R"(
  285. // ColorPicker ok color hs rectangle shader.
  286. uniform float ok_hsl_l = 0.0;
  287. void fragment() {
  288. float h = UV.x;
  289. float s = 1.0 - UV.y;
  290. vec3 col = okhsl_to_srgb(vec3(h, s, ok_hsl_l));
  291. COLOR = vec4(col, 1.0);
  292. })");
  293. rectangle_ok_color_hl_shader.instantiate();
  294. rectangle_ok_color_hl_shader->set_code(OK_COLOR_SHADER + R"(
  295. // ColorPicker ok color hl rectangle shader.
  296. uniform float ok_hsl_s = 0.0;
  297. void fragment() {
  298. float h = UV.x;
  299. float l = 1.0 - UV.y;
  300. vec3 col = okhsl_to_srgb(vec3(h, ok_hsl_s, l));
  301. COLOR = vec4(col, 1.0);
  302. })");
  303. }
  304. void ColorPicker::finish_shaders() {
  305. wheel_shader.unref();
  306. circle_shader.unref();
  307. circle_ok_color_shader.unref();
  308. rectangle_ok_color_hs_shader.unref();
  309. rectangle_ok_color_hl_shader.unref();
  310. }
  311. void ColorPicker::set_focus_on_line_edit() {
  312. callable_mp((Control *)c_text, &Control::grab_focus).call_deferred();
  313. }
  314. void ColorPicker::set_focus_on_picker_shape() {
  315. shapes[get_current_shape_index()]->grab_focus();
  316. }
  317. void ColorPicker::_update_controls() {
  318. int mode_sliders_count = modes[current_mode]->get_slider_count();
  319. for (int i = current_slider_count; i < mode_sliders_count; i++) {
  320. sliders[i]->show();
  321. labels[i]->show();
  322. values[i]->show();
  323. }
  324. for (int i = mode_sliders_count; i < current_slider_count; i++) {
  325. sliders[i]->hide();
  326. labels[i]->hide();
  327. values[i]->hide();
  328. }
  329. current_slider_count = mode_sliders_count;
  330. for (int i = 0; i < current_slider_count; i++) {
  331. labels[i]->set_text(modes[current_mode]->get_slider_label(i));
  332. sliders[i]->set_accessibility_name(modes[current_mode]->get_slider_label(i));
  333. values[i]->set_accessibility_name(modes[current_mode]->get_slider_label(i));
  334. }
  335. alpha_label->set_text("A");
  336. alpha_slider->set_accessibility_name(ETR("Alpha"));
  337. alpha_value->set_accessibility_name(ETR("Alpha"));
  338. intensity_label->set_text("I");
  339. intensity_slider->set_accessibility_name(ETR("Intensity"));
  340. intensity_value->set_accessibility_name(ETR("Intensity"));
  341. alpha_value->set_visible(edit_alpha);
  342. alpha_slider->set_visible(edit_alpha);
  343. alpha_label->set_visible(edit_alpha);
  344. intensity_value->set_visible(edit_intensity);
  345. intensity_slider->set_visible(edit_intensity);
  346. intensity_label->set_visible(edit_intensity);
  347. int i = 0;
  348. for (ColorPickerShape *shape : shapes) {
  349. bool is_active = get_current_shape_index() == i;
  350. i++;
  351. if (!shape->is_initialized) {
  352. if (is_active) {
  353. // Controls are initialized on demand, because ColorPicker does not need them all at once.
  354. shape->initialize_controls();
  355. } else {
  356. continue;
  357. }
  358. }
  359. for (Control *control : shape->controls) {
  360. control->set_visible(is_active);
  361. }
  362. }
  363. btn_shape->set_visible(current_shape != SHAPE_NONE);
  364. }
  365. void ColorPicker::_set_pick_color(const Color &p_color, bool p_update_sliders, bool p_calc_intensity) {
  366. if (text_changed) {
  367. add_recent_preset(color);
  368. text_changed = false;
  369. }
  370. color = p_color;
  371. if (p_calc_intensity) {
  372. _copy_color_to_normalized_and_intensity();
  373. }
  374. _copy_normalized_to_hsv_okhsl();
  375. if (!is_inside_tree()) {
  376. return;
  377. }
  378. _update_color(p_update_sliders);
  379. }
  380. void ColorPicker::set_pick_color(const Color &p_color) {
  381. _set_pick_color(p_color, true, true); // Because setters can't have more arguments.
  382. }
  383. void ColorPicker::set_old_color(const Color &p_color) {
  384. old_color = p_color;
  385. }
  386. void ColorPicker::set_display_old_color(bool p_enabled) {
  387. display_old_color = p_enabled;
  388. }
  389. bool ColorPicker::is_displaying_old_color() const {
  390. return display_old_color;
  391. }
  392. void ColorPicker::set_edit_alpha(bool p_show) {
  393. if (edit_alpha == p_show) {
  394. return;
  395. }
  396. edit_alpha = p_show;
  397. _update_controls();
  398. if (!is_inside_tree()) {
  399. return;
  400. }
  401. _update_color();
  402. sample->queue_redraw();
  403. }
  404. bool ColorPicker::is_editing_alpha() const {
  405. return edit_alpha;
  406. }
  407. void ColorPicker::set_edit_intensity(bool p_show) {
  408. if (edit_intensity == p_show) {
  409. return;
  410. }
  411. if (p_show) {
  412. set_pick_color(color);
  413. } else {
  414. _normalized_apply_intensity_to_color();
  415. color_normalized = color;
  416. intensity = 0;
  417. }
  418. edit_intensity = p_show;
  419. _update_controls();
  420. if (!is_inside_tree()) {
  421. return;
  422. }
  423. _update_color();
  424. sample->queue_redraw();
  425. }
  426. bool ColorPicker::is_editing_intensity() const {
  427. return edit_intensity;
  428. }
  429. void ColorPicker::_slider_drag_started() {
  430. currently_dragging = true;
  431. }
  432. void ColorPicker::_slider_value_changed() {
  433. if (updating) {
  434. return;
  435. }
  436. intensity = intensity_value->get_value();
  437. color_normalized = modes[current_mode]->get_color();
  438. if (edit_intensity && is_color_overbright(color_normalized)) {
  439. modes[current_mode]->_greater_value_inputted();
  440. color_normalized = modes[current_mode]->get_color();
  441. }
  442. _normalized_apply_intensity_to_color();
  443. intensity_value->set_prefix(intensity < 0 ? "" : "+");
  444. modes[current_mode]->_value_changed();
  445. _set_pick_color(color, false, false);
  446. if (!deferred_mode_enabled || !currently_dragging) {
  447. emit_signal(SNAME("color_changed"), color);
  448. }
  449. }
  450. void ColorPicker::_slider_drag_ended() {
  451. currently_dragging = false;
  452. if (deferred_mode_enabled) {
  453. emit_signal(SNAME("color_changed"), color);
  454. }
  455. }
  456. void ColorPicker::add_mode(ColorMode *p_mode) {
  457. modes.push_back(p_mode);
  458. }
  459. void ColorPicker::add_shape(ColorPickerShape *p_shape) {
  460. shapes.push_back(p_shape);
  461. }
  462. void ColorPicker::create_slider(GridContainer *gc, int idx) {
  463. Label *lbl = memnew(Label);
  464. lbl->set_v_size_flags(SIZE_SHRINK_CENTER);
  465. lbl->set_auto_translate_mode(AUTO_TRANSLATE_MODE_DISABLED);
  466. gc->add_child(lbl);
  467. HSlider *slider = memnew(HSlider);
  468. slider->set_v_size_flags(SIZE_SHRINK_CENTER);
  469. slider->set_focus_mode(FOCUS_ACCESSIBILITY);
  470. gc->add_child(slider);
  471. SpinBox *val = memnew(SpinBox);
  472. slider->share(val);
  473. val->set_select_all_on_focus(true);
  474. gc->add_child(val);
  475. LineEdit *vle = val->get_line_edit();
  476. vle->connect(SceneStringName(text_changed), callable_mp(this, &ColorPicker::_text_changed));
  477. vle->connect(SceneStringName(gui_input), callable_mp(this, &ColorPicker::_line_edit_input));
  478. vle->set_horizontal_alignment(HORIZONTAL_ALIGNMENT_RIGHT);
  479. val->connect(SceneStringName(gui_input), callable_mp(this, &ColorPicker::_slider_or_spin_input));
  480. slider->set_h_size_flags(SIZE_EXPAND_FILL);
  481. slider->connect("drag_started", callable_mp(this, &ColorPicker::_slider_drag_started));
  482. slider->connect(SceneStringName(value_changed), callable_mp(this, &ColorPicker::_slider_value_changed).unbind(1));
  483. slider->connect("drag_ended", callable_mp(this, &ColorPicker::_slider_drag_ended).unbind(1));
  484. if (idx < MODE_SLIDER_COUNT) {
  485. slider->connect(SceneStringName(draw), callable_mp(this, &ColorPicker::_slider_draw).bind(idx));
  486. } else if (idx == SLIDER_ALPHA) {
  487. slider->connect(SceneStringName(draw), callable_mp(this, &ColorPicker::_alpha_slider_draw));
  488. }
  489. slider->connect(SceneStringName(gui_input), callable_mp(this, &ColorPicker::_slider_or_spin_input));
  490. if (idx < MODE_SLIDER_COUNT) {
  491. sliders[idx] = slider;
  492. values[idx] = val;
  493. labels[idx] = lbl;
  494. } else if (idx == SLIDER_INTENSITY) {
  495. intensity_slider = slider;
  496. intensity_value = val;
  497. intensity_label = lbl;
  498. } else if (idx == SLIDER_ALPHA) {
  499. alpha_slider = slider;
  500. alpha_value = val;
  501. alpha_label = lbl;
  502. }
  503. }
  504. #ifdef TOOLS_ENABLED
  505. void ColorPicker::set_editor_settings(Object *p_editor_settings) {
  506. if (editor_settings) {
  507. return;
  508. }
  509. editor_settings = p_editor_settings;
  510. if (preset_cache.is_empty()) {
  511. PackedColorArray saved_presets = editor_settings->call(SNAME("get_project_metadata"), "color_picker", "presets", PackedColorArray());
  512. for (int i = 0; i < saved_presets.size(); i++) {
  513. preset_cache.push_back(saved_presets[i]);
  514. }
  515. }
  516. for (const Color &preset : preset_cache) {
  517. presets.push_back(preset);
  518. }
  519. if (recent_preset_cache.is_empty()) {
  520. PackedColorArray saved_recent_presets = editor_settings->call(SNAME("get_project_metadata"), "color_picker", "recent_presets", PackedColorArray());
  521. for (int i = 0; i < saved_recent_presets.size(); i++) {
  522. recent_preset_cache.push_back(saved_recent_presets[i]);
  523. }
  524. }
  525. for (const Color &preset : recent_preset_cache) {
  526. recent_presets.push_back(preset);
  527. }
  528. _update_presets();
  529. _update_recent_presets();
  530. }
  531. void ColorPicker::set_quick_open_callback(const Callable &p_file_selected) {
  532. quick_open_callback = p_file_selected;
  533. }
  534. void ColorPicker::set_palette_saved_callback(const Callable &p_palette_saved) {
  535. palette_saved_callback = p_palette_saved;
  536. }
  537. #endif
  538. HSlider *ColorPicker::get_slider(int p_idx) {
  539. ERR_FAIL_INDEX_V(p_idx, MODE_MAX, nullptr);
  540. return sliders[p_idx];
  541. }
  542. Vector<float> ColorPicker::get_active_slider_values() {
  543. Vector<float> cur_values;
  544. for (int i = 0; i < current_slider_count; i++) {
  545. cur_values.push_back(sliders[i]->get_value());
  546. }
  547. cur_values.push_back(alpha_slider->get_value());
  548. return cur_values;
  549. }
  550. void ColorPicker::_copy_normalized_to_hsv_okhsl() {
  551. if (!okhsl_cached) {
  552. ok_hsl_h = color_normalized.get_ok_hsl_h();
  553. ok_hsl_s = color_normalized.get_ok_hsl_s();
  554. ok_hsl_l = color_normalized.get_ok_hsl_l();
  555. }
  556. if (!hsv_cached) {
  557. h = color_normalized.get_h();
  558. s = color_normalized.get_s();
  559. v = color_normalized.get_v();
  560. }
  561. hsv_cached = false;
  562. okhsl_cached = false;
  563. }
  564. void ColorPicker::_copy_hsv_okhsl_to_normalized() {
  565. if (current_shape != SHAPE_NONE && shapes[get_current_shape_index()]->is_ok_hsl()) {
  566. color_normalized.set_ok_hsl(ok_hsl_h, ok_hsl_s, ok_hsl_l, color_normalized.a);
  567. } else {
  568. color_normalized.set_hsv(h, s, v, color_normalized.a);
  569. }
  570. }
  571. Color ColorPicker::_color_apply_intensity(const Color &col) const {
  572. Color linear_color = col.srgb_to_linear();
  573. Color result;
  574. float multiplier = Math::pow(2, intensity);
  575. for (int i = 0; i < 3; i++) {
  576. result.components[i] = linear_color.components[i] * multiplier;
  577. }
  578. result.a = col.a;
  579. return result.linear_to_srgb();
  580. }
  581. void ColorPicker::_normalized_apply_intensity_to_color() {
  582. color = _color_apply_intensity(color_normalized);
  583. }
  584. void ColorPicker::_copy_color_to_normalized_and_intensity() {
  585. Color linear_color = color.srgb_to_linear();
  586. float multiplier = MAX(1, MAX(MAX(linear_color.r, linear_color.g), linear_color.b));
  587. for (int i = 0; i < 3; i++) {
  588. color_normalized.components[i] = linear_color.components[i] / multiplier;
  589. }
  590. color_normalized.a = linear_color.a;
  591. color_normalized = color_normalized.linear_to_srgb();
  592. intensity = Math::log2(multiplier);
  593. }
  594. void ColorPicker::_select_from_preset_container(const Color &p_color) {
  595. if (preset_group->get_pressed_button()) {
  596. preset_group->get_pressed_button()->set_pressed(false);
  597. }
  598. for (int i = 1; i < preset_container->get_child_count(); i++) {
  599. ColorPresetButton *current_btn = Object::cast_to<ColorPresetButton>(preset_container->get_child(i));
  600. if (current_btn && p_color == current_btn->get_preset_color()) {
  601. current_btn->set_pressed(true);
  602. break;
  603. }
  604. }
  605. }
  606. bool ColorPicker::_select_from_recent_preset_hbc(const Color &p_color) {
  607. for (int i = 0; i < recent_preset_hbc->get_child_count(); i++) {
  608. ColorPresetButton *current_btn = Object::cast_to<ColorPresetButton>(recent_preset_hbc->get_child(i));
  609. if (current_btn && p_color == current_btn->get_preset_color()) {
  610. current_btn->set_pressed(true);
  611. return true;
  612. }
  613. }
  614. return false;
  615. }
  616. void ColorPicker::_reset_sliders_theme() {
  617. Ref<StyleBoxFlat> style_box_flat(memnew(StyleBoxFlat));
  618. style_box_flat->set_content_margin(SIDE_TOP, 16 * theme_cache.base_scale);
  619. style_box_flat->set_bg_color(Color(0.2, 0.23, 0.31).lerp(Color(0, 0, 0, 1), 0.3).clamp());
  620. for (int i = 0; i < MODE_SLIDER_COUNT; i++) {
  621. sliders[i]->begin_bulk_theme_override();
  622. sliders[i]->add_theme_icon_override(SNAME("grabber"), theme_cache.bar_arrow);
  623. sliders[i]->add_theme_icon_override(SNAME("grabber_highlight"), theme_cache.bar_arrow);
  624. sliders[i]->add_theme_constant_override(SNAME("grabber_offset"), 8 * theme_cache.base_scale);
  625. if (!colorize_sliders) {
  626. sliders[i]->add_theme_style_override(SNAME("slider"), style_box_flat);
  627. }
  628. sliders[i]->end_bulk_theme_override();
  629. }
  630. alpha_slider->begin_bulk_theme_override();
  631. alpha_slider->add_theme_icon_override(SNAME("grabber"), theme_cache.bar_arrow);
  632. alpha_slider->add_theme_icon_override(SNAME("grabber_highlight"), theme_cache.bar_arrow);
  633. alpha_slider->add_theme_constant_override(SNAME("grabber_offset"), 8 * theme_cache.base_scale);
  634. if (!colorize_sliders) {
  635. alpha_slider->add_theme_style_override(SNAME("slider"), style_box_flat);
  636. }
  637. alpha_slider->end_bulk_theme_override();
  638. }
  639. void ColorPicker::_html_submitted(const String &p_html) {
  640. if (updating) {
  641. return;
  642. }
  643. Color new_color = color;
  644. if (text_is_constructor || !is_color_valid_hex(color)) {
  645. Ref<Expression> expr;
  646. expr.instantiate();
  647. Error err = expr->parse(p_html);
  648. if (err == OK) {
  649. Variant result = expr->execute(Array(), nullptr, false, true);
  650. // This is basically the same as Variant::operator Color(), but remains original color if Color::from_string() fails
  651. if (result.get_type() == Variant::COLOR) {
  652. new_color = result;
  653. } else if (result.get_type() == Variant::STRING) {
  654. new_color = Color::from_string(result, color);
  655. } else if (result.get_type() == Variant::INT) {
  656. new_color = Color::hex(result);
  657. }
  658. }
  659. } else {
  660. new_color = Color::from_string(p_html.strip_edges(), color);
  661. String html_no_prefix = p_html.strip_edges().trim_prefix("#");
  662. if (html_no_prefix.is_valid_hex_number(false)) {
  663. // Convert invalid HTML color codes that software like Figma supports.
  664. if (html_no_prefix.length() == 1) {
  665. // Turn `#1` into `#111111`.
  666. html_no_prefix = html_no_prefix.repeat(6);
  667. } else if (html_no_prefix.length() == 2) {
  668. // Turn `#12` into `#121212`.
  669. html_no_prefix = html_no_prefix.repeat(3);
  670. } else if (html_no_prefix.length() == 5) {
  671. // Turn `#12345` into `#11223344`.
  672. html_no_prefix = html_no_prefix.left(4);
  673. } else if (html_no_prefix.length() == 7) {
  674. // Turn `#1234567` into `#123456`.
  675. html_no_prefix = html_no_prefix.left(6);
  676. }
  677. }
  678. new_color = Color::from_string(html_no_prefix, new_color);
  679. }
  680. if (!is_editing_alpha()) {
  681. new_color.a = color.a;
  682. }
  683. if (new_color == color) {
  684. return;
  685. }
  686. color = new_color;
  687. if (!is_inside_tree()) {
  688. return;
  689. }
  690. set_pick_color(color);
  691. emit_signal(SNAME("color_changed"), color);
  692. }
  693. void ColorPicker::_update_color(bool p_update_sliders) {
  694. updating = true;
  695. if (p_update_sliders) {
  696. float step = modes[current_mode]->get_slider_step();
  697. float spinbox_arrow_step = modes[current_mode]->get_spinbox_arrow_step();
  698. for (int i = 0; i < current_slider_count; i++) {
  699. sliders[i]->set_max(modes[current_mode]->get_slider_max(i));
  700. sliders[i]->set_step(step);
  701. sliders[i]->set_value(modes[current_mode]->get_slider_value(i));
  702. values[i]->set_custom_arrow_step(spinbox_arrow_step);
  703. values[i]->set_allow_greater(modes[current_mode]->get_allow_greater());
  704. }
  705. alpha_slider->set_max(modes[current_mode]->get_alpha_slider_max());
  706. alpha_slider->set_step(step);
  707. alpha_slider->set_value(modes[current_mode]->get_alpha_slider_value());
  708. intensity_slider->set_value(intensity);
  709. intensity_value->set_prefix(intensity < 0 ? "" : "+");
  710. }
  711. _update_text_value();
  712. if (current_shape != SHAPE_NONE) {
  713. for (Control *control : shapes[get_current_shape_index()]->controls) {
  714. control->queue_redraw();
  715. }
  716. }
  717. sample->queue_redraw();
  718. for (int i = 0; i < current_slider_count; i++) {
  719. sliders[i]->queue_redraw();
  720. }
  721. alpha_slider->queue_redraw();
  722. updating = false;
  723. queue_accessibility_update();
  724. }
  725. void ColorPicker::_update_presets() {
  726. int preset_size = _get_preset_size();
  727. // Only update the preset button size if it has changed.
  728. if (preset_size != prev_preset_size) {
  729. prev_preset_size = preset_size;
  730. btn_add_preset->set_custom_minimum_size(Size2(preset_size, preset_size));
  731. for (int i = 1; i < preset_container->get_child_count(); i++) {
  732. ColorPresetButton *cpb = Object::cast_to<ColorPresetButton>(preset_container->get_child(i));
  733. cpb->set_custom_minimum_size(Size2(preset_size, preset_size));
  734. }
  735. }
  736. #ifdef TOOLS_ENABLED
  737. if (editor_settings) {
  738. String cached_name = editor_settings->call(SNAME("get_project_metadata"), "color_picker", "palette_name", String());
  739. palette_path = editor_settings->call(SNAME("get_project_metadata"), "color_picker", "palette_path", String());
  740. bool palette_edited = editor_settings->call(SNAME("get_project_metadata"), "color_picker", "palette_edited", false);
  741. if (!cached_name.is_empty()) {
  742. palette_name->set_text(cached_name);
  743. if (btn_preset->is_pressed() && !presets.is_empty()) {
  744. palette_name->show();
  745. }
  746. if (palette_edited) {
  747. palette_name->set_text(vformat("%s*", palette_name->get_text().remove_char('*')));
  748. palette_name->set_tooltip_text(TTRC("The changes to this palette have not been saved to a file."));
  749. }
  750. }
  751. }
  752. #endif
  753. // Rebuild swatch color buttons, keeping the add-preset button in the first position.
  754. for (int i = 1; i < preset_container->get_child_count(); i++) {
  755. preset_container->get_child(i)->queue_free();
  756. }
  757. presets = preset_cache;
  758. for (const Color &preset : preset_cache) {
  759. _add_preset_button(preset_size, preset);
  760. }
  761. _notification(NOTIFICATION_VISIBILITY_CHANGED);
  762. }
  763. void ColorPicker::_update_recent_presets() {
  764. #ifdef TOOLS_ENABLED
  765. if (editor_settings) {
  766. int recent_preset_count = recent_preset_hbc->get_child_count();
  767. for (int i = 0; i < recent_preset_count; i++) {
  768. memdelete(recent_preset_hbc->get_child(0));
  769. }
  770. recent_presets.clear();
  771. for (const Color &preset : recent_preset_cache) {
  772. recent_presets.push_back(preset);
  773. }
  774. int preset_size = _get_preset_size();
  775. for (const Color &preset : recent_presets) {
  776. _add_recent_preset_button(preset_size, preset);
  777. }
  778. _notification(NOTIFICATION_VISIBILITY_CHANGED);
  779. }
  780. #endif
  781. }
  782. #ifdef TOOLS_ENABLED
  783. void ColorPicker::_text_type_toggled() {
  784. text_is_constructor = !text_is_constructor;
  785. if (text_is_constructor) {
  786. hex_label->set_text(ETR("Expr"));
  787. text_type->set_text("");
  788. text_type->set_button_icon(theme_cache.color_script);
  789. c_text->set_tooltip_text(RTR("Execute an expression as a color."));
  790. } else {
  791. hex_label->set_text(ETR("Hex"));
  792. text_type->set_text("#");
  793. text_type->set_button_icon(nullptr);
  794. c_text->set_tooltip_text(ETR("Enter a hex code (\"#ff0000\") or named color (\"red\")."));
  795. }
  796. _update_color();
  797. }
  798. #endif // TOOLS_ENABLED
  799. Color ColorPicker::get_pick_color() const {
  800. return color;
  801. }
  802. Color ColorPicker::get_old_color() const {
  803. return old_color;
  804. }
  805. void ColorPicker::set_picker_shape(PickerShapeType p_shape) {
  806. ERR_FAIL_INDEX(p_shape, SHAPE_MAX);
  807. if (p_shape == current_shape) {
  808. return;
  809. }
  810. if (current_shape != SHAPE_NONE) {
  811. shape_popup->set_item_checked(get_current_shape_index(), false);
  812. }
  813. if (p_shape != SHAPE_NONE) {
  814. shape_popup->set_item_checked(shape_to_index(p_shape), true);
  815. btn_shape->set_button_icon(shape_popup->get_item_icon(shape_to_index(p_shape)));
  816. }
  817. current_shape = p_shape;
  818. #ifdef TOOLS_ENABLED
  819. if (editor_settings) {
  820. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "picker_shape", current_shape);
  821. }
  822. #endif
  823. _update_controls();
  824. _update_color();
  825. }
  826. ColorPicker::PickerShapeType ColorPicker::get_picker_shape() const {
  827. return current_shape;
  828. }
  829. inline int ColorPicker::_get_preset_size() {
  830. return (int(get_minimum_size().width) - (preset_container->get_h_separation() * (PRESET_COLUMN_COUNT - 1))) / PRESET_COLUMN_COUNT;
  831. }
  832. void ColorPicker::_add_preset_button(int p_size, const Color &p_color) {
  833. ColorPresetButton *btn_preset_new = memnew(ColorPresetButton(p_color, p_size, false));
  834. SET_DRAG_FORWARDING_GCDU(btn_preset_new, ColorPicker);
  835. btn_preset_new->set_button_group(preset_group);
  836. preset_container->add_child(btn_preset_new);
  837. btn_preset_new->set_pressed(true);
  838. btn_preset_new->connect(SceneStringName(gui_input), callable_mp(this, &ColorPicker::_preset_input).bind(p_color));
  839. }
  840. void ColorPicker::_add_recent_preset_button(int p_size, const Color &p_color) {
  841. ColorPresetButton *btn_preset_new = memnew(ColorPresetButton(p_color, p_size, true));
  842. btn_preset_new->set_button_group(recent_preset_group);
  843. recent_preset_hbc->add_child(btn_preset_new);
  844. recent_preset_hbc->move_child(btn_preset_new, 0);
  845. btn_preset_new->set_pressed(true);
  846. btn_preset_new->connect(SceneStringName(toggled), callable_mp(this, &ColorPicker::_recent_preset_pressed).bind(btn_preset_new));
  847. }
  848. void ColorPicker::_load_palette() {
  849. List<String> extensions;
  850. ResourceLoader::get_recognized_extensions_for_type("ColorPalette", &extensions);
  851. file_dialog->set_title(ETR("Load Color Palette"));
  852. file_dialog->clear_filters();
  853. for (const String &K : extensions) {
  854. file_dialog->add_filter("*." + K);
  855. }
  856. file_dialog->set_file_mode(FileDialog::FILE_MODE_OPEN_FILE);
  857. file_dialog->set_current_file("");
  858. file_dialog->popup_centered_ratio();
  859. }
  860. void ColorPicker::_save_palette(bool p_is_save_as) {
  861. if (!p_is_save_as && !palette_path.is_empty()) {
  862. file_dialog->set_file_mode(FileDialog::FILE_MODE_SAVE_FILE);
  863. _palette_file_selected(palette_path);
  864. return;
  865. } else {
  866. List<String> extensions;
  867. ResourceLoader::get_recognized_extensions_for_type("ColorPalette", &extensions);
  868. file_dialog->set_title(ETR("Save Color Palette"));
  869. file_dialog->clear_filters();
  870. for (const String &K : extensions) {
  871. file_dialog->add_filter("*." + K);
  872. }
  873. file_dialog->set_file_mode(FileDialog::FILE_MODE_SAVE_FILE);
  874. file_dialog->set_current_file("new_palette.tres");
  875. file_dialog->popup_centered_ratio();
  876. }
  877. }
  878. #ifdef TOOLS_ENABLED
  879. void ColorPicker::_quick_open_palette_file_selected(const String &p_path) {
  880. _ensure_file_dialog();
  881. file_dialog->set_file_mode(FileDialog::FILE_MODE_OPEN_FILE);
  882. _palette_file_selected(p_path);
  883. }
  884. GridContainer *ColorPicker::get_slider_container() {
  885. return slider_gc;
  886. }
  887. #endif // ifdef TOOLS_ENABLED
  888. void ColorPicker::_palette_file_selected(const String &p_path) {
  889. switch (file_dialog->get_file_mode()) {
  890. case FileDialog::FileMode::FILE_MODE_OPEN_FILE: {
  891. Ref<ColorPalette> palette = ResourceLoader::load(p_path, "", ResourceFormatLoader::CACHE_MODE_IGNORE);
  892. ERR_FAIL_COND_MSG(palette.is_null(), vformat("Cannot open color palette file for reading at: %s", p_path));
  893. preset_cache.clear();
  894. presets.clear();
  895. PackedColorArray saved_presets = palette->get_colors();
  896. for (const Color &saved_preset : saved_presets) {
  897. preset_cache.push_back(saved_preset);
  898. presets.push_back(saved_preset);
  899. }
  900. #ifdef TOOLS_ENABLED
  901. if (editor_settings) {
  902. const StringName set_project_metadata = SNAME("set_project_metadata");
  903. editor_settings->call(set_project_metadata, "color_picker", "presets", saved_presets);
  904. editor_settings->call(set_project_metadata, "color_picker", "palette_edited", false);
  905. }
  906. #endif
  907. } break;
  908. case FileDialog::FileMode::FILE_MODE_SAVE_FILE: {
  909. ColorPalette *palette = memnew(ColorPalette);
  910. palette->set_colors(get_presets());
  911. Error error = ResourceSaver::save(palette, p_path);
  912. ERR_FAIL_COND_MSG(error != Error::OK, vformat("Cannot open color palette file for writing at: %s", p_path));
  913. #ifdef TOOLS_ENABLED
  914. if (palette_saved_callback.is_valid()) {
  915. palette_saved_callback.call_deferred(p_path);
  916. }
  917. #endif // TOOLS_ENABLED
  918. } break;
  919. default:
  920. break;
  921. }
  922. palette_name->set_text(p_path.get_file().get_basename());
  923. palette_name->set_tooltip_text("");
  924. palette_name->show();
  925. palette_path = p_path;
  926. btn_preset->set_pressed(true);
  927. #ifdef TOOLS_ENABLED
  928. if (editor_settings) {
  929. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "palette_name", palette_name->get_text());
  930. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "palette_path", palette_path);
  931. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "palette_edited", false);
  932. }
  933. #endif
  934. _update_presets();
  935. }
  936. void ColorPicker::_show_hide_preset(const bool &p_is_btn_pressed, Button *p_btn_preset, Container *p_preset_container) {
  937. if (p_is_btn_pressed) {
  938. p_preset_container->show();
  939. } else {
  940. p_preset_container->hide();
  941. }
  942. _update_drop_down_arrow(p_is_btn_pressed, p_btn_preset);
  943. palette_name->hide();
  944. if (btn_preset->is_pressed() && !palette_name->get_text().is_empty()) {
  945. palette_name->show();
  946. }
  947. }
  948. void ColorPicker::_update_drop_down_arrow(const bool &p_is_btn_pressed, Button *p_btn_preset) {
  949. if (p_is_btn_pressed) {
  950. p_btn_preset->set_button_icon(theme_cache.expanded_arrow);
  951. } else {
  952. p_btn_preset->set_button_icon(theme_cache.folded_arrow);
  953. }
  954. }
  955. void ColorPicker::_set_mode_popup_value(ColorModeType p_mode) {
  956. ERR_FAIL_INDEX(p_mode, MODE_MAX + 1);
  957. if (p_mode == MODE_MAX) {
  958. set_colorize_sliders(!colorize_sliders);
  959. } else {
  960. set_color_mode(p_mode);
  961. }
  962. }
  963. Variant ColorPicker::_get_drag_data_fw(const Point2 &p_point, Control *p_from_control) {
  964. ColorPresetButton *dragged_preset_button = Object::cast_to<ColorPresetButton>(p_from_control);
  965. if (!dragged_preset_button) {
  966. return Variant();
  967. }
  968. ColorPresetButton *drag_preview = memnew(ColorPresetButton(dragged_preset_button->get_preset_color(), _get_preset_size(), false));
  969. set_drag_preview(drag_preview);
  970. Dictionary drag_data;
  971. drag_data["type"] = "color_preset";
  972. drag_data["color_preset"] = dragged_preset_button->get_index();
  973. return drag_data;
  974. }
  975. bool ColorPicker::_can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from_control) const {
  976. Dictionary d = p_data;
  977. if (!d.has("type") || String(d["type"]) != "color_preset") {
  978. return false;
  979. }
  980. return true;
  981. }
  982. void ColorPicker::_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from_control) {
  983. Dictionary d = p_data;
  984. if (!d.has("type")) {
  985. return;
  986. }
  987. if (String(d["type"]) == "color_preset") {
  988. int preset_from_id = d["color_preset"];
  989. int hover_now = p_from_control->get_index();
  990. if (preset_from_id == hover_now || hover_now == -1) {
  991. return;
  992. }
  993. preset_container->move_child(preset_container->get_child(preset_from_id), hover_now);
  994. }
  995. }
  996. void ColorPicker::_ensure_file_dialog() {
  997. if (file_dialog) {
  998. return;
  999. }
  1000. file_dialog = memnew(FileDialog);
  1001. file_dialog->set_mode_overrides_title(false);
  1002. file_dialog->set_access(FileDialog::ACCESS_FILESYSTEM);
  1003. file_dialog->set_current_dir(Engine::get_singleton()->is_editor_hint() ? "res://" : "user://");
  1004. add_child(file_dialog, false, INTERNAL_MODE_FRONT);
  1005. file_dialog->connect("file_selected", callable_mp(this, &ColorPicker::_palette_file_selected));
  1006. }
  1007. void ColorPicker::add_preset(const Color &p_color) {
  1008. List<Color>::Element *e = presets.find(p_color);
  1009. if (e) {
  1010. presets.move_to_back(e);
  1011. preset_container->move_child(preset_group->get_pressed_button(), preset_container->get_child_count() - 1);
  1012. } else {
  1013. presets.push_back(p_color);
  1014. _add_preset_button(_get_preset_size(), p_color);
  1015. }
  1016. List<Color>::Element *cache_e = preset_cache.find(p_color);
  1017. if (cache_e) {
  1018. preset_cache.move_to_back(cache_e);
  1019. } else {
  1020. preset_cache.push_back(p_color);
  1021. }
  1022. if (!palette_name->get_text().is_empty()) {
  1023. palette_name->set_text(vformat("%s*", palette_name->get_text().trim_suffix("*")));
  1024. palette_name->set_tooltip_text(ETR("The changes to this palette have not been saved to a file."));
  1025. }
  1026. #ifdef TOOLS_ENABLED
  1027. if (editor_settings) {
  1028. PackedColorArray arr_to_save = get_presets();
  1029. const StringName set_project_metadata = SNAME("set_project_metadata");
  1030. editor_settings->call(set_project_metadata, "color_picker", "presets", arr_to_save);
  1031. editor_settings->call(set_project_metadata, "color_picker", "palette_edited", true);
  1032. }
  1033. #endif
  1034. }
  1035. void ColorPicker::add_recent_preset(const Color &p_color) {
  1036. if (!_select_from_recent_preset_hbc(p_color)) {
  1037. if (recent_preset_hbc->get_child_count() >= PRESET_COLUMN_COUNT) {
  1038. recent_preset_cache.pop_front();
  1039. recent_presets.pop_front();
  1040. recent_preset_hbc->get_child(PRESET_COLUMN_COUNT - 1)->queue_free();
  1041. }
  1042. recent_presets.push_back(p_color);
  1043. recent_preset_cache.push_back(p_color);
  1044. _add_recent_preset_button(_get_preset_size(), p_color);
  1045. }
  1046. _select_from_preset_container(p_color);
  1047. #ifdef TOOLS_ENABLED
  1048. if (editor_settings) {
  1049. PackedColorArray arr_to_save = get_recent_presets();
  1050. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "recent_presets", arr_to_save);
  1051. }
  1052. #endif
  1053. }
  1054. void ColorPicker::erase_preset(const Color &p_color) {
  1055. List<Color>::Element *e = presets.find(p_color);
  1056. if (e) {
  1057. presets.erase(e);
  1058. preset_cache.erase(preset_cache.find(p_color));
  1059. // Find preset button to remove.
  1060. for (int i = 1; i < preset_container->get_child_count(); i++) {
  1061. ColorPresetButton *current_btn = Object::cast_to<ColorPresetButton>(preset_container->get_child(i));
  1062. if (current_btn && p_color == current_btn->get_preset_color()) {
  1063. current_btn->queue_free();
  1064. // Removing focused control loose the focus totally. We focus on previous button to keep it possible to navigate with keyboard/joypad.
  1065. Control *focus_target = Object::cast_to<Control>(preset_container->get_child(i - 1));
  1066. focus_target->grab_focus();
  1067. break;
  1068. }
  1069. }
  1070. palette_name->set_text(vformat("%s*", palette_name->get_text().remove_char('*')));
  1071. palette_name->set_tooltip_text(ETR("The changes to this palette have not been saved to a file."));
  1072. if (presets.is_empty()) {
  1073. palette_name->set_text("");
  1074. palette_path = String();
  1075. palette_name->hide();
  1076. }
  1077. #ifdef TOOLS_ENABLED
  1078. if (editor_settings) {
  1079. PackedColorArray arr_to_save = get_presets();
  1080. const StringName set_project_metadata = SNAME("set_project_metadata");
  1081. editor_settings->call(set_project_metadata, "color_picker", "presets", arr_to_save);
  1082. editor_settings->call(set_project_metadata, "color_picker", "palette_edited", true);
  1083. editor_settings->call(set_project_metadata, "color_picker", "palette_name", palette_name->get_text());
  1084. editor_settings->call(set_project_metadata, "color_picker", "palette_path", palette_path);
  1085. }
  1086. #endif
  1087. }
  1088. }
  1089. void ColorPicker::erase_recent_preset(const Color &p_color) {
  1090. List<Color>::Element *e = recent_presets.find(p_color);
  1091. if (e) {
  1092. recent_presets.erase(e);
  1093. recent_preset_cache.erase(recent_preset_cache.find(p_color));
  1094. // Find recent preset button to remove.
  1095. for (int i = 1; i < recent_preset_hbc->get_child_count(); i++) {
  1096. ColorPresetButton *current_btn = Object::cast_to<ColorPresetButton>(recent_preset_hbc->get_child(i));
  1097. if (current_btn && p_color == current_btn->get_preset_color()) {
  1098. current_btn->queue_free();
  1099. break;
  1100. }
  1101. }
  1102. #ifdef TOOLS_ENABLED
  1103. if (editor_settings) {
  1104. PackedColorArray arr_to_save = get_recent_presets();
  1105. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "recent_presets", arr_to_save);
  1106. }
  1107. #endif
  1108. }
  1109. }
  1110. PackedColorArray ColorPicker::get_presets() const {
  1111. PackedColorArray arr;
  1112. arr.resize(presets.size());
  1113. int i = 0;
  1114. for (List<Color>::ConstIterator itr = presets.begin(); itr != presets.end(); ++itr, ++i) {
  1115. arr.set(i, *itr);
  1116. }
  1117. return arr;
  1118. }
  1119. PackedColorArray ColorPicker::get_recent_presets() const {
  1120. PackedColorArray arr;
  1121. arr.resize(recent_presets.size());
  1122. int i = 0;
  1123. for (List<Color>::ConstIterator itr = recent_presets.begin(); itr != recent_presets.end(); ++itr, ++i) {
  1124. arr.set(i, *itr);
  1125. }
  1126. return arr;
  1127. }
  1128. void ColorPicker::set_color_mode(ColorModeType p_mode) {
  1129. ERR_FAIL_INDEX(p_mode, MODE_MAX);
  1130. if (current_mode == p_mode) {
  1131. return;
  1132. }
  1133. mode_popup->set_item_checked(current_mode, false);
  1134. mode_popup->set_item_checked(p_mode, true);
  1135. if (p_mode < MODE_BUTTON_COUNT) {
  1136. mode_btns[p_mode]->set_pressed(true);
  1137. } else if (current_mode < MODE_BUTTON_COUNT) {
  1138. mode_btns[current_mode]->set_pressed(false);
  1139. }
  1140. current_mode = p_mode;
  1141. #ifdef TOOLS_ENABLED
  1142. if (editor_settings) {
  1143. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "color_mode", current_mode);
  1144. }
  1145. #endif
  1146. if (!is_inside_tree()) {
  1147. return;
  1148. }
  1149. _update_controls();
  1150. _update_color();
  1151. }
  1152. ColorPicker::ColorModeType ColorPicker::get_color_mode() const {
  1153. return current_mode;
  1154. }
  1155. void ColorPicker::set_colorize_sliders(bool p_colorize_sliders) {
  1156. if (colorize_sliders == p_colorize_sliders) {
  1157. return;
  1158. }
  1159. colorize_sliders = p_colorize_sliders;
  1160. mode_popup->set_item_checked(MODE_MAX + 1, colorize_sliders);
  1161. if (colorize_sliders) {
  1162. Ref<StyleBoxEmpty> style_box_empty(memnew(StyleBoxEmpty));
  1163. for (int i = 0; i < MODE_SLIDER_COUNT; i++) {
  1164. sliders[i]->add_theme_style_override("slider", style_box_empty);
  1165. }
  1166. alpha_slider->add_theme_style_override("slider", style_box_empty);
  1167. } else {
  1168. Ref<StyleBoxFlat> style_box_flat(memnew(StyleBoxFlat));
  1169. style_box_flat->set_content_margin(SIDE_TOP, 16 * theme_cache.base_scale);
  1170. style_box_flat->set_bg_color(Color(0.2, 0.23, 0.31).lerp(Color(0, 0, 0, 1), 0.3).clamp());
  1171. for (int i = 0; i < MODE_SLIDER_COUNT; i++) {
  1172. sliders[i]->add_theme_style_override("slider", style_box_flat);
  1173. }
  1174. alpha_slider->add_theme_style_override("slider", style_box_flat);
  1175. }
  1176. }
  1177. bool ColorPicker::is_colorizing_sliders() const {
  1178. return colorize_sliders;
  1179. }
  1180. void ColorPicker::set_deferred_mode(bool p_enabled) {
  1181. deferred_mode_enabled = p_enabled;
  1182. }
  1183. bool ColorPicker::is_deferred_mode() const {
  1184. return deferred_mode_enabled;
  1185. }
  1186. void ColorPicker::_update_text_value() {
  1187. if (text_is_constructor || !is_color_valid_hex(color)) {
  1188. String t = "Color" + color_to_string(color, edit_alpha && color.a < 1, true);
  1189. text_type->set_text("");
  1190. text_type->set_button_icon(theme_cache.color_script);
  1191. text_type->set_disabled(!is_color_valid_hex(color));
  1192. hex_label->set_text(ETR("Expr"));
  1193. c_text->set_text(t);
  1194. } else {
  1195. text_type->set_text("#");
  1196. text_type->set_button_icon(nullptr);
  1197. text_type->set_disabled(false);
  1198. hex_label->set_text(ETR("Hex"));
  1199. c_text->set_text(color.to_html(edit_alpha && color.a < 1));
  1200. }
  1201. }
  1202. void ColorPicker::_sample_input(const Ref<InputEvent> &p_event) {
  1203. if (!display_old_color) {
  1204. return;
  1205. }
  1206. const Ref<InputEventMouseButton> mb = p_event;
  1207. if (mb.is_valid() && mb->is_pressed() && mb->get_button_index() == MouseButton::LEFT) {
  1208. const Rect2 rect_old = Rect2(Point2(), Size2(sample->get_size().width * 0.5, sample->get_size().height * 0.95));
  1209. if (rect_old.has_point(mb->get_position())) {
  1210. // Revert to the old color when left-clicking the old color sample.
  1211. set_pick_color(old_color);
  1212. sample->set_focus_mode(FOCUS_NONE);
  1213. emit_signal(SNAME("color_changed"), color);
  1214. }
  1215. }
  1216. if (p_event->is_action_pressed(SNAME("ui_accept"), false, true)) {
  1217. set_pick_color(old_color);
  1218. emit_signal(SNAME("color_changed"), color);
  1219. }
  1220. }
  1221. void ColorPicker::_sample_draw() {
  1222. // Covers the right half of the sample if the old color is being displayed,
  1223. // or the whole sample if it's not being displayed.
  1224. Rect2 rect_new;
  1225. Rect2 rect_old;
  1226. if (display_old_color) {
  1227. rect_new = Rect2(Point2(sample->get_size().width * 0.5, 0), Size2(sample->get_size().width * 0.5, sample->get_size().height * 0.95));
  1228. // Draw both old and new colors for easier comparison (only if spawned from a ColorPickerButton).
  1229. rect_old = Rect2(Point2(), Size2(sample->get_size().width * 0.5, sample->get_size().height * 0.95));
  1230. if (old_color.a < 1.0) {
  1231. sample->draw_texture_rect(theme_cache.sample_bg, rect_old, true);
  1232. }
  1233. sample->draw_rect(rect_old, old_color);
  1234. if (!old_color.is_equal_approx(color)) {
  1235. // Draw a revert indicator to indicate that the old sample can be clicked to revert to this old color.
  1236. // Adapt icon color to the background color (taking alpha checkerboard into account) so that it's always visible.
  1237. sample->draw_texture(theme_cache.sample_revert,
  1238. rect_old.size * 0.5 - theme_cache.sample_revert->get_size() * 0.5,
  1239. Math::lerp(0.75f, old_color.get_luminance(), old_color.a) < 0.455 ? Color(1, 1, 1) : (Color(0.01, 0.01, 0.01)));
  1240. sample->set_focus_mode(FOCUS_ALL);
  1241. } else {
  1242. sample->set_focus_mode(FOCUS_NONE);
  1243. }
  1244. if (is_color_overbright(color)) {
  1245. // Draw an indicator to denote that the old color is "overbright" and can't be displayed accurately in the preview.
  1246. sample->draw_texture(theme_cache.overbright_indicator, Point2());
  1247. }
  1248. } else {
  1249. rect_new = Rect2(Point2(), Size2(sample->get_size().width, sample->get_size().height * 0.95));
  1250. }
  1251. if (color.a < 1.0) {
  1252. sample->draw_texture_rect(theme_cache.sample_bg, rect_new, true);
  1253. }
  1254. sample->draw_rect(rect_new, color);
  1255. if (display_old_color && !old_color.is_equal_approx(color) && sample->has_focus()) {
  1256. RID ci = sample->get_canvas_item();
  1257. theme_cache.sample_focus->draw(ci, rect_old);
  1258. }
  1259. if (is_color_overbright(color)) {
  1260. // Draw an indicator to denote that the new color is "overbright" and can't be displayed accurately in the preview.
  1261. sample->draw_texture(theme_cache.overbright_indicator, Point2(sample->get_size().width * 0.5, 0));
  1262. }
  1263. }
  1264. void ColorPicker::_slider_draw(int p_which) {
  1265. if (colorize_sliders) {
  1266. modes[current_mode]->slider_draw(p_which);
  1267. }
  1268. }
  1269. void ColorPicker::_alpha_slider_draw() {
  1270. if (!colorize_sliders) {
  1271. return;
  1272. }
  1273. Vector<Vector2> pos;
  1274. pos.resize(4);
  1275. Vector<Color> col;
  1276. col.resize(4);
  1277. Size2 size = alpha_slider->get_size();
  1278. Color left_color;
  1279. Color right_color;
  1280. const real_t margin = 16 * theme_cache.base_scale;
  1281. alpha_slider->draw_texture_rect(theme_cache.sample_bg, Rect2(Point2(0, 0), Size2(size.x, margin)), true);
  1282. left_color = color_normalized;
  1283. left_color.a = 0;
  1284. right_color = color_normalized;
  1285. right_color.a = 1;
  1286. col.set(0, left_color);
  1287. col.set(1, right_color);
  1288. col.set(2, right_color);
  1289. col.set(3, left_color);
  1290. pos.set(0, Vector2(0, 0));
  1291. pos.set(1, Vector2(size.x, 0));
  1292. pos.set(2, Vector2(size.x, margin));
  1293. pos.set(3, Vector2(0, margin));
  1294. alpha_slider->draw_polygon(pos, col);
  1295. }
  1296. void ColorPicker::_slider_or_spin_input(const Ref<InputEvent> &p_event) {
  1297. if (line_edit_mouse_release) {
  1298. line_edit_mouse_release = false;
  1299. return;
  1300. }
  1301. Ref<InputEventMouseButton> bev = p_event;
  1302. if (bev.is_valid() && !bev->is_pressed() && bev->get_button_index() == MouseButton::LEFT) {
  1303. add_recent_preset(color);
  1304. }
  1305. }
  1306. void ColorPicker::_line_edit_input(const Ref<InputEvent> &p_event) {
  1307. Ref<InputEventMouseButton> bev = p_event;
  1308. if (bev.is_valid() && !bev->is_pressed() && bev->get_button_index() == MouseButton::LEFT) {
  1309. line_edit_mouse_release = true;
  1310. }
  1311. }
  1312. void ColorPicker::_preset_input(const Ref<InputEvent> &p_event, const Color &p_color) {
  1313. Ref<InputEventMouseButton> bev = p_event;
  1314. if (bev.is_valid()) {
  1315. if (bev->is_pressed() && bev->get_button_index() == MouseButton::LEFT) {
  1316. set_pick_color(p_color);
  1317. add_recent_preset(color);
  1318. emit_signal(SNAME("color_changed"), p_color);
  1319. } else if (bev->is_pressed() && bev->get_button_index() == MouseButton::RIGHT && can_add_swatches) {
  1320. erase_preset(p_color);
  1321. emit_signal(SNAME("preset_removed"), p_color);
  1322. }
  1323. }
  1324. if (p_event->is_action_pressed(SNAME("ui_accept"), false, true)) {
  1325. set_pick_color(p_color);
  1326. add_recent_preset(color);
  1327. emit_signal(SNAME("color_changed"), p_color);
  1328. } else if (p_event->is_action_pressed(SNAME("ui_colorpicker_delete_preset"), false, true) && can_add_swatches) {
  1329. erase_preset(p_color);
  1330. emit_signal(SNAME("preset_removed"), p_color);
  1331. }
  1332. }
  1333. void ColorPicker::_recent_preset_pressed(const bool p_pressed, ColorPresetButton *p_preset) {
  1334. if (!p_pressed) {
  1335. return;
  1336. }
  1337. // Avoid applying and recalculating the intensity for non-overbright color if it doesn't change.
  1338. if (color != p_preset->get_preset_color()) {
  1339. set_pick_color(p_preset->get_preset_color());
  1340. }
  1341. recent_presets.move_to_back(recent_presets.find(p_preset->get_preset_color()));
  1342. List<Color>::Element *e = recent_preset_cache.find(p_preset->get_preset_color());
  1343. if (e) {
  1344. recent_preset_cache.move_to_back(e);
  1345. }
  1346. recent_preset_hbc->move_child(p_preset, 0);
  1347. emit_signal(SNAME("color_changed"), p_preset->get_preset_color());
  1348. }
  1349. void ColorPicker::_text_changed(const String &) {
  1350. text_changed = true;
  1351. }
  1352. void ColorPicker::_add_preset_pressed() {
  1353. add_preset(color);
  1354. emit_signal(SNAME("preset_added"), color);
  1355. }
  1356. void ColorPicker::_pick_button_pressed_native() {
  1357. if (!DisplayServer::get_singleton()->color_picker(callable_mp(this, &ColorPicker::_native_cb))) {
  1358. // Fallback to default/legacy picker.
  1359. if (DisplayServer::get_singleton()->has_feature(DisplayServer::FEATURE_SCREEN_CAPTURE) && !get_tree()->get_root()->is_embedding_subwindows()) {
  1360. _pick_button_pressed();
  1361. } else {
  1362. _pick_button_pressed_legacy();
  1363. }
  1364. }
  1365. }
  1366. void ColorPicker::_native_cb(bool p_status, const Color &p_color) {
  1367. if (p_status) {
  1368. set_pick_color(p_color);
  1369. if (!deferred_mode_enabled) {
  1370. emit_signal(SNAME("color_changed"), color);
  1371. }
  1372. }
  1373. }
  1374. void ColorPicker::_pick_button_pressed() {
  1375. is_picking_color = true;
  1376. pre_picking_color = color;
  1377. if (!picker_window) {
  1378. picker_window = memnew(Popup);
  1379. bool has_feature_exclude_from_capture = DisplayServer::get_singleton()->has_feature(DisplayServer::FEATURE_SCREEN_EXCLUDE_FROM_CAPTURE);
  1380. if (!has_feature_exclude_from_capture) {
  1381. picker_window->set_size(Vector2i(28, 28));
  1382. } else {
  1383. picker_window->set_size(Vector2i(55, 72));
  1384. picker_window->set_flag(Window::FLAG_EXCLUDE_FROM_CAPTURE, true); // Only supported on MacOS and Windows.
  1385. }
  1386. picker_window->connect(SceneStringName(visibility_changed), callable_mp(this, &ColorPicker::_pick_finished));
  1387. picker_window->connect(SceneStringName(window_input), callable_mp(this, &ColorPicker::_target_gui_input));
  1388. picker_preview = memnew(Panel);
  1389. picker_preview->set_mouse_filter(MOUSE_FILTER_IGNORE);
  1390. picker_preview->set_size(Vector2i(55, 72));
  1391. picker_window->add_child(picker_preview);
  1392. picker_preview_color = memnew(Panel);
  1393. picker_preview_color->set_mouse_filter(MOUSE_FILTER_IGNORE);
  1394. if (!has_feature_exclude_from_capture) {
  1395. picker_preview_color->set_size(Vector2i(24, 24));
  1396. picker_preview_color->set_position(Vector2i(2, 2));
  1397. } else {
  1398. picker_preview_color->set_size(Vector2i(51, 15));
  1399. picker_preview_color->set_position(Vector2i(2, 55));
  1400. }
  1401. picker_preview->add_child(picker_preview_color);
  1402. if (has_feature_exclude_from_capture) {
  1403. picker_texture_zoom = memnew(TextureRect);
  1404. picker_texture_zoom->set_mouse_filter(MOUSE_FILTER_IGNORE);
  1405. picker_texture_zoom->set_custom_minimum_size(Vector2i(51, 51));
  1406. picker_texture_zoom->set_position(Vector2i(2, 2));
  1407. picker_texture_zoom->set_texture_filter(CanvasItem::TEXTURE_FILTER_NEAREST);
  1408. picker_preview->add_child(picker_texture_zoom);
  1409. }
  1410. picker_preview_style_box.instantiate();
  1411. picker_preview->add_theme_style_override(SceneStringName(panel), picker_preview_style_box);
  1412. picker_preview_style_box_color.instantiate();
  1413. picker_preview_color->add_theme_style_override(SceneStringName(panel), picker_preview_style_box_color);
  1414. add_child(picker_window, false, INTERNAL_MODE_FRONT);
  1415. }
  1416. set_process_internal(true);
  1417. picker_window->popup();
  1418. }
  1419. void ColorPicker::_target_gui_input(const Ref<InputEvent> &p_event) {
  1420. const Ref<InputEventMouseButton> mouse_event = p_event;
  1421. if (mouse_event.is_null()) {
  1422. return;
  1423. }
  1424. if (mouse_event->get_button_index() == MouseButton::LEFT) {
  1425. if (mouse_event->is_pressed()) {
  1426. picker_window->hide();
  1427. _pick_finished();
  1428. }
  1429. } else if (mouse_event->get_button_index() == MouseButton::RIGHT) {
  1430. set_pick_color(pre_picking_color); // Cancel.
  1431. is_picking_color = false;
  1432. set_process_internal(false);
  1433. picker_window->hide();
  1434. } else {
  1435. Window *w = picker_window->get_parent_visible_window();
  1436. while (w) {
  1437. Point2i win_mpos = w->get_mouse_position(); // Mouse position local to the window.
  1438. Size2i win_size = w->get_size();
  1439. if (win_mpos.x >= 0 && win_mpos.y >= 0 && win_mpos.x <= win_size.x && win_mpos.y <= win_size.y) {
  1440. // Mouse event inside window bounds, forward this event to the window.
  1441. Ref<InputEventMouseButton> new_ev = p_event->duplicate();
  1442. new_ev->set_position(win_mpos);
  1443. new_ev->set_global_position(win_mpos);
  1444. w->push_input(new_ev, true);
  1445. return;
  1446. }
  1447. w = w->get_parent_visible_window();
  1448. }
  1449. }
  1450. }
  1451. void ColorPicker::_pick_finished() {
  1452. if (picker_window->is_visible()) {
  1453. return;
  1454. }
  1455. if (Input::get_singleton()->is_action_just_pressed(SNAME("ui_cancel"))) {
  1456. set_pick_color(pre_picking_color);
  1457. } else {
  1458. emit_signal(SNAME("color_changed"), color);
  1459. }
  1460. is_picking_color = false;
  1461. set_process_internal(false);
  1462. picker_window->hide();
  1463. }
  1464. void ColorPicker::_update_menu_items() {
  1465. options_menu->clear();
  1466. options_menu->reset_size();
  1467. options_menu->add_icon_item(get_theme_icon(SNAME("save"), SNAME("FileDialog")), ETR("Save"), static_cast<int>(MenuOption::MENU_SAVE));
  1468. options_menu->set_item_tooltip(-1, ETR("Save the current color palette to reuse later."));
  1469. options_menu->set_item_disabled(-1, presets.is_empty());
  1470. options_menu->add_icon_item(get_theme_icon(SNAME("save"), SNAME("FileDialog")), ETR("Save As"), static_cast<int>(MenuOption::MENU_SAVE_AS));
  1471. options_menu->set_item_tooltip(-1, ETR("Save the current color palette as a new to reuse later."));
  1472. options_menu->set_item_disabled(-1, palette_path.is_empty());
  1473. options_menu->add_icon_item(get_theme_icon(SNAME("load"), SNAME("FileDialog")), ETR("Load"), static_cast<int>(MenuOption::MENU_LOAD));
  1474. options_menu->set_item_tooltip(-1, ETR("Load existing color palette."));
  1475. #ifdef TOOLS_ENABLED
  1476. if (Engine::get_singleton()->is_editor_hint()) {
  1477. options_menu->add_icon_item(get_theme_icon(SNAME("load"), SNAME("FileDialog")), TTRC("Quick Load"), static_cast<int>(MenuOption::MENU_QUICKLOAD));
  1478. options_menu->set_item_tooltip(-1, TTRC("Load existing color palette."));
  1479. }
  1480. #endif // TOOLS_ENABLED
  1481. options_menu->add_icon_item(get_theme_icon(SNAME("clear"), SNAME("FileDialog")), ETR("Clear"), static_cast<int>(MenuOption::MENU_CLEAR));
  1482. options_menu->set_item_tooltip(-1, ETR("Clear the currently loaded color palettes in the picker."));
  1483. options_menu->set_item_disabled(-1, presets.is_empty());
  1484. }
  1485. void ColorPicker::_options_menu_cbk(int p_which) {
  1486. _ensure_file_dialog();
  1487. MenuOption option = static_cast<MenuOption>(p_which);
  1488. switch (option) {
  1489. case MenuOption::MENU_SAVE:
  1490. _save_palette(false);
  1491. break;
  1492. case MenuOption::MENU_SAVE_AS:
  1493. _save_palette(true);
  1494. break;
  1495. case MenuOption::MENU_LOAD:
  1496. _load_palette();
  1497. break;
  1498. #ifdef TOOLS_ENABLED
  1499. case MenuOption::MENU_QUICKLOAD:
  1500. if (quick_open_callback.is_valid()) {
  1501. file_dialog->set_file_mode(FileDialog::FILE_MODE_OPEN_FILE);
  1502. quick_open_callback.call_deferred();
  1503. }
  1504. break;
  1505. #endif // TOOLS_ENABLED
  1506. case MenuOption::MENU_CLEAR: {
  1507. PackedColorArray colors = get_presets();
  1508. for (Color c : colors) {
  1509. erase_preset(c);
  1510. }
  1511. palette_name->set_text("");
  1512. palette_name->set_tooltip_text("");
  1513. palette_path = String();
  1514. btn_preset->set_pressed(false);
  1515. #ifdef TOOLS_ENABLED
  1516. if (editor_settings) {
  1517. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "palette_name", palette_name->get_text());
  1518. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "palette_path", palette_path);
  1519. editor_settings->call(SNAME("set_project_metadata"), "color_picker", "palette_edited", false);
  1520. }
  1521. #endif // TOOLS_ENABLED
  1522. }
  1523. break;
  1524. default:
  1525. break;
  1526. }
  1527. }
  1528. void ColorPicker::_block_input_on_popup_show() {
  1529. if (!get_tree()->get_root()->is_embedding_subwindows()) {
  1530. get_viewport()->set_disable_input(true);
  1531. }
  1532. }
  1533. void ColorPicker::_enable_input_on_popup_hide() {
  1534. if (!get_tree()->get_root()->is_embedding_subwindows()) {
  1535. get_viewport()->set_disable_input(false);
  1536. }
  1537. }
  1538. void ColorPicker::_pick_button_pressed_legacy() {
  1539. if (!is_inside_tree()) {
  1540. return;
  1541. }
  1542. pre_picking_color = color;
  1543. if (!picker_window) {
  1544. picker_window = memnew(Popup);
  1545. picker_window->hide();
  1546. picker_window->set_transient(true);
  1547. add_child(picker_window, false, INTERNAL_MODE_FRONT);
  1548. picker_texture_rect = memnew(TextureRect);
  1549. picker_texture_rect->set_anchors_preset(Control::PRESET_FULL_RECT);
  1550. picker_texture_rect->set_expand_mode(TextureRect::EXPAND_IGNORE_SIZE);
  1551. picker_texture_rect->set_default_cursor_shape(Control::CURSOR_CROSS);
  1552. picker_window->add_child(picker_texture_rect);
  1553. picker_texture_rect->connect(SceneStringName(gui_input), callable_mp(this, &ColorPicker::_picker_texture_input));
  1554. picker_preview = memnew(Panel);
  1555. picker_preview->set_mouse_filter(MOUSE_FILTER_IGNORE);
  1556. picker_preview->set_size(Vector2i(55, 72));
  1557. picker_window->add_child(picker_preview);
  1558. picker_preview_color = memnew(Panel);
  1559. picker_preview_color->set_mouse_filter(MOUSE_FILTER_IGNORE);
  1560. picker_preview_color->set_size(Vector2i(51, 15));
  1561. picker_preview_color->set_position(Vector2i(2, 55));
  1562. picker_preview->add_child(picker_preview_color);
  1563. picker_texture_zoom = memnew(TextureRect);
  1564. picker_texture_zoom->set_mouse_filter(MOUSE_FILTER_IGNORE);
  1565. picker_texture_zoom->set_custom_minimum_size(Vector2i(51, 51));
  1566. picker_texture_zoom->set_position(Vector2i(2, 2));
  1567. picker_texture_zoom->set_texture_filter(CanvasItem::TEXTURE_FILTER_NEAREST);
  1568. picker_preview->add_child(picker_texture_zoom);
  1569. picker_preview_style_box.instantiate();
  1570. picker_preview->add_theme_style_override(SceneStringName(panel), picker_preview_style_box);
  1571. picker_preview_style_box_color.instantiate();
  1572. picker_preview_color->add_theme_style_override(SceneStringName(panel), picker_preview_style_box_color);
  1573. }
  1574. Rect2i screen_rect;
  1575. if (picker_window->is_embedded()) {
  1576. Ref<ImageTexture> tx = ImageTexture::create_from_image(picker_window->get_embedder()->get_texture()->get_image());
  1577. screen_rect = picker_window->get_embedder()->get_visible_rect();
  1578. picker_window->set_position(Point2i());
  1579. picker_texture_rect->set_texture(tx);
  1580. Vector2 ofs = picker_window->get_mouse_position();
  1581. picker_preview->set_position(ofs - Vector2(28, 28));
  1582. Vector2 scale = screen_rect.size / tx->get_image()->get_size();
  1583. ofs /= scale;
  1584. Ref<AtlasTexture> atlas;
  1585. atlas.instantiate();
  1586. atlas->set_atlas(tx);
  1587. atlas->set_region(Rect2i(ofs.x - 8, ofs.y - 8, 17, 17));
  1588. picker_texture_zoom->set_texture(atlas);
  1589. } else {
  1590. screen_rect = picker_window->get_parent_rect();
  1591. picker_window->set_position(screen_rect.position);
  1592. Ref<Image> target_image = Image::create_empty(screen_rect.size.x, screen_rect.size.y, false, Image::FORMAT_RGB8);
  1593. DisplayServer *ds = DisplayServer::get_singleton();
  1594. // Add the Texture of each Window to the Image.
  1595. Vector<DisplayServer::WindowID> wl = ds->get_window_list();
  1596. // FIXME: sort windows by visibility.
  1597. for (const DisplayServer::WindowID &window_id : wl) {
  1598. Window *w = Window::get_from_id(window_id);
  1599. if (!w) {
  1600. continue;
  1601. }
  1602. Ref<Image> img = w->get_texture()->get_image();
  1603. if (img.is_null() || img->is_empty()) {
  1604. continue;
  1605. }
  1606. img->convert(Image::FORMAT_RGB8);
  1607. target_image->blit_rect(img, Rect2i(Point2i(0, 0), img->get_size()), w->get_position());
  1608. }
  1609. Ref<ImageTexture> tx = ImageTexture::create_from_image(target_image);
  1610. picker_texture_rect->set_texture(tx);
  1611. Vector2 ofs = screen_rect.position - DisplayServer::get_singleton()->mouse_get_position();
  1612. picker_preview->set_position(ofs - Vector2(28, 28));
  1613. Ref<AtlasTexture> atlas;
  1614. atlas.instantiate();
  1615. atlas->set_atlas(tx);
  1616. atlas->set_region(Rect2i(ofs.x - 8, ofs.y - 8, 17, 17));
  1617. picker_texture_zoom->set_texture(atlas);
  1618. }
  1619. picker_window->set_size(screen_rect.size);
  1620. picker_window->popup();
  1621. }
  1622. void ColorPicker::_picker_texture_input(const Ref<InputEvent> &p_event) {
  1623. if (!is_inside_tree()) {
  1624. return;
  1625. }
  1626. Ref<InputEventMouseButton> bev = p_event;
  1627. if (bev.is_valid() && bev->get_button_index() == MouseButton::LEFT && !bev->is_pressed()) {
  1628. set_pick_color(picker_color);
  1629. emit_signal(SNAME("color_changed"), color);
  1630. picker_window->hide();
  1631. }
  1632. Ref<InputEventMouseMotion> mev = p_event;
  1633. if (mev.is_valid()) {
  1634. Ref<Image> img = picker_texture_rect->get_texture()->get_image();
  1635. if (img.is_valid() && !img->is_empty()) {
  1636. Vector2 ofs = mev->get_position();
  1637. picker_preview->set_position(ofs - Vector2(28, 28));
  1638. Vector2 scale = picker_texture_rect->get_size() / img->get_size();
  1639. ofs /= scale;
  1640. picker_color = img->get_pixel(ofs.x, ofs.y);
  1641. picker_preview_style_box_color->set_bg_color(picker_color);
  1642. picker_preview_style_box->set_bg_color(picker_color.get_luminance() < 0.5 ? Color(1.0f, 1.0f, 1.0f) : Color(0.0f, 0.0f, 0.0f));
  1643. Ref<AtlasTexture> atlas = picker_texture_zoom->get_texture();
  1644. if (atlas.is_valid()) {
  1645. atlas->set_region(Rect2i(ofs.x - 8, ofs.y - 8, 17, 17));
  1646. }
  1647. }
  1648. }
  1649. }
  1650. void ColorPicker::_html_focus_exit() {
  1651. if (c_text->is_menu_visible()) {
  1652. return;
  1653. }
  1654. if (is_visible_in_tree()) {
  1655. _html_submitted(c_text->get_text());
  1656. } else {
  1657. _update_text_value();
  1658. }
  1659. }
  1660. void ColorPicker::set_can_add_swatches(bool p_enabled) {
  1661. if (can_add_swatches == p_enabled) {
  1662. return;
  1663. }
  1664. can_add_swatches = p_enabled;
  1665. if (!p_enabled) {
  1666. btn_add_preset->set_disabled(true);
  1667. btn_add_preset->set_focus_mode(FOCUS_NONE);
  1668. } else {
  1669. btn_add_preset->set_disabled(false);
  1670. btn_add_preset->set_focus_mode(FOCUS_ALL);
  1671. }
  1672. }
  1673. bool ColorPicker::are_swatches_enabled() const {
  1674. return can_add_swatches;
  1675. }
  1676. void ColorPicker::set_presets_visible(bool p_visible) {
  1677. if (presets_visible == p_visible) {
  1678. return;
  1679. }
  1680. presets_visible = p_visible;
  1681. swatches_vbc->set_visible(p_visible);
  1682. }
  1683. bool ColorPicker::are_presets_visible() const {
  1684. return presets_visible;
  1685. }
  1686. void ColorPicker::set_modes_visible(bool p_visible) {
  1687. if (color_modes_visible == p_visible) {
  1688. return;
  1689. }
  1690. color_modes_visible = p_visible;
  1691. mode_hbc->set_visible(p_visible);
  1692. }
  1693. bool ColorPicker::are_modes_visible() const {
  1694. return color_modes_visible;
  1695. }
  1696. void ColorPicker::set_sampler_visible(bool p_visible) {
  1697. if (sampler_visible == p_visible) {
  1698. return;
  1699. }
  1700. sampler_visible = p_visible;
  1701. sample_hbc->set_visible(p_visible);
  1702. }
  1703. bool ColorPicker::is_sampler_visible() const {
  1704. return sampler_visible;
  1705. }
  1706. void ColorPicker::set_sliders_visible(bool p_visible) {
  1707. if (sliders_visible == p_visible) {
  1708. return;
  1709. }
  1710. sliders_visible = p_visible;
  1711. slider_gc->set_visible(p_visible);
  1712. }
  1713. bool ColorPicker::are_sliders_visible() const {
  1714. return sliders_visible;
  1715. }
  1716. void ColorPicker::set_hex_visible(bool p_visible) {
  1717. if (hex_visible == p_visible) {
  1718. return;
  1719. }
  1720. hex_visible = p_visible;
  1721. hex_hbc->set_visible(p_visible);
  1722. }
  1723. bool ColorPicker::is_hex_visible() const {
  1724. return hex_visible;
  1725. }
  1726. void ColorPicker::_bind_methods() {
  1727. ClassDB::bind_method(D_METHOD("set_pick_color", "color"), &ColorPicker::set_pick_color);
  1728. ClassDB::bind_method(D_METHOD("get_pick_color"), &ColorPicker::get_pick_color);
  1729. ClassDB::bind_method(D_METHOD("set_deferred_mode", "mode"), &ColorPicker::set_deferred_mode);
  1730. ClassDB::bind_method(D_METHOD("is_deferred_mode"), &ColorPicker::is_deferred_mode);
  1731. ClassDB::bind_method(D_METHOD("set_color_mode", "color_mode"), &ColorPicker::set_color_mode);
  1732. ClassDB::bind_method(D_METHOD("get_color_mode"), &ColorPicker::get_color_mode);
  1733. ClassDB::bind_method(D_METHOD("set_edit_alpha", "show"), &ColorPicker::set_edit_alpha);
  1734. ClassDB::bind_method(D_METHOD("is_editing_alpha"), &ColorPicker::is_editing_alpha);
  1735. ClassDB::bind_method(D_METHOD("set_edit_intensity", "show"), &ColorPicker::set_edit_intensity);
  1736. ClassDB::bind_method(D_METHOD("is_editing_intensity"), &ColorPicker::is_editing_intensity);
  1737. ClassDB::bind_method(D_METHOD("set_can_add_swatches", "enabled"), &ColorPicker::set_can_add_swatches);
  1738. ClassDB::bind_method(D_METHOD("are_swatches_enabled"), &ColorPicker::are_swatches_enabled);
  1739. ClassDB::bind_method(D_METHOD("set_presets_visible", "visible"), &ColorPicker::set_presets_visible);
  1740. ClassDB::bind_method(D_METHOD("are_presets_visible"), &ColorPicker::are_presets_visible);
  1741. ClassDB::bind_method(D_METHOD("set_modes_visible", "visible"), &ColorPicker::set_modes_visible);
  1742. ClassDB::bind_method(D_METHOD("are_modes_visible"), &ColorPicker::are_modes_visible);
  1743. ClassDB::bind_method(D_METHOD("set_sampler_visible", "visible"), &ColorPicker::set_sampler_visible);
  1744. ClassDB::bind_method(D_METHOD("is_sampler_visible"), &ColorPicker::is_sampler_visible);
  1745. ClassDB::bind_method(D_METHOD("set_sliders_visible", "visible"), &ColorPicker::set_sliders_visible);
  1746. ClassDB::bind_method(D_METHOD("are_sliders_visible"), &ColorPicker::are_sliders_visible);
  1747. ClassDB::bind_method(D_METHOD("set_hex_visible", "visible"), &ColorPicker::set_hex_visible);
  1748. ClassDB::bind_method(D_METHOD("is_hex_visible"), &ColorPicker::is_hex_visible);
  1749. ClassDB::bind_method(D_METHOD("add_preset", "color"), &ColorPicker::add_preset);
  1750. ClassDB::bind_method(D_METHOD("erase_preset", "color"), &ColorPicker::erase_preset);
  1751. ClassDB::bind_method(D_METHOD("get_presets"), &ColorPicker::get_presets);
  1752. ClassDB::bind_method(D_METHOD("add_recent_preset", "color"), &ColorPicker::add_recent_preset);
  1753. ClassDB::bind_method(D_METHOD("erase_recent_preset", "color"), &ColorPicker::erase_recent_preset);
  1754. ClassDB::bind_method(D_METHOD("get_recent_presets"), &ColorPicker::get_recent_presets);
  1755. ClassDB::bind_method(D_METHOD("set_picker_shape", "shape"), &ColorPicker::set_picker_shape);
  1756. ClassDB::bind_method(D_METHOD("get_picker_shape"), &ColorPicker::get_picker_shape);
  1757. ADD_PROPERTY(PropertyInfo(Variant::COLOR, "color"), "set_pick_color", "get_pick_color");
  1758. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "edit_alpha"), "set_edit_alpha", "is_editing_alpha");
  1759. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "edit_intensity"), "set_edit_intensity", "is_editing_intensity");
  1760. ADD_PROPERTY(PropertyInfo(Variant::INT, "color_mode", PROPERTY_HINT_ENUM, "RGB,HSV,LINEAR,OKHSL"), "set_color_mode", "get_color_mode");
  1761. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "deferred_mode"), "set_deferred_mode", "is_deferred_mode");
  1762. ADD_PROPERTY(PropertyInfo(Variant::INT, "picker_shape", PROPERTY_HINT_ENUM, "HSV Rectangle,HSV Rectangle Wheel,VHS Circle,OKHSL Circle,OK HS Rectangle:5,OK HL Rectangle,None:4"), "set_picker_shape", "get_picker_shape");
  1763. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "can_add_swatches"), "set_can_add_swatches", "are_swatches_enabled");
  1764. ADD_GROUP("Customization", "");
  1765. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "sampler_visible"), "set_sampler_visible", "is_sampler_visible");
  1766. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "color_modes_visible"), "set_modes_visible", "are_modes_visible");
  1767. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "sliders_visible"), "set_sliders_visible", "are_sliders_visible");
  1768. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "hex_visible"), "set_hex_visible", "is_hex_visible");
  1769. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "presets_visible"), "set_presets_visible", "are_presets_visible");
  1770. ADD_SIGNAL(MethodInfo("color_changed", PropertyInfo(Variant::COLOR, "color")));
  1771. ADD_SIGNAL(MethodInfo("preset_added", PropertyInfo(Variant::COLOR, "color")));
  1772. ADD_SIGNAL(MethodInfo("preset_removed", PropertyInfo(Variant::COLOR, "color")));
  1773. BIND_ENUM_CONSTANT(MODE_RGB);
  1774. BIND_ENUM_CONSTANT(MODE_HSV);
  1775. #ifndef DISABLE_DEPRECATED
  1776. BIND_ENUM_CONSTANT(MODE_RAW);
  1777. #endif
  1778. BIND_ENUM_CONSTANT(MODE_LINEAR);
  1779. BIND_ENUM_CONSTANT(MODE_OKHSL);
  1780. BIND_ENUM_CONSTANT(SHAPE_HSV_RECTANGLE);
  1781. BIND_ENUM_CONSTANT(SHAPE_HSV_WHEEL);
  1782. BIND_ENUM_CONSTANT(SHAPE_VHS_CIRCLE);
  1783. BIND_ENUM_CONSTANT(SHAPE_OKHSL_CIRCLE);
  1784. BIND_ENUM_CONSTANT(SHAPE_NONE);
  1785. BIND_ENUM_CONSTANT(SHAPE_OK_HS_RECTANGLE);
  1786. BIND_ENUM_CONSTANT(SHAPE_OK_HL_RECTANGLE);
  1787. BIND_THEME_ITEM_CUSTOM(Theme::DATA_TYPE_CONSTANT, ColorPicker, content_margin, "margin");
  1788. BIND_THEME_ITEM(Theme::DATA_TYPE_CONSTANT, ColorPicker, label_width);
  1789. BIND_THEME_ITEM(Theme::DATA_TYPE_CONSTANT, ColorPicker, sv_width);
  1790. BIND_THEME_ITEM(Theme::DATA_TYPE_CONSTANT, ColorPicker, sv_height);
  1791. BIND_THEME_ITEM(Theme::DATA_TYPE_CONSTANT, ColorPicker, h_width);
  1792. BIND_THEME_ITEM(Theme::DATA_TYPE_CONSTANT, ColorPicker, center_slider_grabbers);
  1793. BIND_THEME_ITEM(Theme::DATA_TYPE_STYLEBOX, ColorPicker, sample_focus);
  1794. BIND_THEME_ITEM(Theme::DATA_TYPE_STYLEBOX, ColorPicker, picker_focus_rectangle);
  1795. BIND_THEME_ITEM(Theme::DATA_TYPE_STYLEBOX, ColorPicker, picker_focus_circle);
  1796. BIND_THEME_ITEM(Theme::DATA_TYPE_COLOR, ColorPicker, focused_not_editing_cursor_color);
  1797. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, menu_option);
  1798. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, screen_picker);
  1799. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, expanded_arrow);
  1800. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, folded_arrow);
  1801. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, add_preset);
  1802. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, shape_rect);
  1803. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, shape_rect_wheel);
  1804. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, shape_circle);
  1805. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, bar_arrow);
  1806. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, sample_bg);
  1807. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, sample_revert);
  1808. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, overbright_indicator);
  1809. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, picker_cursor);
  1810. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, picker_cursor_bg);
  1811. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, color_hue);
  1812. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPicker, color_script);
  1813. BIND_THEME_ITEM_EXT(Theme::DATA_TYPE_STYLEBOX, ColorPicker, mode_button_normal, "tab_unselected", "TabContainer");
  1814. BIND_THEME_ITEM_EXT(Theme::DATA_TYPE_STYLEBOX, ColorPicker, mode_button_pressed, "tab_selected", "TabContainer");
  1815. BIND_THEME_ITEM_EXT(Theme::DATA_TYPE_STYLEBOX, ColorPicker, mode_button_hover, "tab_selected", "TabContainer");
  1816. ADD_CLASS_DEPENDENCY("LineEdit");
  1817. ADD_CLASS_DEPENDENCY("MenuButton");
  1818. ADD_CLASS_DEPENDENCY("PopupMenu");
  1819. }
  1820. ColorPicker::ColorPicker() {
  1821. internal_margin = memnew(MarginContainer);
  1822. add_child(internal_margin, false, INTERNAL_MODE_FRONT);
  1823. VBoxContainer *real_vbox = memnew(VBoxContainer);
  1824. internal_margin->add_child(real_vbox);
  1825. shape_container = memnew(HBoxContainer);
  1826. shape_container->set_v_size_flags(SIZE_SHRINK_BEGIN);
  1827. real_vbox->add_child(shape_container);
  1828. sample_hbc = memnew(HBoxContainer);
  1829. real_vbox->add_child(sample_hbc);
  1830. btn_pick = memnew(Button);
  1831. btn_pick->set_icon_alignment(HORIZONTAL_ALIGNMENT_CENTER);
  1832. btn_pick->set_accessibility_name(ETR("Pick"));
  1833. sample_hbc->add_child(btn_pick);
  1834. sample = memnew(TextureRect);
  1835. sample_hbc->add_child(sample);
  1836. sample->set_h_size_flags(SIZE_EXPAND_FILL);
  1837. sample->connect(SceneStringName(gui_input), callable_mp(this, &ColorPicker::_sample_input));
  1838. sample->connect(SceneStringName(draw), callable_mp(this, &ColorPicker::_sample_draw));
  1839. btn_shape = memnew(MenuButton);
  1840. btn_shape->set_flat(false);
  1841. sample_hbc->add_child(btn_shape);
  1842. btn_shape->set_toggle_mode(true);
  1843. btn_shape->set_tooltip_text(ETR("Select a picker shape."));
  1844. btn_shape->set_icon_alignment(HORIZONTAL_ALIGNMENT_CENTER);
  1845. btn_shape->set_focus_mode(FOCUS_ALL);
  1846. add_shape(memnew(ColorPickerShapeRectangle(this)));
  1847. add_shape(memnew(ColorPickerShapeWheel(this)));
  1848. add_shape(memnew(ColorPickerShapeVHSCircle(this)));
  1849. add_shape(memnew(ColorPickerShapeOKHSLCircle(this)));
  1850. add_shape(memnew(ColorPickerShapeOKHSRectangle(this)));
  1851. add_shape(memnew(ColorPickerShapeOKHLRectangle(this)));
  1852. shape_popup = btn_shape->get_popup();
  1853. {
  1854. int i = 0;
  1855. for (const ColorPickerShape *shape : shapes) {
  1856. shape_popup->add_radio_check_item(shape->get_name(), index_to_shape(i));
  1857. i++;
  1858. }
  1859. }
  1860. shape_popup->set_item_checked(get_current_shape_index(), true);
  1861. shape_popup->connect(SceneStringName(id_pressed), callable_mp(this, &ColorPicker::set_picker_shape));
  1862. shape_popup->connect("about_to_popup", callable_mp(this, &ColorPicker::_block_input_on_popup_show));
  1863. shape_popup->connect(SNAME("popup_hide"), callable_mp(this, &ColorPicker::_enable_input_on_popup_hide));
  1864. add_mode(memnew(ColorModeRGB(this)));
  1865. add_mode(memnew(ColorModeHSV(this)));
  1866. add_mode(memnew(ColorModeLinear(this)));
  1867. add_mode(memnew(ColorModeOKHSL(this)));
  1868. mode_hbc = memnew(HBoxContainer);
  1869. real_vbox->add_child(mode_hbc);
  1870. mode_group.instantiate();
  1871. for (int i = 0; i < MODE_BUTTON_COUNT; i++) {
  1872. mode_btns[i] = memnew(Button);
  1873. mode_hbc->add_child(mode_btns[i]);
  1874. mode_btns[i]->set_focus_mode(FOCUS_ALL);
  1875. mode_btns[i]->set_h_size_flags(SIZE_EXPAND_FILL);
  1876. mode_btns[i]->set_toggle_mode(true);
  1877. mode_btns[i]->set_text(modes[i]->get_name());
  1878. mode_btns[i]->set_button_group(mode_group);
  1879. mode_btns[i]->connect(SceneStringName(pressed), callable_mp(this, &ColorPicker::set_color_mode).bind((ColorModeType)i));
  1880. }
  1881. mode_btns[0]->set_pressed(true);
  1882. btn_mode = memnew(MenuButton);
  1883. btn_mode->set_flat(false);
  1884. mode_hbc->add_child(btn_mode);
  1885. btn_mode->set_toggle_mode(true);
  1886. btn_mode->set_accessibility_name(ETR("Select a picker mode."));
  1887. btn_mode->set_tooltip_text(ETR("Select a picker mode."));
  1888. btn_mode->set_focus_mode(FOCUS_ALL);
  1889. mode_popup = btn_mode->get_popup();
  1890. {
  1891. int i = 0;
  1892. for (const ColorMode *mode : modes) {
  1893. mode_popup->add_radio_check_item(mode->get_name(), i);
  1894. i++;
  1895. }
  1896. }
  1897. mode_popup->add_separator();
  1898. mode_popup->add_check_item(ETR("Colorized Sliders"), MODE_MAX);
  1899. mode_popup->set_item_checked(current_mode, true);
  1900. mode_popup->set_item_checked(MODE_MAX + 1, true);
  1901. mode_popup->connect(SceneStringName(id_pressed), callable_mp(this, &ColorPicker::_set_mode_popup_value));
  1902. mode_popup->connect("about_to_popup", callable_mp(this, &ColorPicker::_block_input_on_popup_show));
  1903. mode_popup->connect(SNAME("popup_hide"), callable_mp(this, &ColorPicker::_enable_input_on_popup_hide));
  1904. slider_gc = memnew(GridContainer);
  1905. real_vbox->add_child(slider_gc);
  1906. slider_gc->set_h_size_flags(SIZE_EXPAND_FILL);
  1907. slider_gc->set_columns(3);
  1908. for (int i = 0; i < SLIDER_MAX; i++) {
  1909. create_slider(slider_gc, i);
  1910. }
  1911. alpha_label->set_text("A");
  1912. intensity_label->set_text("I");
  1913. intensity_slider->set_min(-10);
  1914. intensity_slider->set_max(10);
  1915. intensity_slider->set_step(0.001);
  1916. intensity_value->set_allow_greater(true);
  1917. intensity_value->set_custom_arrow_step(1);
  1918. hex_hbc = memnew(HBoxContainer);
  1919. hex_hbc->set_alignment(ALIGNMENT_BEGIN);
  1920. real_vbox->add_child(hex_hbc);
  1921. hex_label = memnew(Label(ETR("Hex")));
  1922. hex_hbc->add_child(hex_label);
  1923. text_type = memnew(Button);
  1924. hex_hbc->add_child(text_type);
  1925. text_type->set_auto_translate_mode(AUTO_TRANSLATE_MODE_DISABLED);
  1926. text_type->set_tooltip_auto_translate_mode(AUTO_TRANSLATE_MODE_ALWAYS);
  1927. text_type->set_text("#");
  1928. #ifdef TOOLS_ENABLED
  1929. if (Engine::get_singleton()->is_editor_hint()) {
  1930. text_type->set_tooltip_text(TTRC("Switch between hexadecimal and code values."));
  1931. text_type->connect(SceneStringName(pressed), callable_mp(this, &ColorPicker::_text_type_toggled));
  1932. } else {
  1933. text_type->set_accessibility_name(ETR("Hexadecimal Values"));
  1934. #else
  1935. {
  1936. text_type->set_accessibility_name(ETR("Hexadecimal Values"));
  1937. #endif // TOOLS_ENABLED
  1938. text_type->set_flat(true);
  1939. }
  1940. c_text = memnew(LineEdit);
  1941. hex_hbc->add_child(c_text);
  1942. c_text->set_h_size_flags(SIZE_EXPAND_FILL);
  1943. c_text->set_select_all_on_focus(true);
  1944. c_text->set_accessibility_name(ETR("Hex code or named color"));
  1945. c_text->set_tooltip_text(ETR("Enter a hex code (\"#ff0000\") or named color (\"red\")."));
  1946. c_text->set_placeholder(ETR("Hex code or named color"));
  1947. c_text->connect(SceneStringName(text_submitted), callable_mp(this, &ColorPicker::_html_submitted));
  1948. c_text->connect(SceneStringName(text_changed), callable_mp(this, &ColorPicker::_text_changed));
  1949. c_text->connect(SceneStringName(focus_exited), callable_mp(this, &ColorPicker::_html_focus_exit));
  1950. _update_controls();
  1951. updating = false;
  1952. swatches_vbc = memnew(VBoxContainer);
  1953. real_vbox->add_child(swatches_vbc);
  1954. preset_container = memnew(GridContainer);
  1955. preset_container->set_h_size_flags(SIZE_EXPAND_FILL);
  1956. preset_container->set_columns(PRESET_COLUMN_COUNT);
  1957. preset_container->hide();
  1958. preset_group.instantiate();
  1959. HBoxContainer *palette_box = memnew(HBoxContainer);
  1960. palette_box->set_h_size_flags(SIZE_EXPAND_FILL);
  1961. swatches_vbc->add_child(palette_box);
  1962. btn_preset = memnew(Button);
  1963. btn_preset->set_text(ETR("Swatches"));
  1964. btn_preset->set_flat(true);
  1965. btn_preset->set_toggle_mode(true);
  1966. btn_preset->set_focus_mode(FOCUS_ALL);
  1967. btn_preset->set_text_alignment(HORIZONTAL_ALIGNMENT_LEFT);
  1968. btn_preset->set_h_size_flags(SIZE_EXPAND_FILL);
  1969. btn_preset->connect(SceneStringName(toggled), callable_mp(this, &ColorPicker::_show_hide_preset).bind(btn_preset, preset_container));
  1970. palette_box->add_child(btn_preset);
  1971. menu_btn = memnew(MenuButton);
  1972. menu_btn->set_flat(false);
  1973. menu_btn->set_focus_mode(FOCUS_ALL);
  1974. menu_btn->set_tooltip_text(ETR("Show all options available."));
  1975. menu_btn->connect("about_to_popup", callable_mp(this, &ColorPicker::_update_menu_items));
  1976. palette_box->add_child(menu_btn);
  1977. options_menu = menu_btn->get_popup();
  1978. options_menu->connect(SceneStringName(id_pressed), callable_mp(this, &ColorPicker::_options_menu_cbk));
  1979. options_menu->connect("about_to_popup", callable_mp(this, &ColorPicker::_block_input_on_popup_show));
  1980. options_menu->connect(SNAME("popup_hide"), callable_mp(this, &ColorPicker::_enable_input_on_popup_hide));
  1981. palette_name = memnew(Label);
  1982. palette_name->hide();
  1983. palette_name->set_mouse_filter(MOUSE_FILTER_PASS);
  1984. swatches_vbc->add_child(palette_name);
  1985. swatches_vbc->add_child(preset_container);
  1986. recent_preset_hbc = memnew(HBoxContainer);
  1987. recent_preset_hbc->set_v_size_flags(SIZE_SHRINK_BEGIN);
  1988. recent_preset_hbc->hide();
  1989. recent_preset_group.instantiate();
  1990. btn_recent_preset = memnew(Button(ETR("Recent Colors")));
  1991. btn_recent_preset->set_flat(true);
  1992. btn_recent_preset->set_toggle_mode(true);
  1993. btn_recent_preset->set_focus_mode(FOCUS_ALL);
  1994. btn_recent_preset->set_text_alignment(HORIZONTAL_ALIGNMENT_LEFT);
  1995. btn_recent_preset->connect(SceneStringName(toggled), callable_mp(this, &ColorPicker::_show_hide_preset).bind(btn_recent_preset, recent_preset_hbc));
  1996. swatches_vbc->add_child(btn_recent_preset);
  1997. swatches_vbc->add_child(recent_preset_hbc);
  1998. set_pick_color(Color(1, 1, 1));
  1999. btn_add_preset = memnew(Button);
  2000. btn_add_preset->set_icon_alignment(HORIZONTAL_ALIGNMENT_CENTER);
  2001. btn_add_preset->set_tooltip_text(ETR("Add current color as a preset."));
  2002. btn_add_preset->connect(SceneStringName(pressed), callable_mp(this, &ColorPicker::_add_preset_pressed));
  2003. preset_container->add_child(btn_add_preset);
  2004. perm_hb = memnew(HBoxContainer);
  2005. perm_hb->set_alignment(BoxContainer::ALIGNMENT_CENTER);
  2006. LinkButton *perm_link = memnew(LinkButton);
  2007. perm_link->set_text(ETR("Screen Recording permission missing!"));
  2008. perm_link->set_tooltip_text(ETR("Screen Recording permission is required to pick colors from the other application windows.\nClick here to request access..."));
  2009. perm_link->connect(SceneStringName(pressed), callable_mp(this, &ColorPicker::_req_permission));
  2010. perm_hb->add_child(perm_link);
  2011. real_vbox->add_child(perm_hb);
  2012. perm_hb->set_visible(false);
  2013. }
  2014. void ColorPicker::_req_permission() {
  2015. #ifdef MACOS_ENABLED
  2016. OS::get_singleton()->request_permission("macos.permission.RECORD_SCREEN");
  2017. #endif
  2018. }
  2019. ColorPicker::~ColorPicker() {
  2020. for (ColorMode *mode : modes) {
  2021. memdelete(mode);
  2022. }
  2023. for (ColorPickerShape *shape : shapes) {
  2024. memdelete(shape);
  2025. }
  2026. }
  2027. /////////////////
  2028. void ColorPickerPopupPanel::_input_from_window(const Ref<InputEvent> &p_event) {
  2029. if (p_event->is_action_pressed(SNAME("ui_accept"), false, true)) {
  2030. _close_pressed();
  2031. }
  2032. PopupPanel::_input_from_window(p_event);
  2033. }
  2034. /////////////////
  2035. void ColorPickerButton::_about_to_popup() {
  2036. if (!get_tree()->get_root()->is_embedding_subwindows()) {
  2037. get_viewport()->set_disable_input(true);
  2038. }
  2039. set_pressed(true);
  2040. if (picker) {
  2041. picker->set_old_color(color);
  2042. }
  2043. }
  2044. void ColorPickerButton::_color_changed(const Color &p_color) {
  2045. color = p_color;
  2046. queue_accessibility_update();
  2047. queue_redraw();
  2048. emit_signal(SNAME("color_changed"), color);
  2049. }
  2050. void ColorPickerButton::_modal_closed() {
  2051. if (Input::get_singleton()->is_action_just_pressed(SNAME("ui_cancel"))) {
  2052. set_pick_color(picker->get_old_color());
  2053. emit_signal(SNAME("color_changed"), color);
  2054. }
  2055. emit_signal(SNAME("popup_closed"));
  2056. set_pressed(false);
  2057. if (!get_tree()->get_root()->is_embedding_subwindows()) {
  2058. get_viewport()->set_disable_input(false);
  2059. }
  2060. }
  2061. void ColorPickerButton::pressed() {
  2062. _update_picker();
  2063. Size2 minsize = popup->get_contents_minimum_size();
  2064. float viewport_height = get_viewport_rect().size.y;
  2065. popup->reset_size();
  2066. picker->_update_presets();
  2067. picker->_update_recent_presets();
  2068. // Determine in which direction to show the popup. By default popup horizontally centered below the button.
  2069. // But if the popup doesn't fit below and the button is in the bottom half of the viewport, show above.
  2070. bool show_above = false;
  2071. if (get_global_position().y + get_size().y + minsize.y > viewport_height && get_global_position().y * 2 + get_size().y > viewport_height) {
  2072. show_above = true;
  2073. }
  2074. float h_offset = (get_size().x - minsize.x) / 2;
  2075. float v_offset = show_above ? -minsize.y : get_size().y;
  2076. popup->set_position(get_screen_position() + Vector2(h_offset, v_offset));
  2077. popup->popup();
  2078. if (!picker->is_hex_visible() && picker->get_picker_shape() != ColorPicker::SHAPE_NONE) {
  2079. callable_mp(picker, &ColorPicker::set_focus_on_picker_shape).call_deferred();
  2080. } else if (DisplayServer::get_singleton()->has_hardware_keyboard()) {
  2081. picker->set_focus_on_line_edit();
  2082. }
  2083. }
  2084. void ColorPickerButton::_notification(int p_what) {
  2085. switch (p_what) {
  2086. case NOTIFICATION_ACCESSIBILITY_UPDATE: {
  2087. RID ae = get_accessibility_element();
  2088. ERR_FAIL_COND(ae.is_null());
  2089. DisplayServer::get_singleton()->accessibility_update_set_role(ae, DisplayServer::AccessibilityRole::ROLE_BUTTON);
  2090. DisplayServer::get_singleton()->accessibility_update_set_popup_type(ae, DisplayServer::AccessibilityPopupType::POPUP_DIALOG);
  2091. DisplayServer::get_singleton()->accessibility_update_set_color_value(ae, color);
  2092. } break;
  2093. case NOTIFICATION_DRAW: {
  2094. const Rect2 r = Rect2(theme_cache.normal_style->get_offset(), get_size() - theme_cache.normal_style->get_minimum_size());
  2095. draw_texture_rect(theme_cache.background_icon, r, true);
  2096. draw_rect(r, color);
  2097. if (color.r > 1 || color.g > 1 || color.b > 1) {
  2098. // Draw an indicator to denote that the color is "overbright" and can't be displayed accurately in the preview
  2099. draw_texture(theme_cache.overbright_indicator, theme_cache.normal_style->get_offset());
  2100. }
  2101. } break;
  2102. case NOTIFICATION_WM_CLOSE_REQUEST: {
  2103. if (popup) {
  2104. popup->hide();
  2105. }
  2106. } break;
  2107. case NOTIFICATION_VISIBILITY_CHANGED: {
  2108. if (popup && !is_visible_in_tree()) {
  2109. popup->hide();
  2110. }
  2111. } break;
  2112. }
  2113. }
  2114. void ColorPickerButton::set_pick_color(const Color &p_color) {
  2115. if (color == p_color) {
  2116. return;
  2117. }
  2118. color = p_color;
  2119. if (picker) {
  2120. picker->set_pick_color(p_color);
  2121. }
  2122. queue_accessibility_update();
  2123. queue_redraw();
  2124. }
  2125. Color ColorPickerButton::get_pick_color() const {
  2126. return color;
  2127. }
  2128. void ColorPickerButton::set_edit_alpha(bool p_show) {
  2129. if (edit_alpha == p_show) {
  2130. return;
  2131. }
  2132. edit_alpha = p_show;
  2133. if (picker) {
  2134. picker->set_edit_alpha(p_show);
  2135. }
  2136. }
  2137. bool ColorPickerButton::is_editing_alpha() const {
  2138. return edit_alpha;
  2139. }
  2140. void ColorPickerButton::set_edit_intensity(bool p_show) {
  2141. if (edit_intensity == p_show) {
  2142. return;
  2143. }
  2144. edit_intensity = p_show;
  2145. if (picker) {
  2146. picker->set_edit_intensity(p_show);
  2147. }
  2148. }
  2149. bool ColorPickerButton::is_editing_intensity() const {
  2150. return edit_intensity;
  2151. }
  2152. ColorPicker *ColorPickerButton::get_picker() {
  2153. _update_picker();
  2154. return picker;
  2155. }
  2156. PopupPanel *ColorPickerButton::get_popup() {
  2157. _update_picker();
  2158. return popup;
  2159. }
  2160. void ColorPickerButton::_update_picker() {
  2161. if (!picker) {
  2162. popup = memnew(ColorPickerPopupPanel);
  2163. popup->set_wrap_controls(true);
  2164. picker = memnew(ColorPicker);
  2165. picker->set_anchors_and_offsets_preset(PRESET_FULL_RECT);
  2166. popup->add_child(picker);
  2167. add_child(popup, false, INTERNAL_MODE_FRONT);
  2168. picker->connect("color_changed", callable_mp(this, &ColorPickerButton::_color_changed));
  2169. popup->connect("about_to_popup", callable_mp(this, &ColorPickerButton::_about_to_popup));
  2170. popup->connect("popup_hide", callable_mp(this, &ColorPickerButton::_modal_closed));
  2171. picker->connect(SceneStringName(minimum_size_changed), callable_mp((Window *)popup, &Window::reset_size));
  2172. picker->set_pick_color(color);
  2173. picker->set_edit_alpha(edit_alpha);
  2174. picker->set_edit_intensity(edit_intensity);
  2175. picker->set_display_old_color(true);
  2176. emit_signal(SNAME("picker_created"));
  2177. }
  2178. }
  2179. void ColorPickerButton::_bind_methods() {
  2180. ClassDB::bind_method(D_METHOD("set_pick_color", "color"), &ColorPickerButton::set_pick_color);
  2181. ClassDB::bind_method(D_METHOD("get_pick_color"), &ColorPickerButton::get_pick_color);
  2182. ClassDB::bind_method(D_METHOD("get_picker"), &ColorPickerButton::get_picker);
  2183. ClassDB::bind_method(D_METHOD("get_popup"), &ColorPickerButton::get_popup);
  2184. ClassDB::bind_method(D_METHOD("set_edit_alpha", "show"), &ColorPickerButton::set_edit_alpha);
  2185. ClassDB::bind_method(D_METHOD("is_editing_alpha"), &ColorPickerButton::is_editing_alpha);
  2186. ClassDB::bind_method(D_METHOD("set_edit_intensity", "show"), &ColorPickerButton::set_edit_intensity);
  2187. ClassDB::bind_method(D_METHOD("is_editing_intensity"), &ColorPickerButton::is_editing_intensity);
  2188. ClassDB::bind_method(D_METHOD("_about_to_popup"), &ColorPickerButton::_about_to_popup);
  2189. ADD_SIGNAL(MethodInfo("color_changed", PropertyInfo(Variant::COLOR, "color")));
  2190. ADD_SIGNAL(MethodInfo("popup_closed"));
  2191. ADD_SIGNAL(MethodInfo("picker_created"));
  2192. ADD_PROPERTY(PropertyInfo(Variant::COLOR, "color"), "set_pick_color", "get_pick_color");
  2193. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "edit_alpha"), "set_edit_alpha", "is_editing_alpha");
  2194. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "edit_intensity"), "set_edit_intensity", "is_editing_intensity");
  2195. BIND_THEME_ITEM_CUSTOM(Theme::DATA_TYPE_STYLEBOX, ColorPickerButton, normal_style, "normal");
  2196. BIND_THEME_ITEM_CUSTOM(Theme::DATA_TYPE_ICON, ColorPickerButton, background_icon, "bg");
  2197. BIND_THEME_ITEM_EXT(Theme::DATA_TYPE_ICON, ColorPickerButton, overbright_indicator, "overbright_indicator", "ColorPicker");
  2198. }
  2199. ColorPickerButton::ColorPickerButton(const String &p_text) :
  2200. Button(p_text) {
  2201. set_toggle_mode(true);
  2202. }
  2203. /////////////////
  2204. void ColorPresetButton::_notification(int p_what) {
  2205. switch (p_what) {
  2206. case NOTIFICATION_ACCESSIBILITY_UPDATE: {
  2207. RID ae = get_accessibility_element();
  2208. ERR_FAIL_COND(ae.is_null());
  2209. DisplayServer::get_singleton()->accessibility_update_set_role(ae, DisplayServer::AccessibilityRole::ROLE_BUTTON);
  2210. DisplayServer::get_singleton()->accessibility_update_set_color_value(ae, preset_color);
  2211. } break;
  2212. case NOTIFICATION_DRAW: {
  2213. const Rect2 r = Rect2(Point2(0, 0), get_size());
  2214. Ref<StyleBox> sb_raw = theme_cache.foreground_style->duplicate();
  2215. Ref<StyleBoxFlat> sb_flat = sb_raw;
  2216. Ref<StyleBoxTexture> sb_texture = sb_raw;
  2217. if (sb_flat.is_valid()) {
  2218. sb_flat->set_border_width(SIDE_BOTTOM, 2);
  2219. if (get_draw_mode() == DRAW_PRESSED || get_draw_mode() == DRAW_HOVER_PRESSED) {
  2220. sb_flat->set_border_color(Color(1, 1, 1, 1));
  2221. } else {
  2222. sb_flat->set_border_color(Color(0, 0, 0, 1));
  2223. }
  2224. if (preset_color.a < 1) {
  2225. // Draw a background pattern when the color is transparent.
  2226. sb_flat->set_bg_color(Color(1, 1, 1));
  2227. sb_flat->draw(get_canvas_item(), r);
  2228. Rect2 bg_texture_rect = r.grow_side(SIDE_LEFT, -sb_flat->get_margin(SIDE_LEFT));
  2229. bg_texture_rect = bg_texture_rect.grow_side(SIDE_RIGHT, -sb_flat->get_margin(SIDE_RIGHT));
  2230. bg_texture_rect = bg_texture_rect.grow_side(SIDE_TOP, -sb_flat->get_margin(SIDE_TOP));
  2231. bg_texture_rect = bg_texture_rect.grow_side(SIDE_BOTTOM, -sb_flat->get_margin(SIDE_BOTTOM));
  2232. draw_texture_rect(theme_cache.background_icon, bg_texture_rect, true);
  2233. sb_flat->set_bg_color(preset_color);
  2234. }
  2235. sb_flat->set_bg_color(preset_color);
  2236. sb_flat->draw(get_canvas_item(), r);
  2237. } else if (sb_texture.is_valid()) {
  2238. if (preset_color.a < 1) {
  2239. // Draw a background pattern when the color is transparent.
  2240. 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);
  2241. draw_texture_rect(theme_cache.background_icon, r, use_tile_texture);
  2242. }
  2243. sb_texture->set_modulate(preset_color);
  2244. sb_texture->draw(get_canvas_item(), r);
  2245. } else {
  2246. WARN_PRINT("Unsupported StyleBox used for ColorPresetButton. Use StyleBoxFlat or StyleBoxTexture instead.");
  2247. }
  2248. if (has_focus()) {
  2249. RID ci = get_canvas_item();
  2250. theme_cache.focus_style->draw(ci, Rect2(Point2(), get_size()));
  2251. }
  2252. if (is_color_overbright(preset_color)) {
  2253. // Draw an indicator to denote that the color is "overbright" and can't be displayed accurately in the preview
  2254. draw_texture(theme_cache.overbright_indicator, Vector2(0, 0));
  2255. }
  2256. } break;
  2257. }
  2258. }
  2259. void ColorPresetButton::set_preset_color(const Color &p_color) {
  2260. preset_color = p_color;
  2261. queue_accessibility_update();
  2262. }
  2263. Color ColorPresetButton::get_preset_color() const {
  2264. return preset_color;
  2265. }
  2266. String ColorPresetButton::get_tooltip(const Point2 &p_pos) const {
  2267. Color color = get_preset_color();
  2268. if (recent) {
  2269. return vformat(atr(ETR("Color: %s\nLMB: Apply color")), color_to_string(color, color.a < 1));
  2270. }
  2271. return vformat(atr(ETR("Color: %s\nLMB: Apply color\nRMB: Remove preset")), color_to_string(color, color.a < 1));
  2272. }
  2273. void ColorPresetButton::_bind_methods() {
  2274. BIND_THEME_ITEM_CUSTOM(Theme::DATA_TYPE_STYLEBOX, ColorPresetButton, foreground_style, "preset_fg");
  2275. BIND_THEME_ITEM_CUSTOM(Theme::DATA_TYPE_STYLEBOX, ColorPresetButton, focus_style, "preset_focus");
  2276. BIND_THEME_ITEM_CUSTOM(Theme::DATA_TYPE_ICON, ColorPresetButton, background_icon, "preset_bg");
  2277. BIND_THEME_ITEM(Theme::DATA_TYPE_ICON, ColorPresetButton, overbright_indicator);
  2278. }
  2279. ColorPresetButton::ColorPresetButton(Color p_color, int p_size, bool p_recent) {
  2280. preset_color = p_color;
  2281. recent = p_recent;
  2282. set_toggle_mode(true);
  2283. set_custom_minimum_size(Size2(p_size, p_size));
  2284. set_accessibility_name(vformat(atr(ETR("Color: %s")), color_to_string(p_color, p_color.a < 1)));
  2285. set_tooltip_auto_translate_mode(AUTO_TRANSLATE_MODE_DISABLED);
  2286. }
  2287. ColorPresetButton::~ColorPresetButton() {
  2288. }