#include <string.h>
#include "MEM_guardedalloc.h"
#include "BLI_utildefines.h"
#include "BLI_bitmap.h"
#include "BLI_math.h"
#include "BLT_translation.h"
#include "DNA_defaults.h"
#include "DNA_mesh_types.h"
#include "DNA_meshdata_types.h"
#include "DNA_object_types.h"
#include "DNA_screen_types.h"
#include "BKE_context.h"
#include "BKE_deform.h"
#include "BKE_lib_id.h"
#include "BKE_lib_query.h"
#include "BKE_mesh.h"
#include "BKE_screen.h"
#include "UI_interface.h"
#include "UI_resources.h"
#include "RNA_access.h"
#include "DEG_depsgraph_query.h"
#include "MOD_ui_common.h"
#include "MOD_util.h"

Functions
static void	generate_vert_coordinates (Mesh mesh, Object ob, Object ob_center, const float offset[3], const int num_verts, float(r_cos)[3], float r_size[3])

static void	mix_normals (const float mix_factor, MDeformVert dvert, const int defgrp_index, const bool use_invert_vgroup, const float mix_limit, const short mix_mode, const int num_verts, MLoop mloop, float(nos_old)[3], float(nos_new)[3], const int num_loops)

static bool	polygons_check_flip (MLoop mloop, float(nos)[3], CustomData ldata, MPoly mpoly, float(*polynors)[3], const int num_polys)

static void	normalEditModifier_do_radial (NormalEditModifierData enmd, const ModifierEvalContext UNUSED(ctx), Object ob, Mesh mesh, short(clnors)[2], float(loopnors)[3], float(polynors)[3], const short mix_mode, const float mix_factor, const float mix_limit, MDeformVert dvert, const int defgrp_index, const bool use_invert_vgroup, MVert mvert, const int num_verts, MEdge medge, const int num_edges, MLoop mloop, const int num_loops, MPoly mpoly, const int num_polys)

static void	normalEditModifier_do_directional (NormalEditModifierData enmd, const ModifierEvalContext UNUSED(ctx), Object ob, Mesh mesh, short(clnors)[2], float(loopnors)[3], float(polynors)[3], const short mix_mode, const float mix_factor, const float mix_limit, MDeformVert dvert, const int defgrp_index, const bool use_invert_vgroup, MVert mvert, const int num_verts, MEdge medge, const int num_edges, MLoop mloop, const int num_loops, MPoly mpoly, const int num_polys)

static bool	is_valid_target (NormalEditModifierData *enmd)

static bool	is_valid_target_with_error (const Object ob, NormalEditModifierData enmd)

static Mesh *	normalEditModifier_do (NormalEditModifierData enmd, const ModifierEvalContext ctx, Object ob, Mesh mesh)

static void	initData (ModifierData *md)

static void	requiredDataMask (Object UNUSED(ob), ModifierData md, CustomData_MeshMasks *r_cddata_masks)

static bool	dependsOnNormals (ModifierData *UNUSED(md))

static void	foreachIDLink (ModifierData md, Object ob, IDWalkFunc walk, void *userData)

static bool	isDisabled (const struct Scene UNUSED(scene), ModifierData md, bool UNUSED(useRenderParams))

static void	updateDepsgraph (ModifierData md, const ModifierUpdateDepsgraphContext ctx)

static Mesh *	modifyMesh (ModifierData md, const ModifierEvalContext ctx, Mesh *mesh)

static void	panel_draw (const bContext UNUSED(C), Panel panel)

static void	mix_mode_panel_draw (const bContext UNUSED(C), Panel panel)

static void	offset_panel_draw (const bContext UNUSED(C), Panel panel)

static void	panelRegister (ARegionType *region_type)

Variables
ModifierTypeInfo	modifierType_NormalEdit

Function Documentation

◆ dependsOnNormals()

static bool dependsOnNormals ( ModifierData * UNUSEDmd )

static

Definition at line 678 of file MOD_normal_edit.c.

◆ foreachIDLink()

static void foreachIDLink	(	ModifierData *	md,
		Object *	ob,
		IDWalkFunc	walk,
		void *	userData
	)

static

Definition at line 683 of file MOD_normal_edit.c.

References IDWALK_CB_NOP, and NormalEditModifierData::target.

◆ generate_vert_coordinates()

