X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fmesh.py;h=3f9ab02e2bd3dabf8493088b114f398c77983659;hb=f241ccca7d0d3f3073a948377bd557a70aa8f1d2;hp=8f19676ac1127627e2a78771e949d89a9ce5b9c4;hpb=5bf3c7175c359ad8509702a4e0790609b39e6aec;p=libs%2Fgl.git diff --git a/blender/io_mspgl/mesh.py b/blender/io_mspgl/mesh.py index 8f19676a..3f9ab02e 100644 --- a/blender/io_mspgl/mesh.py +++ b/blender/io_mspgl/mesh.py @@ -48,11 +48,9 @@ class Vertex: if vertex.__class__==Vertex: self.uvs = vertex.uvs[:] self.tan = vertex.tan - self.bino = vertex.bino else: self.uvs = [] self.tan = None - self.bino = None self.index = vertex.index self.co = mathutils.Vector(vertex.co) self.normal = mathutils.Vector(vertex.normal) @@ -160,8 +158,8 @@ class Mesh: self.winding_test = mesh.winding_test self.smoothing = mesh.smoothing self.use_uv = mesh.use_uv - self.tbn_vecs = mesh.tbn_vecs - self.tbn_uvtex = mesh.tbn_uvtex + self.tangent_uvtex = mesh.tangent_uvtex + self.use_strips = mesh.use_strips self.vertex_groups = mesh.vertex_groups # Clone basic data @@ -231,6 +229,21 @@ class Mesh: else: self.lines = [] + # Check if tangent vectors are needed + if mesh.tangent_vecs=='NO': + self.tangent_vecs = False + elif mesh.tangent_vecs=='YES': + self.tangent_vecs = True + elif mesh.tangent_vecs=='AUTO': + from .material import Material + self.tangent_vecs = False + for m in self.materials: + mat = Material(m) + if mat.type=="pbr": + normal_prop = next((p for p in mat.properties if p.tex_keyword=="normal_map"), None) + if normal_prop and normal_prop.texture: + self.tangent_vecs = True + self.vertex_sequence = [] def transform(self, matrix): @@ -447,26 +460,31 @@ class Mesh: prog_count = len(self.uv_layers) prog_step = 0 - # Split by the UV layer used for TBN vectors first so connectivity - # remains intact for TBN vector computation - tbn_layer_index = -1 - if self.tbn_vecs: - uv_names = [u.name for u in self.uv_layers] - if self.tbn_uvtex in uv_names: + # Split by the UV layer used for tangent vectors first so connectivity + # remains intact for tangent vector computation + tangent_layer_index = -1 + if self.tangent_vecs: + if self.tangent_uvtex: + uv_names = [u.name for u in self.uv_layers] + if self.tangent_uvtex in uv_names: + tangent_layer_index = uv_names.index(self.tangent_uvtex) + elif self.uv_layers[0].unit==0: + tangent_layer_index = 0 + + if tangent_layer_index>=0: prog_count += 1 - tbn_layer_index = uv_names.index(self.tbn_uvtex) - progress.push_task_slice("Computing TBN", 0, prog_count) - self.split_vertices(self.find_uv_group, progress, tbn_layer_index) - progress.set_task_slice(self.tbn_uvtex, 1, prog_count) - self.compute_tbn(tbn_layer_index, progress) + progress.push_task_slice("Computing tangents", 0, prog_count) + self.split_vertices(self.find_uv_group, progress, tangent_layer_index) + progress.set_task_slice(self.tangent_uvtex, 1, prog_count) + self.compute_tangents(tangent_layer_index, progress) progress.pop_task() prog_step = 2 else: - raise Exception("TBN UV layer not found") + raise Exception("Tangent UV layer not found") # Split by the remaining UV layers for i, u in enumerate(self.uv_layers): - if i==tbn_layer_index: + if i==tangent_layer_index: continue progress.push_task_slice(u.name, prog_step, prog_count) @@ -615,14 +633,11 @@ class Mesh: progress.set_progress(i/len(self.vertices)) - def compute_tbn(self, index, progress): - # This function is called at an early stage during UV preparation when - # face UVs are not available yet + def compute_tangents(self, index, progress): layer_uvs = self.uv_layers[index].uvs for i, v in enumerate(self.vertices): v.tan = mathutils.Vector() - v.bino = mathutils.Vector() for f in v.faces: vi = f.pivot_vertex(v) uv0 = layer_uvs[f.loop_indices[vi[0]]] @@ -638,12 +653,9 @@ class Mesh: if div: mul = edge1.angle(edge2)/div v.tan += (edge1*dv2-edge2*dv1)*mul - v.bino += (edge2*du1-edge1*du2)*mul if v.tan.length: v.tan.normalize() - if v.bino.length: - v.bino.normalize() progress.set_progress(i/len(self.vertices)) @@ -783,7 +795,7 @@ class Mesh: e.key = make_edge_key(e.vertices[0].index, e.vertices[1].index) -def create_mesh_from_object(context, obj, progress, *, material_atlas=None): +def create_mesh_from_object(context, obj, material_atlas, progress): if obj.type!="MESH": raise Exception("Object is not a mesh") @@ -812,8 +824,8 @@ def create_mesh_from_object(context, obj, progress, *, material_atlas=None): bmesh.vertex_groups = o.data.vertex_groups bmesh.max_groups_per_vertex = o.data.max_groups_per_vertex bmesh.use_uv = o.data.use_uv - bmesh.tbn_vecs = o.data.tbn_vecs - bmesh.tbn_uvtex = o.data.tbn_uvtex + bmesh.tangent_vecs = o.data.tangent_vecs + bmesh.tangent_uvtex = o.data.tangent_uvtex me = Mesh(bmesh) me.transform(m)