UV mirror tool (copies UVs from one side of the mesh to the other)
WIP, suffers from editmode bug where editmode python tools cant have redo-options set. and needs options for precission.
This commit is contained in:
parent
32f4877c8c
commit
58f13d469e
@ -388,10 +388,19 @@ class MeshEdge(StructRNA):
|
|||||||
|
|
||||||
class MeshFace(StructRNA):
|
class MeshFace(StructRNA):
|
||||||
__slots__ = ()
|
__slots__ = ()
|
||||||
|
@property
|
||||||
|
def center(self):
|
||||||
|
"""The midpoint of the face."""
|
||||||
|
face_verts = self.verts[:]
|
||||||
|
mesh_verts = self.id_data.verts
|
||||||
|
if len(face_verts) == 3:
|
||||||
|
return (mesh_verts[face_verts[0]].co + mesh_verts[face_verts[1]].co + mesh_verts[face_verts[2]].co) / 3.0
|
||||||
|
else:
|
||||||
|
return (mesh_verts[face_verts[0]].co + mesh_verts[face_verts[1]].co + mesh_verts[face_verts[2]].co + mesh_verts[face_verts[3]].co) / 4.0
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def edge_keys(self):
|
def edge_keys(self):
|
||||||
verts = tuple(self.verts)
|
verts = self.verts[:]
|
||||||
if len(verts) == 3:
|
if len(verts) == 3:
|
||||||
return ord_ind(verts[0], verts[1]), ord_ind(verts[1], verts[2]), ord_ind(verts[2], verts[0])
|
return ord_ind(verts[0], verts[1]), ord_ind(verts[1], verts[2]), ord_ind(verts[2], verts[0])
|
||||||
|
|
||||||
|
@ -20,37 +20,6 @@
|
|||||||
|
|
||||||
import bpy
|
import bpy
|
||||||
|
|
||||||
|
|
||||||
def main(context):
|
|
||||||
ob = context.active_object
|
|
||||||
bpy.ops.mesh.selection_type(type='FACE')
|
|
||||||
is_editmode = (ob.mode == 'EDIT')
|
|
||||||
if is_editmode:
|
|
||||||
bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
|
|
||||||
|
|
||||||
mesh = ob.data
|
|
||||||
|
|
||||||
face_list = [face for face in mesh.faces]
|
|
||||||
face_edge_keys = [face.edge_keys for face in face_list]
|
|
||||||
|
|
||||||
edge_face_count = mesh.edge_face_count_dict
|
|
||||||
|
|
||||||
def test_interior(index):
|
|
||||||
for key in face_edge_keys[index]:
|
|
||||||
if edge_face_count[key] < 3:
|
|
||||||
return False
|
|
||||||
return True
|
|
||||||
|
|
||||||
for index, face in enumerate(face_list):
|
|
||||||
if(test_interior(index)):
|
|
||||||
face.selected = True
|
|
||||||
else:
|
|
||||||
face.selected = False
|
|
||||||
|
|
||||||
if is_editmode:
|
|
||||||
bpy.ops.object.mode_set(mode='EDIT', toggle=False)
|
|
||||||
|
|
||||||
|
|
||||||
class MeshSelectInteriorFaces(bpy.types.Operator):
|
class MeshSelectInteriorFaces(bpy.types.Operator):
|
||||||
'''Select faces where all edges have more then 2 face users.'''
|
'''Select faces where all edges have more then 2 face users.'''
|
||||||
|
|
||||||
@ -64,12 +33,149 @@ class MeshSelectInteriorFaces(bpy.types.Operator):
|
|||||||
return (ob and ob.type == 'MESH')
|
return (ob and ob.type == 'MESH')
|
||||||
|
|
||||||
def execute(self, context):
|
def execute(self, context):
|
||||||
main(context)
|
ob = context.active_object
|
||||||
|
bpy.ops.mesh.selection_type(type='FACE')
|
||||||
|
is_editmode = (ob.mode == 'EDIT')
|
||||||
|
if is_editmode:
|
||||||
|
bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
|
||||||
|
|
||||||
|
mesh = ob.data
|
||||||
|
|
||||||
|
face_list = [face for face in mesh.faces]
|
||||||
|
face_edge_keys = [face.edge_keys for face in face_list]
|
||||||
|
|
||||||
|
edge_face_count = mesh.edge_face_count_dict
|
||||||
|
|
||||||
|
def test_interior(index):
|
||||||
|
for key in face_edge_keys[index]:
|
||||||
|
if edge_face_count[key] < 3:
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
for index, face in enumerate(face_list):
|
||||||
|
if(test_interior(index)):
|
||||||
|
face.selected = True
|
||||||
|
else:
|
||||||
|
face.selected = False
|
||||||
|
|
||||||
|
if is_editmode:
|
||||||
|
bpy.ops.object.mode_set(mode='EDIT', toggle=False)
|
||||||
return {'FINISHED'}
|
return {'FINISHED'}
|
||||||
|
|
||||||
|
|
||||||
|
class MeshMirrorUV(bpy.types.Operator):
|
||||||
|
'''Copy mirror UV coordinates on the X axis based on a mirrored mesh'''
|
||||||
|
bl_idname = "mesh.faces_miror_uv"
|
||||||
|
bl_label = "Copy Mirrored UV coords"
|
||||||
|
bl_register = True
|
||||||
|
bl_undo = True
|
||||||
|
|
||||||
|
def poll(self, context):
|
||||||
|
ob = context.active_object
|
||||||
|
return (ob and ob.type == 'MESH')
|
||||||
|
|
||||||
|
def execute(self, context):
|
||||||
|
DIR = 1 # TODO, make an option
|
||||||
|
|
||||||
|
from Mathutils import Vector
|
||||||
|
|
||||||
|
ob = context.active_object
|
||||||
|
is_editmode = (ob.mode == 'EDIT')
|
||||||
|
if is_editmode:
|
||||||
|
bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
|
||||||
|
|
||||||
|
mesh = ob.data
|
||||||
|
|
||||||
|
# mirror lookups
|
||||||
|
mirror_gt = {}
|
||||||
|
mirror_lt = {}
|
||||||
|
|
||||||
|
vcos = [v.co.toTuple(5) for v in mesh.verts]
|
||||||
|
|
||||||
|
for i, co in enumerate(vcos):
|
||||||
|
if co[0] > 0.0:
|
||||||
|
mirror_gt[co] = i
|
||||||
|
elif co[0] < 0.0:
|
||||||
|
mirror_lt[co] = i
|
||||||
|
else:
|
||||||
|
mirror_gt[co] = i
|
||||||
|
mirror_lt[co] = i
|
||||||
|
|
||||||
|
#for i, v in enumerate(mesh.verts):
|
||||||
|
vmap = {}
|
||||||
|
for mirror_a, mirror_b in (mirror_gt, mirror_lt), (mirror_lt, mirror_gt):
|
||||||
|
for co, i in mirror_a.items():
|
||||||
|
nco = (-co[0], co[1], co[2])
|
||||||
|
j = mirror_b.get(nco)
|
||||||
|
if j is not None:
|
||||||
|
vmap[i] = j
|
||||||
|
|
||||||
|
|
||||||
|
active_uv_layer = None
|
||||||
|
for lay in mesh.uv_textures:
|
||||||
|
if lay.active:
|
||||||
|
active_uv_layer = lay.data
|
||||||
|
break
|
||||||
|
|
||||||
|
fuvs = [(uv.uv1, uv.uv2, uv.uv3, uv.uv4) for uv in active_uv_layer]
|
||||||
|
fuvs_cpy = [(uv[0].copy(), uv[1].copy(), uv[2].copy(), uv[3].copy()) for uv in fuvs]
|
||||||
|
|
||||||
|
# as a list
|
||||||
|
faces = mesh.faces[:]
|
||||||
|
|
||||||
|
fuvsel = [(False not in uv.uv_selected) for uv in active_uv_layer]
|
||||||
|
fcents = [f.center for f in faces]
|
||||||
|
|
||||||
|
# find mirror faces
|
||||||
|
mirror_fm = {}
|
||||||
|
for i, f in enumerate(faces):
|
||||||
|
verts = f.verts[:]
|
||||||
|
verts.sort()
|
||||||
|
verts = tuple(verts)
|
||||||
|
mirror_fm[verts] = i
|
||||||
|
|
||||||
|
fmap = {}
|
||||||
|
for i, f in enumerate(faces):
|
||||||
|
verts = [vmap.get(j) for j in f.verts]
|
||||||
|
if None not in verts:
|
||||||
|
verts.sort()
|
||||||
|
j = mirror_fm.get(tuple(verts))
|
||||||
|
if j is not None:
|
||||||
|
fmap[i] = j
|
||||||
|
|
||||||
|
done = [False] * len(faces)
|
||||||
|
for i, j in fmap.items():
|
||||||
|
|
||||||
|
if not fuvsel[i] or not fuvsel[j]:
|
||||||
|
continue
|
||||||
|
elif DIR == 0 and fcents[i][0] < 0.0:
|
||||||
|
continue
|
||||||
|
elif DIR == 1 and fcents[i][0] > 0.0:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# copy UVs
|
||||||
|
uv1= fuvs[i]
|
||||||
|
uv2= fuvs_cpy[j]
|
||||||
|
|
||||||
|
# get the correct rotation
|
||||||
|
v1 = faces[j].verts[:]
|
||||||
|
v2 = [vmap[k] for k in faces[i].verts[:]]
|
||||||
|
|
||||||
|
|
||||||
|
for k in range(len(uv1)):
|
||||||
|
k_map = v1.index(v2[k])
|
||||||
|
uv1[k].x = -(uv2[k_map].x-0.5) + 0.5
|
||||||
|
uv1[k].y = uv2[k_map].y
|
||||||
|
|
||||||
|
if is_editmode:
|
||||||
|
bpy.ops.object.mode_set(mode='EDIT', toggle=False)
|
||||||
|
|
||||||
|
return {'FINISHED'}
|
||||||
|
|
||||||
# Register the operator
|
# Register the operator
|
||||||
bpy.types.register(MeshSelectInteriorFaces)
|
bpy.types.register(MeshSelectInteriorFaces)
|
||||||
|
bpy.types.register(MeshMirrorUV)
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
bpy.ops.mesh.faces_select_interior()
|
# bpy.ops.mesh.faces_select_interior()
|
||||||
|
bpy.ops.mesh.faces_miror_uv()
|
||||||
|
@ -219,6 +219,7 @@ class IMAGE_MT_uvs(bpy.types.Menu):
|
|||||||
layout.operator("uv.average_islands_scale")
|
layout.operator("uv.average_islands_scale")
|
||||||
layout.operator("uv.minimize_stretch")
|
layout.operator("uv.minimize_stretch")
|
||||||
layout.operator("uv.stitch")
|
layout.operator("uv.stitch")
|
||||||
|
layout.operator("mesh.faces_miror_uv")
|
||||||
|
|
||||||
layout.separator()
|
layout.separator()
|
||||||
|
|
||||||
|
@ -1012,7 +1012,6 @@ static int ed_marker_make_links_scene_exec(bContext *C, wmOperator *op)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/* one day multiple scenes will be visible, then we should have some update function for them */
|
|
||||||
return OPERATOR_FINISHED;
|
return OPERATOR_FINISHED;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -99,7 +99,7 @@ TARGET_LINK_LIBRARIES(makesrna bf_dna)
|
|||||||
# Output rna_*_gen.c
|
# Output rna_*_gen.c
|
||||||
ADD_CUSTOM_COMMAND(
|
ADD_CUSTOM_COMMAND(
|
||||||
OUTPUT ${GENSRC}
|
OUTPUT ${GENSRC}
|
||||||
# with crashes try add this after COMMEND: valgrind --leak-check=full --track-origins=yes
|
# with crashes try add this after COMMAND: valgrind --leak-check=full --track-origins=yes
|
||||||
COMMAND ${CMAKE_BINARY_DIR}/bin/${CMAKE_CFG_INTDIR}/makesrna ${CMAKE_CURRENT_BINARY_DIR}/
|
COMMAND ${CMAKE_BINARY_DIR}/bin/${CMAKE_CFG_INTDIR}/makesrna ${CMAKE_CURRENT_BINARY_DIR}/
|
||||||
DEPENDS makesrna
|
DEPENDS makesrna
|
||||||
)
|
)
|
||||||
|
Loading…
Reference in New Issue
Block a user