Cleanup: Remove unnecessary step in calling snap callback

This commit is contained in:
Germano Cavalcante 2020-05-31 18:09:50 -03:00
parent 14af27e63d
commit fcccee0c33
1 changed files with 202 additions and 264 deletions

View File

@ -359,7 +359,7 @@ static SnapObjectData *snap_object_data_editmesh_get(SnapObjectContext *sctx,
* \{ */
typedef void (*IterSnapObjsCallback)(SnapObjectContext *sctx,
bool is_obedit,
bool use_obedit,
bool use_backface_culling,
Object *ob,
float obmat[4][4],
@ -920,128 +920,6 @@ static bool raycastEditMesh(SnapObjectContext *sctx,
return retval;
}
/**
* \param use_obedit: Uses the coordinates of BMesh (if any) to do the snapping;
*
* \note Duplicate args here are documented at #snapObjectsRay
*/
static bool raycastObj(SnapObjectContext *sctx,
const float ray_start[3],
const float ray_dir[3],
Object *ob,
const float obmat[4][4],
const uint ob_index,
bool use_obedit,
bool use_occlusion_test,
bool use_backface_culling,
/* read/write args */
float *ray_depth,
/* return args */
float r_loc[3],
float r_no[3],
int *r_index,
Object **r_ob,
float r_obmat[4][4],
ListBase *r_hit_list)
{
bool retval = false;
if (use_occlusion_test) {
if (use_obedit && sctx->use_v3d && XRAY_FLAG_ENABLED(sctx->v3d_data.v3d)) {
/* Use of occlude geometry in editing mode disabled. */
return false;
}
if (ELEM(ob->dt, OB_BOUNDBOX, OB_WIRE)) {
/* Do not hit objects that are in wire or bounding box
* display mode. */
return false;
}
}
switch (ob->type) {
case OB_MESH: {
Mesh *me = ob->data;
bool use_hide = false;
if (BKE_object_is_in_editmode(ob)) {
if (use_obedit) {
/* Operators only update the editmesh looptris of the original mesh. */
BMEditMesh *em_orig = BKE_editmesh_from_object(DEG_get_original_object(ob));
retval = raycastEditMesh(sctx,
ray_start,
ray_dir,
ob,
em_orig,
obmat,
ob_index,
use_backface_culling,
ray_depth,
r_loc,
r_no,
r_index,
r_hit_list);
break;
}
else {
BMEditMesh *em = BKE_editmesh_from_object(ob);
if (em->mesh_eval_final) {
me = em->mesh_eval_final;
use_hide = true;
}
}
}
retval = raycastMesh(sctx,
ray_start,
ray_dir,
ob,
me,
obmat,
ob_index,
use_hide,
use_backface_culling,
ray_depth,
r_loc,
r_no,
r_index,
r_hit_list);
break;
}
case OB_CURVE:
case OB_SURF:
case OB_FONT: {
Mesh *mesh_eval = BKE_object_get_evaluated_mesh(ob);
if (mesh_eval) {
retval = raycastMesh(sctx,
ray_start,
ray_dir,
ob,
mesh_eval,
obmat,
ob_index,
false,
use_backface_culling,
ray_depth,
r_loc,
r_no,
r_index,
r_hit_list);
break;
}
}
}
if (retval) {
if (r_ob) {
*r_ob = ob;
}
if (r_obmat) {
copy_m4_m4(r_obmat, obmat);
}
return true;
}
return false;
}
struct RaycastObjUserData {
const float *ray_start;
const float *ray_dir;
@ -1059,7 +937,12 @@ struct RaycastObjUserData {
bool ret;
};
static void raycast_obj_cb(SnapObjectContext *sctx,
/**
* \param use_obedit: Uses the coordinates of BMesh (if any) to do the snapping;
*
* \note Duplicate args here are documented at #snapObjectsRay
*/
static void raycast_obj_fn(SnapObjectContext *sctx,
bool use_obedit,
bool use_backface_culling,
Object *ob,
@ -1067,23 +950,105 @@ static void raycast_obj_cb(SnapObjectContext *sctx,
void *data)
{
struct RaycastObjUserData *dt = data;
const uint ob_index = dt->ob_index++;
bool use_occlusion_test = dt->use_occlusion_test;
/* read/write args */
float *ray_depth = dt->ray_depth;
dt->ret |= raycastObj(sctx,
dt->ray_start,
dt->ray_dir,
ob,
obmat,
dt->ob_index++,
use_obedit,
dt->use_occlusion_test,
use_backface_culling,
dt->ray_depth,
dt->r_loc,
dt->r_no,
dt->r_index,
dt->r_ob,
dt->r_obmat,
dt->r_hit_list);
bool retval = false;
if (use_occlusion_test) {
if (use_obedit && sctx->use_v3d && XRAY_FLAG_ENABLED(sctx->v3d_data.v3d)) {
/* Use of occlude geometry in editing mode disabled. */
return;
}
if (ELEM(ob->dt, OB_BOUNDBOX, OB_WIRE)) {
/* Do not hit objects that are in wire or bounding box
* display mode. */
return;
}
}
switch (ob->type) {
case OB_MESH: {
Mesh *me = ob->data;
bool use_hide = false;
if (BKE_object_is_in_editmode(ob)) {
if (use_obedit) {
/* Operators only update the editmesh looptris of the original mesh. */
BMEditMesh *em_orig = BKE_editmesh_from_object(DEG_get_original_object(ob));
retval = raycastEditMesh(sctx,
dt->ray_start,
dt->ray_dir,
ob,
em_orig,
obmat,
ob_index,
use_backface_culling,
ray_depth,
dt->r_loc,
dt->r_no,
dt->r_index,
dt->r_hit_list);
break;
}
else {
BMEditMesh *em = BKE_editmesh_from_object(ob);
if (em->mesh_eval_final) {
me = em->mesh_eval_final;
use_hide = true;
}
}
}
retval = raycastMesh(sctx,
dt->ray_start,
dt->ray_dir,
ob,
me,
obmat,
ob_index,
use_hide,
use_backface_culling,
ray_depth,
dt->r_loc,
dt->r_no,
dt->r_index,
dt->r_hit_list);
break;
}
case OB_CURVE:
case OB_SURF:
case OB_FONT: {
Mesh *mesh_eval = BKE_object_get_evaluated_mesh(ob);
if (mesh_eval) {
retval = raycastMesh(sctx,
dt->ray_start,
dt->ray_dir,
ob,
mesh_eval,
obmat,
ob_index,
false,
use_backface_culling,
ray_depth,
dt->r_loc,
dt->r_no,
dt->r_index,
dt->r_hit_list);
break;
}
}
}
if (retval) {
if (dt->r_ob) {
*dt->r_ob = ob;
}
if (dt->r_obmat) {
copy_m4_m4(dt->r_obmat, obmat);
}
dt->ret = true;
}
}
/**
@ -1144,7 +1109,7 @@ static bool raycastObjects(SnapObjectContext *sctx,
.ret = false,
};
iter_snap_objects(sctx, depsgraph, params, raycast_obj_cb, &data);
iter_snap_objects(sctx, depsgraph, params, raycast_obj_fn, &data);
return data.ret;
}
@ -2644,110 +2609,6 @@ static short snapEditMesh(SnapObjectContext *sctx,
return 0;
}
/**
* \param use_obedit: Uses the coordinates of BMesh (if any) to do the snapping;
*
* \note Duplicate args here are documented at #snapObjectsRay
*/
static short snapObject(SnapObjectContext *sctx,
SnapData *snapdata,
Object *ob,
float obmat[4][4],
bool use_obedit,
bool use_backface_culling,
/* read/write args */
float *dist_px,
/* return args */
float r_loc[3],
float r_no[3],
int *r_index,
Object **r_ob,
float r_obmat[4][4])
{
short retval = 0;
switch (ob->type) {
case OB_MESH: {
Mesh *me = ob->data;
if (BKE_object_is_in_editmode(ob)) {
if (use_obedit) {
/* Operators only update the editmesh looptris of the original mesh. */
BMEditMesh *em_orig = BKE_editmesh_from_object(DEG_get_original_object(ob));
retval = snapEditMesh(sctx,
snapdata,
ob,
em_orig,
obmat,
use_backface_culling,
dist_px,
r_loc,
r_no,
r_index);
break;
}
else {
BMEditMesh *em = BKE_editmesh_from_object(ob);
if (em->mesh_eval_final) {
me = em->mesh_eval_final;
}
}
}
else if (ob->dt == OB_BOUNDBOX) {
/* Do not snap to objects that are in bounding box display mode */
return 0;
}
retval = snapMesh(
sctx, snapdata, ob, me, obmat, use_backface_culling, dist_px, r_loc, r_no, r_index);
break;
}
case OB_ARMATURE:
retval = snapArmature(snapdata, ob, obmat, use_obedit, dist_px, r_loc, r_no, r_index);
break;
case OB_CURVE:
retval = snapCurve(snapdata, ob, obmat, use_obedit, dist_px, r_loc, r_no, r_index);
break; /* Use ATTR_FALLTHROUGH if we want to snap to the generated mesh. */
case OB_SURF:
case OB_FONT: {
Mesh *mesh_eval = BKE_object_get_evaluated_mesh(ob);
if (mesh_eval) {
retval |= snapMesh(sctx,
snapdata,
ob,
mesh_eval,
obmat,
use_backface_culling,
dist_px,
r_loc,
r_no,
r_index);
}
break;
}
case OB_EMPTY:
retval = snapEmpty(snapdata, ob, obmat, dist_px, r_loc, r_no, r_index);
break;
case OB_GPENCIL:
retval = snapEmpty(snapdata, ob, obmat, dist_px, r_loc, r_no, r_index);
break;
case OB_CAMERA:
retval = snapCamera(sctx, snapdata, ob, obmat, dist_px, r_loc, r_no, r_index);
break;
}
if (retval) {
if (r_ob) {
*r_ob = ob;
}
if (r_obmat) {
copy_m4_m4(r_obmat, obmat);
}
return retval;
}
return 0;
}
struct SnapObjUserData {
SnapData *snapdata;
/* read/write args */
@ -2761,32 +2622,109 @@ struct SnapObjUserData {
short ret;
};
static void sanp_obj_cb(SnapObjectContext *sctx,
bool is_obedit,
/**
* \param use_obedit: Uses the coordinates of BMesh (if any) to do the snapping;
*
* \note Duplicate args here are documented at #snapObjectsRay
*/
static void sanp_obj_fn(SnapObjectContext *sctx,
bool use_obedit,
bool use_backface_culling,
Object *ob,
float obmat[4][4],
void *data)
{
struct SnapObjUserData *dt = data;
short retval = 0;
short elem = snapObject(sctx,
dt->snapdata,
ob,
obmat,
is_obedit,
use_backface_culling,
/* read/write args */
dt->dist_px,
/* return args */
dt->r_loc,
dt->r_no,
dt->r_index,
dt->r_ob,
dt->r_obmat);
switch (ob->type) {
case OB_MESH: {
Mesh *me = ob->data;
if (BKE_object_is_in_editmode(ob)) {
if (use_obedit) {
/* Operators only update the editmesh looptris of the original mesh. */
BMEditMesh *em_orig = BKE_editmesh_from_object(DEG_get_original_object(ob));
retval = snapEditMesh(sctx,
dt->snapdata,
ob,
em_orig,
obmat,
use_backface_culling,
dt->dist_px,
dt->r_loc,
dt->r_no,
dt->r_index);
break;
}
else {
BMEditMesh *em = BKE_editmesh_from_object(ob);
if (em->mesh_eval_final) {
me = em->mesh_eval_final;
}
}
}
else if (ob->dt == OB_BOUNDBOX) {
/* Do not snap to objects that are in bounding box display mode */
return;
}
if (elem) {
dt->ret = elem;
retval = snapMesh(sctx,
dt->snapdata,
ob,
me,
obmat,
use_backface_culling,
dt->dist_px,
dt->r_loc,
dt->r_no,
dt->r_index);
break;
}
case OB_ARMATURE:
retval = snapArmature(
dt->snapdata, ob, obmat, use_obedit, dt->dist_px, dt->r_loc, dt->r_no, dt->r_index);
break;
case OB_CURVE:
retval = snapCurve(
dt->snapdata, ob, obmat, use_obedit, dt->dist_px, dt->r_loc, dt->r_no, dt->r_index);
break; /* Use ATTR_FALLTHROUGH if we want to snap to the generated mesh. */
case OB_SURF:
case OB_FONT: {
Mesh *mesh_eval = BKE_object_get_evaluated_mesh(ob);
if (mesh_eval) {
retval |= snapMesh(sctx,
dt->snapdata,
ob,
mesh_eval,
obmat,
use_backface_culling,
dt->dist_px,
dt->r_loc,
dt->r_no,
dt->r_index);
}
break;
}
case OB_EMPTY:
retval = snapEmpty(dt->snapdata, ob, obmat, dt->dist_px, dt->r_loc, dt->r_no, dt->r_index);
break;
case OB_GPENCIL:
retval = snapEmpty(dt->snapdata, ob, obmat, dt->dist_px, dt->r_loc, dt->r_no, dt->r_index);
break;
case OB_CAMERA:
retval = snapCamera(
sctx, dt->snapdata, ob, obmat, dt->dist_px, dt->r_loc, dt->r_no, dt->r_index);
break;
}
if (retval) {
if (dt->r_ob) {
*dt->r_ob = ob;
}
if (dt->r_obmat) {
copy_m4_m4(dt->r_obmat, obmat);
}
dt->ret = retval;
}
}
@ -2839,7 +2777,7 @@ static short snapObjectsRay(SnapObjectContext *sctx,
.ret = 0,
};
iter_snap_objects(sctx, depsgraph, params, sanp_obj_cb, &data);
iter_snap_objects(sctx, depsgraph, params, sanp_obj_fn, &data);
return data.ret;
}