/* * ***** BEGIN GPL LICENSE BLOCK ***** * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version 2 * of the License, or (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software Foundation, * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * * The Original Code is Copyright (C) 2014 by Blender Foundation * All rights reserved. * * ***** END GPL LICENSE BLOCK ***** */ /** \file blender/editors/object/object_random.c * \ingroup edobj */ #include "MEM_guardedalloc.h" #include "DNA_layer_types.h" #include "DNA_object_types.h" #include "BLI_math.h" #include "BLI_rand.h" #include "BKE_context.h" #include "BKE_layer.h" #include "RNA_access.h" #include "RNA_define.h" #include "WM_api.h" #include "WM_types.h" #include "ED_transverts.h" #include "object_intern.h" /** * Generic randomize vertices function */ static bool object_rand_transverts( TransVertStore *tvs, const float offset, const float uniform, const float normal_factor, const unsigned int seed) { bool use_normal = (normal_factor != 0.0f); struct RNG *rng; TransVert *tv; int a; if (!tvs || !(tvs->transverts)) { return false; } rng = BLI_rng_new(seed); tv = tvs->transverts; for (a = 0; a < tvs->transverts_tot; a++, tv++) { const float t = max_ff(0.0f, uniform + ((1.0f - uniform) * BLI_rng_get_float(rng))); float vec[3]; BLI_rng_get_float_unit_v3(rng, vec); if (use_normal && (tv->flag & TX_VERT_USE_NORMAL)) { float no[3]; /* avoid >90d rotation to align with normal */ if (dot_v3v3(vec, tv->normal) < 0.0f) { negate_v3_v3(no, tv->normal); } else { copy_v3_v3(no, tv->normal); } interp_v3_v3v3_slerp_safe(vec, vec, no, normal_factor); } madd_v3_v3fl(tv->loc, vec, offset * t); } BLI_rng_free(rng); return true; } static int object_rand_verts_exec(bContext *C, wmOperator *op) { ViewLayer *view_layer = CTX_data_view_layer(C); Object *ob_active = CTX_data_edit_object(C); const int ob_mode = ob_active->mode; const float offset = RNA_float_get(op->ptr, "offset"); const float uniform = RNA_float_get(op->ptr, "uniform"); const float normal_factor = RNA_float_get(op->ptr, "normal"); const unsigned int seed = RNA_int_get(op->ptr, "seed"); bool changed_multi = false; uint objects_len = 0; Object **objects = BKE_view_layer_array_from_objects_in_mode_unique_data(view_layer, CTX_wm_view3d(C), &objects_len, ob_mode); for (uint ob_index = 0; ob_index < objects_len; ob_index++) { Object *ob_iter = objects[ob_index]; TransVertStore tvs = { NULL }; if (ob_iter) { int mode = TM_ALL_JOINTS; if (normal_factor != 0.0f) { mode |= TX_VERT_USE_NORMAL; } ED_transverts_create_from_obedit(&tvs, ob_iter, mode); if (tvs.transverts_tot == 0) { continue; } int seed_iter = seed; /* This gives a consistent result regardless of object order. */ if (ob_index) { seed_iter += BLI_ghashutil_strhash_p(ob_iter->id.name); } object_rand_transverts(&tvs, offset, uniform, normal_factor, seed_iter); ED_transverts_update_obedit(&tvs, ob_iter); ED_transverts_free(&tvs); WM_event_add_notifier(C, NC_OBJECT | ND_DRAW, ob_iter); changed_multi = true; } } MEM_freeN(objects); return changed_multi ? OPERATOR_FINISHED : OPERATOR_CANCELLED; } void TRANSFORM_OT_vertex_random(struct wmOperatorType *ot) { /* identifiers */ ot->name = "Randomize"; ot->description = "Randomize vertices"; ot->idname = "TRANSFORM_OT_vertex_random"; /* api callbacks */ ot->exec = object_rand_verts_exec; ot->poll = ED_transverts_poll; /* flags */ ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO; /* props */ ot->prop = RNA_def_float( ot->srna, "offset", 0.1f, -FLT_MAX, FLT_MAX, "Amount", "Distance to offset", -10.0f, 10.0f); RNA_def_float(ot->srna, "uniform", 0.0f, 0.0f, 1.0f, "Uniform", "Increase for uniform offset distance", 0.0f, 1.0f); RNA_def_float(ot->srna, "normal", 0.0f, 0.0f, 1.0f, "Normal", "Align offset direction to normals", 0.0f, 1.0f); RNA_def_int(ot->srna, "seed", 0, 0, 10000, "Random Seed", "Seed for the random number generator", 0, 50); }