/**************************************************************************/ /* surface_tool.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 "surface_tool.h" #include "core/method_bind_ext.gen.inc" #define _VERTEX_SNAP 0.0001 #define EQ_VERTEX_DIST 0.00001 bool SurfaceTool::Vertex::operator==(const Vertex &p_vertex) const { if (vertex != p_vertex.vertex) { return false; } if (uv != p_vertex.uv) { return false; } if (uv2 != p_vertex.uv2) { return false; } if (normal != p_vertex.normal) { return false; } if (binormal != p_vertex.binormal) { return false; } if (color != p_vertex.color) { return false; } if (bones.size() != p_vertex.bones.size()) { return false; } for (int i = 0; i < bones.size(); i++) { if (bones[i] != p_vertex.bones[i]) { return false; } } for (int i = 0; i < weights.size(); i++) { if (weights[i] != p_vertex.weights[i]) { return false; } } return true; } uint32_t SurfaceTool::VertexHasher::hash(const Vertex &p_vtx) { uint32_t h = hash_djb2_buffer((const uint8_t *)&p_vtx.vertex, sizeof(real_t) * 3); h = hash_djb2_buffer((const uint8_t *)&p_vtx.normal, sizeof(real_t) * 3, h); h = hash_djb2_buffer((const uint8_t *)&p_vtx.binormal, sizeof(real_t) * 3, h); h = hash_djb2_buffer((const uint8_t *)&p_vtx.tangent, sizeof(real_t) * 3, h); h = hash_djb2_buffer((const uint8_t *)&p_vtx.uv, sizeof(real_t) * 2, h); h = hash_djb2_buffer((const uint8_t *)&p_vtx.uv2, sizeof(real_t) * 2, h); h = hash_djb2_buffer((const uint8_t *)&p_vtx.color, sizeof(real_t) * 4, h); h = hash_djb2_buffer((const uint8_t *)p_vtx.bones.ptr(), p_vtx.bones.size() * sizeof(int), h); h = hash_djb2_buffer((const uint8_t *)p_vtx.weights.ptr(), p_vtx.weights.size() * sizeof(float), h); return h; } void SurfaceTool::begin(Mesh::PrimitiveType p_primitive) { clear(); primitive = p_primitive; begun = true; first = true; } void SurfaceTool::add_vertex(const Vector3 &p_vertex) { ERR_FAIL_COND(!begun); Vertex vtx; vtx.vertex = p_vertex; vtx.color = last_color; vtx.normal = last_normal; vtx.uv = last_uv; vtx.uv2 = last_uv2; vtx.weights = last_weights; vtx.bones = last_bones; vtx.tangent = last_tangent.normal; vtx.binormal = last_normal.cross(last_tangent.normal).normalized() * last_tangent.d; const int expected_vertices = 4; if ((format & Mesh::ARRAY_FORMAT_WEIGHTS || format & Mesh::ARRAY_FORMAT_BONES) && (vtx.weights.size() != expected_vertices || vtx.bones.size() != expected_vertices)) { //ensure vertices are the expected amount ERR_FAIL_COND(vtx.weights.size() != vtx.bones.size()); if (vtx.weights.size() < expected_vertices) { //less than required, fill for (int i = vtx.weights.size(); i < expected_vertices; i++) { vtx.weights.push_back(0); vtx.bones.push_back(0); } } else if (vtx.weights.size() > expected_vertices) { //more than required, sort, cap and normalize. Vector weights; for (int i = 0; i < vtx.weights.size(); i++) { WeightSort ws; ws.index = vtx.bones[i]; ws.weight = vtx.weights[i]; weights.push_back(ws); } //sort weights.sort(); //cap weights.resize(expected_vertices); //renormalize float total = 0; for (int i = 0; i < expected_vertices; i++) { total += weights[i].weight; } vtx.weights.resize(expected_vertices); vtx.bones.resize(expected_vertices); for (int i = 0; i < expected_vertices; i++) { if (total > 0) { vtx.weights.write[i] = weights[i].weight / total; } else { vtx.weights.write[i] = 0; } vtx.bones.write[i] = weights[i].index; } } } vertex_array.push_back(vtx); first = false; format |= Mesh::ARRAY_FORMAT_VERTEX; } void SurfaceTool::add_color(Color p_color) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_COLOR)); format |= Mesh::ARRAY_FORMAT_COLOR; last_color = p_color; } void SurfaceTool::add_normal(const Vector3 &p_normal) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_NORMAL)); format |= Mesh::ARRAY_FORMAT_NORMAL; last_normal = p_normal; } void SurfaceTool::add_tangent(const Plane &p_tangent) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_TANGENT)); format |= Mesh::ARRAY_FORMAT_TANGENT; last_tangent = p_tangent; } void SurfaceTool::add_uv(const Vector2 &p_uv) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_TEX_UV)); format |= Mesh::ARRAY_FORMAT_TEX_UV; last_uv = p_uv; } void SurfaceTool::add_uv2(const Vector2 &p_uv2) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_TEX_UV2)); format |= Mesh::ARRAY_FORMAT_TEX_UV2; last_uv2 = p_uv2; } void SurfaceTool::add_bones(const Vector &p_bones) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_BONES)); format |= Mesh::ARRAY_FORMAT_BONES; last_bones = p_bones; } void SurfaceTool::add_weights(const Vector &p_weights) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_WEIGHTS)); format |= Mesh::ARRAY_FORMAT_WEIGHTS; last_weights = p_weights; } void SurfaceTool::add_smooth_group(bool p_smooth) { ERR_FAIL_COND(!begun); if (index_array.size()) { smooth_groups[index_array.size()] = p_smooth; } else { smooth_groups[vertex_array.size()] = p_smooth; } } void SurfaceTool::add_triangle_fan(const Vector &p_vertices, const Vector &p_uvs, const Vector &p_colors, const Vector &p_uv2s, const Vector &p_normals, const Vector &p_tangents) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(primitive != Mesh::PRIMITIVE_TRIANGLES); ERR_FAIL_COND(p_vertices.size() < 3); #define ADD_POINT(n) \ { \ if (p_colors.size() > n) \ add_color(p_colors[n]); \ if (p_uvs.size() > n) \ add_uv(p_uvs[n]); \ if (p_uv2s.size() > n) \ add_uv2(p_uv2s[n]); \ if (p_normals.size() > n) \ add_normal(p_normals[n]); \ if (p_tangents.size() > n) \ add_tangent(p_tangents[n]); \ add_vertex(p_vertices[n]); \ } for (int i = 0; i < p_vertices.size() - 2; i++) { ADD_POINT(0); ADD_POINT(i + 1); ADD_POINT(i + 2); } #undef ADD_POINT } void SurfaceTool::add_index(int p_index) { ERR_FAIL_COND(!begun); ERR_FAIL_COND(p_index < 0); format |= Mesh::ARRAY_FORMAT_INDEX; index_array.push_back(p_index); } Array SurfaceTool::commit_to_arrays() { int varr_len = vertex_array.size(); Array a; a.resize(Mesh::ARRAY_MAX); for (int i = 0; i < Mesh::ARRAY_MAX; i++) { if (!(format & (1 << i))) { continue; //not in format } switch (i) { case Mesh::ARRAY_VERTEX: case Mesh::ARRAY_NORMAL: { PoolVector array; array.resize(varr_len); PoolVector::Write w = array.write(); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next(), idx++) { const Vertex &v = E->get(); switch (i) { case Mesh::ARRAY_VERTEX: { w[idx] = v.vertex; } break; case Mesh::ARRAY_NORMAL: { w[idx] = v.normal; } break; } } w.release(); a[i] = array; } break; case Mesh::ARRAY_TEX_UV: case Mesh::ARRAY_TEX_UV2: { PoolVector array; array.resize(varr_len); PoolVector::Write w = array.write(); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next(), idx++) { const Vertex &v = E->get(); switch (i) { case Mesh::ARRAY_TEX_UV: { w[idx] = v.uv; } break; case Mesh::ARRAY_TEX_UV2: { w[idx] = v.uv2; } break; } } w.release(); a[i] = array; } break; case Mesh::ARRAY_TANGENT: { PoolVector array; array.resize(varr_len * 4); PoolVector::Write w = array.write(); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next(), idx += 4) { const Vertex &v = E->get(); w[idx + 0] = v.tangent.x; w[idx + 1] = v.tangent.y; w[idx + 2] = v.tangent.z; //float d = v.tangent.dot(v.binormal,v.normal); float d = v.binormal.dot(v.normal.cross(v.tangent)); w[idx + 3] = d < 0 ? -1 : 1; } w.release(); a[i] = array; } break; case Mesh::ARRAY_COLOR: { PoolVector array; array.resize(varr_len); PoolVector::Write w = array.write(); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next(), idx++) { const Vertex &v = E->get(); w[idx] = v.color; } w.release(); a[i] = array; } break; case Mesh::ARRAY_BONES: { PoolVector array; array.resize(varr_len * 4); PoolVector::Write w = array.write(); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next(), idx += 4) { const Vertex &v = E->get(); ERR_CONTINUE(v.bones.size() != 4); for (int j = 0; j < 4; j++) { w[idx + j] = v.bones[j]; } } w.release(); a[i] = array; } break; case Mesh::ARRAY_WEIGHTS: { PoolVector array; array.resize(varr_len * 4); PoolVector::Write w = array.write(); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next(), idx += 4) { const Vertex &v = E->get(); ERR_CONTINUE(v.weights.size() != 4); for (int j = 0; j < 4; j++) { w[idx + j] = v.weights[j]; } } w.release(); a[i] = array; } break; case Mesh::ARRAY_INDEX: { ERR_CONTINUE(index_array.size() == 0); PoolVector array; array.resize(index_array.size()); PoolVector::Write w = array.write(); int idx = 0; for (List::Element *E = index_array.front(); E; E = E->next(), idx++) { w[idx] = E->get(); } w.release(); a[i] = array; } break; default: { } } } return a; } Ref SurfaceTool::commit(const Ref &p_existing, uint32_t p_flags) { Ref mesh; if (p_existing.is_valid()) { mesh = p_existing; } else { mesh.instance(); } int varr_len = vertex_array.size(); if (varr_len == 0) { return mesh; } int surface = mesh->get_surface_count(); Array a = commit_to_arrays(); mesh->add_surface_from_arrays(primitive, a, Array(), p_flags); if (material.is_valid()) { mesh->surface_set_material(surface, material); } return mesh; } void SurfaceTool::index() { if (index_array.size()) { return; //already indexed } HashMap indices; List new_vertices; for (List::Element *E = vertex_array.front(); E; E = E->next()) { int *idxptr = indices.getptr(E->get()); int idx; if (!idxptr) { idx = indices.size(); new_vertices.push_back(E->get()); indices[E->get()] = idx; } else { idx = *idxptr; } index_array.push_back(idx); } vertex_array.clear(); vertex_array = new_vertices; format |= Mesh::ARRAY_FORMAT_INDEX; } void SurfaceTool::deindex() { if (index_array.size() == 0) { return; //nothing to deindex } Vector varr; varr.resize(vertex_array.size()); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next()) { varr.write[idx++] = E->get(); } vertex_array.clear(); for (List::Element *E = index_array.front(); E; E = E->next()) { ERR_FAIL_INDEX(E->get(), varr.size()); vertex_array.push_back(varr[E->get()]); } format &= ~Mesh::ARRAY_FORMAT_INDEX; index_array.clear(); } void SurfaceTool::_create_list(const Ref &p_existing, int p_surface, List *r_vertex, List *r_index, int &lformat) { ERR_FAIL_COND_MSG(p_existing.is_null(), "First argument in SurfaceTool::_create_list() must be a valid object of type Mesh"); Array arr = p_existing->surface_get_arrays(p_surface); ERR_FAIL_COND(arr.size() != VS::ARRAY_MAX); _create_list_from_arrays(arr, r_vertex, r_index, lformat); } Vector SurfaceTool::create_vertex_array_from_triangle_arrays(const Array &p_arrays) { Vector ret; PoolVector varr = p_arrays[VS::ARRAY_VERTEX]; PoolVector narr = p_arrays[VS::ARRAY_NORMAL]; PoolVector tarr = p_arrays[VS::ARRAY_TANGENT]; PoolVector carr = p_arrays[VS::ARRAY_COLOR]; PoolVector uvarr = p_arrays[VS::ARRAY_TEX_UV]; PoolVector uv2arr = p_arrays[VS::ARRAY_TEX_UV2]; PoolVector barr = p_arrays[VS::ARRAY_BONES]; PoolVector warr = p_arrays[VS::ARRAY_WEIGHTS]; int vc = varr.size(); if (vc == 0) { return ret; } int lformat = 0; PoolVector::Read rv; if (varr.size()) { lformat |= VS::ARRAY_FORMAT_VERTEX; rv = varr.read(); } PoolVector::Read rn; if (narr.size()) { lformat |= VS::ARRAY_FORMAT_NORMAL; rn = narr.read(); } PoolVector::Read rt; if (tarr.size()) { lformat |= VS::ARRAY_FORMAT_TANGENT; rt = tarr.read(); } PoolVector::Read rc; if (carr.size()) { lformat |= VS::ARRAY_FORMAT_COLOR; rc = carr.read(); } PoolVector::Read ruv; if (uvarr.size()) { lformat |= VS::ARRAY_FORMAT_TEX_UV; ruv = uvarr.read(); } PoolVector::Read ruv2; if (uv2arr.size()) { lformat |= VS::ARRAY_FORMAT_TEX_UV2; ruv2 = uv2arr.read(); } PoolVector::Read rb; if (barr.size()) { lformat |= VS::ARRAY_FORMAT_BONES; rb = barr.read(); } PoolVector::Read rw; if (warr.size()) { lformat |= VS::ARRAY_FORMAT_WEIGHTS; rw = warr.read(); } for (int i = 0; i < vc; i++) { Vertex v; if (lformat & VS::ARRAY_FORMAT_VERTEX) { v.vertex = varr[i]; } if (lformat & VS::ARRAY_FORMAT_NORMAL) { v.normal = narr[i]; } if (lformat & VS::ARRAY_FORMAT_TANGENT) { Plane p(tarr[i * 4 + 0], tarr[i * 4 + 1], tarr[i * 4 + 2], tarr[i * 4 + 3]); v.tangent = p.normal; v.binormal = p.normal.cross(v.tangent).normalized() * p.d; } if (lformat & VS::ARRAY_FORMAT_COLOR) { v.color = carr[i]; } if (lformat & VS::ARRAY_FORMAT_TEX_UV) { v.uv = uvarr[i]; } if (lformat & VS::ARRAY_FORMAT_TEX_UV2) { v.uv2 = uv2arr[i]; } if (lformat & VS::ARRAY_FORMAT_BONES) { Vector b; b.resize(4); b.write[0] = barr[i * 4 + 0]; b.write[1] = barr[i * 4 + 1]; b.write[2] = barr[i * 4 + 2]; b.write[3] = barr[i * 4 + 3]; v.bones = b; } if (lformat & VS::ARRAY_FORMAT_WEIGHTS) { Vector w; w.resize(4); w.write[0] = warr[i * 4 + 0]; w.write[1] = warr[i * 4 + 1]; w.write[2] = warr[i * 4 + 2]; w.write[3] = warr[i * 4 + 3]; v.weights = w; } ret.push_back(v); } return ret; } void SurfaceTool::_create_list_from_arrays(Array arr, List *r_vertex, List *r_index, int &lformat) { PoolVector varr = arr[VS::ARRAY_VERTEX]; PoolVector narr = arr[VS::ARRAY_NORMAL]; PoolVector tarr = arr[VS::ARRAY_TANGENT]; PoolVector carr = arr[VS::ARRAY_COLOR]; PoolVector uvarr = arr[VS::ARRAY_TEX_UV]; PoolVector uv2arr = arr[VS::ARRAY_TEX_UV2]; PoolVector barr = arr[VS::ARRAY_BONES]; PoolVector warr = arr[VS::ARRAY_WEIGHTS]; int vc = varr.size(); if (vc == 0) { return; } lformat = 0; PoolVector::Read rv; if (varr.size()) { lformat |= VS::ARRAY_FORMAT_VERTEX; rv = varr.read(); } PoolVector::Read rn; if (narr.size()) { lformat |= VS::ARRAY_FORMAT_NORMAL; rn = narr.read(); } PoolVector::Read rt; if (tarr.size()) { lformat |= VS::ARRAY_FORMAT_TANGENT; rt = tarr.read(); } PoolVector::Read rc; if (carr.size()) { lformat |= VS::ARRAY_FORMAT_COLOR; rc = carr.read(); } PoolVector::Read ruv; if (uvarr.size()) { lformat |= VS::ARRAY_FORMAT_TEX_UV; ruv = uvarr.read(); } PoolVector::Read ruv2; if (uv2arr.size()) { lformat |= VS::ARRAY_FORMAT_TEX_UV2; ruv2 = uv2arr.read(); } PoolVector::Read rb; if (barr.size()) { lformat |= VS::ARRAY_FORMAT_BONES; rb = barr.read(); } PoolVector::Read rw; if (warr.size()) { lformat |= VS::ARRAY_FORMAT_WEIGHTS; rw = warr.read(); } for (int i = 0; i < vc; i++) { Vertex v; if (lformat & VS::ARRAY_FORMAT_VERTEX) { v.vertex = varr[i]; } if (lformat & VS::ARRAY_FORMAT_NORMAL) { v.normal = narr[i]; } if (lformat & VS::ARRAY_FORMAT_TANGENT) { Plane p(tarr[i * 4 + 0], tarr[i * 4 + 1], tarr[i * 4 + 2], tarr[i * 4 + 3]); v.tangent = p.normal; v.binormal = p.normal.cross(v.tangent).normalized() * p.d; } if (lformat & VS::ARRAY_FORMAT_COLOR) { v.color = carr[i]; } if (lformat & VS::ARRAY_FORMAT_TEX_UV) { v.uv = uvarr[i]; } if (lformat & VS::ARRAY_FORMAT_TEX_UV2) { v.uv2 = uv2arr[i]; } if (lformat & VS::ARRAY_FORMAT_BONES) { Vector b; b.resize(4); b.write[0] = barr[i * 4 + 0]; b.write[1] = barr[i * 4 + 1]; b.write[2] = barr[i * 4 + 2]; b.write[3] = barr[i * 4 + 3]; v.bones = b; } if (lformat & VS::ARRAY_FORMAT_WEIGHTS) { Vector w; w.resize(4); w.write[0] = warr[i * 4 + 0]; w.write[1] = warr[i * 4 + 1]; w.write[2] = warr[i * 4 + 2]; w.write[3] = warr[i * 4 + 3]; v.weights = w; } r_vertex->push_back(v); } //indices PoolVector idx = arr[VS::ARRAY_INDEX]; int is = idx.size(); if (is) { lformat |= VS::ARRAY_FORMAT_INDEX; PoolVector::Read iarr = idx.read(); for (int i = 0; i < is; i++) { r_index->push_back(iarr[i]); } } } void SurfaceTool::create_from_triangle_arrays(const Array &p_arrays) { clear(); primitive = Mesh::PRIMITIVE_TRIANGLES; _create_list_from_arrays(p_arrays, &vertex_array, &index_array, format); } void SurfaceTool::create_from(const Ref &p_existing, int p_surface) { ERR_FAIL_COND_MSG(p_existing.is_null(), "First argument in SurfaceTool::create_from() must be a valid object of type Mesh"); clear(); primitive = p_existing->surface_get_primitive_type(p_surface); _create_list(p_existing, p_surface, &vertex_array, &index_array, format); material = p_existing->surface_get_material(p_surface); } void SurfaceTool::create_from_blend_shape(const Ref &p_existing, int p_surface, const String &p_blend_shape_name) { ERR_FAIL_COND_MSG(p_existing.is_null(), "First argument in SurfaceTool::create_from_blend_shape() must be a valid object of type Mesh"); clear(); primitive = p_existing->surface_get_primitive_type(p_surface); Array arr = p_existing->surface_get_blend_shape_arrays(p_surface); Array blend_shape_names; int32_t shape_idx = -1; for (int32_t i = 0; i < p_existing->get_blend_shape_count(); i++) { String name = p_existing->get_blend_shape_name(i); if (name == p_blend_shape_name) { shape_idx = i; break; } } ERR_FAIL_COND(shape_idx == -1); ERR_FAIL_COND(shape_idx >= arr.size()); Array mesh = arr[shape_idx]; ERR_FAIL_COND(mesh.size() != VS::ARRAY_MAX); _create_list_from_arrays(arr[shape_idx], &vertex_array, &index_array, format); } void SurfaceTool::append_from(const Ref &p_existing, int p_surface, const Transform &p_xform) { ERR_FAIL_COND_MSG(p_existing.is_null(), "First argument in SurfaceTool::append_from() must be a valid object of type Mesh"); if (vertex_array.size() == 0) { primitive = p_existing->surface_get_primitive_type(p_surface); format = 0; } int nformat; List nvertices; List nindices; _create_list(p_existing, p_surface, &nvertices, &nindices, nformat); format |= nformat; int vfrom = vertex_array.size(); for (List::Element *E = nvertices.front(); E; E = E->next()) { Vertex v = E->get(); v.vertex = p_xform.xform(v.vertex); if (nformat & VS::ARRAY_FORMAT_NORMAL) { v.normal = p_xform.basis.xform(v.normal); } if (nformat & VS::ARRAY_FORMAT_TANGENT) { v.tangent = p_xform.basis.xform(v.tangent); v.binormal = p_xform.basis.xform(v.binormal); } vertex_array.push_back(v); } for (List::Element *E = nindices.front(); E; E = E->next()) { int dst_index = E->get() + vfrom; index_array.push_back(dst_index); } if (index_array.size() % 3) { WARN_PRINT("SurfaceTool: Index array not a multiple of 3."); } } //mikktspace callbacks namespace { struct TangentGenerationContextUserData { Vector::Element *> vertices; Vector::Element *> indices; }; } // namespace int SurfaceTool::mikktGetNumFaces(const SMikkTSpaceContext *pContext) { TangentGenerationContextUserData &triangle_data = *reinterpret_cast(pContext->m_pUserData); if (triangle_data.indices.size() > 0) { return triangle_data.indices.size() / 3; } else { return triangle_data.vertices.size() / 3; } } int SurfaceTool::mikktGetNumVerticesOfFace(const SMikkTSpaceContext *pContext, const int iFace) { return 3; //always 3 } void SurfaceTool::mikktGetPosition(const SMikkTSpaceContext *pContext, float fvPosOut[], const int iFace, const int iVert) { TangentGenerationContextUserData &triangle_data = *reinterpret_cast(pContext->m_pUserData); Vector3 v; if (triangle_data.indices.size() > 0) { int index = triangle_data.indices[iFace * 3 + iVert]->get(); if (index < triangle_data.vertices.size()) { v = triangle_data.vertices[index]->get().vertex; } } else { v = triangle_data.vertices[iFace * 3 + iVert]->get().vertex; } fvPosOut[0] = v.x; fvPosOut[1] = v.y; fvPosOut[2] = v.z; } void SurfaceTool::mikktGetNormal(const SMikkTSpaceContext *pContext, float fvNormOut[], const int iFace, const int iVert) { TangentGenerationContextUserData &triangle_data = *reinterpret_cast(pContext->m_pUserData); Vector3 v; if (triangle_data.indices.size() > 0) { int index = triangle_data.indices[iFace * 3 + iVert]->get(); if (index < triangle_data.vertices.size()) { v = triangle_data.vertices[index]->get().normal; } } else { v = triangle_data.vertices[iFace * 3 + iVert]->get().normal; } fvNormOut[0] = v.x; fvNormOut[1] = v.y; fvNormOut[2] = v.z; } void SurfaceTool::mikktGetTexCoord(const SMikkTSpaceContext *pContext, float fvTexcOut[], const int iFace, const int iVert) { TangentGenerationContextUserData &triangle_data = *reinterpret_cast(pContext->m_pUserData); Vector2 v; if (triangle_data.indices.size() > 0) { int index = triangle_data.indices[iFace * 3 + iVert]->get(); if (index < triangle_data.vertices.size()) { v = triangle_data.vertices[index]->get().uv; } } else { v = triangle_data.vertices[iFace * 3 + iVert]->get().uv; } fvTexcOut[0] = v.x; fvTexcOut[1] = v.y; } void SurfaceTool::mikktSetTSpaceDefault(const SMikkTSpaceContext *pContext, const float fvTangent[], const float fvBiTangent[], const float fMagS, const float fMagT, const tbool bIsOrientationPreserving, const int iFace, const int iVert) { TangentGenerationContextUserData &triangle_data = *reinterpret_cast(pContext->m_pUserData); Vertex *vtx = nullptr; if (triangle_data.indices.size() > 0) { int index = triangle_data.indices[iFace * 3 + iVert]->get(); if (index < triangle_data.vertices.size()) { vtx = &triangle_data.vertices[index]->get(); } } else { vtx = &triangle_data.vertices[iFace * 3 + iVert]->get(); } if (vtx != nullptr) { vtx->tangent = Vector3(fvTangent[0], fvTangent[1], fvTangent[2]); vtx->binormal = Vector3(-fvBiTangent[0], -fvBiTangent[1], -fvBiTangent[2]); // for some reason these are reversed, something with the coordinate system in Godot } } void SurfaceTool::generate_tangents() { ERR_FAIL_COND(!(format & Mesh::ARRAY_FORMAT_TEX_UV)); ERR_FAIL_COND(!(format & Mesh::ARRAY_FORMAT_NORMAL)); SMikkTSpaceInterface mkif; mkif.m_getNormal = mikktGetNormal; mkif.m_getNumFaces = mikktGetNumFaces; mkif.m_getNumVerticesOfFace = mikktGetNumVerticesOfFace; mkif.m_getPosition = mikktGetPosition; mkif.m_getTexCoord = mikktGetTexCoord; mkif.m_setTSpace = mikktSetTSpaceDefault; mkif.m_setTSpaceBasic = nullptr; SMikkTSpaceContext msc; msc.m_pInterface = &mkif; TangentGenerationContextUserData triangle_data; triangle_data.vertices.resize(vertex_array.size()); int idx = 0; for (List::Element *E = vertex_array.front(); E; E = E->next()) { triangle_data.vertices.write[idx++] = E; E->get().binormal = Vector3(); E->get().tangent = Vector3(); } triangle_data.indices.resize(index_array.size()); idx = 0; for (List::Element *E = index_array.front(); E; E = E->next()) { triangle_data.indices.write[idx++] = E; } msc.m_pUserData = &triangle_data; bool res = genTangSpaceDefault(&msc); ERR_FAIL_COND(!res); format |= Mesh::ARRAY_FORMAT_TANGENT; } void SurfaceTool::generate_normals(bool p_flip) { ERR_FAIL_COND(primitive != Mesh::PRIMITIVE_TRIANGLES); bool was_indexed = index_array.size(); deindex(); HashMap vertex_hash; int count = 0; bool smooth = false; if (smooth_groups.has(0)) { smooth = smooth_groups[0]; } List::Element *B = vertex_array.front(); for (List::Element *E = B; E;) { List::Element *v[3]; v[0] = E; v[1] = v[0]->next(); ERR_FAIL_COND(!v[1]); v[2] = v[1]->next(); ERR_FAIL_COND(!v[2]); E = v[2]->next(); Vector3 normal; if (!p_flip) { normal = Plane(v[0]->get().vertex, v[1]->get().vertex, v[2]->get().vertex).normal; } else { normal = Plane(v[2]->get().vertex, v[1]->get().vertex, v[0]->get().vertex).normal; } if (smooth) { for (int i = 0; i < 3; i++) { Vector3 *lv = vertex_hash.getptr(v[i]->get()); if (!lv) { vertex_hash.set(v[i]->get(), normal); } else { (*lv) += normal; } } } else { for (int i = 0; i < 3; i++) { v[i]->get().normal = normal; } } count += 3; if (smooth_groups.has(count) || !E) { if (vertex_hash.size()) { while (B != E) { Vector3 *lv = vertex_hash.getptr(B->get()); if (lv) { B->get().normal = lv->normalized(); } B = B->next(); } } else { B = E; } vertex_hash.clear(); if (E) { smooth = smooth_groups[count]; } } } format |= Mesh::ARRAY_FORMAT_NORMAL; if (was_indexed) { index(); smooth_groups.clear(); } } void SurfaceTool::set_material(const Ref &p_material) { material = p_material; } void SurfaceTool::clear() { begun = false; primitive = Mesh::PRIMITIVE_LINES; format = 0; last_bones.clear(); last_weights.clear(); index_array.clear(); vertex_array.clear(); smooth_groups.clear(); material.unref(); } void SurfaceTool::_bind_methods() { ClassDB::bind_method(D_METHOD("begin", "primitive"), &SurfaceTool::begin); ClassDB::bind_method(D_METHOD("add_vertex", "vertex"), &SurfaceTool::add_vertex); ClassDB::bind_method(D_METHOD("add_color", "color"), &SurfaceTool::add_color); ClassDB::bind_method(D_METHOD("add_normal", "normal"), &SurfaceTool::add_normal); ClassDB::bind_method(D_METHOD("add_tangent", "tangent"), &SurfaceTool::add_tangent); ClassDB::bind_method(D_METHOD("add_uv", "uv"), &SurfaceTool::add_uv); ClassDB::bind_method(D_METHOD("add_uv2", "uv2"), &SurfaceTool::add_uv2); ClassDB::bind_method(D_METHOD("add_bones", "bones"), &SurfaceTool::add_bones); ClassDB::bind_method(D_METHOD("add_weights", "weights"), &SurfaceTool::add_weights); ClassDB::bind_method(D_METHOD("add_smooth_group", "smooth"), &SurfaceTool::add_smooth_group); ClassDB::bind_method(D_METHOD("add_triangle_fan", "vertices", "uvs", "colors", "uv2s", "normals", "tangents"), &SurfaceTool::add_triangle_fan, DEFVAL(Vector()), DEFVAL(Vector()), DEFVAL(Vector()), DEFVAL(Vector()), DEFVAL(Vector())); ClassDB::bind_method(D_METHOD("add_index", "index"), &SurfaceTool::add_index); ClassDB::bind_method(D_METHOD("index"), &SurfaceTool::index); ClassDB::bind_method(D_METHOD("deindex"), &SurfaceTool::deindex); ClassDB::bind_method(D_METHOD("generate_normals", "flip"), &SurfaceTool::generate_normals, DEFVAL(false)); ClassDB::bind_method(D_METHOD("generate_tangents"), &SurfaceTool::generate_tangents); ClassDB::bind_method(D_METHOD("set_material", "material"), &SurfaceTool::set_material); ClassDB::bind_method(D_METHOD("clear"), &SurfaceTool::clear); ClassDB::bind_method(D_METHOD("create_from", "existing", "surface"), &SurfaceTool::create_from); ClassDB::bind_method(D_METHOD("create_from_blend_shape", "existing", "surface", "blend_shape"), &SurfaceTool::create_from_blend_shape); ClassDB::bind_method(D_METHOD("append_from", "existing", "surface", "transform"), &SurfaceTool::append_from); ClassDB::bind_method(D_METHOD("commit", "existing", "flags"), &SurfaceTool::commit, DEFVAL(Variant()), DEFVAL(Mesh::ARRAY_COMPRESS_DEFAULT)); ClassDB::bind_method(D_METHOD("commit_to_arrays"), &SurfaceTool::commit_to_arrays); } SurfaceTool::SurfaceTool() { first = false; begun = false; primitive = Mesh::PRIMITIVE_LINES; format = 0; }