This unifies two almost-identical functions at the expense of having to add one single 'flag' value at one call. This makes copy_softbody() aware of the source/dest objects, allowing it to make a distinction between doing depsgraph evaluation copies and real object copies. This will be used in an upcoming commit to ensure that the pointcache is shared between CoW copies, similar to the current approach for rigidbody simulation.
2103 lines
56 KiB
C
2103 lines
56 KiB
C
/*
|
|
* ***** 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);
|
|
|
|
if (G.background == false) {
|
|
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);
|
|
|
|
if (G.background == false) {
|
|
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) {
|
|
if (base->object->soft) sbFree(base->object->soft);
|
|
BKE_object_copy_softbody(base->object, ob, 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);
|
|
}
|