#include <cfloat>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <ctime>
#include <memory.h>
#include "DNA_gpencil_modifier_types.h"
#include "DNA_mesh_types.h"
#include "DNA_modifier_types.h"
#include "DNA_object_types.h"
#include "BLI_math_geom.h"
#include "BLI_math_matrix.h"
#include "BLI_math_solvers.h"
#include "BLI_math_vector.h"
#include "BLI_task.h"
#include "BLI_utildefines.h"
#include "BKE_attribute.hh"
#include "BKE_mesh_legacy_derived_mesh.hh"
#include "BKE_modifier.hh"
#include "BKE_shrinkwrap.hh"
#include "BKE_deform.hh"
#include "BKE_editmesh.hh"
#include "BKE_mesh.hh"
#include "BKE_mesh_wrapper.hh"
#include "BKE_subsurf.hh"
#include "DEG_depsgraph_query.hh"
#include "MEM_guardedalloc.h"
#include "BLI_strict_flags.h"

Classes
struct	ShrinkwrapCalcData

struct	ShrinkwrapCalcCBData

struct	TargetProjectTriData

Namespaces
namespace	blender

namespace	blender::bke

namespace	blender::bke::shrinkwrap

Macros
#define	TIMEIT_BENCH(expr, id) (expr)

#define	OUT_OF_MEMORY() ((void)printf("Shrinkwrap: Out of memory\n"))

Functions
bool	BKE_shrinkwrap_needs_normals (int shrinkType, int shrinkMode)

bool	BKE_shrinkwrap_init_tree (ShrinkwrapTreeData data, Mesh mesh, int shrinkType, int shrinkMode, bool force_normals)

void	BKE_shrinkwrap_free_tree (ShrinkwrapTreeData *data)

static void	blender::bke::shrinkwrap::merge_vert_dir (ShrinkwrapBoundaryVertData *vdata, MutableSpan< int8_t > status, int index, const float edge_dir[3], signed char side)

static ShrinkwrapBoundaryData	blender::bke::shrinkwrap::shrinkwrap_build_boundary_data (const Mesh &mesh)

const ShrinkwrapBoundaryData &	blender::bke::shrinkwrap::boundary_cache_ensure (const Mesh &mesh)

static void	shrinkwrap_calc_nearest_vertex_cb_ex (void __restrict userdata, const int i, const TaskParallelTLS __restrict tls)

static void	shrinkwrap_calc_nearest_vertex (ShrinkwrapCalcData *calc)

bool	BKE_shrinkwrap_project_normal (char options, const float vert[3], const float dir[3], const float ray_radius, const SpaceTransform transf, ShrinkwrapTreeData tree, BVHTreeRayHit *hit)

static void	shrinkwrap_calc_normal_projection_cb_ex (void __restrict userdata, const int i, const TaskParallelTLS __restrict tls)

static void	shrinkwrap_calc_normal_projection (ShrinkwrapCalcData *calc)

static void	target_project_tri_deviation (void *userdata, const float x[3], float r_delta[3])

static void	target_project_tri_jacobian (void *userdata, const float x[3], float r_jacobian[3][3])

static void	target_project_tri_clamp (float x[3])

static bool	target_project_tri_correct (void *, const float x[3], float step[3], float x_next[3])

static bool	target_project_solve_point_tri (const float *vtri_co[3], const float vtri_no[3][3], const float point_co[3], const float hit_co[3], float hit_dist_sq, float r_hit_co[3], float r_hit_no[3])

static bool	update_hit (BVHTreeNearest *nearest, int index, const float co[3], const float hit_co[3], const float hit_no[3])

static void	target_project_edge (const ShrinkwrapTreeData tree, int index, const float co[3], BVHTreeNearest nearest, int eidx)

static void	mesh_corner_tris_target_project (void userdata, int index, const float co[3], BVHTreeNearest nearest)

void	BKE_shrinkwrap_find_nearest_surface (ShrinkwrapTreeData tree, BVHTreeNearest nearest, float co[3], int type)

static void	shrinkwrap_calc_nearest_surface_point_cb_ex (void __restrict userdata, const int i, const TaskParallelTLS __restrict tls)

void	BKE_shrinkwrap_compute_smooth_normal (const ShrinkwrapTreeData tree, const SpaceTransform transform, int corner_tri_idx, const float hit_co[3], const float hit_no[3], float r_no[3])

