fa36e48d47b64a5468ec21c992001942481f8c57
[blender-staging.git] / release / scripts / startup / bl_operators / 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-80 compliant>
20
21 import bpy
22 from bpy.types import Operator
23 from mathutils import Vector
24
25
26 def randomize_selected(seed, delta, loc, rot, scale, scale_even, scale_min):
27
28     import random
29     from random import uniform
30
31     random.seed(seed)
32
33     def rand_vec(vec_range):
34         return Vector(uniform(-val, val) for val in vec_range)
35
36     for obj in bpy.context.selected_objects:
37
38         if loc:
39             if delta:
40                 obj.delta_location += rand_vec(loc)
41             else:
42                 obj.location += rand_vec(loc)
43         else:  # otherwise the values change under us
44             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
45
46         if rot:
47             vec = rand_vec(rot)
48
49             rotation_mode = obj.rotation_mode
50             if rotation_mode in {'QUATERNION', 'AXIS_ANGLE'}:
51                 obj.rotation_mode = 'XYZ'
52                 
53             if delta:
54                 obj.delta_rotation_euler[0] += vec[0]
55                 obj.delta_rotation_euler[1] += vec[1]
56                 obj.delta_rotation_euler[2] += vec[2]
57             else:
58                 obj.rotation_euler[0] += vec[0]
59                 obj.rotation_euler[1] += vec[1]
60                 obj.rotation_euler[2] += vec[2]
61             obj.rotation_mode = rotation_mode
62         else:
63             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
64
65         if scale:
66             if delta:
67                 org_sca_x, org_sca_y, org_sca_z = obj.delta_scale
68             else:
69                 org_sca_x, org_sca_y, org_sca_z = obj.scale
70
71             sca_x, sca_y, sca_z = (uniform(-scale[0] + 2.0, scale[0]),
72                                    uniform(-scale[1] + 2.0, scale[1]),
73                                    uniform(-scale[2] + 2.0, scale[2]))
74
75             if scale_even:
76                 aX = sca_x * org_sca_x
77                 aY = sca_x * org_sca_y
78                 aZ = sca_x * org_sca_z
79             else:
80                 aX = sca_x * org_sca_x
81                 aY = sca_y * org_sca_y
82                 aZ = sca_z * org_sca_z
83
84             if delta:
85                 obj.delta_scale = aX, aY, aZ
86             else:
87                 obj.scale = aX, aY, aZ
88         else:
89             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
90
91
92 from bpy.props import (IntProperty,
93                        BoolProperty,
94                        FloatVectorProperty)
95 from math import pi
96
97
98 class RandomizeLocRotSize(Operator):
99     '''Randomize objects loc/rot/scale'''
100     bl_idname = "object.randomize_transform"
101     bl_label = "Randomize Transform"
102     bl_options = {'REGISTER', 'UNDO'}
103
104     random_seed = IntProperty(
105             name="Random Seed",
106             description="Seed value for the random generator",
107             min=0,
108             max=10000,
109             default=0,
110             )
111     use_delta = BoolProperty(
112             name="Transform Delta",
113             description=("Randomize delta transform values "
114                          "instead of regular transform"),
115             default=False,
116             )
117     use_loc = BoolProperty(
118             name="Randomize Location",
119             description="Randomize the location values",
120             default=True,
121             )
122     loc = FloatVectorProperty(
123             name="Location",
124             description=("Maximun distance the objects "
125                          "can spread over each axis"),
126             min=-100.0,
127             max=100.0,
128             default=(0.0, 0.0, 0.0),
129             subtype='TRANSLATION',
130             )
131     use_rot = BoolProperty(
132             name="Randomize Rotation",
133             description="Randomize the rotation values",
134             default=True,
135             )
136     rot = FloatVectorProperty(
137             name="Rotation",
138             description="Maximun rotation over each axis",
139             min=-pi,
140             max=pi,
141             default=(0.0, 0.0, 0.0),
142             subtype='EULER',
143             )
144     use_scale = BoolProperty(
145             name="Randomize Scale",
146             description="Randomize the scale values",
147             default=True,
148             )
149     scale_even = BoolProperty(
150             name="Scale Even",
151             description="Use the same scale value for all axis",
152             default=False,
153             )
154
155     '''scale_min = FloatProperty(
156             name="Minimun Scale Factor",
157             description="Lowest scale percentage possible",
158             min=-1.0, max=1.0, precision=3,
159             default=0.15,
160             )'''
161
162     scale = FloatVectorProperty(
163             name="Scale",
164             description="Maximum scale randomization over each axis",
165             min=-100.0,
166             max=100.0,
167             default=(1.0, 1.0, 1.0),
168             subtype='TRANSLATION',
169             )
170
171     def execute(self, context):
172         seed = self.random_seed
173
174         delta = self.use_delta
175
176         loc = None if not self.use_loc else self.loc
177         rot = None if not self.use_rot else Vector(self.rot)
178         scale = None if not self.use_scale else self.scale
179
180         scale_even = self.scale_even
181         #scale_min = self.scale_min
182         scale_min = 0
183
184         randomize_selected(seed, delta, loc, rot, scale, scale_even, scale_min)
185
186         return {'FINISHED'}