EditMesh: add cube interactive tool

Initial add-cube tool which uses the scale cage to resize.

The 3D cursor is currently used to project the cursor onto.

We'll likely have more orientation options in the future.

4 files changed, 434 insertions, 0 deletions

diff --git a/source/blender/editors/mesh/CMakeLists.txt b/source/blender/editors/mesh/CMakeLists.txt
index 1d861257397..5cd768b4fe3 100644
--- a/source/blender/editors/mesh/CMakeLists.txt
+++ b/source/blender/editors/mesh/CMakeLists.txt
@@ -44,6 +44,7 @@ set(INC_SYS
 set(SRC
 	editface.c
 	editmesh_add.c
+	editmesh_add_manipulator.c
 	editmesh_bevel.c
 	editmesh_bisect.c
 	editmesh_extrude.c
diff --git a/source/blender/editors/mesh/editmesh_add_manipulator.c b/source/blender/editors/mesh/editmesh_add_manipulator.c
new file mode 100644
index 00000000000..3434ec82bb9
--- /dev/null
+++ b/source/blender/editors/mesh/editmesh_add_manipulator.c
@@ -0,0 +1,428 @@
+/*
+ * ***** BEGIN GPL LICENSE BLOCK *****
+ *
+ * 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.
+ *
+ * ***** END GPL LICENSE BLOCK *****
+ */
+
+/** \file blender/editors/mesh/editmesh_add_manipulator.c
+ *  \ingroup edmesh
+ *
+ * Creation manipulators.
+ */
+
+#include "MEM_guardedalloc.h"
+
+#include "BLI_math.h"
+
+#include "DNA_object_types.h"
+#include "DNA_scene_types.h"
+
+#include "BKE_context.h"
+#include "BKE_editmesh.h"
+
+#include "ED_manipulator_library.h"
+#include "ED_mesh.h"
+#include "ED_mesh.h"
+#include "ED_object.h"
+#include "ED_screen.h"
+#include "ED_undo.h"
+#include "ED_view3d.h"
+
+#include "RNA_access.h"
+#include "RNA_define.h"
+#include "RNA_define.h"
+
+#include "WM_api.h"
+#include "WM_types.h"
+#include "WM_types.h"
+
+#include "UI_resources.h"
+
+#include "BLT_translation.h"
+
+#include "mesh_intern.h"  /* own include */
+
+/* -------------------------------------------------------------------- */
+/** \name Helper Functions
+ * \{ */
+
+/**
+ * When we place a shape, pick a plane.
+ *
+ * We may base this choice on context,
+ * for now pick the "ground" based on the 3D cursor's dominant plane pointing down relative to the view.
+ */
+static void calc_initial_placement_point_from_view(
+        bContext *C, const float mval[2],
+        float r_location[3], float r_rotation[3][3])
+{
+
+	Scene *scene = CTX_data_scene(C);
+	View3D *v3d = CTX_wm_view3d(C);
+	bool use_mouse_project = true; /* TODO: make optional */
+
+	float cursor_matrix[4][4];
+	ED_view3d_cursor3d_calc_mat4(scene, v3d, cursor_matrix);
+
+	if (use_mouse_project) {
+		ARegion *ar = CTX_wm_region(C);
+		float ray_co[3], ray_no[3];
+
+		float point_on_plane[3][3];
+
+		if (ED_view3d_win_to_ray(
+		            CTX_data_depsgraph(C),
+		            ar, v3d, mval,
+		            ray_co, ray_no, false))
+		{
+			/* Pick the plane which gives the least distant projection,
+			 * this avoids. */
+			float dist_best_sq = FLT_MAX;
+			int axis_best = -1;
+			for (int i = 0; i < 3; i++) {
+				float plane[4];
+				plane_from_point_normal_v3(plane, cursor_matrix[3], cursor_matrix[i]);
+				float lambda;
+				if (isect_ray_plane_v3(ray_co, ray_no, plane, &lambda, true)) {
+					madd_v3_v3v3fl(point_on_plane[i], ray_co, ray_no, lambda);
+					float dist_test_sq = len_squared_v3v3(cursor_matrix[3], point_on_plane[i]);
+					if (dist_test_sq < dist_best_sq) {
+						dist_best_sq = dist_test_sq;
+						axis_best = i;
+					}
+				}
+			}
+			if (axis_best != -1) {
+				copy_v3_v3(r_location, point_on_plane[axis_best]);
+				/* For now always use same orientation. */
+				copy_m3_m4(r_rotation, cursor_matrix);
+			}
+			return;
+		}
+	}
+
+	/* fallback */
+	copy_v3_v3(r_location, cursor_matrix[3]);
+	copy_m3_m4(r_rotation, cursor_matrix);
+}
+
+/** \} */
+
+/* -------------------------------------------------------------------- */
+/** \name Placement Manipulator
+ * \{ */
+
+typedef struct ManipulatorPlacementGroup {
+	struct wmManipulator *cage;
+	struct {
+		bContext *context;
+		wmOperator *op;
+		PropertyRNA *prop_matrix;
+	} data;
+} ManipulatorPlacementGroup;
+
+/**
+ * \warning Calling redo from property updates is not great.
+ * This is needed because changing the RNA doesn't cause a redo
+ * and we're not using operator UI which does just this.
+ */
+static void manipulator_placement_exec(ManipulatorPlacementGroup *man)
+{
+	wmOperator *op = man->data.op;
+	if (op == WM_operator_last_redo((bContext *)man->data.context)) {
+		ED_undo_operator_repeat((bContext *)man->data.context, op);
+	}
+}
+
+static void manipulator_mesh_placement_update_from_op(ManipulatorPlacementGroup *man)
+{
+	wmOperator *op = man->data.op;
+	UNUSED_VARS(op);
+	/* For now don't read back from the operator. */
+#if 0
+	RNA_property_float_get_array(op->ptr, man->data.prop_matrix, &man->cage->matrix_offset[0][0]); 
+#endif
+}
+
+/* translate callbacks */
+static void manipulator_placement_prop_matrix_get(
+        const wmManipulator *mpr, wmManipulatorProperty *mpr_prop,
+        void *value_p)
+{
+	ManipulatorPlacementGroup *man = mpr->parent_mgroup->customdata;
+	wmOperator *op = man->data.op;
+	float *value = value_p;
+	BLI_assert(mpr_prop->type->array_length == 16);
+	UNUSED_VARS_NDEBUG(mpr_prop);
+
+	if (value_p != man->cage->matrix_offset) {
+		mul_m4_m4m4(value_p, man->cage->matrix_basis, man->cage->matrix_offset);
+		RNA_property_float_get_array(op->ptr, man->data.prop_matrix, value);
+	}
+}
+
+static void manipulator_placement_prop_matrix_set(
+        const wmManipulator *mpr, wmManipulatorProperty *mpr_prop,
+        const void *value)
+{
+	ManipulatorPlacementGroup *man = mpr->parent_mgroup->customdata;
+	wmOperator *op = man->data.op;
+
+	BLI_assert(mpr_prop->type->array_length == 16);
+	UNUSED_VARS_NDEBUG(mpr_prop);
+
+	float mat[4][4];
+	mul_m4_m4m4(mat, man->cage->matrix_basis, value);
+
+	if (is_negative_m4(mat)) {
+		negate_v3(mat[0]);
+		negate_v3(mat[1]);
+		negate_v3(mat[2]);
+	}
+
+	RNA_property_float_set_array(op->ptr, man->data.prop_matrix, &mat[0][0]);
+
+	manipulator_placement_exec(man);
+}
+
+static bool manipulator_mesh_placement_poll(const bContext *C, wmManipulatorGroupType *wgt)
+{
+	wmOperator *op = WM_operator_last_redo(C);
+	if (op == NULL || !STREQ(op->type->idname, "MESH_OT_primitive_cube_add_manipulator")) {
+		WM_manipulator_group_type_unlink_delayed_ptr(wgt);
+		return false;
+	}
+	return true;
+}
+
+static void manipulator_mesh_placement_modal_from_setup(
+        const bContext *C, wmManipulatorGroup *mgroup)
+{
+	ManipulatorPlacementGroup *man = mgroup->customdata;
+
+	/* Initial size. */
+	{
+		wmManipulator *mpr = man->cage;
+		zero_m4(mpr->matrix_offset);
+
+		/* TODO: support zero scaled matrix in 'MANIPULATOR_WT_cage_3d'. */
+		mpr->matrix_offset[0][0] = 0.01;
+		mpr->matrix_offset[1][1] = 0.01;
+		mpr->matrix_offset[2][2] = 0.01;
+		mpr->matrix_offset[3][3] = 1.0f;
+	}
+
+	/* Start off dragging. */
+	{
+		wmWindow *win = CTX_wm_window(C);
+		ARegion *ar = CTX_wm_region(C);
+		wmManipulator *mpr = man->cage;
+
+		{
+			float mat3[3][3];
+			float location[3];
+			calc_initial_placement_point_from_view(
+			        (bContext *)C, (float[2]){
+			            win->eventstate->x - ar->winrct.xmin,
+			            win->eventstate->y - ar->winrct.ymin,
+			        },
+			        location, mat3);
+			copy_m4_m3(mpr->matrix_basis, mat3);
+			copy_v3_v3(mpr->matrix_basis[3], location);
+		}
+
+		if (1) {
+			wmManipulatorMap *mmap = mgroup->parent_mmap;
+			WM_manipulator_modal_set_from_setup(
+			        mmap, (bContext *)C, man->cage, ED_MANIPULATOR_CAGE3D_PART_SCALE_MAX_X_MAX_Y_MAX_Z, win->eventstate);
+		}
+	}
+}
+
+static void manipulator_mesh_placement_setup(const bContext *C, wmManipulatorGroup *mgroup)
+{
+	wmOperator *op = WM_operator_last_redo(C);
+
+	if (op == NULL || !STREQ(op->type->idname, "MESH_OT_primitive_cube_add_manipulator")) {
+		return;
+	}
+
+	struct ManipulatorPlacementGroup *man = MEM_callocN(sizeof(ManipulatorPlacementGroup), __func__);
+	mgroup->customdata = man;
+
+	const wmManipulatorType *wt_cage = WM_manipulatortype_find("MANIPULATOR_WT_cage_3d", true);
+
+	man->cage = WM_manipulator_new_ptr(wt_cage, mgroup, NULL);
+
+	UI_GetThemeColor3fv(TH_MANIPULATOR_PRIMARY, man->cage->color);
+
+	RNA_enum_set(man->cage->ptr, "transform",
+	             ED_MANIPULATOR_CAGE2D_XFORM_FLAG_SCALE |
+	             ED_MANIPULATOR_CAGE2D_XFORM_FLAG_TRANSLATE |
+	             ED_MANIPULATOR_CAGE2D_XFORM_FLAG_SCALE_SIGNED);
+
+	WM_manipulator_set_flag(man->cage, WM_MANIPULATOR_DRAW_VALUE, true);
+
+	man->data.context = (bContext *)C;
+	man->data.op = op;
+	man->data.prop_matrix = RNA_struct_find_property(op->ptr, "matrix");
+
+	manipulator_mesh_placement_update_from_op(man);
+
+	/* Setup property callbacks */
+	{
+		WM_manipulator_target_property_def_func(
+		        man->cage, "matrix",
+		        &(const struct wmManipulatorPropertyFnParams) {
+		            .value_get_fn = manipulator_placement_prop_matrix_get,
+		            .value_set_fn = manipulator_placement_prop_matrix_set,
+		            .range_get_fn = NULL,
+		            .user_data = NULL,
+		        });
+	}
+
+	manipulator_mesh_placement_modal_from_setup(C, mgroup);
+}
+
+static void manipulator_mesh_placement_draw_prepare(
+        const bContext *UNUSED(C), wmManipulatorGroup *mgroup)
+{
+	ManipulatorPlacementGroup *man = mgroup->customdata;
+	if (man->data.op->next) {
+		man->data.op = WM_operator_last_redo((bContext *)man->data.context);
+	}
+	manipulator_mesh_placement_update_from_op(man);
+}
+
+static void MESH_WGT_add_bounds(struct wmManipulatorGroupType *wgt)
+{
+	wgt->name = "Mesh Add Bounds";
+	wgt->idname = "MESH_WGT_add_bounds";
+
+	wgt->flag = WM_MANIPULATORGROUPTYPE_3D;
+
+	wgt->mmap_params.spaceid = SPACE_VIEW3D;
+	wgt->mmap_params.regionid = RGN_TYPE_WINDOW;
+
+	wgt->poll = manipulator_mesh_placement_poll;
+	wgt->setup = manipulator_mesh_placement_setup;
+	wgt->draw_prepare = manipulator_mesh_placement_draw_prepare;
+}
+
+/** \} */
+
+/* -------------------------------------------------------------------- */
+/** \name Add Cube Manipulator-Operator
+ *
+ * For now we use a separate operator to add a cube,
+ * we can try to merge then however they are invoked differently
+ * and share the same BMesh creation code.
+ * \{ */
+
+
+static int add_primitive_cube_manipulator_exec(bContext *C, wmOperator *op)
+{
+	Object *obedit = CTX_data_edit_object(C);;
+	BMEditMesh *em = BKE_editmesh_from_object(obedit);
+	float matrix[4][4];
+
+	/* Get the matrix that defines the cube bounds (as set by the manipulator cage). */
+	{
+		PropertyRNA *prop_matrix = RNA_struct_find_property(op->ptr, "matrix");
+		if (RNA_property_is_set(op->ptr, prop_matrix)) {
+			RNA_property_float_get_array(op->ptr, prop_matrix, &matrix[0][0]);
+			invert_m4_m4(obedit->imat, obedit->obmat);
+			mul_m4_m4m4(matrix, obedit->imat, matrix);
+		}
+		else {
+			/* For the first update the widget may not set the matrix. */
+			return OPERATOR_FINISHED;
+		}
+	}
+
+	const bool calc_uvs = RNA_boolean_get(op->ptr, "calc_uvs");
+
+	if (calc_uvs) {
+		ED_mesh_uv_texture_ensure(obedit->data, NULL);
+	}
+
+	if (!EDBM_op_call_and_selectf(
+	        em, op, "verts.out", false,
+	        "create_cube matrix=%m4 size=%f calc_uvs=%b",
+	        matrix, 1.0f, calc_uvs))
+	{
+		return OPERATOR_CANCELLED;
+	}
+
+	EDBM_selectmode_flush_ex(em, SCE_SELECT_VERTEX);
+	EDBM_update_generic(em, true, true);
+
+	return OPERATOR_FINISHED;
+}
+
+static int add_primitive_cube_manipulator_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
+{
+	View3D *v3d = CTX_wm_view3d(C);
+
+	int ret = add_primitive_cube_manipulator_exec(C, op);
+	if (ret & OPERATOR_FINISHED) {
+		/* Setup manipulators */
+		if (v3d && (v3d->twflag & V3D_MANIPULATOR_DRAW)) {
+			ARegion *ar = CTX_wm_region(C);
+			wmManipulatorMap *mmap = ar->manipulator_map;
+			wmManipulatorGroupType *wgt = WM_manipulatorgrouptype_find("MESH_WGT_add_bounds", false);
+			wmManipulatorGroup *mgroup = WM_manipulatormap_group_find_ptr(mmap, wgt);
+			if (mgroup != NULL) {
+				ManipulatorPlacementGroup *man = mgroup->customdata;
+				man->data.op = op;
+				manipulator_mesh_placement_modal_from_setup(C, mgroup);
+			}
+			else {
+				WM_manipulator_group_type_ensure_ptr(wgt);
+			}
+		}
+	}
+
+	return ret;
+}
+
+void MESH_OT_primitive_cube_add_manipulator(wmOperatorType *ot)
+{
+	/* identifiers */
+	ot->name = "Add Cube";
+	ot->description = "Construct a cube mesh";
+	ot->idname = "MESH_OT_primitive_cube_add_manipulator";
+
+	/* api callbacks */
+	ot->invoke = add_primitive_cube_manipulator_invoke;
+	ot->exec = add_primitive_cube_manipulator_exec;
+	ot->poll = ED_operator_editmesh_view3d;
+
+	/* flags */
+	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
+
+	ED_object_add_mesh_props(ot);
+	ED_object_add_generic_props(ot, true);
+
+	/* hidden props */
+	PropertyRNA *prop = RNA_def_float_matrix(ot->srna, "matrix", 4, 4, NULL, 0.0f, 0.0f, "Matrix", "", 0.0f, 0.0f);
+	RNA_def_property_flag(prop, PROP_HIDDEN | PROP_SKIP_SAVE);
+
+	WM_manipulatorgrouptype_append(MESH_WGT_add_bounds);
+}
+
+/** \} */
diff --git a/source/blender/editors/mesh/mesh_intern.h b/source/blender/editors/mesh/mesh_intern.h
index 43677bc6912..b9e9ae9c1fd 100644
--- a/source/blender/editors/mesh/mesh_intern.h
+++ b/source/blender/editors/mesh/mesh_intern.h
@@ -92,6 +92,8 @@ void MESH_OT_primitive_monkey_add(struct wmOperatorType *ot);
 void MESH_OT_primitive_uv_sphere_add(struct wmOperatorType *ot);
 void MESH_OT_primitive_ico_sphere_add(struct wmOperatorType *ot);
 
+/* *** editmesh_add_manipulator.c *** */
+void MESH_OT_primitive_cube_add_manipulator(struct wmOperatorType *ot);
 
 /* *** editmesh_bevel.c *** */
 void MESH_OT_bevel(struct wmOperatorType *ot);
diff --git a/source/blender/editors/mesh/mesh_ops.c b/source/blender/editors/mesh/mesh_ops.c
index 6552062a0d9..dc9cd29a0af 100644
--- a/source/blender/editors/mesh/mesh_ops.c
+++ b/source/blender/editors/mesh/mesh_ops.c
@@ -77,6 +77,9 @@ void ED_operatortypes_mesh(void)
 	WM_operatortype_append(MESH_OT_primitive_monkey_add);
 	WM_operatortype_append(MESH_OT_primitive_uv_sphere_add);
 	WM_operatortype_append(MESH_OT_primitive_ico_sphere_add);
+
+	WM_operatortype_append(MESH_OT_primitive_cube_add_manipulator);
+
 	WM_operatortype_append(MESH_OT_duplicate);
 	WM_operatortype_append(MESH_OT_remove_doubles);
 	WM_operatortype_append(MESH_OT_spin);