static void	shrinkwrap_snap_with_side (float r_point_co[3], const float point_co[3], const float hit_co[3], const float hit_no[3], float goal_dist, float forcesign, bool forcesnap)

void	BKE_shrinkwrap_snap_point_to_surface (const ShrinkwrapTreeData tree, const SpaceTransform transform, int mode, int hit_idx, const float hit_co[3], const float hit_no[3], float goal_dist, const float point_co[3], float r_point_co[3])

static void	shrinkwrap_calc_nearest_surface_point (ShrinkwrapCalcData *calc)

void	shrinkwrapModifier_deform (ShrinkwrapModifierData smd, const ModifierEvalContext ctx, Scene scene, Object ob, Mesh mesh, const MDeformVert dvert, const int defgrp_index, float(*vertexCos)[3], int numVerts)

void	shrinkwrapGpencilModifier_deform (ShrinkwrapGpencilModifierData mmd, Object ob, MDeformVert dvert, const int defgrp_index, float(vertexCos)[3], int numVerts)

void	shrinkwrapParams_deform (const ShrinkwrapParams &params, Object &object, ShrinkwrapTreeData &tree, const blender::Span< MDeformVert > dvert, const int defgrp_index, const blender::MutableSpan< blender::float3 > positions)

void	BKE_shrinkwrap_mesh_nearest_surface_deform (Depsgraph depsgraph, Scene scene, Object ob_source, Object ob_target)

void	BKE_shrinkwrap_remesh_target_project (Mesh src_me, Mesh target_me, Object *ob_target)

Macro Definition Documentation

◆ OUT_OF_MEMORY

#define OUT_OF_MEMORY ( ) ((void)printf("Shrinkwrap: Out of memory\n"))

Definition at line 53 of file shrinkwrap.cc.

◆ TIMEIT_BENCH

#define TIMEIT_BENCH	(		expr,
			id ) (expr)

Definition at line 49 of file shrinkwrap.cc.

Referenced by BKE_shrinkwrap_remesh_target_project(), shrinkwrapGpencilModifier_deform(), shrinkwrapModifier_deform(), and shrinkwrapParams_deform().

Function Documentation

◆ BKE_shrinkwrap_compute_smooth_normal()

void BKE_shrinkwrap_compute_smooth_normal	(	const ShrinkwrapTreeData *	tree,
		const SpaceTransform *	transform,
		int	tri_idx,
		const float	hit_co[3],
		const float	hit_no[3],
		float	r_no[3] )

Compute a smooth normal of the target (if applicable) at the hit location.

Parameters

tree	information about the mesh.
transform	transform from the hit coordinate space to the object space; may be null.
r_no	output in hit coordinate space; may be shared with inputs.

Definition at line 1153 of file shrinkwrap.cc.

References BLI_space_transform_apply(), BLI_space_transform_invert_normal(), copy_v3_v3(), BVHTreeFromMesh::corner_tris, BVHTreeFromMesh::corner_verts, interp_v3_v3v3v3(), interp_weights_tri_v3(), normalize_v3(), tree, BVHTreeFromMesh::vert_positions, and w().

Referenced by BKE_shrinkwrap_snap_point_to_surface(), and shrinkwrap_get_tarmat().

◆ BKE_shrinkwrap_find_nearest_surface()

void BKE_shrinkwrap_find_nearest_surface	(	ShrinkwrapTreeData *	tree,
		BVHTreeNearest *	nearest,
		float	co[3],
		int	type )

Maps the point to the nearest surface, either by simple nearest, or by target normal projection.

Definition at line 1049 of file shrinkwrap.cc.

References BLI_bvhtree_find_nearest(), BLI_bvhtree_find_nearest_ex(), BVH_NEAREST_OPTIMAL_ORDER, mesh_corner_tris_target_project(), MOD_SHRINKWRAP_TARGET_PROJECT, BVHTreeFromMesh::nearest_callback, printf, and tree.

Referenced by shrinkwrap_calc_nearest_surface_point_cb_ex(), and shrinkwrap_get_tarmat().

◆ BKE_shrinkwrap_free_tree()

void BKE_shrinkwrap_free_tree ( ShrinkwrapTreeData * data )

