/*
 * ***** 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.
 *
 * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
 * All rights reserved.
 *
 * Contributor(s): Blender Foundation, 2002-2008 full recode
 *
 * ***** END GPL LICENSE BLOCK *****
 */

/** \file blender/editors/object/object_edit.c
 *  \ingroup edobj
 */

#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include <float.h>
#include <ctype.h>
#include <stddef.h> //for offsetof

#include "MEM_guardedalloc.h"

#include "BLI_blenlib.h"
#include "BLI_math.h"
#include "BLI_utildefines.h"
#include "BLI_ghash.h"
#include "BLI_string_utils.h"

#include "BLT_translation.h"

#include "DNA_armature_types.h"
#include "DNA_curve_types.h"
#include "DNA_gpencil_types.h"
#include "DNA_group_types.h"
#include "DNA_material_types.h"
#include "DNA_meta_types.h"
#include "DNA_scene_types.h"
#include "DNA_object_types.h"
#include "DNA_object_force_types.h"
#include "DNA_meshdata_types.h"
#include "DNA_vfont_types.h"
#include "DNA_mesh_types.h"
#include "DNA_lattice_types.h"
#include "DNA_workspace_types.h"

#include "IMB_imbuf_types.h"

#include "BKE_anim.h"
#include "BKE_collection.h"
#include "BKE_constraint.h"
#include "BKE_context.h"
#include "BKE_curve.h"
#include "BKE_effect.h"
#include "BKE_global.h"
#include "BKE_image.h"
#include "BKE_lattice.h"
#include "BKE_library.h"
#include "BKE_main.h"
#include "BKE_material.h"
#include "BKE_mball.h"
#include "BKE_mesh.h"
#include "BKE_object.h"
#include "BKE_paint.h"
#include "BKE_pointcache.h"
#include "BKE_softbody.h"
#include "BKE_modifier.h"
#include "BKE_editlattice.h"
#include "BKE_editmesh.h"
#include "BKE_report.h"
#include "BKE_workspace.h"
#include "BKE_layer.h"

#include "DEG_depsgraph.h"
#include "DEG_depsgraph_build.h"

#include "ED_armature.h"
#include "ED_curve.h"
#include "ED_mesh.h"
#include "ED_mball.h"
#include "ED_lattice.h"
#include "ED_object.h"
#include "ED_outliner.h"
#include "ED_screen.h"
#include "ED_undo.h"
#include "ED_image.h"

#include "RNA_access.h"
#include "RNA_define.h"
#include "RNA_enum_types.h"

/* for menu/popup icons etc etc*/

#include "UI_interface.h"
#include "UI_resources.h"

#include "WM_api.h"
#include "WM_types.h"
#include "WM_message.h"
#include "WM_toolsystem.h"

#include "object_intern.h"  // own include

/* prototypes */
typedef struct MoveToCollectionData MoveToCollectionData;
static void move_to_collection_menus_items(struct uiLayout *layout, struct MoveToCollectionData *menu);

/* ************* XXX **************** */
static void error(const char *UNUSED(arg)) {}
static void waitcursor(int UNUSED(val)) {}
static int pupmenu(const char *UNUSED(msg)) { return 0; }

/* port over here */
static void error_libdata(void) {}

Object *ED_object_context(bContext *C)
{
	return CTX_data_pointer_get_type(C, "object", &RNA_Object).data;
}

/* find the correct active object per context
 * note: context can be NULL when called from a enum with PROP_ENUM_NO_CONTEXT */
Object *ED_object_active_context(bContext *C)
{
	Object *ob = NULL;
	if (C) {
		ob = ED_object_context(C);
		if (!ob) ob = CTX_data_active_object(C);
	}
	return ob;
}

/* ********************** object hiding *************************** */

static bool object_hide_poll(bContext *C)
{
	if (CTX_wm_space_outliner(C) != NULL) {
		return ED_outliner_collections_editor_poll(C);
	}
	else {
		return ED_operator_view3d_active(C);
	}
}

static int object_hide_view_clear_exec(bContext *C, wmOperator *op)
{
	Scene *scene = CTX_data_scene(C);
	ViewLayer *view_layer = CTX_data_view_layer(C);
	const bool select = RNA_boolean_get(op->ptr, "select");
	bool changed = false;

	for (Base *base = view_layer->object_bases.first; base; base = base->next) {
		if (base->flag & BASE_HIDDEN) {
			base->flag &= ~BASE_HIDDEN;
			changed = true;

			if (select) {
				ED_object_base_select(base, BA_SELECT);
			}
		}
	}

	if (!changed) {
		return OPERATOR_CANCELLED;
	}

	BKE_layer_collection_sync(scene, view_layer);
	DEG_id_tag_update(&scene->id, DEG_TAG_BASE_FLAGS_UPDATE);
	WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene);

	return OPERATOR_FINISHED;
}

void OBJECT_OT_hide_view_clear(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Show Hidden Objects";
	ot->description = "Reveal temporarily hidden objects";
	ot->idname = "OBJECT_OT_hide_view_clear";

	/* api callbacks */
	ot->exec = object_hide_view_clear_exec;
	ot->poll = object_hide_poll;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;

	PropertyRNA *prop = RNA_def_boolean(ot->srna, "select", false, "Select", "");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE | PROP_HIDDEN);
}

static int object_hide_view_set_exec(bContext *C, wmOperator *op)
{
	Scene *scene = CTX_data_scene(C);
	ViewLayer *view_layer = CTX_data_view_layer(C);
	const bool unselected = RNA_boolean_get(op->ptr, "unselected");

	/* Do nothing if no objects was selected. */
	bool have_selected = false;
	for (Base *base = view_layer->object_bases.first; base; base = base->next) {
		if (base->flag & BASE_VISIBLE) {
			if (base->flag & BASE_SELECTED) {
				have_selected = true;
				break;
			}
		}
	}

	if (!have_selected) {
		return OPERATOR_CANCELLED;
	}

	/* Hide selected or unselected objects. */
	for (Base *base = view_layer->object_bases.first; base; base = base->next) {
		if (!(base->flag & BASE_VISIBLE)) {
			continue;
		}

		if (!unselected) {
			if (base->flag & BASE_SELECTED) {
				ED_object_base_select(base, BA_DESELECT);
				base->flag |= BASE_HIDDEN;
			}
		}
		else {
			if (!(base->flag & BASE_SELECTED)) {
				ED_object_base_select(base, BA_DESELECT);
				base->flag |= BASE_HIDDEN;
			}
		}
	}

	BKE_layer_collection_sync(scene, view_layer);
	DEG_id_tag_update(&scene->id, DEG_TAG_BASE_FLAGS_UPDATE);
	WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene);

	return OPERATOR_FINISHED;
}

void OBJECT_OT_hide_view_set(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Hide Objects";
	ot->description = "Temporarily hide objects from the viewport";
	ot->idname = "OBJECT_OT_hide_view_set";

	/* api callbacks */
	ot->exec = object_hide_view_set_exec;
	ot->poll = object_hide_poll;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;

	PropertyRNA *prop;
	prop = RNA_def_boolean(ot->srna, "unselected", 0, "Unselected", "Hide unselected rather than selected objects");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE | PROP_HIDDEN);
}

static int object_hide_collection_exec(bContext *C, wmOperator *op)
{
	int index = RNA_int_get(op->ptr, "collection_index");
	bool extend = (CTX_wm_window(C)->eventstate->shift != 0);

	if (CTX_wm_window(C)->eventstate->alt != 0) {
		index += 10;
	}

	Scene *scene = CTX_data_scene(C);
	ViewLayer *view_layer = CTX_data_view_layer(C);
	LayerCollection *lc = BKE_layer_collection_from_index(view_layer, index);

	if (!lc) {
		return OPERATOR_CANCELLED;
	}

	BKE_layer_collection_set_visible(scene, view_layer, lc, extend);

	DEG_id_tag_update(&scene->id, DEG_TAG_BASE_FLAGS_UPDATE);
	WM_event_add_notifier(C, NC_SCENE | ND_OB_SELECT, scene);

	return OPERATOR_FINISHED;
}

#define COLLECTION_INVALID_INDEX -1

