Workspace: Move engines to workspace and Properties Editor cleanup
[blender.git] / release / scripts / templates_py / operator_mesh_add.py
1 import bpy
2 import bmesh
3
4
5 def add_box(width, height, depth):
6     """
7     This function takes inputs and returns vertex and face arrays.
8     no actual mesh data creation is done here.
9     """
10
11     verts = [(+1.0, +1.0, -1.0),
12              (+1.0, -1.0, -1.0),
13              (-1.0, -1.0, -1.0),
14              (-1.0, +1.0, -1.0),
15              (+1.0, +1.0, +1.0),
16              (+1.0, -1.0, +1.0),
17              (-1.0, -1.0, +1.0),
18              (-1.0, +1.0, +1.0),
19              ]
20
21     faces = [(0, 1, 2, 3),
22              (4, 7, 6, 5),
23              (0, 4, 5, 1),
24              (1, 5, 6, 2),
25              (2, 6, 7, 3),
26              (4, 0, 3, 7),
27             ]
28
29     # apply size
30     for i, v in enumerate(verts):
31         verts[i] = v[0] * width, v[1] * depth, v[2] * height
32
33     return verts, faces
34
35
36 from bpy.props import (
37         BoolProperty,
38         BoolVectorProperty,
39         FloatProperty,
40         FloatVectorProperty,
41         )
42
43
44 class AddBox(bpy.types.Operator):
45     """Add a simple box mesh"""
46     bl_idname = "mesh.primitive_box_add"
47     bl_label = "Add Box"
48     bl_options = {'REGISTER', 'UNDO'}
49
50     width = FloatProperty(
51             name="Width",
52             description="Box Width",
53             min=0.01, max=100.0,
54             default=1.0,
55             )
56     height = FloatProperty(
57             name="Height",
58             description="Box Height",
59             min=0.01, max=100.0,
60             default=1.0,
61             )
62     depth = FloatProperty(
63             name="Depth",
64             description="Box Depth",
65             min=0.01, max=100.0,
66             default=1.0,
67             )
68     layers = BoolVectorProperty(
69             name="Layers",
70             description="Object Layers",
71             size=20,
72             options={'HIDDEN', 'SKIP_SAVE'},
73             )
74
75     # generic transform props
76     view_align = BoolProperty(
77             name="Align to View",
78             default=False,
79             )
80     location = FloatVectorProperty(
81             name="Location",
82             subtype='TRANSLATION',
83             )
84     rotation = FloatVectorProperty(
85             name="Rotation",
86             subtype='EULER',
87             )
88
89     def execute(self, context):
90
91         verts_loc, faces = add_box(self.width,
92                                    self.height,
93                                    self.depth,
94                                    )
95
96         mesh = bpy.data.meshes.new("Box")
97
98         bm = bmesh.new()
99
100         for v_co in verts_loc:
101             bm.verts.new(v_co)
102
103         bm.verts.ensure_lookup_table()
104         for f_idx in faces:
105             bm.faces.new([bm.verts[i] for i in f_idx])
106
107         bm.to_mesh(mesh)
108         mesh.update()
109
110         # add the mesh as an object into the scene with this utility module
111         from bpy_extras import object_utils
112         object_utils.object_data_add(context, mesh, operator=self)
113
114         return {'FINISHED'}
115
116
117 def menu_func(self, context):
118     self.layout.operator(AddBox.bl_idname, icon='MESH_CUBE')
119
120
121 def register():
122     bpy.utils.register_class(AddBox)
123     bpy.types.INFO_MT_mesh_add.append(menu_func)
124
125
126 def unregister():
127     bpy.utils.unregister_class(AddBox)
128     bpy.types.INFO_MT_mesh_add.remove(menu_func)
129
130 if __name__ == "__main__":
131     register()
132
133     # test call
134     bpy.ops.mesh.primitive_box_add()