Frees the tree data if necessary.

Definition at line 152 of file shrinkwrap.cc.

References free_bvhtree_from_mesh().

Referenced by BKE_shrinkwrap_remesh_target_project(), blender::ensure_shrinkwrap_cache_data(), blender::free_data(), gpencil_modifier_free_data(), shrinkwrap_calc_normal_projection(), shrinkwrap_get_tarmat(), and shrinkwrapModifier_deform().

◆ BKE_shrinkwrap_init_tree()

bool BKE_shrinkwrap_init_tree	(	ShrinkwrapTreeData *	data,
		Mesh *	mesh,
		int	shrinkType,
		int	shrinkMode,
		bool	force_normals )

Initializes the mesh data structure from the given mesh and settings.

Definition at line 96 of file shrinkwrap.cc.

References BKE_bvhtree_from_mesh_get(), BKE_mesh_wrapper_ensure_mdata(), BKE_shrinkwrap_needs_normals(), blender::bke::shrinkwrap::boundary_cache_ensure(), BVHTREE_FROM_CORNER_TRIS, BVHTREE_FROM_VERTS, blender::bke::Corner, blender::bke::AttributeAccessor::lookup(), mesh, MOD_SHRINKWRAP_NEAREST_VERTEX, and MOD_SHRINKWRAP_TARGET_PROJECT.

Referenced by BKE_shrinkwrap_remesh_target_project(), blender::ensure_shrinkwrap_cache_data(), shrinkwrap_calc_normal_projection(), shrinkwrap_get_tarmat(), and shrinkwrapModifier_deform().

◆ BKE_shrinkwrap_mesh_nearest_surface_deform()

void BKE_shrinkwrap_mesh_nearest_surface_deform	(	Depsgraph *	depsgraph,
		Scene *	scene,
		Object *	ob_source,
		Object *	ob_target )

Used in editmesh_mask_extract.cc to shrink-wrap the extracted mesh to the sculpt.

Definition at line 1566 of file shrinkwrap.cc.

References Object::data, depsgraph, ShrinkwrapModifierData::keepDist, MOD_SHRINKWRAP_NEAREST_SURFACE, MOD_SHRINKWRAP_ON_SURFACE, ShrinkwrapModifierData::shrinkMode, ShrinkwrapModifierData::shrinkType, shrinkwrapModifier_deform(), ShrinkwrapModifierData::target, and Mesh::verts_num.

Referenced by geometry_extract_apply().

◆ BKE_shrinkwrap_needs_normals()

bool BKE_shrinkwrap_needs_normals	(	int	shrinkType,
		int	shrinkMode )

Checks if the modifier needs target normals with these settings.

Definition at line 89 of file shrinkwrap.cc.

References MOD_SHRINKWRAP_ABOVE_SURFACE, MOD_SHRINKWRAP_NEAREST_VERTEX, and MOD_SHRINKWRAP_TARGET_PROJECT.

Referenced by BKE_shrinkwrap_init_tree(), blender::deg::DepsgraphRelationBuilder::build_constraints(), blender::update_depsgraph(), and update_depsgraph().

◆ BKE_shrinkwrap_project_normal()

bool BKE_shrinkwrap_project_normal	(	char	options,
		const float	vert[3],
		const float	dir[3],
		float	ray_radius,
		const SpaceTransform *	transf,
		ShrinkwrapTreeData *	tree,
		BVHTreeRayHit *	hit )

This function ray-cast a single vertex and updates the hit if the "hit" is considered valid.

Parameters

options	Opts control whether an hit is valid or not. Supported options are: MOD_SHRINKWRAP_CULL_TARGET_FRONTFACE (front faces hits are ignored) MOD_SHRINKWRAP_CULL_TARGET_BACKFACE (back faces hits are ignored)
transf	Take into consideration the space_transform, that is: if `transf` was configured with `SPACE_TRANSFORM_SETUP( &transf, ob1, ob2)` then the input (vert, dir, BVHTreeRayHit) must be defined in ob1 coordinates space and the BVHTree must be built in ob2 coordinate space. Thus it provides an easy way to cast the same ray across several trees (where each tree was built on its own coords space).

Returns: true if "hit" was updated.

Definition at line 390 of file shrinkwrap.cc.

