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
  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(
  97             name="Random Seed",
  98             description="Seed value for the random generator",
  99             min=0,
 100             max=1000,
 101             default=0,
 102             )
 103     use_delta = BoolProperty(
 104             name="Transform Delta",
 105             description=("Randomize delta transform values "
 106                          "instead of regular transform"),
 107             default=False,
 108             )
 109     use_loc = BoolProperty(
 110             name="Randomize Location",
 111             description="Randomize the location values",
 112             default=True,
 113             )
 114     loc = FloatVectorProperty(
 115             name="Location",
 116             description=("Maximun distance the objects "
 117                          "can spread over each axis"),
 118             min=-100.0,
 119             max=100.0,
 120             default=(0.0, 0.0, 0.0),
 121             subtype='TRANSLATION',
 122             )
 123     use_rot = BoolProperty(
 124             name="Randomize Rotation",
 125             description="Randomize the rotation values",
 126             default=True,
 127             )
 128     rot = FloatVectorProperty(
 129             name="Rotation",
 130             description="Maximun rotation over each axis",
 131             min=-180.0,
 132             max=180.0,
 133             default=(0.0, 0.0, 0.0),
 134             subtype='TRANSLATION',
 135             )
 136     use_scale = BoolProperty(
 137             name="Randomize Scale",
 138             description="Randomize the scale values",
 139             default=True,
 140             )
 141     scale_even = BoolProperty(
 142             name="Scale Even",
 143             description="Use the same scale value for all axis",
 144             default=False,
 145             )
 146
 147     '''scale_min = FloatProperty(name="Minimun Scale Factor",
 148         description="Lowest scale percentage possible",
 149         default=0.15, min=-1.0, max=1.0, precision=3)'''
 150
 151     scale = FloatVectorProperty(
 152             name="Scale",
 153             description="Maximum scale randomization over each axis",
 154             min=-100.0,
 155             max=100.0,
 156             default=(0.0, 0.0, 0.0),
 157             subtype='TRANSLATION',
 158             )
 159
 160     def execute(self, context):
 161         from math import radians
 162
 163         seed = self.random_seed
 164
 165         delta = self.use_delta
 166
 167         loc = None if not self.use_loc else self.loc
 168         rot = None if not self.use_rot else self.rot * radians(1.0)
 169         scale = None if not self.use_scale else self.scale
 170
 171         scale_even = self.scale_even
 172         #scale_min = self.scale_min
 173
 174         randomize_selected(seed, delta, loc, rot, scale, scale_even)
 175
 176         return {'FINISHED'}