void ED_hide_collections_menu_draw(const bContext *C, uiLayout *layout)
{
	ViewLayer *view_layer = CTX_data_view_layer(C);
	LayerCollection *lc_scene = view_layer->layer_collections.first;

	uiLayoutSetOperatorContext(layout, WM_OP_EXEC_REGION_WIN);

	for (LayerCollection *lc = lc_scene->layer_collections.first; lc; lc = lc->next) {
		int index = BKE_layer_collection_findindex(view_layer, lc);
		uiLayout *row = uiLayoutRow(layout, false);

		if (lc->collection->flag & COLLECTION_RESTRICT_VIEW) {
			continue;
		}

		if ((view_layer->runtime_flag & VIEW_LAYER_HAS_HIDE) &&
		    !(lc->runtime_flag & LAYER_COLLECTION_HAS_VISIBLE_OBJECTS))
		{
			uiLayoutSetActive(row, false);
		}

		int icon = ICON_NONE;
		if (BKE_layer_collection_has_selected_objects(view_layer, lc)) {
			icon = ICON_LAYER_ACTIVE;
		}
		else if (lc->runtime_flag & LAYER_COLLECTION_HAS_OBJECTS) {
			icon = ICON_LAYER_USED;
		}

		uiItemIntO(row,
		           lc->collection->id.name + 2,
		           icon,
		           "OBJECT_OT_hide_collection",
		           "collection_index",
		           index);
	}
}

static int object_hide_collection_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
{
	/* Immediately execute if collection index was specified. */
	int index = RNA_int_get(op->ptr, "collection_index");
	if (index != COLLECTION_INVALID_INDEX) {
		return object_hide_collection_exec(C, op);
	}

	/* Open popup menu. */
	const char *title = CTX_IFACE_(op->type->translation_context, op->type->name);
	uiPopupMenu *pup = UI_popup_menu_begin(C, title, ICON_GROUP);
	uiLayout *layout = UI_popup_menu_layout(pup);

	ED_hide_collections_menu_draw(C, layout);

	UI_popup_menu_end(C, pup);

	return OPERATOR_INTERFACE;
}

void OBJECT_OT_hide_collection(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Hide Objects By Collection";
	ot->description = "Show only objects in collection (Shift to extend)";
	ot->idname = "OBJECT_OT_hide_collection";

	/* api callbacks */
	ot->exec = object_hide_collection_exec;
	ot->invoke = object_hide_collection_invoke;
	ot->poll = ED_operator_view3d_active;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;

	/* Properties. */
	PropertyRNA *prop;
	prop = RNA_def_int(ot->srna, "collection_index", COLLECTION_INVALID_INDEX, COLLECTION_INVALID_INDEX, INT_MAX,
	                   "Collection Index", "Index of the collection to change visibility", 0, INT_MAX);
	RNA_def_property_flag(prop, PROP_SKIP_SAVE | PROP_HIDDEN);
}

/* ******************* toggle editmode operator  ***************** */

static bool mesh_needs_keyindex(Main *bmain, const Mesh *me)
{
	if (me->key) {
		return false;  /* will be added */
	}

	for (const Object *ob = bmain->object.first; ob; ob = ob->id.next) {
		if ((ob->parent) && (ob->parent->data == me) && ELEM(ob->partype, PARVERT1, PARVERT3)) {
			return true;
		}
		if (ob->data == me) {
			for (const ModifierData *md = ob->modifiers.first; md; md = md->next) {
				if (md->type == eModifierType_Hook) {
					return true;
				}
			}
		}
	}
	return false;
}

/**
 * Load EditMode data back into the object,
 * optionally freeing the editmode data.
 */
static bool ED_object_editmode_load_ex(Main *bmain, Object *obedit, const bool freedata)
{
	if (obedit == NULL) {
		return false;
	}

	if (obedit->type == OB_MESH) {
		Mesh *me = obedit->data;
		if (me->edit_btmesh == NULL) {
			return false;
		}

		if (me->edit_btmesh->bm->totvert > MESH_MAX_VERTS) {
			error("Too many vertices");
			return false;
		}

		EDBM_mesh_load(bmain, obedit);

		if (freedata) {
			EDBM_mesh_free(me->edit_btmesh);
			MEM_freeN(me->edit_btmesh);
			me->edit_btmesh = NULL;
		}
		/* will be recalculated as needed. */
		{
			ED_mesh_mirror_spatial_table(NULL, NULL, NULL, NULL, 'e');
			ED_mesh_mirror_topo_table(NULL, NULL, 'e');
		}
	}
	else if (obedit->type == OB_ARMATURE) {
		const bArmature *arm = obedit->data;
		if (arm->edbo == NULL) {
			return false;
		}
		ED_armature_from_edit(bmain, obedit->data);
		if (freedata) {
			ED_armature_edit_free(obedit->data);
		}
		/* TODO(sergey): Pose channels might have been changed, so need
		 * to inform dependency graph about this. But is it really the
		 * best place to do this?
		 */
		DEG_relations_tag_update(bmain);
	}
	else if (ELEM(obedit->type, OB_CURVE, OB_SURF)) {
		const Curve *cu = obedit->data;
		if (cu->editnurb == NULL) {
			return false;
		}
		ED_curve_editnurb_load(bmain, obedit);
		if (freedata) {
			ED_curve_editnurb_free(obedit);
		}
	}
	else if (obedit->type == OB_FONT) {
		const Curve *cu = obedit->data;
		if (cu->editfont == NULL) {
			return false;
		}
		ED_curve_editfont_load(obedit);
		if (freedata) {
			ED_curve_editfont_free(obedit);
		}
	}
	else if (obedit->type == OB_LATTICE) {
		const Lattice *lt = obedit->data;
		if (lt->editlatt == NULL) {
			return false;
		}
		BKE_editlattice_load(obedit);
		if (freedata) {
			BKE_editlattice_free(obedit);
		}
	}
	else if (obedit->type == OB_MBALL) {
		const MetaBall *mb = obedit->data;
		if (mb->editelems == NULL) {
			return false;
		}
		ED_mball_editmball_load(obedit);
		if (freedata) {
			ED_mball_editmball_free(obedit);
		}
	}

	return true;
}

bool ED_object_editmode_load(Main *bmain, Object *obedit)
{
	return ED_object_editmode_load_ex(bmain, obedit, false);
}

/**
 * \param flag:
 * - If #EM_FREEDATA isn't in the flag, use ED_object_editmode_load directly.
 */
bool ED_object_editmode_exit_ex(Main *bmain, Scene *scene, Object *obedit, int flag)
{
	const bool freedata = (flag & EM_FREEDATA) != 0;

	if (flag & EM_WAITCURSOR) waitcursor(1);

	if (ED_object_editmode_load_ex(bmain, obedit, freedata) == false) {
		/* in rare cases (background mode) its possible active object
		 * is flagged for editmode, without 'obedit' being set [#35489] */
		if (UNLIKELY(obedit && obedit->mode & OB_MODE_EDIT)) {
			obedit->mode &= ~OB_MODE_EDIT;
		}
		if (flag & EM_WAITCURSOR) waitcursor(0);
		return true;
	}

	/* freedata only 0 now on file saves and render */
	if (freedata) {
		ListBase pidlist;
		PTCacheID *pid;

		/* flag object caches as outdated */
		BKE_ptcache_ids_from_object(&pidlist, obedit, scene, 0);
		for (pid = pidlist.first; pid; pid = pid->next) {
			if (pid->type != PTCACHE_TYPE_PARTICLES) /* particles don't need reset on geometry change */
				pid->cache->flag |= PTCACHE_OUTDATED;
		}
		BLI_freelistN(&pidlist);

		BKE_ptcache_object_reset(scene, obedit, PTCACHE_RESET_OUTDATED);

		/* also flush ob recalc, doesn't take much overhead, but used for particles */
		DEG_id_tag_update(&obedit->id, OB_RECALC_OB | OB_RECALC_DATA);

		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_MODE_OBJECT, scene);

		obedit->mode &= ~OB_MODE_EDIT;
	}

	if (flag & EM_WAITCURSOR) waitcursor(0);

	return (obedit->mode & OB_MODE_EDIT) == 0;
}

bool ED_object_editmode_exit(bContext *C, int flag)
{
	Main *bmain = CTX_data_main(C);
	Scene *scene = CTX_data_scene(C);
	Object *obedit = CTX_data_edit_object(C);
	return ED_object_editmode_exit_ex(bmain, scene, obedit, flag);
}