References BLI_assert, BLI_bvhtree_ray_cast(), BLI_space_transform_apply(), BLI_space_transform_apply_normal(), BLI_space_transform_invert(), BLI_space_transform_invert_normal(), BVHTreeRayHit::co, copy_v3_v3(), BVHTreeRayHit::dist, dot_v3v3(), BVHTreeRayHit::index, len_v3v3(), mat4_to_scale(), MOD_SHRINKWRAP_CULL_TARGET_BACKFACE, MOD_SHRINKWRAP_CULL_TARGET_FRONTFACE, MOD_SHRINKWRAP_CULL_TARGET_MASK, BVHTreeRayHit::no, options, and tree.

Referenced by shrinkwrap_calc_normal_projection_cb_ex(), and shrinkwrap_get_tarmat().

◆ BKE_shrinkwrap_remesh_target_project()

void BKE_shrinkwrap_remesh_target_project	(	Mesh *	src_me,
		Mesh *	target_me,
		Object *	ob_target )

Used in object_remesh.cc to preserve the details and volume in the voxel remesher.

Definition at line 1594 of file shrinkwrap.cc.

References BKE_shrinkwrap_free_tree(), BKE_shrinkwrap_init_tree(), BLI_SPACE_TRANSFORM_SETUP, ShrinkwrapCalcData::keepDist, ShrinkwrapModifierData::keepDist, ShrinkwrapCalcData::local2target, MOD_SHRINKWRAP_ON_SURFACE, MOD_SHRINKWRAP_PROJECT, MOD_SHRINKWRAP_PROJECT_ALLOW_NEG_DIR, MOD_SHRINKWRAP_PROJECT_ALLOW_POS_DIR, NULL_ShrinkwrapCalcData, ShrinkwrapCalcData::numVerts, ShrinkwrapModifierData::projLimit, Mesh::remesh_voxel_size, ShrinkwrapModifierData::shrinkMode, ShrinkwrapModifierData::shrinkOpts, ShrinkwrapModifierData::shrinkType, shrinkwrap_calc_normal_projection(), ShrinkwrapCalcData::smd, ShrinkwrapCalcData::target, ShrinkwrapModifierData::target, TIMEIT_BENCH, ShrinkwrapCalcData::tree, tree, ShrinkwrapCalcData::vert_normals, ShrinkwrapCalcData::vert_positions, ShrinkwrapCalcData::vertexCos, Mesh::verts_num, and ShrinkwrapCalcData::vgroup.

Referenced by blender::ed::object::voxel_remesh_exec().

◆ BKE_shrinkwrap_snap_point_to_surface()

void BKE_shrinkwrap_snap_point_to_surface	(	const ShrinkwrapTreeData *	tree,
		const SpaceTransform *	transform,
		int	mode,
		int	hit_idx,
		const float	hit_co[3],
		const float	hit_no[3],
		float	goal_dist,
		const float	point_co[3],
		float	r_point_co[3] )

Apply the shrink to surface modes to the given original coordinates and nearest point.

Parameters

tree	mesh data for smooth normals.
transform	transform from the hit coordinate space to the object space; may be null.
r_point_co	may be the same memory location as `point_co`, `hit_co`, or `hit_no`.

Definition at line 1275 of file shrinkwrap.cc.

References BKE_shrinkwrap_compute_smooth_normal(), copy_v3_v3(), madd_v3_v3v3fl(), MOD_SHRINKWRAP_ABOVE_SURFACE, MOD_SHRINKWRAP_INSIDE, MOD_SHRINKWRAP_ON_SURFACE, MOD_SHRINKWRAP_OUTSIDE, MOD_SHRINKWRAP_OUTSIDE_SURFACE, printf, shrinkwrap_snap_with_side(), and tree.

Referenced by shrinkwrap_calc_nearest_surface_point_cb_ex(), shrinkwrap_calc_normal_projection_cb_ex(), and shrinkwrap_get_tarmat().

◆ mesh_corner_tris_target_project()

static void mesh_corner_tris_target_project	(	void *	userdata,
		int	index,
		const float	co[3],
		BVHTreeNearest *	nearest )

static

Definition at line 985 of file shrinkwrap.cc.

References closest_on_tri_to_point_v3(), copy_v3_v3(), len_squared_v3v3(), printf, target_project_edge(), target_project_solve_point_tri(), tree, UNPACK3, and update_hit().

