forked from bookyakuno/Blender-Scramble-Addon
-
Notifications
You must be signed in to change notification settings - Fork 0
/
VIEW3D_MT_mesh_add.py
189 lines (169 loc) · 6.75 KB
/
VIEW3D_MT_mesh_add.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
# 「3Dビュー」エリア > 「追加」メニュー > 「メッシュ」メニュー
# "3D View" Area > "Add" Menu > "Mesh" Menu
import bpy, bmesh
import math
from bpy.props import *
################
# オペレーター #
################
class AddSphereOnlySquare(bpy.types.Operator):
bl_idname = "mesh.add_sphere_only_square"
bl_label = "Squared-Polygon Sphere"
bl_description = "Add a sphere mesh composed of only quadrilateral polygons"
bl_options = {'REGISTER', 'UNDO'}
level : IntProperty(name="Number of Subdivisions", default=2, step=1, min=1, max=6, soft_min=1, soft_max=6)
radius : FloatProperty(name="Radius", default=1.0, step=10, precision=3, min=0.001, max=100, soft_min=0.001, soft_max=100)
view_align : BoolProperty(name="Align View", default=False)
location : FloatVectorProperty(name="Location", default=(0.0, 0.0, 0.0), step=10, precision=3, subtype='XYZ', min=-100, max=100, soft_min=-100, soft_max=100)
rotation : IntVectorProperty(name="Rotation", default=(0, 0, 0), step=1, subtype='XYZ', min=-360, max=360, soft_min=-360, soft_max=360)
enter_editmode = False
def execute(self, context):
isEdited = False
if (context.mode == 'EDIT_MESH'):
isEdited = True
activeObj = context.active_object
try:
bpy.ops.object.mode_set(mode="OBJECT")
except RuntimeError: pass
if (self.view_align):
bpy.ops.mesh.primitive_cube_add(size=self.radius*2, view_align=True, location=self.location)
else:
rotation = self.rotation
rotation = (math.radians(rotation[0]), math.radians(rotation[1]), math.radians(rotation[2]))
bpy.ops.mesh.primitive_cube_add(size=self.radius*2, location=self.location, rotation=rotation)
context.active_object.name = "SquarePolySphere"
subsurf = context.active_object.modifiers.new("temp", "SUBSURF")
subsurf.levels = self.level
bpy.ops.object.modifier_apply(modifier=subsurf.name)
bpy.ops.object.mode_set(mode="EDIT")
bpy.ops.transform.tosphere(value=1)
bpy.ops.object.mode_set(mode="OBJECT")
if (isEdited and False):
activeObj.select_set(True)
bpy.context.view_layer.objects.active = activeObj
bpy.ops.object.join()
bpy.ops.object.mode_set(mode="EDIT")
return {'FINISHED'}
class AddVertexOnlyObject(bpy.types.Operator):
bl_idname = "mesh.add_vertex_only_object"
bl_label = "Only One Vertex"
bl_description = "Add a mesh composed of only one vertex at the 3D cursor's location"
bl_options = {'REGISTER', 'UNDO'}
def execute(self, context):
if (context.mode != 'OBJECT'):
bpy.ops.object.mode_set(mode="OBJECT")
me = bpy.data.meshes.new("Vertex")
me.from_pydata([(0, 0, 0)], [], [])
me.update()
obj = bpy.data.objects.new("Vertex", me)
obj.data = me
bpy.context.collection.objects.link(obj)
bpy.ops.object.select_all(action='DESELECT')
obj.select_set(True)
bpy.context.view_layer.objects.active = obj
obj.location = context.scene.cursor.location[:]
bpy.ops.object.mode_set(mode="EDIT")
context.tool_settings.mesh_select_mode = (True, False, False)
return {'FINISHED'}
class CreateVertexGroupSplits(bpy.types.Operator):
bl_idname = "mesh.create_vertex_group_splits"
bl_label = "Separated Copy"
bl_description = "Add the active mesh's copy separated into parts based on its vertex groups"
bl_options = {'REGISTER', 'UNDO'}
threshold : FloatProperty(name="Threshold Weight for Inclusion in Group", default=0.5, min=0, max=1, soft_min=0, soft_max=1, step=3, precision=2)
delete_source : BoolProperty(name="Delete Original Object", default=False)
@classmethod
def poll(cls, context):
if (context.mode == 'OBJECT'):
for obj in context.selected_objects:
if (obj.type == 'MESH'):
if (len(obj.vertex_groups)):
return True
return False
def invoke(self, context, event):
return context.window_manager.invoke_props_dialog(self, width=300)
def draw(self, context):
sp = self.layout.split(factor=0.7)
sp.label(text="Threshold Weight of Included Vertices")
sp.prop(self, 'threshold', text="")
self.layout.prop(self, 'delete_source')
def execute(self, context):
for obj in context.selected_objects:
obj.select_set(False)
me = obj.data
bm = bmesh.new()
bm.from_mesh(me)
for vertex_group in obj.vertex_groups:
new_verts = []
new_verts_index = []
new_faces = []
for index, vert in enumerate(bm.verts):
for group in me.vertices[index].groups:
if (obj.vertex_groups[group.group].name == vertex_group.name):
if (self.threshold <= group.weight):
new_verts.append(vert.co[:])
new_verts_index.append(index)
break
for face in bm.faces:
for vert in face.verts:
if (vert.index not in new_verts_index):
break
else:
faces = []
for vert in face.verts:
faces.append(new_verts_index.index(vert.index))
new_faces.append(faces)
if (len(new_verts) and len(new_faces)):
new_me = bpy.data.meshes.new(obj.name +":"+ vertex_group.name)
new_me.from_pydata(new_verts, [], new_faces)
new_obj = bpy.data.objects.new(obj.name +":"+ vertex_group.name, new_me)
context.view_layer.active_layer_collection.collection.objects.link(new_obj)
new_obj.select_set(True)
bpy.context.view_layer.objects.active = new_obj
new_obj.location = obj.location[:]
new_obj.rotation_mode = obj.rotation_mode
if (obj.rotation_mode == 'QUATERNION'):
new_obj.rotation_quaternion = obj.rotation_quaternion[:]
elif (obj.rotation_mode == 'AXIS_ANGLE'):
new_obj.rotation_axis_angle = obj.rotation_axis_angle[:]
else:
new_obj.rotation_euler = obj.rotation_euler[:]
new_obj.scale = obj.scale[:]
if (self.delete_source):
context.scene.objects.unlink(obj)
return {'FINISHED'}
################
# クラスの登録 #
################
classes = [
AddSphereOnlySquare,
AddVertexOnlyObject,
CreateVertexGroupSplits
]
def register():
for cls in classes:
bpy.utils.register_class(cls)
def unregister():
for cls in classes:
bpy.utils.unregister_class(cls)
################
# メニュー追加 #
################
# メニューのオン/オフの判定
def IsMenuEnable(self_id):
for id in bpy.context.preferences.addons[__name__.partition('.')[0]].preferences.disabled_menu.split(','):
if (id == self_id):
return False
else:
return True
# メニューを登録する関数
def menu(self, context):
if (IsMenuEnable(__name__.split('.')[-1])):
self.layout.separator()
self.layout.operator(AddVertexOnlyObject.bl_idname, icon='PLUGIN')
self.layout.operator(AddSphereOnlySquare.bl_idname, icon='PLUGIN').location = context.scene.cursor.location
self.layout.separator()
self.layout.operator(CreateVertexGroupSplits.bl_idname, icon='PLUGIN')
if (context.preferences.addons[__name__.partition('.')[0]].preferences.use_disabled_menu):
self.layout.separator()
self.layout.operator('wm.toggle_menu_enable', icon='CANCEL').id = __name__.split('.')[-1]