20f0ffdbe60bfce626d591f45c3672ab6a9a6d17
[blender.git] / release / scripts / modules / add_object_utils.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 compliant>
20
21 import bpy
22 import mathutils
23
24
25 def add_object_align_init(context, operator):
26
27     if operator and operator.properties.is_property_set("location") and operator.properties.is_property_set("rotation"):
28         location = mathutils.Matrix.Translation(mathutils.Vector(operator.properties.location))
29         rotation = mathutils.Euler(operator.properties.rotation).to_matrix().resize4x4()
30     else:
31         # TODO, local view cursor!
32         location = mathutils.Matrix.Translation(context.scene.cursor_location)
33
34         if context.user_preferences.edit.object_align == 'VIEW' and context.space_data.type == 'VIEW_3D':
35             rotation = context.space_data.region_3d.view_matrix.rotation_part().invert().resize4x4()
36         else:
37             rotation = mathutils.Matrix()
38
39         # set the operator properties
40         if operator:
41             operator.properties.location = location.translation_part()
42             operator.properties.rotation = rotation.to_euler()
43
44     return location * rotation
45
46
47 def object_data_add(context, obdata, operator=None):
48
49     scene = context.scene
50
51     # ugh, could be made nicer
52     for ob in scene.objects:
53         ob.select = False
54
55     obj_new = bpy.data.objects.new(obdata.name, obdata)
56
57     base = scene.objects.link(obj_new)
58     base.select = True
59
60     if context.space_data and context.space_data.type == 'VIEW_3D':
61         base.layers_from_view(context.space_data)
62
63     obj_new.matrix_world = add_object_align_init(context, operator)
64
65     obj_act = scene.objects.active
66
67     if obj_act and obj_act.mode == 'EDIT' and obj_act.type == obj_new.type:
68         bpy.ops.object.mode_set(mode='OBJECT')
69
70         obj_act.select = True
71         scene.update()  # apply location
72         #scene.objects.active = obj_new
73
74         bpy.ops.object.join()  # join into the active.
75
76         bpy.ops.object.mode_set(mode='EDIT')
77     else:
78         scene.objects.active = obj_new
79         if context.user_preferences.edit.use_enter_edit_mode:
80             bpy.ops.object.mode_set(mode='EDIT')
81
82     return base