Cleanup: move mask queries into own file
Similar functions to lookup nearest mask points were in mask_add.c & mask_edit.c
This commit is contained in:
@@ -36,6 +36,13 @@ struct bContext;
|
||||
struct wmKeyConfig;
|
||||
|
||||
/* mask_edit.c */
|
||||
void ED_mask_deselect_all(const struct bContext *C);
|
||||
|
||||
void ED_operatortypes_mask(void);
|
||||
void ED_keymap_mask(struct wmKeyConfig *keyconf);
|
||||
void ED_operatormacros_mask(void);
|
||||
|
||||
/* mask_query.c */
|
||||
void ED_mask_get_size(struct ScrArea *sa, int *width, int *height);
|
||||
void ED_mask_zoom(struct ScrArea *sa, struct ARegion *region, float *zoomx, float *zoomy);
|
||||
void ED_mask_get_aspect(struct ScrArea *sa, struct ARegion *region, float *aspx, float *aspy);
|
||||
@@ -54,12 +61,6 @@ void ED_mask_point_pos__reverse(
|
||||
void ED_mask_cursor_location_get(struct ScrArea *sa, float cursor[2]);
|
||||
bool ED_mask_selected_minmax(const struct bContext *C, float min[2], float max[2]);
|
||||
|
||||
void ED_mask_deselect_all(const struct bContext *C);
|
||||
|
||||
void ED_operatortypes_mask(void);
|
||||
void ED_keymap_mask(struct wmKeyConfig *keyconf);
|
||||
void ED_operatormacros_mask(void);
|
||||
|
||||
/* mask_draw.c */
|
||||
void ED_mask_draw(const struct bContext *C, const char draw_flag, const char draw_type);
|
||||
void ED_mask_draw_region(struct Depsgraph *depsgraph,
|
||||
|
||||
@@ -40,6 +40,7 @@ set(SRC
|
||||
mask_edit.c
|
||||
mask_editaction.c
|
||||
mask_ops.c
|
||||
mask_query.c
|
||||
mask_relationships.c
|
||||
mask_select.c
|
||||
mask_shapekey.c
|
||||
|
||||
@@ -47,158 +47,6 @@
|
||||
|
||||
#include "mask_intern.h" /* own include */
|
||||
|
||||
bool ED_mask_find_nearest_diff_point(const bContext *C,
|
||||
struct Mask *mask_orig,
|
||||
const float normal_co[2],
|
||||
int threshold,
|
||||
bool feather,
|
||||
float tangent[2],
|
||||
const bool use_deform,
|
||||
const bool use_project,
|
||||
MaskLayer **r_mask_layer,
|
||||
MaskSpline **r_spline,
|
||||
MaskSplinePoint **r_point,
|
||||
float *r_u,
|
||||
float *r_score)
|
||||
{
|
||||
ScrArea *sa = CTX_wm_area(C);
|
||||
ARegion *region = CTX_wm_region(C);
|
||||
|
||||
MaskLayer *point_mask_layer;
|
||||
MaskSpline *point_spline;
|
||||
MaskSplinePoint *point = NULL;
|
||||
float dist_best_sq = FLT_MAX, co[2];
|
||||
int width, height;
|
||||
float u = 0.0f;
|
||||
float scalex, scaley;
|
||||
|
||||
Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
|
||||
Mask *mask_eval = (Mask *)DEG_get_evaluated_id(depsgraph, &mask_orig->id);
|
||||
|
||||
ED_mask_get_size(sa, &width, &height);
|
||||
ED_mask_pixelspace_factor(sa, region, &scalex, &scaley);
|
||||
|
||||
co[0] = normal_co[0] * scalex;
|
||||
co[1] = normal_co[1] * scaley;
|
||||
|
||||
for (MaskLayer *mask_layer_orig = mask_orig->masklayers.first,
|
||||
*mask_layer_eval = mask_eval->masklayers.first;
|
||||
mask_layer_orig != NULL;
|
||||
mask_layer_orig = mask_layer_orig->next, mask_layer_eval = mask_layer_eval->next) {
|
||||
if (mask_layer_orig->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
for (MaskSpline *spline_orig = mask_layer_orig->splines.first,
|
||||
*spline_eval = mask_layer_eval->splines.first;
|
||||
spline_orig != NULL;
|
||||
spline_orig = spline_orig->next, spline_eval = spline_eval->next) {
|
||||
int i;
|
||||
MaskSplinePoint *cur_point_eval;
|
||||
|
||||
for (i = 0, cur_point_eval = use_deform ? spline_eval->points_deform : spline_eval->points;
|
||||
i < spline_eval->tot_point;
|
||||
i++, cur_point_eval++) {
|
||||
unsigned int tot_diff_point;
|
||||
float *diff_points = BKE_mask_point_segment_diff(
|
||||
spline_eval, cur_point_eval, width, height, &tot_diff_point);
|
||||
|
||||
if (diff_points) {
|
||||
int j, tot_point;
|
||||
unsigned int tot_feather_point;
|
||||
float *feather_points = NULL, *points;
|
||||
|
||||
if (feather) {
|
||||
feather_points = BKE_mask_point_segment_feather_diff(
|
||||
spline_eval, cur_point_eval, width, height, &tot_feather_point);
|
||||
|
||||
points = feather_points;
|
||||
tot_point = tot_feather_point;
|
||||
}
|
||||
else {
|
||||
points = diff_points;
|
||||
tot_point = tot_diff_point;
|
||||
}
|
||||
|
||||
for (j = 0; j < tot_point - 1; j++) {
|
||||
float dist_sq, a[2], b[2];
|
||||
|
||||
a[0] = points[2 * j] * scalex;
|
||||
a[1] = points[2 * j + 1] * scaley;
|
||||
|
||||
b[0] = points[2 * j + 2] * scalex;
|
||||
b[1] = points[2 * j + 3] * scaley;
|
||||
|
||||
dist_sq = dist_squared_to_line_segment_v2(co, a, b);
|
||||
|
||||
if (dist_sq < dist_best_sq) {
|
||||
if (tangent) {
|
||||
sub_v2_v2v2(tangent, &diff_points[2 * j + 2], &diff_points[2 * j]);
|
||||
}
|
||||
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point_spline = spline_orig;
|
||||
point = use_deform ?
|
||||
&spline_orig->points[(cur_point_eval - spline_eval->points_deform)] :
|
||||
&spline_orig->points[(cur_point_eval - spline_eval->points)];
|
||||
dist_best_sq = dist_sq;
|
||||
u = (float)j / tot_point;
|
||||
}
|
||||
}
|
||||
|
||||
if (feather_points != NULL) {
|
||||
MEM_freeN(feather_points);
|
||||
}
|
||||
MEM_freeN(diff_points);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (point && dist_best_sq < threshold) {
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = point_mask_layer;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = point_spline;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = point;
|
||||
}
|
||||
|
||||
if (r_u) {
|
||||
/* TODO(sergey): Projection fails in some weirdo cases.. */
|
||||
if (use_project) {
|
||||
u = BKE_mask_spline_project_co(point_spline, point, u, normal_co, MASK_PROJ_ANY);
|
||||
}
|
||||
|
||||
*r_u = u;
|
||||
}
|
||||
|
||||
if (r_score) {
|
||||
*r_score = dist_best_sq;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = NULL;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = NULL;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = NULL;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/******************** add vertex *********************/
|
||||
|
||||
static void setup_vertex_point(Mask *mask,
|
||||
|
||||
@@ -47,7 +47,9 @@
|
||||
|
||||
#include "mask_intern.h" /* own include */
|
||||
|
||||
/********************** generic poll functions *********************/
|
||||
/* -------------------------------------------------------------------- */
|
||||
/** \name Poll Functions
|
||||
* \{ */
|
||||
|
||||
bool ED_maskedit_poll(bContext *C)
|
||||
{
|
||||
@@ -81,344 +83,11 @@ bool ED_maskedit_mask_poll(bContext *C)
|
||||
return false;
|
||||
}
|
||||
|
||||
/********************** registration *********************/
|
||||
/** \} */
|
||||
|
||||
/* takes event->mval */
|
||||
void ED_mask_mouse_pos(ScrArea *sa, ARegion *region, const int mval[2], float co[2])
|
||||
{
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_clip_mouse_pos(sc, region, mval, co);
|
||||
BKE_mask_coord_from_movieclip(sc->clip, &sc->user, co, co);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
UI_view2d_region_to_view(®ion->v2d, mval[0], mval[1], &co[0], &co[1]);
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_image_mouse_pos(sima, region, mval, co);
|
||||
BKE_mask_coord_from_image(sima->image, &sima->iuser, co, co);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
}
|
||||
}
|
||||
|
||||
/* input: x/y - mval space
|
||||
* output: xr/yr - mask point space */
|
||||
void ED_mask_point_pos(ScrArea *sa, ARegion *region, float x, float y, float *xr, float *yr)
|
||||
{
|
||||
float co[2];
|
||||
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_clip_point_stable_pos(sc, region, x, y, &co[0], &co[1]);
|
||||
BKE_mask_coord_from_movieclip(sc->clip, &sc->user, co, co);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ:
|
||||
zero_v2(co); /* MASKTODO */
|
||||
break;
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_image_point_pos(sima, region, x, y, &co[0], &co[1]);
|
||||
BKE_mask_coord_from_image(sima->image, &sima->iuser, co, co);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
}
|
||||
|
||||
*xr = co[0];
|
||||
*yr = co[1];
|
||||
}
|
||||
|
||||
void ED_mask_point_pos__reverse(
|
||||
ScrArea *sa, ARegion *region, float x, float y, float *xr, float *yr)
|
||||
{
|
||||
float co[2];
|
||||
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
co[0] = x;
|
||||
co[1] = y;
|
||||
BKE_mask_coord_to_movieclip(sc->clip, &sc->user, co, co);
|
||||
ED_clip_point_stable_pos__reverse(sc, region, co, co);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ:
|
||||
zero_v2(co); /* MASKTODO */
|
||||
break;
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
co[0] = x;
|
||||
co[1] = y;
|
||||
BKE_mask_coord_to_image(sima->image, &sima->iuser, co, co);
|
||||
ED_image_point_pos__reverse(sima, region, co, co);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
}
|
||||
|
||||
*xr = co[0];
|
||||
*yr = co[1];
|
||||
}
|
||||
|
||||
void ED_mask_get_size(ScrArea *sa, int *width, int *height)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_space_clip_get_size(sc, width, height);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
// Scene *scene = CTX_data_scene(C);
|
||||
// *width = (scene->r.size * scene->r.xsch) / 100;
|
||||
// *height = (scene->r.size * scene->r.ysch) / 100;
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_space_image_get_size(sima, width, height);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*width = 0;
|
||||
*height = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*width = 0;
|
||||
*height = 0;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_zoom(ScrArea *sa, ARegion *region, float *zoomx, float *zoomy)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_space_clip_get_zoom(sc, region, zoomx, zoomy);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
*zoomx = *zoomy = 1.0f;
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_space_image_get_zoom(sima, region, zoomx, zoomy);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*zoomx = *zoomy = 1.0f;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*zoomx = *zoomy = 1.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_get_aspect(ScrArea *sa, ARegion *UNUSED(region), float *aspx, float *aspy)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_space_clip_get_aspect(sc, aspx, aspy);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
*aspx = *aspy = 1.0f; /* MASKTODO - render aspect? */
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_space_image_get_aspect(sima, aspx, aspy);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*aspx = *aspy = 1.0f;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*aspx = *aspy = 1.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_pixelspace_factor(ScrArea *sa, ARegion *region, float *scalex, float *scaley)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
float aspx, aspy;
|
||||
|
||||
UI_view2d_scale_get(®ion->v2d, scalex, scaley);
|
||||
ED_space_clip_get_aspect(sc, &aspx, &aspy);
|
||||
|
||||
*scalex *= aspx;
|
||||
*scaley *= aspy;
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
*scalex = *scaley = 1.0f; /* MASKTODO? */
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
float aspx, aspy;
|
||||
|
||||
UI_view2d_scale_get(®ion->v2d, scalex, scaley);
|
||||
ED_space_image_get_aspect(sima, &aspx, &aspy);
|
||||
|
||||
*scalex *= aspx;
|
||||
*scaley *= aspy;
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*scalex = *scaley = 1.0f;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*scalex = *scaley = 1.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_cursor_location_get(ScrArea *sa, float cursor[2])
|
||||
{
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *space_clip = sa->spacedata.first;
|
||||
copy_v2_v2(cursor, space_clip->cursor);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
zero_v2(cursor);
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *space_image = sa->spacedata.first;
|
||||
copy_v2_v2(cursor, space_image->cursor);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(cursor);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(cursor);
|
||||
}
|
||||
}
|
||||
|
||||
bool ED_mask_selected_minmax(const bContext *C, float min[2], float max[2])
|
||||
{
|
||||
Mask *mask = CTX_data_edit_mask(C);
|
||||
bool ok = false;
|
||||
|
||||
if (mask == NULL) {
|
||||
return ok;
|
||||
}
|
||||
|
||||
INIT_MINMAX2(min, max);
|
||||
for (MaskLayer *mask_layer = mask->masklayers.first; mask_layer != NULL;
|
||||
mask_layer = mask_layer->next) {
|
||||
if (mask_layer->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
for (MaskSpline *spline = mask_layer->splines.first; spline != NULL; spline = spline->next) {
|
||||
MaskSplinePoint *points_array = BKE_mask_spline_point_array(spline);
|
||||
for (int i = 0; i < spline->tot_point; i++) {
|
||||
MaskSplinePoint *point = &spline->points[i];
|
||||
MaskSplinePoint *deform_point = &points_array[i];
|
||||
BezTriple *bezt = &point->bezt;
|
||||
float handle[2];
|
||||
if (!MASKPOINT_ISSEL_ANY(point)) {
|
||||
continue;
|
||||
}
|
||||
if (bezt->f2 & SELECT) {
|
||||
minmax_v2v2_v2(min, max, deform_point->bezt.vec[1]);
|
||||
}
|
||||
if (BKE_mask_point_handles_mode_get(point) == MASK_HANDLE_MODE_STICK) {
|
||||
BKE_mask_point_handle(deform_point, MASK_WHICH_HANDLE_STICK, handle);
|
||||
minmax_v2v2_v2(min, max, handle);
|
||||
}
|
||||
else {
|
||||
if ((bezt->f1 & SELECT) && (bezt->h1 != HD_VECT)) {
|
||||
BKE_mask_point_handle(deform_point, MASK_WHICH_HANDLE_LEFT, handle);
|
||||
minmax_v2v2_v2(min, max, handle);
|
||||
}
|
||||
if ((bezt->f3 & SELECT) && (bezt->h2 != HD_VECT)) {
|
||||
BKE_mask_point_handle(deform_point, MASK_WHICH_HANDLE_RIGHT, handle);
|
||||
minmax_v2v2_v2(min, max, handle);
|
||||
}
|
||||
}
|
||||
ok = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
return ok;
|
||||
}
|
||||
|
||||
/********************** registration *********************/
|
||||
/* -------------------------------------------------------------------- */
|
||||
/** \name Registration
|
||||
* \{ */
|
||||
|
||||
void ED_operatortypes_mask(void)
|
||||
{
|
||||
@@ -522,3 +191,5 @@ void ED_operatormacros_mask(void)
|
||||
RNA_boolean_set(otmacro->ptr, "use_proportional_edit", false);
|
||||
RNA_boolean_set(otmacro->ptr, "mirror", false);
|
||||
}
|
||||
|
||||
/** \} */
|
||||
|
||||
@@ -31,20 +31,6 @@ struct wmOperatorType;
|
||||
/* internal exports only */
|
||||
|
||||
/* mask_add.c */
|
||||
bool ED_mask_find_nearest_diff_point(const struct bContext *C,
|
||||
struct Mask *mask,
|
||||
const float normal_co[2],
|
||||
int threshold,
|
||||
bool feather,
|
||||
float tangent[2],
|
||||
const bool use_deform,
|
||||
const bool use_project,
|
||||
struct MaskLayer **r_mask_layer,
|
||||
struct MaskSpline **r_spline,
|
||||
struct MaskSplinePoint **r_point,
|
||||
float *r_u,
|
||||
float *r_score);
|
||||
|
||||
void MASK_OT_add_vertex(struct wmOperatorType *ot);
|
||||
void MASK_OT_add_feather_vertex(struct wmOperatorType *ot);
|
||||
void MASK_OT_primitive_circle_add(struct wmOperatorType *ot);
|
||||
@@ -72,25 +58,6 @@ void MASK_OT_normals_make_consistent(struct wmOperatorType *ot);
|
||||
|
||||
void MASK_OT_handle_type_set(struct wmOperatorType *ot);
|
||||
|
||||
bool ED_mask_feather_find_nearest(const struct bContext *C,
|
||||
struct Mask *mask,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
struct MaskLayer **r_mask_layer,
|
||||
struct MaskSpline **r_spline,
|
||||
struct MaskSplinePoint **r_point,
|
||||
struct MaskSplinePointUW **r_uw,
|
||||
float *r_score);
|
||||
|
||||
struct MaskSplinePoint *ED_mask_point_find_nearest(const struct bContext *C,
|
||||
struct Mask *mask,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
struct MaskLayer **r_mask_layer,
|
||||
struct MaskSpline **r_spline,
|
||||
eMaskWhichHandle *r_which_handle,
|
||||
float *r_score);
|
||||
|
||||
void MASK_OT_layer_move(struct wmOperatorType *ot);
|
||||
|
||||
void MASK_OT_duplicate(struct wmOperatorType *ot);
|
||||
@@ -126,6 +93,38 @@ void ED_mask_select_flush_all(struct Mask *mask);
|
||||
bool ED_maskedit_poll(struct bContext *C);
|
||||
bool ED_maskedit_mask_poll(struct bContext *C);
|
||||
|
||||
/* mask_query.c */
|
||||
bool ED_mask_find_nearest_diff_point(const struct bContext *C,
|
||||
struct Mask *mask,
|
||||
const float normal_co[2],
|
||||
int threshold,
|
||||
bool feather,
|
||||
float tangent[2],
|
||||
const bool use_deform,
|
||||
const bool use_project,
|
||||
struct MaskLayer **r_mask_layer,
|
||||
struct MaskSpline **r_spline,
|
||||
struct MaskSplinePoint **r_point,
|
||||
float *r_u,
|
||||
float *r_score);
|
||||
bool ED_mask_feather_find_nearest(const struct bContext *C,
|
||||
struct Mask *mask,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
struct MaskLayer **r_mask_layer,
|
||||
struct MaskSpline **r_spline,
|
||||
struct MaskSplinePoint **r_point,
|
||||
struct MaskSplinePointUW **r_uw,
|
||||
float *r_score);
|
||||
struct MaskSplinePoint *ED_mask_point_find_nearest(const struct bContext *C,
|
||||
struct Mask *mask,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
struct MaskLayer **r_mask_layer,
|
||||
struct MaskSpline **r_spline,
|
||||
eMaskWhichHandle *r_which_handle,
|
||||
float *r_score);
|
||||
|
||||
/* mask_shapekey.c */
|
||||
void MASK_OT_shape_key_insert(struct wmOperatorType *ot);
|
||||
void MASK_OT_shape_key_clear(struct wmOperatorType *ot);
|
||||
|
||||
@@ -52,298 +52,6 @@
|
||||
|
||||
#include "mask_intern.h" /* own include */
|
||||
|
||||
/******************** utility functions *********************/
|
||||
|
||||
static void mask_point_scaled_handle(/*const*/ MaskSplinePoint *point,
|
||||
/*const*/ eMaskWhichHandle which_handle,
|
||||
const float scalex,
|
||||
const float scaley,
|
||||
float handle[2])
|
||||
{
|
||||
BKE_mask_point_handle(point, which_handle, handle);
|
||||
handle[0] *= scalex;
|
||||
handle[1] *= scaley;
|
||||
}
|
||||
|
||||
MaskSplinePoint *ED_mask_point_find_nearest(const bContext *C,
|
||||
Mask *mask_orig,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
MaskLayer **r_mask_layer,
|
||||
MaskSpline **r_spline,
|
||||
eMaskWhichHandle *r_which_handle,
|
||||
float *r_score)
|
||||
{
|
||||
ScrArea *sa = CTX_wm_area(C);
|
||||
ARegion *region = CTX_wm_region(C);
|
||||
|
||||
MaskLayer *point_mask_layer = NULL;
|
||||
MaskSpline *point_spline = NULL;
|
||||
MaskSplinePoint *point = NULL;
|
||||
float co[2];
|
||||
const float threshold_sq = threshold * threshold;
|
||||
float len_sq = FLT_MAX, scalex, scaley;
|
||||
eMaskWhichHandle which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
int width, height;
|
||||
|
||||
Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
|
||||
Mask *mask_eval = (Mask *)DEG_get_evaluated_id(depsgraph, &mask_orig->id);
|
||||
|
||||
ED_mask_get_size(sa, &width, &height);
|
||||
ED_mask_pixelspace_factor(sa, region, &scalex, &scaley);
|
||||
|
||||
co[0] = normal_co[0] * scalex;
|
||||
co[1] = normal_co[1] * scaley;
|
||||
|
||||
for (MaskLayer *mask_layer_orig = mask_orig->masklayers.first,
|
||||
*mask_layer_eval = mask_eval->masklayers.first;
|
||||
mask_layer_orig != NULL;
|
||||
mask_layer_orig = mask_layer_orig->next, mask_layer_eval = mask_layer_eval->next) {
|
||||
|
||||
if (mask_layer_orig->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
for (MaskSpline *spline_orig = mask_layer_orig->splines.first,
|
||||
*spline_eval = mask_layer_eval->splines.first;
|
||||
spline_orig != NULL;
|
||||
spline_orig = spline_orig->next, spline_eval = spline_eval->next) {
|
||||
MaskSplinePoint *points_array = BKE_mask_spline_point_array(spline_eval);
|
||||
|
||||
for (int i = 0; i < spline_orig->tot_point; i++) {
|
||||
MaskSplinePoint *cur_point_orig = &spline_orig->points[i];
|
||||
MaskSplinePoint *cur_point_deform_eval = &points_array[i];
|
||||
eMaskWhichHandle cur_which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
BezTriple *bezt = &cur_point_deform_eval->bezt;
|
||||
float cur_len_sq, vec[2];
|
||||
|
||||
vec[0] = bezt->vec[1][0] * scalex;
|
||||
vec[1] = bezt->vec[1][1] * scaley;
|
||||
|
||||
cur_len_sq = len_squared_v2v2(co, vec);
|
||||
|
||||
if (cur_len_sq < len_sq) {
|
||||
point_spline = spline_orig;
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point = cur_point_orig;
|
||||
len_sq = cur_len_sq;
|
||||
which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
}
|
||||
|
||||
if (BKE_mask_point_handles_mode_get(cur_point_deform_eval) == MASK_HANDLE_MODE_STICK) {
|
||||
float handle[2];
|
||||
mask_point_scaled_handle(
|
||||
cur_point_deform_eval, MASK_WHICH_HANDLE_STICK, scalex, scaley, handle);
|
||||
cur_len_sq = len_squared_v2v2(co, handle);
|
||||
cur_which_handle = MASK_WHICH_HANDLE_STICK;
|
||||
}
|
||||
else {
|
||||
float handle_left[2], handle_right[2];
|
||||
float len_left_sq, len_right_sq;
|
||||
mask_point_scaled_handle(
|
||||
cur_point_deform_eval, MASK_WHICH_HANDLE_LEFT, scalex, scaley, handle_left);
|
||||
mask_point_scaled_handle(
|
||||
cur_point_deform_eval, MASK_WHICH_HANDLE_RIGHT, scalex, scaley, handle_right);
|
||||
|
||||
len_left_sq = len_squared_v2v2(co, handle_left);
|
||||
len_right_sq = len_squared_v2v2(co, handle_right);
|
||||
if (i == 0) {
|
||||
if (len_left_sq <= len_right_sq) {
|
||||
if (bezt->h1 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_LEFT;
|
||||
cur_len_sq = len_left_sq;
|
||||
}
|
||||
}
|
||||
else if (bezt->h2 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_RIGHT;
|
||||
cur_len_sq = len_right_sq;
|
||||
}
|
||||
}
|
||||
else {
|
||||
if (len_right_sq <= len_left_sq) {
|
||||
if (bezt->h2 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_RIGHT;
|
||||
cur_len_sq = len_right_sq;
|
||||
}
|
||||
}
|
||||
else if (bezt->h1 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_LEFT;
|
||||
cur_len_sq = len_left_sq;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (cur_len_sq <= len_sq && cur_which_handle != MASK_WHICH_HANDLE_NONE) {
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point_spline = spline_orig;
|
||||
point = cur_point_orig;
|
||||
len_sq = cur_len_sq;
|
||||
which_handle = cur_which_handle;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (len_sq < threshold_sq) {
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = point_mask_layer;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = point_spline;
|
||||
}
|
||||
|
||||
if (r_which_handle) {
|
||||
*r_which_handle = which_handle;
|
||||
}
|
||||
|
||||
if (r_score) {
|
||||
*r_score = sqrtf(len_sq);
|
||||
}
|
||||
|
||||
return point;
|
||||
}
|
||||
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = NULL;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = NULL;
|
||||
}
|
||||
|
||||
if (r_which_handle) {
|
||||
*r_which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
bool ED_mask_feather_find_nearest(const bContext *C,
|
||||
Mask *mask_orig,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
MaskLayer **r_mask_layer,
|
||||
MaskSpline **r_spline,
|
||||
MaskSplinePoint **r_point,
|
||||
MaskSplinePointUW **r_uw,
|
||||
float *r_score)
|
||||
{
|
||||
ScrArea *sa = CTX_wm_area(C);
|
||||
ARegion *region = CTX_wm_region(C);
|
||||
|
||||
MaskLayer *point_mask_layer = NULL;
|
||||
MaskSpline *point_spline = NULL;
|
||||
MaskSplinePoint *point = NULL;
|
||||
MaskSplinePointUW *uw = NULL;
|
||||
const float threshold_sq = threshold * threshold;
|
||||
float len = FLT_MAX, co[2];
|
||||
float scalex, scaley;
|
||||
int width, height;
|
||||
|
||||
Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
|
||||
Mask *mask_eval = (Mask *)DEG_get_evaluated_id(depsgraph, &mask_orig->id);
|
||||
|
||||
ED_mask_get_size(sa, &width, &height);
|
||||
ED_mask_pixelspace_factor(sa, region, &scalex, &scaley);
|
||||
|
||||
co[0] = normal_co[0] * scalex;
|
||||
co[1] = normal_co[1] * scaley;
|
||||
|
||||
for (MaskLayer *mask_layer_orig = mask_orig->masklayers.first,
|
||||
*mask_layer_eval = mask_eval->masklayers.first;
|
||||
mask_layer_orig != NULL;
|
||||
mask_layer_orig = mask_layer_orig->next, mask_layer_eval = mask_layer_eval->next) {
|
||||
|
||||
for (MaskSpline *spline_orig = mask_layer_orig->splines.first,
|
||||
*spline_eval = mask_layer_eval->splines.first;
|
||||
spline_orig != NULL;
|
||||
spline_orig = spline_orig->next, spline_eval = spline_eval->next) {
|
||||
// MaskSplinePoint *points_array = BKE_mask_spline_point_array(spline);
|
||||
|
||||
int i, tot_feather_point;
|
||||
float(*feather_points)[2], (*fp)[2];
|
||||
|
||||
if (mask_layer_orig->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
feather_points = fp = BKE_mask_spline_feather_points(spline_eval, &tot_feather_point);
|
||||
|
||||
for (i = 0; i < spline_orig->tot_point; i++) {
|
||||
int j;
|
||||
MaskSplinePoint *cur_point_orig = &spline_orig->points[i];
|
||||
MaskSplinePoint *cur_point_eval = &spline_eval->points[i];
|
||||
|
||||
for (j = 0; j <= cur_point_eval->tot_uw; j++) {
|
||||
float cur_len_sq, vec[2];
|
||||
|
||||
vec[0] = (*fp)[0] * scalex;
|
||||
vec[1] = (*fp)[1] * scaley;
|
||||
|
||||
cur_len_sq = len_squared_v2v2(vec, co);
|
||||
|
||||
if (point == NULL || cur_len_sq < len) {
|
||||
if (j == 0) {
|
||||
uw = NULL;
|
||||
}
|
||||
else {
|
||||
uw = &cur_point_orig->uw[j - 1];
|
||||
}
|
||||
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point_spline = spline_orig;
|
||||
point = cur_point_orig;
|
||||
len = cur_len_sq;
|
||||
}
|
||||
|
||||
fp++;
|
||||
}
|
||||
}
|
||||
|
||||
MEM_freeN(feather_points);
|
||||
}
|
||||
}
|
||||
|
||||
if (len < threshold_sq) {
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = point_mask_layer;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = point_spline;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = point;
|
||||
}
|
||||
|
||||
if (r_uw) {
|
||||
*r_uw = uw;
|
||||
}
|
||||
|
||||
if (r_score) {
|
||||
*r_score = sqrtf(len);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = NULL;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = NULL;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = NULL;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/******************** create new mask *********************/
|
||||
|
||||
Mask *ED_mask_new(bContext *C, const char *name)
|
||||
|
||||
841
source/blender/editors/mask/mask_query.c
Normal file
841
source/blender/editors/mask/mask_query.c
Normal file
@@ -0,0 +1,841 @@
|
||||
/*
|
||||
* 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) 2012 Blender Foundation.
|
||||
* All rights reserved.
|
||||
*/
|
||||
|
||||
/** \file
|
||||
* \ingroup edmask
|
||||
*/
|
||||
|
||||
#include "MEM_guardedalloc.h"
|
||||
|
||||
#include "BLI_math.h"
|
||||
|
||||
#include "BKE_context.h"
|
||||
#include "BKE_mask.h"
|
||||
|
||||
#include "DEG_depsgraph.h"
|
||||
#include "DEG_depsgraph_query.h"
|
||||
|
||||
#include "DNA_mask_types.h"
|
||||
#include "DNA_scene_types.h"
|
||||
#include "DNA_screen_types.h"
|
||||
|
||||
#include "WM_api.h"
|
||||
#include "WM_types.h"
|
||||
|
||||
#include "ED_clip.h"
|
||||
#include "ED_image.h"
|
||||
#include "ED_mask.h" /* own include */
|
||||
#include "ED_screen.h"
|
||||
#include "ED_select_utils.h"
|
||||
|
||||
#include "UI_view2d.h"
|
||||
|
||||
#include "RNA_access.h"
|
||||
#include "RNA_define.h"
|
||||
|
||||
#include "mask_intern.h" /* own include */
|
||||
|
||||
/* -------------------------------------------------------------------- */
|
||||
/** \name Spatial Queries
|
||||
* \{ */
|
||||
|
||||
bool ED_mask_find_nearest_diff_point(const bContext *C,
|
||||
struct Mask *mask_orig,
|
||||
const float normal_co[2],
|
||||
int threshold,
|
||||
bool feather,
|
||||
float tangent[2],
|
||||
const bool use_deform,
|
||||
const bool use_project,
|
||||
MaskLayer **r_mask_layer,
|
||||
MaskSpline **r_spline,
|
||||
MaskSplinePoint **r_point,
|
||||
float *r_u,
|
||||
float *r_score)
|
||||
{
|
||||
ScrArea *sa = CTX_wm_area(C);
|
||||
ARegion *region = CTX_wm_region(C);
|
||||
|
||||
MaskLayer *point_mask_layer;
|
||||
MaskSpline *point_spline;
|
||||
MaskSplinePoint *point = NULL;
|
||||
float dist_best_sq = FLT_MAX, co[2];
|
||||
int width, height;
|
||||
float u = 0.0f;
|
||||
float scalex, scaley;
|
||||
|
||||
Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
|
||||
Mask *mask_eval = (Mask *)DEG_get_evaluated_id(depsgraph, &mask_orig->id);
|
||||
|
||||
ED_mask_get_size(sa, &width, &height);
|
||||
ED_mask_pixelspace_factor(sa, region, &scalex, &scaley);
|
||||
|
||||
co[0] = normal_co[0] * scalex;
|
||||
co[1] = normal_co[1] * scaley;
|
||||
|
||||
for (MaskLayer *mask_layer_orig = mask_orig->masklayers.first,
|
||||
*mask_layer_eval = mask_eval->masklayers.first;
|
||||
mask_layer_orig != NULL;
|
||||
mask_layer_orig = mask_layer_orig->next, mask_layer_eval = mask_layer_eval->next) {
|
||||
if (mask_layer_orig->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
for (MaskSpline *spline_orig = mask_layer_orig->splines.first,
|
||||
*spline_eval = mask_layer_eval->splines.first;
|
||||
spline_orig != NULL;
|
||||
spline_orig = spline_orig->next, spline_eval = spline_eval->next) {
|
||||
int i;
|
||||
MaskSplinePoint *cur_point_eval;
|
||||
|
||||
for (i = 0, cur_point_eval = use_deform ? spline_eval->points_deform : spline_eval->points;
|
||||
i < spline_eval->tot_point;
|
||||
i++, cur_point_eval++) {
|
||||
unsigned int tot_diff_point;
|
||||
float *diff_points = BKE_mask_point_segment_diff(
|
||||
spline_eval, cur_point_eval, width, height, &tot_diff_point);
|
||||
|
||||
if (diff_points) {
|
||||
int j, tot_point;
|
||||
unsigned int tot_feather_point;
|
||||
float *feather_points = NULL, *points;
|
||||
|
||||
if (feather) {
|
||||
feather_points = BKE_mask_point_segment_feather_diff(
|
||||
spline_eval, cur_point_eval, width, height, &tot_feather_point);
|
||||
|
||||
points = feather_points;
|
||||
tot_point = tot_feather_point;
|
||||
}
|
||||
else {
|
||||
points = diff_points;
|
||||
tot_point = tot_diff_point;
|
||||
}
|
||||
|
||||
for (j = 0; j < tot_point - 1; j++) {
|
||||
float dist_sq, a[2], b[2];
|
||||
|
||||
a[0] = points[2 * j] * scalex;
|
||||
a[1] = points[2 * j + 1] * scaley;
|
||||
|
||||
b[0] = points[2 * j + 2] * scalex;
|
||||
b[1] = points[2 * j + 3] * scaley;
|
||||
|
||||
dist_sq = dist_squared_to_line_segment_v2(co, a, b);
|
||||
|
||||
if (dist_sq < dist_best_sq) {
|
||||
if (tangent) {
|
||||
sub_v2_v2v2(tangent, &diff_points[2 * j + 2], &diff_points[2 * j]);
|
||||
}
|
||||
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point_spline = spline_orig;
|
||||
point = use_deform ?
|
||||
&spline_orig->points[(cur_point_eval - spline_eval->points_deform)] :
|
||||
&spline_orig->points[(cur_point_eval - spline_eval->points)];
|
||||
dist_best_sq = dist_sq;
|
||||
u = (float)j / tot_point;
|
||||
}
|
||||
}
|
||||
|
||||
if (feather_points != NULL) {
|
||||
MEM_freeN(feather_points);
|
||||
}
|
||||
MEM_freeN(diff_points);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (point && dist_best_sq < threshold) {
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = point_mask_layer;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = point_spline;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = point;
|
||||
}
|
||||
|
||||
if (r_u) {
|
||||
/* TODO(sergey): Projection fails in some weirdo cases.. */
|
||||
if (use_project) {
|
||||
u = BKE_mask_spline_project_co(point_spline, point, u, normal_co, MASK_PROJ_ANY);
|
||||
}
|
||||
|
||||
*r_u = u;
|
||||
}
|
||||
|
||||
if (r_score) {
|
||||
*r_score = dist_best_sq;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = NULL;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = NULL;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = NULL;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
static void mask_point_scaled_handle(/*const*/ MaskSplinePoint *point,
|
||||
/*const*/ eMaskWhichHandle which_handle,
|
||||
const float scalex,
|
||||
const float scaley,
|
||||
float handle[2])
|
||||
{
|
||||
BKE_mask_point_handle(point, which_handle, handle);
|
||||
handle[0] *= scalex;
|
||||
handle[1] *= scaley;
|
||||
}
|
||||
|
||||
MaskSplinePoint *ED_mask_point_find_nearest(const bContext *C,
|
||||
Mask *mask_orig,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
MaskLayer **r_mask_layer,
|
||||
MaskSpline **r_spline,
|
||||
eMaskWhichHandle *r_which_handle,
|
||||
float *r_score)
|
||||
{
|
||||
ScrArea *sa = CTX_wm_area(C);
|
||||
ARegion *region = CTX_wm_region(C);
|
||||
|
||||
MaskLayer *point_mask_layer = NULL;
|
||||
MaskSpline *point_spline = NULL;
|
||||
MaskSplinePoint *point = NULL;
|
||||
float co[2];
|
||||
const float threshold_sq = threshold * threshold;
|
||||
float len_sq = FLT_MAX, scalex, scaley;
|
||||
eMaskWhichHandle which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
int width, height;
|
||||
|
||||
Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
|
||||
Mask *mask_eval = (Mask *)DEG_get_evaluated_id(depsgraph, &mask_orig->id);
|
||||
|
||||
ED_mask_get_size(sa, &width, &height);
|
||||
ED_mask_pixelspace_factor(sa, region, &scalex, &scaley);
|
||||
|
||||
co[0] = normal_co[0] * scalex;
|
||||
co[1] = normal_co[1] * scaley;
|
||||
|
||||
for (MaskLayer *mask_layer_orig = mask_orig->masklayers.first,
|
||||
*mask_layer_eval = mask_eval->masklayers.first;
|
||||
mask_layer_orig != NULL;
|
||||
mask_layer_orig = mask_layer_orig->next, mask_layer_eval = mask_layer_eval->next) {
|
||||
|
||||
if (mask_layer_orig->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
for (MaskSpline *spline_orig = mask_layer_orig->splines.first,
|
||||
*spline_eval = mask_layer_eval->splines.first;
|
||||
spline_orig != NULL;
|
||||
spline_orig = spline_orig->next, spline_eval = spline_eval->next) {
|
||||
MaskSplinePoint *points_array = BKE_mask_spline_point_array(spline_eval);
|
||||
|
||||
for (int i = 0; i < spline_orig->tot_point; i++) {
|
||||
MaskSplinePoint *cur_point_orig = &spline_orig->points[i];
|
||||
MaskSplinePoint *cur_point_deform_eval = &points_array[i];
|
||||
eMaskWhichHandle cur_which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
BezTriple *bezt = &cur_point_deform_eval->bezt;
|
||||
float cur_len_sq, vec[2];
|
||||
|
||||
vec[0] = bezt->vec[1][0] * scalex;
|
||||
vec[1] = bezt->vec[1][1] * scaley;
|
||||
|
||||
cur_len_sq = len_squared_v2v2(co, vec);
|
||||
|
||||
if (cur_len_sq < len_sq) {
|
||||
point_spline = spline_orig;
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point = cur_point_orig;
|
||||
len_sq = cur_len_sq;
|
||||
which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
}
|
||||
|
||||
if (BKE_mask_point_handles_mode_get(cur_point_deform_eval) == MASK_HANDLE_MODE_STICK) {
|
||||
float handle[2];
|
||||
mask_point_scaled_handle(
|
||||
cur_point_deform_eval, MASK_WHICH_HANDLE_STICK, scalex, scaley, handle);
|
||||
cur_len_sq = len_squared_v2v2(co, handle);
|
||||
cur_which_handle = MASK_WHICH_HANDLE_STICK;
|
||||
}
|
||||
else {
|
||||
float handle_left[2], handle_right[2];
|
||||
float len_left_sq, len_right_sq;
|
||||
mask_point_scaled_handle(
|
||||
cur_point_deform_eval, MASK_WHICH_HANDLE_LEFT, scalex, scaley, handle_left);
|
||||
mask_point_scaled_handle(
|
||||
cur_point_deform_eval, MASK_WHICH_HANDLE_RIGHT, scalex, scaley, handle_right);
|
||||
|
||||
len_left_sq = len_squared_v2v2(co, handle_left);
|
||||
len_right_sq = len_squared_v2v2(co, handle_right);
|
||||
if (i == 0) {
|
||||
if (len_left_sq <= len_right_sq) {
|
||||
if (bezt->h1 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_LEFT;
|
||||
cur_len_sq = len_left_sq;
|
||||
}
|
||||
}
|
||||
else if (bezt->h2 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_RIGHT;
|
||||
cur_len_sq = len_right_sq;
|
||||
}
|
||||
}
|
||||
else {
|
||||
if (len_right_sq <= len_left_sq) {
|
||||
if (bezt->h2 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_RIGHT;
|
||||
cur_len_sq = len_right_sq;
|
||||
}
|
||||
}
|
||||
else if (bezt->h1 != HD_VECT) {
|
||||
cur_which_handle = MASK_WHICH_HANDLE_LEFT;
|
||||
cur_len_sq = len_left_sq;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (cur_len_sq <= len_sq && cur_which_handle != MASK_WHICH_HANDLE_NONE) {
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point_spline = spline_orig;
|
||||
point = cur_point_orig;
|
||||
len_sq = cur_len_sq;
|
||||
which_handle = cur_which_handle;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (len_sq < threshold_sq) {
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = point_mask_layer;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = point_spline;
|
||||
}
|
||||
|
||||
if (r_which_handle) {
|
||||
*r_which_handle = which_handle;
|
||||
}
|
||||
|
||||
if (r_score) {
|
||||
*r_score = sqrtf(len_sq);
|
||||
}
|
||||
|
||||
return point;
|
||||
}
|
||||
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = NULL;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = NULL;
|
||||
}
|
||||
|
||||
if (r_which_handle) {
|
||||
*r_which_handle = MASK_WHICH_HANDLE_NONE;
|
||||
}
|
||||
|
||||
return NULL;
|
||||
}
|
||||
|
||||
bool ED_mask_feather_find_nearest(const bContext *C,
|
||||
Mask *mask_orig,
|
||||
const float normal_co[2],
|
||||
const float threshold,
|
||||
MaskLayer **r_mask_layer,
|
||||
MaskSpline **r_spline,
|
||||
MaskSplinePoint **r_point,
|
||||
MaskSplinePointUW **r_uw,
|
||||
float *r_score)
|
||||
{
|
||||
ScrArea *sa = CTX_wm_area(C);
|
||||
ARegion *region = CTX_wm_region(C);
|
||||
|
||||
MaskLayer *point_mask_layer = NULL;
|
||||
MaskSpline *point_spline = NULL;
|
||||
MaskSplinePoint *point = NULL;
|
||||
MaskSplinePointUW *uw = NULL;
|
||||
const float threshold_sq = threshold * threshold;
|
||||
float len = FLT_MAX, co[2];
|
||||
float scalex, scaley;
|
||||
int width, height;
|
||||
|
||||
Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
|
||||
Mask *mask_eval = (Mask *)DEG_get_evaluated_id(depsgraph, &mask_orig->id);
|
||||
|
||||
ED_mask_get_size(sa, &width, &height);
|
||||
ED_mask_pixelspace_factor(sa, region, &scalex, &scaley);
|
||||
|
||||
co[0] = normal_co[0] * scalex;
|
||||
co[1] = normal_co[1] * scaley;
|
||||
|
||||
for (MaskLayer *mask_layer_orig = mask_orig->masklayers.first,
|
||||
*mask_layer_eval = mask_eval->masklayers.first;
|
||||
mask_layer_orig != NULL;
|
||||
mask_layer_orig = mask_layer_orig->next, mask_layer_eval = mask_layer_eval->next) {
|
||||
|
||||
for (MaskSpline *spline_orig = mask_layer_orig->splines.first,
|
||||
*spline_eval = mask_layer_eval->splines.first;
|
||||
spline_orig != NULL;
|
||||
spline_orig = spline_orig->next, spline_eval = spline_eval->next) {
|
||||
// MaskSplinePoint *points_array = BKE_mask_spline_point_array(spline);
|
||||
|
||||
int i, tot_feather_point;
|
||||
float(*feather_points)[2], (*fp)[2];
|
||||
|
||||
if (mask_layer_orig->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
feather_points = fp = BKE_mask_spline_feather_points(spline_eval, &tot_feather_point);
|
||||
|
||||
for (i = 0; i < spline_orig->tot_point; i++) {
|
||||
int j;
|
||||
MaskSplinePoint *cur_point_orig = &spline_orig->points[i];
|
||||
MaskSplinePoint *cur_point_eval = &spline_eval->points[i];
|
||||
|
||||
for (j = 0; j <= cur_point_eval->tot_uw; j++) {
|
||||
float cur_len_sq, vec[2];
|
||||
|
||||
vec[0] = (*fp)[0] * scalex;
|
||||
vec[1] = (*fp)[1] * scaley;
|
||||
|
||||
cur_len_sq = len_squared_v2v2(vec, co);
|
||||
|
||||
if (point == NULL || cur_len_sq < len) {
|
||||
if (j == 0) {
|
||||
uw = NULL;
|
||||
}
|
||||
else {
|
||||
uw = &cur_point_orig->uw[j - 1];
|
||||
}
|
||||
|
||||
point_mask_layer = mask_layer_orig;
|
||||
point_spline = spline_orig;
|
||||
point = cur_point_orig;
|
||||
len = cur_len_sq;
|
||||
}
|
||||
|
||||
fp++;
|
||||
}
|
||||
}
|
||||
|
||||
MEM_freeN(feather_points);
|
||||
}
|
||||
}
|
||||
|
||||
if (len < threshold_sq) {
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = point_mask_layer;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = point_spline;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = point;
|
||||
}
|
||||
|
||||
if (r_uw) {
|
||||
*r_uw = uw;
|
||||
}
|
||||
|
||||
if (r_score) {
|
||||
*r_score = sqrtf(len);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
if (r_mask_layer) {
|
||||
*r_mask_layer = NULL;
|
||||
}
|
||||
|
||||
if (r_spline) {
|
||||
*r_spline = NULL;
|
||||
}
|
||||
|
||||
if (r_point) {
|
||||
*r_point = NULL;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/* takes event->mval */
|
||||
void ED_mask_mouse_pos(ScrArea *sa, ARegion *region, const int mval[2], float co[2])
|
||||
{
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_clip_mouse_pos(sc, region, mval, co);
|
||||
BKE_mask_coord_from_movieclip(sc->clip, &sc->user, co, co);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
UI_view2d_region_to_view(®ion->v2d, mval[0], mval[1], &co[0], &co[1]);
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_image_mouse_pos(sima, region, mval, co);
|
||||
BKE_mask_coord_from_image(sima->image, &sima->iuser, co, co);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
}
|
||||
}
|
||||
|
||||
/* input: x/y - mval space
|
||||
* output: xr/yr - mask point space */
|
||||
void ED_mask_point_pos(ScrArea *sa, ARegion *region, float x, float y, float *xr, float *yr)
|
||||
{
|
||||
float co[2];
|
||||
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_clip_point_stable_pos(sc, region, x, y, &co[0], &co[1]);
|
||||
BKE_mask_coord_from_movieclip(sc->clip, &sc->user, co, co);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ:
|
||||
zero_v2(co); /* MASKTODO */
|
||||
break;
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_image_point_pos(sima, region, x, y, &co[0], &co[1]);
|
||||
BKE_mask_coord_from_image(sima->image, &sima->iuser, co, co);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
}
|
||||
|
||||
*xr = co[0];
|
||||
*yr = co[1];
|
||||
}
|
||||
|
||||
void ED_mask_point_pos__reverse(
|
||||
ScrArea *sa, ARegion *region, float x, float y, float *xr, float *yr)
|
||||
{
|
||||
float co[2];
|
||||
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
co[0] = x;
|
||||
co[1] = y;
|
||||
BKE_mask_coord_to_movieclip(sc->clip, &sc->user, co, co);
|
||||
ED_clip_point_stable_pos__reverse(sc, region, co, co);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ:
|
||||
zero_v2(co); /* MASKTODO */
|
||||
break;
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
co[0] = x;
|
||||
co[1] = y;
|
||||
BKE_mask_coord_to_image(sima->image, &sima->iuser, co, co);
|
||||
ED_image_point_pos__reverse(sima, region, co, co);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(co);
|
||||
}
|
||||
|
||||
*xr = co[0];
|
||||
*yr = co[1];
|
||||
}
|
||||
|
||||
bool ED_mask_selected_minmax(const bContext *C, float min[2], float max[2])
|
||||
{
|
||||
Mask *mask = CTX_data_edit_mask(C);
|
||||
bool ok = false;
|
||||
|
||||
if (mask == NULL) {
|
||||
return ok;
|
||||
}
|
||||
|
||||
INIT_MINMAX2(min, max);
|
||||
for (MaskLayer *mask_layer = mask->masklayers.first; mask_layer != NULL;
|
||||
mask_layer = mask_layer->next) {
|
||||
if (mask_layer->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
|
||||
continue;
|
||||
}
|
||||
for (MaskSpline *spline = mask_layer->splines.first; spline != NULL; spline = spline->next) {
|
||||
MaskSplinePoint *points_array = BKE_mask_spline_point_array(spline);
|
||||
for (int i = 0; i < spline->tot_point; i++) {
|
||||
MaskSplinePoint *point = &spline->points[i];
|
||||
MaskSplinePoint *deform_point = &points_array[i];
|
||||
BezTriple *bezt = &point->bezt;
|
||||
float handle[2];
|
||||
if (!MASKPOINT_ISSEL_ANY(point)) {
|
||||
continue;
|
||||
}
|
||||
if (bezt->f2 & SELECT) {
|
||||
minmax_v2v2_v2(min, max, deform_point->bezt.vec[1]);
|
||||
}
|
||||
if (BKE_mask_point_handles_mode_get(point) == MASK_HANDLE_MODE_STICK) {
|
||||
BKE_mask_point_handle(deform_point, MASK_WHICH_HANDLE_STICK, handle);
|
||||
minmax_v2v2_v2(min, max, handle);
|
||||
}
|
||||
else {
|
||||
if ((bezt->f1 & SELECT) && (bezt->h1 != HD_VECT)) {
|
||||
BKE_mask_point_handle(deform_point, MASK_WHICH_HANDLE_LEFT, handle);
|
||||
minmax_v2v2_v2(min, max, handle);
|
||||
}
|
||||
if ((bezt->f3 & SELECT) && (bezt->h2 != HD_VECT)) {
|
||||
BKE_mask_point_handle(deform_point, MASK_WHICH_HANDLE_RIGHT, handle);
|
||||
minmax_v2v2_v2(min, max, handle);
|
||||
}
|
||||
}
|
||||
ok = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
return ok;
|
||||
}
|
||||
|
||||
/** \} */
|
||||
|
||||
/* -------------------------------------------------------------------- */
|
||||
/** \name Generic 2D View Queries
|
||||
* \{ */
|
||||
|
||||
void ED_mask_get_size(ScrArea *sa, int *width, int *height)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_space_clip_get_size(sc, width, height);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
// Scene *scene = CTX_data_scene(C);
|
||||
// *width = (scene->r.size * scene->r.xsch) / 100;
|
||||
// *height = (scene->r.size * scene->r.ysch) / 100;
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_space_image_get_size(sima, width, height);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*width = 0;
|
||||
*height = 0;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*width = 0;
|
||||
*height = 0;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_zoom(ScrArea *sa, ARegion *region, float *zoomx, float *zoomy)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_space_clip_get_zoom(sc, region, zoomx, zoomy);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
*zoomx = *zoomy = 1.0f;
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_space_image_get_zoom(sima, region, zoomx, zoomy);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*zoomx = *zoomy = 1.0f;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*zoomx = *zoomy = 1.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_get_aspect(ScrArea *sa, ARegion *UNUSED(region), float *aspx, float *aspy)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
ED_space_clip_get_aspect(sc, aspx, aspy);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
*aspx = *aspy = 1.0f; /* MASKTODO - render aspect? */
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
ED_space_image_get_aspect(sima, aspx, aspy);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*aspx = *aspy = 1.0f;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*aspx = *aspy = 1.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_pixelspace_factor(ScrArea *sa, ARegion *region, float *scalex, float *scaley)
|
||||
{
|
||||
if (sa && sa->spacedata.first) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *sc = sa->spacedata.first;
|
||||
float aspx, aspy;
|
||||
|
||||
UI_view2d_scale_get(®ion->v2d, scalex, scaley);
|
||||
ED_space_clip_get_aspect(sc, &aspx, &aspy);
|
||||
|
||||
*scalex *= aspx;
|
||||
*scaley *= aspy;
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
*scalex = *scaley = 1.0f; /* MASKTODO? */
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *sima = sa->spacedata.first;
|
||||
float aspx, aspy;
|
||||
|
||||
UI_view2d_scale_get(®ion->v2d, scalex, scaley);
|
||||
ED_space_image_get_aspect(sima, &aspx, &aspy);
|
||||
|
||||
*scalex *= aspx;
|
||||
*scaley *= aspy;
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
*scalex = *scaley = 1.0f;
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
*scalex = *scaley = 1.0f;
|
||||
}
|
||||
}
|
||||
|
||||
void ED_mask_cursor_location_get(ScrArea *sa, float cursor[2])
|
||||
{
|
||||
if (sa) {
|
||||
switch (sa->spacetype) {
|
||||
case SPACE_CLIP: {
|
||||
SpaceClip *space_clip = sa->spacedata.first;
|
||||
copy_v2_v2(cursor, space_clip->cursor);
|
||||
break;
|
||||
}
|
||||
case SPACE_SEQ: {
|
||||
zero_v2(cursor);
|
||||
break;
|
||||
}
|
||||
case SPACE_IMAGE: {
|
||||
SpaceImage *space_image = sa->spacedata.first;
|
||||
copy_v2_v2(cursor, space_image->cursor);
|
||||
break;
|
||||
}
|
||||
default:
|
||||
/* possible other spaces from which mask editing is available */
|
||||
BLI_assert(0);
|
||||
zero_v2(cursor);
|
||||
break;
|
||||
}
|
||||
}
|
||||
else {
|
||||
BLI_assert(0);
|
||||
zero_v2(cursor);
|
||||
}
|
||||
}
|
||||
|
||||
/** \} */
|
||||
Reference in New Issue
Block a user