bool ED_object_editmode_enter_ex(Main *bmain, Scene *scene, Object *ob, int flag)
{
	bool ok = false;

	if (ELEM(NULL, ob, ob->data) || ID_IS_LINKED(ob)) {
		return false;
	}

	/* this checks actual object->data, for cases when other scenes have it in editmode context */
	if (BKE_object_is_in_editmode(ob)) {
		return true;
	}

	if (BKE_object_obdata_is_libdata(ob)) {
		error_libdata();
		return false;
	}

	if (flag & EM_WAITCURSOR) waitcursor(1);

	ob->restore_mode = ob->mode;

	ob->mode = OB_MODE_EDIT;

	if (ob->type == OB_MESH) {
		BMEditMesh *em;
		ok = 1;

		const bool use_key_index = mesh_needs_keyindex(bmain, ob->data);

		EDBM_mesh_make(ob, scene->toolsettings->selectmode, use_key_index);

		em = BKE_editmesh_from_object(ob);
		if (LIKELY(em)) {
			/* order doesn't matter */
			EDBM_mesh_normals_update(em);
			BKE_editmesh_tessface_calc(em);
		}

		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_EDITMODE_MESH, NULL);
	}
	else if (ob->type == OB_ARMATURE) {
		ok = 1;
		ED_armature_to_edit(ob->data);
		/* to ensure all goes in restposition and without striding */
		DEG_id_tag_update(&ob->id, OB_RECALC_OB | OB_RECALC_DATA | OB_RECALC_TIME); /* XXX: should this be OB_RECALC_DATA? */

		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_EDITMODE_ARMATURE, scene);
	}
	else if (ob->type == OB_FONT) {
		ok = 1;
		ED_curve_editfont_make(ob);

		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_EDITMODE_TEXT, scene);
	}
	else if (ob->type == OB_MBALL) {
		ok = 1;
		ED_mball_editmball_make(ob);

		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_EDITMODE_MBALL, scene);
	}
	else if (ob->type == OB_LATTICE) {
		ok = 1;
		BKE_editlattice_make(ob);

		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_EDITMODE_LATTICE, scene);
	}
	else if (ob->type == OB_SURF || ob->type == OB_CURVE) {
		ok = 1;
		ED_curve_editnurb_make(ob);

		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_EDITMODE_CURVE, scene);
	}

	if (ok) {
		DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
	}
	else {
		if ((flag & EM_NO_CONTEXT) == 0) {
			ob->mode &= ~OB_MODE_EDIT;
		}
		WM_main_add_notifier(NC_SCENE | ND_MODE | NS_MODE_OBJECT, scene);
	}

	if (flag & EM_WAITCURSOR) waitcursor(0);

	return (ob->mode & OB_MODE_EDIT) != 0;
}

bool ED_object_editmode_enter(bContext *C, int flag)
{
	Main *bmain = CTX_data_main(C);
	Scene *scene = CTX_data_scene(C);
	ViewLayer *view_layer = CTX_data_view_layer(C);
	Object *ob;

	if ((flag & EM_IGNORE_LAYER) == 0) {
		ob = CTX_data_active_object(C); /* active layer checked here for view3d */
	}
	else {
		ob = view_layer->basact->object;
	}
	if ((ob == NULL) || ID_IS_LINKED(ob)) {
		return false;
	}
	return ED_object_editmode_enter_ex(bmain, scene, ob, flag);
}

static int editmode_toggle_exec(bContext *C, wmOperator *op)
{
	struct wmMsgBus *mbus = CTX_wm_message_bus(C);
	const int mode_flag = OB_MODE_EDIT;
	const bool is_mode_set = (CTX_data_edit_object(C) != NULL);
	Main *bmain = CTX_data_main(C);
	Scene *scene =  CTX_data_scene(C);
	ViewLayer *view_layer = CTX_data_view_layer(C);
	Object *obact = OBACT(view_layer);

	if (!is_mode_set) {
		if (!ED_object_mode_compat_set(C, obact, mode_flag, op->reports)) {
			return OPERATOR_CANCELLED;
		}
	}

	if (!is_mode_set) {
		ED_object_editmode_enter(C, EM_WAITCURSOR);
		if (obact->mode & mode_flag) {
			FOREACH_SELECTED_OBJECT_BEGIN(view_layer, ob)
			{
				if ((ob != obact) && (ob->type == obact->type)) {
					ED_object_editmode_enter_ex(bmain, scene, ob, EM_WAITCURSOR | EM_NO_CONTEXT);
				}
			}
			FOREACH_SELECTED_OBJECT_END;
		}
	}
	else {
		ED_object_editmode_exit(C, EM_FREEDATA | EM_WAITCURSOR);
		if ((obact->mode & mode_flag) == 0) {
			FOREACH_OBJECT_BEGIN(view_layer, ob)
			{
				if ((ob != obact) && (ob->type == obact->type)) {
					ED_object_editmode_exit_ex(bmain, scene, ob, EM_FREEDATA | EM_WAITCURSOR);
				}
			}
			FOREACH_OBJECT_END;
		}
	}

	ED_space_image_uv_sculpt_update(bmain, CTX_wm_manager(C), scene);

	WM_msg_publish_rna_prop(mbus, &obact->id, obact, Object, mode);

	WM_toolsystem_update_from_context_view3d(C);

	return OPERATOR_FINISHED;
}

static bool editmode_toggle_poll(bContext *C)
{
	Object *ob = CTX_data_active_object(C);

	/* covers proxies too */
	if (ELEM(NULL, ob, ob->data) || ID_IS_LINKED(ob->data))
		return 0;

	/* if hidden but in edit mode, we still display */
	if ((ob->restrictflag & OB_RESTRICT_VIEW) && !(ob->mode & OB_MODE_EDIT)) {
		return 0;
	}

	return OB_TYPE_SUPPORT_EDITMODE(ob->type);
}

void OBJECT_OT_editmode_toggle(wmOperatorType *ot)
{

	/* identifiers */
	ot->name = "Toggle Editmode";
	ot->description = "Toggle object's editmode";
	ot->idname = "OBJECT_OT_editmode_toggle";

	/* api callbacks */
	ot->exec = editmode_toggle_exec;
	ot->poll = editmode_toggle_poll;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
}

/* *************************** */

static int posemode_exec(bContext *C, wmOperator *op)
{
	struct wmMsgBus *mbus = CTX_wm_message_bus(C);
	Base *base = CTX_data_active_base(C);
	Object *obact = base->object;
	const int mode_flag = OB_MODE_POSE;
	bool is_mode_set = (obact->mode & mode_flag) != 0;

	if (!is_mode_set) {
		if (!ED_object_mode_compat_set(C, obact, mode_flag, op->reports)) {
			return OPERATOR_CANCELLED;
		}
	}

	if (obact->type != OB_ARMATURE) {
		return OPERATOR_PASS_THROUGH;
	}

	if (obact == CTX_data_edit_object(C)) {
		ED_object_editmode_exit(C, EM_FREEDATA);
		is_mode_set = false;
	}

	if (is_mode_set) {
		bool ok = ED_object_posemode_exit(C, obact);
		if (ok) {
			struct Main *bmain = CTX_data_main(C);
			ViewLayer *view_layer = CTX_data_view_layer(C);
			FOREACH_OBJECT_BEGIN(view_layer, ob)
			{
				if ((ob != obact) &&
				    (ob->type == OB_ARMATURE) &&
				    (ob->mode & mode_flag))
				{
					ED_object_posemode_exit_ex(bmain, ob);
				}
			}
			FOREACH_OBJECT_END;
		}
	}
	else {
		bool ok = ED_object_posemode_enter(C, obact);
		if (ok) {
			struct Main *bmain = CTX_data_main(C);
			ViewLayer *view_layer = CTX_data_view_layer(C);
			FOREACH_SELECTED_OBJECT_BEGIN(view_layer, ob)
			{
				if ((ob != obact) &&
				    (ob->type == OB_ARMATURE) &&
				    (ob->mode == OB_MODE_OBJECT) &&
				    (!ID_IS_LINKED(ob)))
				{
					ED_object_posemode_enter_ex(bmain, ob);
				}
			}
			FOREACH_SELECTED_OBJECT_END;
		}
	}

	WM_msg_publish_rna_prop(mbus, &obact->id, obact, Object, mode);

	WM_toolsystem_update_from_context_view3d(C);

	return OPERATOR_FINISHED;
}

void OBJECT_OT_posemode_toggle(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Toggle Pose Mode";
	ot->idname = "OBJECT_OT_posemode_toggle";
	ot->description = "Enable or disable posing/selecting bones";

	/* api callbacks */
	ot->exec = posemode_exec;
	ot->poll = ED_operator_object_active_editable;

	/* flag */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
}

