mirror of
https://github.com/godotengine/godot.git
synced 2024-12-09 10:09:20 +08:00
f3a6e04f8b
Fixes potential crashes in navmesh geometry functions.
192 lines
8.3 KiB
C++
192 lines
8.3 KiB
C++
/**************************************************************************/
|
|
/* navigation_mesh_source_geometry_data_3d.cpp */
|
|
/**************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/**************************************************************************/
|
|
/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
|
|
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
|
|
/* */
|
|
/* 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 "navigation_mesh_source_geometry_data_3d.h"
|
|
|
|
void NavigationMeshSourceGeometryData3D::set_vertices(const Vector<float> &p_vertices) {
|
|
vertices = p_vertices;
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::set_indices(const Vector<int> &p_indices) {
|
|
indices = p_indices;
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::clear() {
|
|
vertices.clear();
|
|
indices.clear();
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::_add_vertex(const Vector3 &p_vec3) {
|
|
vertices.push_back(p_vec3.x);
|
|
vertices.push_back(p_vec3.y);
|
|
vertices.push_back(p_vec3.z);
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::_add_mesh(const Ref<Mesh> &p_mesh, const Transform3D &p_xform) {
|
|
int current_vertex_count;
|
|
for (int i = 0; i < p_mesh->get_surface_count(); i++) {
|
|
current_vertex_count = vertices.size() / 3;
|
|
|
|
if (p_mesh->surface_get_primitive_type(i) != Mesh::PRIMITIVE_TRIANGLES) {
|
|
continue;
|
|
}
|
|
|
|
int index_count = 0;
|
|
if (p_mesh->surface_get_format(i) & Mesh::ARRAY_FORMAT_INDEX) {
|
|
index_count = p_mesh->surface_get_array_index_len(i);
|
|
} else {
|
|
index_count = p_mesh->surface_get_array_len(i);
|
|
}
|
|
|
|
ERR_CONTINUE((index_count == 0 || (index_count % 3) != 0));
|
|
|
|
int face_count = index_count / 3;
|
|
|
|
Array a = p_mesh->surface_get_arrays(i);
|
|
ERR_CONTINUE(a.is_empty() || (a.size() != Mesh::ARRAY_MAX));
|
|
|
|
Vector<Vector3> mesh_vertices = a[Mesh::ARRAY_VERTEX];
|
|
ERR_CONTINUE(mesh_vertices.is_empty());
|
|
const Vector3 *vr = mesh_vertices.ptr();
|
|
|
|
if (p_mesh->surface_get_format(i) & Mesh::ARRAY_FORMAT_INDEX) {
|
|
Vector<int> mesh_indices = a[Mesh::ARRAY_INDEX];
|
|
ERR_CONTINUE(mesh_indices.is_empty() || (mesh_indices.size() != index_count));
|
|
const int *ir = mesh_indices.ptr();
|
|
|
|
for (int j = 0; j < mesh_vertices.size(); j++) {
|
|
_add_vertex(p_xform.xform(vr[j]));
|
|
}
|
|
|
|
for (int j = 0; j < face_count; j++) {
|
|
// CCW
|
|
indices.push_back(current_vertex_count + (ir[j * 3 + 0]));
|
|
indices.push_back(current_vertex_count + (ir[j * 3 + 2]));
|
|
indices.push_back(current_vertex_count + (ir[j * 3 + 1]));
|
|
}
|
|
} else {
|
|
ERR_CONTINUE(mesh_vertices.size() != index_count);
|
|
face_count = mesh_vertices.size() / 3;
|
|
for (int j = 0; j < face_count; j++) {
|
|
_add_vertex(p_xform.xform(vr[j * 3 + 0]));
|
|
_add_vertex(p_xform.xform(vr[j * 3 + 2]));
|
|
_add_vertex(p_xform.xform(vr[j * 3 + 1]));
|
|
|
|
indices.push_back(current_vertex_count + (j * 3 + 0));
|
|
indices.push_back(current_vertex_count + (j * 3 + 1));
|
|
indices.push_back(current_vertex_count + (j * 3 + 2));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::_add_mesh_array(const Array &p_mesh_array, const Transform3D &p_xform) {
|
|
ERR_FAIL_COND(p_mesh_array.size() != Mesh::ARRAY_MAX);
|
|
|
|
Vector<Vector3> mesh_vertices = p_mesh_array[Mesh::ARRAY_VERTEX];
|
|
ERR_FAIL_COND(mesh_vertices.is_empty());
|
|
const Vector3 *vr = mesh_vertices.ptr();
|
|
|
|
Vector<int> mesh_indices = p_mesh_array[Mesh::ARRAY_INDEX];
|
|
ERR_FAIL_COND(mesh_indices.is_empty());
|
|
const int *ir = mesh_indices.ptr();
|
|
|
|
const int face_count = mesh_indices.size() / 3;
|
|
const int current_vertex_count = vertices.size() / 3;
|
|
|
|
for (int j = 0; j < mesh_vertices.size(); j++) {
|
|
_add_vertex(p_xform.xform(vr[j]));
|
|
}
|
|
|
|
for (int j = 0; j < face_count; j++) {
|
|
// CCW
|
|
indices.push_back(current_vertex_count + (ir[j * 3 + 0]));
|
|
indices.push_back(current_vertex_count + (ir[j * 3 + 2]));
|
|
indices.push_back(current_vertex_count + (ir[j * 3 + 1]));
|
|
}
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::_add_faces(const PackedVector3Array &p_faces, const Transform3D &p_xform) {
|
|
ERR_FAIL_COND(p_faces.is_empty());
|
|
ERR_FAIL_COND(p_faces.size() % 3 != 0);
|
|
int face_count = p_faces.size() / 3;
|
|
int current_vertex_count = vertices.size() / 3;
|
|
|
|
for (int j = 0; j < face_count; j++) {
|
|
_add_vertex(p_xform.xform(p_faces[j * 3 + 0]));
|
|
_add_vertex(p_xform.xform(p_faces[j * 3 + 1]));
|
|
_add_vertex(p_xform.xform(p_faces[j * 3 + 2]));
|
|
|
|
indices.push_back(current_vertex_count + (j * 3 + 0));
|
|
indices.push_back(current_vertex_count + (j * 3 + 2));
|
|
indices.push_back(current_vertex_count + (j * 3 + 1));
|
|
}
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::add_mesh(const Ref<Mesh> &p_mesh, const Transform3D &p_xform) {
|
|
ERR_FAIL_COND(!p_mesh.is_valid());
|
|
_add_mesh(p_mesh, root_node_transform * p_xform);
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::add_mesh_array(const Array &p_mesh_array, const Transform3D &p_xform) {
|
|
ERR_FAIL_COND(p_mesh_array.size() != Mesh::ARRAY_MAX);
|
|
_add_mesh_array(p_mesh_array, root_node_transform * p_xform);
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::add_faces(const PackedVector3Array &p_faces, const Transform3D &p_xform) {
|
|
ERR_FAIL_COND(p_faces.size() % 3 != 0);
|
|
_add_faces(p_faces, root_node_transform * p_xform);
|
|
}
|
|
|
|
void NavigationMeshSourceGeometryData3D::_bind_methods() {
|
|
ClassDB::bind_method(D_METHOD("set_vertices", "vertices"), &NavigationMeshSourceGeometryData3D::set_vertices);
|
|
ClassDB::bind_method(D_METHOD("get_vertices"), &NavigationMeshSourceGeometryData3D::get_vertices);
|
|
|
|
ClassDB::bind_method(D_METHOD("set_indices", "indices"), &NavigationMeshSourceGeometryData3D::set_indices);
|
|
ClassDB::bind_method(D_METHOD("get_indices"), &NavigationMeshSourceGeometryData3D::get_indices);
|
|
|
|
ClassDB::bind_method(D_METHOD("clear"), &NavigationMeshSourceGeometryData3D::clear);
|
|
ClassDB::bind_method(D_METHOD("has_data"), &NavigationMeshSourceGeometryData3D::has_data);
|
|
|
|
ClassDB::bind_method(D_METHOD("add_mesh", "mesh", "xform"), &NavigationMeshSourceGeometryData3D::add_mesh);
|
|
ClassDB::bind_method(D_METHOD("add_mesh_array", "mesh_array", "xform"), &NavigationMeshSourceGeometryData3D::add_mesh_array);
|
|
ClassDB::bind_method(D_METHOD("add_faces", "faces", "xform"), &NavigationMeshSourceGeometryData3D::add_faces);
|
|
|
|
ADD_PROPERTY(PropertyInfo(Variant::PACKED_VECTOR3_ARRAY, "vertices", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR | PROPERTY_USAGE_INTERNAL), "set_vertices", "get_vertices");
|
|
ADD_PROPERTY(PropertyInfo(Variant::PACKED_INT32_ARRAY, "indices", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR | PROPERTY_USAGE_INTERNAL), "set_indices", "get_indices");
|
|
}
|
|
|
|
NavigationMeshSourceGeometryData3D::NavigationMeshSourceGeometryData3D() {
|
|
}
|
|
|
|
NavigationMeshSourceGeometryData3D::~NavigationMeshSourceGeometryData3D() {
|
|
clear();
|
|
}
|