color_picker.cpp 92 KB

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