/* both pointers should exist */
static void copy_texture_space(Object *to, Object *ob)
{
	float *poin1 = NULL, *poin2 = NULL;
	short texflag = 0;

	if (ob->type == OB_MESH) {
		texflag = ((Mesh *)ob->data)->texflag;
		poin2 = ((Mesh *)ob->data)->loc;
	}
	else if (ELEM(ob->type, OB_CURVE, OB_SURF, OB_FONT)) {
		texflag = ((Curve *)ob->data)->texflag;
		poin2 = ((Curve *)ob->data)->loc;
	}
	else if (ob->type == OB_MBALL) {
		texflag = ((MetaBall *)ob->data)->texflag;
		poin2 = ((MetaBall *)ob->data)->loc;
	}
	else
		return;

	if (to->type == OB_MESH) {
		((Mesh *)to->data)->texflag = texflag;
		poin1 = ((Mesh *)to->data)->loc;
	}
	else if (ELEM(to->type, OB_CURVE, OB_SURF, OB_FONT)) {
		((Curve *)to->data)->texflag = texflag;
		poin1 = ((Curve *)to->data)->loc;
	}
	else if (to->type == OB_MBALL) {
		((MetaBall *)to->data)->texflag = texflag;
		poin1 = ((MetaBall *)to->data)->loc;
	}
	else
		return;

	memcpy(poin1, poin2, 9 * sizeof(float));  /* this was noted in DNA_mesh, curve, mball */

	if (to->type == OB_MESH) {
		/* pass */
	}
	else if (to->type == OB_MBALL) {
		BKE_mball_texspace_calc(to);
	}
	else {
		BKE_curve_texspace_calc(to->data);
	}

}

