68f97f63c735776bcc9070f4ec74ec3cd9c304b7
[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 compliant>
20
21 import bpy
22 from blf import gettext as _
23
24 def randomize_selected(seed, delta, loc, rot, scale, scale_even):
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             if delta:
39                 obj.delta_location += rand_vec(loc)
40             else:
41                 obj.location += rand_vec(loc)
42         else:  # otherwise the values change under us
43             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
44
45         if rot:  # TODO, non euler's
46             vec = rand_vec(rot)
47             if delta:
48                 obj.delta_rotation_euler[0] += vec[0]
49                 obj.delta_rotation_euler[1] += vec[1]
50                 obj.delta_rotation_euler[2] += vec[2]
51             else:
52                 obj.rotation_euler[0] += vec[0]
53                 obj.rotation_euler[1] += vec[1]
54                 obj.rotation_euler[2] += vec[2]
55         else:
56             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
57
58         if scale:
59             if delta:
60                 org_sca_x, org_sca_y, org_sca_z = obj.delta_scale
61             else:
62                 org_sca_x, org_sca_y, org_sca_z = obj.scale
63
64             if scale_even:
65                 sca_x = sca_y = sca_z = uniform(scale[0], - scale[0])
66                 uniform(0.0, 0.0), uniform(0.0, 0.0)
67             else:
68                 sca_x, sca_y, sca_z = rand_vec(scale)
69
70             if scale_even:
71                 aX = -(sca_x * org_sca_x) + org_sca_x
72                 aY = -(sca_x * org_sca_y) + org_sca_y
73                 aZ = -(sca_x * org_sca_z) + org_sca_z
74             else:
75                 aX = sca_x + org_sca_x
76                 aY = sca_y + org_sca_y
77                 aZ = sca_z + org_sca_z
78
79             if delta:
80                 obj.delta_scale = aX, aY, aZ
81             else:
82                 obj.scale = aX, aY, aZ
83         else:
84             uniform(0.0, 0.0), uniform(0.0, 0.0), uniform(0.0, 0.0)
85
86
87 from bpy.props import IntProperty, BoolProperty, FloatVectorProperty
88
89
90 class RandomizeLocRotSize(bpy.types.Operator):
91     '''Randomize objects loc/rot/scale'''
92     bl_idname = "object.randomize_transform"
93     bl_label = _("Randomize Transform")
94     bl_options = {'REGISTER', 'UNDO'}
95
96     random_seed = IntProperty(name="Random Seed",
97         description="Seed value for the random generator",
98         default=0, min=0, max=1000)
99
100     use_delta = BoolProperty(name="Transform Delta",
101         description="Randomize delta transform values instead of regular transform", default=False)
102
103     use_loc = BoolProperty(name="Randomize Location",
104         description="Randomize the location values", default=True)
105
106     loc = FloatVectorProperty(name="Location",
107         description="Maximun distance the objects can spread over each axis",
108         default=(0.0, 0.0, 0.0), min=-100.0, max=100.0, subtype='TRANSLATION')
109
110     use_rot = BoolProperty(name="Randomize Rotation",
111         description="Randomize the rotation values", default=True)
112
113     rot = FloatVectorProperty(name="Rotation",
114         description="Maximun rotation over each axis",
115         default=(0.0, 0.0, 0.0), min=-180.0, max=180.0, subtype='TRANSLATION')
116
117     use_scale = BoolProperty(name="Randomize Scale",
118         description="Randomize the scale values", default=True)
119
120     scale_even = BoolProperty(name="Scale Even",
121         description="Use the same scale value for all axis", default=False)
122
123     '''scale_min = FloatProperty(name="Minimun Scale Factor",
124         description="Lowest scale percentage possible",
125         default=0.15, min=-1.0, max=1.0, precision=3)'''
126
127     scale = FloatVectorProperty(name="Scale",
128         description="Maximum scale randomization over each axis",
129         default=(0.0, 0.0, 0.0), min=-100.0, max=100.0, subtype='TRANSLATION')
130
131     def execute(self, context):
132         from math import radians
133
134         seed = self.random_seed
135
136         delta = self.use_delta
137
138         loc = None if not self.use_loc else self.loc
139         rot = None if not self.use_rot else self.rot * radians(1.0)
140         scale = None if not self.use_scale else self.scale
141
142         scale_even = self.scale_even
143         #scale_min = self.scale_min
144
145         randomize_selected(seed, delta, loc, rot, scale, scale_even)
146
147         return {'FINISHED'}