static void generate_vert_coordinates	(	Mesh *	mesh,
		Object *	ob,
		Object *	ob_center,
		const float	offset[3],
		const int	num_verts,
		float(*)	r_cos[3],
		float	r_size[3]
	)

static

Definition at line 55 of file MOD_normal_edit.c.

References abs_v3_v3(), add_v3_v3(), CLAMP_MIN, copy_v3_v3(), KDL::diff(), INIT_MINMAX, invert_m4_m4(), is_zero_v3(), mesh, minmax_v3v3_v3(), mul_v3_m4v3(), Mesh::mvert, negate_v3(), negate_v3_v3(), NULL, Object::obmat, Object::scale, sub_v3_v3v3(), and Mesh::totvert.

Referenced by normalEditModifier_do_directional(), and normalEditModifier_do_radial().

◆ initData()

static void initData ( ModifierData * md )

static

Definition at line 655 of file MOD_normal_edit.c.

References BLI_assert, DNA_struct_default_get, MEMCMP_STRUCT_AFTER_IS_ZERO, and MEMCPY_STRUCT_AFTER.

◆ is_valid_target()

static bool is_valid_target ( NormalEditModifierData * enmd )

static

Definition at line 471 of file MOD_normal_edit.c.

References MOD_NORMALEDIT_MODE_DIRECTIONAL, MOD_NORMALEDIT_MODE_RADIAL, NormalEditModifierData::mode, and NormalEditModifierData::target.

Referenced by is_valid_target_with_error(), and isDisabled().

◆ is_valid_target_with_error()

static bool is_valid_target_with_error	(	const Object *	ob,
		NormalEditModifierData *	enmd
	)

static

Definition at line 482 of file MOD_normal_edit.c.

References BKE_modifier_set_error(), and is_valid_target().

Referenced by normalEditModifier_do().

◆ isDisabled()

static bool isDisabled	(	const struct Scene *	UNUSEDscene,
		ModifierData *	md,
		bool	UNUSEDuseRenderParams
	)

static

Definition at line 690 of file MOD_normal_edit.c.

References is_valid_target().

◆ mix_mode_panel_draw()

static void mix_mode_panel_draw	(	const bContext *	UNUSEDC,
		Panel *	panel
	)

static

Definition at line 737 of file MOD_normal_edit.c.

References Panel::layout, modifier_panel_get_property_pointers(), modifier_vgroup_ui(), NULL, ptr, RNA_boolean_get(), uiItemR(), uiLayoutRow(), and uiLayoutSetPropSep().

Referenced by panelRegister().

◆ mix_normals()

static void mix_normals	(	const float	mix_factor,
		MDeformVert *	dvert,
		const int	defgrp_index,
		const bool	use_invert_vgroup,
		const float	mix_limit,
		const short	mix_mode,
		const int	num_verts,
		MLoop *	mloop,
		float(*)	nos_old[3],
		float(*)	nos_new[3],
		const int	num_loops
	)

static

Definition at line 132 of file MOD_normal_edit.c.

References add_v3_v3(), angle_v3v3(), BKE_defvert_extract_vgroup_to_loopweights(), float(), interp_v3_v3v3_slerp_safe(), M_PI, MEM_malloc_arrayN, MEM_SAFE_FREE, min_ff(), MOD_NORMALEDIT_MIX_ADD, MOD_NORMALEDIT_MIX_COPY, MOD_NORMALEDIT_MIX_MUL, MOD_NORMALEDIT_MIX_SUB, mul_v3_v3(), normalize_v3(), NULL, and sub_v3_v3().

Referenced by normalEditModifier_do_directional(), and normalEditModifier_do_radial().

◆ modifyMesh()

static Mesh* modifyMesh	(	ModifierData *	md,
		const ModifierEvalContext *	ctx,
		Mesh *	mesh
	)

static

Definition at line 708 of file MOD_normal_edit.c.

References mesh, normalEditModifier_do(), and ModifierEvalContext::object.

◆ normalEditModifier_do()

static Mesh* normalEditModifier_do	(	NormalEditModifierData *	enmd,
		const ModifierEvalContext *	ctx,
		Object *	ob,
		Mesh *	mesh
	)

static

Definition at line 491 of file MOD_normal_edit.c.

