editor_resource_picker.cpp 47 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391
  1. /**************************************************************************/
  2. /* editor_resource_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 "editor_resource_picker.h"
  31. #include "editor/audio_stream_preview.h"
  32. #include "editor/editor_help.h"
  33. #include "editor/editor_node.h"
  34. #include "editor/editor_resource_preview.h"
  35. #include "editor/editor_settings.h"
  36. #include "editor/editor_string_names.h"
  37. #include "editor/filesystem_dock.h"
  38. #include "editor/gui/editor_file_dialog.h"
  39. #include "editor/gui/editor_quick_open_dialog.h"
  40. #include "editor/plugins/editor_resource_conversion_plugin.h"
  41. #include "editor/plugins/script_editor_plugin.h"
  42. #include "editor/scene_tree_dock.h"
  43. #include "editor/themes/editor_scale.h"
  44. #include "scene/gui/button.h"
  45. #include "scene/gui/texture_rect.h"
  46. #include "scene/resources/gradient_texture.h"
  47. #include "scene/resources/image_texture.h"
  48. void EditorResourcePicker::_update_resource() {
  49. String resource_path;
  50. if (edited_resource.is_valid() && edited_resource->get_path().is_resource_file()) {
  51. resource_path = edited_resource->get_path() + "\n";
  52. }
  53. String class_name = _get_resource_type(edited_resource);
  54. if (preview_rect) {
  55. preview_rect->set_texture(Ref<Texture2D>());
  56. assign_button->set_custom_minimum_size(assign_button_min_size);
  57. if (edited_resource == Ref<Resource>()) {
  58. assign_button->set_button_icon(Ref<Texture2D>());
  59. assign_button->set_text(TTR("<empty>"));
  60. assign_button->set_tooltip_text("");
  61. } else {
  62. assign_button->set_button_icon(EditorNode::get_singleton()->get_object_icon(edited_resource.operator->(), SNAME("Object")));
  63. if (!edited_resource->get_name().is_empty()) {
  64. assign_button->set_text(edited_resource->get_name());
  65. } else if (edited_resource->get_path().is_resource_file()) {
  66. assign_button->set_text(edited_resource->get_path().get_file());
  67. } else {
  68. assign_button->set_text(class_name);
  69. }
  70. if (edited_resource->get_path().is_resource_file()) {
  71. resource_path = edited_resource->get_path() + "\n";
  72. }
  73. assign_button->set_tooltip_text(resource_path + TTR("Type:") + " " + class_name);
  74. // Preview will override the above, so called at the end.
  75. EditorResourcePreview::get_singleton()->queue_edited_resource_preview(edited_resource, this, "_update_resource_preview", edited_resource->get_instance_id());
  76. }
  77. } else if (edited_resource.is_valid()) {
  78. assign_button->set_tooltip_text(resource_path + TTR("Type:") + " " + edited_resource->get_class());
  79. }
  80. assign_button->set_disabled(!editable && edited_resource.is_null());
  81. }
  82. void EditorResourcePicker::_update_resource_preview(const String &p_path, const Ref<Texture2D> &p_preview, const Ref<Texture2D> &p_small_preview, ObjectID p_obj) {
  83. if (edited_resource.is_null() || edited_resource->get_instance_id() != p_obj) {
  84. return;
  85. }
  86. if (preview_rect) {
  87. Ref<Script> scr = edited_resource;
  88. if (scr.is_valid()) {
  89. assign_button->set_text(scr->get_path().get_file());
  90. return;
  91. }
  92. if (p_preview.is_valid()) {
  93. preview_rect->set_offset(SIDE_LEFT, assign_button->get_button_icon()->get_width() + assign_button->get_theme_stylebox(CoreStringName(normal))->get_content_margin(SIDE_LEFT) + get_theme_constant(SNAME("h_separation"), SNAME("Button")));
  94. // Resource-specific stretching.
  95. if (Ref<GradientTexture1D>(edited_resource).is_valid() || Ref<Gradient>(edited_resource).is_valid()) {
  96. preview_rect->set_stretch_mode(TextureRect::STRETCH_SCALE);
  97. assign_button->set_custom_minimum_size(assign_button_min_size);
  98. } else {
  99. preview_rect->set_stretch_mode(TextureRect::STRETCH_KEEP_ASPECT_CENTERED);
  100. int thumbnail_size = EDITOR_GET("filesystem/file_dialog/thumbnail_size");
  101. thumbnail_size *= EDSCALE;
  102. assign_button->set_custom_minimum_size(assign_button_min_size.max(Size2(1, thumbnail_size)));
  103. }
  104. preview_rect->set_texture(p_preview);
  105. assign_button->set_text("");
  106. }
  107. }
  108. }
  109. void EditorResourcePicker::_resource_selected() {
  110. if (edited_resource.is_null()) {
  111. edit_button->set_pressed(true);
  112. _update_menu();
  113. return;
  114. }
  115. emit_signal(SNAME("resource_selected"), edited_resource, false);
  116. }
  117. void EditorResourcePicker::_resource_changed() {
  118. emit_signal(SNAME("resource_changed"), edited_resource);
  119. _update_resource();
  120. }
  121. void EditorResourcePicker::_file_selected(const String &p_path) {
  122. Ref<Resource> loaded_resource = ResourceLoader::load(p_path);
  123. ERR_FAIL_COND_MSG(loaded_resource.is_null(), "Cannot load resource from path '" + p_path + "'.");
  124. if (!base_type.is_empty()) {
  125. bool any_type_matches = false;
  126. String res_type = loaded_resource->get_class();
  127. Ref<Script> res_script = loaded_resource->get_script();
  128. bool is_global_class = false;
  129. if (res_script.is_valid()) {
  130. String script_type = EditorNode::get_editor_data().script_class_get_name(res_script->get_path());
  131. if (!script_type.is_empty()) {
  132. is_global_class = true;
  133. res_type = script_type;
  134. }
  135. }
  136. for (int i = 0; i < base_type.get_slice_count(","); i++) {
  137. String base = base_type.get_slice(",", i);
  138. any_type_matches = is_global_class ? EditorNode::get_editor_data().script_class_is_parent(res_type, base) : loaded_resource->is_class(base);
  139. if (any_type_matches) {
  140. break;
  141. }
  142. }
  143. if (!any_type_matches) {
  144. EditorNode::get_singleton()->show_warning(vformat(TTR("The selected resource (%s) does not match any type expected for this property (%s)."), res_type, base_type));
  145. return;
  146. }
  147. }
  148. edited_resource = loaded_resource;
  149. _resource_changed();
  150. }
  151. void EditorResourcePicker::_resource_saved(Object *p_resource) {
  152. if (edited_resource.is_valid() && p_resource == edited_resource.ptr()) {
  153. emit_signal(SNAME("resource_changed"), edited_resource);
  154. _update_resource();
  155. }
  156. }
  157. void EditorResourcePicker::_update_menu() {
  158. _update_menu_items();
  159. Rect2 gt = edit_button->get_screen_rect();
  160. edit_menu->reset_size();
  161. int ms = edit_menu->get_contents_minimum_size().width;
  162. Vector2 popup_pos = gt.get_end() - Vector2(ms, 0);
  163. edit_menu->set_position(popup_pos);
  164. edit_menu->popup();
  165. }
  166. void EditorResourcePicker::_update_menu_items() {
  167. _ensure_resource_menu();
  168. edit_menu->clear();
  169. // Add options for creating specific subtypes of the base resource type.
  170. if (is_editable()) {
  171. set_create_options(edit_menu);
  172. // Add an option to load a resource from a file using the QuickOpen dialog.
  173. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Load")), TTR("Quick Load..."), OBJ_MENU_QUICKLOAD);
  174. edit_menu->set_item_tooltip(-1, TTR("Opens a quick menu to select from a list of allowed Resource files."));
  175. // Add an option to load a resource from a file using the regular file dialog.
  176. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Load")), TTR("Load..."), OBJ_MENU_LOAD);
  177. }
  178. // Add options for changing existing value of the resource.
  179. if (edited_resource.is_valid()) {
  180. // Determine if the edited resource is part of another scene (foreign) which was imported
  181. bool is_edited_resource_foreign_import = EditorNode::get_singleton()->is_resource_read_only(edited_resource, true);
  182. // If the resource is determined to be foreign and imported, change the menu entry's description to 'inspect' rather than 'edit'
  183. // since will only be able to view its properties in read-only mode.
  184. if (is_edited_resource_foreign_import) {
  185. // The 'Search' icon is a magnifying glass, which seems appropriate, but maybe a bespoke icon is preferred here.
  186. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Search")), TTR("Inspect"), OBJ_MENU_INSPECT);
  187. } else {
  188. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Edit")), TTR("Edit"), OBJ_MENU_INSPECT);
  189. }
  190. if (is_editable()) {
  191. if (!_is_custom_type_script()) {
  192. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Clear")), TTR("Clear"), OBJ_MENU_CLEAR);
  193. }
  194. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Duplicate")), TTR("Make Unique"), OBJ_MENU_MAKE_UNIQUE);
  195. // Check whether the resource has subresources.
  196. List<PropertyInfo> property_list;
  197. edited_resource->get_property_list(&property_list);
  198. bool has_subresources = false;
  199. for (PropertyInfo &p : property_list) {
  200. if ((p.type == Variant::OBJECT) && (p.hint == PROPERTY_HINT_RESOURCE_TYPE) && (p.name != "script") && ((Object *)edited_resource->get(p.name) != nullptr)) {
  201. has_subresources = true;
  202. break;
  203. }
  204. }
  205. if (has_subresources) {
  206. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Duplicate")), TTR("Make Unique (Recursive)"), OBJ_MENU_MAKE_UNIQUE_RECURSIVE);
  207. }
  208. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Save")), TTR("Save"), OBJ_MENU_SAVE);
  209. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("Save")), TTR("Save As..."), OBJ_MENU_SAVE_AS);
  210. }
  211. if (edited_resource->get_path().is_resource_file()) {
  212. edit_menu->add_separator();
  213. edit_menu->add_icon_item(get_editor_theme_icon(SNAME("ShowInFileSystem")), TTR("Show in FileSystem"), OBJ_MENU_SHOW_IN_FILE_SYSTEM);
  214. }
  215. }
  216. // Add options to copy/paste resource.
  217. Ref<Resource> cb = EditorSettings::get_singleton()->get_resource_clipboard();
  218. bool paste_valid = false;
  219. if (is_editable() && cb.is_valid()) {
  220. if (base_type.is_empty()) {
  221. paste_valid = true;
  222. } else {
  223. String res_type = _get_resource_type(cb);
  224. for (int i = 0; i < base_type.get_slice_count(","); i++) {
  225. String base = base_type.get_slice(",", i);
  226. paste_valid = ClassDB::is_parent_class(res_type, base) || EditorNode::get_editor_data().script_class_is_parent(res_type, base);
  227. if (paste_valid) {
  228. break;
  229. }
  230. }
  231. }
  232. }
  233. if (edited_resource.is_valid() || paste_valid) {
  234. edit_menu->add_separator();
  235. if (edited_resource.is_valid()) {
  236. edit_menu->add_item(TTR("Copy"), OBJ_MENU_COPY);
  237. }
  238. if (paste_valid) {
  239. edit_menu->add_item(TTR("Paste"), OBJ_MENU_PASTE);
  240. }
  241. }
  242. // Add options to convert existing resource to another type of resource.
  243. if (is_editable() && edited_resource.is_valid()) {
  244. Vector<Ref<EditorResourceConversionPlugin>> conversions = EditorNode::get_singleton()->find_resource_conversion_plugin_for_resource(edited_resource);
  245. if (!conversions.is_empty()) {
  246. edit_menu->add_separator();
  247. }
  248. int relative_id = 0;
  249. for (const Ref<EditorResourceConversionPlugin> &conversion : conversions) {
  250. String what = conversion->converts_to();
  251. Ref<Texture2D> icon;
  252. if (has_theme_icon(what, EditorStringName(EditorIcons))) {
  253. icon = get_editor_theme_icon(what);
  254. } else {
  255. icon = get_editor_theme_icon(SNAME("Object"));
  256. }
  257. edit_menu->add_icon_item(icon, vformat(TTR("Convert to %s"), what), CONVERT_BASE_ID + relative_id);
  258. relative_id++;
  259. }
  260. }
  261. }
  262. void EditorResourcePicker::_edit_menu_cbk(int p_which) {
  263. switch (p_which) {
  264. case OBJ_MENU_LOAD: {
  265. List<String> extensions;
  266. for (int i = 0; i < base_type.get_slice_count(","); i++) {
  267. String base = base_type.get_slice(",", i);
  268. ResourceLoader::get_recognized_extensions_for_type(base, &extensions);
  269. if (ScriptServer::is_global_class(base)) {
  270. ResourceLoader::get_recognized_extensions_for_type(ScriptServer::get_global_class_native_base(base), &extensions);
  271. }
  272. }
  273. HashSet<String> valid_extensions;
  274. for (const String &E : extensions) {
  275. valid_extensions.insert(E);
  276. }
  277. if (!file_dialog) {
  278. file_dialog = memnew(EditorFileDialog);
  279. file_dialog->set_file_mode(EditorFileDialog::FILE_MODE_OPEN_FILE);
  280. add_child(file_dialog);
  281. file_dialog->connect("file_selected", callable_mp(this, &EditorResourcePicker::_file_selected));
  282. }
  283. file_dialog->clear_filters();
  284. for (const String &E : valid_extensions) {
  285. file_dialog->add_filter("*." + E, E.to_upper());
  286. }
  287. file_dialog->popup_file_dialog();
  288. } break;
  289. case OBJ_MENU_QUICKLOAD: {
  290. const Vector<String> &base_types_string = base_type.split(",");
  291. Vector<StringName> base_types;
  292. for (const String &type : base_types_string) {
  293. base_types.push_back(type);
  294. }
  295. EditorNode::get_singleton()->get_quick_open_dialog()->popup_dialog(base_types, callable_mp(this, &EditorResourcePicker::_file_selected));
  296. } break;
  297. case OBJ_MENU_INSPECT: {
  298. if (edited_resource.is_valid()) {
  299. emit_signal(SNAME("resource_selected"), edited_resource, true);
  300. }
  301. } break;
  302. case OBJ_MENU_CLEAR: {
  303. edited_resource = Ref<Resource>();
  304. _resource_changed();
  305. } break;
  306. case OBJ_MENU_MAKE_UNIQUE: {
  307. if (edited_resource.is_null()) {
  308. return;
  309. }
  310. Ref<Resource> unique_resource = edited_resource->duplicate();
  311. ERR_FAIL_COND(unique_resource.is_null()); // duplicate() may fail.
  312. edited_resource = unique_resource;
  313. _resource_changed();
  314. } break;
  315. case OBJ_MENU_MAKE_UNIQUE_RECURSIVE: {
  316. if (edited_resource.is_null()) {
  317. return;
  318. }
  319. if (!duplicate_resources_dialog) {
  320. duplicate_resources_dialog = memnew(ConfirmationDialog);
  321. add_child(duplicate_resources_dialog);
  322. duplicate_resources_dialog->set_title(TTR("Make Unique (Recursive)"));
  323. duplicate_resources_dialog->connect(SceneStringName(confirmed), callable_mp(this, &EditorResourcePicker::_duplicate_selected_resources));
  324. VBoxContainer *vb = memnew(VBoxContainer);
  325. duplicate_resources_dialog->add_child(vb);
  326. Label *label = memnew(Label(TTR("Select resources to make unique:")));
  327. vb->add_child(label);
  328. duplicate_resources_tree = memnew(Tree);
  329. duplicate_resources_tree->set_auto_translate_mode(AUTO_TRANSLATE_MODE_DISABLED);
  330. vb->add_child(duplicate_resources_tree);
  331. duplicate_resources_tree->set_columns(2);
  332. duplicate_resources_tree->set_v_size_flags(SIZE_EXPAND_FILL);
  333. }
  334. duplicate_resources_tree->clear();
  335. TreeItem *root = duplicate_resources_tree->create_item();
  336. _gather_resources_to_duplicate(edited_resource, root);
  337. duplicate_resources_dialog->reset_size();
  338. duplicate_resources_dialog->popup_centered(Vector2(500, 400) * EDSCALE);
  339. } break;
  340. case OBJ_MENU_SAVE: {
  341. if (edited_resource.is_null()) {
  342. return;
  343. }
  344. EditorNode::get_singleton()->save_resource(edited_resource);
  345. } break;
  346. case OBJ_MENU_SAVE_AS: {
  347. if (edited_resource.is_null()) {
  348. return;
  349. }
  350. Callable resource_saved = callable_mp(this, &EditorResourcePicker::_resource_saved);
  351. if (!EditorNode::get_singleton()->is_connected("resource_saved", resource_saved)) {
  352. EditorNode::get_singleton()->connect("resource_saved", resource_saved);
  353. }
  354. EditorNode::get_singleton()->save_resource_as(edited_resource);
  355. } break;
  356. case OBJ_MENU_COPY: {
  357. EditorSettings::get_singleton()->set_resource_clipboard(edited_resource);
  358. } break;
  359. case OBJ_MENU_PASTE: {
  360. edited_resource = EditorSettings::get_singleton()->get_resource_clipboard();
  361. if (edited_resource->is_built_in() && EditorNode::get_singleton()->get_edited_scene() &&
  362. edited_resource->get_path().get_slice("::", 0) != EditorNode::get_singleton()->get_edited_scene()->get_scene_file_path()) {
  363. // Automatically make resource unique if it belongs to another scene.
  364. _edit_menu_cbk(OBJ_MENU_MAKE_UNIQUE);
  365. return;
  366. }
  367. _resource_changed();
  368. } break;
  369. case OBJ_MENU_SHOW_IN_FILE_SYSTEM: {
  370. FileSystemDock::get_singleton()->navigate_to_path(edited_resource->get_path());
  371. } break;
  372. default: {
  373. // Allow subclasses to handle their own options first, only then fallback on the default branch logic.
  374. if (handle_menu_selected(p_which)) {
  375. break;
  376. }
  377. if (p_which >= CONVERT_BASE_ID) {
  378. int to_type = p_which - CONVERT_BASE_ID;
  379. Vector<Ref<EditorResourceConversionPlugin>> conversions = EditorNode::get_singleton()->find_resource_conversion_plugin_for_resource(edited_resource);
  380. ERR_FAIL_INDEX(to_type, conversions.size());
  381. edited_resource = conversions[to_type]->convert(edited_resource);
  382. _resource_changed();
  383. break;
  384. }
  385. ERR_FAIL_COND(inheritors_array.is_empty());
  386. String intype = inheritors_array[p_which - TYPE_BASE_ID];
  387. Variant obj;
  388. if (ScriptServer::is_global_class(intype)) {
  389. obj = EditorNode::get_editor_data().script_class_instance(intype);
  390. } else {
  391. obj = ClassDB::instantiate(intype);
  392. }
  393. if (!obj) {
  394. obj = EditorNode::get_editor_data().instantiate_custom_type(intype, "Resource");
  395. }
  396. Resource *resp = Object::cast_to<Resource>(obj);
  397. ERR_BREAK(!resp);
  398. EditorNode::get_editor_data().instantiate_object_properties(obj);
  399. // Prevent freeing of the object until the end of the update of the resource (GH-88286).
  400. Ref<Resource> old_edited_resource = edited_resource;
  401. edited_resource = Ref<Resource>(resp);
  402. _resource_changed();
  403. } break;
  404. }
  405. }
  406. void EditorResourcePicker::set_create_options(Object *p_menu_node) {
  407. _ensure_resource_menu();
  408. // If a subclass implements this method, use it to replace all create items.
  409. if (GDVIRTUAL_CALL(_set_create_options, p_menu_node)) {
  410. return;
  411. }
  412. // By default provide generic "New ..." options.
  413. if (!base_type.is_empty()) {
  414. int idx = 0;
  415. _ensure_allowed_types();
  416. HashSet<StringName> allowed_types = allowed_types_without_convert;
  417. for (const StringName &E : allowed_types) {
  418. const String &t = E;
  419. if (!ClassDB::can_instantiate(t)) {
  420. continue;
  421. }
  422. inheritors_array.push_back(t);
  423. Ref<Texture2D> icon = EditorNode::get_singleton()->get_class_icon(t, "Object");
  424. int id = TYPE_BASE_ID + idx;
  425. edit_menu->add_icon_item(icon, vformat(TTR("New %s"), t), id);
  426. HashMap<String, DocData::ClassDoc>::Iterator class_doc = EditorHelp::get_doc_data()->class_list.find(t);
  427. if (class_doc) {
  428. edit_menu->set_item_tooltip(-1, DTR(class_doc->value.brief_description));
  429. }
  430. idx++;
  431. }
  432. if (edit_menu->get_item_count()) {
  433. edit_menu->add_separator();
  434. }
  435. }
  436. }
  437. bool EditorResourcePicker::handle_menu_selected(int p_which) {
  438. bool success = false;
  439. GDVIRTUAL_CALL(_handle_menu_selected, p_which, success);
  440. return success;
  441. }
  442. void EditorResourcePicker::_button_draw() {
  443. if (dropping) {
  444. Color color = get_theme_color(SNAME("accent_color"), EditorStringName(Editor));
  445. assign_button->draw_rect(Rect2(Point2(), assign_button->get_size()), color, false);
  446. }
  447. }
  448. void EditorResourcePicker::_button_input(const Ref<InputEvent> &p_event) {
  449. Ref<InputEventMouseButton> mb = p_event;
  450. if (mb.is_valid() && mb->is_pressed() && mb->get_button_index() == MouseButton::RIGHT) {
  451. // Only attempt to update and show the menu if we have
  452. // a valid resource or the Picker is editable, as
  453. // there will otherwise be nothing to display.
  454. if (edited_resource.is_valid() || is_editable()) {
  455. _update_menu_items();
  456. Vector2 pos = get_screen_position() + mb->get_position();
  457. edit_menu->reset_size();
  458. edit_menu->set_position(pos);
  459. edit_menu->popup();
  460. }
  461. }
  462. }
  463. String EditorResourcePicker::_get_resource_type(const Ref<Resource> &p_resource) const {
  464. if (p_resource.is_null()) {
  465. return String();
  466. }
  467. String res_type = p_resource->get_class();
  468. Ref<Script> res_script = p_resource->get_script();
  469. if (res_script.is_null()) {
  470. return res_type;
  471. }
  472. // TODO: Replace with EditorFileSystem when PR #60606 is merged to use cached resource type.
  473. String script_type = EditorNode::get_editor_data().script_class_get_name(res_script->get_path());
  474. if (!script_type.is_empty()) {
  475. res_type = script_type;
  476. }
  477. return res_type;
  478. }
  479. static void _add_allowed_type(const StringName &p_type, List<StringName> *p_vector) {
  480. if (p_vector->find(p_type)) {
  481. // Already added.
  482. return;
  483. }
  484. if (ClassDB::class_exists(p_type)) {
  485. // Engine class,
  486. if (!ClassDB::is_virtual(p_type)) {
  487. p_vector->push_back(p_type);
  488. }
  489. List<StringName> inheriters;
  490. ClassDB::get_inheriters_from_class(p_type, &inheriters);
  491. for (const StringName &S : inheriters) {
  492. _add_allowed_type(S, p_vector);
  493. }
  494. } else {
  495. // Script class.
  496. p_vector->push_back(p_type);
  497. }
  498. List<StringName> inheriters;
  499. ScriptServer::get_inheriters_list(p_type, &inheriters);
  500. for (const StringName &S : inheriters) {
  501. _add_allowed_type(S, p_vector);
  502. }
  503. }
  504. void EditorResourcePicker::_ensure_allowed_types() const {
  505. if (!allowed_types_without_convert.is_empty()) {
  506. return;
  507. }
  508. List<StringName> final_allowed;
  509. Vector<String> allowed_types = base_type.split(",");
  510. int size = allowed_types.size();
  511. for (const String &S : allowed_types) {
  512. const String base = S.strip_edges();
  513. if (base.begins_with("-")) {
  514. final_allowed.erase(base.right(-1));
  515. continue;
  516. }
  517. _add_allowed_type(base, &final_allowed);
  518. }
  519. for (const StringName &SN : final_allowed) {
  520. allowed_types_without_convert.insert(SN);
  521. }
  522. allowed_types_with_convert = HashSet<StringName>(allowed_types_without_convert);
  523. for (int i = 0; i < size; i++) {
  524. const String base = allowed_types[i].strip_edges();
  525. if (base == "BaseMaterial3D") {
  526. allowed_types_with_convert.insert("Texture2D");
  527. } else if (ClassDB::is_parent_class("ShaderMaterial", base)) {
  528. allowed_types_with_convert.insert("Shader");
  529. } else if (ClassDB::is_parent_class("ImageTexture", base)) {
  530. allowed_types_with_convert.insert("Image");
  531. }
  532. }
  533. }
  534. bool EditorResourcePicker::_is_drop_valid(const Dictionary &p_drag_data) const {
  535. {
  536. const ObjectID source_picker = p_drag_data.get("source_picker", ObjectID());
  537. if (source_picker == get_instance_id()) {
  538. return false;
  539. }
  540. }
  541. if (base_type.is_empty()) {
  542. return true;
  543. }
  544. Dictionary drag_data = p_drag_data;
  545. Ref<Resource> res;
  546. if (drag_data.has("type") && String(drag_data["type"]) == "script_list_element") {
  547. ScriptEditorBase *se = Object::cast_to<ScriptEditorBase>(drag_data["script_list_element"]);
  548. if (se) {
  549. res = se->get_edited_resource();
  550. }
  551. } else if (drag_data.has("type") && String(drag_data["type"]) == "resource") {
  552. res = drag_data["resource"];
  553. } else if (drag_data.has("type") && String(drag_data["type"]) == "files") {
  554. Vector<String> files = drag_data["files"];
  555. // TODO: Extract the typename of the dropped filepath's resource in a more performant way, without fully loading it.
  556. if (files.size() == 1) {
  557. res = ResourceLoader::load(files[0]);
  558. }
  559. }
  560. _ensure_allowed_types();
  561. HashSet<StringName> allowed_types = allowed_types_with_convert;
  562. if (res.is_valid()) {
  563. String res_type = _get_resource_type(res);
  564. if (_is_type_valid(res_type, allowed_types)) {
  565. return true;
  566. }
  567. StringName custom_class = EditorNode::get_singleton()->get_object_custom_type_name(res.ptr());
  568. if (_is_type_valid(custom_class, allowed_types)) {
  569. return true;
  570. }
  571. }
  572. return false;
  573. }
  574. bool EditorResourcePicker::_is_type_valid(const String &p_type_name, const HashSet<StringName> &p_allowed_types) const {
  575. for (const StringName &E : p_allowed_types) {
  576. String at = E;
  577. if (p_type_name == at || ClassDB::is_parent_class(p_type_name, at) || EditorNode::get_editor_data().script_class_is_parent(p_type_name, at)) {
  578. return true;
  579. }
  580. }
  581. return false;
  582. }
  583. bool EditorResourcePicker::_is_custom_type_script() const {
  584. Ref<Script> resource_as_script = edited_resource;
  585. if (resource_as_script.is_valid() && resource_owner && resource_owner->has_meta(SceneStringName(_custom_type_script))) {
  586. return true;
  587. }
  588. return false;
  589. }
  590. Variant EditorResourcePicker::get_drag_data_fw(const Point2 &p_point, Control *p_from) {
  591. if (edited_resource.is_valid()) {
  592. Dictionary drag_data = EditorNode::get_singleton()->drag_resource(edited_resource, p_from);
  593. drag_data["source_picker"] = get_instance_id();
  594. return drag_data;
  595. }
  596. return Variant();
  597. }
  598. bool EditorResourcePicker::can_drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) const {
  599. return editable && _is_drop_valid(p_data);
  600. }
  601. void EditorResourcePicker::drop_data_fw(const Point2 &p_point, const Variant &p_data, Control *p_from) {
  602. ERR_FAIL_COND(!_is_drop_valid(p_data));
  603. Dictionary drag_data = p_data;
  604. Ref<Resource> dropped_resource;
  605. if (drag_data.has("type") && String(drag_data["type"]) == "script_list_element") {
  606. ScriptEditorBase *se = Object::cast_to<ScriptEditorBase>(drag_data["script_list_element"]);
  607. if (se) {
  608. dropped_resource = se->get_edited_resource();
  609. }
  610. } else if (drag_data.has("type") && String(drag_data["type"]) == "resource") {
  611. dropped_resource = drag_data["resource"];
  612. }
  613. if (dropped_resource.is_null() && drag_data.has("type") && String(drag_data["type"]) == "files") {
  614. Vector<String> files = drag_data["files"];
  615. if (files.size() == 1) {
  616. dropped_resource = ResourceLoader::load(files[0]);
  617. }
  618. }
  619. if (dropped_resource.is_valid()) {
  620. _ensure_allowed_types();
  621. HashSet<StringName> allowed_types = allowed_types_without_convert;
  622. String res_type = _get_resource_type(dropped_resource);
  623. // If the accepted dropped resource is from the extended list, it requires conversion.
  624. if (!_is_type_valid(res_type, allowed_types)) {
  625. for (const StringName &E : allowed_types) {
  626. String at = E;
  627. if (at == "BaseMaterial3D" && Ref<Texture2D>(dropped_resource).is_valid()) {
  628. // Use existing resource if possible and only replace its data.
  629. Ref<StandardMaterial3D> mat = edited_resource;
  630. if (mat.is_null()) {
  631. mat.instantiate();
  632. }
  633. mat->set_texture(StandardMaterial3D::TextureParam::TEXTURE_ALBEDO, dropped_resource);
  634. dropped_resource = mat;
  635. break;
  636. }
  637. if (at == "ShaderMaterial" && Ref<Shader>(dropped_resource).is_valid()) {
  638. Ref<ShaderMaterial> mat = edited_resource;
  639. if (mat.is_null()) {
  640. mat.instantiate();
  641. }
  642. mat->set_shader(dropped_resource);
  643. dropped_resource = mat;
  644. break;
  645. }
  646. if (at == "ImageTexture" && Ref<Image>(dropped_resource).is_valid()) {
  647. Ref<ImageTexture> texture = edited_resource;
  648. if (texture.is_null()) {
  649. texture.instantiate();
  650. }
  651. texture->set_image(dropped_resource);
  652. dropped_resource = texture;
  653. break;
  654. }
  655. }
  656. }
  657. edited_resource = dropped_resource;
  658. _resource_changed();
  659. }
  660. }
  661. void EditorResourcePicker::_bind_methods() {
  662. ClassDB::bind_method(D_METHOD("_update_resource_preview"), &EditorResourcePicker::_update_resource_preview);
  663. ClassDB::bind_method(D_METHOD("set_base_type", "base_type"), &EditorResourcePicker::set_base_type);
  664. ClassDB::bind_method(D_METHOD("get_base_type"), &EditorResourcePicker::get_base_type);
  665. ClassDB::bind_method(D_METHOD("get_allowed_types"), &EditorResourcePicker::get_allowed_types);
  666. ClassDB::bind_method(D_METHOD("set_edited_resource", "resource"), &EditorResourcePicker::set_edited_resource);
  667. ClassDB::bind_method(D_METHOD("get_edited_resource"), &EditorResourcePicker::get_edited_resource);
  668. ClassDB::bind_method(D_METHOD("set_toggle_mode", "enable"), &EditorResourcePicker::set_toggle_mode);
  669. ClassDB::bind_method(D_METHOD("is_toggle_mode"), &EditorResourcePicker::is_toggle_mode);
  670. ClassDB::bind_method(D_METHOD("set_toggle_pressed", "pressed"), &EditorResourcePicker::set_toggle_pressed);
  671. ClassDB::bind_method(D_METHOD("set_editable", "enable"), &EditorResourcePicker::set_editable);
  672. ClassDB::bind_method(D_METHOD("is_editable"), &EditorResourcePicker::is_editable);
  673. GDVIRTUAL_BIND(_set_create_options, "menu_node");
  674. GDVIRTUAL_BIND(_handle_menu_selected, "id");
  675. ADD_PROPERTY(PropertyInfo(Variant::STRING, "base_type"), "set_base_type", "get_base_type");
  676. ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "edited_resource", PROPERTY_HINT_RESOURCE_TYPE, "Resource", PROPERTY_USAGE_NONE), "set_edited_resource", "get_edited_resource");
  677. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "editable"), "set_editable", "is_editable");
  678. ADD_PROPERTY(PropertyInfo(Variant::BOOL, "toggle_mode"), "set_toggle_mode", "is_toggle_mode");
  679. ADD_SIGNAL(MethodInfo("resource_selected", PropertyInfo(Variant::OBJECT, "resource", PROPERTY_HINT_RESOURCE_TYPE, "Resource"), PropertyInfo(Variant::BOOL, "inspect")));
  680. ADD_SIGNAL(MethodInfo("resource_changed", PropertyInfo(Variant::OBJECT, "resource", PROPERTY_HINT_RESOURCE_TYPE, "Resource")));
  681. }
  682. void EditorResourcePicker::_notification(int p_what) {
  683. switch (p_what) {
  684. case NOTIFICATION_ENTER_TREE: {
  685. _update_resource();
  686. [[fallthrough]];
  687. }
  688. case NOTIFICATION_THEME_CHANGED: {
  689. const int icon_width = get_theme_constant(SNAME("class_icon_size"), EditorStringName(Editor));
  690. assign_button->add_theme_constant_override("icon_max_width", icon_width);
  691. if (edit_menu) {
  692. edit_menu->add_theme_constant_override("icon_max_width", icon_width);
  693. }
  694. edit_button->set_button_icon(get_theme_icon(SNAME("select_arrow"), SNAME("Tree")));
  695. } break;
  696. case NOTIFICATION_DRAW: {
  697. draw_style_box(get_theme_stylebox(SceneStringName(panel), SNAME("Tree")), Rect2(Point2(), get_size()));
  698. } break;
  699. case NOTIFICATION_DRAG_BEGIN: {
  700. if (editable && _is_drop_valid(get_viewport()->gui_get_drag_data())) {
  701. dropping = true;
  702. assign_button->queue_redraw();
  703. }
  704. } break;
  705. case NOTIFICATION_DRAG_END: {
  706. if (dropping) {
  707. dropping = false;
  708. assign_button->queue_redraw();
  709. }
  710. } break;
  711. case NOTIFICATION_EXIT_TREE: {
  712. Callable resource_saved = callable_mp(this, &EditorResourcePicker::_resource_saved);
  713. if (EditorNode::get_singleton()->is_connected("resource_saved", resource_saved)) {
  714. EditorNode::get_singleton()->disconnect("resource_saved", resource_saved);
  715. }
  716. } break;
  717. }
  718. }
  719. void EditorResourcePicker::set_assign_button_min_size(const Size2i &p_size) {
  720. assign_button_min_size = p_size;
  721. assign_button->set_custom_minimum_size(assign_button_min_size);
  722. }
  723. void EditorResourcePicker::set_base_type(const String &p_base_type) {
  724. base_type = p_base_type;
  725. // There is a possibility that the new base type is conflicting with the existing value.
  726. // Keep the value, but warn the user that there is a potential mistake.
  727. if (!base_type.is_empty() && edited_resource.is_valid()) {
  728. _ensure_allowed_types();
  729. HashSet<StringName> allowed_types = allowed_types_with_convert;
  730. StringName custom_class;
  731. bool is_custom = false;
  732. if (edited_resource->get_script()) {
  733. custom_class = EditorNode::get_singleton()->get_object_custom_type_name(edited_resource->get_script());
  734. is_custom = _is_type_valid(custom_class, allowed_types);
  735. }
  736. if (!is_custom && !_is_type_valid(edited_resource->get_class(), allowed_types)) {
  737. String class_str = (custom_class == StringName() ? edited_resource->get_class() : vformat("%s (%s)", custom_class, edited_resource->get_class()));
  738. WARN_PRINT(vformat("Value mismatch between the new base type of this EditorResourcePicker, '%s', and the type of the value it already has, '%s'.", base_type, class_str));
  739. }
  740. }
  741. }
  742. String EditorResourcePicker::get_base_type() const {
  743. return base_type;
  744. }
  745. Vector<String> EditorResourcePicker::get_allowed_types() const {
  746. _ensure_allowed_types();
  747. HashSet<StringName> allowed_types = allowed_types_without_convert;
  748. Vector<String> types;
  749. types.resize(allowed_types.size());
  750. int i = 0;
  751. String *w = types.ptrw();
  752. for (const StringName &E : allowed_types) {
  753. w[i] = E;
  754. i++;
  755. }
  756. return types;
  757. }
  758. void EditorResourcePicker::set_edited_resource(Ref<Resource> p_resource) {
  759. if (p_resource.is_null()) {
  760. edited_resource = Ref<Resource>();
  761. _update_resource();
  762. return;
  763. }
  764. if (!base_type.is_empty()) {
  765. _ensure_allowed_types();
  766. HashSet<StringName> allowed_types = allowed_types_with_convert;
  767. StringName custom_class;
  768. bool is_custom = false;
  769. if (p_resource->get_script()) {
  770. custom_class = EditorNode::get_singleton()->get_object_custom_type_name(p_resource->get_script());
  771. is_custom = _is_type_valid(custom_class, allowed_types);
  772. }
  773. if (!is_custom && !_is_type_valid(p_resource->get_class(), allowed_types)) {
  774. String class_str = (custom_class == StringName() ? p_resource->get_class() : vformat("%s (%s)", custom_class, p_resource->get_class()));
  775. ERR_FAIL_MSG(vformat("Failed to set a resource of the type '%s' because this EditorResourcePicker only accepts '%s' and its derivatives.", class_str, base_type));
  776. }
  777. }
  778. set_edited_resource_no_check(p_resource);
  779. }
  780. void EditorResourcePicker::set_edited_resource_no_check(Ref<Resource> p_resource) {
  781. edited_resource = p_resource;
  782. _update_resource();
  783. }
  784. Ref<Resource> EditorResourcePicker::get_edited_resource() {
  785. return edited_resource;
  786. }
  787. void EditorResourcePicker::set_toggle_mode(bool p_enable) {
  788. assign_button->set_toggle_mode(p_enable);
  789. }
  790. bool EditorResourcePicker::is_toggle_mode() const {
  791. return assign_button->is_toggle_mode();
  792. }
  793. void EditorResourcePicker::set_toggle_pressed(bool p_pressed) {
  794. if (!is_toggle_mode()) {
  795. return;
  796. }
  797. assign_button->set_pressed(p_pressed);
  798. }
  799. bool EditorResourcePicker::is_toggle_pressed() const {
  800. return assign_button->is_pressed();
  801. }
  802. void EditorResourcePicker::set_resource_owner(Object *p_object) {
  803. resource_owner = p_object;
  804. }
  805. void EditorResourcePicker::set_editable(bool p_editable) {
  806. editable = p_editable;
  807. assign_button->set_disabled(!editable && edited_resource.is_null());
  808. edit_button->set_visible(editable);
  809. }
  810. bool EditorResourcePicker::is_editable() const {
  811. return editable;
  812. }
  813. void EditorResourcePicker::_ensure_resource_menu() {
  814. if (edit_menu) {
  815. return;
  816. }
  817. edit_menu = memnew(PopupMenu);
  818. edit_menu->add_theme_constant_override("icon_max_width", get_theme_constant(SNAME("class_icon_size"), EditorStringName(Editor)));
  819. add_child(edit_menu);
  820. edit_menu->connect(SceneStringName(id_pressed), callable_mp(this, &EditorResourcePicker::_edit_menu_cbk));
  821. edit_menu->connect("popup_hide", callable_mp((BaseButton *)edit_button, &BaseButton::set_pressed).bind(false));
  822. }
  823. void EditorResourcePicker::_gather_resources_to_duplicate(const Ref<Resource> p_resource, TreeItem *p_item, const String &p_property_name) const {
  824. p_item->set_cell_mode(0, TreeItem::CELL_MODE_CHECK);
  825. String res_name = p_resource->get_name();
  826. if (res_name.is_empty() && !p_resource->is_built_in()) {
  827. res_name = p_resource->get_path().get_file();
  828. }
  829. if (res_name.is_empty()) {
  830. p_item->set_text(0, p_resource->get_class());
  831. } else {
  832. p_item->set_text(0, vformat("%s (%s)", p_resource->get_class(), res_name));
  833. }
  834. p_item->set_icon(0, EditorNode::get_singleton()->get_object_icon(p_resource.ptr()));
  835. p_item->set_editable(0, true);
  836. Array meta;
  837. meta.append(p_resource);
  838. p_item->set_metadata(0, meta);
  839. if (!p_property_name.is_empty()) {
  840. p_item->set_text(1, p_property_name);
  841. }
  842. static Vector<String> unique_exceptions = { "Image", "Shader", "Mesh", "FontFile" };
  843. if (!unique_exceptions.has(p_resource->get_class())) {
  844. // Automatically select resource, unless it's something that shouldn't be duplicated.
  845. p_item->set_checked(0, true);
  846. }
  847. List<PropertyInfo> plist;
  848. p_resource->get_property_list(&plist);
  849. for (const PropertyInfo &E : plist) {
  850. if (!(E.usage & PROPERTY_USAGE_STORAGE) || E.type != Variant::OBJECT || E.hint != PROPERTY_HINT_RESOURCE_TYPE) {
  851. continue;
  852. }
  853. Ref<Resource> res = p_resource->get(E.name);
  854. if (res.is_null()) {
  855. continue;
  856. }
  857. TreeItem *child = p_item->create_child();
  858. _gather_resources_to_duplicate(res, child, E.name);
  859. meta = child->get_metadata(0);
  860. // Remember property name.
  861. meta.append(E.name);
  862. if ((E.usage & PROPERTY_USAGE_NEVER_DUPLICATE)) {
  863. // The resource can't be duplicated, but make it appear on the list anyway.
  864. child->set_checked(0, false);
  865. child->set_editable(0, false);
  866. }
  867. }
  868. }
  869. void EditorResourcePicker::_duplicate_selected_resources() {
  870. for (TreeItem *item = duplicate_resources_tree->get_root(); item; item = item->get_next_in_tree()) {
  871. if (!item->is_checked(0)) {
  872. continue;
  873. }
  874. Array meta = item->get_metadata(0);
  875. Ref<Resource> res = meta[0];
  876. Ref<Resource> unique_resource = res->duplicate();
  877. ERR_FAIL_COND(unique_resource.is_null()); // duplicate() may fail.
  878. meta[0] = unique_resource;
  879. if (meta.size() == 1) { // Root.
  880. edited_resource = unique_resource;
  881. _resource_changed();
  882. } else {
  883. Array parent_meta = item->get_parent()->get_metadata(0);
  884. Ref<Resource> parent = parent_meta[0];
  885. parent->set(meta[1], unique_resource);
  886. }
  887. }
  888. }
  889. EditorResourcePicker::EditorResourcePicker(bool p_hide_assign_button_controls) {
  890. assign_button = memnew(Button);
  891. assign_button->set_flat(true);
  892. assign_button->set_h_size_flags(SIZE_EXPAND_FILL);
  893. assign_button->set_expand_icon(true);
  894. assign_button->set_clip_text(true);
  895. assign_button->set_auto_translate_mode(AUTO_TRANSLATE_MODE_DISABLED);
  896. SET_DRAG_FORWARDING_GCD(assign_button, EditorResourcePicker);
  897. add_child(assign_button);
  898. assign_button->connect(SceneStringName(pressed), callable_mp(this, &EditorResourcePicker::_resource_selected));
  899. assign_button->connect(SceneStringName(draw), callable_mp(this, &EditorResourcePicker::_button_draw));
  900. assign_button->connect(SceneStringName(gui_input), callable_mp(this, &EditorResourcePicker::_button_input));
  901. if (!p_hide_assign_button_controls) {
  902. preview_rect = memnew(TextureRect);
  903. preview_rect->set_expand_mode(TextureRect::EXPAND_IGNORE_SIZE);
  904. preview_rect->set_anchors_and_offsets_preset(PRESET_FULL_RECT);
  905. preview_rect->set_offset(SIDE_TOP, 1);
  906. preview_rect->set_offset(SIDE_BOTTOM, -1);
  907. preview_rect->set_offset(SIDE_RIGHT, -1);
  908. preview_rect->set_texture_filter(TEXTURE_FILTER_NEAREST_WITH_MIPMAPS);
  909. assign_button->add_child(preview_rect);
  910. }
  911. edit_button = memnew(Button);
  912. edit_button->set_flat(false);
  913. edit_button->set_toggle_mode(true);
  914. edit_button->set_action_mode(BaseButton::ACTION_MODE_BUTTON_PRESS);
  915. edit_button->connect(SceneStringName(pressed), callable_mp(this, &EditorResourcePicker::_update_menu));
  916. add_child(edit_button);
  917. edit_button->connect(SceneStringName(gui_input), callable_mp(this, &EditorResourcePicker::_button_input));
  918. add_theme_constant_override("separation", 0);
  919. }
  920. // EditorScriptPicker
  921. void EditorScriptPicker::set_create_options(Object *p_menu_node) {
  922. PopupMenu *menu_node = Object::cast_to<PopupMenu>(p_menu_node);
  923. if (!menu_node) {
  924. return;
  925. }
  926. if (!(script_owner && script_owner->has_meta(SceneStringName(_custom_type_script)))) {
  927. menu_node->add_icon_item(get_editor_theme_icon(SNAME("ScriptCreate")), TTR("New Script..."), OBJ_MENU_NEW_SCRIPT);
  928. }
  929. if (script_owner) {
  930. Ref<Script> scr = script_owner->get_script();
  931. if (scr.is_valid()) {
  932. menu_node->add_icon_item(get_editor_theme_icon(SNAME("ScriptExtend")), TTR("Extend Script..."), OBJ_MENU_EXTEND_SCRIPT);
  933. }
  934. }
  935. menu_node->add_separator();
  936. }
  937. bool EditorScriptPicker::handle_menu_selected(int p_which) {
  938. switch (p_which) {
  939. case OBJ_MENU_NEW_SCRIPT: {
  940. if (script_owner) {
  941. SceneTreeDock::get_singleton()->open_script_dialog(script_owner, false);
  942. }
  943. return true;
  944. }
  945. case OBJ_MENU_EXTEND_SCRIPT: {
  946. if (script_owner) {
  947. SceneTreeDock::get_singleton()->open_script_dialog(script_owner, true);
  948. }
  949. return true;
  950. }
  951. }
  952. return false;
  953. }
  954. void EditorScriptPicker::set_script_owner(Node *p_owner) {
  955. script_owner = p_owner;
  956. }
  957. Node *EditorScriptPicker::get_script_owner() const {
  958. return script_owner;
  959. }
  960. void EditorScriptPicker::_bind_methods() {
  961. ClassDB::bind_method(D_METHOD("set_script_owner", "owner_node"), &EditorScriptPicker::set_script_owner);
  962. ClassDB::bind_method(D_METHOD("get_script_owner"), &EditorScriptPicker::get_script_owner);
  963. ADD_PROPERTY(PropertyInfo(Variant::OBJECT, "script_owner", PROPERTY_HINT_RESOURCE_TYPE, "Node", PROPERTY_USAGE_NONE), "set_script_owner", "get_script_owner");
  964. }
  965. EditorScriptPicker::EditorScriptPicker() {
  966. }
  967. // EditorShaderPicker
  968. void EditorShaderPicker::set_create_options(Object *p_menu_node) {
  969. PopupMenu *menu_node = Object::cast_to<PopupMenu>(p_menu_node);
  970. if (!menu_node) {
  971. return;
  972. }
  973. menu_node->add_icon_item(get_editor_theme_icon(SNAME("Shader")), TTR("New Shader..."), OBJ_MENU_NEW_SHADER);
  974. menu_node->add_separator();
  975. }
  976. bool EditorShaderPicker::handle_menu_selected(int p_which) {
  977. Ref<ShaderMaterial> ed_material = Ref<ShaderMaterial>(get_edited_material());
  978. switch (p_which) {
  979. case OBJ_MENU_NEW_SHADER: {
  980. if (ed_material.is_valid()) {
  981. SceneTreeDock::get_singleton()->open_shader_dialog(ed_material, preferred_mode);
  982. return true;
  983. }
  984. } break;
  985. default:
  986. break;
  987. }
  988. return false;
  989. }
  990. void EditorShaderPicker::set_edited_material(ShaderMaterial *p_material) {
  991. edited_material = p_material;
  992. }
  993. ShaderMaterial *EditorShaderPicker::get_edited_material() const {
  994. return edited_material;
  995. }
  996. void EditorShaderPicker::set_preferred_mode(int p_mode) {
  997. preferred_mode = p_mode;
  998. }
  999. EditorShaderPicker::EditorShaderPicker() {
  1000. }
  1001. //////////////
  1002. void EditorAudioStreamPicker::_notification(int p_what) {
  1003. switch (p_what) {
  1004. case NOTIFICATION_READY:
  1005. case NOTIFICATION_THEME_CHANGED: {
  1006. _update_resource();
  1007. } break;
  1008. case NOTIFICATION_INTERNAL_PROCESS: {
  1009. Ref<AudioStream> audio_stream = get_edited_resource();
  1010. if (audio_stream.is_valid()) {
  1011. if (audio_stream->get_length() > 0) {
  1012. Ref<AudioStreamPreview> preview = AudioStreamPreviewGenerator::get_singleton()->generate_preview(audio_stream);
  1013. if (preview.is_valid()) {
  1014. if (preview->get_version() != last_preview_version) {
  1015. stream_preview_rect->queue_redraw();
  1016. last_preview_version = preview->get_version();
  1017. }
  1018. }
  1019. }
  1020. uint64_t tagged_frame = audio_stream->get_tagged_frame();
  1021. uint64_t diff_frames = AudioServer::get_singleton()->get_mixed_frames() - tagged_frame;
  1022. uint64_t diff_msec = diff_frames * 1000 / AudioServer::get_singleton()->get_mix_rate();
  1023. if (diff_msec < 300) {
  1024. uint32_t count = audio_stream->get_tagged_frame_count();
  1025. bool differ = false;
  1026. if (count != tagged_frame_offset_count) {
  1027. differ = true;
  1028. }
  1029. float offsets[MAX_TAGGED_FRAMES];
  1030. for (uint32_t i = 0; i < MIN(count, uint32_t(MAX_TAGGED_FRAMES)); i++) {
  1031. offsets[i] = audio_stream->get_tagged_frame_offset(i);
  1032. if (offsets[i] != tagged_frame_offsets[i]) {
  1033. differ = true;
  1034. }
  1035. }
  1036. if (differ) {
  1037. tagged_frame_offset_count = count;
  1038. for (uint32_t i = 0; i < count; i++) {
  1039. tagged_frame_offsets[i] = offsets[i];
  1040. }
  1041. }
  1042. stream_preview_rect->queue_redraw();
  1043. } else {
  1044. if (tagged_frame_offset_count != 0) {
  1045. stream_preview_rect->queue_redraw();
  1046. }
  1047. tagged_frame_offset_count = 0;
  1048. }
  1049. }
  1050. } break;
  1051. }
  1052. }
  1053. void EditorAudioStreamPicker::_update_resource() {
  1054. EditorResourcePicker::_update_resource();
  1055. Ref<Font> font = get_theme_font(SceneStringName(font), SNAME("Label"));
  1056. int font_size = get_theme_font_size(SceneStringName(font_size), SNAME("Label"));
  1057. Ref<AudioStream> audio_stream = get_edited_resource();
  1058. if (audio_stream.is_valid() && audio_stream->get_length() > 0.0) {
  1059. set_assign_button_min_size(Size2(1, font->get_height(font_size) * 3));
  1060. } else {
  1061. set_assign_button_min_size(Size2(1, font->get_height(font_size) * 1.5));
  1062. }
  1063. stream_preview_rect->queue_redraw();
  1064. }
  1065. void EditorAudioStreamPicker::_preview_draw() {
  1066. Ref<AudioStream> audio_stream = get_edited_resource();
  1067. if (audio_stream.is_null()) {
  1068. get_assign_button()->set_text(TTR("<empty>"));
  1069. return;
  1070. }
  1071. int font_size = get_theme_font_size(SceneStringName(font_size), SNAME("Label"));
  1072. get_assign_button()->set_text("");
  1073. Size2i size = stream_preview_rect->get_size();
  1074. Ref<Font> font = get_theme_font(SceneStringName(font), SNAME("Label"));
  1075. Rect2 rect(Point2(), size);
  1076. if (audio_stream->get_length() > 0 && size.width > 0) {
  1077. rect.size.height *= 0.5;
  1078. Ref<AudioStreamPreview> preview = AudioStreamPreviewGenerator::get_singleton()->generate_preview(audio_stream);
  1079. float preview_len = preview->get_length();
  1080. Vector<Vector2> points;
  1081. points.resize(size.width * 2);
  1082. for (int i = 0; i < size.width; i++) {
  1083. float ofs = i * preview_len / size.width;
  1084. float ofs_n = (i + 1) * preview_len / size.width;
  1085. float max = preview->get_max(ofs, ofs_n) * 0.5 + 0.5;
  1086. float min = preview->get_min(ofs, ofs_n) * 0.5 + 0.5;
  1087. int idx = i;
  1088. points.write[idx * 2 + 0] = Vector2(i + 1, rect.position.y + min * rect.size.y);
  1089. points.write[idx * 2 + 1] = Vector2(i + 1, rect.position.y + max * rect.size.y);
  1090. }
  1091. Vector<Color> colors = { get_theme_color(SNAME("contrast_color_2"), EditorStringName(Editor)) };
  1092. RS::get_singleton()->canvas_item_add_multiline(stream_preview_rect->get_canvas_item(), points, colors);
  1093. if (tagged_frame_offset_count) {
  1094. Color accent = get_theme_color(SNAME("accent_color"), EditorStringName(Editor));
  1095. for (uint32_t i = 0; i < tagged_frame_offset_count; i++) {
  1096. int x = CLAMP(tagged_frame_offsets[i] * size.width / preview_len, 0, size.width);
  1097. if (x == 0) {
  1098. continue; // Because some may always return 0, ignore offset 0.
  1099. }
  1100. stream_preview_rect->draw_rect(Rect2i(x, 0, 2, rect.size.height), accent);
  1101. }
  1102. }
  1103. rect.position.y += rect.size.height;
  1104. }
  1105. Ref<Texture2D> icon;
  1106. Color icon_modulate(1, 1, 1, 1);
  1107. if (tagged_frame_offset_count > 0) {
  1108. icon = get_editor_theme_icon(SNAME("Play"));
  1109. if ((OS::get_singleton()->get_ticks_msec() % 500) > 250) {
  1110. icon_modulate = Color(1, 0.5, 0.5, 1); // get_theme_color(SNAME("accent_color"), EditorStringName(Editor));
  1111. }
  1112. } else {
  1113. icon = EditorNode::get_singleton()->get_object_icon(audio_stream.operator->(), "Object");
  1114. }
  1115. String text;
  1116. if (!audio_stream->get_name().is_empty()) {
  1117. text = audio_stream->get_name();
  1118. } else if (audio_stream->get_path().is_resource_file()) {
  1119. text = audio_stream->get_path().get_file();
  1120. } else {
  1121. text = audio_stream->get_class().replace_first("AudioStream", "");
  1122. }
  1123. stream_preview_rect->draw_texture(icon, Point2i(EDSCALE * 4, rect.position.y + (rect.size.height - icon->get_height()) / 2), icon_modulate);
  1124. stream_preview_rect->draw_string(font, Point2i(EDSCALE * 4 + icon->get_width(), rect.position.y + font->get_ascent(font_size) + (rect.size.height - font->get_height(font_size)) / 2), text, HORIZONTAL_ALIGNMENT_CENTER, size.width - 4 * EDSCALE - icon->get_width(), font_size, get_theme_color(SceneStringName(font_color), EditorStringName(Editor)));
  1125. }
  1126. EditorAudioStreamPicker::EditorAudioStreamPicker() :
  1127. EditorResourcePicker(true) {
  1128. stream_preview_rect = memnew(Control);
  1129. stream_preview_rect->set_anchors_and_offsets_preset(PRESET_FULL_RECT);
  1130. stream_preview_rect->set_offset(SIDE_TOP, 1);
  1131. stream_preview_rect->set_offset(SIDE_BOTTOM, -1);
  1132. stream_preview_rect->set_offset(SIDE_RIGHT, -1);
  1133. stream_preview_rect->set_mouse_filter(MOUSE_FILTER_IGNORE);
  1134. stream_preview_rect->connect(SceneStringName(draw), callable_mp(this, &EditorAudioStreamPicker::_preview_draw));
  1135. get_assign_button()->add_child(stream_preview_rect);
  1136. get_assign_button()->move_child(stream_preview_rect, 0);
  1137. set_process_internal(true);
  1138. }