Fix some inconsistencies in object visibility/selectability tests.
[blender.git] / source / blender / editors / object / object_random.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2014 by Blender Foundation
19  * All rights reserved.
20  *
21  * ***** END GPL LICENSE BLOCK *****
22  */
23
24 /** \file blender/editors/object/object_random.c
25  *  \ingroup edobj
26  */
27
28 #include "MEM_guardedalloc.h"
29
30 #include "DNA_layer_types.h"
31 #include "DNA_object_types.h"
32
33 #include "BLI_math.h"
34 #include "BLI_rand.h"
35
36
37 #include "BKE_context.h"
38 #include "BKE_layer.h"
39
40 #include "RNA_access.h"
41 #include "RNA_define.h"
42
43 #include "WM_api.h"
44 #include "WM_types.h"
45
46 #include "ED_transverts.h"
47
48 #include "object_intern.h"
49
50
51 /**
52  * Generic randomize vertices function
53  */
54
55 static bool object_rand_transverts(
56         TransVertStore *tvs,
57         const float offset, const float uniform, const float normal_factor,
58         const unsigned int seed)
59 {
60         bool use_normal = (normal_factor != 0.0f);
61         struct RNG *rng;
62         TransVert *tv;
63         int a;
64
65         if (!tvs || !(tvs->transverts)) {
66                 return false;
67         }
68
69         rng = BLI_rng_new(seed);
70
71         tv = tvs->transverts;
72         for (a = 0; a < tvs->transverts_tot; a++, tv++) {
73                 const float t = max_ff(0.0f, uniform + ((1.0f - uniform) * BLI_rng_get_float(rng)));
74                 float vec[3];
75                 BLI_rng_get_float_unit_v3(rng, vec);
76
77                 if (use_normal && (tv->flag & TX_VERT_USE_NORMAL)) {
78                         float no[3];
79
80                         /* avoid >90d rotation to align with normal */
81                         if (dot_v3v3(vec, tv->normal) < 0.0f) {
82                                 negate_v3_v3(no, tv->normal);
83                         }
84                         else {
85                                 copy_v3_v3(no, tv->normal);
86                         }
87
88                         interp_v3_v3v3_slerp_safe(vec, vec, no, normal_factor);
89                 }
90
91                 madd_v3_v3fl(tv->loc, vec, offset * t);
92         }
93
94         BLI_rng_free(rng);
95
96         return true;
97 }
98
99 static int object_rand_verts_exec(bContext *C, wmOperator *op)
100 {
101         ViewLayer *view_layer = CTX_data_view_layer(C);
102         Object *ob_active = CTX_data_edit_object(C);
103         const int ob_mode = ob_active->mode;
104
105         const float offset = RNA_float_get(op->ptr, "offset");
106         const float uniform = RNA_float_get(op->ptr, "uniform");
107         const float normal_factor = RNA_float_get(op->ptr, "normal");
108         const unsigned int seed = RNA_int_get(op->ptr, "seed");
109
110         bool changed_multi = false;
111         uint objects_len = 0;
112         Object **objects = BKE_view_layer_array_from_objects_in_mode_unique_data(view_layer, CTX_wm_view3d(C), &objects_len, ob_mode);
113         for (uint ob_index = 0; ob_index < objects_len; ob_index++) {
114                 Object *ob_iter = objects[ob_index];
115
116                 TransVertStore tvs = { NULL };
117
118                 if (ob_iter) {
119                         int mode = TM_ALL_JOINTS;
120
121                         if (normal_factor != 0.0f) {
122                                 mode |= TX_VERT_USE_NORMAL;
123                         }
124
125                         ED_transverts_create_from_obedit(&tvs, ob_iter, mode);
126                         if (tvs.transverts_tot == 0) {
127                                 continue;
128                         }
129
130                         int seed_iter = seed;
131                         /* This gives a consistent result regardless of object order. */
132                         if (ob_index) {
133                                 seed_iter += BLI_ghashutil_strhash_p(ob_iter->id.name);
134                         }
135
136                         object_rand_transverts(&tvs, offset, uniform, normal_factor, seed_iter);
137
138                         ED_transverts_update_obedit(&tvs, ob_iter);
139                         ED_transverts_free(&tvs);
140
141                         WM_event_add_notifier(C, NC_OBJECT | ND_DRAW, ob_iter);
142                         changed_multi = true;
143                 }
144         }
145         MEM_freeN(objects);
146
147         return changed_multi ? OPERATOR_FINISHED : OPERATOR_CANCELLED;
148 }
149
150 void TRANSFORM_OT_vertex_random(struct wmOperatorType *ot)
151 {
152         /* identifiers */
153         ot->name = "Randomize";
154         ot->description = "Randomize vertices";
155         ot->idname = "TRANSFORM_OT_vertex_random";
156
157         /* api callbacks */
158         ot->exec = object_rand_verts_exec;
159         ot->poll = ED_transverts_poll;
160
161         /* flags */
162         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
163
164         /* props */
165         ot->prop = RNA_def_float(
166                 ot->srna, "offset",  0.1f, -FLT_MAX, FLT_MAX,
167                 "Amount", "Distance to offset", -10.0f, 10.0f);
168         RNA_def_float(ot->srna, "uniform",  0.0f, 0.0f, 1.0f, "Uniform",
169                       "Increase for uniform offset distance", 0.0f, 1.0f);
170         RNA_def_float(ot->srna, "normal",  0.0f, 0.0f, 1.0f, "Normal",
171                       "Align offset direction to normals", 0.0f, 1.0f);
172         RNA_def_int(ot->srna, "seed", 0, 0, 10000, "Random Seed", "Seed for the random number generator", 0, 50);
173 }