/* UNUSED, keep in case we want to copy functionality for use elsewhere */
static void copy_attr(Main *bmain, Scene *scene, ViewLayer *view_layer, short event)
{
	Object *ob;
	Base *base;
	Curve *cu, *cu1;
	Nurb *nu;

	if (ID_IS_LINKED(scene)) return;

	if (!(ob = OBACT(view_layer))) return;

	if (BKE_object_is_in_editmode(ob)) {
		/* obedit_copymenu(); */
		return;
	}

	if (event == 24) {
		/* moved to BKE_object_link_modifiers */
		/* copymenu_modifiers(bmain, scene, v3d, ob); */
		return;
	}

	for (base = FIRSTBASE(view_layer); base; base = base->next) {
		if (base != BASACT(view_layer)) {
			if (TESTBASELIB(base)) {
				DEG_id_tag_update(&base->object->id, OB_RECALC_DATA);

				if (event == 1) {  /* loc */
					copy_v3_v3(base->object->loc, ob->loc);
					copy_v3_v3(base->object->dloc, ob->dloc);
				}
				else if (event == 2) {  /* rot */
					copy_v3_v3(base->object->rot, ob->rot);
					copy_v3_v3(base->object->drot, ob->drot);

					copy_qt_qt(base->object->quat, ob->quat);
					copy_qt_qt(base->object->dquat, ob->dquat);
				}
				else if (event == 3) {  /* size */
					copy_v3_v3(base->object->size, ob->size);
					copy_v3_v3(base->object->dscale, ob->dscale);
				}
				else if (event == 4) {  /* drawtype */
					base->object->dt = ob->dt;
					base->object->dtx = ob->dtx;
					base->object->empty_drawtype = ob->empty_drawtype;
					base->object->empty_drawsize = ob->empty_drawsize;
				}
				else if (event == 5) {  /* time offs */
					base->object->sf = ob->sf;
				}
				else if (event == 6) {  /* dupli */
					base->object->dupon = ob->dupon;
					base->object->dupoff = ob->dupoff;
					base->object->dupsta = ob->dupsta;
					base->object->dupend = ob->dupend;

					base->object->transflag &= ~OB_DUPLI;
					base->object->transflag |= (ob->transflag & OB_DUPLI);

					base->object->dup_group = ob->dup_group;
					if (ob->dup_group)
						id_us_plus(&ob->dup_group->id);
				}
				else if (event == 17) {   /* tex space */
					copy_texture_space(base->object, ob);
				}
				else if (event == 18) {   /* font settings */

					if (base->object->type == ob->type) {
						cu = ob->data;
						cu1 = base->object->data;

						cu1->spacemode = cu->spacemode;
						cu1->align_y = cu->align_y;
						cu1->spacing = cu->spacing;
						cu1->linedist = cu->linedist;
						cu1->shear = cu->shear;
						cu1->fsize = cu->fsize;
						cu1->xof = cu->xof;
						cu1->yof = cu->yof;
						cu1->textoncurve = cu->textoncurve;
						cu1->wordspace = cu->wordspace;
						cu1->ulpos = cu->ulpos;
						cu1->ulheight = cu->ulheight;
						if (cu1->vfont)
							id_us_min(&cu1->vfont->id);
						cu1->vfont = cu->vfont;
						id_us_plus((ID *)cu1->vfont);
						if (cu1->vfontb)
							id_us_min(&cu1->vfontb->id);
						cu1->vfontb = cu->vfontb;
						id_us_plus((ID *)cu1->vfontb);
						if (cu1->vfonti)
							id_us_min(&cu1->vfonti->id);
						cu1->vfonti = cu->vfonti;
						id_us_plus((ID *)cu1->vfonti);
						if (cu1->vfontbi)
							id_us_min(&cu1->vfontbi->id);
						cu1->vfontbi = cu->vfontbi;
						id_us_plus((ID *)cu1->vfontbi);

						BLI_strncpy(cu1->family, cu->family, sizeof(cu1->family));

						DEG_id_tag_update(&base->object->id, OB_RECALC_DATA);
					}
				}
				else if (event == 19) {   /* bevel settings */

					if (ELEM(base->object->type, OB_CURVE, OB_FONT)) {
						cu = ob->data;
						cu1 = base->object->data;

						cu1->bevobj = cu->bevobj;
						cu1->taperobj = cu->taperobj;
						cu1->width = cu->width;
						cu1->bevresol = cu->bevresol;
						cu1->ext1 = cu->ext1;
						cu1->ext2 = cu->ext2;

						DEG_id_tag_update(&base->object->id, OB_RECALC_DATA);
					}
				}
				else if (event == 25) {   /* curve resolution */

					if (ELEM(base->object->type, OB_CURVE, OB_FONT)) {
						cu = ob->data;
						cu1 = base->object->data;

						cu1->resolu = cu->resolu;
						cu1->resolu_ren = cu->resolu_ren;

						nu = cu1->nurb.first;

						while (nu) {
							nu->resolu = cu1->resolu;
							nu = nu->next;
						}

						DEG_id_tag_update(&base->object->id, OB_RECALC_DATA);
					}
				}
				else if (event == 21) {
					if (base->object->type == OB_MESH) {
						ModifierData *md = modifiers_findByType(ob, eModifierType_Subsurf);

						if (md) {
							ModifierData *tmd = modifiers_findByType(base->object, eModifierType_Subsurf);

							if (!tmd) {
								tmd = modifier_new(eModifierType_Subsurf);
								BLI_addtail(&base->object->modifiers, tmd);
							}

							modifier_copyData(md, tmd);
							DEG_id_tag_update(&base->object->id, OB_RECALC_DATA);
						}
					}
				}
				else if (event == 22) {
					/* Copy the constraint channels over */
					BKE_constraints_copy(&base->object->constraints, &ob->constraints, true);
					DEG_id_tag_update(&base->object->id, DEG_TAG_COPY_ON_WRITE);
					DEG_relations_tag_update(bmain);
				}
				else if (event == 23) {
					base->object->softflag = ob->softflag;
					if (base->object->soft) sbFree(base->object->soft);

					base->object->soft = copy_softbody(ob->soft, 0);

					if (!modifiers_findByType(base->object, eModifierType_Softbody)) {
						BLI_addhead(&base->object->modifiers, modifier_new(eModifierType_Softbody));
					}

					DEG_id_tag_update(&base->object->id, DEG_TAG_COPY_ON_WRITE);
					DEG_relations_tag_update(bmain);
				}
				else if (event == 26) {
#if 0 // XXX old animation system
					BKE_nlastrip_copy(s(&base->object->nlastrips, &ob->nlastrips);
#endif // XXX old animation system
				}
				else if (event == 27) {   /* autosmooth */
					if (base->object->type == OB_MESH) {
						Mesh *me = ob->data;
						Mesh *cme = base->object->data;
						cme->smoothresh = me->smoothresh;
						if (me->flag & ME_AUTOSMOOTH)
							cme->flag |= ME_AUTOSMOOTH;
						else
							cme->flag &= ~ME_AUTOSMOOTH;
					}
				}
				else if (event == 28) { /* UV orco */
					if (ELEM(base->object->type, OB_CURVE, OB_SURF)) {
						cu = ob->data;
						cu1 = base->object->data;

						if (cu->flag & CU_UV_ORCO)
							cu1->flag |= CU_UV_ORCO;
						else
							cu1->flag &= ~CU_UV_ORCO;
					}
				}
				else if (event == 29) { /* protected bits */
					base->object->protectflag = ob->protectflag;
				}
				else if (event == 30) { /* index object */
					base->object->index = ob->index;
				}
				else if (event == 31) { /* object color */
					copy_v4_v4(base->object->col, ob->col);
				}
			}
		}
	}
}

static void UNUSED_FUNCTION(copy_attr_menu) (Main *bmain, Scene *scene, ViewLayer *view_layer, Object *obedit)
{
	Object *ob;
	short event;
	char str[512];

	if (!(ob = OBACT(view_layer))) return;

	if (obedit) {
/*		if (ob->type == OB_MESH) */
/* XXX			mesh_copy_menu(); */
		return;
	}

	/* Object Mode */

	/* If you change this menu, don't forget to update the menu in header_view3d.c
	 * view3d_edit_object_copyattrmenu() and in toolbox.c
	 */

	strcpy(str,
	       "Copy Attributes %t|Location %x1|Rotation %x2|Size %x3|Draw Options %x4|"
	       "Time Offset %x5|Dupli %x6|Object Color %x31|%l|Mass %x7|Damping %x8|All Physical Attributes %x11|Properties %x9|"
	       "Logic Bricks %x10|Protected Transform %x29|%l");

	strcat(str, "|Object Constraints %x22");
	strcat(str, "|NLA Strips %x26");

/* XXX	if (OB_TYPE_SUPPORT_MATERIAL(ob->type)) { */
/*		strcat(str, "|Texture Space %x17"); */
/*	} */

	if (ob->type == OB_FONT) strcat(str, "|Font Settings %x18|Bevel Settings %x19");
	if (ob->type == OB_CURVE) strcat(str, "|Bevel Settings %x19|UV Orco %x28");

	if ((ob->type == OB_FONT) || (ob->type == OB_CURVE)) {
		strcat(str, "|Curve Resolution %x25");
	}

	if (ob->type == OB_MESH) {
		strcat(str, "|Subsurf Settings %x21|AutoSmooth %x27");
	}

	if (ob->soft) strcat(str, "|Soft Body Settings %x23");

	strcat(str, "|Pass Index %x30");

	if (ob->type == OB_MESH || ob->type == OB_CURVE || ob->type == OB_LATTICE || ob->type == OB_SURF) {
		strcat(str, "|Modifiers ... %x24");
	}

	event = pupmenu(str);
	if (event <= 0) return;

	copy_attr(bmain, scene, view_layer, event);
}

/* ******************* force field toggle operator ***************** */

void ED_object_check_force_modifiers(Main *bmain, Scene *scene, Object *object)
{
	PartDeflect *pd = object->pd;
	ModifierData *md = modifiers_findByType(object, eModifierType_Surface);

	/* add/remove modifier as needed */
	if (!md) {
		if (pd && (pd->shape == PFIELD_SHAPE_SURFACE) && !ELEM(pd->forcefield, 0, PFIELD_GUIDE, PFIELD_TEXTURE)) {
			if (ELEM(object->type, OB_MESH, OB_SURF, OB_FONT, OB_CURVE)) {
				ED_object_modifier_add(NULL, bmain, scene, object, NULL, eModifierType_Surface);
			}
		}
	}
	else {
		if (!pd || (pd->shape != PFIELD_SHAPE_SURFACE) || ELEM(pd->forcefield, 0, PFIELD_GUIDE, PFIELD_TEXTURE)) {
			ED_object_modifier_remove(NULL, bmain, object, md);
		}
	}
}

static int forcefield_toggle_exec(bContext *C, wmOperator *UNUSED(op))
{
	Object *ob = CTX_data_active_object(C);

	if (ob->pd == NULL)
		ob->pd = object_add_collision_fields(PFIELD_FORCE);
	else if (ob->pd->forcefield == 0)
		ob->pd->forcefield = PFIELD_FORCE;
	else
		ob->pd->forcefield = 0;

	ED_object_check_force_modifiers(CTX_data_main(C), CTX_data_scene(C), ob);
	WM_event_add_notifier(C, NC_OBJECT | ND_DRAW, ob);
	WM_event_add_notifier(C, NC_OBJECT | ND_MODIFIER, ob);

	return OPERATOR_FINISHED;
}

void OBJECT_OT_forcefield_toggle(wmOperatorType *ot)
{

	/* identifiers */
	ot->name = "Toggle Force Field";
	ot->description = "Toggle object's force field";
	ot->idname = "OBJECT_OT_forcefield_toggle";

	/* api callbacks */
	ot->exec = forcefield_toggle_exec;
	ot->poll = ED_operator_object_active_editable;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
}

/* ********************************************** */
/* Motion Paths */

/* For the objects with animation: update paths for those that have got them
 * This should selectively update paths that exist...
 *
 * To be called from various tools that do incremental updates
 */
void ED_objects_recalculate_paths(bContext *C, Scene *scene)
{
	struct Main *bmain = CTX_data_main(C);
	Depsgraph *depsgraph = CTX_data_depsgraph(C);
	ListBase targets = {NULL, NULL};

	/* loop over objects in scene */
	CTX_DATA_BEGIN(C, Object *, ob, selected_editable_objects)
	{
		/* set flag to force recalc, then grab path(s) from object */
		ob->avs.recalc |= ANIMVIZ_RECALC_PATHS;
		animviz_get_object_motionpaths(ob, &targets);
	}
	CTX_DATA_END;

	/* recalculate paths, then free */
	animviz_calc_motionpaths(depsgraph, bmain, scene, &targets);
	BLI_freelistN(&targets);

	/* tag objects for copy on write - so paths will draw/redraw */
	CTX_DATA_BEGIN(C, Object *, ob, selected_editable_objects)
	{
		if (ob->mpath) {
			DEG_id_tag_update(&ob->id, DEG_TAG_COPY_ON_WRITE);
		}
	}
	CTX_DATA_END;
}


/* show popup to determine settings */
static int object_calculate_paths_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
{
	Object *ob = CTX_data_active_object(C);

	if (ob == NULL)
		return OPERATOR_CANCELLED;

	/* set default settings from existing/stored settings */
	{
		bAnimVizSettings *avs = &ob->avs;

		RNA_int_set(op->ptr, "start_frame", avs->path_sf);
		RNA_int_set(op->ptr, "end_frame", avs->path_ef);
	}

	/* show popup dialog to allow editing of range... */
	/* FIXME: hardcoded dimensions here are just arbitrary */
	return WM_operator_props_dialog_popup(C, op, 10 * UI_UNIT_X, 10 * UI_UNIT_Y);
}

/* Calculate/recalculate whole paths (avs.path_sf to avs.path_ef) */
static int object_calculate_paths_exec(bContext *C, wmOperator *op)
{
	Scene *scene = CTX_data_scene(C);
	int start = RNA_int_get(op->ptr, "start_frame");
	int end = RNA_int_get(op->ptr, "end_frame");

	/* set up path data for bones being calculated */
	CTX_DATA_BEGIN(C, Object *, ob, selected_editable_objects)
	{
		bAnimVizSettings *avs = &ob->avs;

		/* grab baking settings from operator settings */
		avs->path_sf = start;
		avs->path_ef = end;

		/* verify that the selected object has the appropriate settings */
		animviz_verify_motionpaths(op->reports, scene, ob, NULL);
	}
	CTX_DATA_END;

	/* calculate the paths for objects that have them (and are tagged to get refreshed) */
	ED_objects_recalculate_paths(C, scene);

	/* notifiers for updates */
	WM_event_add_notifier(C, NC_OBJECT | ND_TRANSFORM, NULL);

	return OPERATOR_FINISHED;
}

void OBJECT_OT_paths_calculate(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Calculate Object Paths";
	ot->idname = "OBJECT_OT_paths_calculate";
	ot->description = "Calculate motion paths for the selected objects";

	/* api callbacks */
	ot->invoke = object_calculate_paths_invoke;
	ot->exec = object_calculate_paths_exec;
	ot->poll = ED_operator_object_active_editable;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;

	/* properties */
	RNA_def_int(ot->srna, "start_frame", 1, MINAFRAME, MAXFRAME, "Start",
	            "First frame to calculate object paths on", MINFRAME, MAXFRAME / 2.0);
	RNA_def_int(ot->srna, "end_frame", 250, MINAFRAME, MAXFRAME, "End",
	            "Last frame to calculate object paths on", MINFRAME, MAXFRAME / 2.0);
}

/* --------- */

static bool object_update_paths_poll(bContext *C)
{
	if (ED_operator_object_active_editable(C)) {
		Object *ob = ED_object_active_context(C);
		return (ob->avs.path_bakeflag & MOTIONPATH_BAKE_HAS_PATHS) != 0;
	}

	return false;
}

static int object_update_paths_exec(bContext *C, wmOperator *UNUSED(op))
{
	Scene *scene = CTX_data_scene(C);

	if (scene == NULL)
		return OPERATOR_CANCELLED;

	/* calculate the paths for objects that have them (and are tagged to get refreshed) */
	ED_objects_recalculate_paths(C, scene);

	/* notifiers for updates */
	WM_event_add_notifier(C, NC_OBJECT | ND_TRANSFORM, NULL);

	return OPERATOR_FINISHED;
}

void OBJECT_OT_paths_update(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Update Object Paths";
	ot->idname = "OBJECT_OT_paths_update";
	ot->description = "Recalculate paths for selected objects";

	/* api callbakcs */
	ot->exec = object_update_paths_exec;
	ot->poll = object_update_paths_poll;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
}

/* --------- */

/* Helper for ED_objects_clear_paths() */
static void object_clear_mpath(Object *ob)
{
	if (ob->mpath) {
		animviz_free_motionpath(ob->mpath);
		ob->mpath = NULL;
		ob->avs.path_bakeflag &= ~MOTIONPATH_BAKE_HAS_PATHS;
	}
}

/* Clear motion paths for all objects */
void ED_objects_clear_paths(bContext *C, bool only_selected)
{
	if (only_selected) {
		/* loop over all selected + sedtiable objects in scene */
		CTX_DATA_BEGIN(C, Object *, ob, selected_editable_objects)
		{
			object_clear_mpath(ob);
		}
		CTX_DATA_END;
	}
	else {
		/* loop over all edtiable objects in scene */
		CTX_DATA_BEGIN(C, Object *, ob, editable_objects)
		{
			object_clear_mpath(ob);
		}
		CTX_DATA_END;
	}
}

/* operator callback for this */
static int object_clear_paths_exec(bContext *C, wmOperator *op)
{
	bool only_selected = RNA_boolean_get(op->ptr, "only_selected");

	/* use the backend function for this */
	ED_objects_clear_paths(C, only_selected);

	/* notifiers for updates */
	WM_event_add_notifier(C, NC_OBJECT | ND_TRANSFORM, NULL);

	return OPERATOR_FINISHED;
}

/* operator callback/wrapper */
static int object_clear_paths_invoke(bContext *C, wmOperator *op, const wmEvent *evt)
{
	if ((evt->shift) && !RNA_struct_property_is_set(op->ptr, "only_selected")) {
		RNA_boolean_set(op->ptr, "only_selected", true);
	}
	return object_clear_paths_exec(C, op);
}

void OBJECT_OT_paths_clear(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Clear Object Paths";
	ot->idname = "OBJECT_OT_paths_clear";
	ot->description = "Clear path caches for all objects, hold Shift key for selected objects only";

	/* api callbacks */
	ot->invoke = object_clear_paths_invoke;
	ot->exec = object_clear_paths_exec;
	ot->poll = ED_operator_object_active_editable;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;

	/* properties */
	ot->prop = RNA_def_boolean(ot->srna, "only_selected", false, "Only Selected",
	                           "Only clear paths from selected objects");
	RNA_def_property_flag(ot->prop, PROP_SKIP_SAVE);
}


/********************** Smooth/Flat *********************/

static int shade_smooth_exec(bContext *C, wmOperator *op)
{
	ID *data;
	Curve *cu;
	Nurb *nu;
	int clear = (STREQ(op->idname, "OBJECT_OT_shade_flat"));
	bool done = false, linked_data = false;

	CTX_DATA_BEGIN(C, Object *, ob, selected_editable_objects)
	{
		data = ob->data;

		if (data && ID_IS_LINKED(data)) {
			linked_data = true;
			continue;
		}

		if (ob->type == OB_MESH) {
			BKE_mesh_smooth_flag_set(ob, !clear);

			BKE_mesh_batch_cache_dirty(ob->data, BKE_MESH_BATCH_DIRTY_ALL);
			DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
			WM_event_add_notifier(C, NC_OBJECT | ND_DRAW, ob);

			done = true;
		}
		else if (ELEM(ob->type, OB_SURF, OB_CURVE)) {
			cu = ob->data;

			for (nu = cu->nurb.first; nu; nu = nu->next) {
				if (!clear) nu->flag |= ME_SMOOTH;
				else nu->flag &= ~ME_SMOOTH;
			}

			DEG_id_tag_update(&ob->id, OB_RECALC_DATA);
			WM_event_add_notifier(C, NC_OBJECT | ND_DRAW, ob);

			done = true;
		}
	}
	CTX_DATA_END;

	if (linked_data)
		BKE_report(op->reports, RPT_WARNING, "Can't edit linked mesh or curve data");

	return (done) ? OPERATOR_FINISHED : OPERATOR_CANCELLED;
}

static bool shade_poll(bContext *C)
{
	return (CTX_data_edit_object(C) == NULL);
}

void OBJECT_OT_shade_flat(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Shade Flat";
	ot->description = "Render and display faces uniform, using Face Normals";
	ot->idname = "OBJECT_OT_shade_flat";

	/* api callbacks */
	ot->poll = shade_poll;
	ot->exec = shade_smooth_exec;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
}

void OBJECT_OT_shade_smooth(wmOperatorType *ot)
{
	/* identifiers */
	ot->name = "Shade Smooth";
	ot->description = "Render and display faces smooth, using interpolated Vertex Normals";
	ot->idname = "OBJECT_OT_shade_smooth";

	/* api callbacks */
	ot->poll = shade_poll;
	ot->exec = shade_smooth_exec;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
}

/* ********************** */

static const EnumPropertyItem *object_mode_set_itemsf(
        bContext *C, PointerRNA *UNUSED(ptr), PropertyRNA *UNUSED(prop), bool *r_free)
{
	const EnumPropertyItem *input = rna_enum_object_mode_items;
	EnumPropertyItem *item = NULL;
	Object *ob;
	bGPdata *gpd;
	int totitem = 0;

	if (!C) /* needed for docs */
		return rna_enum_object_mode_items;

	ob = CTX_data_active_object(C);
	if (ob) {
		const bool use_mode_particle_edit = (BLI_listbase_is_empty(&ob->particlesystem) == false) ||
		                                    (ob->soft != NULL) ||
		                                    (modifiers_findByType(ob, eModifierType_Cloth) != NULL);
		while (input->identifier) {
			if ((input->value == OB_MODE_EDIT && OB_TYPE_SUPPORT_EDITMODE(ob->type)) ||
			    (input->value == OB_MODE_POSE && (ob->type == OB_ARMATURE)) ||
			    (input->value == OB_MODE_PARTICLE_EDIT && use_mode_particle_edit) ||
			    (ELEM(input->value, OB_MODE_SCULPT, OB_MODE_VERTEX_PAINT,
			           OB_MODE_WEIGHT_PAINT, OB_MODE_TEXTURE_PAINT) && (ob->type == OB_MESH)) ||
			    (input->value == OB_MODE_OBJECT))
			{
				RNA_enum_item_add(&item, &totitem, input);
			}
			input++;
		}
	}
	else {
		/* We need at least this one! */
		RNA_enum_items_add_value(&item, &totitem, input, OB_MODE_OBJECT);
	}

	/* On top of all the rest, GPencil Stroke Edit Mode
	 * is available if there's a valid gp datablock...
	 */
	gpd = CTX_data_gpencil_data(C);
	if (gpd) {
		RNA_enum_items_add_value(&item, &totitem, rna_enum_object_mode_items, OB_MODE_GPENCIL);
	}

	RNA_enum_item_end(&item, &totitem);

	*r_free = true;

	return item;
}

static bool object_mode_set_poll(bContext *C)
{
	/* Since Grease Pencil editmode is also handled here,
	 * we have a special exception for allowing this operator
	 * to still work in that case when there's no active object
	 * so that users can exit editmode this way as per normal.
	 */
	if (ED_operator_object_active_editable(C))
		return true;
	else
		return (CTX_data_gpencil_data(C) != NULL);
}

static int object_mode_set_exec(bContext *C, wmOperator *op)
{
	bool use_submode = STREQ(op->idname, "OBJECT_OT_mode_set_or_submode");
	Object *ob = CTX_data_active_object(C);
	bGPdata *gpd = CTX_data_gpencil_data(C);
	eObjectMode mode = RNA_enum_get(op->ptr, "mode");
	eObjectMode restore_mode = (ob) ? ob->mode : OB_MODE_OBJECT;
	const bool toggle = RNA_boolean_get(op->ptr, "toggle");

	if (use_submode) {
		/* When not changing modes use submodes, see: T55162. */
		if (toggle == false) {
			if (mode == restore_mode) {
				switch (mode) {
					case OB_MODE_EDIT:
						WM_menu_name_call(C, "VIEW3D_MT_edit_mesh_select_mode", WM_OP_INVOKE_REGION_WIN);
						return OPERATOR_INTERFACE;
					default:
						break;
				}
			}
		}
	}

	if (gpd) {
		/* GP Mode is not bound to a specific object. Therefore,
		 * we don't want it to be actually saved on any objects,
		 * as weirdness can happen if you select other objects,
		 * or load old files.
		 *
		 * Instead, we use the following 2 rules to ensure that
		 * the mode selector works as expected:
		 *  1) If there's no object, we want to enter editmode.
		 *     (i.e. with no object, we're in object mode)
		 *  2) Otherwise, exit stroke editmode, so that we can
		 *     enter another mode...
		 */
		if (!ob || (gpd->flag & GP_DATA_STROKE_EDITMODE)) {
			WM_operator_name_call(C, "GPENCIL_OT_editmode_toggle", WM_OP_EXEC_REGION_WIN, NULL);
		}
	}

	if (!ob || !ED_object_mode_compat_test(ob, mode))
		return OPERATOR_PASS_THROUGH;

	if (ob->mode != mode) {
		/* we should be able to remove this call, each operator calls  */
		ED_object_mode_compat_set(C, ob, mode, op->reports);
	}

	/* Exit current mode if it's not the mode we're setting */
	if (mode != OB_MODE_OBJECT && (ob->mode != mode || toggle)) {
		/* Enter new mode */
		ED_object_mode_toggle(C, mode);
	}

	if (toggle) {
		/* Special case for Object mode! */
		if (mode == OB_MODE_OBJECT && restore_mode == OB_MODE_OBJECT && ob->restore_mode != OB_MODE_OBJECT) {
			ED_object_mode_toggle(C, ob->restore_mode);
		}
		else if (ob->mode == mode) {
			/* For toggling, store old mode so we know what to go back to */
			ob->restore_mode = restore_mode;
		}
		else if (ob->restore_mode != OB_MODE_OBJECT && ob->restore_mode != mode) {
			ED_object_mode_toggle(C, ob->restore_mode);
		}
	}

	return OPERATOR_FINISHED;
}

void OBJECT_OT_mode_set(wmOperatorType *ot)
{
	PropertyRNA *prop;

	/* identifiers */
	ot->name = "Set Object Mode";
	ot->description = "Sets the object interaction mode";
	ot->idname = "OBJECT_OT_mode_set";

	/* api callbacks */
	ot->exec = object_mode_set_exec;

	ot->poll = object_mode_set_poll; //ED_operator_object_active_editable;

	/* flags */
	ot->flag = 0; /* no register/undo here, leave it to operators being called */

	ot->prop = RNA_def_enum(ot->srna, "mode", rna_enum_object_mode_items, OB_MODE_OBJECT, "Mode", "");
	RNA_def_enum_funcs(ot->prop, object_mode_set_itemsf);
	RNA_def_property_flag(ot->prop, PROP_SKIP_SAVE);

	prop = RNA_def_boolean(ot->srna, "toggle", 0, "Toggle", "");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE);
}

