But in the future the selection code may also be used in object mode (eg for snapping). So to avoid using too much VRAM resources, it is good to avoid drawing all objects in the viewport. The solution was to create an array with only objects that are detected within the selection area. If the selection operator is modal, objects already detected are not removed from the array until view3d is moved or orbited. To detect the object, its BoundBox is tested. Since the Select Engine does not have a dedicated depth texture, whenever a new object is "found" the depth of the objects in the array already drawn is redrawn. Reviewers: campbellbarton, fclem Reviewed By: fclem Differential Revision: https://developer.blender.org/D5435
817 lines
23 KiB
C
817 lines
23 KiB
C
/*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
|
|
* All rights reserved.
|
|
*/
|
|
|
|
/** \file
|
|
* \ingroup edsculpt
|
|
*/
|
|
|
|
#include <math.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "DNA_mesh_types.h"
|
|
#include "DNA_meshdata_types.h"
|
|
#include "DNA_object_types.h"
|
|
#include "DNA_material_types.h"
|
|
|
|
#include "DNA_scene_types.h"
|
|
#include "DNA_brush_types.h"
|
|
|
|
#include "BLI_math.h"
|
|
#include "BLI_math_color.h"
|
|
#include "BLI_utildefines.h"
|
|
#include "BLI_listbase.h"
|
|
#include "BLI_rect.h"
|
|
|
|
#include "BLT_translation.h"
|
|
|
|
#include "BKE_brush.h"
|
|
#include "BKE_context.h"
|
|
#include "BKE_customdata.h"
|
|
#include "BKE_image.h"
|
|
#include "BKE_material.h"
|
|
#include "BKE_mesh_runtime.h"
|
|
#include "BKE_object.h"
|
|
#include "BKE_paint.h"
|
|
#include "BKE_report.h"
|
|
|
|
#include "DEG_depsgraph.h"
|
|
#include "DEG_depsgraph_query.h"
|
|
|
|
#include "RNA_access.h"
|
|
#include "RNA_define.h"
|
|
|
|
#include "GPU_glew.h"
|
|
#include "GPU_matrix.h"
|
|
#include "GPU_state.h"
|
|
|
|
#include "IMB_colormanagement.h"
|
|
#include "IMB_imbuf_types.h"
|
|
#include "IMB_imbuf.h"
|
|
|
|
#include "RE_render_ext.h"
|
|
|
|
#include "ED_view3d.h"
|
|
#include "ED_screen.h"
|
|
|
|
#include "BLI_sys_types.h"
|
|
#include "ED_mesh.h" /* for face mask functions */
|
|
|
|
#include "DRW_select_buffer.h"
|
|
|
|
#include "WM_api.h"
|
|
#include "WM_types.h"
|
|
|
|
#include "paint_intern.h"
|
|
|
|
/* Convert the object-space axis-aligned bounding box (expressed as
|
|
* its minimum and maximum corners) into a screen-space rectangle,
|
|
* returns zero if the result is empty */
|
|
bool paint_convert_bb_to_rect(rcti *rect,
|
|
const float bb_min[3],
|
|
const float bb_max[3],
|
|
const ARegion *ar,
|
|
RegionView3D *rv3d,
|
|
Object *ob)
|
|
{
|
|
float projection_mat[4][4];
|
|
int i, j, k;
|
|
|
|
BLI_rcti_init_minmax(rect);
|
|
|
|
/* return zero if the bounding box has non-positive volume */
|
|
if (bb_min[0] > bb_max[0] || bb_min[1] > bb_max[1] || bb_min[2] > bb_max[2]) {
|
|
return 0;
|
|
}
|
|
|
|
ED_view3d_ob_project_mat_get(rv3d, ob, projection_mat);
|
|
|
|
for (i = 0; i < 2; ++i) {
|
|
for (j = 0; j < 2; ++j) {
|
|
for (k = 0; k < 2; ++k) {
|
|
float vec[3], proj[2];
|
|
int proj_i[2];
|
|
vec[0] = i ? bb_min[0] : bb_max[0];
|
|
vec[1] = j ? bb_min[1] : bb_max[1];
|
|
vec[2] = k ? bb_min[2] : bb_max[2];
|
|
/* convert corner to screen space */
|
|
ED_view3d_project_float_v2_m4(ar, vec, proj, projection_mat);
|
|
/* expand 2D rectangle */
|
|
|
|
/* we could project directly to int? */
|
|
proj_i[0] = proj[0];
|
|
proj_i[1] = proj[1];
|
|
|
|
BLI_rcti_do_minmax_v(rect, proj_i);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* return false if the rectangle has non-positive area */
|
|
return rect->xmin < rect->xmax && rect->ymin < rect->ymax;
|
|
}
|
|
|
|
/* Get four planes in object-space that describe the projection of
|
|
* screen_rect from screen into object-space (essentially converting a
|
|
* 2D screens-space bounding box into four 3D planes) */
|
|
void paint_calc_redraw_planes(float planes[4][4],
|
|
const ARegion *ar,
|
|
Object *ob,
|
|
const rcti *screen_rect)
|
|
{
|
|
BoundBox bb;
|
|
rcti rect;
|
|
|
|
/* use some extra space just in case */
|
|
rect = *screen_rect;
|
|
rect.xmin -= 2;
|
|
rect.xmax += 2;
|
|
rect.ymin -= 2;
|
|
rect.ymax += 2;
|
|
|
|
ED_view3d_clipping_calc(&bb, planes, ar, ob, &rect);
|
|
negate_m4(planes);
|
|
}
|
|
|
|
float paint_calc_object_space_radius(ViewContext *vc, const float center[3], float pixel_radius)
|
|
{
|
|
Object *ob = vc->obact;
|
|
float delta[3], scale, loc[3];
|
|
const float mval_f[2] = {pixel_radius, 0.0f};
|
|
float zfac;
|
|
|
|
mul_v3_m4v3(loc, ob->obmat, center);
|
|
|
|
zfac = ED_view3d_calc_zfac(vc->rv3d, loc, NULL);
|
|
ED_view3d_win_to_delta(vc->ar, mval_f, delta, zfac);
|
|
|
|
scale = fabsf(mat4_to_scale(ob->obmat));
|
|
scale = (scale == 0.0f) ? 1.0f : scale;
|
|
|
|
return len_v3(delta) / scale;
|
|
}
|
|
|
|
float paint_get_tex_pixel(const MTex *mtex, float u, float v, struct ImagePool *pool, int thread)
|
|
{
|
|
float intensity, rgba[4];
|
|
float co[3] = {u, v, 0.0f};
|
|
|
|
externtex(mtex, co, &intensity, rgba, rgba + 1, rgba + 2, rgba + 3, thread, pool, false, false);
|
|
|
|
return intensity;
|
|
}
|
|
|
|
void paint_get_tex_pixel_col(const MTex *mtex,
|
|
float u,
|
|
float v,
|
|
float rgba[4],
|
|
struct ImagePool *pool,
|
|
int thread,
|
|
bool convert_to_linear,
|
|
struct ColorSpace *colorspace)
|
|
{
|
|
float co[3] = {u, v, 0.0f};
|
|
int hasrgb;
|
|
float intensity;
|
|
|
|
hasrgb = externtex(
|
|
mtex, co, &intensity, rgba, rgba + 1, rgba + 2, rgba + 3, thread, pool, false, false);
|
|
if (!hasrgb) {
|
|
rgba[0] = intensity;
|
|
rgba[1] = intensity;
|
|
rgba[2] = intensity;
|
|
rgba[3] = 1.0f;
|
|
}
|
|
|
|
if (convert_to_linear) {
|
|
IMB_colormanagement_colorspace_to_scene_linear_v3(rgba, colorspace);
|
|
}
|
|
|
|
linearrgb_to_srgb_v3_v3(rgba, rgba);
|
|
|
|
CLAMP(rgba[0], 0.0f, 1.0f);
|
|
CLAMP(rgba[1], 0.0f, 1.0f);
|
|
CLAMP(rgba[2], 0.0f, 1.0f);
|
|
CLAMP(rgba[3], 0.0f, 1.0f);
|
|
}
|
|
|
|
void paint_stroke_operator_properties(wmOperatorType *ot)
|
|
{
|
|
static const EnumPropertyItem stroke_mode_items[] = {
|
|
{BRUSH_STROKE_NORMAL, "NORMAL", 0, "Regular", "Apply brush normally"},
|
|
{BRUSH_STROKE_INVERT,
|
|
"INVERT",
|
|
0,
|
|
"Invert",
|
|
"Invert action of brush for duration of stroke"},
|
|
{BRUSH_STROKE_SMOOTH,
|
|
"SMOOTH",
|
|
0,
|
|
"Smooth",
|
|
"Switch brush to smooth mode for duration of stroke"},
|
|
{0},
|
|
};
|
|
|
|
PropertyRNA *prop;
|
|
|
|
prop = RNA_def_collection_runtime(ot->srna, "stroke", &RNA_OperatorStrokeElement, "Stroke", "");
|
|
RNA_def_property_flag(prop, PROP_HIDDEN | PROP_SKIP_SAVE);
|
|
|
|
RNA_def_enum(ot->srna,
|
|
"mode",
|
|
stroke_mode_items,
|
|
BRUSH_STROKE_NORMAL,
|
|
"Stroke Mode",
|
|
"Action taken when a paint stroke is made");
|
|
}
|
|
|
|
/* 3D Paint */
|
|
|
|
static void imapaint_project(float matrix[4][4], const float co[3], float pco[4])
|
|
{
|
|
copy_v3_v3(pco, co);
|
|
pco[3] = 1.0f;
|
|
|
|
mul_m4_v4(matrix, pco);
|
|
}
|
|
|
|
static void imapaint_tri_weights(float matrix[4][4],
|
|
GLint view[4],
|
|
const float v1[3],
|
|
const float v2[3],
|
|
const float v3[3],
|
|
const float co[2],
|
|
float w[3])
|
|
{
|
|
float pv1[4], pv2[4], pv3[4], h[3], divw;
|
|
float wmat[3][3], invwmat[3][3];
|
|
|
|
/* compute barycentric coordinates */
|
|
|
|
/* project the verts */
|
|
imapaint_project(matrix, v1, pv1);
|
|
imapaint_project(matrix, v2, pv2);
|
|
imapaint_project(matrix, v3, pv3);
|
|
|
|
/* do inverse view mapping, see gluProject man page */
|
|
h[0] = (co[0] - view[0]) * 2.0f / view[2] - 1.0f;
|
|
h[1] = (co[1] - view[1]) * 2.0f / view[3] - 1.0f;
|
|
h[2] = 1.0f;
|
|
|
|
/* solve for (w1,w2,w3)/perspdiv in:
|
|
* h * perspdiv = Project * Model * (w1 * v1 + w2 * v2 + w3 * v3) */
|
|
|
|
wmat[0][0] = pv1[0];
|
|
wmat[1][0] = pv2[0];
|
|
wmat[2][0] = pv3[0];
|
|
wmat[0][1] = pv1[1];
|
|
wmat[1][1] = pv2[1];
|
|
wmat[2][1] = pv3[1];
|
|
wmat[0][2] = pv1[3];
|
|
wmat[1][2] = pv2[3];
|
|
wmat[2][2] = pv3[3];
|
|
|
|
invert_m3_m3(invwmat, wmat);
|
|
mul_m3_v3(invwmat, h);
|
|
|
|
copy_v3_v3(w, h);
|
|
|
|
/* w is still divided by perspdiv, make it sum to one */
|
|
divw = w[0] + w[1] + w[2];
|
|
if (divw != 0.0f) {
|
|
mul_v3_fl(w, 1.0f / divw);
|
|
}
|
|
}
|
|
|
|
/* compute uv coordinates of mouse in face */
|
|
static void imapaint_pick_uv(Mesh *me_eval,
|
|
Scene *scene,
|
|
Object *ob_eval,
|
|
unsigned int faceindex,
|
|
const int xy[2],
|
|
float uv[2])
|
|
{
|
|
int i, findex;
|
|
float p[2], w[3], absw, minabsw;
|
|
float matrix[4][4], proj[4][4];
|
|
GLint view[4];
|
|
const eImagePaintMode mode = scene->toolsettings->imapaint.mode;
|
|
|
|
const MLoopTri *lt = BKE_mesh_runtime_looptri_ensure(me_eval);
|
|
const int tottri = me_eval->runtime.looptris.len;
|
|
|
|
const MVert *mvert = me_eval->mvert;
|
|
const MPoly *mpoly = me_eval->mpoly;
|
|
const MLoop *mloop = me_eval->mloop;
|
|
const int *index_mp_to_orig = CustomData_get_layer(&me_eval->pdata, CD_ORIGINDEX);
|
|
|
|
/* get the needed opengl matrices */
|
|
GPU_viewport_size_get_i(view);
|
|
GPU_matrix_model_view_get(matrix);
|
|
GPU_matrix_projection_get(proj);
|
|
view[0] = view[1] = 0;
|
|
mul_m4_m4m4(matrix, matrix, ob_eval->obmat);
|
|
mul_m4_m4m4(matrix, proj, matrix);
|
|
|
|
minabsw = 1e10;
|
|
uv[0] = uv[1] = 0.0;
|
|
|
|
/* test all faces in the derivedmesh with the original index of the picked face */
|
|
/* face means poly here, not triangle, indeed */
|
|
for (i = 0; i < tottri; i++, lt++) {
|
|
findex = index_mp_to_orig ? index_mp_to_orig[lt->poly] : lt->poly;
|
|
|
|
if (findex == faceindex) {
|
|
const MLoopUV *mloopuv;
|
|
const MPoly *mp = &mpoly[lt->poly];
|
|
const MLoopUV *tri_uv[3];
|
|
float tri_co[3][3];
|
|
|
|
for (int j = 3; j--;) {
|
|
copy_v3_v3(tri_co[j], mvert[mloop[lt->tri[j]].v].co);
|
|
}
|
|
|
|
if (mode == IMAGEPAINT_MODE_MATERIAL) {
|
|
const Material *ma;
|
|
const TexPaintSlot *slot;
|
|
|
|
ma = give_current_material(ob_eval, mp->mat_nr + 1);
|
|
slot = &ma->texpaintslot[ma->paint_active_slot];
|
|
|
|
if (!(slot && slot->uvname &&
|
|
(mloopuv = CustomData_get_layer_named(&me_eval->ldata, CD_MLOOPUV, slot->uvname)))) {
|
|
mloopuv = CustomData_get_layer(&me_eval->ldata, CD_MLOOPUV);
|
|
}
|
|
}
|
|
else {
|
|
mloopuv = CustomData_get_layer(&me_eval->ldata, CD_MLOOPUV);
|
|
}
|
|
|
|
tri_uv[0] = &mloopuv[lt->tri[0]];
|
|
tri_uv[1] = &mloopuv[lt->tri[1]];
|
|
tri_uv[2] = &mloopuv[lt->tri[2]];
|
|
|
|
p[0] = xy[0];
|
|
p[1] = xy[1];
|
|
|
|
imapaint_tri_weights(matrix, view, UNPACK3(tri_co), p, w);
|
|
absw = fabsf(w[0]) + fabsf(w[1]) + fabsf(w[2]);
|
|
if (absw < minabsw) {
|
|
uv[0] = tri_uv[0]->uv[0] * w[0] + tri_uv[1]->uv[0] * w[1] + tri_uv[2]->uv[0] * w[2];
|
|
uv[1] = tri_uv[0]->uv[1] * w[0] + tri_uv[1]->uv[1] * w[1] + tri_uv[2]->uv[1] * w[2];
|
|
minabsw = absw;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* returns 0 if not found, otherwise 1 */
|
|
static int imapaint_pick_face(ViewContext *vc,
|
|
const int mval[2],
|
|
unsigned int *r_index,
|
|
unsigned int totpoly)
|
|
{
|
|
if (totpoly == 0) {
|
|
return 0;
|
|
}
|
|
|
|
/* sample only on the exact position */
|
|
ED_view3d_select_id_validate(vc);
|
|
*r_index = DRW_select_buffer_sample_point(vc->depsgraph, vc->ar, vc->v3d, mval);
|
|
|
|
if ((*r_index) == 0 || (*r_index) > (unsigned int)totpoly) {
|
|
return 0;
|
|
}
|
|
|
|
(*r_index)--;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static Image *imapaint_face_image(Object *ob, Mesh *me, int face_index)
|
|
{
|
|
Image *ima;
|
|
MPoly *mp = me->mpoly + face_index;
|
|
Material *ma = give_current_material(ob, mp->mat_nr + 1);
|
|
ima = ma && ma->texpaintslot ? ma->texpaintslot[ma->paint_active_slot].ima : NULL;
|
|
|
|
return ima;
|
|
}
|
|
|
|
/* Uses symm to selectively flip any axis of a coordinate. */
|
|
void flip_v3_v3(float out[3], const float in[3], const char symm)
|
|
{
|
|
if (symm & PAINT_SYMM_X) {
|
|
out[0] = -in[0];
|
|
}
|
|
else {
|
|
out[0] = in[0];
|
|
}
|
|
if (symm & PAINT_SYMM_Y) {
|
|
out[1] = -in[1];
|
|
}
|
|
else {
|
|
out[1] = in[1];
|
|
}
|
|
if (symm & PAINT_SYMM_Z) {
|
|
out[2] = -in[2];
|
|
}
|
|
else {
|
|
out[2] = in[2];
|
|
}
|
|
}
|
|
|
|
void flip_qt_qt(float out[4], const float in[4], const char symm)
|
|
{
|
|
float axis[3], angle;
|
|
|
|
quat_to_axis_angle(axis, &angle, in);
|
|
normalize_v3(axis);
|
|
|
|
if (symm & PAINT_SYMM_X) {
|
|
axis[0] *= -1.0f;
|
|
angle *= -1.0f;
|
|
}
|
|
if (symm & PAINT_SYMM_Y) {
|
|
axis[1] *= -1.0f;
|
|
angle *= -1.0f;
|
|
}
|
|
if (symm & PAINT_SYMM_Z) {
|
|
axis[2] *= -1.0f;
|
|
angle *= -1.0f;
|
|
}
|
|
|
|
axis_angle_normalized_to_quat(out, axis, angle);
|
|
}
|
|
|
|
/* used for both 3d view and image window */
|
|
void paint_sample_color(
|
|
bContext *C, ARegion *ar, int x, int y, bool texpaint_proj, bool use_palette)
|
|
{
|
|
Scene *scene = CTX_data_scene(C);
|
|
Depsgraph *depsgraph = CTX_data_ensure_evaluated_depsgraph(C);
|
|
Paint *paint = BKE_paint_get_active_from_context(C);
|
|
Palette *palette = BKE_paint_palette(paint);
|
|
PaletteColor *color = NULL;
|
|
Brush *br = BKE_paint_brush(BKE_paint_get_active_from_context(C));
|
|
unsigned int col;
|
|
const unsigned char *cp;
|
|
|
|
CLAMP(x, 0, ar->winx);
|
|
CLAMP(y, 0, ar->winy);
|
|
|
|
if (use_palette) {
|
|
if (!palette) {
|
|
palette = BKE_palette_add(CTX_data_main(C), "Palette");
|
|
BKE_paint_palette_set(paint, palette);
|
|
}
|
|
|
|
color = BKE_palette_color_add(palette);
|
|
palette->active_color = BLI_listbase_count(&palette->colors) - 1;
|
|
}
|
|
|
|
if (CTX_wm_view3d(C) && texpaint_proj) {
|
|
/* first try getting a color directly from the mesh faces if possible */
|
|
ViewLayer *view_layer = CTX_data_view_layer(C);
|
|
Object *ob = OBACT(view_layer);
|
|
Object *ob_eval = DEG_get_evaluated_object(depsgraph, ob);
|
|
bool sample_success = false;
|
|
ImagePaintSettings *imapaint = &scene->toolsettings->imapaint;
|
|
bool use_material = (imapaint->mode == IMAGEPAINT_MODE_MATERIAL);
|
|
|
|
if (ob) {
|
|
CustomData_MeshMasks cddata_masks = CD_MASK_BAREMESH;
|
|
cddata_masks.pmask |= CD_MASK_ORIGINDEX;
|
|
Mesh *me = (Mesh *)ob->data;
|
|
Mesh *me_eval = mesh_get_eval_final(depsgraph, scene, ob_eval, &cddata_masks);
|
|
|
|
ViewContext vc;
|
|
const int mval[2] = {x, y};
|
|
unsigned int faceindex;
|
|
unsigned int totpoly = me->totpoly;
|
|
|
|
if (CustomData_has_layer(&me_eval->ldata, CD_MLOOPUV)) {
|
|
ED_view3d_viewcontext_init(C, &vc);
|
|
|
|
view3d_operator_needs_opengl(C);
|
|
|
|
if (imapaint_pick_face(&vc, mval, &faceindex, totpoly)) {
|
|
Image *image;
|
|
|
|
if (use_material) {
|
|
image = imapaint_face_image(ob_eval, me_eval, faceindex);
|
|
}
|
|
else {
|
|
image = imapaint->canvas;
|
|
}
|
|
|
|
if (image) {
|
|
ImBuf *ibuf = BKE_image_acquire_ibuf(image, NULL, NULL);
|
|
if (ibuf && ibuf->rect) {
|
|
float uv[2];
|
|
float u, v;
|
|
imapaint_pick_uv(me_eval, scene, ob_eval, faceindex, mval, uv);
|
|
sample_success = true;
|
|
|
|
u = fmodf(uv[0], 1.0f);
|
|
v = fmodf(uv[1], 1.0f);
|
|
|
|
if (u < 0.0f) {
|
|
u += 1.0f;
|
|
}
|
|
if (v < 0.0f) {
|
|
v += 1.0f;
|
|
}
|
|
|
|
u = u * ibuf->x;
|
|
v = v * ibuf->y;
|
|
|
|
if (ibuf->rect_float) {
|
|
float rgba_f[4];
|
|
bilinear_interpolation_color_wrap(ibuf, NULL, rgba_f, u, v);
|
|
straight_to_premul_v4(rgba_f);
|
|
if (use_palette) {
|
|
linearrgb_to_srgb_v3_v3(color->rgb, rgba_f);
|
|
}
|
|
else {
|
|
linearrgb_to_srgb_v3_v3(rgba_f, rgba_f);
|
|
BKE_brush_color_set(scene, br, rgba_f);
|
|
}
|
|
}
|
|
else {
|
|
unsigned char rgba[4];
|
|
bilinear_interpolation_color_wrap(ibuf, rgba, NULL, u, v);
|
|
if (use_palette) {
|
|
rgb_uchar_to_float(color->rgb, rgba);
|
|
}
|
|
else {
|
|
float rgba_f[3];
|
|
rgb_uchar_to_float(rgba_f, rgba);
|
|
BKE_brush_color_set(scene, br, rgba_f);
|
|
}
|
|
}
|
|
}
|
|
|
|
BKE_image_release_ibuf(image, ibuf, NULL);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!sample_success) {
|
|
glReadBuffer(GL_FRONT);
|
|
glReadPixels(
|
|
x + ar->winrct.xmin, y + ar->winrct.ymin, 1, 1, GL_RGBA, GL_UNSIGNED_BYTE, &col);
|
|
glReadBuffer(GL_BACK);
|
|
}
|
|
else {
|
|
return;
|
|
}
|
|
}
|
|
else {
|
|
glReadBuffer(GL_FRONT);
|
|
glReadPixels(x + ar->winrct.xmin, y + ar->winrct.ymin, 1, 1, GL_RGBA, GL_UNSIGNED_BYTE, &col);
|
|
glReadBuffer(GL_BACK);
|
|
}
|
|
cp = (unsigned char *)&col;
|
|
|
|
if (use_palette) {
|
|
rgb_uchar_to_float(color->rgb, cp);
|
|
}
|
|
else {
|
|
float rgba_f[3];
|
|
rgb_uchar_to_float(rgba_f, cp);
|
|
BKE_brush_color_set(scene, br, rgba_f);
|
|
}
|
|
}
|
|
|
|
static int brush_curve_preset_exec(bContext *C, wmOperator *op)
|
|
{
|
|
Brush *br = BKE_paint_brush(BKE_paint_get_active_from_context(C));
|
|
|
|
if (br) {
|
|
Scene *scene = CTX_data_scene(C);
|
|
ViewLayer *view_layer = CTX_data_view_layer(C);
|
|
BKE_brush_curve_preset(br, RNA_enum_get(op->ptr, "shape"));
|
|
BKE_paint_invalidate_cursor_overlay(scene, view_layer, br->curve);
|
|
}
|
|
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
static bool brush_curve_preset_poll(bContext *C)
|
|
{
|
|
Brush *br = BKE_paint_brush(BKE_paint_get_active_from_context(C));
|
|
|
|
return br && br->curve;
|
|
}
|
|
|
|
void BRUSH_OT_curve_preset(wmOperatorType *ot)
|
|
{
|
|
PropertyRNA *prop;
|
|
static const EnumPropertyItem prop_shape_items[] = {
|
|
{CURVE_PRESET_SHARP, "SHARP", 0, "Sharp", ""},
|
|
{CURVE_PRESET_SMOOTH, "SMOOTH", 0, "Smooth", ""},
|
|
{CURVE_PRESET_MAX, "MAX", 0, "Max", ""},
|
|
{CURVE_PRESET_LINE, "LINE", 0, "Line", ""},
|
|
{CURVE_PRESET_ROUND, "ROUND", 0, "Round", ""},
|
|
{CURVE_PRESET_ROOT, "ROOT", 0, "Root", ""},
|
|
{0, NULL, 0, NULL, NULL},
|
|
};
|
|
|
|
ot->name = "Preset";
|
|
ot->description = "Set brush shape";
|
|
ot->idname = "BRUSH_OT_curve_preset";
|
|
|
|
ot->exec = brush_curve_preset_exec;
|
|
ot->poll = brush_curve_preset_poll;
|
|
|
|
prop = RNA_def_enum(ot->srna, "shape", prop_shape_items, CURVE_PRESET_SMOOTH, "Mode", "");
|
|
RNA_def_property_translation_context(prop, BLT_I18NCONTEXT_ID_CURVE); /* Abusing id_curve :/ */
|
|
}
|
|
|
|
/* face-select ops */
|
|
static int paint_select_linked_exec(bContext *C, wmOperator *UNUSED(op))
|
|
{
|
|
paintface_select_linked(C, CTX_data_active_object(C), NULL, true);
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
void PAINT_OT_face_select_linked(wmOperatorType *ot)
|
|
{
|
|
ot->name = "Select Linked";
|
|
ot->description = "Select linked faces";
|
|
ot->idname = "PAINT_OT_face_select_linked";
|
|
|
|
ot->exec = paint_select_linked_exec;
|
|
ot->poll = facemask_paint_poll;
|
|
|
|
ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
|
|
}
|
|
|
|
static int paint_select_linked_pick_invoke(bContext *C, wmOperator *op, const wmEvent *event)
|
|
{
|
|
const bool select = !RNA_boolean_get(op->ptr, "deselect");
|
|
view3d_operator_needs_opengl(C);
|
|
paintface_select_linked(C, CTX_data_active_object(C), event->mval, select);
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
void PAINT_OT_face_select_linked_pick(wmOperatorType *ot)
|
|
{
|
|
ot->name = "Select Linked Pick";
|
|
ot->description = "Select linked faces under the cursor";
|
|
ot->idname = "PAINT_OT_face_select_linked_pick";
|
|
|
|
ot->invoke = paint_select_linked_pick_invoke;
|
|
ot->poll = facemask_paint_poll;
|
|
|
|
ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
|
|
|
|
RNA_def_boolean(ot->srna, "deselect", 0, "Deselect", "Deselect rather than select items");
|
|
}
|
|
|
|
static int face_select_all_exec(bContext *C, wmOperator *op)
|
|
{
|
|
Object *ob = CTX_data_active_object(C);
|
|
if (paintface_deselect_all_visible(C, ob, RNA_enum_get(op->ptr, "action"), true)) {
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
return OPERATOR_CANCELLED;
|
|
}
|
|
|
|
void PAINT_OT_face_select_all(wmOperatorType *ot)
|
|
{
|
|
ot->name = "(De)select All";
|
|
ot->description = "Change selection for all faces";
|
|
ot->idname = "PAINT_OT_face_select_all";
|
|
|
|
ot->exec = face_select_all_exec;
|
|
ot->poll = facemask_paint_poll;
|
|
|
|
ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
|
|
|
|
WM_operator_properties_select_all(ot);
|
|
}
|
|
|
|
static int vert_select_all_exec(bContext *C, wmOperator *op)
|
|
{
|
|
Object *ob = CTX_data_active_object(C);
|
|
paintvert_deselect_all_visible(ob, RNA_enum_get(op->ptr, "action"), true);
|
|
paintvert_tag_select_update(C, ob);
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
void PAINT_OT_vert_select_all(wmOperatorType *ot)
|
|
{
|
|
ot->name = "(De)select All";
|
|
ot->description = "Change selection for all vertices";
|
|
ot->idname = "PAINT_OT_vert_select_all";
|
|
|
|
ot->exec = vert_select_all_exec;
|
|
ot->poll = vert_paint_poll;
|
|
|
|
ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
|
|
|
|
WM_operator_properties_select_all(ot);
|
|
}
|
|
|
|
static int vert_select_ungrouped_exec(bContext *C, wmOperator *op)
|
|
{
|
|
Object *ob = CTX_data_active_object(C);
|
|
Mesh *me = ob->data;
|
|
|
|
if (BLI_listbase_is_empty(&ob->defbase) || (me->dvert == NULL)) {
|
|
BKE_report(op->reports, RPT_ERROR, "No weights/vertex groups on object");
|
|
return OPERATOR_CANCELLED;
|
|
}
|
|
|
|
paintvert_select_ungrouped(ob, RNA_boolean_get(op->ptr, "extend"), true);
|
|
paintvert_tag_select_update(C, ob);
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
void PAINT_OT_vert_select_ungrouped(wmOperatorType *ot)
|
|
{
|
|
/* identifiers */
|
|
ot->name = "Select Ungrouped";
|
|
ot->idname = "PAINT_OT_vert_select_ungrouped";
|
|
ot->description = "Select vertices without a group";
|
|
|
|
/* api callbacks */
|
|
ot->exec = vert_select_ungrouped_exec;
|
|
ot->poll = vert_paint_poll;
|
|
|
|
/* flags */
|
|
ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
|
|
|
|
RNA_def_boolean(ot->srna, "extend", false, "Extend", "Extend the selection");
|
|
}
|
|
|
|
static int face_select_hide_exec(bContext *C, wmOperator *op)
|
|
{
|
|
const bool unselected = RNA_boolean_get(op->ptr, "unselected");
|
|
Object *ob = CTX_data_active_object(C);
|
|
paintface_hide(C, ob, unselected);
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
void PAINT_OT_face_select_hide(wmOperatorType *ot)
|
|
{
|
|
ot->name = "Face Select Hide";
|
|
ot->description = "Hide selected faces";
|
|
ot->idname = "PAINT_OT_face_select_hide";
|
|
|
|
ot->exec = face_select_hide_exec;
|
|
ot->poll = facemask_paint_poll;
|
|
|
|
ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
|
|
|
|
RNA_def_boolean(
|
|
ot->srna, "unselected", 0, "Unselected", "Hide unselected rather than selected objects");
|
|
}
|
|
|
|
static int face_select_reveal_exec(bContext *C, wmOperator *op)
|
|
{
|
|
const bool select = RNA_boolean_get(op->ptr, "select");
|
|
Object *ob = CTX_data_active_object(C);
|
|
paintface_reveal(C, ob, select);
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
return OPERATOR_FINISHED;
|
|
}
|
|
|
|
void PAINT_OT_face_select_reveal(wmOperatorType *ot)
|
|
{
|
|
ot->name = "Face Select Reveal";
|
|
ot->description = "Reveal hidden faces";
|
|
ot->idname = "PAINT_OT_face_select_reveal";
|
|
|
|
ot->exec = face_select_reveal_exec;
|
|
ot->poll = facemask_paint_poll;
|
|
|
|
ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
|
|
|
|
RNA_def_boolean(ot->srna, "select", true, "Select", "");
|
|
}
|