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