void OBJECT_OT_mode_set_or_submode(wmOperatorType *ot)
{
	PropertyRNA *prop;

	/* identifiers */
	ot->name = "Set Object Mode or Submode";
	ot->description = "Sets the object interaction mode";
	ot->idname = "OBJECT_OT_mode_set_or_submode";

	/* api callbacks */
	ot->exec = object_mode_set_exec;

	ot->poll = object_mode_set_poll; //ED_operator_object_active_editable;

	/* flags */
	ot->flag = 0; /* no register/undo here, leave it to operators being called */

	ot->prop = RNA_def_enum(ot->srna, "mode", rna_enum_object_mode_items, OB_MODE_OBJECT, "Mode", "");
	RNA_def_enum_funcs(ot->prop, object_mode_set_itemsf);
	RNA_def_property_flag(ot->prop, PROP_SKIP_SAVE);

	prop = RNA_def_boolean(ot->srna, "toggle", 0, "Toggle", "");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE);
}

bool ED_object_editmode_calc_active_center(Object *obedit, const bool select_only, float r_center[3])
{
	switch (obedit->type) {
		case OB_MESH:
		{
			BMEditMesh *em = BKE_editmesh_from_object(obedit);
			BMEditSelection ese;

			if (BM_select_history_active_get(em->bm, &ese)) {
				BM_editselection_center(&ese, r_center);
				return true;
			}
			break;
		}
		case OB_ARMATURE:
		{
			bArmature *arm = obedit->data;
			EditBone *ebo = arm->act_edbone;

			if (ebo && (!select_only || (ebo->flag & (BONE_SELECTED | BONE_ROOTSEL)))) {
				copy_v3_v3(r_center, ebo->head);
				return true;
			}

			break;
		}
		case OB_CURVE:
		case OB_SURF:
		{
			Curve *cu = obedit->data;

			if (ED_curve_active_center(cu, r_center)) {
				return true;
			}
			break;
		}
		case OB_MBALL:
		{
			MetaBall *mb = obedit->data;
			MetaElem *ml_act = mb->lastelem;

			if (ml_act && (!select_only || (ml_act->flag & SELECT))) {
				copy_v3_v3(r_center, &ml_act->x);
				return true;
			}
			break;
		}
		case OB_LATTICE:
		{
			BPoint *actbp = BKE_lattice_active_point_get(obedit->data);

			if (actbp) {
				copy_v3_v3(r_center, actbp->vec);
				return true;
			}
			break;
		}
	}

	return false;
}