Referenced by BKE_shrinkwrap_find_nearest_surface().

◆ shrinkwrap_calc_nearest_surface_point()

static void shrinkwrap_calc_nearest_surface_point ( ShrinkwrapCalcData * calc )

static

Definition at line 1332 of file shrinkwrap.cc.

References BLI_parallel_range_settings_defaults(), BLI_task_parallel_range(), ShrinkwrapCalcCBData::calc, FLT_MAX, BVHTreeNearest::index, NULL_BVHTreeNearest, ShrinkwrapCalcData::numVerts, shrinkwrap_calc_nearest_surface_point_cb_ex(), and ShrinkwrapCalcData::tree.

Referenced by shrinkwrapGpencilModifier_deform(), shrinkwrapModifier_deform(), and shrinkwrapParams_deform().

◆ shrinkwrap_calc_nearest_surface_point_cb_ex()

static void shrinkwrap_calc_nearest_surface_point_cb_ex	(	void *__restrict	userdata,
		const int	i,
		const TaskParallelTLS *__restrict	tls )

static

Shrink-wrap moving vertices to the nearest surface point on the target.

It builds a BVHTree from the target mesh and then performs a NN matches for each vertex

Definition at line 1084 of file shrinkwrap.cc.

Referenced by shrinkwrap_calc_nearest_surface_point().

◆ shrinkwrap_calc_nearest_vertex()

static void shrinkwrap_calc_nearest_vertex ( ShrinkwrapCalcData * calc )

static

Definition at line 370 of file shrinkwrap.cc.

References BLI_parallel_range_settings_defaults(), BLI_task_parallel_range(), ShrinkwrapCalcCBData::calc, FLT_MAX, BVHTreeNearest::index, NULL_BVHTreeNearest, ShrinkwrapCalcData::numVerts, shrinkwrap_calc_nearest_vertex_cb_ex(), and ShrinkwrapCalcData::tree.

Referenced by shrinkwrapGpencilModifier_deform(), shrinkwrapModifier_deform(), and shrinkwrapParams_deform().

◆ shrinkwrap_calc_nearest_vertex_cb_ex()

static void shrinkwrap_calc_nearest_vertex_cb_ex	(	void *__restrict	userdata,
		const int	i,
		const TaskParallelTLS *__restrict	tls )

static

Shrink-wrap to the nearest vertex

it builds a BVH-tree of vertices we can attach to and then for each vertex performs a nearest vertex search on the tree.

Definition at line 308 of file shrinkwrap.cc.

References BKE_defvert_array_find_weight_safe(), BLI_bvhtree_find_nearest(), BLI_space_transform_apply(), BLI_space_transform_invert(), copy_v3_v3(), ShrinkwrapCalcData::dvert, FLT_MAX, interp_v3_v3v3(), ShrinkwrapCalcData::invert_vgroup, ShrinkwrapCalcData::keepDist, len_squared_v3v3(), ShrinkwrapCalcData::local2target, BVHTreeFromMesh::nearest_callback, sqrtf, BVHTreeFromMesh::tree, ShrinkwrapCalcData::vert_positions, ShrinkwrapCalcData::vertexCos, and ShrinkwrapCalcData::vgroup.

Referenced by shrinkwrap_calc_nearest_vertex().

◆ shrinkwrap_calc_normal_projection()

static void shrinkwrap_calc_normal_projection ( ShrinkwrapCalcData * calc )

static

Note: 'hit.dist' is kept in the targets space, this is only used for finding the best hit, to get the real dist, measure the len_v3v3() from the input coord to hit.co

Definition at line 592 of file shrinkwrap.cc.

