pandemonium_engine/modules/mesh_data_resource/editor/mdi_gizmo.cpp

2231 lines
55 KiB
C++

/*
Copyright (c) 2019-2022 Péter Magyar
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 "mdi_gizmo.h"
#include "../mesh_data_resource.h"
#include "../nodes/mesh_data_instance.h"
#include "./utilities/mdr_ed_mesh_decompose.h"
#include "./utilities/mdr_ed_mesh_outline.h"
#include "./utilities/mdr_ed_mesh_utils.h"
#include "core/math/geometry.h"
#include "editor/editor_node.h"
#include "modules/mesh_utils/mesh_utils.h"
#include "scene/3d/camera.h"
void MDIGizmo::set_visible(const bool visible) {
_visible = visible;
redraw();
}
void MDIGizmo::setup() {
MeshDataInstance *mdi = Object::cast_to<MeshDataInstance>(get_spatial_node());
ERR_FAIL_COND(!mdi);
mdi->connect("mesh_data_resource_changed", this, "on_mesh_data_resource_changed");
on_mesh_data_resource_changed(mdi->get_mesh_data());
}
void MDIGizmo::set_editor_plugin(EditorPlugin *editor_plugin) {
_editor_plugin = editor_plugin;
_undo_redo = EditorNode::get_undo_redo();
}
void MDIGizmo::set_handle(int index, bool secondary, Camera *camera, const Point2 &point) {
Vector2 relative = point - previous_point;
if (!_handle_drag_op) {
relative = Vector2();
_handle_drag_op = true;
if (edit_mode == EditMode::EDIT_MODE_SCALE) {
_drag_op_accumulator = Vector3(1, 1, 1);
} else {
_drag_op_accumulator = Vector3();
}
_drag_op_accumulator_quat = Quaternion();
_drag_op_orig_verices = copy_mdr_verts_array();
setup_op_drag_indices();
_drag_op_pivot = get_drag_op_pivot();
}
if (edit_mode == EditMode::EDIT_MODE_NONE) {
return;
} else if (edit_mode == EditMode::EDIT_MODE_TRANSLATE) {
Vector3 ofs;
ofs = camera->get_global_transform().basis.get_axis(0);
if ((axis_constraint & AXIS_CONSTRAINT_X) != 0) {
ofs.x *= relative.x * 0.01;
} else {
ofs.x = 0;
}
if ((axis_constraint & AXIS_CONSTRAINT_Y) != 0) {
ofs.y = relative.y * -0.01;
} else {
ofs.y = 0;
}
if ((axis_constraint & AXIS_CONSTRAINT_Z) != 0) {
ofs.z *= relative.x * 0.01;
} else {
ofs.z = 0;
}
_drag_op_accumulator += ofs;
add_to_all_selected(_drag_op_accumulator);
apply();
redraw();
} else if (edit_mode == EditMode::EDIT_MODE_SCALE) {
float r = ((relative.x + relative.y) * 0.05);
Vector3 vs;
if ((axis_constraint & AXIS_CONSTRAINT_X) != 0) {
vs.x = r;
}
if ((axis_constraint & AXIS_CONSTRAINT_Y) != 0) {
vs.y = r;
}
if ((axis_constraint & AXIS_CONSTRAINT_Z) != 0) {
vs.z = r;
}
_drag_op_accumulator += vs;
Basis b = Basis().scaled(_drag_op_accumulator);
Transform t = Transform(Basis(), _drag_op_pivot);
t *= Transform(b, Vector3());
t *= Transform(Basis(), _drag_op_pivot).inverse();
mul_all_selected_with_transform(t);
apply();
redraw();
} else if (edit_mode == EditMode::EDIT_MODE_ROTATE) {
Quaternion yrot = Quaternion(Vector3(0, 1, 0), relative.x * 0.01);
Quaternion xrot = Quaternion(camera->get_global_transform().basis.get_axis(0), relative.y * 0.01);
_drag_op_accumulator_quat *= yrot;
_drag_op_accumulator_quat *= xrot;
_drag_op_accumulator_quat = _drag_op_accumulator_quat.normalized();
Basis b = Basis(_drag_op_accumulator_quat);
Transform t = Transform(Basis(), _drag_op_pivot);
t *= Transform(b, Vector3());
t *= Transform(Basis(), _drag_op_pivot).inverse();
mul_all_selected_with_transform(t);
apply();
redraw();
}
previous_point = point;
}
void MDIGizmo::redraw() {
clear();
if (!_visible) {
return;
}
if (!_mdr.is_valid()) {
return;
}
Array array = _mdr->get_array();
if (array.size() != ArrayMesh::ARRAY_MAX) {
return;
}
if (!get_plugin().is_valid()) {
return;
}
Ref<SpatialMaterial> handles_material = get_plugin()->get_material("handles", Ref<EditorSpatialGizmo>(this));
Ref<SpatialMaterial> material = get_plugin()->get_material("main", Ref<EditorSpatialGizmo>(this));
Ref<SpatialMaterial> seam_material = get_plugin()->get_material("seam", Ref<EditorSpatialGizmo>(this));
_mesh_outline_generator->setup(_mdr);
if (selection_mode == SELECTION_MODE_EDGE) {
_mesh_outline_generator->generate_mark_edges(visual_indicator_outline, visual_indicator_handle);
} else if (selection_mode == SELECTION_MODE_FACE) {
_mesh_outline_generator->generate_mark_faces(visual_indicator_outline, visual_indicator_handle);
} else {
_mesh_outline_generator->generate(visual_indicator_outline, visual_indicator_handle);
}
if (visual_indicator_outline || visual_indicator_handle) {
add_lines(_mesh_outline_generator->lines, material, false);
}
if (visual_indicator_seam) {
add_lines(_mesh_outline_generator->seam_lines, seam_material, false);
}
if (_selected_points.size() > 0) {
Vector<Vector3> vs;
for (int i = 0; i < _selected_points.size(); ++i) {
vs.push_back(_handle_points[_selected_points[i]]);
}
add_handles(vs, handles_material);
}
}
void MDIGizmo::apply() {
if (!_mdr.is_valid()) {
return;
}
disable_change_event();
Array arrs = _mdr->get_array();
arrs[ArrayMesh::ARRAY_VERTEX] = _vertices;
arrs[ArrayMesh::ARRAY_INDEX] = _indices;
_mdr->set_array(arrs);
enable_change_event();
}
void MDIGizmo::select_all() {
if (_selected_points.size() == _handle_points.size()) {
return;
}
_selected_points.resize(_handle_points.size());
PoolIntArray::Write w = _selected_points.write();
for (int i = 0; i < _selected_points.size(); ++i) {
w[i] = i;
}
redraw();
}
bool MDIGizmo::selection_click(Camera *camera, const Ref<InputEventMouse> &event) {
if (handle_selection_type == HANDLE_SELECTION_TYPE_FRONT) {
return selection_click_select_front_or_back(camera, event);
} else if (handle_selection_type == HANDLE_SELECTION_TYPE_BACK) {
return selection_click_select_front_or_back(camera, event);
} else {
return selection_click_select_through(camera, event);
}
return false;
}
bool MDIGizmo::is_point_visible(const Vector3 &point_orig, const Vector3 &camera_pos, const Transform &gt) {
Vector3 point = gt.xform(point_orig);
// go from the given point to the origin (camera_pos -> camera)
Vector3 dir = camera_pos - point;
dir = dir.normalized();
// Might need to reduce z fighting
//point += dir * 0.5
for (int i = 0; i < _indices.size(); i += 3) {
int i0 = _indices[i];
int i1 = _indices[i + 1];
int i2 = _indices[i + 2];
Vector3 v0 = _vertices[i0];
Vector3 v1 = _vertices[i1];
Vector3 v2 = _vertices[i2];
v0 = gt.xform(v0);
v1 = gt.xform(v1);
v2 = gt.xform(v2);
bool intersects = Geometry::ray_intersects_triangle(point, dir, v0, v1, v2);
if (intersects) {
return false;
}
}
return true;
}
bool MDIGizmo::selection_click_select_front_or_back(Camera *camera, const Ref<InputEventMouse> &event) {
Transform gt = get_spatial_node()->get_global_transform();
Vector3 ray_from = camera->get_global_transform().origin;
Vector2 gpoint = event->get_position();
float grab_threshold = 8;
// select vertex
int closest_idx = -1;
float closest_dist = 1e10;
for (int i = 0; i < _handle_points.size(); ++i) {
Vector3 vert_pos_3d = gt.xform(_handle_points[i]);
Vector2 vert_pos_2d = camera->unproject_position(vert_pos_3d);
float dist_3d = ray_from.distance_to(vert_pos_3d);
float dist_2d = gpoint.distance_to(vert_pos_2d);
if (dist_2d < grab_threshold && dist_3d < closest_dist) {
bool point_visible = is_point_visible(_handle_points[i], ray_from, gt);
if (handle_selection_type == HANDLE_SELECTION_TYPE_FRONT) {
if (!point_visible) {
continue;
}
} else if (handle_selection_type == HANDLE_SELECTION_TYPE_BACK) {
if (point_visible) {
continue;
}
}
closest_dist = dist_3d;
closest_idx = i;
}
}
if (closest_idx >= 0) {
for (int si = 0; si < _selected_points.size(); ++si) {
if (_selected_points[si] == closest_idx) {
if (event->get_alt() || event->get_control()) {
_selected_points.remove(si);
return true;
}
return false;
}
}
if (event->get_alt() || event->get_control()) {
return false;
}
if (event->get_shift()) {
_selected_points.append(closest_idx);
} else {
// Select new point only
_selected_points.resize(0);
_selected_points.append(closest_idx);
}
apply();
redraw();
} else {
// Don't unselect all if either control or shift is held down
if (event->get_shift() || event->get_control() || event->get_alt()) {
return false;
}
if (_selected_points.size() == 0) {
return false;
}
//Unselect all
_selected_points.resize(0);
redraw();
}
return false;
}
bool MDIGizmo::selection_click_select_through(Camera *camera, const Ref<InputEventMouse> &event) {
Transform gt = get_spatial_node()->get_global_transform();
Vector3 ray_from = camera->get_global_transform().origin;
Vector2 gpoint = event->get_position();
float grab_threshold = 8;
// select vertex
int closest_idx = -1;
float closest_dist = 1e10;
for (int i = 0; i < _handle_points.size(); ++i) {
Vector3 vert_pos_3d = gt.xform(_handle_points[i]);
Vector2 vert_pos_2d = camera->unproject_position(vert_pos_3d);
float dist_3d = ray_from.distance_to(vert_pos_3d);
float dist_2d = gpoint.distance_to(vert_pos_2d);
if (dist_2d < grab_threshold && dist_3d < closest_dist) {
closest_dist = dist_3d;
closest_idx = i;
}
}
if (closest_idx >= 0) {
for (int si = 0; si < _selected_points.size(); ++si) {
if (_selected_points[si] == closest_idx) {
if (event->get_alt() || event->get_control()) {
_selected_points.remove(si);
return true;
}
return false;
}
}
if (event->get_alt() || event->get_control()) {
return false;
}
if (event->get_shift()) {
_selected_points.append(closest_idx);
} else {
// Select new point only
_selected_points.resize(0);
_selected_points.append(closest_idx);
}
apply();
redraw();
} else {
// Don't unselect all if either control or shift is held down
if (event->get_shift() || event->get_control() || event->get_alt()) {
return false;
}
if (_selected_points.size() == 0) {
return false;
}
//Unselect all
_selected_points.resize(0);
redraw();
}
return false;
}
void MDIGizmo::selection_drag(Camera *camera, const Ref<InputEventMouse> &event) {
if (handle_selection_type == HANDLE_SELECTION_TYPE_FRONT) {
selection_drag_rect_select_front_back(camera, event);
} else if (handle_selection_type == HANDLE_SELECTION_TYPE_BACK) {
selection_drag_rect_select_front_back(camera, event);
} else {
selection_drag_rect_select_through(camera, event);
}
}
void MDIGizmo::selection_drag_rect_select_front_back(Camera *camera, const Ref<InputEventMouse> &event) {
Transform gt = get_spatial_node()->get_global_transform();
Vector3 ray_from = camera->get_global_transform().origin;
Vector2 mouse_pos = event->get_position();
Vector2 rect_size = _rect_drag_start_point - mouse_pos;
rect_size.x = ABS(rect_size.x);
rect_size.y = ABS(rect_size.y);
Rect2 rect = Rect2(_rect_drag_start_point, rect_size);
// This is needed so selection works even when you drag from bottom to top, and from right to left
Vector2 rect_ofs = _rect_drag_start_point - mouse_pos;
if (rect_ofs.x > 0) {
rect.position.x -= rect_ofs.x;
}
if (rect_ofs.y > 0) {
rect.position.y -= rect_ofs.y;
}
PoolIntArray selected;
for (int i = 0; i < _handle_points.size(); ++i) {
Vector3 vert_pos_3d = gt.xform(_handle_points[i]);
Vector2 vert_pos_2d = camera->unproject_position(vert_pos_3d);
if (rect.has_point(vert_pos_2d)) {
bool point_visible = is_point_visible(_handle_points[i], ray_from, gt);
if (handle_selection_type == HANDLE_SELECTION_TYPE_FRONT) {
if (!point_visible) {
continue;
}
} else if (handle_selection_type == HANDLE_SELECTION_TYPE_BACK) {
if (point_visible) {
continue;
}
}
selected.push_back(i);
}
}
if (event->get_alt() || event->get_control()) {
PoolIntArray::Read r = selected.read();
for (int is = 0; is < selected.size(); ++is) {
int isel = r[is];
for (int i = 0; i < _selected_points.size(); ++i) {
if (_selected_points[i] == isel) {
_selected_points.remove(i);
break;
}
}
}
r.release();
redraw();
return;
}
if (event->get_shift()) {
PoolIntArray::Read r = selected.read();
for (int is = 0; is < selected.size(); ++is) {
int isel = r[is];
if (!pool_int_arr_contains(_selected_points, isel)) {
_selected_points.push_back(isel);
}
}
r.release();
redraw();
return;
}
_selected_points.resize(0);
_selected_points.append_array(selected);
redraw();
}
void MDIGizmo::selection_drag_rect_select_through(Camera *camera, const Ref<InputEventMouse> &event) {
Transform gt = get_spatial_node()->get_global_transform();
Vector2 mouse_pos = event->get_position();
Vector2 rect_size = _rect_drag_start_point - mouse_pos;
rect_size.x = ABS(rect_size.x);
rect_size.y = ABS(rect_size.y);
Rect2 rect = Rect2(_rect_drag_start_point, rect_size);
// This is needed so selection works even when you drag from bottom to top, and from right to left
Vector2 rect_ofs = _rect_drag_start_point - mouse_pos;
if (rect_ofs.x > 0) {
rect.position.x -= rect_ofs.x;
}
if (rect_ofs.y > 0) {
rect.position.y -= rect_ofs.y;
}
PoolIntArray selected;
for (int i = 0; i < _handle_points.size(); ++i) {
Vector3 vert_pos_3d = gt.xform(_handle_points[i]);
Vector2 vert_pos_2d = camera->unproject_position(vert_pos_3d);
if (rect.has_point(vert_pos_2d)) {
selected.push_back(i);
}
}
if (event->get_alt() || event->get_control()) {
for (int ii = 0; ii < selected.size(); ++ii) {
int isel = selected[ii];
for (int i = 0; i < _selected_points.size(); ++i) {
if (_selected_points[i] == isel) {
_selected_points.remove(i);
break;
}
}
}
redraw();
return;
}
if (event->get_shift()) {
for (int ii = 0; ii < selected.size(); ++ii) {
int isel = selected[ii];
if (!pool_int_arr_contains(_selected_points, isel)) {
_selected_points.push_back(isel);
}
}
redraw();
return;
}
_selected_points.resize(0);
_selected_points.append_array(selected);
redraw();
}
EditorPlugin::AfterGUIInput MDIGizmo::forward_spatial_gui_input(Camera *camera, const Ref<InputEvent> &event) {
_last_known_camera_facing = camera->get_transform().basis.xform(Vector3(0, 0, -1));
Ref<InputEventMouseButton> event_button = event;
if (event_button.is_valid()) {
if (event_button->get_button_index() == BUTTON_LEFT) {
if (_handle_drag_op) {
if (!event_button->is_pressed()) {
_handle_drag_op = false;
// If a handle was being dragged only run these
if (_mdr.is_valid() && _mdr->get_array().size() == ArrayMesh::ARRAY_MAX && !_mdr->get_array()[ArrayMesh::ARRAY_VERTEX].is_null()) {
Array arrs = _mdr->get_array();
PoolVector3Array vertices = arrs[ArrayMesh::ARRAY_VERTEX];
if (vertices.size() == _drag_op_orig_verices.size()) {
_undo_redo->create_action("Drag");
_undo_redo->add_do_method(this, "apply_vertex_array", _mdr, vertices);
_undo_redo->add_undo_method(this, "apply_vertex_array", _mdr, _drag_op_orig_verices);
_undo_redo->commit_action();
}
}
}
// Dont consume the event here, because the handles will get stuck
// to the mouse pointer if we return true
return EditorPlugin::AFTER_GUI_INPUT_NO_DESELECT;
}
if (!event_button->is_pressed()) {
// See whether we should check for a click or a selection box
Vector2 mouse_pos = event_button->get_position();
Vector2 rect_size = _rect_drag_start_point - mouse_pos;
rect_size.x = ABS(rect_size.x);
rect_size.y = ABS(rect_size.y);
bool had_rect_drag = false;
if (rect_size.x > _rect_drag_min_ofset || rect_size.y > _rect_drag_min_ofset) {
had_rect_drag = true;
}
if (!had_rect_drag) {
if (selection_click(camera, event)) {
return EditorPlugin::AFTER_GUI_INPUT_STOP;
} else {
return EditorPlugin::AFTER_GUI_INPUT_NO_DESELECT;
}
} else {
selection_drag(camera, event_button);
// Always return false here, so the drag rect thing disappears in the editor
return EditorPlugin::AFTER_GUI_INPUT_NO_DESELECT;
}
} else {
// event is pressed
_rect_drag = true;
_rect_drag_start_point = event_button->get_position();
}
}
}
return EditorPlugin::AFTER_GUI_INPUT_NO_DESELECT;
}
void MDIGizmo::add_to_all_selected(const Vector3 &ofs) {
for (int i = 0; i < _selected_points.size(); ++i) {
int indx = _selected_points[i];
Vector3 v = _handle_points[indx];
v += ofs;
_handle_points.set(indx, v);
}
for (int i = 0; i < _drag_op_indices.size(); ++i) {
int indx = _drag_op_indices[i];
Vector3 v = _drag_op_orig_verices[indx];
v += ofs;
_vertices.set(indx, v);
}
}
void MDIGizmo::mul_all_selected_with_basis(const Basis &b) {
for (int i = 0; i < _selected_points.size(); ++i) {
int indx = _selected_points[i];
Vector3 v = _handle_points[indx];
v = b.xform(v);
_handle_points.set(indx, v);
}
for (int i = 0; i < _drag_op_indices.size(); ++i) {
int indx = _drag_op_indices[i];
Vector3 v = _drag_op_orig_verices[indx];
v = b.xform(v);
_vertices.set(indx, v);
}
}
void MDIGizmo::mul_all_selected_with_transform(const Transform &t) {
for (int i = 0; i < _selected_points.size(); ++i) {
int indx = _selected_points[i];
Vector3 v = _handle_points[indx];
v = t.xform(v);
_handle_points.set(indx, v);
}
for (int i = 0; i < _drag_op_indices.size(); ++i) {
int indx = _drag_op_indices[i];
Vector3 v = _drag_op_orig_verices[indx];
v = t.xform(v);
_vertices.set(indx, v);
}
}
void MDIGizmo::mul_all_selected_with_transform_acc(const Transform &t) {
for (int i = 0; i < _selected_points.size(); ++i) {
int indx = _selected_points[i];
Vector3 v = _handle_points[indx];
v = t.xform(v);
_handle_points.set(indx, v);
}
for (int i = 0; i < _drag_op_indices.size(); ++i) {
int indx = _drag_op_indices[i];
Vector3 v = _vertices[indx];
v = t.xform(v);
_vertices.set(indx, v);
}
}
void MDIGizmo::set_translate() {
edit_mode = EDIT_MODE_TRANSLATE;
}
void MDIGizmo::set_scale() {
edit_mode = EDIT_MODE_SCALE;
}
void MDIGizmo::set_rotate() {
edit_mode = EDIT_MODE_ROTATE;
}
void MDIGizmo::set_edit_mode(int em) {
edit_mode = em;
}
void MDIGizmo::set_axis_x(bool on) {
if (on) {
axis_constraint |= AXIS_CONSTRAINT_X;
} else {
if ((axis_constraint & AXIS_CONSTRAINT_X) != 0) {
axis_constraint ^= AXIS_CONSTRAINT_X;
}
}
}
void MDIGizmo::set_axis_y(bool on) {
if (on) {
axis_constraint |= AXIS_CONSTRAINT_Y;
} else {
if ((axis_constraint & AXIS_CONSTRAINT_Y) != 0) {
axis_constraint ^= AXIS_CONSTRAINT_Y;
}
}
}
void MDIGizmo::set_axis_z(bool on) {
if (on) {
axis_constraint |= AXIS_CONSTRAINT_Z;
} else {
if ((axis_constraint & AXIS_CONSTRAINT_Z) != 0) {
axis_constraint ^= AXIS_CONSTRAINT_Z;
}
}
}
void MDIGizmo::set_selection_mode_vertex() {
if (selection_mode == SELECTION_MODE_VERTEX) {
return;
}
selection_mode = SELECTION_MODE_VERTEX;
_selected_points.resize(0);
recalculate_handle_points();
redraw();
}
void MDIGizmo::set_selection_mode_edge() {
if (selection_mode == SELECTION_MODE_EDGE) {
return;
}
selection_mode = SELECTION_MODE_EDGE;
_selected_points.resize(0);
recalculate_handle_points();
redraw();
}
void MDIGizmo::set_selection_mode_face() {
if (selection_mode == SELECTION_MODE_FACE) {
return;
}
selection_mode = SELECTION_MODE_FACE;
_selected_points.resize(0);
recalculate_handle_points();
redraw();
}
void MDIGizmo::recalculate_handle_points() {
if (!_mdr.is_valid()) {
_handle_points.resize(0);
_handle_to_vertex_map.resize(0);
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
_handle_points.resize(0);
_handle_to_vertex_map.resize(0);
return;
}
PoolVector3Array vertices = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (vertices.size() == 0) {
_handle_points.resize(0);
_handle_to_vertex_map.resize(0);
return;
}
Array arr;
arr.resize(ArrayMesh::ARRAY_MAX);
arr[ArrayMesh::ARRAY_VERTEX] = mdr_arr[ArrayMesh::ARRAY_VERTEX];
arr[ArrayMesh::ARRAY_INDEX] = mdr_arr[ArrayMesh::ARRAY_INDEX];
if (selection_mode == SELECTION_MODE_VERTEX) {
Array merged_arrays = MeshUtils::get_singleton()->merge_mesh_array(arr);
_handle_points = merged_arrays[ArrayMesh::ARRAY_VERTEX];
_handle_to_vertex_map = MDREDMeshDecompose::get_handle_vertex_to_vertex_map(mdr_arr, _handle_points);
} else if (selection_mode == SELECTION_MODE_EDGE) {
MDREDMeshDecompose::HandleVertexMapResult result = MDREDMeshDecompose::get_handle_edge_to_vertex_map(arr);
_handle_points = result.handle_points;
_handle_to_vertex_map = result.handle_to_vertex_map;
} else if (selection_mode == SELECTION_MODE_FACE) {
MDREDMeshDecompose::HandleVertexMapResult result = MDREDMeshDecompose::get_handle_face_to_vertex_map(arr);
_handle_points = result.handle_points;
_handle_to_vertex_map = result.handle_to_vertex_map;
}
}
void MDIGizmo::on_mesh_data_resource_changed(Ref<MeshDataResource> mdr) {
if (_mdr.is_valid()) {
_mdr->disconnect("changed", this, "on_mdr_changed");
}
_mdr = mdr;
if (_mdr.is_valid()) {
_mdr->connect("changed", this, "on_mdr_changed");
} else {
_vertices.resize(0);
_indices.resize(0);
recalculate_handle_points();
redraw();
return;
}
Array arrs = _mdr->get_array();
if (arrs.size() == ArrayMesh::ARRAY_MAX && !arrs[ArrayMesh::ARRAY_VERTEX].is_null()) {
_vertices = arrs[ArrayMesh::ARRAY_VERTEX];
_indices = arrs[ArrayMesh::ARRAY_INDEX];
} else {
_vertices.resize(0);
_indices.resize(0);
}
recalculate_handle_points();
redraw();
}
void MDIGizmo::on_mdr_changed() {
if (!_mdr.is_valid()) {
_vertices.resize(0);
_indices.resize(0);
recalculate_handle_points();
redraw();
}
Array arr = _mdr->get_array();
if (arr.size() == ArrayMesh::ARRAY_MAX && !arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
_vertices = arr[ArrayMesh::ARRAY_VERTEX];
_indices = arr[ArrayMesh::ARRAY_INDEX];
} else {
_vertices.resize(0);
_indices.resize(0);
}
recalculate_handle_points();
redraw();
}
void MDIGizmo::disable_change_event() {
_mdr->disconnect("changed", this, "on_mdr_changed");
}
void MDIGizmo::enable_change_event(bool update) {
_mdr->connect("changed", this, "on_mdr_changed");
if (update) {
on_mdr_changed();
}
}
void MDIGizmo::add_triangle() {
if (_mdr.is_valid()) {
Array orig_arr = copy_arrays(_mdr->get_array());
MDREDMeshUtils::add_triangle(_mdr);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Add Triangle");
}
}
void MDIGizmo::add_quad() {
if (_mdr.is_valid()) {
Array orig_arr = copy_arrays(_mdr->get_array());
MDREDMeshUtils::add_quad(_mdr);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Add Quad");
}
}
bool MDIGizmo::is_verts_equal(const Vector3 &v0, const Vector3 &v1) {
return Math::is_equal_approx(v0.x, v1.x) && Math::is_equal_approx(v0.y, v1.y) && Math::is_equal_approx(v0.z, v1.z);
}
Vector3 MDIGizmo::find_other_vertex_for_edge(const int edge, const Vector3 &v0) {
PoolIntArray ps = _handle_to_vertex_map[edge];
Vector3 vert;
for (int i = 0; i < ps.size(); ++i) {
vert = _vertices[ps[i]];
if (!is_verts_equal(v0, vert)) {
return vert;
}
}
return v0;
}
Vector<PoolIntArray> MDIGizmo::split_edge_indices(const int edge) {
PoolIntArray ps = _handle_to_vertex_map[edge];
if (ps.size() == 0) {
return Vector<PoolIntArray>();
}
Vector3 v0 = _vertices[ps[0]];
PoolIntArray v0ei;
v0ei.append(ps[0]);
PoolIntArray v1ei;
for (int i = 1; i < ps.size(); ++i) {
Vector3 vert = _vertices[ps[i]];
if (is_verts_equal(v0, vert)) {
v0ei.append(ps[i]);
} else {
v1ei.append(ps[i]);
}
}
Vector<PoolIntArray> arr;
arr.push_back(v0ei);
arr.push_back(v1ei);
return arr;
}
bool MDIGizmo::pool_int_arr_contains(const PoolIntArray &arr, const int val) {
PoolIntArray::Read r = arr.read();
for (int i = 0; i < arr.size(); ++i) {
if (r[i] == val) {
return true;
}
}
return false;
}
PoolIntArray MDIGizmo::find_triangles_for_edge(int edge) {
Vector<PoolIntArray> eisarr = split_edge_indices(edge);
if (eisarr.size() == 0) {
return PoolIntArray();
}
// these should have the same size
PoolIntArray v0ei = eisarr[0];
PoolIntArray v1ei = eisarr[1];
PoolIntArray res;
for (int i = 0; i < _indices.size(); i += 3) {
int i0 = _indices[i];
int i1 = _indices[i + 1];
int i2 = _indices[i + 2];
if (pool_int_arr_contains(v0ei, i0) || pool_int_arr_contains(v0ei, i1) || pool_int_arr_contains(v0ei, i2)) {
if (pool_int_arr_contains(v1ei, i0) || pool_int_arr_contains(v1ei, i1) || pool_int_arr_contains(v1ei, i2)) {
res.append(i / 3);
}
}
}
return res;
}
int MDIGizmo::find_first_triangle_for_edge(int edge) {
Vector<PoolIntArray> eisarr = split_edge_indices(edge);
if (eisarr.size() == 0) {
return -1;
}
// these should have the same size
PoolIntArray v0ei = eisarr[0];
PoolIntArray v1ei = eisarr[1];
for (int i = 0; i < _indices.size(); i += 3) {
int i0 = _indices[i];
int i1 = _indices[i + 1];
int i2 = _indices[i + 2];
if (pool_int_arr_contains(v0ei, i0) || pool_int_arr_contains(v0ei, i1) || pool_int_arr_contains(v0ei, i2)) {
if (pool_int_arr_contains(v1ei, i0) || pool_int_arr_contains(v1ei, i1) || pool_int_arr_contains(v1ei, i2)) {
return i / 3;
}
}
}
return -1;
}
void MDIGizmo::add_triangle_to_edge(int edge) {
int triangle_index = find_first_triangle_for_edge(edge);
int inds = triangle_index * 3;
int ti0 = _indices[inds];
int ti1 = _indices[inds + 1];
int ti2 = _indices[inds + 2];
PoolIntArray ps = _handle_to_vertex_map[edge];
if (ps.size() == 0) {
return;
}
int ei0 = 0;
int ei1 = 0;
int erefind = 0;
if (!pool_int_arr_contains(ps, ti0)) {
ei0 = ti1;
ei1 = ti2;
erefind = ti0;
} else if (!pool_int_arr_contains(ps, ti1)) {
ei0 = ti0;
ei1 = ti2;
erefind = ti1;
} else if (!pool_int_arr_contains(ps, ti2)) {
ei0 = ti0;
ei1 = ti1;
erefind = ti2;
}
Vector3 fo = MDREDMeshUtils::get_face_normal(_vertices[ti0], _vertices[ti1], _vertices[ti2]);
Vector3 fn = MDREDMeshUtils::get_face_normal(_vertices[ei0], _vertices[ei1], _vertices[erefind]);
if (fo.dot(fn) < 0) {
int t = ei0;
ei0 = ei1;
ei1 = t;
}
MDREDMeshUtils::append_triangle_to_tri_edge(_mdr, _vertices[ei0], _vertices[ei1], _vertices[erefind]);
}
void MDIGizmo::add_quad_to_edge(int edge) {
int triangle_index = find_first_triangle_for_edge(edge);
int inds = triangle_index * 3;
int ti0 = _indices[inds];
int ti1 = _indices[inds + 1];
int ti2 = _indices[inds + 2];
PoolIntArray ps = _handle_to_vertex_map[edge];
if (ps.size() == 0) {
return;
}
int ei0 = 0;
int ei1 = 0;
int erefind = 0;
if (!pool_int_arr_contains(ps, ti0)) {
ei0 = ti1;
ei1 = ti2;
erefind = ti0;
} else if (!pool_int_arr_contains(ps, ti1)) {
ei0 = ti0;
ei1 = ti2;
erefind = ti1;
} else if (!pool_int_arr_contains(ps, ti2)) {
ei0 = ti0;
ei1 = ti1;
erefind = ti2;
}
Vector3 fo = MDREDMeshUtils::get_face_normal(_vertices[ti0], _vertices[ti1], _vertices[ti2]);
Vector3 fn = MDREDMeshUtils::get_face_normal(_vertices[ei0], _vertices[ei1], _vertices[erefind]);
if (fo.dot(fn) < 0) {
int t = ei0;
ei0 = ei1;
ei1 = t;
}
MDREDMeshUtils::append_quad_to_tri_edge(_mdr, _vertices[ei0], _vertices[ei1], _vertices[erefind]);
}
void MDIGizmo::add_triangle_at() {
if (!_mdr.is_valid()) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
} else if (selection_mode == SELECTION_MODE_EDGE) {
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolIntArray::Read r = _selected_points.read();
for (int i = 0; i < _selected_points.size(); ++i) {
add_triangle_to_edge(r[i]);
}
r.release();
_selected_points.resize(0);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Add Triangle At");
enable_change_event();
} else {
add_triangle();
}
}
void MDIGizmo::add_quad_at() {
if (!_mdr.is_valid()) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
} else if (selection_mode == SELECTION_MODE_EDGE) {
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolIntArray::Read r = _selected_points.read();
for (int i = 0; i < _selected_points.size(); ++i) {
add_quad_to_edge(r[i]);
}
r.release();
_selected_points.resize(0);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Add Triangle At");
enable_change_event();
} else {
add_quad();
}
}
void MDIGizmo::extrude() {
if (!_mdr.is_valid()) {
return;
}
Array arr = _mdr->get_array();
if (arr.size() != ArrayMesh::ARRAY_MAX || arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
} else if (selection_mode == SELECTION_MODE_EDGE) {
disable_change_event();
Array orig_arr = copy_arrays(arr);
PoolVector3Array voa = orig_arr[ArrayMesh::ARRAY_VERTEX];
int original_size = voa.size();
PoolIntArray::Read r = _selected_points.read();
for (int i = 0; i < _selected_points.size(); ++i) {
add_quad_to_edge(r[i]);
}
r.release();
arr = _mdr->get_array();
// Note: This algorithm depends heavily depends on the inner workings of add_quad_to_edge!
PoolVector3Array new_verts = arr[ArrayMesh::ARRAY_VERTEX];
// every 4 vertex is a quad
// 1 ---- 2
// | |
// | |
// 0 ---- 3
// vertex 1, and 2 are the created new ones, 0, and 3 are duplicated from the original edge
// Don't reallocate it every time
PoolIntArray found_verts;
// Go through every new created 0th vertex
for (int i = original_size; i < new_verts.size(); i += 4) {
Vector3 v0 = new_verts[i];
found_verts.resize(0);
// Find a pair for it (has to be the 3th).
for (int j = original_size; j < new_verts.size(); j += 4) {
if (i == j) {
continue;
}
// +3 offset to 3rd vert
Vector3 v3 = new_verts[j + 3];
if (is_verts_equal(v0, v3)) {
// +2 offset to 2rd vert
found_verts.append(j + 2);
}
}
if (found_verts.size() == 0) {
continue;
}
// Also append the first vertex index to simplify logic
found_verts.append(i + 1);
// Calculate avg
Vector3 vavg;
for (int ic = 0; ic < found_verts.size(); ++ic) {
vavg += new_verts[found_verts[ic]];
}
vavg /= found_verts.size();
// set back
for (int ic = 0; ic < found_verts.size(); ++ic) {
new_verts.set(found_verts[ic], vavg);
}
}
arr[ArrayMesh::ARRAY_VERTEX] = new_verts;
_mdr->set_array(arr);
_selected_points.resize(0);
add_mesh_change_undo_redo(orig_arr, arr, "Extrude");
enable_change_event();
// The selection alo will take care of the duplicates
PoolVector3Array new_handle_points;
for (int i = original_size; i < new_verts.size(); i += 4) {
Vector3 vavg = new_verts[i + 1];
vavg += new_verts[i + 2];
vavg /= 2;
new_handle_points.append(vavg);
}
select_handle_points(new_handle_points);
} else {
add_quad();
}
}
void MDIGizmo::add_box() {
if (_mdr.is_valid()) {
Array orig_arr = copy_arrays(_mdr->get_array());
MDREDMeshUtils::add_box(_mdr);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Add Box");
}
}
void MDIGizmo::split() {
}
void MDIGizmo::disconnect_action() {
}
int MDIGizmo::get_first_triangle_index_for_vertex(int indx) {
PoolIntArray::Read r = _indices.read();
for (int i = 0; i < _indices.size(); ++i) {
if (r[i] == indx) {
return i / 3;
}
}
return -1;
}
void MDIGizmo::create_face() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() <= 2) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolVector3Array points;
for (int i = 0; i < _selected_points.size(); ++i) {
points.push_back(_handle_points[_selected_points[i]]);
}
if (points.size() == 3) {
int i0 = _handle_to_vertex_map[_selected_points[0]][0];
int i1 = _handle_to_vertex_map[_selected_points[1]][0];
int i2 = _handle_to_vertex_map[_selected_points[2]][0];
Vector3 v0 = points[0];
Vector3 v1 = points[1];
Vector3 v2 = points[2];
Vector3 tfn;
PoolVector3Array orig_arr_normals = orig_arr[ArrayMesh::ARRAY_NORMAL];
PoolVector3Array orig_arr_verts = orig_arr[ArrayMesh::ARRAY_VERTEX];
if (!orig_arr[ArrayMesh::ARRAY_NORMAL].is_null() && orig_arr_normals.size() == orig_arr_verts.size()) {
PoolVector3Array normals = orig_arr[ArrayMesh::ARRAY_NORMAL];
tfn += normals[i0];
tfn += normals[i1];
tfn += normals[i2];
tfn /= 3;
tfn = tfn.normalized();
} else {
tfn = MDREDMeshUtils::get_face_normal(_vertices[i0], _vertices[i1], _vertices[i2]);
}
bool flip = !MDREDMeshUtils::should_triangle_flip(v0, v1, v2, tfn);
MDREDMeshUtils::add_triangle_at(_mdr, v0, v1, v2, flip);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Create Face");
enable_change_event();
return;
}
if (!MDREDMeshUtils::add_triangulated_mesh_from_points_delaunay(_mdr, points, _last_known_camera_facing)) {
enable_change_event();
return;
}
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Create Face");
//_selected_points.resize(0)
enable_change_event();
} else if (selection_mode == SELECTION_MODE_EDGE) {
} else if (selection_mode == SELECTION_MODE_FACE) {
}
}
Vector<PoolIntArray> MDIGizmo::split_face_indices(int face) {
Vector<PoolIntArray> ret;
PoolIntArray ps = _handle_to_vertex_map[face];
if (ps.size() == 0) {
return ret;
}
Vector3 v0 = _vertices[ps[0]];
Vector3 v1;
bool v1found = false;
PoolIntArray v0ei;
v0ei.append(ps[0]);
PoolIntArray v1ei;
PoolIntArray v2ei;
PoolIntArray::Read r = ps.read();
for (int i = 1; i < ps.size(); ++i) {
Vector3 vert = _vertices[ps[i]];
if (is_verts_equal(v0, vert)) {
v0ei.append(ps[i]);
} else {
if (v1found) {
if (is_verts_equal(v1, vert)) {
v1ei.append(ps[i]);
} else {
v2ei.append(ps[i]);
}
} else {
v1found = true;
v1 = _vertices[ps[i]];
v1ei.append(ps[i]);
}
}
}
ret.push_back(v0ei);
ret.push_back(v1ei);
ret.push_back(v2ei);
return ret;
}
int MDIGizmo::find_first_triangle_index_for_face(int face) {
Vector<PoolIntArray> split_indices_arr = split_face_indices(face);
if (split_indices_arr.size() == 0) {
return -1;
}
PoolIntArray v0ei = split_indices_arr[0];
PoolIntArray v1ei = split_indices_arr[1];
PoolIntArray v2ei = split_indices_arr[2];
//int tri_index = -1;
for (int i = 0; i < _indices.size(); i += 3) {
int i0 = _indices[i];
int i1 = _indices[i + 1];
int i2 = _indices[i + 2];
if (pool_int_arr_contains(v0ei, i0) || pool_int_arr_contains(v0ei, i1) || pool_int_arr_contains(v0ei, i2)) {
if (pool_int_arr_contains(v1ei, i0) || pool_int_arr_contains(v1ei, i1) || pool_int_arr_contains(v1ei, i2)) {
if (pool_int_arr_contains(v2ei, i0) || pool_int_arr_contains(v2ei, i1) || pool_int_arr_contains(v2ei, i2)) {
return i / 3;
}
}
}
}
return -1;
}
void MDIGizmo::delete_selected() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() == 0) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
} else if (selection_mode == SELECTION_MODE_EDGE) {
} else if (selection_mode == SELECTION_MODE_FACE) {
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
Array triangle_indexes;
for (int i = 0; i < _selected_points.size(); ++i) {
int triangle_index = find_first_triangle_index_for_face(_selected_points[i]);
triangle_indexes.append(triangle_index);
}
//delete in reverse triangle index order
triangle_indexes.sort();
for (int i = triangle_indexes.size() - 1; i >= 0; --i) {
int triangle_index = triangle_indexes[i];
MDREDMeshUtils::remove_triangle(_mdr, triangle_index);
}
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Delete");
_selected_points.resize(0);
enable_change_event();
}
}
void MDIGizmo::generate_normals() {
if (!_mdr.is_valid()) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array verts = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (verts.size() == 0) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolIntArray orig_seams = copy_pool_int_array(_mdr->get_seams());
PoolVector3Array seam_points = MDREDMeshUtils::seams_to_points(_mdr);
MDREDMeshUtils::generate_normals_mdr(_mdr);
MDREDMeshUtils::points_to_seams(_mdr, seam_points);
add_mesh_seam_change_undo_redo(orig_arr, orig_seams, _mdr->get_array(), _mdr->get_seams(), "Generate Normals");
enable_change_event();
}
void MDIGizmo::generate_tangents() {
if (!_mdr.is_valid()) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array verts = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (verts.size() == 0) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolIntArray orig_seams = copy_pool_int_array(_mdr->get_seams());
PoolVector3Array seam_points = MDREDMeshUtils::seams_to_points(_mdr);
MDREDMeshUtils::generate_tangents(_mdr);
MDREDMeshUtils::points_to_seams(_mdr, seam_points);
add_mesh_seam_change_undo_redo(orig_arr, orig_seams, _mdr->get_array(), _mdr->get_seams(), "Generate Tangents");
enable_change_event();
}
void MDIGizmo::remove_doubles() {
if (!_mdr.is_valid()) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array verts = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (verts.size() == 0) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolIntArray orig_seams = copy_pool_int_array(_mdr->get_seams());
PoolVector3Array seam_points = MDREDMeshUtils::seams_to_points(_mdr);
Array merged_arrays = MeshUtils::get_singleton()->remove_doubles(mdr_arr);
_mdr->set_array(merged_arrays);
MDREDMeshUtils::points_to_seams(_mdr, seam_points);
add_mesh_seam_change_undo_redo(orig_arr, orig_seams, _mdr->get_array(), _mdr->get_seams(), "Remove Doubles");
enable_change_event();
}
void MDIGizmo::merge_optimize() {
if (!_mdr.is_valid()) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array verts = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (verts.size() == 0) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolIntArray orig_seams = copy_pool_int_array(_mdr->get_seams());
PoolVector3Array seam_points = MDREDMeshUtils::seams_to_points(_mdr);
Array merged_arrays = MeshUtils::get_singleton()->merge_mesh_array(mdr_arr);
_mdr->set_array(merged_arrays);
MDREDMeshUtils::points_to_seams(_mdr, seam_points);
add_mesh_seam_change_undo_redo(orig_arr, orig_seams, _mdr->get_array(), _mdr->get_seams(), "Merge Optimize");
enable_change_event();
}
void MDIGizmo::connect_to_first_selected() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() < 2) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array vertices = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (vertices.size() == 0) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
if (selection_mode == SELECTION_MODE_VERTEX) {
Vector3 mpos = _handle_points[_selected_points[0]];
for (int i = 1; i < _selected_points.size(); ++i) {
PoolIntArray ps = _handle_to_vertex_map[_selected_points[i]];
for (int j = 0; j < ps.size(); ++j) {
vertices.set(ps[j], mpos);
}
}
_selected_points.resize(0);
mdr_arr[ArrayMesh::ARRAY_VERTEX] = vertices;
_mdr->set_array(mdr_arr);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Connect to first selected");
enable_change_event();
} else if (selection_mode == SELECTION_MODE_EDGE) {
} else if (selection_mode == SELECTION_MODE_FACE) {
}
}
void MDIGizmo::connect_to_avg() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() < 2) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array vertices = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (vertices.size() == 0) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
if (selection_mode == SELECTION_MODE_VERTEX) {
Vector3 mpos;
for (int i = 0; i < _selected_points.size(); ++i) {
mpos += _handle_points[_selected_points[i]];
}
mpos /= _selected_points.size();
for (int i = 0; i < _selected_points.size(); ++i) {
PoolIntArray ps = _handle_to_vertex_map[_selected_points[i]];
for (int j = 0; j < ps.size(); ++j) {
vertices.set(ps[j], mpos);
}
}
_selected_points.resize(0);
mdr_arr[ArrayMesh::ARRAY_VERTEX] = vertices;
_mdr->set_array(mdr_arr);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Connect to average");
enable_change_event();
} else if (selection_mode == SELECTION_MODE_EDGE) {
} else if (selection_mode == SELECTION_MODE_FACE) {
}
}
void MDIGizmo::connect_to_last_selected() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() < 2) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array vertices = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (vertices.size() == 0) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
if (selection_mode == SELECTION_MODE_VERTEX) {
Vector3 mpos = _handle_points[_selected_points[_selected_points.size() - 1]];
for (int i = 0; i < _selected_points.size(); ++i) {
PoolIntArray ps = _handle_to_vertex_map[_selected_points[i]];
for (int j = 0; j < ps.size(); ++j) {
vertices.set(ps[j], mpos);
}
}
_selected_points.resize(0);
mdr_arr[ArrayMesh::ARRAY_VERTEX] = vertices;
_mdr->set_array(mdr_arr);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Connect to last selected");
enable_change_event();
} else if (selection_mode == SELECTION_MODE_EDGE) {
} else if (selection_mode == SELECTION_MODE_FACE) {
}
}
PoolIntArray MDIGizmo::get_first_index_pair_for_edge(int edge) {
PoolIntArray ret;
Vector<PoolIntArray> eisarr = split_edge_indices(edge);
if (eisarr.size() == 0) {
return ret;
}
// these should have the same size
PoolIntArray v0ei = eisarr[0];
PoolIntArray v1ei = eisarr[1];
for (int i = 0; i < _indices.size(); i += 3) {
int i0 = _indices[i];
int i1 = _indices[i + 1];
int i2 = _indices[i + 2];
if (pool_int_arr_contains(v0ei, i0) || pool_int_arr_contains(v0ei, i1) || pool_int_arr_contains(v0ei, i2)) {
if (pool_int_arr_contains(v1ei, i0) || pool_int_arr_contains(v1ei, i1) || pool_int_arr_contains(v1ei, i2)) {
if (pool_int_arr_contains(v0ei, i0)) {
ret.push_back(i0);
} else if (pool_int_arr_contains(v0ei, i1)) {
ret.push_back(i1);
} else if (pool_int_arr_contains(v0ei, i2)) {
ret.push_back(i2);
}
if (pool_int_arr_contains(v1ei, i0)) {
ret.push_back(i0);
} else if (pool_int_arr_contains(v1ei, i1)) {
ret.push_back(i1);
} else if (pool_int_arr_contains(v1ei, i2)) {
ret.push_back(i2);
}
return ret;
}
}
}
return ret;
}
PoolIntArray MDIGizmo::get_all_index_pairs_for_edge(int edge) {
PoolIntArray ret;
Vector<PoolIntArray> eisarr = split_edge_indices(edge);
if (eisarr.size() == 0) {
return ret;
}
// these should have the same size
PoolIntArray v0ei = eisarr[0];
PoolIntArray v1ei = eisarr[1];
for (int i = 0; i < _indices.size(); i += 3) {
int i0 = _indices[i];
int i1 = _indices[i + 1];
int i2 = _indices[i + 2];
if (pool_int_arr_contains(v0ei, i0) || pool_int_arr_contains(v0ei, i1) || pool_int_arr_contains(v0ei, i2)) {
if (pool_int_arr_contains(v1ei, i0) || pool_int_arr_contains(v1ei, i1) || pool_int_arr_contains(v1ei, i2)) {
if (pool_int_arr_contains(v0ei, i0)) {
ret.push_back(i0);
} else if (pool_int_arr_contains(v0ei, i1)) {
ret.push_back(i1);
} else if (pool_int_arr_contains(v0ei, i2)) {
ret.push_back(i2);
}
if (pool_int_arr_contains(v1ei, i0)) {
ret.push_back(i0);
} else if (pool_int_arr_contains(v1ei, i1)) {
ret.push_back(i1);
} else if (pool_int_arr_contains(v1ei, i2)) {
ret.push_back(i2);
}
}
}
}
return ret;
}
void MDIGizmo::mark_seam() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() == 0) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
} else if (selection_mode == SELECTION_MODE_EDGE) {
disable_change_event();
PoolIntArray prev_seams = copy_pool_int_array(_mdr->get_seams());
for (int i = 0; i < _selected_points.size(); ++i) { // se in _selected_points:
int se = _selected_points[i];
PoolIntArray eis = MDREDMeshUtils::order_seam_indices(get_first_index_pair_for_edge(se));
if (eis.size() == 0) {
continue;
}
MDREDMeshUtils::add_seam(_mdr, eis[0], eis[1]);
}
_undo_redo->create_action("mark_seam");
_undo_redo->add_do_method(this, "set_seam", _mdr, copy_pool_int_array(_mdr->get_seams()));
_undo_redo->add_undo_method(this, "set_seam", _mdr, prev_seams);
_undo_redo->commit_action();
enable_change_event();
} else if (selection_mode == SELECTION_MODE_FACE) {
}
}
void MDIGizmo::unmark_seam() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() == 0) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
} else if (selection_mode == SELECTION_MODE_EDGE) {
disable_change_event();
PoolIntArray prev_seams = copy_pool_int_array(_mdr->get_seams());
for (int i = 0; i < _selected_points.size(); ++i) { // se in _selected_points:
int se = _selected_points[i];
PoolIntArray eis = MDREDMeshUtils::order_seam_indices(get_first_index_pair_for_edge(se));
if (eis.size() == 0) {
continue;
}
MDREDMeshUtils::remove_seam(_mdr, eis[0], eis[1]);
}
_undo_redo->create_action("unmark_seam");
_undo_redo->add_do_method(this, "set_seam", _mdr, copy_pool_int_array(_mdr->get_seams()));
_undo_redo->add_undo_method(this, "set_seam", _mdr, prev_seams);
_undo_redo->commit_action();
enable_change_event();
} else if (selection_mode == SELECTION_MODE_FACE) {
}
}
void MDIGizmo::set_seam(Ref<MeshDataResource> mdr, PoolIntArray arr) {
mdr->set_seams(arr);
}
void MDIGizmo::apply_seam() {
if (!_mdr.is_valid()) {
return;
}
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
MDREDMeshUtils::apply_seam(_mdr);
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "apply_seam");
enable_change_event();
}
void MDIGizmo::clean_mesh() {
if (!_mdr.is_valid()) {
return;
}
Array arrays = _mdr->get_array();
if (arrays.size() != ArrayMesh::ARRAY_MAX || arrays[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array vertices = arrays[ArrayMesh::ARRAY_VERTEX];
if (vertices.size() == 0) {
return;
}
//int old_vert_size = vertices.size();
disable_change_event();
Array orig_arr = copy_arrays(arrays);
arrays = MDREDMeshUtils::remove_used_vertices(arrays);
//int new_vert_size = arrays[ArrayMesh::ARRAY_VERTEX].size();
add_mesh_change_undo_redo(orig_arr, arrays, "clean_mesh");
enable_change_event();
//var d : int = old_vert_size - new_vert_size;
//print("MDRED: Removed " + str(d) + " unused vertices.")
}
void MDIGizmo::uv_unwrap() {
if (!_mdr.is_valid()) {
return;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return;
}
PoolVector3Array verts = mdr_arr[ArrayMesh::ARRAY_VERTEX];
if (verts.size() == 0) {
return;
}
disable_change_event();
PoolVector2Array uvs = MeshUtils::get_singleton()->uv_unwrap(mdr_arr);
if (uvs.size() != verts.size()) {
ERR_PRINT("Error: Could not unwrap mesh!");
enable_change_event(false);
return;
}
Array orig_arr = copy_arrays(mdr_arr);
mdr_arr[ArrayMesh::ARRAY_TEX_UV] = uvs;
add_mesh_change_undo_redo(orig_arr, mdr_arr, "uv_unwrap");
enable_change_event();
}
void MDIGizmo::flip_selected_faces() {
if (!_mdr.is_valid()) {
return;
}
if (_selected_points.size() == 0) {
return;
}
if (selection_mode == SELECTION_MODE_VERTEX) {
} else if (selection_mode == SELECTION_MODE_EDGE) {
} else if (selection_mode == SELECTION_MODE_FACE) {
disable_change_event();
Array orig_arr = copy_arrays(_mdr->get_array());
PoolIntArray::Read r = _selected_points.read();
for (int i = 0; i < _selected_points.size(); ++i) {
int sp = r[i];
int triangle_index = find_first_triangle_index_for_face(sp);
MDREDMeshUtils::flip_triangle_ti(_mdr, triangle_index);
}
add_mesh_change_undo_redo(orig_arr, _mdr->get_array(), "Flip Faces");
enable_change_event();
}
}
void MDIGizmo::add_mesh_change_undo_redo(const Array &orig_arr, const Array &new_arr, const String &action_name) {
_undo_redo->create_action(action_name);
Array nac = copy_arrays(new_arr);
_undo_redo->add_do_method(this, "apply_mesh_change", _mdr, nac);
_undo_redo->add_undo_method(this, "apply_mesh_change", _mdr, orig_arr);
_undo_redo->commit_action();
}
void MDIGizmo::add_mesh_seam_change_undo_redo(const Array &orig_arr, const PoolIntArray &orig_seams, const Array &new_arr, const PoolIntArray &new_seams, const String &action_name) {
_undo_redo->create_action(action_name);
Array nac = copy_arrays(new_arr);
_undo_redo->add_do_method(this, "apply_mesh_change", _mdr, nac);
_undo_redo->add_undo_method(this, "apply_mesh_change", _mdr, orig_arr);
_undo_redo->add_do_method(this, "set_seam", _mdr, copy_pool_int_array(new_seams));
_undo_redo->add_undo_method(this, "set_seam", _mdr, orig_seams);
_undo_redo->commit_action();
}
void MDIGizmo::apply_mesh_change(Ref<MeshDataResource> mdr, const Array &arr) {
if (!mdr.is_valid()) {
return;
}
mdr->set_array(copy_arrays(arr));
}
void MDIGizmo::apply_vertex_array(Ref<MeshDataResource> mdr, const PoolVector3Array &verts) {
if (!mdr.is_valid()) {
return;
}
Array mdr_arr = mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX) {
return;
}
mdr_arr[ArrayMesh::ARRAY_VERTEX] = verts;
mdr->set_array(mdr_arr);
}
Array MDIGizmo::copy_arrays(const Array &arr) {
return arr.duplicate(true);
}
PoolIntArray MDIGizmo::copy_pool_int_array(const PoolIntArray &pia) {
PoolIntArray ret;
ret.resize(pia.size());
PoolIntArray::Read r = pia.read();
PoolIntArray::Write w = ret.write();
for (int i = 0; i < pia.size(); ++i) {
w[i] = r[i];
}
r.release();
w.release();
return ret;
}
PoolVector3Array MDIGizmo::copy_mdr_verts_array() {
PoolVector3Array ret;
if (!_mdr.is_valid()) {
return ret;
}
Array mdr_arr = _mdr->get_array();
if (mdr_arr.size() != ArrayMesh::ARRAY_MAX || mdr_arr[ArrayMesh::ARRAY_VERTEX].is_null()) {
return ret;
}
PoolVector3Array vertices = mdr_arr[ArrayMesh::ARRAY_VERTEX];
ret.append_array(vertices);
return ret;
}
void MDIGizmo::setup_op_drag_indices() {
_drag_op_indices.resize(0);
PoolIntArray::Read r = _selected_points.read();
for (int i = 0; i < _selected_points.size(); ++i) {
int sp = r[i];
PoolIntArray pi = _handle_to_vertex_map[sp];
PoolIntArray::Read pir = pi.read();
for (int j = 0; j < pi.size(); ++j) {
int indx = pir[j];
if (!pool_int_arr_contains(_drag_op_indices, indx)) {
_drag_op_indices.append(indx);
}
}
pir.release();
}
}
Vector3 MDIGizmo::get_drag_op_pivot() {
if (pivot_type == PIVOT_TYPE_AVERAGED) {
Vector3 avg = Vector3();
PoolIntArray::Read r = _drag_op_indices.read();
for (int i = 0; i < _drag_op_indices.size(); ++i) {
avg += _vertices[r[i]];
}
r.release();
avg /= _drag_op_indices.size();
return avg;
} else if (pivot_type == PIVOT_TYPE_MDI_ORIGIN) {
return Vector3();
} else if (pivot_type == PIVOT_TYPE_WORLD_ORIGIN) {
return get_spatial_node()->to_local(Vector3());
}
return Vector3();
}
void MDIGizmo::select_handle_points(const PoolVector3Array &points) {
_selected_points.resize(0);
PoolVector3Array::Read r = points.read();
for (int ip = 0; ip < points.size(); ++ip) {
Vector3 p = r[ip];
PoolVector3Array::Read hpr = _handle_points.read();
for (int i = 0; i < _handle_points.size(); ++i) {
if (is_verts_equal(p, hpr[i])) {
if (!pool_int_arr_contains(_selected_points, i)) {
_selected_points.push_back(i);
}
}
}
hpr.release();
}
redraw();
}
void MDIGizmo::set_pivot_averaged() {
pivot_type = PIVOT_TYPE_AVERAGED;
}
void MDIGizmo::set_pivot_mdi_origin() {
pivot_type = PIVOT_TYPE_MDI_ORIGIN;
}
void MDIGizmo::set_pivot_world_origin() {
pivot_type = PIVOT_TYPE_WORLD_ORIGIN;
}
void MDIGizmo::transfer_state_from(const Ref<MDIGizmo> &other) {
edit_mode = other->edit_mode;
pivot_type = other->pivot_type;
axis_constraint = other->axis_constraint;
selection_mode = other->selection_mode;
handle_selection_type = other->handle_selection_type;
visual_indicator_outline = other->visual_indicator_outline;
visual_indicator_seam = other->visual_indicator_seam;
visual_indicator_handle = other->visual_indicator_handle;
}
void MDIGizmo::visual_indicator_outline_set(bool on) {
visual_indicator_outline = on;
redraw();
}
void MDIGizmo::visual_indicator_seam_set(bool on) {
visual_indicator_seam = on;
redraw();
}
void MDIGizmo::visual_indicator_handle_set(bool on) {
visual_indicator_handle = on;
redraw();
}
void MDIGizmo::handle_selection_type_front() {
handle_selection_type = HANDLE_SELECTION_TYPE_FRONT;
}
void MDIGizmo::handle_selection_type_back() {
handle_selection_type = HANDLE_SELECTION_TYPE_BACK;
}
void MDIGizmo::handle_selection_type_all() {
handle_selection_type = HANDLE_SELECTION_TYPE_ALL;
}
MDIGizmo::MDIGizmo() {
gizmo_size = 3.0;
edit_mode = EDIT_MODE_TRANSLATE;
pivot_type = PIVOT_TYPE_AVERAGED;
axis_constraint = AXIS_CONSTRAINT_X | AXIS_CONSTRAINT_Y | AXIS_CONSTRAINT_Z;
selection_mode = SELECTION_MODE_VERTEX;
handle_selection_type = HANDLE_SELECTION_TYPE_FRONT;
visual_indicator_outline = true;
visual_indicator_seam = true;
visual_indicator_handle = true;
_last_known_camera_facing = Vector3(0, 0, -1);
_rect_drag = false;
_rect_drag_min_ofset = 10;
_mesh_outline_generator.instance();
_handle_drag_op = false;
_editor_plugin = nullptr;
_undo_redo = nullptr;
_visible = false;
}
MDIGizmo::~MDIGizmo() {
}
void MDIGizmo::_bind_methods() {
ClassDB::bind_method(D_METHOD("on_mdr_changed"), &MDIGizmo::on_mdr_changed);
ClassDB::bind_method(D_METHOD("on_mesh_data_resource_changed"), &MDIGizmo::on_mesh_data_resource_changed);
ClassDB::bind_method(D_METHOD("apply_mesh_change"), &MDIGizmo::apply_mesh_change);
ClassDB::bind_method(D_METHOD("set_seam"), &MDIGizmo::set_seam);
ClassDB::bind_method(D_METHOD("apply_vertex_array"), &MDIGizmo::apply_vertex_array);
}