static bool move_to_collection_poll(bContext *C)
{
	if (CTX_wm_space_outliner(C) != NULL) {
		return ED_outliner_collections_editor_poll(C);
	}
	else {
		return ED_operator_object_active_editable(C);
	}
}

static int move_to_collection_exec(bContext *C, wmOperator *op)
{
	Main *bmain = CTX_data_main(C);
	Scene *scene = CTX_data_scene(C);
	PropertyRNA *prop = RNA_struct_find_property(op->ptr, "collection_index");
	const bool is_link = STREQ(op->idname, "OBJECT_OT_link_to_collection");
	const bool is_new = RNA_boolean_get(op->ptr, "is_new");
	Collection *collection;
	ListBase objects = {NULL};

	if (!RNA_property_is_set(op->ptr, prop)) {
		BKE_report(op->reports, RPT_ERROR, "No collection selected");
		return OPERATOR_CANCELLED;
	}

	int collection_index = RNA_property_int_get(op->ptr, prop);
	collection = BKE_collection_from_index(CTX_data_scene(C), collection_index);
	if (collection == NULL) {
		BKE_report(op->reports, RPT_ERROR, "Unexpected error, collection not found");
		return OPERATOR_CANCELLED;
	}

	if (CTX_wm_space_outliner(C) != NULL) {
		ED_outliner_selected_objects_get(C, &objects);
	}
	else {
		CTX_DATA_BEGIN (C, Object *, ob, selected_objects)
		{
			BLI_addtail(&objects, BLI_genericNodeN(ob));
		}
		CTX_DATA_END;
	}

	if (is_new) {
		char new_collection_name[MAX_NAME];
		RNA_string_get(op->ptr, "new_collection_name", new_collection_name);
		collection = BKE_collection_add(bmain, collection, new_collection_name);
	}

	Object *single_object = BLI_listbase_is_single(&objects) ?
	                            ((LinkData *)objects.first)->data : NULL;

	if ((single_object != NULL) &&
	    is_link &&
	    BLI_findptr(&collection->gobject, single_object, offsetof(CollectionObject, ob)))
	{
		BKE_reportf(op->reports, RPT_ERROR, "%s already in %s", single_object->id.name + 2, collection->id.name + 2);
		BLI_freelistN(&objects);
		return OPERATOR_CANCELLED;
	}

	for (LinkData *link = objects.first; link; link = link->next) {
		Object *ob = link->data;

		if (!is_link) {
			BKE_collection_object_move(bmain, scene, collection, NULL, ob);
		}
		else {
			BKE_collection_object_add(bmain, collection, ob);
		}
	}
	BLI_freelistN(&objects);

	BKE_reportf(op->reports,
	            RPT_INFO,
	            "%s %s to %s",
	            (single_object != NULL) ? single_object->id.name + 2 : "Objects",
	            is_link ? "linked" : "moved",
	            collection->id.name + 2);

	DEG_relations_tag_update(bmain);
	DEG_id_tag_update(&scene->id, DEG_TAG_COPY_ON_WRITE | DEG_TAG_SELECT_UPDATE);

	WM_event_add_notifier(C, NC_SCENE | ND_LAYER, scene);
	WM_event_add_notifier(C, NC_SCENE | ND_OB_ACTIVE, scene);
	WM_event_add_notifier(C, NC_SCENE | ND_LAYER_CONTENT, scene);

	return OPERATOR_FINISHED;
}

