forked from LuffyTheFox/blender-addons
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmesh_auto_mirror.py
221 lines (180 loc) · 10.4 KB
/
mesh_auto_mirror.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
######################################################################################################
# An simple add-on to auto cut in two and mirror an object #
# Actualy partialy uncommented (see further version) #
# Author: Lapineige #
# License: GPL v3 #
######################################################################################################
############# Add-on description (used by Blender)
bl_info = {
"name": "Auto Mirror",
"description": "Super fast cutting and mirroring for mesh",
"author": "Lapineige",
"version": (2, 4, 1),
"blender": (2, 7, 1),
"location": "View 3D > Toolbar > Tools tab > AutoMirror (panel)",
"warning": "",
"wiki_url": "https://wiki.blender.org/index.php/Extensions:2.6/Py/Scripts/Modeling/AutoMirror",
"tracker_url": "https://developer.blender.org/maniphest/task/edit/form/2/",
"category": "Mesh"}
import bpy
from mathutils import Vector
bpy.types.Scene.AutoMirror_axis = bpy.props.EnumProperty(items = [("x", "X", "", 1),
("y", "Y", "", 2),
("z", "Z", "", 3)],
description="Axis used by the mirror modifier")
bpy.types.Scene.AutoMirror_orientation = bpy.props.EnumProperty(items = [("positive","Positive", "", 1),
("negative", "Negative", "", 2)],
description="Choose the side along the axis of the editable part (+/- coordinates)")
bpy.types.Scene.AutoMirror_threshold = bpy.props.FloatProperty(default= 0.001,
min= 0.001,
description="Vertices closer than this distance are merged on the loopcut")
bpy.types.Scene.AutoMirror_toggle_edit = bpy.props.BoolProperty(default= True,
description="If not in edit mode, change mode to edit")
bpy.types.Scene.AutoMirror_cut = bpy.props.BoolProperty(default= True,
description="If enabeled, cut the mesh in two parts and mirror it. If not, just make a loopcut")
bpy.types.Scene.AutoMirror_clipping = bpy.props.BoolProperty(default=True)
bpy.types.Scene.AutoMirror_use_clip = bpy.props.BoolProperty(default=True,
description="Use clipping for the mirror modifier")
bpy.types.Scene.AutoMirror_show_on_cage = bpy.props.BoolProperty(default=True,
description="Enable to edit the cage (it's the classical modifier's option)")
bpy.types.Scene.AutoMirror_apply_mirror = bpy.props.BoolProperty(description="Apply the mirror modifier (useful to symmetrise the mesh)")
############### Operator
class AlignVertices(bpy.types.Operator):
""" """
bl_idname = "object.align_vertices"
bl_label = "Align Vertices on 1 Axis"
@classmethod
def poll(cls, context):
return True
def execute(self, context):
bpy.ops.object.mode_set(mode = 'OBJECT')
x1,y1,z1 = bpy.context.scene.cursor_location
bpy.ops.view3d.snap_cursor_to_selected()
x2,y2,z2 = bpy.context.scene.cursor_location
bpy.context.scene.cursor_location[0],bpy.context.scene.cursor_location[1],bpy.context.scene.cursor_location[2] = 0,0,0
#Vertices coordinate to 0 (local coordinate, so on the origin)
for vert in bpy.context.object.data.vertices:
if vert.select:
if bpy.context.scene.AutoMirror_axis == 'x':
axis = 0
elif bpy.context.scene.AutoMirror_axis == 'y':
axis = 1
elif bpy.context.scene.AutoMirror_axis == 'z':
axis = 2
vert.co[axis] = 0
#
bpy.context.scene.cursor_location = x2,y2,z2
bpy.ops.object.origin_set(type='ORIGIN_CURSOR')
bpy.context.scene.cursor_location = x1,y1,z1
bpy.ops.object.mode_set(mode = 'EDIT')
return {'FINISHED'}
class AutoMirror(bpy.types.Operator):
""" Automatically cut an object along an axis """
bl_idname = "object.automirror"
bl_label = "AutoMirror"
bl_options = {'REGISTER'} # 'UNDO' ?
@classmethod
def poll(cls, context):
return True
def draw(self, context):
layout = self.layout
if bpy.context.object and bpy.context.object.type == 'MESH':
layout.prop(context.scene, "AutoMirror_axis", text="Mirror axis")
layout.prop(context.scene, "AutoMirror_orientation", text="Orientation")
layout.prop(context.scene, "AutoMirror_threshold", text="Threshold")
layout.prop(context.scene, "AutoMirror_toggle_edit", text="Toggle edit")
layout.prop(context.scene, "AutoMirror_cut", text="Cut and mirror")
if bpy.context.scene.AutoMirror_cut:
layout.prop(context.scene, "AutoMirror_clipping", text="Clipping")
layout.prop(context.scene, "AutoMirror_apply_mirror", text="Apply mirror")
else:
layout.label(icon="ERROR", text="No mesh selected")
def get_local_axis_vector(self, context, X, Y, Z, orientation):
loc = context.object.location
bpy.ops.object.mode_set(mode="OBJECT") # Needed to avoid to translate vertices
v1 = Vector((loc[0],loc[1],loc[2]))
bpy.ops.transform.translate(value=(X*orientation, Y*orientation, Z*orientation),
constraint_axis=((X==1), (Y==1), (Z==1)),
constraint_orientation='LOCAL')
v2 = Vector((loc[0],loc[1],loc[2]))
bpy.ops.transform.translate(value=(-X*orientation, -Y*orientation, -Z*orientation),
constraint_axis=((X==1), (Y==1), (Z==1)),
constraint_orientation='LOCAL')
bpy.ops.object.mode_set(mode="EDIT")
return v2-v1
def execute(self, context):
X,Y,Z = 0,0,0
if bpy.context.scene.AutoMirror_axis == 'x':
X = 1
elif bpy.context.scene.AutoMirror_axis == 'y':
Y = 1
elif bpy.context.scene.AutoMirror_axis == 'z':
Z = 1
current_mode = bpy.context.object.mode # Save the current mode
if bpy.context.object.mode != "EDIT":
bpy.ops.object.mode_set(mode="EDIT") # Go to edit mode
bpy.ops.mesh.select_all(action='SELECT') # Select all the vertices
if bpy.context.scene.AutoMirror_orientation == 'positive':
orientation = 1
else:
orientation = -1
cut_normal = self.get_local_axis_vector(context, X, Y, Z, orientation)
bpy.ops.mesh.bisect(plane_co=(bpy.context.object.location[0],
bpy.context.object.location[1],
bpy.context.object.location[2]),
plane_no=cut_normal,
use_fill= False,
clear_inner= bpy.context.scene.AutoMirror_cut,
clear_outer= 0,
threshold= bpy.context.scene.AutoMirror_threshold) # Cut the mesh
bpy.ops.object.align_vertices() # Use to align the vertices on the origin, needed by the "threshold"
if not bpy.context.scene.AutoMirror_toggle_edit:
bpy.ops.object.mode_set(mode=current_mode) # Reload previous mode
if bpy.context.scene.AutoMirror_cut:
bpy.ops.object.modifier_add(type='MIRROR') # Add a mirror modifier
bpy.context.object.modifiers[-1].use_x = X # Choose the axis to use, based on the cut's axis
bpy.context.object.modifiers[-1].use_y = Y
bpy.context.object.modifiers[-1].use_z = Z
bpy.context.object.modifiers[-1].use_clip = context.scene.AutoMirror_use_clip
bpy.context.object.modifiers[-1].show_on_cage = context.scene.AutoMirror_show_on_cage
if bpy.context.scene.AutoMirror_apply_mirror:
bpy.ops.object.mode_set(mode='OBJECT')
bpy.ops.object.modifier_apply(apply_as= 'DATA', modifier= bpy.context.object.modifiers[-1].name)
if bpy.context.scene.AutoMirror_toggle_edit:
bpy.ops.object.mode_set(mode='EDIT')
else:
bpy.ops.object.mode_set(mode=current_mode)
return {'FINISHED'}
#################### Panel
class BisectMirror(bpy.types.Panel):
""" The AutoMirror panel on the toolbar tab 'Tools' """
bl_label = "Auto Mirror"
bl_space_type = 'VIEW_3D'
bl_region_type = 'TOOLS'
bl_category = "Tools"
def draw(self, context):
layout = self.layout
if bpy.context.object and bpy.context.object.type == 'MESH':
layout.operator("object.automirror")
layout.prop(context.scene, "AutoMirror_axis", text="Mirror Axis", expand=True)
layout.prop(context.scene, "AutoMirror_orientation", text="Orientation")
layout.prop(context.scene, "AutoMirror_threshold", text="Threshold")
layout.prop(context.scene, "AutoMirror_toggle_edit", text="Toggle Edit")
layout.prop(context.scene, "AutoMirror_cut", text="Cut and Mirror")
if bpy.context.scene.AutoMirror_cut:
row = layout.row()
row.prop(context.scene, "AutoMirror_use_clip", text="Use Clip")
row.prop(context.scene, "AutoMirror_show_on_cage", text="Editable")
layout.prop(context.scene, "AutoMirror_apply_mirror", text="Apply Mirror")
else:
layout.label(icon="ERROR", text="No mesh selected")
def register():
bpy.utils.register_class(BisectMirror)
bpy.utils.register_class(AutoMirror)
bpy.utils.register_class(AlignVertices)
def unregister():
bpy.utils.unregister_class(BisectMirror)
bpy.utils.unregister_class(AutoMirror)
bpy.utils.unregister_class(AlignVertices)
if __name__ == "__main__":
register()