change python scripts so modules which register with blender have a register() functi...
[blender.git] / release / scripts / op / object_randomize_transform.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
23
24 def randomize_selected(seed, loc, rot, scale, scale_even, scale_min):
25
26     import random
27     from random import uniform
28     from Mathutils import Vector
29
30     random.seed(seed)
31
32     def rand_vec(vec_range):
33         return Vector([uniform(- val, val) for val in vec_range])
34
35     for obj in bpy.context.selected_objects:
36
37         if loc:
38             obj.location += rand_vec(loc)
39         else: # otherwise the values change under us
40             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
41
42         if rot: # TODO, non euler's
43             vec = rand_vec(rot)
44             obj.rotation_euler[0] += vec[0]
45             obj.rotation_euler[1] += vec[1]
46             obj.rotation_euler[2] += vec[2]
47         else:
48             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
49
50         if scale:
51             org_sca_x, org_sca_y, org_sca_z = obj.scale
52
53             if scale_even:
54                 sca_x = sca_y = sca_z = uniform(scale[0], - scale[0])
55                 uniform(0.0, 0.0), uniform(0.0, 0.0)
56             else:
57                 sca_x, sca_y, sca_z = rand_vec(scale)
58
59             aX = sca_x + org_sca_x
60             bX = org_sca_x * scale_min
61
62             aY = sca_y + org_sca_y
63             bY = org_sca_y * scale_min
64
65             aZ = sca_z + org_sca_z
66             bZ = org_sca_z * scale_min
67
68             if aX < bX:
69                 aX = bX
70             if aY < bY:
71                 aY = bY
72             if aZ < bZ:
73                 aZ = bZ
74
75             obj.scale = aX, aY, aZ
76         else:
77             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
78
79 from bpy.props import *
80
81
82 class RandomizeLocRotSize(bpy.types.Operator):
83     '''Randomize objects loc/rot/scale'''
84     bl_idname = "object.randomize_transform"
85     bl_label = "Randomize Transform"
86     bl_register = True
87     bl_undo = True
88
89     random_seed = IntProperty(name="Random Seed",
90         description="Seed value for the random generator",
91         default=0, min=0, max=1000)
92
93     use_loc = BoolProperty(name="Randomize Location",
94         description="Randomize the scale values", default=True)
95
96     loc = FloatVectorProperty(name="Location",
97         description="Maximun distance the objects can spread over each axis",
98         default=(0.0, 0.0, 0.0), min=-100.0, max=100.0, subtype='TRANSLATION')
99
100     use_rot = BoolProperty(name="Randomize Rotation",
101         description="Randomize the rotation values", default=True)
102
103     rot = FloatVectorProperty(name="Rotation",
104         description="Maximun rotation over each axis",
105         default=(0.0, 0.0, 0.0), min=-180.0, max=180.0, subtype='TRANSLATION')
106
107     use_scale = BoolProperty(name="Randomize Scale",
108         description="Randomize the scale values", default=True)
109
110     scale_even = BoolProperty(name="Scale Even",
111         description="Use the same scale value for all axis", default=False)
112
113     scale_min = FloatProperty(name="Minimun Scale Factor",
114         description="Lowest scale percentage possible",
115         default=0.15, min=-1.0, max=1.0, precision=3)
116
117     scale = FloatVectorProperty(name="Scale",
118         description="Maximum scale randomization over each axis",
119         default=(0.0, 0.0, 0.0), min=-100.0, max=100.0, subtype='TRANSLATION')
120
121     def execute(self, context):
122         from math import radians
123         seed = self.properties.random_seed
124
125         loc = None if not self.properties.use_loc else self.properties.loc
126         rot = None if not self.properties.use_rot else self.properties.rot * radians(1.0)
127         scale = None if not self.properties.use_scale else self.properties.scale
128
129         scale_even = self.properties.scale_even
130         scale_min = self.properties.scale_min
131
132         randomize_selected(seed, loc, rot, scale, scale_even, scale_min)
133
134         return {'FINISHED'}
135
136
137 def menu_func(self, context):
138     if context.mode == 'OBJECT':
139         self.layout.operator(RandomizeLocRotSize.bl_idname,
140         text="Randomize Transform")
141
142
143 def register():
144     bpy.types.register(RandomizeLocRotSize)
145     bpy.types.VIEW3D_MT_transform.append(menu_func)
146
147 def unregister():
148     bpy.types.unregister(RandomizeLocRotSize)
149     bpy.types.VIEW3D_MT_transform.remove(menu_func)
150