2009-02-19 23:53:40 +00:00
|
|
|
|
|
|
|
#include <math.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
|
|
|
|
#include "DNA_mesh_types.h"
|
|
|
|
#include "DNA_meshdata_types.h"
|
|
|
|
#include "DNA_object_types.h"
|
2009-08-14 04:45:29 +00:00
|
|
|
|
2009-02-19 23:53:40 +00:00
|
|
|
#include "DNA_scene_types.h"
|
|
|
|
#include "DNA_screen_types.h"
|
|
|
|
#include "DNA_view3d_types.h"
|
|
|
|
|
2009-08-14 04:45:29 +00:00
|
|
|
#include "RNA_access.h"
|
|
|
|
#include "RNA_define.h"
|
|
|
|
|
2009-11-10 20:43:45 +00:00
|
|
|
#include "BLI_math.h"
|
2009-02-19 23:53:40 +00:00
|
|
|
|
2009-03-11 00:43:08 +00:00
|
|
|
#include "BKE_brush.h"
|
2009-08-14 04:45:29 +00:00
|
|
|
#include "BKE_context.h"
|
2009-02-19 23:53:40 +00:00
|
|
|
#include "BKE_DerivedMesh.h"
|
|
|
|
#include "BKE_global.h"
|
2009-08-16 19:50:00 +00:00
|
|
|
#include "BKE_paint.h"
|
|
|
|
|
2009-02-19 23:53:40 +00:00
|
|
|
#include "BKE_utildefines.h"
|
|
|
|
|
|
|
|
#include "BIF_gl.h"
|
|
|
|
|
|
|
|
#include "ED_view3d.h"
|
2009-11-01 00:06:53 +00:00
|
|
|
#include "ED_screen.h"
|
|
|
|
|
|
|
|
#include "BLO_sys_types.h"
|
|
|
|
#include "ED_mesh.h" /* for face mask functions */
|
2009-02-19 23:53:40 +00:00
|
|
|
|
2009-08-14 04:45:29 +00:00
|
|
|
#include "WM_api.h"
|
|
|
|
#include "WM_types.h"
|
|
|
|
|
2009-02-19 23:53:40 +00:00
|
|
|
#include "paint_intern.h"
|
|
|
|
|
|
|
|
/* 3D Paint */
|
|
|
|
|
|
|
|
static void imapaint_project(Object *ob, float *model, float *proj, float *co, float *pco)
|
|
|
|
{
|
|
|
|
VECCOPY(pco, co);
|
|
|
|
pco[3]= 1.0f;
|
|
|
|
|
2009-11-10 20:43:45 +00:00
|
|
|
mul_m4_v3(ob->obmat, pco);
|
|
|
|
mul_m4_v3((float(*)[4])model, pco);
|
|
|
|
mul_m4_v4((float(*)[4])proj, pco);
|
2009-02-19 23:53:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void imapaint_tri_weights(Object *ob, float *v1, float *v2, float *v3, float *co, float *w)
|
|
|
|
{
|
|
|
|
float pv1[4], pv2[4], pv3[4], h[3], divw;
|
|
|
|
float model[16], proj[16], wmat[3][3], invwmat[3][3];
|
|
|
|
GLint view[4];
|
|
|
|
|
|
|
|
/* compute barycentric coordinates */
|
|
|
|
|
|
|
|
/* get the needed opengl matrices */
|
|
|
|
glGetIntegerv(GL_VIEWPORT, view);
|
|
|
|
glGetFloatv(GL_MODELVIEW_MATRIX, model);
|
|
|
|
glGetFloatv(GL_PROJECTION_MATRIX, proj);
|
|
|
|
view[0] = view[1] = 0;
|
|
|
|
|
|
|
|
/* project the verts */
|
|
|
|
imapaint_project(ob, model, proj, v1, pv1);
|
|
|
|
imapaint_project(ob, model, proj, v2, pv2);
|
|
|
|
imapaint_project(ob, model, proj, v3, pv3);
|
|
|
|
|
|
|
|
/* do inverse view mapping, see gluProject man page */
|
|
|
|
h[0]= (co[0] - view[0])*2.0f/view[2] - 1;
|
|
|
|
h[1]= (co[1] - view[1])*2.0f/view[3] - 1;
|
|
|
|
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];
|
|
|
|
|
2009-11-10 20:43:45 +00:00
|
|
|
invert_m3_m3(invwmat, wmat);
|
|
|
|
mul_m3_v3(invwmat, h);
|
2009-02-19 23:53:40 +00:00
|
|
|
|
|
|
|
VECCOPY(w, h);
|
|
|
|
|
|
|
|
/* w is still divided by perspdiv, make it sum to one */
|
|
|
|
divw= w[0] + w[1] + w[2];
|
|
|
|
if(divw != 0.0f)
|
2009-11-10 20:43:45 +00:00
|
|
|
mul_v3_fl(w, 1.0f/divw);
|
2009-02-19 23:53:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* compute uv coordinates of mouse in face */
|
|
|
|
void imapaint_pick_uv(Scene *scene, Object *ob, Mesh *mesh, unsigned int faceindex, int *xy, float *uv)
|
|
|
|
{
|
|
|
|
DerivedMesh *dm = mesh_get_derived_final(scene, ob, CD_MASK_BAREMESH);
|
(NOTE: DO NOT TEST)
Start of planned DerivedMesh refactoring. The mface
interfaces in DerivedMesh have been renamed to reflect
their new status as tesselated face interfaces (rather
then the primary ones, which are now stored in mpolys).
short review: mpolys store "primary" face data, while
mfaces store the tesselated form of the mesh (generally
as triangles). mpolys are defined by mloops, and each
mpoly defines a range of loops it "owns" in the main
mloop array.
I've also added basic read-only face iterators, which
are implemented for CDDM, ccgsubsurf, and the bmeditmesh
derivedmesh. Since faces are now variable-length things,
trying to implement the same interface as mfaces would not
have worked well (especially since faces are stored as
an mpoly + a range of mloops).
I figure first we can evaluate these simple read-only
face iterators, then decide if a) we like using iterators
in DerivedMesh, b) how much of it should use them, and c)
if we want write-capable iterators.
I plan to write official docs on this design after I get
it more stable; I'm committing now because there's a rather
lot of changes, and I might do a merge soon.
2009-06-10 10:06:25 +00:00
|
|
|
int *index = dm->getTessFaceDataArray(dm, CD_ORIGINDEX);
|
|
|
|
MTFace *tface = dm->getTessFaceDataArray(dm, CD_MTFACE), *tf;
|
|
|
|
int numfaces = dm->getNumTessFaces(dm), a;
|
2009-02-19 23:53:40 +00:00
|
|
|
float p[2], w[3], absw, minabsw;
|
|
|
|
MFace mf;
|
|
|
|
MVert mv[4];
|
|
|
|
|
|
|
|
minabsw = 1e10;
|
|
|
|
uv[0] = uv[1] = 0.0;
|
|
|
|
|
|
|
|
/* test all faces in the derivedmesh with the original index of the picked face */
|
|
|
|
for(a = 0; a < numfaces; a++) {
|
|
|
|
if(index[a] == faceindex) {
|
(NOTE: DO NOT TEST)
Start of planned DerivedMesh refactoring. The mface
interfaces in DerivedMesh have been renamed to reflect
their new status as tesselated face interfaces (rather
then the primary ones, which are now stored in mpolys).
short review: mpolys store "primary" face data, while
mfaces store the tesselated form of the mesh (generally
as triangles). mpolys are defined by mloops, and each
mpoly defines a range of loops it "owns" in the main
mloop array.
I've also added basic read-only face iterators, which
are implemented for CDDM, ccgsubsurf, and the bmeditmesh
derivedmesh. Since faces are now variable-length things,
trying to implement the same interface as mfaces would not
have worked well (especially since faces are stored as
an mpoly + a range of mloops).
I figure first we can evaluate these simple read-only
face iterators, then decide if a) we like using iterators
in DerivedMesh, b) how much of it should use them, and c)
if we want write-capable iterators.
I plan to write official docs on this design after I get
it more stable; I'm committing now because there's a rather
lot of changes, and I might do a merge soon.
2009-06-10 10:06:25 +00:00
|
|
|
dm->getTessFace(dm, a, &mf);
|
2009-02-19 23:53:40 +00:00
|
|
|
|
|
|
|
dm->getVert(dm, mf.v1, &mv[0]);
|
|
|
|
dm->getVert(dm, mf.v2, &mv[1]);
|
|
|
|
dm->getVert(dm, mf.v3, &mv[2]);
|
|
|
|
if(mf.v4)
|
|
|
|
dm->getVert(dm, mf.v4, &mv[3]);
|
|
|
|
|
|
|
|
tf= &tface[a];
|
|
|
|
|
|
|
|
p[0]= xy[0];
|
|
|
|
p[1]= xy[1];
|
|
|
|
|
|
|
|
if(mf.v4) {
|
|
|
|
/* the triangle with the largest absolute values is the one
|
|
|
|
with the most negative weights */
|
|
|
|
imapaint_tri_weights(ob, mv[0].co, mv[1].co, mv[3].co, p, w);
|
|
|
|
absw= fabs(w[0]) + fabs(w[1]) + fabs(w[2]);
|
|
|
|
if(absw < minabsw) {
|
|
|
|
uv[0]= tf->uv[0][0]*w[0] + tf->uv[1][0]*w[1] + tf->uv[3][0]*w[2];
|
|
|
|
uv[1]= tf->uv[0][1]*w[0] + tf->uv[1][1]*w[1] + tf->uv[3][1]*w[2];
|
|
|
|
minabsw = absw;
|
|
|
|
}
|
|
|
|
|
|
|
|
imapaint_tri_weights(ob, mv[1].co, mv[2].co, mv[3].co, p, w);
|
|
|
|
absw= fabs(w[0]) + fabs(w[1]) + fabs(w[2]);
|
|
|
|
if(absw < minabsw) {
|
|
|
|
uv[0]= tf->uv[1][0]*w[0] + tf->uv[2][0]*w[1] + tf->uv[3][0]*w[2];
|
|
|
|
uv[1]= tf->uv[1][1]*w[0] + tf->uv[2][1]*w[1] + tf->uv[3][1]*w[2];
|
|
|
|
minabsw = absw;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
imapaint_tri_weights(ob, mv[0].co, mv[1].co, mv[2].co, p, w);
|
|
|
|
absw= fabs(w[0]) + fabs(w[1]) + fabs(w[2]);
|
|
|
|
if(absw < minabsw) {
|
|
|
|
uv[0]= tf->uv[0][0]*w[0] + tf->uv[1][0]*w[1] + tf->uv[2][0]*w[2];
|
|
|
|
uv[1]= tf->uv[0][1]*w[0] + tf->uv[1][1]*w[1] + tf->uv[2][1]*w[2];
|
|
|
|
minabsw = absw;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
dm->release(dm);
|
|
|
|
}
|
|
|
|
|
|
|
|
///* returns 0 if not found, otherwise 1 */
|
|
|
|
int imapaint_pick_face(ViewContext *vc, Mesh *me, int *mval, unsigned int *index)
|
|
|
|
{
|
|
|
|
if(!me || me->totface==0)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
/* sample only on the exact position */
|
|
|
|
*index = view3d_sample_backbuf(vc, mval[0], mval[1]);
|
|
|
|
|
|
|
|
if((*index)<=0 || (*index)>(unsigned int)me->totface)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
(*index)--;
|
|
|
|
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* used for both 3d view and image window */
|
|
|
|
void paint_sample_color(Scene *scene, ARegion *ar, int x, int y) /* frontbuf */
|
|
|
|
{
|
2009-08-16 19:50:00 +00:00
|
|
|
Brush *br = paint_brush(paint_get_active(scene));
|
2009-02-19 23:53:40 +00:00
|
|
|
unsigned int col;
|
|
|
|
char *cp;
|
|
|
|
|
|
|
|
CLAMP(x, 0, ar->winx);
|
|
|
|
CLAMP(y, 0, ar->winy);
|
|
|
|
|
|
|
|
glReadBuffer(GL_FRONT);
|
|
|
|
glReadPixels(x+ar->winrct.xmin, y+ar->winrct.ymin, 1, 1, GL_RGBA, GL_UNSIGNED_BYTE, &col);
|
|
|
|
glReadBuffer(GL_BACK);
|
|
|
|
|
|
|
|
cp = (char *)&col;
|
|
|
|
|
2009-08-16 19:50:00 +00:00
|
|
|
if(br) {
|
|
|
|
br->rgb[0]= cp[0]/255.0f;
|
|
|
|
br->rgb[1]= cp[1]/255.0f;
|
|
|
|
br->rgb[2]= cp[2]/255.0f;
|
2009-02-19 23:53:40 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2009-08-14 04:45:29 +00:00
|
|
|
static int brush_curve_preset_exec(bContext *C, wmOperator *op)
|
|
|
|
{
|
2009-08-16 19:50:00 +00:00
|
|
|
Brush *br = paint_brush(paint_get_active(CTX_data_scene(C)));
|
2009-08-14 04:45:29 +00:00
|
|
|
brush_curve_preset(br, RNA_enum_get(op->ptr, "shape"));
|
|
|
|
|
|
|
|
return OPERATOR_FINISHED;
|
|
|
|
}
|
|
|
|
|
|
|
|
static int brush_curve_preset_poll(bContext *C)
|
|
|
|
{
|
2009-08-16 19:50:00 +00:00
|
|
|
Brush *br = paint_brush(paint_get_active(CTX_data_scene(C)));
|
2009-08-14 04:45:29 +00:00
|
|
|
|
2009-08-16 19:50:00 +00:00
|
|
|
return br && br->curve;
|
2009-08-14 04:45:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void BRUSH_OT_curve_preset(wmOperatorType *ot)
|
|
|
|
{
|
|
|
|
static EnumPropertyItem prop_shape_items[] = {
|
|
|
|
{BRUSH_PRESET_SHARP, "SHARP", 0, "Sharp", ""},
|
|
|
|
{BRUSH_PRESET_SMOOTH, "SMOOTH", 0, "Smooth", ""},
|
|
|
|
{BRUSH_PRESET_MAX, "MAX", 0, "Max", ""},
|
|
|
|
{0, NULL, 0, NULL, NULL}};
|
|
|
|
|
|
|
|
ot->name= "Preset";
|
2009-09-16 17:43:09 +00:00
|
|
|
ot->description= "Set brush shape.";
|
2009-08-14 04:45:29 +00:00
|
|
|
ot->idname= "BRUSH_OT_curve_preset";
|
|
|
|
|
|
|
|
ot->exec= brush_curve_preset_exec;
|
|
|
|
ot->poll= brush_curve_preset_poll;
|
|
|
|
|
|
|
|
ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
|
|
|
|
|
|
|
|
RNA_def_enum(ot->srna, "shape", prop_shape_items, BRUSH_PRESET_SHARP, "Mode", "");
|
|
|
|
}
|
2009-11-01 00:06:53 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* face-select ops */
|
|
|
|
static int paint_select_linked_exec(bContext *C, wmOperator *op)
|
|
|
|
{
|
|
|
|
select_linked_tfaces(C, CTX_data_active_object(C), NULL, 2);
|
|
|
|
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, wmEvent *event)
|
|
|
|
{
|
|
|
|
int mode= RNA_boolean_get(op->ptr, "extend") ? 1:0;
|
|
|
|
select_linked_tfaces(C, CTX_data_active_object(C), event->mval, mode);
|
|
|
|
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.";
|
|
|
|
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, "extend", 0, "Extend", "Extend the existing selection");
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static int face_deselect_all_exec(bContext *C, wmOperator *op)
|
|
|
|
{
|
|
|
|
deselectall_tface(CTX_data_active_object(C));
|
|
|
|
ED_region_tag_redraw(CTX_wm_region(C));
|
|
|
|
return OPERATOR_FINISHED;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void PAINT_OT_face_deselect_all(wmOperatorType *ot)
|
|
|
|
{
|
|
|
|
ot->name= "Select Linked";
|
|
|
|
ot->description= "Select linked faces under the mouse.";
|
|
|
|
ot->idname= "PAINT_OT_face_deselect_all";
|
|
|
|
|
|
|
|
ot->exec= face_deselect_all_exec;
|
|
|
|
ot->poll= facemask_paint_poll;
|
|
|
|
|
|
|
|
ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
|
|
|
|
}
|