From 5fe6984639b5db9efa14103698e489040315ebfb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20Verschelde?= Date: Mon, 28 Mar 2022 14:10:28 +0200 Subject: Modules: Don't build editor-specific classes in templates They're moved to an `editor` subfolder so that we can easily handle them separately. --- modules/gridmap/SCsub | 3 + modules/gridmap/editor/grid_map_editor_plugin.cpp | 1485 +++++++++++++++++++++ modules/gridmap/editor/grid_map_editor_plugin.h | 256 ++++ modules/gridmap/grid_map_editor_plugin.cpp | 1481 -------------------- modules/gridmap/grid_map_editor_plugin.h | 252 ---- modules/gridmap/register_types.cpp | 12 +- 6 files changed, 1752 insertions(+), 1737 deletions(-) create mode 100644 modules/gridmap/editor/grid_map_editor_plugin.cpp create mode 100644 modules/gridmap/editor/grid_map_editor_plugin.h delete mode 100644 modules/gridmap/grid_map_editor_plugin.cpp delete mode 100644 modules/gridmap/grid_map_editor_plugin.h (limited to 'modules/gridmap') diff --git a/modules/gridmap/SCsub b/modules/gridmap/SCsub index 970ce534f0..52777235b8 100644 --- a/modules/gridmap/SCsub +++ b/modules/gridmap/SCsub @@ -5,4 +5,7 @@ Import("env_modules") env_gridmap = env_modules.Clone() +# Godot's own source files env_gridmap.add_source_files(env.modules_sources, "*.cpp") +if env["tools"]: + env_gridmap.add_source_files(env.modules_sources, "editor/*.cpp") diff --git a/modules/gridmap/editor/grid_map_editor_plugin.cpp b/modules/gridmap/editor/grid_map_editor_plugin.cpp new file mode 100644 index 0000000000..68968325dd --- /dev/null +++ b/modules/gridmap/editor/grid_map_editor_plugin.cpp @@ -0,0 +1,1485 @@ +/*************************************************************************/ +/* grid_map_editor_plugin.cpp */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* https://godotengine.org */ +/*************************************************************************/ +/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */ +/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */ +/* */ +/* Permission is hereby granted, free of charge, to any person obtaining */ +/* a copy of this software and associated documentation files (the */ +/* "Software"), to deal in the Software without restriction, including */ +/* without limitation the rights to use, copy, modify, merge, publish, */ +/* distribute, sublicense, and/or sell copies of the Software, and to */ +/* permit persons to whom the Software is furnished to do so, subject to */ +/* the following conditions: */ +/* */ +/* The above copyright notice and this permission notice shall be */ +/* included in all copies or substantial portions of the Software. */ +/* */ +/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ +/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ +/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/ +/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ +/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ +/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ +/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ +/*************************************************************************/ + +#include "grid_map_editor_plugin.h" + +#ifdef TOOLS_ENABLED + +#include "core/input/input.h" +#include "core/os/keyboard.h" +#include "editor/editor_node.h" +#include "editor/editor_scale.h" +#include "editor/editor_settings.h" +#include "editor/plugins/node_3d_editor_plugin.h" +#include "scene/3d/camera_3d.h" +#include "scene/main/window.h" + +void GridMapEditor::_node_removed(Node *p_node) { + if (p_node == node) { + node = nullptr; + } +} + +void GridMapEditor::_configure() { + if (!node) { + return; + } + + update_grid(); +} + +void GridMapEditor::_menu_option(int p_option) { + switch (p_option) { + case MENU_OPTION_PREV_LEVEL: { + floor->set_value(floor->get_value() - 1); + } break; + + case MENU_OPTION_NEXT_LEVEL: { + floor->set_value(floor->get_value() + 1); + } break; + + case MENU_OPTION_X_AXIS: + case MENU_OPTION_Y_AXIS: + case MENU_OPTION_Z_AXIS: { + int new_axis = p_option - MENU_OPTION_X_AXIS; + for (int i = 0; i < 3; i++) { + int idx = options->get_popup()->get_item_index(MENU_OPTION_X_AXIS + i); + options->get_popup()->set_item_checked(idx, i == new_axis); + } + + if (edit_axis != new_axis) { + int item1 = options->get_popup()->get_item_index(MENU_OPTION_NEXT_LEVEL); + int item2 = options->get_popup()->get_item_index(MENU_OPTION_PREV_LEVEL); + if (edit_axis == Vector3::AXIS_Y) { + options->get_popup()->set_item_text(item1, TTR("Next Plane")); + options->get_popup()->set_item_text(item2, TTR("Previous Plane")); + spin_box_label->set_text(TTR("Plane:")); + } else if (new_axis == Vector3::AXIS_Y) { + options->get_popup()->set_item_text(item1, TTR("Next Floor")); + options->get_popup()->set_item_text(item2, TTR("Previous Floor")); + spin_box_label->set_text(TTR("Floor:")); + } + } + edit_axis = Vector3::Axis(new_axis); + update_grid(); + + } break; + case MENU_OPTION_CURSOR_ROTATE_Y: { + Basis r; + if (input_action == INPUT_PASTE) { + r.set_orthogonal_index(paste_indicator.orientation); + r.rotate(Vector3(0, 1, 0), -Math_PI / 2.0); + paste_indicator.orientation = r.get_orthogonal_index(); + _update_paste_indicator(); + break; + } + + r.set_orthogonal_index(cursor_rot); + r.rotate(Vector3(0, 1, 0), -Math_PI / 2.0); + cursor_rot = r.get_orthogonal_index(); + _update_cursor_transform(); + } break; + case MENU_OPTION_CURSOR_ROTATE_X: { + Basis r; + if (input_action == INPUT_PASTE) { + r.set_orthogonal_index(paste_indicator.orientation); + r.rotate(Vector3(1, 0, 0), -Math_PI / 2.0); + paste_indicator.orientation = r.get_orthogonal_index(); + _update_paste_indicator(); + break; + } + + r.set_orthogonal_index(cursor_rot); + r.rotate(Vector3(1, 0, 0), -Math_PI / 2.0); + cursor_rot = r.get_orthogonal_index(); + _update_cursor_transform(); + } break; + case MENU_OPTION_CURSOR_ROTATE_Z: { + Basis r; + if (input_action == INPUT_PASTE) { + r.set_orthogonal_index(paste_indicator.orientation); + r.rotate(Vector3(0, 0, 1), -Math_PI / 2.0); + paste_indicator.orientation = r.get_orthogonal_index(); + _update_paste_indicator(); + break; + } + + r.set_orthogonal_index(cursor_rot); + r.rotate(Vector3(0, 0, 1), -Math_PI / 2.0); + cursor_rot = r.get_orthogonal_index(); + _update_cursor_transform(); + } break; + case MENU_OPTION_CURSOR_BACK_ROTATE_Y: { + Basis r; + if (input_action == INPUT_PASTE) { + r.set_orthogonal_index(paste_indicator.orientation); + r.rotate(Vector3(0, 1, 0), Math_PI / 2.0); + paste_indicator.orientation = r.get_orthogonal_index(); + _update_paste_indicator(); + break; + } + + r.set_orthogonal_index(cursor_rot); + r.rotate(Vector3(0, 1, 0), Math_PI / 2.0); + cursor_rot = r.get_orthogonal_index(); + _update_cursor_transform(); + } break; + case MENU_OPTION_CURSOR_BACK_ROTATE_X: { + Basis r; + if (input_action == INPUT_PASTE) { + r.set_orthogonal_index(paste_indicator.orientation); + r.rotate(Vector3(1, 0, 0), Math_PI / 2.0); + paste_indicator.orientation = r.get_orthogonal_index(); + _update_paste_indicator(); + break; + } + + r.set_orthogonal_index(cursor_rot); + r.rotate(Vector3(1, 0, 0), Math_PI / 2.0); + cursor_rot = r.get_orthogonal_index(); + _update_cursor_transform(); + } break; + case MENU_OPTION_CURSOR_BACK_ROTATE_Z: { + Basis r; + if (input_action == INPUT_PASTE) { + r.set_orthogonal_index(paste_indicator.orientation); + r.rotate(Vector3(0, 0, 1), Math_PI / 2.0); + paste_indicator.orientation = r.get_orthogonal_index(); + _update_paste_indicator(); + break; + } + + r.set_orthogonal_index(cursor_rot); + r.rotate(Vector3(0, 0, 1), Math_PI / 2.0); + cursor_rot = r.get_orthogonal_index(); + _update_cursor_transform(); + } break; + case MENU_OPTION_CURSOR_CLEAR_ROTATION: { + if (input_action == INPUT_PASTE) { + paste_indicator.orientation = 0; + _update_paste_indicator(); + break; + } + + cursor_rot = 0; + _update_cursor_transform(); + } break; + + case MENU_OPTION_PASTE_SELECTS: { + int idx = options->get_popup()->get_item_index(MENU_OPTION_PASTE_SELECTS); + options->get_popup()->set_item_checked(idx, !options->get_popup()->is_item_checked(idx)); + } break; + + case MENU_OPTION_SELECTION_DUPLICATE: + case MENU_OPTION_SELECTION_CUT: { + if (!(selection.active && input_action == INPUT_NONE)) { + break; + } + + _set_clipboard_data(); + + if (p_option == MENU_OPTION_SELECTION_CUT) { + _delete_selection(); + } + + input_action = INPUT_PASTE; + paste_indicator.click = selection.begin; + paste_indicator.current = selection.begin; + paste_indicator.begin = selection.begin; + paste_indicator.end = selection.end; + paste_indicator.orientation = 0; + _update_paste_indicator(); + } break; + case MENU_OPTION_SELECTION_CLEAR: { + if (!selection.active) { + break; + } + + _delete_selection(); + + } break; + case MENU_OPTION_SELECTION_FILL: { + if (!selection.active) { + return; + } + + _fill_selection(); + + } break; + case MENU_OPTION_GRIDMAP_SETTINGS: { + settings_dialog->popup_centered(settings_vbc->get_combined_minimum_size() + Size2(50, 50) * EDSCALE); + } break; + } +} + +void GridMapEditor::_update_cursor_transform() { + cursor_transform = Transform3D(); + cursor_transform.origin = cursor_origin; + cursor_transform.basis.set_orthogonal_index(cursor_rot); + cursor_transform.basis *= node->get_cell_scale(); + cursor_transform = node->get_global_transform() * cursor_transform; + + if (selected_palette >= 0) { + if (node && !node->get_mesh_library().is_null()) { + cursor_transform *= node->get_mesh_library()->get_item_mesh_transform(selected_palette); + } + } + + if (cursor_instance.is_valid()) { + RenderingServer::get_singleton()->instance_set_transform(cursor_instance, cursor_transform); + RenderingServer::get_singleton()->instance_set_visible(cursor_instance, cursor_visible); + } +} + +void GridMapEditor::_update_selection_transform() { + Transform3D xf_zero; + xf_zero.basis.set_zero(); + + if (!selection.active) { + RenderingServer::get_singleton()->instance_set_transform(selection_instance, xf_zero); + for (int i = 0; i < 3; i++) { + RenderingServer::get_singleton()->instance_set_transform(selection_level_instance[i], xf_zero); + } + return; + } + + Transform3D xf; + xf.scale((Vector3(1, 1, 1) + (selection.end - selection.begin)) * node->get_cell_size()); + xf.origin = selection.begin * node->get_cell_size(); + + RenderingServer::get_singleton()->instance_set_transform(selection_instance, node->get_global_transform() * xf); + + for (int i = 0; i < 3; i++) { + if (i != edit_axis || (edit_floor[edit_axis] < selection.begin[edit_axis]) || (edit_floor[edit_axis] > selection.end[edit_axis] + 1)) { + RenderingServer::get_singleton()->instance_set_transform(selection_level_instance[i], xf_zero); + } else { + Vector3 scale = (selection.end - selection.begin + Vector3(1, 1, 1)); + scale[edit_axis] = 1.0; + Vector3 position = selection.begin; + position[edit_axis] = edit_floor[edit_axis]; + + scale *= node->get_cell_size(); + position *= node->get_cell_size(); + + Transform3D xf2; + xf2.basis.scale(scale); + xf2.origin = position; + + RenderingServer::get_singleton()->instance_set_transform(selection_level_instance[i], xf2); + } + } +} + +void GridMapEditor::_validate_selection() { + if (!selection.active) { + return; + } + selection.begin = selection.click; + selection.end = selection.current; + + if (selection.begin.x > selection.end.x) { + SWAP(selection.begin.x, selection.end.x); + } + if (selection.begin.y > selection.end.y) { + SWAP(selection.begin.y, selection.end.y); + } + if (selection.begin.z > selection.end.z) { + SWAP(selection.begin.z, selection.end.z); + } + + _update_selection_transform(); +} + +void GridMapEditor::_set_selection(bool p_active, const Vector3 &p_begin, const Vector3 &p_end) { + selection.active = p_active; + selection.begin = p_begin; + selection.end = p_end; + selection.click = p_begin; + selection.current = p_end; + + if (is_visible_in_tree()) { + _update_selection_transform(); + } + + options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_CLEAR), !selection.active); + options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_CUT), !selection.active); + options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_DUPLICATE), !selection.active); + options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_FILL), !selection.active); +} + +bool GridMapEditor::do_input_action(Camera3D *p_camera, const Point2 &p_point, bool p_click) { + if (!spatial_editor) { + return false; + } + + if (selected_palette < 0 && input_action != INPUT_PICK && input_action != INPUT_SELECT && input_action != INPUT_PASTE) { + return false; + } + Ref mesh_library = node->get_mesh_library(); + if (mesh_library.is_null()) { + return false; + } + if (input_action != INPUT_PICK && input_action != INPUT_SELECT && input_action != INPUT_PASTE && !mesh_library->has_item(selected_palette)) { + return false; + } + + Camera3D *camera = p_camera; + Vector3 from = camera->project_ray_origin(p_point); + Vector3 normal = camera->project_ray_normal(p_point); + Transform3D local_xform = node->get_global_transform().affine_inverse(); + Vector planes = camera->get_frustum(); + from = local_xform.xform(from); + normal = local_xform.basis.xform(normal).normalized(); + + Plane p; + p.normal[edit_axis] = 1.0; + p.d = edit_floor[edit_axis] * node->get_cell_size()[edit_axis]; + + Vector3 inters; + if (!p.intersects_segment(from, from + normal * settings_pick_distance->get_value(), &inters)) { + return false; + } + + // Make sure the intersection is inside the frustum planes, to avoid + // Painting on invisible regions. + for (int i = 0; i < planes.size(); i++) { + Plane fp = local_xform.xform(planes[i]); + if (fp.is_point_over(inters)) { + return false; + } + } + + int cell[3]; + Vector3 cell_size = node->get_cell_size(); + + for (int i = 0; i < 3; i++) { + if (i == edit_axis) { + cell[i] = edit_floor[i]; + } else { + cell[i] = inters[i] / cell_size[i]; + if (inters[i] < 0) { + cell[i] -= 1; // Compensate negative. + } + grid_ofs[i] = cell[i] * cell_size[i]; + } + } + + RS::get_singleton()->instance_set_transform(grid_instance[edit_axis], node->get_global_transform() * edit_grid_xform); + + if (cursor_instance.is_valid()) { + cursor_origin = (Vector3(cell[0], cell[1], cell[2]) + Vector3(0.5 * node->get_center_x(), 0.5 * node->get_center_y(), 0.5 * node->get_center_z())) * node->get_cell_size(); + cursor_visible = true; + + if (input_action == INPUT_SELECT || input_action == INPUT_PASTE) { + cursor_visible = false; + } + + _update_cursor_transform(); + } + + if (input_action == INPUT_PASTE) { + paste_indicator.current = Vector3i(cell[0], cell[1], cell[2]); + _update_paste_indicator(); + + } else if (input_action == INPUT_SELECT) { + selection.current = Vector3i(cell[0], cell[1], cell[2]); + if (p_click) { + selection.click = selection.current; + } + selection.active = true; + _validate_selection(); + + return true; + } else if (input_action == INPUT_PICK) { + int item = node->get_cell_item(Vector3i(cell[0], cell[1], cell[2])); + if (item >= 0) { + selected_palette = item; + mesh_library_palette->set_current(item); + update_palette(); + _update_cursor_instance(); + } + return true; + } + + if (input_action == INPUT_PAINT) { + SetItem si; + si.position = Vector3i(cell[0], cell[1], cell[2]); + si.new_value = selected_palette; + si.new_orientation = cursor_rot; + si.old_value = node->get_cell_item(Vector3i(cell[0], cell[1], cell[2])); + si.old_orientation = node->get_cell_item_orientation(Vector3i(cell[0], cell[1], cell[2])); + set_items.push_back(si); + node->set_cell_item(Vector3i(cell[0], cell[1], cell[2]), selected_palette, cursor_rot); + return true; + } else if (input_action == INPUT_ERASE) { + SetItem si; + si.position = Vector3i(cell[0], cell[1], cell[2]); + si.new_value = -1; + si.new_orientation = 0; + si.old_value = node->get_cell_item(Vector3i(cell[0], cell[1], cell[2])); + si.old_orientation = node->get_cell_item_orientation(Vector3i(cell[0], cell[1], cell[2])); + set_items.push_back(si); + node->set_cell_item(Vector3i(cell[0], cell[1], cell[2]), -1); + return true; + } + + return false; +} + +void GridMapEditor::_delete_selection() { + if (!selection.active) { + return; + } + + undo_redo->create_action(TTR("GridMap Delete Selection")); + for (int i = selection.begin.x; i <= selection.end.x; i++) { + for (int j = selection.begin.y; j <= selection.end.y; j++) { + for (int k = selection.begin.z; k <= selection.end.z; k++) { + Vector3i selected = Vector3i(i, j, k); + undo_redo->add_do_method(node, "set_cell_item", selected, GridMap::INVALID_CELL_ITEM); + undo_redo->add_undo_method(node, "set_cell_item", selected, node->get_cell_item(selected), node->get_cell_item_orientation(selected)); + } + } + } + undo_redo->add_do_method(this, "_set_selection", !selection.active, selection.begin, selection.end); + undo_redo->add_undo_method(this, "_set_selection", selection.active, selection.begin, selection.end); + undo_redo->commit_action(); +} + +void GridMapEditor::_fill_selection() { + if (!selection.active) { + return; + } + + undo_redo->create_action(TTR("GridMap Fill Selection")); + for (int i = selection.begin.x; i <= selection.end.x; i++) { + for (int j = selection.begin.y; j <= selection.end.y; j++) { + for (int k = selection.begin.z; k <= selection.end.z; k++) { + Vector3i selected = Vector3i(i, j, k); + undo_redo->add_do_method(node, "set_cell_item", selected, selected_palette, cursor_rot); + undo_redo->add_undo_method(node, "set_cell_item", selected, node->get_cell_item(selected), node->get_cell_item_orientation(selected)); + } + } + } + undo_redo->add_do_method(this, "_set_selection", !selection.active, selection.begin, selection.end); + undo_redo->add_undo_method(this, "_set_selection", selection.active, selection.begin, selection.end); + undo_redo->commit_action(); +} + +void GridMapEditor::_clear_clipboard_data() { + for (const ClipboardItem &E : clipboard_items) { + RenderingServer::get_singleton()->free(E.instance); + } + + clipboard_items.clear(); +} + +void GridMapEditor::_set_clipboard_data() { + _clear_clipboard_data(); + + Ref meshLibrary = node->get_mesh_library(); + + for (int i = selection.begin.x; i <= selection.end.x; i++) { + for (int j = selection.begin.y; j <= selection.end.y; j++) { + for (int k = selection.begin.z; k <= selection.end.z; k++) { + Vector3i selected = Vector3i(i, j, k); + int itm = node->get_cell_item(selected); + if (itm == GridMap::INVALID_CELL_ITEM) { + continue; + } + + Ref mesh = meshLibrary->get_item_mesh(itm); + + ClipboardItem item; + item.cell_item = itm; + item.grid_offset = Vector3(selected) - selection.begin; + item.orientation = node->get_cell_item_orientation(selected); + item.instance = RenderingServer::get_singleton()->instance_create2(mesh->get_rid(), get_tree()->get_root()->get_world_3d()->get_scenario()); + + clipboard_items.push_back(item); + } + } + } +} + +void GridMapEditor::_update_paste_indicator() { + if (input_action != INPUT_PASTE) { + Transform3D xf; + xf.basis.set_zero(); + RenderingServer::get_singleton()->instance_set_transform(paste_instance, xf); + return; + } + + Vector3 center = 0.5 * Vector3(real_t(node->get_center_x()), real_t(node->get_center_y()), real_t(node->get_center_z())); + Vector3 scale = (Vector3(1, 1, 1) + (paste_indicator.end - paste_indicator.begin)) * node->get_cell_size(); + Transform3D xf; + xf.scale(scale); + xf.origin = (paste_indicator.begin + (paste_indicator.current - paste_indicator.click) + center) * node->get_cell_size(); + Basis rot; + rot.set_orthogonal_index(paste_indicator.orientation); + xf.basis = rot * xf.basis; + xf.translate((-center * node->get_cell_size()) / scale); + + RenderingServer::get_singleton()->instance_set_transform(paste_instance, node->get_global_transform() * xf); + + for (const ClipboardItem &item : clipboard_items) { + xf = Transform3D(); + xf.origin = (paste_indicator.begin + (paste_indicator.current - paste_indicator.click) + center) * node->get_cell_size(); + xf.basis = rot * xf.basis; + xf.translate(item.grid_offset * node->get_cell_size()); + + Basis item_rot; + item_rot.set_orthogonal_index(item.orientation); + xf.basis = item_rot * xf.basis * node->get_cell_scale(); + + RenderingServer::get_singleton()->instance_set_transform(item.instance, node->get_global_transform() * xf); + } +} + +void GridMapEditor::_do_paste() { + int idx = options->get_popup()->get_item_index(MENU_OPTION_PASTE_SELECTS); + bool reselect = options->get_popup()->is_item_checked(idx); + + Basis rot; + rot.set_orthogonal_index(paste_indicator.orientation); + + Vector3 ofs = paste_indicator.current - paste_indicator.click; + undo_redo->create_action(TTR("GridMap Paste Selection")); + + for (const ClipboardItem &item : clipboard_items) { + Vector3 position = rot.xform(item.grid_offset) + paste_indicator.begin + ofs; + + Basis orm; + orm.set_orthogonal_index(item.orientation); + orm = rot * orm; + + undo_redo->add_do_method(node, "set_cell_item", position, item.cell_item, orm.get_orthogonal_index()); + undo_redo->add_undo_method(node, "set_cell_item", position, node->get_cell_item(position), node->get_cell_item_orientation(position)); + } + + if (reselect) { + undo_redo->add_do_method(this, "_set_selection", true, paste_indicator.begin + ofs, paste_indicator.end + ofs); + undo_redo->add_undo_method(this, "_set_selection", selection.active, selection.begin, selection.end); + } + + undo_redo->commit_action(); + + _clear_clipboard_data(); +} + +EditorPlugin::AfterGUIInput GridMapEditor::forward_spatial_input_event(Camera3D *p_camera, const Ref &p_event) { + if (!node) { + return EditorPlugin::AFTER_GUI_INPUT_PASS; + } + + Ref mb = p_event; + + if (mb.is_valid()) { + if (mb->get_button_index() == MouseButton::WHEEL_UP && (mb->is_command_pressed() || mb->is_shift_pressed())) { + if (mb->is_pressed()) { + floor->set_value(floor->get_value() + mb->get_factor()); + } + + return EditorPlugin::AFTER_GUI_INPUT_STOP; // Eaten. + } else if (mb->get_button_index() == MouseButton::WHEEL_DOWN && (mb->is_command_pressed() || mb->is_shift_pressed())) { + if (mb->is_pressed()) { + floor->set_value(floor->get_value() - mb->get_factor()); + } + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + + if (mb->is_pressed()) { + Node3DEditorViewport::NavigationScheme nav_scheme = (Node3DEditorViewport::NavigationScheme)EditorSettings::get_singleton()->get("editors/3d/navigation/navigation_scheme").operator int(); + if ((nav_scheme == Node3DEditorViewport::NAVIGATION_MAYA || nav_scheme == Node3DEditorViewport::NAVIGATION_MODO) && mb->is_alt_pressed()) { + input_action = INPUT_NONE; + } else if (mb->get_button_index() == MouseButton::LEFT) { + bool can_edit = (node && node->get_mesh_library().is_valid()); + if (input_action == INPUT_PASTE) { + _do_paste(); + input_action = INPUT_NONE; + _update_paste_indicator(); + } else if (mb->is_shift_pressed() && can_edit) { + input_action = INPUT_SELECT; + last_selection = selection; + } else if (mb->is_command_pressed() && can_edit) { + input_action = INPUT_PICK; + } else { + input_action = INPUT_PAINT; + set_items.clear(); + } + } else if (mb->get_button_index() == MouseButton::RIGHT) { + if (input_action == INPUT_PASTE) { + _clear_clipboard_data(); + input_action = INPUT_NONE; + _update_paste_indicator(); + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } else if (selection.active) { + _set_selection(false); + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } else { + input_action = INPUT_ERASE; + set_items.clear(); + } + } else { + return EditorPlugin::AFTER_GUI_INPUT_PASS; + } + + if (do_input_action(p_camera, Point2(mb->get_position().x, mb->get_position().y), true)) { + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + return EditorPlugin::AFTER_GUI_INPUT_PASS; + } else { + if ((mb->get_button_index() == MouseButton::RIGHT && input_action == INPUT_ERASE) || (mb->get_button_index() == MouseButton::LEFT && input_action == INPUT_PAINT)) { + if (set_items.size()) { + undo_redo->create_action(TTR("GridMap Paint")); + for (const SetItem &si : set_items) { + undo_redo->add_do_method(node, "set_cell_item", si.position, si.new_value, si.new_orientation); + } + for (List::Element *E = set_items.back(); E; E = E->prev()) { + const SetItem &si = E->get(); + undo_redo->add_undo_method(node, "set_cell_item", si.position, si.old_value, si.old_orientation); + } + + undo_redo->commit_action(); + } + set_items.clear(); + input_action = INPUT_NONE; + + if (set_items.size() > 0) { + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + return EditorPlugin::AFTER_GUI_INPUT_PASS; + } + + if (mb->get_button_index() == MouseButton::LEFT && input_action == INPUT_SELECT) { + undo_redo->create_action(TTR("GridMap Selection")); + undo_redo->add_do_method(this, "_set_selection", selection.active, selection.begin, selection.end); + undo_redo->add_undo_method(this, "_set_selection", last_selection.active, last_selection.begin, last_selection.end); + undo_redo->commit_action(); + } + + if (mb->get_button_index() == MouseButton::LEFT && input_action != INPUT_NONE) { + set_items.clear(); + input_action = INPUT_NONE; + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + if (mb->get_button_index() == MouseButton::RIGHT && (input_action == INPUT_ERASE || input_action == INPUT_PASTE)) { + input_action = INPUT_NONE; + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + } + } + + Ref mm = p_event; + + if (mm.is_valid()) { + if (do_input_action(p_camera, mm->get_position(), false)) { + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + return EditorPlugin::AFTER_GUI_INPUT_PASS; + } + + Ref k = p_event; + + if (k.is_valid()) { + if (k->is_pressed()) { + if (k->get_keycode() == Key::ESCAPE) { + if (input_action == INPUT_PASTE) { + _clear_clipboard_data(); + input_action = INPUT_NONE; + _update_paste_indicator(); + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } else if (selection.active) { + _set_selection(false); + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } else { + selected_palette = -1; + mesh_library_palette->deselect_all(); + update_palette(); + _update_cursor_instance(); + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + } + + if (k->is_shift_pressed() && selection.active && input_action != INPUT_PASTE) { + if (k->get_keycode() == (Key)options->get_popup()->get_item_accelerator(options->get_popup()->get_item_index(MENU_OPTION_PREV_LEVEL))) { + selection.click[edit_axis]--; + _validate_selection(); + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + if (k->get_keycode() == (Key)options->get_popup()->get_item_accelerator(options->get_popup()->get_item_index(MENU_OPTION_NEXT_LEVEL))) { + selection.click[edit_axis]++; + _validate_selection(); + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + } + } + } + + Ref pan_gesture = p_event; + if (pan_gesture.is_valid()) { + if (pan_gesture->is_alt_pressed() && (pan_gesture->is_command_pressed() || pan_gesture->is_shift_pressed())) { + const real_t delta = pan_gesture->get_delta().y * 0.5; + accumulated_floor_delta += delta; + int step = 0; + if (ABS(accumulated_floor_delta) > 1.0) { + step = SIGN(accumulated_floor_delta); + accumulated_floor_delta -= step; + } + if (step) { + floor->set_value(floor->get_value() + step); + } + return EditorPlugin::AFTER_GUI_INPUT_STOP; + } + } + accumulated_floor_delta = 0.0; + + return EditorPlugin::AFTER_GUI_INPUT_PASS; +} + +struct _CGMEItemSort { + String name; + int id = 0; + _FORCE_INLINE_ bool operator<(const _CGMEItemSort &r_it) const { return name < r_it.name; } +}; + +void GridMapEditor::_set_display_mode(int p_mode) { + if (display_mode == p_mode) { + return; + } + + if (p_mode == DISPLAY_LIST) { + mode_list->set_pressed(true); + mode_thumbnail->set_pressed(false); + } else if (p_mode == DISPLAY_THUMBNAIL) { + mode_list->set_pressed(false); + mode_thumbnail->set_pressed(true); + } + + display_mode = p_mode; + + update_palette(); +} + +void GridMapEditor::_text_changed(const String &p_text) { + update_palette(); +} + +void GridMapEditor::_sbox_input(const Ref &p_ie) { + const Ref k = p_ie; + + if (k.is_valid() && (k->get_keycode() == Key::UP || k->get_keycode() == Key::DOWN || k->get_keycode() == Key::PAGEUP || k->get_keycode() == Key::PAGEDOWN)) { + // Forward the key input to the ItemList so it can be scrolled + mesh_library_palette->gui_input(k); + search_box->accept_event(); + } +} + +void GridMapEditor::_mesh_library_palette_input(const Ref &p_ie) { + const Ref mb = p_ie; + + // Zoom in/out using Ctrl + mouse wheel + if (mb.is_valid() && mb->is_pressed() && mb->is_command_pressed()) { + if (mb->is_pressed() && mb->get_button_index() == MouseButton::WHEEL_UP) { + size_slider->set_value(size_slider->get_value() + 0.2); + } + + if (mb->is_pressed() && mb->get_button_index() == MouseButton::WHEEL_DOWN) { + size_slider->set_value(size_slider->get_value() - 0.2); + } + } +} + +void GridMapEditor::_icon_size_changed(float p_value) { + mesh_library_palette->set_icon_scale(p_value); + update_palette(); +} + +void GridMapEditor::update_palette() { + int selected = mesh_library_palette->get_current(); + + float min_size = EDITOR_GET("editors/grid_map/preview_size"); + min_size *= EDSCALE; + + mesh_library_palette->clear(); + if (display_mode == DISPLAY_THUMBNAIL) { + mesh_library_palette->set_max_columns(0); + mesh_library_palette->set_icon_mode(ItemList::ICON_MODE_TOP); + mesh_library_palette->set_fixed_column_width(min_size * MAX(size_slider->get_value(), 1.5)); + } else if (display_mode == DISPLAY_LIST) { + mesh_library_palette->set_max_columns(1); + mesh_library_palette->set_icon_mode(ItemList::ICON_MODE_LEFT); + mesh_library_palette->set_fixed_column_width(0); + } + + mesh_library_palette->set_fixed_icon_size(Size2(min_size, min_size)); + mesh_library_palette->set_max_text_lines(2); + + Ref mesh_library = node->get_mesh_library(); + + if (mesh_library.is_null()) { + last_mesh_library = nullptr; + search_box->set_text(""); + search_box->set_editable(false); + info_message->show(); + return; + } + + search_box->set_editable(true); + info_message->hide(); + + Vector ids; + ids = mesh_library->get_item_list(); + + List<_CGMEItemSort> il; + for (int i = 0; i < ids.size(); i++) { + _CGMEItemSort is; + is.id = ids[i]; + is.name = mesh_library->get_item_name(ids[i]); + il.push_back(is); + } + il.sort(); + + String filter = search_box->get_text().strip_edges(); + + int item = 0; + + for (_CGMEItemSort &E : il) { + int id = E.id; + String name = mesh_library->get_item_name(id); + Ref preview = mesh_library->get_item_preview(id); + + if (name.is_empty()) { + name = "#" + itos(id); + } + + if (!filter.is_empty() && !filter.is_subsequence_ofn(name)) { + continue; + } + + mesh_library_palette->add_item(""); + if (!preview.is_null()) { + mesh_library_palette->set_item_icon(item, preview); + mesh_library_palette->set_item_tooltip(item, name); + } + mesh_library_palette->set_item_text(item, name); + mesh_library_palette->set_item_metadata(item, id); + + item++; + } + + if (selected != -1 && mesh_library_palette->get_item_count() > 0) { + mesh_library_palette->select(selected); + } + + last_mesh_library = mesh_library.operator->(); +} + +void GridMapEditor::edit(GridMap *p_gridmap) { + if (!p_gridmap && node) { + node->disconnect("cell_size_changed", callable_mp(this, &GridMapEditor::_draw_grids)); + } + + node = p_gridmap; + + input_action = INPUT_NONE; + selection.active = false; + _update_selection_transform(); + _update_paste_indicator(); + + spatial_editor = Object::cast_to(EditorNode::get_singleton()->get_editor_plugin_screen()); + + if (!node) { + set_process(false); + for (int i = 0; i < 3; i++) { + RenderingServer::get_singleton()->instance_set_visible(grid_instance[i], false); + } + + if (cursor_instance.is_valid()) { + RenderingServer::get_singleton()->instance_set_visible(cursor_instance, false); + } + + return; + } + + update_palette(); + + set_process(true); + + _draw_grids(node->get_cell_size()); + update_grid(); + + node->connect("cell_size_changed", callable_mp(this, &GridMapEditor::_draw_grids)); +} + +void GridMapEditor::update_grid() { + grid_xform.origin.x -= 1; // Force update in hackish way. + + grid_ofs[edit_axis] = edit_floor[edit_axis] * node->get_cell_size()[edit_axis]; + + edit_grid_xform.origin = grid_ofs; + edit_grid_xform.basis = Basis(); + + for (int i = 0; i < 3; i++) { + RenderingServer::get_singleton()->instance_set_visible(grid_instance[i], i == edit_axis); + } + + updating = true; + floor->set_value(edit_floor[edit_axis]); + updating = false; +} + +void GridMapEditor::_draw_grids(const Vector3 &cell_size) { + Vector3 edited_floor = node->has_meta("_editor_floor_") ? node->get_meta("_editor_floor_") : Variant(); + + for (int i = 0; i < 3; i++) { + RS::get_singleton()->mesh_clear(grid[i]); + edit_floor[i] = edited_floor[i]; + } + + Vector grid_points[3]; + Vector grid_colors[3]; + + for (int i = 0; i < 3; i++) { + Vector3 axis; + axis[i] = 1; + Vector3 axis_n1; + axis_n1[(i + 1) % 3] = cell_size[(i + 1) % 3]; + Vector3 axis_n2; + axis_n2[(i + 2) % 3] = cell_size[(i + 2) % 3]; + + for (int j = -GRID_CURSOR_SIZE; j <= GRID_CURSOR_SIZE; j++) { + for (int k = -GRID_CURSOR_SIZE; k <= GRID_CURSOR_SIZE; k++) { + Vector3 p = axis_n1 * j + axis_n2 * k; + float trans = Math::pow(MAX(0, 1.0 - (Vector2(j, k).length() / GRID_CURSOR_SIZE)), 2); + + Vector3 pj = axis_n1 * (j + 1) + axis_n2 * k; + float transj = Math::pow(MAX(0, 1.0 - (Vector2(j + 1, k).length() / GRID_CURSOR_SIZE)), 2); + + Vector3 pk = axis_n1 * j + axis_n2 * (k + 1); + float transk = Math::pow(MAX(0, 1.0 - (Vector2(j, k + 1).length() / GRID_CURSOR_SIZE)), 2); + + grid_points[i].push_back(p); + grid_points[i].push_back(pk); + grid_colors[i].push_back(Color(1, 1, 1, trans)); + grid_colors[i].push_back(Color(1, 1, 1, transk)); + + grid_points[i].push_back(p); + grid_points[i].push_back(pj); + grid_colors[i].push_back(Color(1, 1, 1, trans)); + grid_colors[i].push_back(Color(1, 1, 1, transj)); + } + } + + Array d; + d.resize(RS::ARRAY_MAX); + d[RS::ARRAY_VERTEX] = grid_points[i]; + d[RS::ARRAY_COLOR] = grid_colors[i]; + RenderingServer::get_singleton()->mesh_add_surface_from_arrays(grid[i], RenderingServer::PRIMITIVE_LINES, d); + RenderingServer::get_singleton()->mesh_surface_set_material(grid[i], 0, indicator_mat->get_rid()); + } +} + +void GridMapEditor::_update_theme() { + options->set_icon(get_theme_icon(SNAME("GridMap"), SNAME("EditorIcons"))); + search_box->set_right_icon(get_theme_icon(SNAME("Search"), SNAME("EditorIcons"))); + mode_thumbnail->set_icon(get_theme_icon(SNAME("FileThumbnail"), SNAME("EditorIcons"))); + mode_list->set_icon(get_theme_icon(SNAME("FileList"), SNAME("EditorIcons"))); +} + +void GridMapEditor::_notification(int p_what) { + switch (p_what) { + case NOTIFICATION_ENTER_TREE: { + get_tree()->connect("node_removed", callable_mp(this, &GridMapEditor::_node_removed)); + mesh_library_palette->connect("item_selected", callable_mp(this, &GridMapEditor::_item_selected_cbk)); + for (int i = 0; i < 3; i++) { + grid[i] = RS::get_singleton()->mesh_create(); + grid_instance[i] = RS::get_singleton()->instance_create2(grid[i], get_tree()->get_root()->get_world_3d()->get_scenario()); + RenderingServer::get_singleton()->instance_set_layer_mask(grid_instance[i], 1 << Node3DEditorViewport::MISC_TOOL_LAYER); + selection_level_instance[i] = RenderingServer::get_singleton()->instance_create2(selection_level_mesh[i], get_tree()->get_root()->get_world_3d()->get_scenario()); + RenderingServer::get_singleton()->instance_set_layer_mask(selection_level_instance[i], 1 << Node3DEditorViewport::MISC_TOOL_LAYER); + } + + selection_instance = RenderingServer::get_singleton()->instance_create2(selection_mesh, get_tree()->get_root()->get_world_3d()->get_scenario()); + RenderingServer::get_singleton()->instance_set_layer_mask(selection_instance, 1 << Node3DEditorViewport::MISC_TOOL_LAYER); + paste_instance = RenderingServer::get_singleton()->instance_create2(paste_mesh, get_tree()->get_root()->get_world_3d()->get_scenario()); + RenderingServer::get_singleton()->instance_set_layer_mask(paste_instance, 1 << Node3DEditorViewport::MISC_TOOL_LAYER); + + _update_selection_transform(); + _update_paste_indicator(); + _update_theme(); + } break; + + case NOTIFICATION_EXIT_TREE: { + get_tree()->disconnect("node_removed", callable_mp(this, &GridMapEditor::_node_removed)); + _clear_clipboard_data(); + + for (int i = 0; i < 3; i++) { + RS::get_singleton()->free(grid_instance[i]); + RS::get_singleton()->free(grid[i]); + grid_instance[i] = RID(); + grid[i] = RID(); + RenderingServer::get_singleton()->free(selection_level_instance[i]); + } + + RenderingServer::get_singleton()->free(selection_instance); + RenderingServer::get_singleton()->free(paste_instance); + selection_instance = RID(); + paste_instance = RID(); + } break; + + case NOTIFICATION_PROCESS: { + if (!node) { + return; + } + + Transform3D xf = node->get_global_transform(); + + if (xf != grid_xform) { + for (int i = 0; i < 3; i++) { + RS::get_singleton()->instance_set_transform(grid_instance[i], xf * edit_grid_xform); + } + grid_xform = xf; + } + Ref cgmt = node->get_mesh_library(); + if (cgmt.operator->() != last_mesh_library) { + update_palette(); + } + } break; + + case NOTIFICATION_THEME_CHANGED: { + _update_theme(); + } break; + + case NOTIFICATION_APPLICATION_FOCUS_OUT: { + if (input_action == INPUT_PAINT) { + // Simulate mouse released event to stop drawing when editor focus exists. + Ref release; + release.instantiate(); + release->set_button_index(MouseButton::LEFT); + forward_spatial_input_event(nullptr, release); + } + } break; + } +} + +void GridMapEditor::_update_cursor_instance() { + if (!node) { + return; + } + + if (cursor_instance.is_valid()) { + RenderingServer::get_singleton()->free(cursor_instance); + } + cursor_instance = RID(); + + if (selected_palette >= 0) { + if (node && !node->get_mesh_library().is_null()) { + Ref mesh = node->get_mesh_library()->get_item_mesh(selected_palette); + if (!mesh.is_null() && mesh->get_rid().is_valid()) { + cursor_instance = RenderingServer::get_singleton()->instance_create2(mesh->get_rid(), get_tree()->get_root()->get_world_3d()->get_scenario()); + RenderingServer::get_singleton()->instance_set_transform(cursor_instance, cursor_transform); + } + } + } +} + +void GridMapEditor::_item_selected_cbk(int idx) { + selected_palette = mesh_library_palette->get_item_metadata(idx); + + _update_cursor_instance(); +} + +void GridMapEditor::_floor_changed(float p_value) { + if (updating) { + return; + } + + edit_floor[edit_axis] = p_value; + node->set_meta("_editor_floor_", Vector3(edit_floor[0], edit_floor[1], edit_floor[2])); + update_grid(); + _update_selection_transform(); +} + +void GridMapEditor::_floor_mouse_exited() { + floor->get_line_edit()->release_focus(); +} + +void GridMapEditor::_bind_methods() { + ClassDB::bind_method("_configure", &GridMapEditor::_configure); + ClassDB::bind_method("_set_selection", &GridMapEditor::_set_selection); +} + +GridMapEditor::GridMapEditor() { + undo_redo = EditorNode::get_singleton()->get_undo_redo(); + + int mw = EDITOR_DEF("editors/grid_map/palette_min_width", 230); + Control *ec = memnew(Control); + ec->set_custom_minimum_size(Size2(mw, 0) * EDSCALE); + add_child(ec); + + spatial_editor_hb = memnew(HBoxContainer); + spatial_editor_hb->set_h_size_flags(SIZE_EXPAND_FILL); + spatial_editor_hb->set_alignment(BoxContainer::ALIGNMENT_END); + Node3DEditor::get_singleton()->add_control_to_menu_panel(spatial_editor_hb); + + spin_box_label = memnew(Label); + spin_box_label->set_text(TTR("Floor:")); + spatial_editor_hb->add_child(spin_box_label); + + floor = memnew(SpinBox); + floor->set_min(-32767); + floor->set_max(32767); + floor->set_step(1); + floor->get_line_edit()->add_theme_constant_override("minimum_character_width", 16); + + spatial_editor_hb->add_child(floor); + floor->connect("value_changed", callable_mp(this, &GridMapEditor::_floor_changed)); + floor->connect("mouse_exited", callable_mp(this, &GridMapEditor::_floor_mouse_exited)); + floor->get_line_edit()->connect("mouse_exited", callable_mp(this, &GridMapEditor::_floor_mouse_exited)); + + spatial_editor_hb->add_child(memnew(VSeparator)); + + options = memnew(MenuButton); + spatial_editor_hb->add_child(options); + spatial_editor_hb->hide(); + + options->set_text(TTR("Grid Map")); + options->get_popup()->add_item(TTR("Previous Floor"), MENU_OPTION_PREV_LEVEL, Key::Q); + options->get_popup()->add_item(TTR("Next Floor"), MENU_OPTION_NEXT_LEVEL, Key::E); + options->get_popup()->add_separator(); + options->get_popup()->add_radio_check_item(TTR("Edit X Axis"), MENU_OPTION_X_AXIS, Key::Z); + options->get_popup()->add_radio_check_item(TTR("Edit Y Axis"), MENU_OPTION_Y_AXIS, Key::X); + options->get_popup()->add_radio_check_item(TTR("Edit Z Axis"), MENU_OPTION_Z_AXIS, Key::C); + options->get_popup()->set_item_checked(options->get_popup()->get_item_index(MENU_OPTION_Y_AXIS), true); + options->get_popup()->add_separator(); + options->get_popup()->add_item(TTR("Cursor Rotate X"), MENU_OPTION_CURSOR_ROTATE_X, Key::A); + options->get_popup()->add_item(TTR("Cursor Rotate Y"), MENU_OPTION_CURSOR_ROTATE_Y, Key::S); + options->get_popup()->add_item(TTR("Cursor Rotate Z"), MENU_OPTION_CURSOR_ROTATE_Z, Key::D); + options->get_popup()->add_item(TTR("Cursor Back Rotate X"), MENU_OPTION_CURSOR_BACK_ROTATE_X, KeyModifierMask::SHIFT + Key::A); + options->get_popup()->add_item(TTR("Cursor Back Rotate Y"), MENU_OPTION_CURSOR_BACK_ROTATE_Y, KeyModifierMask::SHIFT + Key::S); + options->get_popup()->add_item(TTR("Cursor Back Rotate Z"), MENU_OPTION_CURSOR_BACK_ROTATE_Z, KeyModifierMask::SHIFT + Key::D); + options->get_popup()->add_item(TTR("Cursor Clear Rotation"), MENU_OPTION_CURSOR_CLEAR_ROTATION, Key::W); + options->get_popup()->add_separator(); + options->get_popup()->add_check_item(TTR("Paste Selects"), MENU_OPTION_PASTE_SELECTS); + options->get_popup()->add_separator(); + options->get_popup()->add_item(TTR("Duplicate Selection"), MENU_OPTION_SELECTION_DUPLICATE, KeyModifierMask::CTRL + Key::C); + options->get_popup()->add_item(TTR("Cut Selection"), MENU_OPTION_SELECTION_CUT, KeyModifierMask::CTRL + Key::X); + options->get_popup()->add_item(TTR("Clear Selection"), MENU_OPTION_SELECTION_CLEAR, Key::KEY_DELETE); + options->get_popup()->add_item(TTR("Fill Selection"), MENU_OPTION_SELECTION_FILL, KeyModifierMask::CTRL + Key::F); + + options->get_popup()->add_separator(); + options->get_popup()->add_item(TTR("Settings..."), MENU_OPTION_GRIDMAP_SETTINGS); + + settings_dialog = memnew(ConfirmationDialog); + settings_dialog->set_title(TTR("GridMap Settings")); + add_child(settings_dialog); + settings_vbc = memnew(VBoxContainer); + settings_vbc->set_custom_minimum_size(Size2(200, 0) * EDSCALE); + settings_dialog->add_child(settings_vbc); + + settings_pick_distance = memnew(SpinBox); + settings_pick_distance->set_max(10000.0f); + settings_pick_distance->set_min(500.0f); + settings_pick_distance->set_step(1.0f); + settings_pick_distance->set_value(EDITOR_GET("editors/grid_map/pick_distance")); + settings_vbc->add_margin_child(TTR("Pick Distance:"), settings_pick_distance); + + options->get_popup()->connect("id_pressed", callable_mp(this, &GridMapEditor::_menu_option)); + + HBoxContainer *hb = memnew(HBoxContainer); + add_child(hb); + hb->set_h_size_flags(SIZE_EXPAND_FILL); + + search_box = memnew(LineEdit); + search_box->set_h_size_flags(SIZE_EXPAND_FILL); + search_box->set_placeholder(TTR("Filter meshes")); + hb->add_child(search_box); + search_box->connect("text_changed", callable_mp(this, &GridMapEditor::_text_changed)); + search_box->connect("gui_input", callable_mp(this, &GridMapEditor::_sbox_input)); + + mode_thumbnail = memnew(Button); + mode_thumbnail->set_flat(true); + mode_thumbnail->set_toggle_mode(true); + mode_thumbnail->set_pressed(true); + hb->add_child(mode_thumbnail); + mode_thumbnail->connect("pressed", callable_mp(this, &GridMapEditor::_set_display_mode), varray(DISPLAY_THUMBNAIL)); + + mode_list = memnew(Button); + mode_list->set_flat(true); + mode_list->set_toggle_mode(true); + mode_list->set_pressed(false); + hb->add_child(mode_list); + mode_list->connect("pressed", callable_mp(this, &GridMapEditor::_set_display_mode), varray(DISPLAY_LIST)); + + size_slider = memnew(HSlider); + size_slider->set_h_size_flags(SIZE_EXPAND_FILL); + size_slider->set_min(0.2f); + size_slider->set_max(4.0f); + size_slider->set_step(0.1f); + size_slider->set_value(1.0f); + size_slider->connect("value_changed", callable_mp(this, &GridMapEditor::_icon_size_changed)); + add_child(size_slider); + + EDITOR_DEF("editors/grid_map/preview_size", 64); + + mesh_library_palette = memnew(ItemList); + add_child(mesh_library_palette); + mesh_library_palette->set_v_size_flags(SIZE_EXPAND_FILL); + mesh_library_palette->connect("gui_input", callable_mp(this, &GridMapEditor::_mesh_library_palette_input)); + + info_message = memnew(Label); + info_message->set_text(TTR("Give a MeshLibrary resource to this GridMap to use its meshes.")); + info_message->set_vertical_alignment(VERTICAL_ALIGNMENT_CENTER); + info_message->set_horizontal_alignment(HORIZONTAL_ALIGNMENT_CENTER); + info_message->set_autowrap_mode(Label::AUTOWRAP_WORD_SMART); + info_message->set_custom_minimum_size(Size2(100 * EDSCALE, 0)); + info_message->set_anchors_and_offsets_preset(PRESET_WIDE, PRESET_MODE_KEEP_SIZE, 8 * EDSCALE); + mesh_library_palette->add_child(info_message); + + edit_axis = Vector3::AXIS_Y; + edit_floor[0] = -1; + edit_floor[1] = -1; + edit_floor[2] = -1; + + selection_mesh = RenderingServer::get_singleton()->mesh_create(); + paste_mesh = RenderingServer::get_singleton()->mesh_create(); + + { + // Selection mesh create. + + Vector lines; + Vector triangles; + Vector square[3]; + + for (int i = 0; i < 6; i++) { + Vector3 face_points[4]; + + for (int j = 0; j < 4; j++) { + float v[3]; + v[0] = 1.0; + v[1] = 1 - 2 * ((j >> 1) & 1); + v[2] = v[1] * (1 - 2 * (j & 1)); + + for (int k = 0; k < 3; k++) { + if (i < 3) { + face_points[j][(i + k) % 3] = v[k]; + } else { + face_points[3 - j][(i + k) % 3] = -v[k]; + } + } + } + + triangles.push_back(face_points[0] * 0.5 + Vector3(0.5, 0.5, 0.5)); + triangles.push_back(face_points[1] * 0.5 + Vector3(0.5, 0.5, 0.5)); + triangles.push_back(face_points[2] * 0.5 + Vector3(0.5, 0.5, 0.5)); + + triangles.push_back(face_points[2] * 0.5 + Vector3(0.5, 0.5, 0.5)); + triangles.push_back(face_points[3] * 0.5 + Vector3(0.5, 0.5, 0.5)); + triangles.push_back(face_points[0] * 0.5 + Vector3(0.5, 0.5, 0.5)); + } + + for (int i = 0; i < 12; i++) { + AABB base(Vector3(0, 0, 0), Vector3(1, 1, 1)); + Vector3 a, b; + base.get_edge(i, a, b); + lines.push_back(a); + lines.push_back(b); + } + + for (int i = 0; i < 3; i++) { + Vector3 points[4]; + for (int j = 0; j < 4; j++) { + static const bool orderx[4] = { false, true, true, false }; + static const bool ordery[4] = { false, false, true, true }; + + Vector3 sp; + if (orderx[j]) { + sp[(i + 1) % 3] = 1.0; + } + if (ordery[j]) { + sp[(i + 2) % 3] = 1.0; + } + + points[j] = sp; + } + + for (int j = 0; j < 4; j++) { + Vector3 ofs; + ofs[i] += 0.01; + square[i].push_back(points[j] - ofs); + square[i].push_back(points[(j + 1) % 4] - ofs); + square[i].push_back(points[j] + ofs); + square[i].push_back(points[(j + 1) % 4] + ofs); + } + } + + Array d; + d.resize(RS::ARRAY_MAX); + + inner_mat.instantiate(); + inner_mat->set_albedo(Color(0.7, 0.7, 1.0, 0.2)); + inner_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); + inner_mat->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA); + + d[RS::ARRAY_VERTEX] = triangles; + RenderingServer::get_singleton()->mesh_add_surface_from_arrays(selection_mesh, RS::PRIMITIVE_TRIANGLES, d); + RenderingServer::get_singleton()->mesh_surface_set_material(selection_mesh, 0, inner_mat->get_rid()); + + outer_mat.instantiate(); + outer_mat->set_albedo(Color(0.7, 0.7, 1.0, 0.8)); + outer_mat->set_on_top_of_alpha(); + + outer_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); + outer_mat->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA); + + selection_floor_mat.instantiate(); + selection_floor_mat->set_albedo(Color(0.80, 0.80, 1.0, 1)); + selection_floor_mat->set_on_top_of_alpha(); + selection_floor_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); + + d[RS::ARRAY_VERTEX] = lines; + RenderingServer::get_singleton()->mesh_add_surface_from_arrays(selection_mesh, RS::PRIMITIVE_LINES, d); + RenderingServer::get_singleton()->mesh_surface_set_material(selection_mesh, 1, outer_mat->get_rid()); + + d[RS::ARRAY_VERTEX] = triangles; + RenderingServer::get_singleton()->mesh_add_surface_from_arrays(paste_mesh, RS::PRIMITIVE_TRIANGLES, d); + RenderingServer::get_singleton()->mesh_surface_set_material(paste_mesh, 0, inner_mat->get_rid()); + + d[RS::ARRAY_VERTEX] = lines; + RenderingServer::get_singleton()->mesh_add_surface_from_arrays(paste_mesh, RS::PRIMITIVE_LINES, d); + RenderingServer::get_singleton()->mesh_surface_set_material(paste_mesh, 1, outer_mat->get_rid()); + + for (int i = 0; i < 3; i++) { + d[RS::ARRAY_VERTEX] = square[i]; + selection_level_mesh[i] = RS::get_singleton()->mesh_create(); + RenderingServer::get_singleton()->mesh_add_surface_from_arrays(selection_level_mesh[i], RS::PRIMITIVE_LINES, d); + RenderingServer::get_singleton()->mesh_surface_set_material(selection_level_mesh[i], 0, selection_floor_mat->get_rid()); + } + } + + _set_selection(false); + + indicator_mat.instantiate(); + indicator_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); + indicator_mat->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA); + indicator_mat->set_flag(StandardMaterial3D::FLAG_SRGB_VERTEX_COLOR, true); + indicator_mat->set_flag(StandardMaterial3D::FLAG_ALBEDO_FROM_VERTEX_COLOR, true); + indicator_mat->set_albedo(Color(0.8, 0.5, 0.1)); +} + +GridMapEditor::~GridMapEditor() { + _clear_clipboard_data(); + + for (int i = 0; i < 3; i++) { + if (grid[i].is_valid()) { + RenderingServer::get_singleton()->free(grid[i]); + } + if (grid_instance[i].is_valid()) { + RenderingServer::get_singleton()->free(grid_instance[i]); + } + if (cursor_instance.is_valid()) { + RenderingServer::get_singleton()->free(cursor_instance); + } + if (selection_level_instance[i].is_valid()) { + RenderingServer::get_singleton()->free(selection_level_instance[i]); + } + if (selection_level_mesh[i].is_valid()) { + RenderingServer::get_singleton()->free(selection_level_mesh[i]); + } + } + + RenderingServer::get_singleton()->free(selection_mesh); + if (selection_instance.is_valid()) { + RenderingServer::get_singleton()->free(selection_instance); + } + + RenderingServer::get_singleton()->free(paste_mesh); + if (paste_instance.is_valid()) { + RenderingServer::get_singleton()->free(paste_instance); + } +} + +void GridMapEditorPlugin::_notification(int p_what) { + switch (p_what) { + case EditorSettings::NOTIFICATION_EDITOR_SETTINGS_CHANGED: { + switch ((int)EditorSettings::get_singleton()->get("editors/grid_map/editor_side")) { + case 0: { // Left. + Node3DEditor::get_singleton()->move_control_to_left_panel(grid_map_editor); + } break; + case 1: { // Right. + Node3DEditor::get_singleton()->move_control_to_right_panel(grid_map_editor); + } break; + } + } break; + } +} + +void GridMapEditorPlugin::edit(Object *p_object) { + grid_map_editor->edit(Object::cast_to(p_object)); +} + +bool GridMapEditorPlugin::handles(Object *p_object) const { + return p_object->is_class("GridMap"); +} + +void GridMapEditorPlugin::make_visible(bool p_visible) { + if (p_visible) { + grid_map_editor->show(); + grid_map_editor->spatial_editor_hb->show(); + grid_map_editor->set_process(true); + } else { + grid_map_editor->spatial_editor_hb->hide(); + grid_map_editor->hide(); + grid_map_editor->edit(nullptr); + grid_map_editor->set_process(false); + } +} + +GridMapEditorPlugin::GridMapEditorPlugin() { + EDITOR_DEF("editors/grid_map/editor_side", 1); + EditorSettings::get_singleton()->add_property_hint(PropertyInfo(Variant::INT, "editors/grid_map/editor_side", PROPERTY_HINT_ENUM, "Left,Right")); + + grid_map_editor = memnew(GridMapEditor); + switch ((int)EditorSettings::get_singleton()->get("editors/grid_map/editor_side")) { + case 0: { // Left. + Node3DEditor::get_singleton()->add_control_to_left_panel(grid_map_editor); + } break; + case 1: { // Right. + Node3DEditor::get_singleton()->add_control_to_right_panel(grid_map_editor); + } break; + } + grid_map_editor->hide(); +} + +GridMapEditorPlugin::~GridMapEditorPlugin() { +} + +#endif // TOOLS_ENABLED diff --git a/modules/gridmap/editor/grid_map_editor_plugin.h b/modules/gridmap/editor/grid_map_editor_plugin.h new file mode 100644 index 0000000000..adb0da6e2c --- /dev/null +++ b/modules/gridmap/editor/grid_map_editor_plugin.h @@ -0,0 +1,256 @@ +/*************************************************************************/ +/* grid_map_editor_plugin.h */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* https://godotengine.org */ +/*************************************************************************/ +/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */ +/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */ +/* */ +/* Permission is hereby granted, free of charge, to any person obtaining */ +/* a copy of this software and associated documentation files (the */ +/* "Software"), to deal in the Software without restriction, including */ +/* without limitation the rights to use, copy, modify, merge, publish, */ +/* distribute, sublicense, and/or sell copies of the Software, and to */ +/* permit persons to whom the Software is furnished to do so, subject to */ +/* the following conditions: */ +/* */ +/* The above copyright notice and this permission notice shall be */ +/* included in all copies or substantial portions of the Software. */ +/* */ +/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ +/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ +/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/ +/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ +/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ +/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ +/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ +/*************************************************************************/ + +#ifndef GRID_MAP_EDITOR_PLUGIN_H +#define GRID_MAP_EDITOR_PLUGIN_H + +#ifdef TOOLS_ENABLED + +#include "../grid_map.h" +#include "editor/editor_plugin.h" +#include "scene/gui/item_list.h" +#include "scene/gui/slider.h" +#include "scene/gui/spin_box.h" + +class Node3DEditorPlugin; + +class GridMapEditor : public VBoxContainer { + GDCLASS(GridMapEditor, VBoxContainer); + + enum { + GRID_CURSOR_SIZE = 50 + }; + + enum InputAction { + INPUT_NONE, + INPUT_PAINT, + INPUT_ERASE, + INPUT_PICK, + INPUT_SELECT, + INPUT_PASTE, + }; + + enum DisplayMode { + DISPLAY_THUMBNAIL, + DISPLAY_LIST + }; + + UndoRedo *undo_redo; + InputAction input_action = INPUT_NONE; + Panel *panel; + MenuButton *options; + SpinBox *floor; + double accumulated_floor_delta = 0.0; + Button *mode_thumbnail; + Button *mode_list; + LineEdit *search_box; + HSlider *size_slider; + HBoxContainer *spatial_editor_hb; + ConfirmationDialog *settings_dialog; + VBoxContainer *settings_vbc; + SpinBox *settings_pick_distance; + Label *spin_box_label; + + struct SetItem { + Vector3i position; + int new_value = 0; + int new_orientation = 0; + int old_value = 0; + int old_orientation = 0; + }; + + List set_items; + + GridMap *node = nullptr; + MeshLibrary *last_mesh_library; + + Transform3D grid_xform; + Transform3D edit_grid_xform; + Vector3::Axis edit_axis; + int edit_floor[3]; + Vector3 grid_ofs; + + RID grid[3]; + RID grid_instance[3]; + RID cursor_instance; + RID selection_mesh; + RID selection_instance; + RID selection_level_mesh[3]; + RID selection_level_instance[3]; + RID paste_mesh; + RID paste_instance; + + struct ClipboardItem { + int cell_item = 0; + Vector3 grid_offset; + int orientation = 0; + RID instance; + }; + + List clipboard_items; + + Ref indicator_mat; + Ref inner_mat; + Ref outer_mat; + Ref selection_floor_mat; + + bool updating = false; + + struct Selection { + Vector3 click; + Vector3 current; + Vector3 begin; + Vector3 end; + bool active = false; + } selection; + Selection last_selection; + + struct PasteIndicator { + Vector3 click; + Vector3 current; + Vector3 begin; + Vector3 end; + int orientation = 0; + }; + PasteIndicator paste_indicator; + + bool cursor_visible = false; + Transform3D cursor_transform; + + Vector3 cursor_origin; + + int display_mode = DISPLAY_THUMBNAIL; + int selected_palette = -1; + int cursor_rot = 0; + + enum Menu { + MENU_OPTION_NEXT_LEVEL, + MENU_OPTION_PREV_LEVEL, + MENU_OPTION_LOCK_VIEW, + MENU_OPTION_X_AXIS, + MENU_OPTION_Y_AXIS, + MENU_OPTION_Z_AXIS, + MENU_OPTION_CURSOR_ROTATE_Y, + MENU_OPTION_CURSOR_ROTATE_X, + MENU_OPTION_CURSOR_ROTATE_Z, + MENU_OPTION_CURSOR_BACK_ROTATE_Y, + MENU_OPTION_CURSOR_BACK_ROTATE_X, + MENU_OPTION_CURSOR_BACK_ROTATE_Z, + MENU_OPTION_CURSOR_CLEAR_ROTATION, + MENU_OPTION_PASTE_SELECTS, + MENU_OPTION_SELECTION_DUPLICATE, + MENU_OPTION_SELECTION_CUT, + MENU_OPTION_SELECTION_CLEAR, + MENU_OPTION_SELECTION_FILL, + MENU_OPTION_GRIDMAP_SETTINGS + + }; + + Node3DEditorPlugin *spatial_editor; + + struct AreaDisplay { + RID mesh; + RID instance; + }; + + ItemList *mesh_library_palette; + Label *info_message; + + void update_grid(); // Change which and where the grid is displayed + void _draw_grids(const Vector3 &cell_size); + void _configure(); + void _menu_option(int); + void update_palette(); + void _set_display_mode(int p_mode); + void _item_selected_cbk(int idx); + void _update_cursor_transform(); + void _update_cursor_instance(); + void _update_theme(); + + void _text_changed(const String &p_text); + void _sbox_input(const Ref &p_ie); + void _mesh_library_palette_input(const Ref &p_ie); + + void _icon_size_changed(float p_value); + + void _clear_clipboard_data(); + void _set_clipboard_data(); + void _update_paste_indicator(); + void _do_paste(); + void _update_selection_transform(); + void _validate_selection(); + void _set_selection(bool p_active, const Vector3 &p_begin = Vector3(), const Vector3 &p_end = Vector3()); + + void _floor_changed(float p_value); + void _floor_mouse_exited(); + + void _delete_selection(); + void _fill_selection(); + + bool do_input_action(Camera3D *p_camera, const Point2 &p_point, bool p_click); + + friend class GridMapEditorPlugin; + +protected: + void _notification(int p_what); + void _node_removed(Node *p_node); + static void _bind_methods(); + +public: + EditorPlugin::AfterGUIInput forward_spatial_input_event(Camera3D *p_camera, const Ref &p_event); + + void edit(GridMap *p_gridmap); + GridMapEditor(); + ~GridMapEditor(); +}; + +class GridMapEditorPlugin : public EditorPlugin { + GDCLASS(GridMapEditorPlugin, EditorPlugin); + + GridMapEditor *grid_map_editor; + +protected: + void _notification(int p_what); + +public: + virtual EditorPlugin::AfterGUIInput forward_spatial_gui_input(Camera3D *p_camera, const Ref &p_event) override { return grid_map_editor->forward_spatial_input_event(p_camera, p_event); } + virtual String get_name() const override { return "GridMap"; } + bool has_main_screen() const override { return false; } + virtual void edit(Object *p_object) override; + virtual bool handles(Object *p_object) const override; + virtual void make_visible(bool p_visible) override; + + GridMapEditorPlugin(); + ~GridMapEditorPlugin(); +}; + +#endif // TOOLS_ENABLED + +#endif // GRID_MAP_EDITOR_PLUGIN_H diff --git a/modules/gridmap/grid_map_editor_plugin.cpp b/modules/gridmap/grid_map_editor_plugin.cpp deleted file mode 100644 index e8e096d52b..0000000000 --- a/modules/gridmap/grid_map_editor_plugin.cpp +++ /dev/null @@ -1,1481 +0,0 @@ -/*************************************************************************/ -/* grid_map_editor_plugin.cpp */ -/*************************************************************************/ -/* This file is part of: */ -/* GODOT ENGINE */ -/* https://godotengine.org */ -/*************************************************************************/ -/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */ -/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */ -/* */ -/* Permission is hereby granted, free of charge, to any person obtaining */ -/* a copy of this software and associated documentation files (the */ -/* "Software"), to deal in the Software without restriction, including */ -/* without limitation the rights to use, copy, modify, merge, publish, */ -/* distribute, sublicense, and/or sell copies of the Software, and to */ -/* permit persons to whom the Software is furnished to do so, subject to */ -/* the following conditions: */ -/* */ -/* The above copyright notice and this permission notice shall be */ -/* included in all copies or substantial portions of the Software. */ -/* */ -/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ -/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ -/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/ -/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ -/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ -/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ -/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ -/*************************************************************************/ - -#include "grid_map_editor_plugin.h" -#include "core/input/input.h" -#include "editor/editor_node.h" -#include "editor/editor_scale.h" -#include "editor/editor_settings.h" -#include "editor/plugins/node_3d_editor_plugin.h" -#include "scene/3d/camera_3d.h" - -#include "core/os/keyboard.h" -#include "scene/main/window.h" - -void GridMapEditor::_node_removed(Node *p_node) { - if (p_node == node) { - node = nullptr; - } -} - -void GridMapEditor::_configure() { - if (!node) { - return; - } - - update_grid(); -} - -void GridMapEditor::_menu_option(int p_option) { - switch (p_option) { - case MENU_OPTION_PREV_LEVEL: { - floor->set_value(floor->get_value() - 1); - } break; - - case MENU_OPTION_NEXT_LEVEL: { - floor->set_value(floor->get_value() + 1); - } break; - - case MENU_OPTION_X_AXIS: - case MENU_OPTION_Y_AXIS: - case MENU_OPTION_Z_AXIS: { - int new_axis = p_option - MENU_OPTION_X_AXIS; - for (int i = 0; i < 3; i++) { - int idx = options->get_popup()->get_item_index(MENU_OPTION_X_AXIS + i); - options->get_popup()->set_item_checked(idx, i == new_axis); - } - - if (edit_axis != new_axis) { - int item1 = options->get_popup()->get_item_index(MENU_OPTION_NEXT_LEVEL); - int item2 = options->get_popup()->get_item_index(MENU_OPTION_PREV_LEVEL); - if (edit_axis == Vector3::AXIS_Y) { - options->get_popup()->set_item_text(item1, TTR("Next Plane")); - options->get_popup()->set_item_text(item2, TTR("Previous Plane")); - spin_box_label->set_text(TTR("Plane:")); - } else if (new_axis == Vector3::AXIS_Y) { - options->get_popup()->set_item_text(item1, TTR("Next Floor")); - options->get_popup()->set_item_text(item2, TTR("Previous Floor")); - spin_box_label->set_text(TTR("Floor:")); - } - } - edit_axis = Vector3::Axis(new_axis); - update_grid(); - - } break; - case MENU_OPTION_CURSOR_ROTATE_Y: { - Basis r; - if (input_action == INPUT_PASTE) { - r.set_orthogonal_index(paste_indicator.orientation); - r.rotate(Vector3(0, 1, 0), -Math_PI / 2.0); - paste_indicator.orientation = r.get_orthogonal_index(); - _update_paste_indicator(); - break; - } - - r.set_orthogonal_index(cursor_rot); - r.rotate(Vector3(0, 1, 0), -Math_PI / 2.0); - cursor_rot = r.get_orthogonal_index(); - _update_cursor_transform(); - } break; - case MENU_OPTION_CURSOR_ROTATE_X: { - Basis r; - if (input_action == INPUT_PASTE) { - r.set_orthogonal_index(paste_indicator.orientation); - r.rotate(Vector3(1, 0, 0), -Math_PI / 2.0); - paste_indicator.orientation = r.get_orthogonal_index(); - _update_paste_indicator(); - break; - } - - r.set_orthogonal_index(cursor_rot); - r.rotate(Vector3(1, 0, 0), -Math_PI / 2.0); - cursor_rot = r.get_orthogonal_index(); - _update_cursor_transform(); - } break; - case MENU_OPTION_CURSOR_ROTATE_Z: { - Basis r; - if (input_action == INPUT_PASTE) { - r.set_orthogonal_index(paste_indicator.orientation); - r.rotate(Vector3(0, 0, 1), -Math_PI / 2.0); - paste_indicator.orientation = r.get_orthogonal_index(); - _update_paste_indicator(); - break; - } - - r.set_orthogonal_index(cursor_rot); - r.rotate(Vector3(0, 0, 1), -Math_PI / 2.0); - cursor_rot = r.get_orthogonal_index(); - _update_cursor_transform(); - } break; - case MENU_OPTION_CURSOR_BACK_ROTATE_Y: { - Basis r; - if (input_action == INPUT_PASTE) { - r.set_orthogonal_index(paste_indicator.orientation); - r.rotate(Vector3(0, 1, 0), Math_PI / 2.0); - paste_indicator.orientation = r.get_orthogonal_index(); - _update_paste_indicator(); - break; - } - - r.set_orthogonal_index(cursor_rot); - r.rotate(Vector3(0, 1, 0), Math_PI / 2.0); - cursor_rot = r.get_orthogonal_index(); - _update_cursor_transform(); - } break; - case MENU_OPTION_CURSOR_BACK_ROTATE_X: { - Basis r; - if (input_action == INPUT_PASTE) { - r.set_orthogonal_index(paste_indicator.orientation); - r.rotate(Vector3(1, 0, 0), Math_PI / 2.0); - paste_indicator.orientation = r.get_orthogonal_index(); - _update_paste_indicator(); - break; - } - - r.set_orthogonal_index(cursor_rot); - r.rotate(Vector3(1, 0, 0), Math_PI / 2.0); - cursor_rot = r.get_orthogonal_index(); - _update_cursor_transform(); - } break; - case MENU_OPTION_CURSOR_BACK_ROTATE_Z: { - Basis r; - if (input_action == INPUT_PASTE) { - r.set_orthogonal_index(paste_indicator.orientation); - r.rotate(Vector3(0, 0, 1), Math_PI / 2.0); - paste_indicator.orientation = r.get_orthogonal_index(); - _update_paste_indicator(); - break; - } - - r.set_orthogonal_index(cursor_rot); - r.rotate(Vector3(0, 0, 1), Math_PI / 2.0); - cursor_rot = r.get_orthogonal_index(); - _update_cursor_transform(); - } break; - case MENU_OPTION_CURSOR_CLEAR_ROTATION: { - if (input_action == INPUT_PASTE) { - paste_indicator.orientation = 0; - _update_paste_indicator(); - break; - } - - cursor_rot = 0; - _update_cursor_transform(); - } break; - - case MENU_OPTION_PASTE_SELECTS: { - int idx = options->get_popup()->get_item_index(MENU_OPTION_PASTE_SELECTS); - options->get_popup()->set_item_checked(idx, !options->get_popup()->is_item_checked(idx)); - } break; - - case MENU_OPTION_SELECTION_DUPLICATE: - case MENU_OPTION_SELECTION_CUT: { - if (!(selection.active && input_action == INPUT_NONE)) { - break; - } - - _set_clipboard_data(); - - if (p_option == MENU_OPTION_SELECTION_CUT) { - _delete_selection(); - } - - input_action = INPUT_PASTE; - paste_indicator.click = selection.begin; - paste_indicator.current = selection.begin; - paste_indicator.begin = selection.begin; - paste_indicator.end = selection.end; - paste_indicator.orientation = 0; - _update_paste_indicator(); - } break; - case MENU_OPTION_SELECTION_CLEAR: { - if (!selection.active) { - break; - } - - _delete_selection(); - - } break; - case MENU_OPTION_SELECTION_FILL: { - if (!selection.active) { - return; - } - - _fill_selection(); - - } break; - case MENU_OPTION_GRIDMAP_SETTINGS: { - settings_dialog->popup_centered(settings_vbc->get_combined_minimum_size() + Size2(50, 50) * EDSCALE); - } break; - } -} - -void GridMapEditor::_update_cursor_transform() { - cursor_transform = Transform3D(); - cursor_transform.origin = cursor_origin; - cursor_transform.basis.set_orthogonal_index(cursor_rot); - cursor_transform.basis *= node->get_cell_scale(); - cursor_transform = node->get_global_transform() * cursor_transform; - - if (selected_palette >= 0) { - if (node && !node->get_mesh_library().is_null()) { - cursor_transform *= node->get_mesh_library()->get_item_mesh_transform(selected_palette); - } - } - - if (cursor_instance.is_valid()) { - RenderingServer::get_singleton()->instance_set_transform(cursor_instance, cursor_transform); - RenderingServer::get_singleton()->instance_set_visible(cursor_instance, cursor_visible); - } -} - -void GridMapEditor::_update_selection_transform() { - Transform3D xf_zero; - xf_zero.basis.set_zero(); - - if (!selection.active) { - RenderingServer::get_singleton()->instance_set_transform(selection_instance, xf_zero); - for (int i = 0; i < 3; i++) { - RenderingServer::get_singleton()->instance_set_transform(selection_level_instance[i], xf_zero); - } - return; - } - - Transform3D xf; - xf.scale((Vector3(1, 1, 1) + (selection.end - selection.begin)) * node->get_cell_size()); - xf.origin = selection.begin * node->get_cell_size(); - - RenderingServer::get_singleton()->instance_set_transform(selection_instance, node->get_global_transform() * xf); - - for (int i = 0; i < 3; i++) { - if (i != edit_axis || (edit_floor[edit_axis] < selection.begin[edit_axis]) || (edit_floor[edit_axis] > selection.end[edit_axis] + 1)) { - RenderingServer::get_singleton()->instance_set_transform(selection_level_instance[i], xf_zero); - } else { - Vector3 scale = (selection.end - selection.begin + Vector3(1, 1, 1)); - scale[edit_axis] = 1.0; - Vector3 position = selection.begin; - position[edit_axis] = edit_floor[edit_axis]; - - scale *= node->get_cell_size(); - position *= node->get_cell_size(); - - Transform3D xf2; - xf2.basis.scale(scale); - xf2.origin = position; - - RenderingServer::get_singleton()->instance_set_transform(selection_level_instance[i], xf2); - } - } -} - -void GridMapEditor::_validate_selection() { - if (!selection.active) { - return; - } - selection.begin = selection.click; - selection.end = selection.current; - - if (selection.begin.x > selection.end.x) { - SWAP(selection.begin.x, selection.end.x); - } - if (selection.begin.y > selection.end.y) { - SWAP(selection.begin.y, selection.end.y); - } - if (selection.begin.z > selection.end.z) { - SWAP(selection.begin.z, selection.end.z); - } - - _update_selection_transform(); -} - -void GridMapEditor::_set_selection(bool p_active, const Vector3 &p_begin, const Vector3 &p_end) { - selection.active = p_active; - selection.begin = p_begin; - selection.end = p_end; - selection.click = p_begin; - selection.current = p_end; - - if (is_visible_in_tree()) { - _update_selection_transform(); - } - - options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_CLEAR), !selection.active); - options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_CUT), !selection.active); - options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_DUPLICATE), !selection.active); - options->get_popup()->set_item_disabled(options->get_popup()->get_item_index(MENU_OPTION_SELECTION_FILL), !selection.active); -} - -bool GridMapEditor::do_input_action(Camera3D *p_camera, const Point2 &p_point, bool p_click) { - if (!spatial_editor) { - return false; - } - - if (selected_palette < 0 && input_action != INPUT_PICK && input_action != INPUT_SELECT && input_action != INPUT_PASTE) { - return false; - } - Ref mesh_library = node->get_mesh_library(); - if (mesh_library.is_null()) { - return false; - } - if (input_action != INPUT_PICK && input_action != INPUT_SELECT && input_action != INPUT_PASTE && !mesh_library->has_item(selected_palette)) { - return false; - } - - Camera3D *camera = p_camera; - Vector3 from = camera->project_ray_origin(p_point); - Vector3 normal = camera->project_ray_normal(p_point); - Transform3D local_xform = node->get_global_transform().affine_inverse(); - Vector planes = camera->get_frustum(); - from = local_xform.xform(from); - normal = local_xform.basis.xform(normal).normalized(); - - Plane p; - p.normal[edit_axis] = 1.0; - p.d = edit_floor[edit_axis] * node->get_cell_size()[edit_axis]; - - Vector3 inters; - if (!p.intersects_segment(from, from + normal * settings_pick_distance->get_value(), &inters)) { - return false; - } - - // Make sure the intersection is inside the frustum planes, to avoid - // Painting on invisible regions. - for (int i = 0; i < planes.size(); i++) { - Plane fp = local_xform.xform(planes[i]); - if (fp.is_point_over(inters)) { - return false; - } - } - - int cell[3]; - Vector3 cell_size = node->get_cell_size(); - - for (int i = 0; i < 3; i++) { - if (i == edit_axis) { - cell[i] = edit_floor[i]; - } else { - cell[i] = inters[i] / cell_size[i]; - if (inters[i] < 0) { - cell[i] -= 1; // Compensate negative. - } - grid_ofs[i] = cell[i] * cell_size[i]; - } - } - - RS::get_singleton()->instance_set_transform(grid_instance[edit_axis], node->get_global_transform() * edit_grid_xform); - - if (cursor_instance.is_valid()) { - cursor_origin = (Vector3(cell[0], cell[1], cell[2]) + Vector3(0.5 * node->get_center_x(), 0.5 * node->get_center_y(), 0.5 * node->get_center_z())) * node->get_cell_size(); - cursor_visible = true; - - if (input_action == INPUT_SELECT || input_action == INPUT_PASTE) { - cursor_visible = false; - } - - _update_cursor_transform(); - } - - if (input_action == INPUT_PASTE) { - paste_indicator.current = Vector3i(cell[0], cell[1], cell[2]); - _update_paste_indicator(); - - } else if (input_action == INPUT_SELECT) { - selection.current = Vector3i(cell[0], cell[1], cell[2]); - if (p_click) { - selection.click = selection.current; - } - selection.active = true; - _validate_selection(); - - return true; - } else if (input_action == INPUT_PICK) { - int item = node->get_cell_item(Vector3i(cell[0], cell[1], cell[2])); - if (item >= 0) { - selected_palette = item; - mesh_library_palette->set_current(item); - update_palette(); - _update_cursor_instance(); - } - return true; - } - - if (input_action == INPUT_PAINT) { - SetItem si; - si.position = Vector3i(cell[0], cell[1], cell[2]); - si.new_value = selected_palette; - si.new_orientation = cursor_rot; - si.old_value = node->get_cell_item(Vector3i(cell[0], cell[1], cell[2])); - si.old_orientation = node->get_cell_item_orientation(Vector3i(cell[0], cell[1], cell[2])); - set_items.push_back(si); - node->set_cell_item(Vector3i(cell[0], cell[1], cell[2]), selected_palette, cursor_rot); - return true; - } else if (input_action == INPUT_ERASE) { - SetItem si; - si.position = Vector3i(cell[0], cell[1], cell[2]); - si.new_value = -1; - si.new_orientation = 0; - si.old_value = node->get_cell_item(Vector3i(cell[0], cell[1], cell[2])); - si.old_orientation = node->get_cell_item_orientation(Vector3i(cell[0], cell[1], cell[2])); - set_items.push_back(si); - node->set_cell_item(Vector3i(cell[0], cell[1], cell[2]), -1); - return true; - } - - return false; -} - -void GridMapEditor::_delete_selection() { - if (!selection.active) { - return; - } - - undo_redo->create_action(TTR("GridMap Delete Selection")); - for (int i = selection.begin.x; i <= selection.end.x; i++) { - for (int j = selection.begin.y; j <= selection.end.y; j++) { - for (int k = selection.begin.z; k <= selection.end.z; k++) { - Vector3i selected = Vector3i(i, j, k); - undo_redo->add_do_method(node, "set_cell_item", selected, GridMap::INVALID_CELL_ITEM); - undo_redo->add_undo_method(node, "set_cell_item", selected, node->get_cell_item(selected), node->get_cell_item_orientation(selected)); - } - } - } - undo_redo->add_do_method(this, "_set_selection", !selection.active, selection.begin, selection.end); - undo_redo->add_undo_method(this, "_set_selection", selection.active, selection.begin, selection.end); - undo_redo->commit_action(); -} - -void GridMapEditor::_fill_selection() { - if (!selection.active) { - return; - } - - undo_redo->create_action(TTR("GridMap Fill Selection")); - for (int i = selection.begin.x; i <= selection.end.x; i++) { - for (int j = selection.begin.y; j <= selection.end.y; j++) { - for (int k = selection.begin.z; k <= selection.end.z; k++) { - Vector3i selected = Vector3i(i, j, k); - undo_redo->add_do_method(node, "set_cell_item", selected, selected_palette, cursor_rot); - undo_redo->add_undo_method(node, "set_cell_item", selected, node->get_cell_item(selected), node->get_cell_item_orientation(selected)); - } - } - } - undo_redo->add_do_method(this, "_set_selection", !selection.active, selection.begin, selection.end); - undo_redo->add_undo_method(this, "_set_selection", selection.active, selection.begin, selection.end); - undo_redo->commit_action(); -} - -void GridMapEditor::_clear_clipboard_data() { - for (const ClipboardItem &E : clipboard_items) { - RenderingServer::get_singleton()->free(E.instance); - } - - clipboard_items.clear(); -} - -void GridMapEditor::_set_clipboard_data() { - _clear_clipboard_data(); - - Ref meshLibrary = node->get_mesh_library(); - - for (int i = selection.begin.x; i <= selection.end.x; i++) { - for (int j = selection.begin.y; j <= selection.end.y; j++) { - for (int k = selection.begin.z; k <= selection.end.z; k++) { - Vector3i selected = Vector3i(i, j, k); - int itm = node->get_cell_item(selected); - if (itm == GridMap::INVALID_CELL_ITEM) { - continue; - } - - Ref mesh = meshLibrary->get_item_mesh(itm); - - ClipboardItem item; - item.cell_item = itm; - item.grid_offset = Vector3(selected) - selection.begin; - item.orientation = node->get_cell_item_orientation(selected); - item.instance = RenderingServer::get_singleton()->instance_create2(mesh->get_rid(), get_tree()->get_root()->get_world_3d()->get_scenario()); - - clipboard_items.push_back(item); - } - } - } -} - -void GridMapEditor::_update_paste_indicator() { - if (input_action != INPUT_PASTE) { - Transform3D xf; - xf.basis.set_zero(); - RenderingServer::get_singleton()->instance_set_transform(paste_instance, xf); - return; - } - - Vector3 center = 0.5 * Vector3(real_t(node->get_center_x()), real_t(node->get_center_y()), real_t(node->get_center_z())); - Vector3 scale = (Vector3(1, 1, 1) + (paste_indicator.end - paste_indicator.begin)) * node->get_cell_size(); - Transform3D xf; - xf.scale(scale); - xf.origin = (paste_indicator.begin + (paste_indicator.current - paste_indicator.click) + center) * node->get_cell_size(); - Basis rot; - rot.set_orthogonal_index(paste_indicator.orientation); - xf.basis = rot * xf.basis; - xf.translate((-center * node->get_cell_size()) / scale); - - RenderingServer::get_singleton()->instance_set_transform(paste_instance, node->get_global_transform() * xf); - - for (const ClipboardItem &item : clipboard_items) { - xf = Transform3D(); - xf.origin = (paste_indicator.begin + (paste_indicator.current - paste_indicator.click) + center) * node->get_cell_size(); - xf.basis = rot * xf.basis; - xf.translate(item.grid_offset * node->get_cell_size()); - - Basis item_rot; - item_rot.set_orthogonal_index(item.orientation); - xf.basis = item_rot * xf.basis * node->get_cell_scale(); - - RenderingServer::get_singleton()->instance_set_transform(item.instance, node->get_global_transform() * xf); - } -} - -void GridMapEditor::_do_paste() { - int idx = options->get_popup()->get_item_index(MENU_OPTION_PASTE_SELECTS); - bool reselect = options->get_popup()->is_item_checked(idx); - - Basis rot; - rot.set_orthogonal_index(paste_indicator.orientation); - - Vector3 ofs = paste_indicator.current - paste_indicator.click; - undo_redo->create_action(TTR("GridMap Paste Selection")); - - for (const ClipboardItem &item : clipboard_items) { - Vector3 position = rot.xform(item.grid_offset) + paste_indicator.begin + ofs; - - Basis orm; - orm.set_orthogonal_index(item.orientation); - orm = rot * orm; - - undo_redo->add_do_method(node, "set_cell_item", position, item.cell_item, orm.get_orthogonal_index()); - undo_redo->add_undo_method(node, "set_cell_item", position, node->get_cell_item(position), node->get_cell_item_orientation(position)); - } - - if (reselect) { - undo_redo->add_do_method(this, "_set_selection", true, paste_indicator.begin + ofs, paste_indicator.end + ofs); - undo_redo->add_undo_method(this, "_set_selection", selection.active, selection.begin, selection.end); - } - - undo_redo->commit_action(); - - _clear_clipboard_data(); -} - -EditorPlugin::AfterGUIInput GridMapEditor::forward_spatial_input_event(Camera3D *p_camera, const Ref &p_event) { - if (!node) { - return EditorPlugin::AFTER_GUI_INPUT_PASS; - } - - Ref mb = p_event; - - if (mb.is_valid()) { - if (mb->get_button_index() == MouseButton::WHEEL_UP && (mb->is_command_pressed() || mb->is_shift_pressed())) { - if (mb->is_pressed()) { - floor->set_value(floor->get_value() + mb->get_factor()); - } - - return EditorPlugin::AFTER_GUI_INPUT_STOP; // Eaten. - } else if (mb->get_button_index() == MouseButton::WHEEL_DOWN && (mb->is_command_pressed() || mb->is_shift_pressed())) { - if (mb->is_pressed()) { - floor->set_value(floor->get_value() - mb->get_factor()); - } - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - - if (mb->is_pressed()) { - Node3DEditorViewport::NavigationScheme nav_scheme = (Node3DEditorViewport::NavigationScheme)EditorSettings::get_singleton()->get("editors/3d/navigation/navigation_scheme").operator int(); - if ((nav_scheme == Node3DEditorViewport::NAVIGATION_MAYA || nav_scheme == Node3DEditorViewport::NAVIGATION_MODO) && mb->is_alt_pressed()) { - input_action = INPUT_NONE; - } else if (mb->get_button_index() == MouseButton::LEFT) { - bool can_edit = (node && node->get_mesh_library().is_valid()); - if (input_action == INPUT_PASTE) { - _do_paste(); - input_action = INPUT_NONE; - _update_paste_indicator(); - } else if (mb->is_shift_pressed() && can_edit) { - input_action = INPUT_SELECT; - last_selection = selection; - } else if (mb->is_command_pressed() && can_edit) { - input_action = INPUT_PICK; - } else { - input_action = INPUT_PAINT; - set_items.clear(); - } - } else if (mb->get_button_index() == MouseButton::RIGHT) { - if (input_action == INPUT_PASTE) { - _clear_clipboard_data(); - input_action = INPUT_NONE; - _update_paste_indicator(); - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } else if (selection.active) { - _set_selection(false); - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } else { - input_action = INPUT_ERASE; - set_items.clear(); - } - } else { - return EditorPlugin::AFTER_GUI_INPUT_PASS; - } - - if (do_input_action(p_camera, Point2(mb->get_position().x, mb->get_position().y), true)) { - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - return EditorPlugin::AFTER_GUI_INPUT_PASS; - } else { - if ((mb->get_button_index() == MouseButton::RIGHT && input_action == INPUT_ERASE) || (mb->get_button_index() == MouseButton::LEFT && input_action == INPUT_PAINT)) { - if (set_items.size()) { - undo_redo->create_action(TTR("GridMap Paint")); - for (const SetItem &si : set_items) { - undo_redo->add_do_method(node, "set_cell_item", si.position, si.new_value, si.new_orientation); - } - for (List::Element *E = set_items.back(); E; E = E->prev()) { - const SetItem &si = E->get(); - undo_redo->add_undo_method(node, "set_cell_item", si.position, si.old_value, si.old_orientation); - } - - undo_redo->commit_action(); - } - set_items.clear(); - input_action = INPUT_NONE; - - if (set_items.size() > 0) { - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - return EditorPlugin::AFTER_GUI_INPUT_PASS; - } - - if (mb->get_button_index() == MouseButton::LEFT && input_action == INPUT_SELECT) { - undo_redo->create_action(TTR("GridMap Selection")); - undo_redo->add_do_method(this, "_set_selection", selection.active, selection.begin, selection.end); - undo_redo->add_undo_method(this, "_set_selection", last_selection.active, last_selection.begin, last_selection.end); - undo_redo->commit_action(); - } - - if (mb->get_button_index() == MouseButton::LEFT && input_action != INPUT_NONE) { - set_items.clear(); - input_action = INPUT_NONE; - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - if (mb->get_button_index() == MouseButton::RIGHT && (input_action == INPUT_ERASE || input_action == INPUT_PASTE)) { - input_action = INPUT_NONE; - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - } - } - - Ref mm = p_event; - - if (mm.is_valid()) { - if (do_input_action(p_camera, mm->get_position(), false)) { - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - return EditorPlugin::AFTER_GUI_INPUT_PASS; - } - - Ref k = p_event; - - if (k.is_valid()) { - if (k->is_pressed()) { - if (k->get_keycode() == Key::ESCAPE) { - if (input_action == INPUT_PASTE) { - _clear_clipboard_data(); - input_action = INPUT_NONE; - _update_paste_indicator(); - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } else if (selection.active) { - _set_selection(false); - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } else { - selected_palette = -1; - mesh_library_palette->deselect_all(); - update_palette(); - _update_cursor_instance(); - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - } - - if (k->is_shift_pressed() && selection.active && input_action != INPUT_PASTE) { - if (k->get_keycode() == (Key)options->get_popup()->get_item_accelerator(options->get_popup()->get_item_index(MENU_OPTION_PREV_LEVEL))) { - selection.click[edit_axis]--; - _validate_selection(); - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - if (k->get_keycode() == (Key)options->get_popup()->get_item_accelerator(options->get_popup()->get_item_index(MENU_OPTION_NEXT_LEVEL))) { - selection.click[edit_axis]++; - _validate_selection(); - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - } - } - } - - Ref pan_gesture = p_event; - if (pan_gesture.is_valid()) { - if (pan_gesture->is_alt_pressed() && (pan_gesture->is_command_pressed() || pan_gesture->is_shift_pressed())) { - const real_t delta = pan_gesture->get_delta().y * 0.5; - accumulated_floor_delta += delta; - int step = 0; - if (ABS(accumulated_floor_delta) > 1.0) { - step = SIGN(accumulated_floor_delta); - accumulated_floor_delta -= step; - } - if (step) { - floor->set_value(floor->get_value() + step); - } - return EditorPlugin::AFTER_GUI_INPUT_STOP; - } - } - accumulated_floor_delta = 0.0; - - return EditorPlugin::AFTER_GUI_INPUT_PASS; -} - -struct _CGMEItemSort { - String name; - int id = 0; - _FORCE_INLINE_ bool operator<(const _CGMEItemSort &r_it) const { return name < r_it.name; } -}; - -void GridMapEditor::_set_display_mode(int p_mode) { - if (display_mode == p_mode) { - return; - } - - if (p_mode == DISPLAY_LIST) { - mode_list->set_pressed(true); - mode_thumbnail->set_pressed(false); - } else if (p_mode == DISPLAY_THUMBNAIL) { - mode_list->set_pressed(false); - mode_thumbnail->set_pressed(true); - } - - display_mode = p_mode; - - update_palette(); -} - -void GridMapEditor::_text_changed(const String &p_text) { - update_palette(); -} - -void GridMapEditor::_sbox_input(const Ref &p_ie) { - const Ref k = p_ie; - - if (k.is_valid() && (k->get_keycode() == Key::UP || k->get_keycode() == Key::DOWN || k->get_keycode() == Key::PAGEUP || k->get_keycode() == Key::PAGEDOWN)) { - // Forward the key input to the ItemList so it can be scrolled - mesh_library_palette->gui_input(k); - search_box->accept_event(); - } -} - -void GridMapEditor::_mesh_library_palette_input(const Ref &p_ie) { - const Ref mb = p_ie; - - // Zoom in/out using Ctrl + mouse wheel - if (mb.is_valid() && mb->is_pressed() && mb->is_command_pressed()) { - if (mb->is_pressed() && mb->get_button_index() == MouseButton::WHEEL_UP) { - size_slider->set_value(size_slider->get_value() + 0.2); - } - - if (mb->is_pressed() && mb->get_button_index() == MouseButton::WHEEL_DOWN) { - size_slider->set_value(size_slider->get_value() - 0.2); - } - } -} - -void GridMapEditor::_icon_size_changed(float p_value) { - mesh_library_palette->set_icon_scale(p_value); - update_palette(); -} - -void GridMapEditor::update_palette() { - int selected = mesh_library_palette->get_current(); - - float min_size = EDITOR_GET("editors/grid_map/preview_size"); - min_size *= EDSCALE; - - mesh_library_palette->clear(); - if (display_mode == DISPLAY_THUMBNAIL) { - mesh_library_palette->set_max_columns(0); - mesh_library_palette->set_icon_mode(ItemList::ICON_MODE_TOP); - mesh_library_palette->set_fixed_column_width(min_size * MAX(size_slider->get_value(), 1.5)); - } else if (display_mode == DISPLAY_LIST) { - mesh_library_palette->set_max_columns(1); - mesh_library_palette->set_icon_mode(ItemList::ICON_MODE_LEFT); - mesh_library_palette->set_fixed_column_width(0); - } - - mesh_library_palette->set_fixed_icon_size(Size2(min_size, min_size)); - mesh_library_palette->set_max_text_lines(2); - - Ref mesh_library = node->get_mesh_library(); - - if (mesh_library.is_null()) { - last_mesh_library = nullptr; - search_box->set_text(""); - search_box->set_editable(false); - info_message->show(); - return; - } - - search_box->set_editable(true); - info_message->hide(); - - Vector ids; - ids = mesh_library->get_item_list(); - - List<_CGMEItemSort> il; - for (int i = 0; i < ids.size(); i++) { - _CGMEItemSort is; - is.id = ids[i]; - is.name = mesh_library->get_item_name(ids[i]); - il.push_back(is); - } - il.sort(); - - String filter = search_box->get_text().strip_edges(); - - int item = 0; - - for (_CGMEItemSort &E : il) { - int id = E.id; - String name = mesh_library->get_item_name(id); - Ref preview = mesh_library->get_item_preview(id); - - if (name.is_empty()) { - name = "#" + itos(id); - } - - if (!filter.is_empty() && !filter.is_subsequence_ofn(name)) { - continue; - } - - mesh_library_palette->add_item(""); - if (!preview.is_null()) { - mesh_library_palette->set_item_icon(item, preview); - mesh_library_palette->set_item_tooltip(item, name); - } - mesh_library_palette->set_item_text(item, name); - mesh_library_palette->set_item_metadata(item, id); - - item++; - } - - if (selected != -1 && mesh_library_palette->get_item_count() > 0) { - mesh_library_palette->select(selected); - } - - last_mesh_library = mesh_library.operator->(); -} - -void GridMapEditor::edit(GridMap *p_gridmap) { - if (!p_gridmap && node) { - node->disconnect("cell_size_changed", callable_mp(this, &GridMapEditor::_draw_grids)); - } - - node = p_gridmap; - - input_action = INPUT_NONE; - selection.active = false; - _update_selection_transform(); - _update_paste_indicator(); - - spatial_editor = Object::cast_to(EditorNode::get_singleton()->get_editor_plugin_screen()); - - if (!node) { - set_process(false); - for (int i = 0; i < 3; i++) { - RenderingServer::get_singleton()->instance_set_visible(grid_instance[i], false); - } - - if (cursor_instance.is_valid()) { - RenderingServer::get_singleton()->instance_set_visible(cursor_instance, false); - } - - return; - } - - update_palette(); - - set_process(true); - - _draw_grids(node->get_cell_size()); - update_grid(); - - node->connect("cell_size_changed", callable_mp(this, &GridMapEditor::_draw_grids)); -} - -void GridMapEditor::update_grid() { - grid_xform.origin.x -= 1; // Force update in hackish way. - - grid_ofs[edit_axis] = edit_floor[edit_axis] * node->get_cell_size()[edit_axis]; - - edit_grid_xform.origin = grid_ofs; - edit_grid_xform.basis = Basis(); - - for (int i = 0; i < 3; i++) { - RenderingServer::get_singleton()->instance_set_visible(grid_instance[i], i == edit_axis); - } - - updating = true; - floor->set_value(edit_floor[edit_axis]); - updating = false; -} - -void GridMapEditor::_draw_grids(const Vector3 &cell_size) { - Vector3 edited_floor = node->has_meta("_editor_floor_") ? node->get_meta("_editor_floor_") : Variant(); - - for (int i = 0; i < 3; i++) { - RS::get_singleton()->mesh_clear(grid[i]); - edit_floor[i] = edited_floor[i]; - } - - Vector grid_points[3]; - Vector grid_colors[3]; - - for (int i = 0; i < 3; i++) { - Vector3 axis; - axis[i] = 1; - Vector3 axis_n1; - axis_n1[(i + 1) % 3] = cell_size[(i + 1) % 3]; - Vector3 axis_n2; - axis_n2[(i + 2) % 3] = cell_size[(i + 2) % 3]; - - for (int j = -GRID_CURSOR_SIZE; j <= GRID_CURSOR_SIZE; j++) { - for (int k = -GRID_CURSOR_SIZE; k <= GRID_CURSOR_SIZE; k++) { - Vector3 p = axis_n1 * j + axis_n2 * k; - float trans = Math::pow(MAX(0, 1.0 - (Vector2(j, k).length() / GRID_CURSOR_SIZE)), 2); - - Vector3 pj = axis_n1 * (j + 1) + axis_n2 * k; - float transj = Math::pow(MAX(0, 1.0 - (Vector2(j + 1, k).length() / GRID_CURSOR_SIZE)), 2); - - Vector3 pk = axis_n1 * j + axis_n2 * (k + 1); - float transk = Math::pow(MAX(0, 1.0 - (Vector2(j, k + 1).length() / GRID_CURSOR_SIZE)), 2); - - grid_points[i].push_back(p); - grid_points[i].push_back(pk); - grid_colors[i].push_back(Color(1, 1, 1, trans)); - grid_colors[i].push_back(Color(1, 1, 1, transk)); - - grid_points[i].push_back(p); - grid_points[i].push_back(pj); - grid_colors[i].push_back(Color(1, 1, 1, trans)); - grid_colors[i].push_back(Color(1, 1, 1, transj)); - } - } - - Array d; - d.resize(RS::ARRAY_MAX); - d[RS::ARRAY_VERTEX] = grid_points[i]; - d[RS::ARRAY_COLOR] = grid_colors[i]; - RenderingServer::get_singleton()->mesh_add_surface_from_arrays(grid[i], RenderingServer::PRIMITIVE_LINES, d); - RenderingServer::get_singleton()->mesh_surface_set_material(grid[i], 0, indicator_mat->get_rid()); - } -} - -void GridMapEditor::_update_theme() { - options->set_icon(get_theme_icon(SNAME("GridMap"), SNAME("EditorIcons"))); - search_box->set_right_icon(get_theme_icon(SNAME("Search"), SNAME("EditorIcons"))); - mode_thumbnail->set_icon(get_theme_icon(SNAME("FileThumbnail"), SNAME("EditorIcons"))); - mode_list->set_icon(get_theme_icon(SNAME("FileList"), SNAME("EditorIcons"))); -} - -void GridMapEditor::_notification(int p_what) { - switch (p_what) { - case NOTIFICATION_ENTER_TREE: { - get_tree()->connect("node_removed", callable_mp(this, &GridMapEditor::_node_removed)); - mesh_library_palette->connect("item_selected", callable_mp(this, &GridMapEditor::_item_selected_cbk)); - for (int i = 0; i < 3; i++) { - grid[i] = RS::get_singleton()->mesh_create(); - grid_instance[i] = RS::get_singleton()->instance_create2(grid[i], get_tree()->get_root()->get_world_3d()->get_scenario()); - RenderingServer::get_singleton()->instance_set_layer_mask(grid_instance[i], 1 << Node3DEditorViewport::MISC_TOOL_LAYER); - selection_level_instance[i] = RenderingServer::get_singleton()->instance_create2(selection_level_mesh[i], get_tree()->get_root()->get_world_3d()->get_scenario()); - RenderingServer::get_singleton()->instance_set_layer_mask(selection_level_instance[i], 1 << Node3DEditorViewport::MISC_TOOL_LAYER); - } - - selection_instance = RenderingServer::get_singleton()->instance_create2(selection_mesh, get_tree()->get_root()->get_world_3d()->get_scenario()); - RenderingServer::get_singleton()->instance_set_layer_mask(selection_instance, 1 << Node3DEditorViewport::MISC_TOOL_LAYER); - paste_instance = RenderingServer::get_singleton()->instance_create2(paste_mesh, get_tree()->get_root()->get_world_3d()->get_scenario()); - RenderingServer::get_singleton()->instance_set_layer_mask(paste_instance, 1 << Node3DEditorViewport::MISC_TOOL_LAYER); - - _update_selection_transform(); - _update_paste_indicator(); - _update_theme(); - } break; - - case NOTIFICATION_EXIT_TREE: { - get_tree()->disconnect("node_removed", callable_mp(this, &GridMapEditor::_node_removed)); - _clear_clipboard_data(); - - for (int i = 0; i < 3; i++) { - RS::get_singleton()->free(grid_instance[i]); - RS::get_singleton()->free(grid[i]); - grid_instance[i] = RID(); - grid[i] = RID(); - RenderingServer::get_singleton()->free(selection_level_instance[i]); - } - - RenderingServer::get_singleton()->free(selection_instance); - RenderingServer::get_singleton()->free(paste_instance); - selection_instance = RID(); - paste_instance = RID(); - } break; - - case NOTIFICATION_PROCESS: { - if (!node) { - return; - } - - Transform3D xf = node->get_global_transform(); - - if (xf != grid_xform) { - for (int i = 0; i < 3; i++) { - RS::get_singleton()->instance_set_transform(grid_instance[i], xf * edit_grid_xform); - } - grid_xform = xf; - } - Ref cgmt = node->get_mesh_library(); - if (cgmt.operator->() != last_mesh_library) { - update_palette(); - } - } break; - - case NOTIFICATION_THEME_CHANGED: { - _update_theme(); - } break; - - case NOTIFICATION_APPLICATION_FOCUS_OUT: { - if (input_action == INPUT_PAINT) { - // Simulate mouse released event to stop drawing when editor focus exists. - Ref release; - release.instantiate(); - release->set_button_index(MouseButton::LEFT); - forward_spatial_input_event(nullptr, release); - } - } break; - } -} - -void GridMapEditor::_update_cursor_instance() { - if (!node) { - return; - } - - if (cursor_instance.is_valid()) { - RenderingServer::get_singleton()->free(cursor_instance); - } - cursor_instance = RID(); - - if (selected_palette >= 0) { - if (node && !node->get_mesh_library().is_null()) { - Ref mesh = node->get_mesh_library()->get_item_mesh(selected_palette); - if (!mesh.is_null() && mesh->get_rid().is_valid()) { - cursor_instance = RenderingServer::get_singleton()->instance_create2(mesh->get_rid(), get_tree()->get_root()->get_world_3d()->get_scenario()); - RenderingServer::get_singleton()->instance_set_transform(cursor_instance, cursor_transform); - } - } - } -} - -void GridMapEditor::_item_selected_cbk(int idx) { - selected_palette = mesh_library_palette->get_item_metadata(idx); - - _update_cursor_instance(); -} - -void GridMapEditor::_floor_changed(float p_value) { - if (updating) { - return; - } - - edit_floor[edit_axis] = p_value; - node->set_meta("_editor_floor_", Vector3(edit_floor[0], edit_floor[1], edit_floor[2])); - update_grid(); - _update_selection_transform(); -} - -void GridMapEditor::_floor_mouse_exited() { - floor->get_line_edit()->release_focus(); -} - -void GridMapEditor::_bind_methods() { - ClassDB::bind_method("_configure", &GridMapEditor::_configure); - ClassDB::bind_method("_set_selection", &GridMapEditor::_set_selection); -} - -GridMapEditor::GridMapEditor() { - undo_redo = EditorNode::get_singleton()->get_undo_redo(); - - int mw = EDITOR_DEF("editors/grid_map/palette_min_width", 230); - Control *ec = memnew(Control); - ec->set_custom_minimum_size(Size2(mw, 0) * EDSCALE); - add_child(ec); - - spatial_editor_hb = memnew(HBoxContainer); - spatial_editor_hb->set_h_size_flags(SIZE_EXPAND_FILL); - spatial_editor_hb->set_alignment(BoxContainer::ALIGNMENT_END); - Node3DEditor::get_singleton()->add_control_to_menu_panel(spatial_editor_hb); - - spin_box_label = memnew(Label); - spin_box_label->set_text(TTR("Floor:")); - spatial_editor_hb->add_child(spin_box_label); - - floor = memnew(SpinBox); - floor->set_min(-32767); - floor->set_max(32767); - floor->set_step(1); - floor->get_line_edit()->add_theme_constant_override("minimum_character_width", 16); - - spatial_editor_hb->add_child(floor); - floor->connect("value_changed", callable_mp(this, &GridMapEditor::_floor_changed)); - floor->connect("mouse_exited", callable_mp(this, &GridMapEditor::_floor_mouse_exited)); - floor->get_line_edit()->connect("mouse_exited", callable_mp(this, &GridMapEditor::_floor_mouse_exited)); - - spatial_editor_hb->add_child(memnew(VSeparator)); - - options = memnew(MenuButton); - spatial_editor_hb->add_child(options); - spatial_editor_hb->hide(); - - options->set_text(TTR("Grid Map")); - options->get_popup()->add_item(TTR("Previous Floor"), MENU_OPTION_PREV_LEVEL, Key::Q); - options->get_popup()->add_item(TTR("Next Floor"), MENU_OPTION_NEXT_LEVEL, Key::E); - options->get_popup()->add_separator(); - options->get_popup()->add_radio_check_item(TTR("Edit X Axis"), MENU_OPTION_X_AXIS, Key::Z); - options->get_popup()->add_radio_check_item(TTR("Edit Y Axis"), MENU_OPTION_Y_AXIS, Key::X); - options->get_popup()->add_radio_check_item(TTR("Edit Z Axis"), MENU_OPTION_Z_AXIS, Key::C); - options->get_popup()->set_item_checked(options->get_popup()->get_item_index(MENU_OPTION_Y_AXIS), true); - options->get_popup()->add_separator(); - options->get_popup()->add_item(TTR("Cursor Rotate X"), MENU_OPTION_CURSOR_ROTATE_X, Key::A); - options->get_popup()->add_item(TTR("Cursor Rotate Y"), MENU_OPTION_CURSOR_ROTATE_Y, Key::S); - options->get_popup()->add_item(TTR("Cursor Rotate Z"), MENU_OPTION_CURSOR_ROTATE_Z, Key::D); - options->get_popup()->add_item(TTR("Cursor Back Rotate X"), MENU_OPTION_CURSOR_BACK_ROTATE_X, KeyModifierMask::SHIFT + Key::A); - options->get_popup()->add_item(TTR("Cursor Back Rotate Y"), MENU_OPTION_CURSOR_BACK_ROTATE_Y, KeyModifierMask::SHIFT + Key::S); - options->get_popup()->add_item(TTR("Cursor Back Rotate Z"), MENU_OPTION_CURSOR_BACK_ROTATE_Z, KeyModifierMask::SHIFT + Key::D); - options->get_popup()->add_item(TTR("Cursor Clear Rotation"), MENU_OPTION_CURSOR_CLEAR_ROTATION, Key::W); - options->get_popup()->add_separator(); - options->get_popup()->add_check_item(TTR("Paste Selects"), MENU_OPTION_PASTE_SELECTS); - options->get_popup()->add_separator(); - options->get_popup()->add_item(TTR("Duplicate Selection"), MENU_OPTION_SELECTION_DUPLICATE, KeyModifierMask::CTRL + Key::C); - options->get_popup()->add_item(TTR("Cut Selection"), MENU_OPTION_SELECTION_CUT, KeyModifierMask::CTRL + Key::X); - options->get_popup()->add_item(TTR("Clear Selection"), MENU_OPTION_SELECTION_CLEAR, Key::KEY_DELETE); - options->get_popup()->add_item(TTR("Fill Selection"), MENU_OPTION_SELECTION_FILL, KeyModifierMask::CTRL + Key::F); - - options->get_popup()->add_separator(); - options->get_popup()->add_item(TTR("Settings..."), MENU_OPTION_GRIDMAP_SETTINGS); - - settings_dialog = memnew(ConfirmationDialog); - settings_dialog->set_title(TTR("GridMap Settings")); - add_child(settings_dialog); - settings_vbc = memnew(VBoxContainer); - settings_vbc->set_custom_minimum_size(Size2(200, 0) * EDSCALE); - settings_dialog->add_child(settings_vbc); - - settings_pick_distance = memnew(SpinBox); - settings_pick_distance->set_max(10000.0f); - settings_pick_distance->set_min(500.0f); - settings_pick_distance->set_step(1.0f); - settings_pick_distance->set_value(EDITOR_GET("editors/grid_map/pick_distance")); - settings_vbc->add_margin_child(TTR("Pick Distance:"), settings_pick_distance); - - options->get_popup()->connect("id_pressed", callable_mp(this, &GridMapEditor::_menu_option)); - - HBoxContainer *hb = memnew(HBoxContainer); - add_child(hb); - hb->set_h_size_flags(SIZE_EXPAND_FILL); - - search_box = memnew(LineEdit); - search_box->set_h_size_flags(SIZE_EXPAND_FILL); - search_box->set_placeholder(TTR("Filter meshes")); - hb->add_child(search_box); - search_box->connect("text_changed", callable_mp(this, &GridMapEditor::_text_changed)); - search_box->connect("gui_input", callable_mp(this, &GridMapEditor::_sbox_input)); - - mode_thumbnail = memnew(Button); - mode_thumbnail->set_flat(true); - mode_thumbnail->set_toggle_mode(true); - mode_thumbnail->set_pressed(true); - hb->add_child(mode_thumbnail); - mode_thumbnail->connect("pressed", callable_mp(this, &GridMapEditor::_set_display_mode), varray(DISPLAY_THUMBNAIL)); - - mode_list = memnew(Button); - mode_list->set_flat(true); - mode_list->set_toggle_mode(true); - mode_list->set_pressed(false); - hb->add_child(mode_list); - mode_list->connect("pressed", callable_mp(this, &GridMapEditor::_set_display_mode), varray(DISPLAY_LIST)); - - size_slider = memnew(HSlider); - size_slider->set_h_size_flags(SIZE_EXPAND_FILL); - size_slider->set_min(0.2f); - size_slider->set_max(4.0f); - size_slider->set_step(0.1f); - size_slider->set_value(1.0f); - size_slider->connect("value_changed", callable_mp(this, &GridMapEditor::_icon_size_changed)); - add_child(size_slider); - - EDITOR_DEF("editors/grid_map/preview_size", 64); - - mesh_library_palette = memnew(ItemList); - add_child(mesh_library_palette); - mesh_library_palette->set_v_size_flags(SIZE_EXPAND_FILL); - mesh_library_palette->connect("gui_input", callable_mp(this, &GridMapEditor::_mesh_library_palette_input)); - - info_message = memnew(Label); - info_message->set_text(TTR("Give a MeshLibrary resource to this GridMap to use its meshes.")); - info_message->set_vertical_alignment(VERTICAL_ALIGNMENT_CENTER); - info_message->set_horizontal_alignment(HORIZONTAL_ALIGNMENT_CENTER); - info_message->set_autowrap_mode(Label::AUTOWRAP_WORD_SMART); - info_message->set_custom_minimum_size(Size2(100 * EDSCALE, 0)); - info_message->set_anchors_and_offsets_preset(PRESET_WIDE, PRESET_MODE_KEEP_SIZE, 8 * EDSCALE); - mesh_library_palette->add_child(info_message); - - edit_axis = Vector3::AXIS_Y; - edit_floor[0] = -1; - edit_floor[1] = -1; - edit_floor[2] = -1; - - selection_mesh = RenderingServer::get_singleton()->mesh_create(); - paste_mesh = RenderingServer::get_singleton()->mesh_create(); - - { - // Selection mesh create. - - Vector lines; - Vector triangles; - Vector square[3]; - - for (int i = 0; i < 6; i++) { - Vector3 face_points[4]; - - for (int j = 0; j < 4; j++) { - float v[3]; - v[0] = 1.0; - v[1] = 1 - 2 * ((j >> 1) & 1); - v[2] = v[1] * (1 - 2 * (j & 1)); - - for (int k = 0; k < 3; k++) { - if (i < 3) { - face_points[j][(i + k) % 3] = v[k]; - } else { - face_points[3 - j][(i + k) % 3] = -v[k]; - } - } - } - - triangles.push_back(face_points[0] * 0.5 + Vector3(0.5, 0.5, 0.5)); - triangles.push_back(face_points[1] * 0.5 + Vector3(0.5, 0.5, 0.5)); - triangles.push_back(face_points[2] * 0.5 + Vector3(0.5, 0.5, 0.5)); - - triangles.push_back(face_points[2] * 0.5 + Vector3(0.5, 0.5, 0.5)); - triangles.push_back(face_points[3] * 0.5 + Vector3(0.5, 0.5, 0.5)); - triangles.push_back(face_points[0] * 0.5 + Vector3(0.5, 0.5, 0.5)); - } - - for (int i = 0; i < 12; i++) { - AABB base(Vector3(0, 0, 0), Vector3(1, 1, 1)); - Vector3 a, b; - base.get_edge(i, a, b); - lines.push_back(a); - lines.push_back(b); - } - - for (int i = 0; i < 3; i++) { - Vector3 points[4]; - for (int j = 0; j < 4; j++) { - static const bool orderx[4] = { false, true, true, false }; - static const bool ordery[4] = { false, false, true, true }; - - Vector3 sp; - if (orderx[j]) { - sp[(i + 1) % 3] = 1.0; - } - if (ordery[j]) { - sp[(i + 2) % 3] = 1.0; - } - - points[j] = sp; - } - - for (int j = 0; j < 4; j++) { - Vector3 ofs; - ofs[i] += 0.01; - square[i].push_back(points[j] - ofs); - square[i].push_back(points[(j + 1) % 4] - ofs); - square[i].push_back(points[j] + ofs); - square[i].push_back(points[(j + 1) % 4] + ofs); - } - } - - Array d; - d.resize(RS::ARRAY_MAX); - - inner_mat.instantiate(); - inner_mat->set_albedo(Color(0.7, 0.7, 1.0, 0.2)); - inner_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); - inner_mat->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA); - - d[RS::ARRAY_VERTEX] = triangles; - RenderingServer::get_singleton()->mesh_add_surface_from_arrays(selection_mesh, RS::PRIMITIVE_TRIANGLES, d); - RenderingServer::get_singleton()->mesh_surface_set_material(selection_mesh, 0, inner_mat->get_rid()); - - outer_mat.instantiate(); - outer_mat->set_albedo(Color(0.7, 0.7, 1.0, 0.8)); - outer_mat->set_on_top_of_alpha(); - - outer_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); - outer_mat->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA); - - selection_floor_mat.instantiate(); - selection_floor_mat->set_albedo(Color(0.80, 0.80, 1.0, 1)); - selection_floor_mat->set_on_top_of_alpha(); - selection_floor_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); - - d[RS::ARRAY_VERTEX] = lines; - RenderingServer::get_singleton()->mesh_add_surface_from_arrays(selection_mesh, RS::PRIMITIVE_LINES, d); - RenderingServer::get_singleton()->mesh_surface_set_material(selection_mesh, 1, outer_mat->get_rid()); - - d[RS::ARRAY_VERTEX] = triangles; - RenderingServer::get_singleton()->mesh_add_surface_from_arrays(paste_mesh, RS::PRIMITIVE_TRIANGLES, d); - RenderingServer::get_singleton()->mesh_surface_set_material(paste_mesh, 0, inner_mat->get_rid()); - - d[RS::ARRAY_VERTEX] = lines; - RenderingServer::get_singleton()->mesh_add_surface_from_arrays(paste_mesh, RS::PRIMITIVE_LINES, d); - RenderingServer::get_singleton()->mesh_surface_set_material(paste_mesh, 1, outer_mat->get_rid()); - - for (int i = 0; i < 3; i++) { - d[RS::ARRAY_VERTEX] = square[i]; - selection_level_mesh[i] = RS::get_singleton()->mesh_create(); - RenderingServer::get_singleton()->mesh_add_surface_from_arrays(selection_level_mesh[i], RS::PRIMITIVE_LINES, d); - RenderingServer::get_singleton()->mesh_surface_set_material(selection_level_mesh[i], 0, selection_floor_mat->get_rid()); - } - } - - _set_selection(false); - - indicator_mat.instantiate(); - indicator_mat->set_shading_mode(StandardMaterial3D::SHADING_MODE_UNSHADED); - indicator_mat->set_transparency(StandardMaterial3D::TRANSPARENCY_ALPHA); - indicator_mat->set_flag(StandardMaterial3D::FLAG_SRGB_VERTEX_COLOR, true); - indicator_mat->set_flag(StandardMaterial3D::FLAG_ALBEDO_FROM_VERTEX_COLOR, true); - indicator_mat->set_albedo(Color(0.8, 0.5, 0.1)); -} - -GridMapEditor::~GridMapEditor() { - _clear_clipboard_data(); - - for (int i = 0; i < 3; i++) { - if (grid[i].is_valid()) { - RenderingServer::get_singleton()->free(grid[i]); - } - if (grid_instance[i].is_valid()) { - RenderingServer::get_singleton()->free(grid_instance[i]); - } - if (cursor_instance.is_valid()) { - RenderingServer::get_singleton()->free(cursor_instance); - } - if (selection_level_instance[i].is_valid()) { - RenderingServer::get_singleton()->free(selection_level_instance[i]); - } - if (selection_level_mesh[i].is_valid()) { - RenderingServer::get_singleton()->free(selection_level_mesh[i]); - } - } - - RenderingServer::get_singleton()->free(selection_mesh); - if (selection_instance.is_valid()) { - RenderingServer::get_singleton()->free(selection_instance); - } - - RenderingServer::get_singleton()->free(paste_mesh); - if (paste_instance.is_valid()) { - RenderingServer::get_singleton()->free(paste_instance); - } -} - -void GridMapEditorPlugin::_notification(int p_what) { - switch (p_what) { - case EditorSettings::NOTIFICATION_EDITOR_SETTINGS_CHANGED: { - switch ((int)EditorSettings::get_singleton()->get("editors/grid_map/editor_side")) { - case 0: { // Left. - Node3DEditor::get_singleton()->move_control_to_left_panel(grid_map_editor); - } break; - case 1: { // Right. - Node3DEditor::get_singleton()->move_control_to_right_panel(grid_map_editor); - } break; - } - } break; - } -} - -void GridMapEditorPlugin::edit(Object *p_object) { - grid_map_editor->edit(Object::cast_to(p_object)); -} - -bool GridMapEditorPlugin::handles(Object *p_object) const { - return p_object->is_class("GridMap"); -} - -void GridMapEditorPlugin::make_visible(bool p_visible) { - if (p_visible) { - grid_map_editor->show(); - grid_map_editor->spatial_editor_hb->show(); - grid_map_editor->set_process(true); - } else { - grid_map_editor->spatial_editor_hb->hide(); - grid_map_editor->hide(); - grid_map_editor->edit(nullptr); - grid_map_editor->set_process(false); - } -} - -GridMapEditorPlugin::GridMapEditorPlugin() { - EDITOR_DEF("editors/grid_map/editor_side", 1); - EditorSettings::get_singleton()->add_property_hint(PropertyInfo(Variant::INT, "editors/grid_map/editor_side", PROPERTY_HINT_ENUM, "Left,Right")); - - grid_map_editor = memnew(GridMapEditor); - switch ((int)EditorSettings::get_singleton()->get("editors/grid_map/editor_side")) { - case 0: { // Left. - Node3DEditor::get_singleton()->add_control_to_left_panel(grid_map_editor); - } break; - case 1: { // Right. - Node3DEditor::get_singleton()->add_control_to_right_panel(grid_map_editor); - } break; - } - grid_map_editor->hide(); -} - -GridMapEditorPlugin::~GridMapEditorPlugin() { -} diff --git a/modules/gridmap/grid_map_editor_plugin.h b/modules/gridmap/grid_map_editor_plugin.h deleted file mode 100644 index c44c4ca7e0..0000000000 --- a/modules/gridmap/grid_map_editor_plugin.h +++ /dev/null @@ -1,252 +0,0 @@ -/*************************************************************************/ -/* grid_map_editor_plugin.h */ -/*************************************************************************/ -/* This file is part of: */ -/* GODOT ENGINE */ -/* https://godotengine.org */ -/*************************************************************************/ -/* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */ -/* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */ -/* */ -/* Permission is hereby granted, free of charge, to any person obtaining */ -/* a copy of this software and associated documentation files (the */ -/* "Software"), to deal in the Software without restriction, including */ -/* without limitation the rights to use, copy, modify, merge, publish, */ -/* distribute, sublicense, and/or sell copies of the Software, and to */ -/* permit persons to whom the Software is furnished to do so, subject to */ -/* the following conditions: */ -/* */ -/* The above copyright notice and this permission notice shall be */ -/* included in all copies or substantial portions of the Software. */ -/* */ -/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ -/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ -/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/ -/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ -/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ -/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ -/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ -/*************************************************************************/ - -#ifndef GRID_MAP_EDITOR_PLUGIN_H -#define GRID_MAP_EDITOR_PLUGIN_H - -#include "editor/editor_plugin.h" -#include "grid_map.h" -#include "scene/gui/item_list.h" -#include "scene/gui/slider.h" -#include "scene/gui/spin_box.h" - -class Node3DEditorPlugin; - -class GridMapEditor : public VBoxContainer { - GDCLASS(GridMapEditor, VBoxContainer); - - enum { - GRID_CURSOR_SIZE = 50 - }; - - enum InputAction { - INPUT_NONE, - INPUT_PAINT, - INPUT_ERASE, - INPUT_PICK, - INPUT_SELECT, - INPUT_PASTE, - }; - - enum DisplayMode { - DISPLAY_THUMBNAIL, - DISPLAY_LIST - }; - - UndoRedo *undo_redo; - InputAction input_action = INPUT_NONE; - Panel *panel; - MenuButton *options; - SpinBox *floor; - double accumulated_floor_delta = 0.0; - Button *mode_thumbnail; - Button *mode_list; - LineEdit *search_box; - HSlider *size_slider; - HBoxContainer *spatial_editor_hb; - ConfirmationDialog *settings_dialog; - VBoxContainer *settings_vbc; - SpinBox *settings_pick_distance; - Label *spin_box_label; - - struct SetItem { - Vector3i position; - int new_value = 0; - int new_orientation = 0; - int old_value = 0; - int old_orientation = 0; - }; - - List set_items; - - GridMap *node = nullptr; - MeshLibrary *last_mesh_library; - - Transform3D grid_xform; - Transform3D edit_grid_xform; - Vector3::Axis edit_axis; - int edit_floor[3]; - Vector3 grid_ofs; - - RID grid[3]; - RID grid_instance[3]; - RID cursor_instance; - RID selection_mesh; - RID selection_instance; - RID selection_level_mesh[3]; - RID selection_level_instance[3]; - RID paste_mesh; - RID paste_instance; - - struct ClipboardItem { - int cell_item = 0; - Vector3 grid_offset; - int orientation = 0; - RID instance; - }; - - List clipboard_items; - - Ref indicator_mat; - Ref inner_mat; - Ref outer_mat; - Ref selection_floor_mat; - - bool updating = false; - - struct Selection { - Vector3 click; - Vector3 current; - Vector3 begin; - Vector3 end; - bool active = false; - } selection; - Selection last_selection; - - struct PasteIndicator { - Vector3 click; - Vector3 current; - Vector3 begin; - Vector3 end; - int orientation = 0; - }; - PasteIndicator paste_indicator; - - bool cursor_visible = false; - Transform3D cursor_transform; - - Vector3 cursor_origin; - - int display_mode = DISPLAY_THUMBNAIL; - int selected_palette = -1; - int cursor_rot = 0; - - enum Menu { - MENU_OPTION_NEXT_LEVEL, - MENU_OPTION_PREV_LEVEL, - MENU_OPTION_LOCK_VIEW, - MENU_OPTION_X_AXIS, - MENU_OPTION_Y_AXIS, - MENU_OPTION_Z_AXIS, - MENU_OPTION_CURSOR_ROTATE_Y, - MENU_OPTION_CURSOR_ROTATE_X, - MENU_OPTION_CURSOR_ROTATE_Z, - MENU_OPTION_CURSOR_BACK_ROTATE_Y, - MENU_OPTION_CURSOR_BACK_ROTATE_X, - MENU_OPTION_CURSOR_BACK_ROTATE_Z, - MENU_OPTION_CURSOR_CLEAR_ROTATION, - MENU_OPTION_PASTE_SELECTS, - MENU_OPTION_SELECTION_DUPLICATE, - MENU_OPTION_SELECTION_CUT, - MENU_OPTION_SELECTION_CLEAR, - MENU_OPTION_SELECTION_FILL, - MENU_OPTION_GRIDMAP_SETTINGS - - }; - - Node3DEditorPlugin *spatial_editor; - - struct AreaDisplay { - RID mesh; - RID instance; - }; - - ItemList *mesh_library_palette; - Label *info_message; - - void update_grid(); // Change which and where the grid is displayed - void _draw_grids(const Vector3 &cell_size); - void _configure(); - void _menu_option(int); - void update_palette(); - void _set_display_mode(int p_mode); - void _item_selected_cbk(int idx); - void _update_cursor_transform(); - void _update_cursor_instance(); - void _update_theme(); - - void _text_changed(const String &p_text); - void _sbox_input(const Ref &p_ie); - void _mesh_library_palette_input(const Ref &p_ie); - - void _icon_size_changed(float p_value); - - void _clear_clipboard_data(); - void _set_clipboard_data(); - void _update_paste_indicator(); - void _do_paste(); - void _update_selection_transform(); - void _validate_selection(); - void _set_selection(bool p_active, const Vector3 &p_begin = Vector3(), const Vector3 &p_end = Vector3()); - - void _floor_changed(float p_value); - void _floor_mouse_exited(); - - void _delete_selection(); - void _fill_selection(); - - bool do_input_action(Camera3D *p_camera, const Point2 &p_point, bool p_click); - - friend class GridMapEditorPlugin; - -protected: - void _notification(int p_what); - void _node_removed(Node *p_node); - static void _bind_methods(); - -public: - EditorPlugin::AfterGUIInput forward_spatial_input_event(Camera3D *p_camera, const Ref &p_event); - - void edit(GridMap *p_gridmap); - GridMapEditor(); - ~GridMapEditor(); -}; - -class GridMapEditorPlugin : public EditorPlugin { - GDCLASS(GridMapEditorPlugin, EditorPlugin); - - GridMapEditor *grid_map_editor; - -protected: - void _notification(int p_what); - -public: - virtual EditorPlugin::AfterGUIInput forward_spatial_gui_input(Camera3D *p_camera, const Ref &p_event) override { return grid_map_editor->forward_spatial_input_event(p_camera, p_event); } - virtual String get_name() const override { return "GridMap"; } - bool has_main_screen() const override { return false; } - virtual void edit(Object *p_object) override; - virtual bool handles(Object *p_object) const override; - virtual void make_visible(bool p_visible) override; - - GridMapEditorPlugin(); - ~GridMapEditorPlugin(); -}; - -#endif // CUBE_GRID_MAP_EDITOR_PLUGIN_H diff --git a/modules/gridmap/register_types.cpp b/modules/gridmap/register_types.cpp index f05c5de18b..d7c9f5c92e 100644 --- a/modules/gridmap/register_types.cpp +++ b/modules/gridmap/register_types.cpp @@ -28,21 +28,25 @@ /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ /*************************************************************************/ -#include "register_types.h" #ifndef _3D_DISABLED + +#include "register_types.h" + #include "core/object/class_db.h" #include "grid_map.h" -#include "grid_map_editor_plugin.h" + +#ifdef TOOLS_ENABLED +#include "editor/grid_map_editor_plugin.h" #endif void register_gridmap_types() { -#ifndef _3D_DISABLED GDREGISTER_CLASS(GridMap); #ifdef TOOLS_ENABLED EditorPlugins::add_by_type(); #endif -#endif } void unregister_gridmap_types() { } + +#endif // _3D_DISABLED -- cgit v1.2.3