References ShrinkwrapCalcData::aux_target, BKE_modifier_get_evaluated_mesh_from_evaluated_object(), BKE_shrinkwrap_free_tree(), BKE_shrinkwrap_init_tree(), BLI_parallel_range_settings_defaults(), BLI_SPACE_TRANSFORM_SETUP, BLI_task_parallel_range(), ShrinkwrapCalcCBData::calc, len_squared_v3(), MOD_SHRINKWRAP_PROJECT_ALLOW_NEG_DIR, MOD_SHRINKWRAP_PROJECT_ALLOW_POS_DIR, MOD_SHRINKWRAP_PROJECT_OVER_NORMAL, MOD_SHRINKWRAP_PROJECT_OVER_X_AXIS, MOD_SHRINKWRAP_PROJECT_OVER_Y_AXIS, MOD_SHRINKWRAP_PROJECT_OVER_Z_AXIS, normalize_v3(), ShrinkwrapCalcData::numVerts, ShrinkwrapCalcData::ob, ShrinkwrapModifierData::projAxis, ShrinkwrapModifierData::shrinkMode, ShrinkwrapModifierData::shrinkOpts, ShrinkwrapModifierData::shrinkType, shrinkwrap_calc_normal_projection_cb_ex(), ShrinkwrapCalcData::smd, ShrinkwrapCalcData::tree, and ShrinkwrapCalcData::vert_positions.

Referenced by BKE_shrinkwrap_remesh_target_project(), shrinkwrapGpencilModifier_deform(), shrinkwrapModifier_deform(), and shrinkwrapParams_deform().

◆ shrinkwrap_calc_normal_projection_cb_ex()

static void shrinkwrap_calc_normal_projection_cb_ex	(	void *__restrict	userdata,
		const int	i,
		const TaskParallelTLS *__restrict	tls )

static

Definition at line 466 of file shrinkwrap.cc.

Referenced by shrinkwrap_calc_normal_projection().

◆ shrinkwrap_snap_with_side()

static void shrinkwrap_snap_with_side	(	float	r_point_co[3],
		const float	point_co[3],
		const float	hit_co[3],
		const float	hit_no[3],
		float	goal_dist,
		float	forcesign,
		bool	forcesnap )

static

Definition at line 1222 of file shrinkwrap.cc.

References copy_v3_v3(), dot_v3v3(), fabsf, interp_v3_v3v3(), len_manhattan_v3(), len_v3(), madd_v3_v3v3fl(), mul_v3_fl(), printf, signf(), and sub_v3_v3v3().

Referenced by BKE_shrinkwrap_snap_point_to_surface().

◆ shrinkwrapGpencilModifier_deform()

void shrinkwrapGpencilModifier_deform	(	ShrinkwrapGpencilModifierData *	mmd,
		Object *	ob,
		MDeformVert *	dvert,
		const int	defgrp_index,
		float(*)	vertexCos[3],
		int	numVerts )

Definition at line 1467 of file shrinkwrap.cc.

◆ shrinkwrapModifier_deform()

void shrinkwrapModifier_deform	(	ShrinkwrapModifierData *	smd,
		const ModifierEvalContext *	ctx,
		Scene *	scene,
		Object *	ob,
		Mesh *	mesh,
		const MDeformVert *	dvert,
		int	defgrp_index,
		float(*)	vertexCos[3],
		int	numVerts )

Main shrink-wrap function (implementation of the shrink-wrap modifier).

Definition at line 1353 of file shrinkwrap.cc.

Referenced by BKE_shrinkwrap_mesh_nearest_surface_deform(), and deform_verts().

◆ shrinkwrapParams_deform()

void shrinkwrapParams_deform	(	const ShrinkwrapParams &	params,
		Object &	object,
		ShrinkwrapTreeData &	tree,
		const blender::Span< MDeformVert >	dvert,
		const int	defgrp_index,
		const blender::MutableSpan< blender::float3 >	positions )

Definition at line 1516 of file shrinkwrap.cc.

Referenced by blender::modify_drawing().

◆ target_project_edge()

static void target_project_edge	(	const ShrinkwrapTreeData *	tree,
		int	index,
		const float	co[3],
		BVHTreeNearest *	nearest,
		int	eidx )

static

Definition at line 909 of file shrinkwrap.cc.

References b, CLAMP, copy_v3_v3(), dot_v3v3(), float, interp_v3_v3v3(), negate_v3(), printf, sqrtf, sub_v3_v3v3(), tree, UNPACK3, and update_hit().

Referenced by mesh_corner_tris_target_project().

◆ target_project_solve_point_tri()

static bool target_project_solve_point_tri	(	const float *	vtri_co[3],
		const float	vtri_no[3][3],
		const float	point_co[3],
		const float	hit_co[3],
		float	hit_dist_sq,
		float	r_hit_co[3],
		float	r_hit_no[3] )