References BKE_id_copy_ex(), BKE_mesh_calc_normals_poly(), BKE_mesh_normals_loop_split(), BKE_modifier_set_error(), CD_CALLOC, CD_CUSTOMLOOPNORMAL, CD_FLAG_TEMPORARY, CD_MASK_NORMAL, CD_NORMAL, CustomData_add_layer(), CustomData_duplicate_referenced_layer(), CustomData_free_layers(), CustomData_get_layer(), CustomData_set_layer_flag(), Object::data, NormalEditModifierData::defgrp_name, Mesh::flag, NormalEditModifierData::flag, float(), Mesh::id, is_valid_target_with_error(), LIB_ID_COPY_LOCALIZE, M_PI, ME_AUTOSMOOTH, Mesh::medge, MEM_malloc_arrayN, MEM_SAFE_FREE, mesh, NormalEditModifierData::mix_factor, NormalEditModifierData::mix_limit, NormalEditModifierData::mix_mode, MOD_get_vgroup(), MOD_NORMALEDIT_INVERT_VGROUP, MOD_NORMALEDIT_MIX_COPY, MOD_NORMALEDIT_MODE_DIRECTIONAL, MOD_NORMALEDIT_MODE_RADIAL, NormalEditModifierData::mode, normalEditModifier_do_directional(), normalEditModifier_do_radial(), NULL, ModifierEvalContext::object, result, and Mesh::totloop.

Referenced by modifyMesh().

◆ normalEditModifier_do_directional()

static void normalEditModifier_do_directional	(	NormalEditModifierData *	enmd,
		const ModifierEvalContext *	UNUSEDctx,
		Object *	ob,
		Mesh *	mesh,
		short(*)	clnors[2],
		float(*)	loopnors[3],
		float(*)	polynors[3],
		const short	mix_mode,
		const float	mix_factor,
		const float	mix_limit,
		MDeformVert *	dvert,
		const int	defgrp_index,
		const bool	use_invert_vgroup,
		MVert *	mvert,
		const int	num_verts,
		MEdge *	medge,
		const int	num_edges,
		MLoop *	mloop,
		const int	num_loops,
		MPoly *	mpoly,
		const int	num_polys
	)

static

◆ normalEditModifier_do_radial()

static void normalEditModifier_do_radial	(	NormalEditModifierData *	enmd,
		const ModifierEvalContext *	UNUSEDctx,
		Object *	ob,
		Mesh *	mesh,
		short(*)	clnors[2],
		float(*)	loopnors[3],
		float(*)	polynors[3],
		const short	mix_mode,
		const float	mix_factor,
		const float	mix_limit,
		MDeformVert *	dvert,
		const int	defgrp_index,
		const bool	use_invert_vgroup,
		MVert *	mvert,
		const int	num_verts,
		MEdge *	medge,
		const int	num_edges,
		MLoop *	mloop,
		const int	num_loops,
		MPoly *	mpoly,
		const int	num_polys
	)

static

size gives us our spheroid coefficients (A, B, C). Then, we want to find out for each vert its (a, b, c) triple (proportional to (A, B, C) one).

Ellipsoid basic equation: (x^2/a^2) + (y^2/b^2) + (z^2/c^2) = 1. Since we want to find (a, b, c) matching this equation and proportional to (A, B, C), we can do:

    m = B / A
    n = C / A

hence:

    (x^2/a^2) + (y^2/b^2) + (z^2/c^2) = 1
 -> b^2*c^2*x^2 + a^2*c^2*y^2 + a^2*b^2*z^2 = a^2*b^2*c^2
    b = ma
    c = na
 -> m^2*a^2*n^2*a^2*x^2 + a^2*n^2*a^2*y^2 + a^2*m^2*a^2*z^2 = a^2*m^2*a^2*n^2*a^2
 -> m^2*n^2*a^4*x^2 + n^2*a^4*y^2 + m^2*a^4*z^2 = m^2*n^2*a^6
 -> a^2 = (m^2*n^2*x^2 + n^2y^2 + m^2z^2) / (m^2*n^2) = x^2 + (y^2 / m^2) + (z^2 / n^2)
 -> b^2 = (m^2*n^2*x^2 + n^2y^2 + m^2z^2) / (n^2)     = (m^2 * x^2) + y^2 + (m^2 * z^2 / n^2)
 -> c^2 = (m^2*n^2*x^2 + n^2y^2 + m^2z^2) / (m^2)     = (n^2 * x^2) + (n^2 * y^2 / m^2) + z^2

All we have to do now is compute normal of the spheroid at that point: