moved select interior faces into a C function (was python)
[blender.git] / release / scripts / startup / bl_operators / mesh.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8-80 compliant>
20
21 import bpy
22 from bpy.types import Operator
23
24 from bpy.props import EnumProperty
25
26
27 class MeshMirrorUV(Operator):
28     '''Copy mirror UV coordinates on the X axis based on a mirrored mesh'''
29     bl_idname = "mesh.faces_mirror_uv"
30     bl_label = "Copy Mirrored UV coords"
31     bl_options = {'REGISTER', 'UNDO'}
32
33     direction = EnumProperty(
34             name="Axis Direction",
35             items=(('POSITIVE', "Positive", ""),
36                    ('NEGATIVE', "Negative", "")),
37             )
38
39     @classmethod
40     def poll(cls, context):
41         obj = context.active_object
42         return (obj and obj.type == 'MESH' and obj.data.uv_textures.active)
43
44     def execute(self, context):
45         DIR = (self.direction == 'NEGATIVE')
46
47         ob = context.active_object
48         is_editmode = (ob.mode == 'EDIT')
49         if is_editmode:
50             bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
51
52         mesh = ob.data
53
54         # mirror lookups
55         mirror_gt = {}
56         mirror_lt = {}
57
58         vcos = [v.co.to_tuple(5) for v in mesh.vertices]
59
60         for i, co in enumerate(vcos):
61             if co[0] > 0.0:
62                 mirror_gt[co] = i
63             elif co[0] < 0.0:
64                 mirror_lt[co] = i
65             else:
66                 mirror_gt[co] = i
67                 mirror_lt[co] = i
68
69         #for i, v in enumerate(mesh.vertices):
70         vmap = {}
71         for mirror_a, mirror_b in ((mirror_gt, mirror_lt),
72                                    (mirror_lt, mirror_gt)):
73             for co, i in mirror_a.items():
74                 nco = (-co[0], co[1], co[2])
75                 j = mirror_b.get(nco)
76                 if j is not None:
77                     vmap[i] = j
78
79         active_uv_layer = mesh.uv_textures.active.data
80         fuvs = [(uv.uv1, uv.uv2, uv.uv3, uv.uv4) for uv in active_uv_layer]
81         fuvs_cpy = [(uv[0].copy(), uv[1].copy(), uv[2].copy(), uv[3].copy())
82                     for uv in fuvs]
83
84         # as a list
85         faces = mesh.faces[:]
86
87         fuvsel = [(False not in uv.select_uv) for uv in active_uv_layer]
88         fcents = [f.center for f in faces]
89
90         # find mirror faces
91         mirror_fm = {}
92         for i, f in enumerate(faces):
93             verts = list(f.vertices)
94             verts.sort()
95             verts = tuple(verts)
96             mirror_fm[verts] = i
97
98         fmap = {}
99         for i, f in enumerate(faces):
100             verts = [vmap.get(j) for j in f.vertices]
101             if None not in verts:
102                 verts.sort()
103                 j = mirror_fm.get(tuple(verts))
104                 if j is not None:
105                     fmap[i] = j
106
107         for i, j in fmap.items():
108
109             if not fuvsel[i] or not fuvsel[j]:
110                 continue
111             elif DIR == 0 and fcents[i][0] < 0.0:
112                 continue
113             elif DIR == 1 and fcents[i][0] > 0.0:
114                 continue
115
116             # copy UVs
117             uv1 = fuvs[i]
118             uv2 = fuvs_cpy[j]
119
120             # get the correct rotation
121             v1 = faces[j].vertices[:]
122             v2 = [vmap[k] for k in faces[i].vertices[:]]
123
124             if len(v1) == len(v2):
125                 for k in range(len(v1)):
126                     k_map = v1.index(v2[k])
127                     uv1[k].xy = - (uv2[k_map].x - 0.5) + 0.5, uv2[k_map].y
128
129         if is_editmode:
130             bpy.ops.object.mode_set(mode='EDIT', toggle=False)
131
132         return {'FINISHED'}