1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042 |
- /*************************************************************************/
- /* surface_tool.cpp */
- /*************************************************************************/
- /* This file is part of: */
- /* GODOT ENGINE */
- /* https://godotengine.org */
- /*************************************************************************/
- /* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
- /* Copyright (c) 2014-2019 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 "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;
- 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<int> &p_bones) {
- ERR_FAIL_COND(!begun);
- ERR_FAIL_COND(p_bones.size() != 4);
- ERR_FAIL_COND(!first && !(format & Mesh::ARRAY_FORMAT_BONES));
- format |= Mesh::ARRAY_FORMAT_BONES;
- last_bones = p_bones;
- }
- void SurfaceTool::add_weights(const Vector<float> &p_weights) {
- ERR_FAIL_COND(!begun);
- ERR_FAIL_COND(p_weights.size() != 4);
- 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<Vector3> &p_vertexes, const Vector<Vector2> &p_uvs, const Vector<Color> &p_colors, const Vector<Vector2> &p_uv2s, const Vector<Vector3> &p_normals, const Vector<Plane> &p_tangents) {
- ERR_FAIL_COND(!begun);
- ERR_FAIL_COND(primitive != Mesh::PRIMITIVE_TRIANGLES);
- ERR_FAIL_COND(p_vertexes.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_vertexes[n]); \
- }
- for (int i = 0; i < p_vertexes.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);
- 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<Vector3> array;
- array.resize(varr_len);
- PoolVector<Vector3>::Write w = array.write();
- int idx = 0;
- for (List<Vertex>::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 = PoolVector<Vector3>::Write();
- a[i] = array;
- } break;
- case Mesh::ARRAY_TEX_UV:
- case Mesh::ARRAY_TEX_UV2: {
- PoolVector<Vector2> array;
- array.resize(varr_len);
- PoolVector<Vector2>::Write w = array.write();
- int idx = 0;
- for (List<Vertex>::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 = PoolVector<Vector2>::Write();
- a[i] = array;
- } break;
- case Mesh::ARRAY_TANGENT: {
- PoolVector<float> array;
- array.resize(varr_len * 4);
- PoolVector<float>::Write w = array.write();
- int idx = 0;
- for (List<Vertex>::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 = PoolVector<float>::Write();
- a[i] = array;
- } break;
- case Mesh::ARRAY_COLOR: {
- PoolVector<Color> array;
- array.resize(varr_len);
- PoolVector<Color>::Write w = array.write();
- int idx = 0;
- for (List<Vertex>::Element *E = vertex_array.front(); E; E = E->next(), idx++) {
- const Vertex &v = E->get();
- w[idx] = v.color;
- }
- w = PoolVector<Color>::Write();
- a[i] = array;
- } break;
- case Mesh::ARRAY_BONES: {
- PoolVector<int> array;
- array.resize(varr_len * 4);
- PoolVector<int>::Write w = array.write();
- int idx = 0;
- for (List<Vertex>::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 = PoolVector<int>::Write();
- a[i] = array;
- } break;
- case Mesh::ARRAY_WEIGHTS: {
- PoolVector<float> array;
- array.resize(varr_len * 4);
- PoolVector<float>::Write w = array.write();
- int idx = 0;
- for (List<Vertex>::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 = PoolVector<float>::Write();
- a[i] = array;
- } break;
- case Mesh::ARRAY_INDEX: {
- ERR_CONTINUE(index_array.size() == 0);
- PoolVector<int> array;
- array.resize(index_array.size());
- PoolVector<int>::Write w = array.write();
- int idx = 0;
- for (List<int>::Element *E = index_array.front(); E; E = E->next(), idx++) {
- w[idx] = E->get();
- }
- w = PoolVector<int>::Write();
- a[i] = array;
- } break;
- default: {}
- }
- }
- return a;
- }
- Ref<ArrayMesh> SurfaceTool::commit(const Ref<ArrayMesh> &p_existing, uint32_t p_flags) {
- Ref<ArrayMesh> 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<Vertex, int, VertexHasher> indices;
- List<Vertex> new_vertices;
- for (List<Vertex>::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<Vertex> varr;
- varr.resize(vertex_array.size());
- int idx = 0;
- for (List<Vertex>::Element *E = vertex_array.front(); E; E = E->next()) {
- varr.write[idx++] = E->get();
- }
- vertex_array.clear();
- for (List<int>::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<Mesh> &p_existing, int p_surface, List<Vertex> *r_vertex, List<int> *r_index, int &lformat) {
- 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::Vertex> SurfaceTool::create_vertex_array_from_triangle_arrays(const Array &p_arrays) {
- Vector<SurfaceTool::Vertex> ret;
- PoolVector<Vector3> varr = p_arrays[VS::ARRAY_VERTEX];
- PoolVector<Vector3> narr = p_arrays[VS::ARRAY_NORMAL];
- PoolVector<float> tarr = p_arrays[VS::ARRAY_TANGENT];
- PoolVector<Color> carr = p_arrays[VS::ARRAY_COLOR];
- PoolVector<Vector2> uvarr = p_arrays[VS::ARRAY_TEX_UV];
- PoolVector<Vector2> uv2arr = p_arrays[VS::ARRAY_TEX_UV2];
- PoolVector<int> barr = p_arrays[VS::ARRAY_BONES];
- PoolVector<float> warr = p_arrays[VS::ARRAY_WEIGHTS];
- int vc = varr.size();
- if (vc == 0)
- return ret;
- int lformat = 0;
- PoolVector<Vector3>::Read rv;
- if (varr.size()) {
- lformat |= VS::ARRAY_FORMAT_VERTEX;
- rv = varr.read();
- }
- PoolVector<Vector3>::Read rn;
- if (narr.size()) {
- lformat |= VS::ARRAY_FORMAT_NORMAL;
- rn = narr.read();
- }
- PoolVector<float>::Read rt;
- if (tarr.size()) {
- lformat |= VS::ARRAY_FORMAT_TANGENT;
- rt = tarr.read();
- }
- PoolVector<Color>::Read rc;
- if (carr.size()) {
- lformat |= VS::ARRAY_FORMAT_COLOR;
- rc = carr.read();
- }
- PoolVector<Vector2>::Read ruv;
- if (uvarr.size()) {
- lformat |= VS::ARRAY_FORMAT_TEX_UV;
- ruv = uvarr.read();
- }
- PoolVector<Vector2>::Read ruv2;
- if (uv2arr.size()) {
- lformat |= VS::ARRAY_FORMAT_TEX_UV2;
- ruv2 = uv2arr.read();
- }
- PoolVector<int>::Read rb;
- if (barr.size()) {
- lformat |= VS::ARRAY_FORMAT_BONES;
- rb = barr.read();
- }
- PoolVector<float>::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<int> 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<float> 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<Vertex> *r_vertex, List<int> *r_index, int &lformat) {
- PoolVector<Vector3> varr = arr[VS::ARRAY_VERTEX];
- PoolVector<Vector3> narr = arr[VS::ARRAY_NORMAL];
- PoolVector<float> tarr = arr[VS::ARRAY_TANGENT];
- PoolVector<Color> carr = arr[VS::ARRAY_COLOR];
- PoolVector<Vector2> uvarr = arr[VS::ARRAY_TEX_UV];
- PoolVector<Vector2> uv2arr = arr[VS::ARRAY_TEX_UV2];
- PoolVector<int> barr = arr[VS::ARRAY_BONES];
- PoolVector<float> warr = arr[VS::ARRAY_WEIGHTS];
- int vc = varr.size();
- if (vc == 0)
- return;
- lformat = 0;
- PoolVector<Vector3>::Read rv;
- if (varr.size()) {
- lformat |= VS::ARRAY_FORMAT_VERTEX;
- rv = varr.read();
- }
- PoolVector<Vector3>::Read rn;
- if (narr.size()) {
- lformat |= VS::ARRAY_FORMAT_NORMAL;
- rn = narr.read();
- }
- PoolVector<float>::Read rt;
- if (tarr.size()) {
- lformat |= VS::ARRAY_FORMAT_TANGENT;
- rt = tarr.read();
- }
- PoolVector<Color>::Read rc;
- if (carr.size()) {
- lformat |= VS::ARRAY_FORMAT_COLOR;
- rc = carr.read();
- }
- PoolVector<Vector2>::Read ruv;
- if (uvarr.size()) {
- lformat |= VS::ARRAY_FORMAT_TEX_UV;
- ruv = uvarr.read();
- }
- PoolVector<Vector2>::Read ruv2;
- if (uv2arr.size()) {
- lformat |= VS::ARRAY_FORMAT_TEX_UV2;
- ruv2 = uv2arr.read();
- }
- PoolVector<int>::Read rb;
- if (barr.size()) {
- lformat |= VS::ARRAY_FORMAT_BONES;
- rb = barr.read();
- }
- PoolVector<float>::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<int> 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<float> 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<int> idx = arr[VS::ARRAY_INDEX];
- int is = idx.size();
- if (is) {
- lformat |= VS::ARRAY_FORMAT_INDEX;
- PoolVector<int>::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<Mesh> &p_existing, int p_surface) {
- 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::append_from(const Ref<Mesh> &p_existing, int p_surface, const Transform &p_xform) {
- if (vertex_array.size() == 0) {
- primitive = p_existing->surface_get_primitive_type(p_surface);
- format = 0;
- }
- int nformat;
- List<Vertex> nvertices;
- List<int> nindices;
- _create_list(p_existing, p_surface, &nvertices, &nindices, nformat);
- format |= nformat;
- int vfrom = vertex_array.size();
- for (List<Vertex>::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<int>::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<List<SurfaceTool::Vertex>::Element *> vertices;
- Vector<List<int>::Element *> indices;
- };
- } // namespace
- int SurfaceTool::mikktGetNumFaces(const SMikkTSpaceContext *pContext) {
- TangentGenerationContextUserData &triangle_data = *reinterpret_cast<TangentGenerationContextUserData *>(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<TangentGenerationContextUserData *>(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<TangentGenerationContextUserData *>(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<TangentGenerationContextUserData *>(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<TangentGenerationContextUserData *>(pContext->m_pUserData);
- Vertex *vtx = NULL;
- 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 != NULL) {
- 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 = NULL;
- SMikkTSpaceContext msc;
- msc.m_pInterface = &mkif;
- TangentGenerationContextUserData triangle_data;
- triangle_data.vertices.resize(vertex_array.size());
- int idx = 0;
- for (List<Vertex>::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<int>::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, Vector3, VertexHasher> vertex_hash;
- int count = 0;
- bool smooth = false;
- if (smooth_groups.has(0))
- smooth = smooth_groups[0];
- List<Vertex>::Element *B = vertex_array.front();
- for (List<Vertex>::Element *E = B; E;) {
- List<Vertex>::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<Material> &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", "vertexes", "uvs", "colors", "uv2s", "normals", "tangents"), &SurfaceTool::add_triangle_fan, DEFVAL(Vector<Vector2>()), DEFVAL(Vector<Color>()), DEFVAL(Vector<Vector2>()), DEFVAL(Vector<Vector3>()), DEFVAL(Vector<Plane>()));
- 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("add_to_format", "flags"), &SurfaceTool::add_to_format);
- 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("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));
- }
- SurfaceTool::SurfaceTool() {
- first = false;
- begun = false;
- primitive = Mesh::PRIMITIVE_LINES;
- format = 0;
- }
|