struct MoveToCollectionData {
	struct MoveToCollectionData *next, *prev;
	int index;
	struct Collection *collection;
	struct ListBase submenus;
	PointerRNA ptr;
	struct wmOperatorType *ot;
};

static int move_to_collection_menus_create(wmOperator *op, MoveToCollectionData *menu)
{
	int index = menu->index;
	for (CollectionChild *child = menu->collection->children.first;
	     child != NULL;
	     child = child->next)
	{
		Collection *collection = child->collection;
		MoveToCollectionData *submenu = MEM_callocN(sizeof(MoveToCollectionData),
		                                            "MoveToCollectionData submenu - expected memleak");
		BLI_addtail(&menu->submenus, submenu);
		submenu->collection = collection;
		submenu->index = ++index;
		index = move_to_collection_menus_create(op, submenu);
		submenu->ot = op->type;
	}
	return index;
}

static void move_to_collection_menus_free_recursive(MoveToCollectionData *menu)
{
	for (MoveToCollectionData *submenu = menu->submenus.first;
	     submenu != NULL;
	     submenu = submenu->next)
	{
		move_to_collection_menus_free_recursive(submenu);
	}
	BLI_freelistN(&menu->submenus);
}

static void move_to_collection_menus_free(MoveToCollectionData **menu)
{
	if (*menu == NULL) {
		return;
	}

	move_to_collection_menus_free_recursive(*menu);
	MEM_freeN(*menu);
	*menu = NULL;
}

static void move_to_collection_menu_create(bContext *UNUSED(C), uiLayout *layout, void *menu_v)
{
	MoveToCollectionData *menu = menu_v;
	const char *name;

	if (menu->collection->flag & COLLECTION_IS_MASTER) {
		name = IFACE_("Scene Collection");
	}
	else {
		name = menu->collection->id.name + 2;
	}

	uiItemIntO(layout,
	           name,
	           ICON_NONE,
	           menu->ot->idname,
	           "collection_index",
	           menu->index);
	uiItemS(layout);

	for (MoveToCollectionData *submenu = menu->submenus.first;
	     submenu != NULL;
	     submenu = submenu->next)
	{
		move_to_collection_menus_items(layout, submenu);
	}

	uiItemS(layout);

	WM_operator_properties_create_ptr(&menu->ptr, menu->ot);
	RNA_int_set(&menu->ptr, "collection_index", menu->index);
	RNA_boolean_set(&menu->ptr, "is_new", true);

	uiItemFullO_ptr(layout,
	                menu->ot,
	                "New Collection",
	                ICON_ZOOMIN,
	                menu->ptr.data,
	                WM_OP_INVOKE_DEFAULT,
	                0,
	                NULL);
}

static void move_to_collection_menus_items(uiLayout *layout, MoveToCollectionData *menu)
{
	if (BLI_listbase_is_empty(&menu->submenus)) {
		uiItemIntO(layout,
		           menu->collection->id.name + 2,
		           ICON_NONE,
		           menu->ot->idname,
		           "collection_index",
		           menu->index);
	}
	else {
		uiItemMenuF(layout,
		            menu->collection->id.name + 2,
		            ICON_NONE,
		            move_to_collection_menu_create,
		            menu);
	}
}

/* This is allocated statically because we need this available for the menus creation callback. */
static MoveToCollectionData *master_collection_menu = NULL;

static int move_to_collection_invoke(bContext *C, wmOperator *op, const wmEvent *UNUSED(event))
{
	Scene *scene = CTX_data_scene(C);

	/* Reset the menus data for the current master collection, and free previously allocated data. */
	move_to_collection_menus_free(&master_collection_menu);

	PropertyRNA *prop;
	prop = RNA_struct_find_property(op->ptr, "collection_index");
	if (RNA_property_is_set(op->ptr, prop)) {
		int collection_index = RNA_property_int_get(op->ptr, prop);

		if (RNA_boolean_get(op->ptr, "is_new")) {
			prop = RNA_struct_find_property(op->ptr, "new_collection_name");
			if (!RNA_property_is_set(op->ptr, prop)) {
				char name[MAX_NAME];
				Collection *collection;

				collection = BKE_collection_from_index(scene, collection_index);
				BKE_collection_new_name_get(collection, name);

				RNA_property_string_set(op->ptr, prop, name);
				return WM_operator_props_dialog_popup(C, op, 10 * UI_UNIT_X, 5 * UI_UNIT_Y);
			}
		}
		return move_to_collection_exec(C, op);
	}

	Collection *master_collection = BKE_collection_master(scene);

	/* We need the data to be allocated so it's available during menu drawing.
	 * Technically we could use wmOperator->customdata. However there is no free callback
	 * called to an operator that exit with OPERATOR_INTERFACE to launch a menu.
	 *
	 * So we are left with a memory that will necessarily leak. It's a small leak though.*/
	if (master_collection_menu == NULL) {
		master_collection_menu = MEM_callocN(sizeof(MoveToCollectionData),
		                                     "MoveToCollectionData menu - expected eventual memleak");
	}

	master_collection_menu->collection = master_collection;
	master_collection_menu->ot = op->type;
	move_to_collection_menus_create(op, master_collection_menu);

	uiPopupMenu *pup;
	uiLayout *layout;

	/* Build the menus. */
	const char *title = CTX_IFACE_(op->type->translation_context, op->type->name);
	pup = UI_popup_menu_begin(C, title, ICON_NONE);
	layout = UI_popup_menu_layout(pup);

	uiLayoutSetOperatorContext(layout, WM_OP_INVOKE_DEFAULT);

	move_to_collection_menu_create(C, layout, master_collection_menu);

	UI_popup_menu_end(C, pup);

	return OPERATOR_INTERFACE;
}

void OBJECT_OT_move_to_collection(wmOperatorType *ot)
{
	PropertyRNA *prop;

	/* identifiers */
	ot->name = "Move to Collection";
	ot->description = "Move objects to a scene collection";
	ot->idname = "OBJECT_OT_move_to_collection";

	/* api callbacks */
	ot->exec = move_to_collection_exec;
	ot->invoke = move_to_collection_invoke;
	ot->poll = move_to_collection_poll;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;

	prop = RNA_def_int(ot->srna, "collection_index", COLLECTION_INVALID_INDEX, COLLECTION_INVALID_INDEX, INT_MAX,
	                   "Collection Index", "Index of the collection to move to", 0, INT_MAX);
	RNA_def_property_flag(prop, PROP_SKIP_SAVE | PROP_HIDDEN);
	prop = RNA_def_boolean(ot->srna, "is_new", false, "New", "Move objects to a new collection");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE | PROP_HIDDEN);
	prop = RNA_def_string(ot->srna, "new_collection_name", NULL, MAX_NAME, "Name",
	                      "Name of the newly added collection");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE);
}

void OBJECT_OT_link_to_collection(wmOperatorType *ot)
{
	PropertyRNA *prop;

	/* identifiers */
	ot->name = "Link to Collection";
	ot->description = "Link objects to a collection";
	ot->idname = "OBJECT_OT_link_to_collection";

	/* api callbacks */
	ot->exec = move_to_collection_exec;
	ot->invoke = move_to_collection_invoke;
	ot->poll = move_to_collection_poll;

	/* flags */
	ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;

	prop = RNA_def_int(ot->srna, "collection_index", COLLECTION_INVALID_INDEX, COLLECTION_INVALID_INDEX, INT_MAX,
	                   "Collection Index", "Index of the collection to move to", 0, INT_MAX);
	RNA_def_property_flag(prop, PROP_SKIP_SAVE | PROP_HIDDEN);
	prop = RNA_def_boolean(ot->srna, "is_new", false, "New", "Move objects to a new collection");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE | PROP_HIDDEN);
	prop = RNA_def_string(ot->srna, "new_collection_name", NULL, MAX_NAME, "Name",
	                      "Name of the newly added collection");
	RNA_def_property_flag(prop, PROP_SKIP_SAVE);
}