Jacques Lucke
efb741b280
Since the initial merge of the geometry nodes project, the modifyPointCloud function already was already modifying a geometry set. The function wasn't renamed back then, because then the merge would have touched many more files. Ref T83357.
315 lines
10 KiB
C
315 lines
10 KiB
C
/*
|
|
* 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) 2005 by the Blender Foundation.
|
|
* All rights reserved.
|
|
*/
|
|
|
|
/** \file
|
|
* \ingroup modifiers
|
|
*/
|
|
|
|
#include <string.h>
|
|
|
|
#include "BLI_listbase.h"
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "BLT_translation.h"
|
|
|
|
#include "DNA_armature_types.h"
|
|
#include "DNA_defaults.h"
|
|
#include "DNA_mesh_types.h"
|
|
#include "DNA_object_types.h"
|
|
#include "DNA_screen_types.h"
|
|
|
|
#include "BKE_action.h"
|
|
#include "BKE_armature.h"
|
|
#include "BKE_context.h"
|
|
#include "BKE_editmesh.h"
|
|
#include "BKE_lib_id.h"
|
|
#include "BKE_lib_query.h"
|
|
#include "BKE_mesh.h"
|
|
#include "BKE_mesh_wrapper.h"
|
|
#include "BKE_modifier.h"
|
|
#include "BKE_screen.h"
|
|
|
|
#include "UI_interface.h"
|
|
#include "UI_resources.h"
|
|
|
|
#include "RNA_access.h"
|
|
|
|
#include "BLO_read_write.h"
|
|
|
|
#include "DEG_depsgraph_query.h"
|
|
|
|
#include "bmesh.h"
|
|
#include "bmesh_tools.h"
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
#include "MOD_ui_common.h"
|
|
#include "MOD_util.h"
|
|
|
|
static void initData(ModifierData *md)
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
|
|
BLI_assert(MEMCMP_STRUCT_AFTER_IS_ZERO(amd, modifier));
|
|
|
|
MEMCPY_STRUCT_AFTER(amd, DNA_struct_default_get(ArmatureModifierData), modifier);
|
|
}
|
|
|
|
static void copyData(const ModifierData *md, ModifierData *target, const int flag)
|
|
{
|
|
#if 0
|
|
const ArmatureModifierData *amd = (const ArmatureModifierData *)md;
|
|
#endif
|
|
ArmatureModifierData *tamd = (ArmatureModifierData *)target;
|
|
|
|
BKE_modifier_copydata_generic(md, target, flag);
|
|
tamd->vert_coords_prev = NULL;
|
|
}
|
|
|
|
static void requiredDataMask(Object *UNUSED(ob),
|
|
ModifierData *UNUSED(md),
|
|
CustomData_MeshMasks *r_cddata_masks)
|
|
{
|
|
/* ask for vertexgroups */
|
|
r_cddata_masks->vmask |= CD_MASK_MDEFORMVERT;
|
|
}
|
|
|
|
static bool isDisabled(const struct Scene *UNUSED(scene),
|
|
ModifierData *md,
|
|
bool UNUSED(useRenderParams))
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
|
|
/* The object type check is only needed here in case we have a placeholder
|
|
* object assigned (because the library containing the armature is missing).
|
|
*
|
|
* In other cases it should be impossible to have a type mismatch.
|
|
*/
|
|
return !amd->object || amd->object->type != OB_ARMATURE;
|
|
}
|
|
|
|
static void foreachIDLink(ModifierData *md, Object *ob, IDWalkFunc walk, void *userData)
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
|
|
walk(userData, ob, (ID **)&amd->object, IDWALK_CB_NOP);
|
|
}
|
|
|
|
static void updateDepsgraph(ModifierData *md, const ModifierUpdateDepsgraphContext *ctx)
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
if (amd->object != NULL) {
|
|
/* If not using envelopes,
|
|
* create relations to individual bones for more rigging flexibility. */
|
|
if ((amd->deformflag & ARM_DEF_ENVELOPE) == 0 && (amd->object->pose != NULL) &&
|
|
ELEM(ctx->object->type, OB_MESH, OB_LATTICE, OB_GPENCIL)) {
|
|
/* If neither vertex groups nor envelopes are used, the modifier has no bone dependencies. */
|
|
if ((amd->deformflag & ARM_DEF_VGROUP) != 0) {
|
|
/* Enumerate groups that match existing bones. */
|
|
LISTBASE_FOREACH (bDeformGroup *, dg, &ctx->object->defbase) {
|
|
if (BKE_pose_channel_find_name(amd->object->pose, dg->name) != NULL) {
|
|
/* Can't check BONE_NO_DEFORM because it can be animated. */
|
|
DEG_add_bone_relation(
|
|
ctx->node, amd->object, dg->name, DEG_OB_COMP_BONE, "Armature Modifier");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
/* Otherwise require the whole pose to be complete. */
|
|
else {
|
|
DEG_add_object_relation(ctx->node, amd->object, DEG_OB_COMP_EVAL_POSE, "Armature Modifier");
|
|
}
|
|
|
|
DEG_add_object_relation(ctx->node, amd->object, DEG_OB_COMP_TRANSFORM, "Armature Modifier");
|
|
}
|
|
DEG_add_modifier_to_transform_relation(ctx->node, "Armature Modifier");
|
|
}
|
|
|
|
static void deformVerts(ModifierData *md,
|
|
const ModifierEvalContext *ctx,
|
|
Mesh *mesh,
|
|
float (*vertexCos)[3],
|
|
int numVerts)
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
|
|
MOD_previous_vcos_store(md, vertexCos); /* if next modifier needs original vertices */
|
|
|
|
BKE_armature_deform_coords_with_mesh(amd->object,
|
|
ctx->object,
|
|
vertexCos,
|
|
NULL,
|
|
numVerts,
|
|
amd->deformflag,
|
|
amd->vert_coords_prev,
|
|
amd->defgrp_name,
|
|
mesh);
|
|
|
|
/* free cache */
|
|
MEM_SAFE_FREE(amd->vert_coords_prev);
|
|
}
|
|
|
|
static void deformVertsEM(ModifierData *md,
|
|
const ModifierEvalContext *ctx,
|
|
struct BMEditMesh *em,
|
|
Mesh *mesh,
|
|
float (*vertexCos)[3],
|
|
int numVerts)
|
|
{
|
|
if (mesh != NULL) {
|
|
deformVerts(md, ctx, mesh, vertexCos, numVerts);
|
|
return;
|
|
}
|
|
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
|
|
MOD_previous_vcos_store(md, vertexCos); /* if next modifier needs original vertices */
|
|
|
|
BKE_armature_deform_coords_with_editmesh(amd->object,
|
|
ctx->object,
|
|
vertexCos,
|
|
NULL,
|
|
numVerts,
|
|
amd->deformflag,
|
|
amd->vert_coords_prev,
|
|
amd->defgrp_name,
|
|
em);
|
|
|
|
/* free cache */
|
|
MEM_SAFE_FREE(amd->vert_coords_prev);
|
|
}
|
|
|
|
static void deformMatricesEM(ModifierData *md,
|
|
const ModifierEvalContext *ctx,
|
|
struct BMEditMesh *em,
|
|
Mesh *UNUSED(mesh),
|
|
float (*vertexCos)[3],
|
|
float (*defMats)[3][3],
|
|
int numVerts)
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
|
|
BKE_armature_deform_coords_with_editmesh(amd->object,
|
|
ctx->object,
|
|
vertexCos,
|
|
defMats,
|
|
numVerts,
|
|
amd->deformflag,
|
|
NULL,
|
|
amd->defgrp_name,
|
|
em);
|
|
}
|
|
|
|
static void deformMatrices(ModifierData *md,
|
|
const ModifierEvalContext *ctx,
|
|
Mesh *mesh,
|
|
float (*vertexCos)[3],
|
|
float (*defMats)[3][3],
|
|
int numVerts)
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
Mesh *mesh_src = MOD_deform_mesh_eval_get(ctx->object, NULL, mesh, NULL, numVerts, false, false);
|
|
|
|
BKE_armature_deform_coords_with_mesh(amd->object,
|
|
ctx->object,
|
|
vertexCos,
|
|
defMats,
|
|
numVerts,
|
|
amd->deformflag,
|
|
NULL,
|
|
amd->defgrp_name,
|
|
mesh_src);
|
|
|
|
if (!ELEM(mesh_src, NULL, mesh)) {
|
|
BKE_id_free(NULL, mesh_src);
|
|
}
|
|
}
|
|
|
|
static void panel_draw(const bContext *UNUSED(C), Panel *panel)
|
|
{
|
|
uiLayout *col;
|
|
uiLayout *layout = panel->layout;
|
|
|
|
PointerRNA ob_ptr;
|
|
PointerRNA *ptr = modifier_panel_get_property_pointers(panel, &ob_ptr);
|
|
|
|
uiLayoutSetPropSep(layout, true);
|
|
|
|
uiItemR(layout, ptr, "object", 0, NULL, ICON_NONE);
|
|
modifier_vgroup_ui(layout, ptr, &ob_ptr, "vertex_group", "invert_vertex_group", NULL);
|
|
|
|
col = uiLayoutColumn(layout, true);
|
|
uiItemR(col, ptr, "use_deform_preserve_volume", 0, NULL, ICON_NONE);
|
|
uiItemR(col, ptr, "use_multi_modifier", 0, NULL, ICON_NONE);
|
|
|
|
col = uiLayoutColumnWithHeading(layout, true, IFACE_("Bind To"));
|
|
uiItemR(col, ptr, "use_vertex_groups", 0, IFACE_("Vertex Groups"), ICON_NONE);
|
|
uiItemR(col, ptr, "use_bone_envelopes", 0, IFACE_("Bone Envelopes"), ICON_NONE);
|
|
|
|
modifier_panel_end(layout, ptr);
|
|
}
|
|
|
|
static void panelRegister(ARegionType *region_type)
|
|
{
|
|
modifier_panel_register(region_type, eModifierType_Armature, panel_draw);
|
|
}
|
|
|
|
static void blendRead(BlendDataReader *UNUSED(reader), ModifierData *md)
|
|
{
|
|
ArmatureModifierData *amd = (ArmatureModifierData *)md;
|
|
|
|
amd->vert_coords_prev = NULL;
|
|
}
|
|
|
|
ModifierTypeInfo modifierType_Armature = {
|
|
/* name */ "Armature",
|
|
/* structName */ "ArmatureModifierData",
|
|
/* structSize */ sizeof(ArmatureModifierData),
|
|
/* srna */ &RNA_ArmatureModifier,
|
|
/* type */ eModifierTypeType_OnlyDeform,
|
|
/* flags */ eModifierTypeFlag_AcceptsCVs | eModifierTypeFlag_AcceptsVertexCosOnly |
|
|
eModifierTypeFlag_SupportsEditmode,
|
|
/* icon */ ICON_MOD_ARMATURE,
|
|
|
|
/* copyData */ copyData,
|
|
|
|
/* deformVerts */ deformVerts,
|
|
/* deformMatrices */ deformMatrices,
|
|
/* deformVertsEM */ deformVertsEM,
|
|
/* deformMatricesEM */ deformMatricesEM,
|
|
/* modifyMesh */ NULL,
|
|
/* modifyHair */ NULL,
|
|
/* modifyGeometrySet */ NULL,
|
|
/* modifyVolume */ NULL,
|
|
|
|
/* initData */ initData,
|
|
/* requiredDataMask */ requiredDataMask,
|
|
/* freeData */ NULL,
|
|
/* isDisabled */ isDisabled,
|
|
/* updateDepsgraph */ updateDepsgraph,
|
|
/* dependsOnTime */ NULL,
|
|
/* dependsOnNormals */ NULL,
|
|
/* foreachIDLink */ foreachIDLink,
|
|
/* foreachTexLink */ NULL,
|
|
/* freeRuntimeData */ NULL,
|
|
/* panelRegister */ panelRegister,
|
|
/* blendWrite */ NULL,
|
|
/* blendRead */ blendRead,
|
|
};
|