/*************************************************************************/ /* mesh_data_tool.cpp */ /*************************************************************************/ /* This file is part of: */ /* GODOT ENGINE */ /* http://www.godotengine.org */ /*************************************************************************/ /* Copyright (c) 2007-2016 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 "mesh_data_tool.h" void MeshDataTool::clear() { vertices.clear(); edges.clear(); faces.clear(); material = Ref<Material>(); format=0; } Error MeshDataTool::create_from_surface(const Ref<Mesh>& p_mesh,int p_surface) { ERR_FAIL_COND_V(p_mesh.is_null(),ERR_INVALID_PARAMETER); ERR_FAIL_COND_V(p_mesh.is_null(),ERR_INVALID_PARAMETER); ERR_FAIL_COND_V(p_mesh->surface_get_primitive_type(p_surface)!=Mesh::PRIMITIVE_TRIANGLES,ERR_INVALID_PARAMETER); Array arrays = p_mesh->surface_get_arrays(p_surface); ERR_FAIL_COND_V( arrays.empty(), ERR_INVALID_PARAMETER ); DVector<Vector3> varray = arrays[Mesh::ARRAY_VERTEX]; int vcount = varray.size(); ERR_FAIL_COND_V( vcount == 0, ERR_INVALID_PARAMETER); clear(); format = p_mesh->surface_get_format(p_surface); material=p_mesh->surface_get_material(p_surface); DVector<Vector3>::Read vr = varray.read(); DVector<Vector3>::Read nr; if (arrays[Mesh::ARRAY_NORMAL].get_type()!=Variant::NIL) nr = arrays[Mesh::ARRAY_NORMAL].operator DVector<Vector3>().read(); DVector<real_t>::Read ta; if (arrays[Mesh::ARRAY_TANGENT].get_type()!=Variant::NIL) ta = arrays[Mesh::ARRAY_TANGENT].operator DVector<real_t>().read(); DVector<Vector2>::Read uv; if (arrays[Mesh::ARRAY_TEX_UV].get_type()!=Variant::NIL) uv = arrays[Mesh::ARRAY_TEX_UV].operator DVector<Vector2>().read(); DVector<Vector2>::Read uv2; if (arrays[Mesh::ARRAY_TEX_UV2].get_type()!=Variant::NIL) uv2 = arrays[Mesh::ARRAY_TEX_UV2].operator DVector<Vector2>().read(); DVector<Color>::Read col; if (arrays[Mesh::ARRAY_COLOR].get_type()!=Variant::NIL) col = arrays[Mesh::ARRAY_COLOR].operator DVector<Color>().read(); DVector<real_t>::Read bo; if (arrays[Mesh::ARRAY_BONES].get_type()!=Variant::NIL) bo = arrays[Mesh::ARRAY_BONES].operator DVector<real_t>().read(); DVector<real_t>::Read we; if (arrays[Mesh::ARRAY_WEIGHTS].get_type()!=Variant::NIL) we = arrays[Mesh::ARRAY_WEIGHTS].operator DVector<real_t>().read(); vertices.resize(vcount); for(int i=0;i<vcount;i++) { Vertex v; v.vertex=vr[i]; if (nr.ptr()) v.normal=nr[i]; if (ta.ptr()) v.tangent=Plane(ta[i*4+0],ta[i*4+1],ta[i*4+2],ta[i*4+3]); if (uv.ptr()) v.uv=uv[i]; if (uv2.ptr()) v.uv2=uv2[i]; if (col.ptr()) v.color=col[i]; if (we.ptr()) { v.weights.push_back(we[i*4+0]); v.weights.push_back(we[i*4+1]); v.weights.push_back(we[i*4+2]); v.weights.push_back(we[i*4+3]); } if (bo.ptr()) { v.bones.push_back(bo[i*4+0]); v.bones.push_back(bo[i*4+1]); v.bones.push_back(bo[i*4+2]); v.bones.push_back(bo[i*4+3]); } vertices[i]=v; } DVector<int> indices; if (arrays[Mesh::ARRAY_INDEX].get_type()!=Variant::NIL) { indices = arrays[Mesh::ARRAY_INDEX]; } else { //make code simpler indices.resize(vcount); DVector<int>::Write iw=indices.write(); for(int i=0;i<vcount;i++) iw[i]=i; } int icount=indices.size(); DVector<int>::Read r = indices.read(); Map<Point2i,int> edge_indices; for(int i=0;i<icount;i+=3) { Vertex*v[3]={ &vertices[r[i+0]], &vertices[r[i+1]], &vertices[r[i+2]] }; int fidx=faces.size(); Face face; for(int j=0;j<3;j++) { face.v[j]=r[i+j]; Point2i edge(r[i+j],r[i+(j+1)%3]); if (edge.x > edge.y) { SWAP(edge.x,edge.y); } if (edge_indices.has(edge)) { face.edges[j]=edge_indices[edge]; } else { face.edges[j]=edge_indices.size(); edge_indices[edge]=face.edges[j]; Edge e; e.vertex[0]=edge.x; e.vertex[1]=edge.y; edges.push_back(e); } edges[face.edges[j]].faces.push_back(fidx); v[j]->faces.push_back(fidx); v[j]->edges.push_back(face.edges[j]); } faces.push_back(face); } return OK; } Error MeshDataTool::commit_to_surface(const Ref<Mesh>& p_mesh) { ERR_FAIL_COND_V(p_mesh.is_null(),ERR_INVALID_PARAMETER); Array arr; arr.resize(Mesh::ARRAY_MAX); int vcount=vertices.size(); DVector<Vector3> v; DVector<Vector3> n; DVector<real_t> t; DVector<Vector2> u; DVector<Vector2> u2; DVector<Color> c; DVector<real_t> b; DVector<real_t> w; DVector<int> in; { v.resize(vcount); DVector<Vector3>::Write vr=v.write(); DVector<Vector3>::Write nr; if (format&Mesh::ARRAY_FORMAT_NORMAL) { n.resize(vcount); nr = n.write(); } DVector<real_t>::Write ta; if (format&Mesh::ARRAY_FORMAT_TANGENT) { t.resize(vcount*4); ta = t.write(); } DVector<Vector2>::Write uv; if (format&Mesh::ARRAY_FORMAT_TEX_UV) { u.resize(vcount); uv = u.write(); } DVector<Vector2>::Write uv2; if (format&Mesh::ARRAY_FORMAT_TEX_UV2) { u2.resize(vcount); uv2 = u2.write(); } DVector<Color>::Write col; if (format&Mesh::ARRAY_FORMAT_COLOR) { c.resize(vcount); col = c.write(); } DVector<real_t>::Write bo; if (format&Mesh::ARRAY_FORMAT_BONES) { b.resize(vcount*4); bo = b.write(); } DVector<real_t>::Write we; if (format&Mesh::ARRAY_FORMAT_WEIGHTS) { w.resize(vcount*4); we = w.write(); } for(int i=0;i<vcount;i++) { Vertex &vtx=vertices[i]; vr[i]=vtx.vertex; if (nr.ptr()) nr[i]=vtx.normal; if (ta.ptr()) { ta[i*4+0]=vtx.tangent.normal.x; ta[i*4+1]=vtx.tangent.normal.y; ta[i*4+2]=vtx.tangent.normal.z; ta[i*4+3]=vtx.tangent.d; } if (uv.ptr()) uv[i]=vtx.uv; if (uv2.ptr()) uv2[i]=vtx.uv2; if (col.ptr()) col[i]=vtx.color; if (we.ptr()) { we[i*4+0]=vtx.weights[0]; we[i*4+1]=vtx.weights[1]; we[i*4+2]=vtx.weights[2]; we[i*4+3]=vtx.weights[3]; } if (bo.ptr()) { bo[i*4+0]=vtx.bones[0]; bo[i*4+1]=vtx.bones[1]; bo[i*4+2]=vtx.bones[2]; bo[i*4+3]=vtx.bones[3]; } } int fc = faces.size(); in.resize(fc*3); DVector<int>::Write iw=in.write(); for(int i=0;i<fc;i++) { iw[i*3+0]=faces[i].v[0]; iw[i*3+1]=faces[i].v[1]; iw[i*3+2]=faces[i].v[2]; } } arr[Mesh::ARRAY_VERTEX]=v; arr[Mesh::ARRAY_INDEX]=in; if (n.size()) arr[Mesh::ARRAY_NORMAL]=n; if (c.size()) arr[Mesh::ARRAY_COLOR]=c; if (u.size()) arr[Mesh::ARRAY_TEX_UV]=u; if (u2.size()) arr[Mesh::ARRAY_TEX_UV2]=u2; if (t.size()) arr[Mesh::ARRAY_TANGENT]=t; if (b.size()) arr[Mesh::ARRAY_BONES]=b; if (w.size()) arr[Mesh::ARRAY_WEIGHTS]=w; Ref<Mesh> ncmesh=p_mesh; int sc = ncmesh->get_surface_count(); ncmesh->add_surface(Mesh::PRIMITIVE_TRIANGLES,arr); ncmesh->surface_set_material(sc,material); return OK; } int MeshDataTool::get_format() const { return format; } int MeshDataTool::get_vertex_count() const { return vertices.size(); } int MeshDataTool::get_edge_count() const{ return edges.size(); } int MeshDataTool::get_face_count() const{ return faces.size(); } Vector3 MeshDataTool::get_vertex(int p_idx) const { ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector3()); return vertices[p_idx].vertex; } void MeshDataTool::set_vertex(int p_idx,const Vector3& p_vertex){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].vertex=p_vertex; } Vector3 MeshDataTool::get_vertex_normal(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector3()); return vertices[p_idx].normal; } void MeshDataTool::set_vertex_normal(int p_idx,const Vector3& p_normal){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].normal=p_normal; format|=Mesh::ARRAY_FORMAT_NORMAL; } Plane MeshDataTool::get_vertex_tangent(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Plane()); return vertices[p_idx].tangent; } void MeshDataTool::set_vertex_tangent(int p_idx,const Plane& p_tangent){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].tangent=p_tangent; format|=Mesh::ARRAY_FORMAT_TANGENT; } Vector2 MeshDataTool::get_vertex_uv(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector2()); return vertices[p_idx].uv; } void MeshDataTool::set_vertex_uv(int p_idx,const Vector2& p_uv){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].uv=p_uv; format|=Mesh::ARRAY_FORMAT_TEX_UV; } Vector2 MeshDataTool::get_vertex_uv2(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector2()); return vertices[p_idx].uv2; } void MeshDataTool::set_vertex_uv2(int p_idx,const Vector2& p_uv2){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].uv2=p_uv2; format|=Mesh::ARRAY_FORMAT_TEX_UV2; } Color MeshDataTool::get_vertex_color(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Color()); return vertices[p_idx].color; } void MeshDataTool::set_vertex_color(int p_idx,const Color& p_color){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].color=p_color; format|=Mesh::ARRAY_FORMAT_COLOR; } Vector<int> MeshDataTool::get_vertex_bones(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector<int>()); return vertices[p_idx].bones; } void MeshDataTool::set_vertex_bones(int p_idx,const Vector<int>& p_bones){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].bones=p_bones; format|=Mesh::ARRAY_FORMAT_BONES; } Vector<float> MeshDataTool::get_vertex_weights(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector<float>()); return vertices[p_idx].weights; } void MeshDataTool::set_vertex_weights(int p_idx,const Vector<float>& p_weights){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].weights=p_weights; format|=Mesh::ARRAY_FORMAT_WEIGHTS; } Variant MeshDataTool::get_vertex_meta(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Variant()); return vertices[p_idx].meta; } void MeshDataTool::set_vertex_meta(int p_idx,const Variant& p_meta){ ERR_FAIL_INDEX(p_idx,vertices.size()); vertices[p_idx].meta=p_meta; } Vector<int> MeshDataTool::get_vertex_edges(int p_idx) const { ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector<int>()); return vertices[p_idx].edges; } Vector<int> MeshDataTool::get_vertex_faces(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,vertices.size(),Vector<int>()); return vertices[p_idx].faces; } int MeshDataTool::get_edge_vertex(int p_edge,int p_vertex) const { ERR_FAIL_INDEX_V(p_edge,edges.size(),-1); ERR_FAIL_INDEX_V(p_vertex,2,-1); return edges[p_edge].vertex[p_vertex]; } Vector<int> MeshDataTool::get_edge_faces(int p_edge) const{ ERR_FAIL_INDEX_V(p_edge,edges.size(),Vector<int>()); return edges[p_edge].faces; } Variant MeshDataTool::get_edge_meta(int p_idx) const{ ERR_FAIL_INDEX_V(p_idx,edges.size(),Variant()); return edges[p_idx].meta; } void MeshDataTool::set_edge_meta(int p_idx,const Variant& p_meta){ ERR_FAIL_INDEX(p_idx,edges.size()); edges[p_idx].meta=p_meta; } int MeshDataTool::get_face_vertex(int p_face,int p_vertex) const { ERR_FAIL_INDEX_V(p_face,faces.size(),-1); ERR_FAIL_INDEX_V(p_vertex,3,-1); return faces[p_face].v[p_vertex]; } int MeshDataTool::get_face_edge(int p_face,int p_vertex) const{ ERR_FAIL_INDEX_V(p_face,faces.size(),-1); ERR_FAIL_INDEX_V(p_vertex,3,-1); return faces[p_face].edges[p_vertex]; } Variant MeshDataTool::get_face_meta(int p_face) const{ ERR_FAIL_INDEX_V(p_face,faces.size(),Variant()); return faces[p_face].meta; } void MeshDataTool::set_face_meta(int p_face,const Variant& p_meta){ ERR_FAIL_INDEX(p_face,faces.size()); faces[p_face].meta=p_meta; } Vector3 MeshDataTool::get_face_normal(int p_face) const { ERR_FAIL_INDEX_V(p_face,faces.size(),Vector3()); Vector3 v0 = vertices[faces[p_face].v[0]].vertex; Vector3 v1 = vertices[faces[p_face].v[1]].vertex; Vector3 v2 = vertices[faces[p_face].v[2]].vertex; return Plane(v0,v1,v2).normal; } Ref<Material> MeshDataTool::get_material() const { return material; } void MeshDataTool::set_material(const Ref<Material> &p_material) { material=p_material; } void MeshDataTool::_bind_methods() { ObjectTypeDB::bind_method(_MD("clear"),&MeshDataTool::clear); ObjectTypeDB::bind_method(_MD("create_from_surface","mesh","surface"),&MeshDataTool::create_from_surface); ObjectTypeDB::bind_method(_MD("commit_to_surface","mesh"),&MeshDataTool::commit_to_surface); ObjectTypeDB::bind_method(_MD("get_format"),&MeshDataTool::get_format); ObjectTypeDB::bind_method(_MD("get_vertex_count"),&MeshDataTool::get_vertex_count); ObjectTypeDB::bind_method(_MD("get_edge_count"),&MeshDataTool::get_edge_count); ObjectTypeDB::bind_method(_MD("get_face_count"),&MeshDataTool::get_face_count); ObjectTypeDB::bind_method(_MD("set_vertex","idx","vertex"),&MeshDataTool::set_vertex); ObjectTypeDB::bind_method(_MD("get_vertex","idx"),&MeshDataTool::get_vertex); ObjectTypeDB::bind_method(_MD("set_vertex_normal","idx","normal"),&MeshDataTool::set_vertex_normal); ObjectTypeDB::bind_method(_MD("get_vertex_normal","idx"),&MeshDataTool::get_vertex_normal); ObjectTypeDB::bind_method(_MD("set_vertex_tangent","idx","tangent"),&MeshDataTool::set_vertex_tangent); ObjectTypeDB::bind_method(_MD("get_vertex_tangent","idx"),&MeshDataTool::get_vertex_tangent); ObjectTypeDB::bind_method(_MD("set_vertex_uv","idx","uv"),&MeshDataTool::set_vertex_uv); ObjectTypeDB::bind_method(_MD("get_vertex_uv","idx"),&MeshDataTool::get_vertex_uv); ObjectTypeDB::bind_method(_MD("set_vertex_uv2","idx","uv2"),&MeshDataTool::set_vertex_uv2); ObjectTypeDB::bind_method(_MD("get_vertex_uv2","idx"),&MeshDataTool::get_vertex_uv2); ObjectTypeDB::bind_method(_MD("set_vertex_color","idx","color"),&MeshDataTool::set_vertex_color); ObjectTypeDB::bind_method(_MD("get_vertex_color","idx"),&MeshDataTool::get_vertex_color); ObjectTypeDB::bind_method(_MD("set_vertex_bones","idx","bones"),&MeshDataTool::set_vertex_bones); ObjectTypeDB::bind_method(_MD("get_vertex_bones","idx"),&MeshDataTool::get_vertex_bones); ObjectTypeDB::bind_method(_MD("set_vertex_weights","idx","weights"),&MeshDataTool::set_vertex_weights); ObjectTypeDB::bind_method(_MD("get_vertex_weights","idx"),&MeshDataTool::get_vertex_weights); ObjectTypeDB::bind_method(_MD("set_vertex_meta","idx","meta"),&MeshDataTool::set_vertex_meta); ObjectTypeDB::bind_method(_MD("get_vertex_meta","idx"),&MeshDataTool::get_vertex_meta); ObjectTypeDB::bind_method(_MD("get_vertex_edges","idx"),&MeshDataTool::get_vertex_edges); ObjectTypeDB::bind_method(_MD("get_vertex_faces","idx"),&MeshDataTool::get_vertex_faces); ObjectTypeDB::bind_method(_MD("get_edge_vertex","idx","vertex"),&MeshDataTool::get_edge_vertex); ObjectTypeDB::bind_method(_MD("get_edge_faces","idx","faces"),&MeshDataTool::get_edge_faces); ObjectTypeDB::bind_method(_MD("set_edge_meta","idx","meta"),&MeshDataTool::set_edge_meta); ObjectTypeDB::bind_method(_MD("get_edge_meta","idx"),&MeshDataTool::get_edge_meta); ObjectTypeDB::bind_method(_MD("get_face_vertex","idx","vertex"),&MeshDataTool::get_face_vertex); ObjectTypeDB::bind_method(_MD("get_face_edge","idx","edge"),&MeshDataTool::get_face_edge); ObjectTypeDB::bind_method(_MD("set_face_meta","idx","meta"),&MeshDataTool::set_face_meta); ObjectTypeDB::bind_method(_MD("get_face_meta","idx"),&MeshDataTool::get_face_meta); ObjectTypeDB::bind_method(_MD("get_face_normal","idx"),&MeshDataTool::get_face_normal); ObjectTypeDB::bind_method(_MD("set_material","material:Material"),&MeshDataTool::set_material); ObjectTypeDB::bind_method(_MD("get_material","material"),&MeshDataTool::get_material); } MeshDataTool::MeshDataTool(){ clear(); }