pep8 compliance for Randomize objects loc/rot/scale.
[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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, 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             else:
56                 sca_x, sca_y, sca_z = rand_vec(scale)
57
58             aX = sca_x + org_sca_x
59             bX = org_sca_x * scale_min
60
61             aY = sca_y + org_sca_y
62             bY = org_sca_y * scale_min
63
64             aZ = sca_z + org_sca_z
65             bZ = org_sca_z * scale_min
66
67             if aX < bX:
68                 aX = bX
69             if aY < bY:
70                 aY = bY
71             if aZ < bZ:
72                 aZ = bZ
73
74             obj.scale = aX, aY, aZ
75         else:
76             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
77
78 from bpy.props import *
79
80
81 class RandomizeLocRotSize(bpy.types.Operator):
82     '''Randomize objects loc/rot/scale.'''
83     bl_idname = "object.randomize_locrotsize"
84     bl_label = "Randomize Loc Rot Size"
85     bl_register = True
86     bl_undo = True
87
88     random_seed = IntProperty(name="Random Seed",
89         description="Seed value for the random generator",
90         default=0, min=0, max=1000)
91
92     use_loc = BoolProperty(name="Randomize Location",
93         description="Randomize the scale values", default=True)
94
95     loc = FloatVectorProperty(name="Location",
96         description="Maximun distance the objects can spread over each axis",
97         default=(0.0, 0.0, 0.0), min=-100.0, max=100.0)
98
99     use_rot = BoolProperty(name="Randomize Rotation",
100         description="Randomize the rotation values", default=True)
101
102     rot = FloatVectorProperty(name="Rotation",
103         description="Maximun rotation over each axis",
104         default=(0.0, 0.0, 0.0), min=-180.0, max=180.0)
105
106     use_scale = BoolProperty(name="Randomize Scale",
107         description="Randomize the scale values", default=True)
108
109     scale_even = BoolProperty(name="Scale Even",
110         description="Use the same scale value for all axis", default=False)
111
112     scale_min = FloatProperty(name="Minimun Scale Factor",
113         description="Lowest scale percentage possible",
114         default=15.0, min=-1.0, max=1.0, precision=3)
115
116     scale = FloatVectorProperty(name="Scale",
117         description="Maximum scale randomization over each axis",
118         default=(0.0, 0.0, 0.0), min=-100.0, max=100.0)
119
120     def execute(self, context):
121         from math import radians
122         seed = self.properties.random_seed
123
124         loc = self.properties.loc if self.properties.use_loc else None
125         rot = self.properties.rot if self.properties.use_rot else None
126         scale = [radians(val) for val in self.properties.scale] if self.properties.use_scale else None
127
128         scale_even = self.properties.scale_even
129         scale_min = self.properties.scale_min
130
131         randomize_selected(seed, loc, rot, scale, scale_even, scale_min)
132
133         return {'FINISHED'}
134
135
136 # Register the operator
137 bpy.types.register(RandomizeLocRotSize)
138
139
140 def menu_func(self, context):
141     if context.mode == 'OBJECT':
142         self.layout.operator(RandomizeLocRotSize.bl_idname,
143         text="Randomize Loc Rot Size")
144
145 bpy.types.VIEW3D_MT_transform.append(menu_func)