re-commit temp workaround [#35920], this still fails for OSX retina display,
[blender-staging.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, IntProperty
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     precision = IntProperty(
40             name="Precision",
41             description=("Tolerance for finding vertex duplicates"),
42             min=1, max=16,
43             soft_min=1, soft_max=16,
44             default=3,
45             )
46
47     @classmethod
48     def poll(cls, context):
49         obj = context.active_object
50         return (obj and obj.type == 'MESH' and obj.data.uv_textures.active)
51
52     def execute(self, context):
53         DIR = (self.direction == 'NEGATIVE')
54         precision = self.precision
55         double_warn = 0
56
57         ob = context.active_object
58         is_editmode = (ob.mode == 'EDIT')
59         if is_editmode:
60             bpy.ops.object.mode_set(mode='OBJECT', toggle=False)
61
62         mesh = ob.data
63
64         # mirror lookups
65         mirror_gt = {}
66         mirror_lt = {}
67
68         vcos = (v.co.to_tuple(precision) for v in mesh.vertices)
69
70         for i, co in enumerate(vcos):
71             if co[0] >= 0.0:
72                 double_warn += co in mirror_gt
73                 mirror_gt[co] = i
74             if co[0] <= 0.0:
75                 double_warn += co in mirror_lt
76                 mirror_lt[co] = i
77
78         #for i, v in enumerate(mesh.vertices):
79         vmap = {}
80         for mirror_a, mirror_b in ((mirror_gt, mirror_lt),
81                                    (mirror_lt, mirror_gt)):
82             for co, i in mirror_a.items():
83                 nco = (-co[0], co[1], co[2])
84                 j = mirror_b.get(nco)
85                 if j is not None:
86                     vmap[i] = j
87
88         polys = mesh.polygons
89         loops = mesh.loops
90         verts = mesh.vertices
91         uv_loops = mesh.uv_layers.active.data
92         nbr_polys = len(polys)
93
94         mirror_pm = {}
95         pmap = {}
96         puvs = [None] * nbr_polys
97         puvs_cpy = [None] * nbr_polys
98         puvsel = [None] * nbr_polys
99         pcents = [None] * nbr_polys
100         vidxs = [None] * nbr_polys
101         for i, p in enumerate(polys):
102             lstart = lend = p.loop_start
103             lend += p.loop_total
104             puvs[i] = tuple(uv.uv for uv in uv_loops[lstart:lend])
105             puvs_cpy[i] = tuple(uv.copy() for uv in puvs[i])
106             puvsel[i] = (False not in
107                          (uv.select for uv in uv_loops[lstart:lend]))
108             # Vert idx of the poly.
109             vidxs[i] = tuple(l.vertex_index for l in loops[lstart:lend])
110             pcents[i] = p.center
111             # Preparing next step finding matching polys.
112             mirror_pm[tuple(sorted(vidxs[i]))] = i
113
114         for i in range(nbr_polys):
115             # Find matching mirror poly.
116             tvidxs = [vmap.get(j) for j in vidxs[i]]
117             if None not in tvidxs:
118                 tvidxs.sort()
119                 j = mirror_pm.get(tuple(tvidxs))
120                 if j is not None:
121                     pmap[i] = j
122
123         for i, j in pmap.items():
124             if not puvsel[i] or not puvsel[j]:
125                 continue
126             elif DIR == 0 and pcents[i][0] < 0.0:
127                 continue
128             elif DIR == 1 and pcents[i][0] > 0.0:
129                 continue
130
131             # copy UVs
132             uv1 = puvs[i]
133             uv2 = puvs_cpy[j]
134
135             # get the correct rotation
136             v1 = vidxs[j]
137             v2 = tuple(vmap[k] for k in vidxs[i])
138
139             if len(v1) == len(v2):
140                 for k in range(len(v1)):
141                     k_map = v1.index(v2[k])
142                     uv1[k].xy = - (uv2[k_map].x - 0.5) + 0.5, uv2[k_map].y
143
144         if is_editmode:
145             bpy.ops.object.mode_set(mode='EDIT', toggle=False)
146
147         if double_warn:
148             self.report({'WARNING'},
149                         "%d duplicates found, mirror may be incomplete" %
150                         double_warn)
151
152         return {'FINISHED'}