diff options
| author | Julian Eisel <julian@blender.org> | 2022-01-13 18:35:32 +0300 |
|---|---|---|
| committer | Julian Eisel <julian@blender.org> | 2022-01-13 19:01:47 +0300 |
| commit | 039cc329178e9f7b14f514850479648c38fe1ea3 (patch) | |
| tree | 6729e63938ab2103606d8048c7a2f52d415e1ec4 /source/blender/editors/space_outliner/outliner_draw.c | |
| parent | 1a4f8ab38934891e76a56a5c31be02b934f68162 (diff) | |
Outliner: Compile all Outliner files in C++
We want to refactor quite some of the Outliner code using C++, this is a
logical step to help the transition to a new architecture.
Includes plenty of fixes to make this compile without warnings, trying
not to change logic. The usual stuff (casts from `void *`, designated
initializers, compound literals, etc.).
Diffstat (limited to 'source/blender/editors/space_outliner/outliner_draw.c')
| -rw-r--r-- | source/blender/editors/space_outliner/outliner_draw.c | 3947 |
1 files changed, 0 insertions, 3947 deletions
diff --git a/source/blender/editors/space_outliner/outliner_draw.c b/source/blender/editors/space_outliner/outliner_draw.c deleted file mode 100644 index 8bcd50161bd..00000000000 --- a/source/blender/editors/space_outliner/outliner_draw.c +++ /dev/null @@ -1,3947 +0,0 @@ -/* - * 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) 2004 Blender Foundation. - * All rights reserved. - */ - -/** \file - * \ingroup spoutliner - */ - -#include "DNA_armature_types.h" -#include "DNA_collection_types.h" -#include "DNA_constraint_types.h" -#include "DNA_gpencil_modifier_types.h" -#include "DNA_gpencil_types.h" -#include "DNA_light_types.h" -#include "DNA_lightprobe_types.h" -#include "DNA_object_force_types.h" -#include "DNA_object_types.h" -#include "DNA_scene_types.h" -#include "DNA_sequence_types.h" -#include "DNA_text_types.h" - -#include "BLI_blenlib.h" -#include "BLI_math.h" -#include "BLI_mempool.h" -#include "BLI_string_utils.h" -#include "BLI_utildefines.h" - -#include "BLT_translation.h" - -#include "BKE_armature.h" -#include "BKE_context.h" -#include "BKE_deform.h" -#include "BKE_gpencil.h" -#include "BKE_idtype.h" -#include "BKE_layer.h" -#include "BKE_lib_id.h" -#include "BKE_library.h" -#include "BKE_main.h" -#include "BKE_modifier.h" -#include "BKE_object.h" -#include "BKE_particle.h" -#include "BKE_report.h" - -#include "DEG_depsgraph.h" -#include "DEG_depsgraph_build.h" - -#include "ED_armature.h" -#include "ED_fileselect.h" -#include "ED_outliner.h" -#include "ED_screen.h" - -#include "WM_api.h" -#include "WM_message.h" -#include "WM_types.h" - -#include "GPU_immediate.h" -#include "GPU_state.h" - -#include "UI_interface.h" -#include "UI_interface_icons.h" -#include "UI_resources.h" -#include "UI_view2d.h" - -#include "RNA_access.h" - -#include "outliner_intern.h" -#include "tree/tree_display.h" - -/* Disable - this is far too slow - campbell. */ -/* #define USE_GROUP_SELECT */ - -/* ****************************************************** */ -/* Tree Size Functions */ - -static void outliner_tree_dimensions_impl(SpaceOutliner *space_outliner, - ListBase *lb, - int *width, - int *height) -{ - LISTBASE_FOREACH (TreeElement *, te, lb) { - *width = MAX2(*width, te->xend); - if (height != NULL) { - *height += UI_UNIT_Y; - } - - TreeStoreElem *tselem = TREESTORE(te); - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_tree_dimensions_impl(space_outliner, &te->subtree, width, height); - } - else { - outliner_tree_dimensions_impl(space_outliner, &te->subtree, width, NULL); - } - } -} - -void outliner_tree_dimensions(SpaceOutliner *space_outliner, int *r_width, int *r_height) -{ - *r_width = 0; - *r_height = 0; - outliner_tree_dimensions_impl(space_outliner, &space_outliner->tree, r_width, r_height); -} - -/** - * The active object is only needed for reference. - */ -static bool is_object_data_in_editmode(const ID *id, const Object *obact) -{ - if (id == NULL) { - return false; - } - - const short id_type = GS(id->name); - - if (id_type == ID_GD && obact && obact->data == id) { - bGPdata *gpd = (bGPdata *)id; - return GPENCIL_EDIT_MODE(gpd); - } - - return ((obact && (obact->mode & OB_MODE_EDIT)) && (id && OB_DATA_SUPPORT_EDITMODE(id_type)) && - (GS(((ID *)obact->data)->name) == id_type) && BKE_object_data_is_in_editmode(id)); -} - -/* ****************************************************** */ - -static void restrictbutton_recursive_ebone(bArmature *arm, - EditBone *ebone_parent, - int flag, - bool set_flag) -{ - EditBone *ebone; - - for (ebone = arm->edbo->first; ebone; ebone = ebone->next) { - if (ED_armature_ebone_is_child_recursive(ebone_parent, ebone)) { - if (set_flag) { - ebone->flag &= ~(BONE_TIPSEL | BONE_SELECTED | BONE_ROOTSEL); - ebone->flag |= flag; - } - else { - ebone->flag &= ~flag; - } - } - } -} - -static void restrictbutton_recursive_bone(Bone *bone_parent, int flag, bool set_flag) -{ - Bone *bone; - for (bone = bone_parent->childbase.first; bone; bone = bone->next) { - if (set_flag) { - bone->flag &= ~(BONE_TIPSEL | BONE_SELECTED | BONE_ROOTSEL); - bone->flag |= flag; - } - else { - bone->flag &= ~flag; - } - restrictbutton_recursive_bone(bone, flag, set_flag); - } -} - -static void restrictbutton_r_lay_fn(bContext *C, void *poin, void *UNUSED(poin2)) -{ - WM_event_add_notifier(C, NC_SCENE | ND_RENDER_OPTIONS, poin); -} - -static void restrictbutton_bone_visibility_fn(bContext *C, void *poin, void *UNUSED(poin2)) -{ - Bone *bone = (Bone *)poin; - - if (CTX_wm_window(C)->eventstate->shift) { - restrictbutton_recursive_bone(bone, BONE_HIDDEN_P, (bone->flag & BONE_HIDDEN_P) != 0); - } -} - -static void restrictbutton_bone_select_fn(bContext *C, void *UNUSED(poin), void *poin2) -{ - Bone *bone = (Bone *)poin2; - if (bone->flag & BONE_UNSELECTABLE) { - bone->flag &= ~(BONE_SELECTED | BONE_TIPSEL | BONE_ROOTSEL); - } - - if (CTX_wm_window(C)->eventstate->shift) { - restrictbutton_recursive_bone(bone, BONE_UNSELECTABLE, (bone->flag & BONE_UNSELECTABLE) != 0); - } - - WM_event_add_notifier(C, NC_OBJECT | ND_POSE, NULL); -} - -static void restrictbutton_ebone_select_fn(bContext *C, void *poin, void *poin2) -{ - bArmature *arm = (bArmature *)poin; - EditBone *ebone = (EditBone *)poin2; - - if (ebone->flag & BONE_UNSELECTABLE) { - ebone->flag &= ~(BONE_SELECTED | BONE_TIPSEL | BONE_ROOTSEL); - } - - if (CTX_wm_window(C)->eventstate->shift) { - restrictbutton_recursive_ebone( - arm, ebone, BONE_UNSELECTABLE, (ebone->flag & BONE_UNSELECTABLE) != 0); - } - - WM_event_add_notifier(C, NC_OBJECT | ND_POSE, NULL); -} - -static void restrictbutton_ebone_visibility_fn(bContext *C, void *poin, void *poin2) -{ - bArmature *arm = (bArmature *)poin; - EditBone *ebone = (EditBone *)poin2; - if (ebone->flag & BONE_HIDDEN_A) { - ebone->flag &= ~(BONE_SELECTED | BONE_TIPSEL | BONE_ROOTSEL); - } - - if (CTX_wm_window(C)->eventstate->shift) { - restrictbutton_recursive_ebone(arm, ebone, BONE_HIDDEN_A, (ebone->flag & BONE_HIDDEN_A) != 0); - } - - WM_event_add_notifier(C, NC_OBJECT | ND_POSE, NULL); -} - -static void restrictbutton_gp_layer_flag_fn(bContext *C, void *poin, void *UNUSED(poin2)) -{ - ID *id = (ID *)poin; - - DEG_id_tag_update(id, ID_RECALC_GEOMETRY); - WM_event_add_notifier(C, NC_GPENCIL | ND_DATA | NA_EDITED, NULL); -} - -static void restrictbutton_id_user_toggle(bContext *UNUSED(C), void *poin, void *UNUSED(poin2)) -{ - ID *id = (ID *)poin; - - BLI_assert(id != NULL); - - if (id->flag & LIB_FAKEUSER) { - id_us_plus(id); - } - else { - id_us_min(id); - } -} - -static void outliner_object_set_flag_recursive_fn(bContext *C, - Base *base, - Object *ob, - const char *propname) -{ - Main *bmain = CTX_data_main(C); - wmWindow *win = CTX_wm_window(C); - Scene *scene = CTX_data_scene(C); - ViewLayer *view_layer = CTX_data_view_layer(C); - PointerRNA ptr; - - bool extend = (win->eventstate->shift != 0); - - if (!extend) { - return; - } - - /* Create PointerRNA and PropertyRNA for either Object or Base. */ - ID *id = ob ? &ob->id : &scene->id; - StructRNA *struct_rna = ob ? &RNA_Object : &RNA_ObjectBase; - void *data = ob ? (void *)ob : (void *)base; - - RNA_pointer_create(id, struct_rna, data, &ptr); - PropertyRNA *base_or_object_prop = RNA_struct_type_find_property(struct_rna, propname); - const bool value = RNA_property_boolean_get(&ptr, base_or_object_prop); - - Object *ob_parent = ob ? ob : base->object; - - for (Object *ob_iter = bmain->objects.first; ob_iter; ob_iter = ob_iter->id.next) { - if (BKE_object_is_child_recursive(ob_parent, ob_iter)) { - if (ob) { - RNA_id_pointer_create(&ob_iter->id, &ptr); - DEG_id_tag_update(&ob_iter->id, ID_RECALC_COPY_ON_WRITE); - } - else { - Base *base_iter = BKE_view_layer_base_find(view_layer, ob_iter); - /* Child can be in a collection excluded from viewlayer. */ - if (base_iter == NULL) { - continue; - } - RNA_pointer_create(&scene->id, &RNA_ObjectBase, base_iter, &ptr); - } - RNA_property_boolean_set(&ptr, base_or_object_prop, value); - } - } - - /* We don't call RNA_property_update() due to performance, so we batch update them. */ - if (ob) { - BKE_main_collection_sync_remap(bmain); - DEG_relations_tag_update(bmain); - } - else { - BKE_layer_collection_sync(scene, view_layer); - DEG_id_tag_update(&scene->id, ID_RECALC_BASE_FLAGS); - } -} - -/** - * Object properties. - */ -static void outliner__object_set_flag_recursive_fn(bContext *C, void *poin, void *poin2) -{ - Object *ob = poin; - char *propname = poin2; - outliner_object_set_flag_recursive_fn(C, NULL, ob, propname); -} - -/** - * Base properties. - */ -static void outliner__base_set_flag_recursive_fn(bContext *C, void *poin, void *poin2) -{ - Base *base = poin; - char *propname = poin2; - outliner_object_set_flag_recursive_fn(C, base, NULL, propname); -} - -/** Create either a RNA_LayerCollection or a RNA_Collection pointer. */ -static void outliner_layer_or_collection_pointer_create(Scene *scene, - LayerCollection *layer_collection, - Collection *collection, - PointerRNA *ptr) -{ - if (collection) { - RNA_id_pointer_create(&collection->id, ptr); - } - else { - RNA_pointer_create(&scene->id, &RNA_LayerCollection, layer_collection, ptr); - } -} - -/** Create either a RNA_ObjectBase or a RNA_Object pointer. */ -static void outliner_base_or_object_pointer_create( - Scene *scene, ViewLayer *view_layer, Collection *collection, Object *ob, PointerRNA *ptr) -{ - if (collection) { - RNA_id_pointer_create(&ob->id, ptr); - } - else { - Base *base = BKE_view_layer_base_find(view_layer, ob); - RNA_pointer_create(&scene->id, &RNA_ObjectBase, base, ptr); - } -} - -/* NOTE: Collection is only valid when we want to change the collection data, otherwise we get it - * from layer collection. Layer collection is valid whenever we are looking at a view layer. */ -static void outliner_collection_set_flag_recursive(Scene *scene, - ViewLayer *view_layer, - LayerCollection *layer_collection, - Collection *collection, - PropertyRNA *layer_or_collection_prop, - PropertyRNA *base_or_object_prop, - const bool value) -{ - if (layer_collection && layer_collection->flag & LAYER_COLLECTION_EXCLUDE) { - return; - } - PointerRNA ptr; - outliner_layer_or_collection_pointer_create(scene, layer_collection, collection, &ptr); - RNA_property_boolean_set(&ptr, layer_or_collection_prop, value); - - /* Set the same flag for the nested objects as well. */ - if (base_or_object_prop) { - /* NOTE: We can't use BKE_collection_object_cache_get() - * otherwise we would not take collection exclusion into account. */ - LISTBASE_FOREACH (CollectionObject *, cob, &layer_collection->collection->gobject) { - - outliner_base_or_object_pointer_create(scene, view_layer, collection, cob->ob, &ptr); - RNA_property_boolean_set(&ptr, base_or_object_prop, value); - - if (collection) { - DEG_id_tag_update(&cob->ob->id, ID_RECALC_COPY_ON_WRITE); - } - } - } - - /* Keep going recursively. */ - ListBase *lb = (layer_collection ? &layer_collection->layer_collections : &collection->children); - LISTBASE_FOREACH (Link *, link, lb) { - LayerCollection *layer_collection_iter = layer_collection ? (LayerCollection *)link : NULL; - Collection *collection_iter = layer_collection ? - (collection ? layer_collection_iter->collection : NULL) : - ((CollectionChild *)link)->collection; - outliner_collection_set_flag_recursive(scene, - view_layer, - layer_collection_iter, - collection_iter, - layer_or_collection_prop, - base_or_object_prop, - value); - } - - if (collection) { - DEG_id_tag_update(&collection->id, ID_RECALC_COPY_ON_WRITE); - } -} - -/** - * Check if collection is already isolated. - * - * A collection is isolated if all its parents and children are "visible". - * All the other collections must be "invisible". - * - * NOTE: We could/should boost performance by iterating over the tree twice. - * First tagging all the children/parent collections, then getting their values and comparing. - * To run BKE_collection_has_collection() so many times is silly and slow. - */ -static bool outliner_collection_is_isolated(Scene *scene, - const LayerCollection *layer_collection_cmp, - const Collection *collection_cmp, - const bool value_cmp, - const PropertyRNA *layer_or_collection_prop, - LayerCollection *layer_collection, - Collection *collection) -{ - PointerRNA ptr; - outliner_layer_or_collection_pointer_create(scene, layer_collection, collection, &ptr); - const bool value = RNA_property_boolean_get(&ptr, (PropertyRNA *)layer_or_collection_prop); - Collection *collection_ensure = collection ? collection : layer_collection->collection; - const Collection *collection_ensure_cmp = collection_cmp ? collection_cmp : - layer_collection_cmp->collection; - - if (collection_ensure->flag & COLLECTION_IS_MASTER) { - } - else if (collection_ensure == collection_ensure_cmp) { - } - else if (BKE_collection_has_collection(collection_ensure, (Collection *)collection_ensure_cmp) || - BKE_collection_has_collection((Collection *)collection_ensure_cmp, collection_ensure)) { - /* This collection is either a parent or a child of the collection. - * We expect it to be set "visible" already. */ - if (value != value_cmp) { - return false; - } - } - else { - /* This collection is neither a parent nor a child of the collection. - * We expect it to be "invisible". */ - if (value == value_cmp) { - return false; - } - } - - /* Keep going recursively. */ - ListBase *lb = (layer_collection ? &layer_collection->layer_collections : &collection->children); - LISTBASE_FOREACH (Link *, link, lb) { - LayerCollection *layer_collection_iter = layer_collection ? (LayerCollection *)link : NULL; - Collection *collection_iter = layer_collection ? - (collection ? layer_collection_iter->collection : NULL) : - ((CollectionChild *)link)->collection; - if (layer_collection_iter && layer_collection_iter->flag & LAYER_COLLECTION_EXCLUDE) { - continue; - } - if (!outliner_collection_is_isolated(scene, - layer_collection_cmp, - collection_cmp, - value_cmp, - layer_or_collection_prop, - layer_collection_iter, - collection_iter)) { - return false; - } - } - - return true; -} - -void outliner_collection_isolate_flag(Scene *scene, - ViewLayer *view_layer, - LayerCollection *layer_collection, - Collection *collection, - PropertyRNA *layer_or_collection_prop, - const char *propname, - const bool value) -{ - PointerRNA ptr; - const bool is_hide = strstr(propname, "hide_") != NULL; - - LayerCollection *top_layer_collection = layer_collection ? view_layer->layer_collections.first : - NULL; - Collection *top_collection = collection ? scene->master_collection : NULL; - - bool was_isolated = (value == is_hide); - was_isolated &= outliner_collection_is_isolated(scene, - layer_collection, - collection, - !is_hide, - layer_or_collection_prop, - top_layer_collection, - top_collection); - - if (was_isolated) { - const bool default_value = RNA_property_boolean_get_default(NULL, layer_or_collection_prop); - /* Make every collection go back to its default "visibility" state. */ - outliner_collection_set_flag_recursive(scene, - view_layer, - top_layer_collection, - top_collection, - layer_or_collection_prop, - NULL, - default_value); - return; - } - - /* Make every collection "invisible". */ - outliner_collection_set_flag_recursive(scene, - view_layer, - top_layer_collection, - top_collection, - layer_or_collection_prop, - NULL, - is_hide); - - /* Make this collection and its children collections the only "visible". */ - outliner_collection_set_flag_recursive( - scene, view_layer, layer_collection, collection, layer_or_collection_prop, NULL, !is_hide); - - /* Make this collection direct parents also "visible". */ - if (layer_collection) { - LayerCollection *lc_parent = layer_collection; - LISTBASE_FOREACH (LayerCollection *, lc_iter, &top_layer_collection->layer_collections) { - if (BKE_layer_collection_has_layer_collection(lc_iter, layer_collection)) { - lc_parent = lc_iter; - break; - } - } - - while (lc_parent != layer_collection) { - outliner_layer_or_collection_pointer_create( - scene, lc_parent, collection ? lc_parent->collection : NULL, &ptr); - RNA_property_boolean_set(&ptr, layer_or_collection_prop, !is_hide); - - LISTBASE_FOREACH (LayerCollection *, lc_iter, &lc_parent->layer_collections) { - if (BKE_layer_collection_has_layer_collection(lc_iter, layer_collection)) { - lc_parent = lc_iter; - break; - } - } - } - } - else { - CollectionParent *parent; - Collection *child = collection; - while ((parent = child->parents.first)) { - if (parent->collection->flag & COLLECTION_IS_MASTER) { - break; - } - RNA_id_pointer_create(&parent->collection->id, &ptr); - RNA_property_boolean_set(&ptr, layer_or_collection_prop, !is_hide); - child = parent->collection; - } - } -} - -static void outliner_collection_set_flag_recursive_fn(bContext *C, - LayerCollection *layer_collection, - Collection *collection, - const char *propname) -{ - Main *bmain = CTX_data_main(C); - wmWindow *win = CTX_wm_window(C); - Scene *scene = CTX_data_scene(C); - ViewLayer *view_layer = CTX_data_view_layer(C); - PointerRNA ptr; - - bool do_isolate = (win->eventstate->ctrl != 0); - bool extend = (win->eventstate->shift != 0); - - if (!ELEM(true, do_isolate, extend)) { - return; - } - - /* Create PointerRNA and PropertyRNA for either Collection or LayerCollection. */ - ID *id = collection ? &collection->id : &scene->id; - StructRNA *struct_rna = collection ? &RNA_Collection : &RNA_LayerCollection; - void *data = collection ? (void *)collection : (void *)layer_collection; - - RNA_pointer_create(id, struct_rna, data, &ptr); - outliner_layer_or_collection_pointer_create(scene, layer_collection, collection, &ptr); - PropertyRNA *layer_or_collection_prop = RNA_struct_type_find_property(struct_rna, propname); - const bool value = RNA_property_boolean_get(&ptr, layer_or_collection_prop); - - PropertyRNA *base_or_object_prop = NULL; - if (layer_collection != NULL) { - /* If we are toggling Layer collections we still want to change the properties of the base - * or the objects. If we have a matching property, toggle it as well, it can be NULL. */ - struct_rna = collection ? &RNA_Object : &RNA_ObjectBase; - base_or_object_prop = RNA_struct_type_find_property(struct_rna, propname); - } - - if (extend) { - outliner_collection_set_flag_recursive(scene, - view_layer, - layer_collection, - collection, - layer_or_collection_prop, - base_or_object_prop, - value); - } - else { - outliner_collection_isolate_flag(scene, - view_layer, - layer_collection, - collection, - layer_or_collection_prop, - propname, - value); - } - - /* We don't call RNA_property_update() due to performance, so we batch update them. */ - BKE_main_collection_sync_remap(bmain); - DEG_relations_tag_update(bmain); -} - -/** - * Layer collection properties called from the ViewLayer mode. - * Change the (non-excluded) collection children, and the objects nested to them all. - */ -static void view_layer__layer_collection_set_flag_recursive_fn(bContext *C, - void *poin, - void *poin2) -{ - LayerCollection *layer_collection = poin; - char *propname = poin2; - outliner_collection_set_flag_recursive_fn(C, layer_collection, NULL, propname); -} - -/** - * Collection properties called from the ViewLayer mode. - * Change the (non-excluded) collection children, and the objects nested to them all. - */ -static void view_layer__collection_set_flag_recursive_fn(bContext *C, void *poin, void *poin2) -{ - LayerCollection *layer_collection = poin; - char *propname = poin2; - outliner_collection_set_flag_recursive_fn( - C, layer_collection, layer_collection->collection, propname); -} - -/** - * Collection properties called from the Scenes mode. - * Change the collection children but no objects. - */ -static void scenes__collection_set_flag_recursive_fn(bContext *C, void *poin, void *poin2) -{ - Collection *collection = poin; - char *propname = poin2; - outliner_collection_set_flag_recursive_fn(C, NULL, collection, propname); -} - -static void namebutton_fn(bContext *C, void *tsep, char *oldname) -{ - Main *bmain = CTX_data_main(C); - SpaceOutliner *space_outliner = CTX_wm_space_outliner(C); - struct wmMsgBus *mbus = CTX_wm_message_bus(C); - BLI_mempool *ts = space_outliner->treestore; - TreeStoreElem *tselem = tsep; - - if (ts && tselem) { - TreeElement *te = outliner_find_tree_element(&space_outliner->tree, tselem); - - if (tselem->type == TSE_SOME_ID) { - BLI_libblock_ensure_unique_name(bmain, tselem->id->name); - - WM_msg_publish_rna_prop(mbus, tselem->id, tselem->id, ID, name); - - switch (GS(tselem->id->name)) { - case ID_MA: - WM_event_add_notifier(C, NC_MATERIAL, NULL); - break; - case ID_TE: - WM_event_add_notifier(C, NC_TEXTURE, NULL); - break; - case ID_IM: - WM_event_add_notifier(C, NC_IMAGE, NULL); - break; - case ID_SCE: - WM_event_add_notifier(C, NC_SCENE, NULL); - break; - case ID_OB: { - Object *ob = (Object *)tselem->id; - if (ob->type == OB_MBALL) { - DEG_id_tag_update(&ob->id, ID_RECALC_GEOMETRY); - } - DEG_id_tag_update(&ob->id, ID_RECALC_COPY_ON_WRITE); - break; - } - default: - break; - } - WM_event_add_notifier(C, NC_ID | NA_RENAME, NULL); - - /* Check the library target exists */ - if (te->idcode == ID_LI) { - Library *lib = (Library *)tselem->id; - char expanded[FILE_MAX]; - - BKE_library_filepath_set(bmain, lib, lib->filepath); - - BLI_strncpy(expanded, lib->filepath, sizeof(expanded)); - BLI_path_abs(expanded, BKE_main_blendfile_path(bmain)); - if (!BLI_exists(expanded)) { - BKE_reportf(CTX_wm_reports(C), - RPT_ERROR, - "Library path '%s' does not exist, correct this before saving", - expanded); - } - else if (lib->id.tag & LIB_TAG_MISSING) { - BKE_reportf(CTX_wm_reports(C), - RPT_INFO, - "Library path '%s' is now valid, please reload the library", - expanded); - lib->id.tag &= ~LIB_TAG_MISSING; - } - } - } - else { - switch (tselem->type) { - case TSE_DEFGROUP: { - Object *ob = (Object *)tselem->id; - bDeformGroup *vg = te->directdata; - BKE_object_defgroup_unique_name(vg, ob); - WM_msg_publish_rna_prop(mbus, &ob->id, vg, VertexGroup, name); - break; - } - case TSE_NLA_ACTION: { - bAction *act = (bAction *)tselem->id; - BLI_libblock_ensure_unique_name(bmain, act->id.name); - WM_msg_publish_rna_prop(mbus, &act->id, &act->id, ID, name); - break; - } - case TSE_EBONE: { - bArmature *arm = (bArmature *)tselem->id; - if (arm->edbo) { - EditBone *ebone = te->directdata; - char newname[sizeof(ebone->name)]; - - /* restore bone name */ - BLI_strncpy(newname, ebone->name, sizeof(ebone->name)); - BLI_strncpy(ebone->name, oldname, sizeof(ebone->name)); - ED_armature_bone_rename(bmain, arm, oldname, newname); - WM_msg_publish_rna_prop(mbus, &arm->id, ebone, EditBone, name); - WM_event_add_notifier(C, NC_OBJECT | ND_POSE, NULL); - } - break; - } - - case TSE_BONE: { - TreeViewContext tvc; - outliner_viewcontext_init(C, &tvc); - - bArmature *arm = (bArmature *)tselem->id; - Bone *bone = te->directdata; - char newname[sizeof(bone->name)]; - - /* always make current object active */ - tree_element_activate(C, &tvc, te, OL_SETSEL_NORMAL, true); - - /* restore bone name */ - BLI_strncpy(newname, bone->name, sizeof(bone->name)); - BLI_strncpy(bone->name, oldname, sizeof(bone->name)); - ED_armature_bone_rename(bmain, arm, oldname, newname); - WM_msg_publish_rna_prop(mbus, &arm->id, bone, Bone, name); - WM_event_add_notifier(C, NC_OBJECT | ND_POSE, NULL); - break; - } - case TSE_POSE_CHANNEL: { - TreeViewContext tvc; - outliner_viewcontext_init(C, &tvc); - - Object *ob = (Object *)tselem->id; - bArmature *arm = (bArmature *)ob->data; - bPoseChannel *pchan = te->directdata; - char newname[sizeof(pchan->name)]; - - /* always make current pose-bone active */ - tree_element_activate(C, &tvc, te, OL_SETSEL_NORMAL, true); - - BLI_assert(ob->type == OB_ARMATURE); - - /* restore bone name */ - BLI_strncpy(newname, pchan->name, sizeof(pchan->name)); - BLI_strncpy(pchan->name, oldname, sizeof(pchan->name)); - ED_armature_bone_rename(bmain, ob->data, oldname, newname); - WM_msg_publish_rna_prop(mbus, &arm->id, pchan->bone, Bone, name); - WM_event_add_notifier(C, NC_OBJECT | ND_POSE, NULL); - break; - } - case TSE_POSEGRP: { - Object *ob = (Object *)tselem->id; /* id = object. */ - bActionGroup *grp = te->directdata; - - BLI_uniquename(&ob->pose->agroups, - grp, - CTX_DATA_(BLT_I18NCONTEXT_ID_ACTION, "Group"), - '.', - offsetof(bActionGroup, name), - sizeof(grp->name)); - WM_msg_publish_rna_prop(mbus, &ob->id, grp, ActionGroup, name); - WM_event_add_notifier(C, NC_OBJECT | ND_POSE, ob); - break; - } - case TSE_GP_LAYER: { - bGPdata *gpd = (bGPdata *)tselem->id; /* id = GP Datablock */ - bGPDlayer *gpl = te->directdata; - - /* always make layer active */ - BKE_gpencil_layer_active_set(gpd, gpl); - - /* XXX: name needs translation stuff. */ - BLI_uniquename( - &gpd->layers, gpl, "GP Layer", '.', offsetof(bGPDlayer, info), sizeof(gpl->info)); - - WM_msg_publish_rna_prop(mbus, &gpd->id, gpl, GPencilLayer, info); - DEG_id_tag_update(&gpd->id, ID_RECALC_GEOMETRY); - WM_event_add_notifier(C, NC_GPENCIL | ND_DATA | NA_SELECTED, gpd); - break; - } - case TSE_R_LAYER: { - Scene *scene = (Scene *)tselem->id; - ViewLayer *view_layer = te->directdata; - - /* Restore old name. */ - char newname[sizeof(view_layer->name)]; - BLI_strncpy(newname, view_layer->name, sizeof(view_layer->name)); - BLI_strncpy(view_layer->name, oldname, sizeof(view_layer->name)); - - /* Rename, preserving animation and compositing data. */ - BKE_view_layer_rename(bmain, scene, view_layer, newname); - WM_msg_publish_rna_prop(mbus, &scene->id, view_layer, ViewLayer, name); - WM_event_add_notifier(C, NC_ID | NA_RENAME, NULL); - break; - } - case TSE_LAYER_COLLECTION: { - /* The ID is a #Collection, not a #LayerCollection */ - Collection *collection = (Collection *)tselem->id; - BLI_libblock_ensure_unique_name(bmain, collection->id.name); - WM_msg_publish_rna_prop(mbus, &collection->id, &collection->id, ID, name); - WM_event_add_notifier(C, NC_ID | NA_RENAME, NULL); - break; - } - } - } - tselem->flag &= ~TSE_TEXTBUT; - } -} - -typedef struct RestrictProperties { - bool initialized; - - PropertyRNA *object_hide_viewport, *object_hide_select, *object_hide_render; - PropertyRNA *base_hide_viewport; - PropertyRNA *collection_hide_viewport, *collection_hide_select, *collection_hide_render; - PropertyRNA *layer_collection_exclude, *layer_collection_holdout, - *layer_collection_indirect_only, *layer_collection_hide_viewport; - PropertyRNA *modifier_show_viewport, *modifier_show_render; - PropertyRNA *constraint_enable; - PropertyRNA *bone_hide_viewport; -} RestrictProperties; - -/* We don't care about the value of the property - * but whether the property should be active or grayed out. */ -typedef struct RestrictPropertiesActive { - bool object_hide_viewport; - bool object_hide_select; - bool object_hide_render; - bool base_hide_viewport; - bool collection_hide_viewport; - bool collection_hide_select; - bool collection_hide_render; - bool layer_collection_exclude; - bool layer_collection_holdout; - bool layer_collection_indirect_only; - bool layer_collection_hide_viewport; - bool modifier_show_viewport; - bool modifier_show_render; - bool constraint_enable; - bool bone_hide_viewport; -} RestrictPropertiesActive; - -static void outliner_restrict_properties_enable_collection_set( - PointerRNA *collection_ptr, RestrictProperties *props, RestrictPropertiesActive *props_active) -{ - if (props_active->collection_hide_render) { - props_active->collection_hide_render = !RNA_property_boolean_get( - collection_ptr, props->collection_hide_render); - if (!props_active->collection_hide_render) { - props_active->layer_collection_holdout = false; - props_active->layer_collection_indirect_only = false; - props_active->object_hide_render = false; - props_active->modifier_show_render = false; - props_active->constraint_enable = false; - } - } - - if (props_active->collection_hide_viewport) { - props_active->collection_hide_viewport = !RNA_property_boolean_get( - collection_ptr, props->collection_hide_viewport); - if (!props_active->collection_hide_viewport) { - props_active->collection_hide_select = false; - props_active->object_hide_select = false; - props_active->layer_collection_hide_viewport = false; - props_active->object_hide_viewport = false; - props_active->base_hide_viewport = false; - props_active->modifier_show_viewport = false; - props_active->constraint_enable = false; - } - } - - if (props_active->collection_hide_select) { - props_active->collection_hide_select = !RNA_property_boolean_get( - collection_ptr, props->collection_hide_select); - if (!props_active->collection_hide_select) { - props_active->object_hide_select = false; - } - } -} - -static void outliner_restrict_properties_enable_layer_collection_set( - PointerRNA *layer_collection_ptr, - PointerRNA *collection_ptr, - RestrictProperties *props, - RestrictPropertiesActive *props_active) -{ - outliner_restrict_properties_enable_collection_set(collection_ptr, props, props_active); - - if (props_active->layer_collection_holdout) { - props_active->layer_collection_holdout = RNA_property_boolean_get( - layer_collection_ptr, props->layer_collection_holdout); - } - - if (props_active->layer_collection_indirect_only) { - props_active->layer_collection_indirect_only = RNA_property_boolean_get( - layer_collection_ptr, props->layer_collection_indirect_only); - } - - if (props_active->layer_collection_hide_viewport) { - props_active->layer_collection_hide_viewport = !RNA_property_boolean_get( - layer_collection_ptr, props->layer_collection_hide_viewport); - - if (!props_active->layer_collection_hide_viewport) { - props_active->base_hide_viewport = false; - props_active->collection_hide_select = false; - props_active->object_hide_select = false; - } - } - - if (props_active->layer_collection_exclude) { - props_active->layer_collection_exclude = !RNA_property_boolean_get( - layer_collection_ptr, props->layer_collection_exclude); - - if (!props_active->layer_collection_exclude) { - props_active->collection_hide_viewport = false; - props_active->collection_hide_select = false; - props_active->collection_hide_render = false; - props_active->layer_collection_hide_viewport = false; - props_active->layer_collection_holdout = false; - props_active->layer_collection_indirect_only = false; - } - } -} - -static bool outliner_restrict_properties_collection_set(Scene *scene, - TreeElement *te, - PointerRNA *collection_ptr, - PointerRNA *layer_collection_ptr, - RestrictProperties *props, - RestrictPropertiesActive *props_active) -{ - TreeStoreElem *tselem = TREESTORE(te); - LayerCollection *layer_collection = (tselem->type == TSE_LAYER_COLLECTION) ? te->directdata : - NULL; - Collection *collection = outliner_collection_from_tree_element(te); - - if (collection->flag & COLLECTION_IS_MASTER) { - return false; - } - - /* Create the PointerRNA. */ - RNA_id_pointer_create(&collection->id, collection_ptr); - if (layer_collection != NULL) { - RNA_pointer_create(&scene->id, &RNA_LayerCollection, layer_collection, layer_collection_ptr); - } - - /* Update the restriction column values for the collection children. */ - if (layer_collection) { - outliner_restrict_properties_enable_layer_collection_set( - layer_collection_ptr, collection_ptr, props, props_active); - } - else { - outliner_restrict_properties_enable_collection_set(collection_ptr, props, props_active); - } - return true; -} - -static void outliner_draw_restrictbuts(uiBlock *block, - Scene *scene, - ViewLayer *view_layer, - ARegion *region, - SpaceOutliner *space_outliner, - ListBase *lb, - RestrictPropertiesActive props_active_parent) -{ - /* Get RNA properties (once for speed). */ - static RestrictProperties props = {false}; - if (!props.initialized) { - props.object_hide_viewport = RNA_struct_type_find_property(&RNA_Object, "hide_viewport"); - props.object_hide_select = RNA_struct_type_find_property(&RNA_Object, "hide_select"); - props.object_hide_render = RNA_struct_type_find_property(&RNA_Object, "hide_render"); - props.base_hide_viewport = RNA_struct_type_find_property(&RNA_ObjectBase, "hide_viewport"); - props.collection_hide_viewport = RNA_struct_type_find_property(&RNA_Collection, - "hide_viewport"); - props.collection_hide_select = RNA_struct_type_find_property(&RNA_Collection, "hide_select"); - props.collection_hide_render = RNA_struct_type_find_property(&RNA_Collection, "hide_render"); - props.layer_collection_exclude = RNA_struct_type_find_property(&RNA_LayerCollection, - "exclude"); - props.layer_collection_holdout = RNA_struct_type_find_property(&RNA_LayerCollection, - "holdout"); - props.layer_collection_indirect_only = RNA_struct_type_find_property(&RNA_LayerCollection, - "indirect_only"); - props.layer_collection_hide_viewport = RNA_struct_type_find_property(&RNA_LayerCollection, - "hide_viewport"); - props.modifier_show_viewport = RNA_struct_type_find_property(&RNA_Modifier, "show_viewport"); - props.modifier_show_render = RNA_struct_type_find_property(&RNA_Modifier, "show_render"); - - props.constraint_enable = RNA_struct_type_find_property(&RNA_Constraint, "mute"); - - props.bone_hide_viewport = RNA_struct_type_find_property(&RNA_Bone, "hide"); - - props.initialized = true; - } - - struct { - int enable; - int select; - int hide; - int viewport; - int render; - int indirect_only; - int holdout; - } restrict_offsets = {0}; - int restrict_column_offset = 0; - - /* This will determine the order of drawing from RIGHT to LEFT. */ - if (space_outliner->outlinevis == SO_VIEW_LAYER) { - if (space_outliner->show_restrict_flags & SO_RESTRICT_INDIRECT_ONLY) { - restrict_offsets.indirect_only = (++restrict_column_offset) * UI_UNIT_X + V2D_SCROLL_WIDTH; - } - if (space_outliner->show_restrict_flags & SO_RESTRICT_HOLDOUT) { - restrict_offsets.holdout = (++restrict_column_offset) * UI_UNIT_X + V2D_SCROLL_WIDTH; - } - } - if (space_outliner->show_restrict_flags & SO_RESTRICT_RENDER) { - restrict_offsets.render = (++restrict_column_offset) * UI_UNIT_X + V2D_SCROLL_WIDTH; - } - if (space_outliner->show_restrict_flags & SO_RESTRICT_VIEWPORT) { - restrict_offsets.viewport = (++restrict_column_offset) * UI_UNIT_X + V2D_SCROLL_WIDTH; - } - if (space_outliner->show_restrict_flags & SO_RESTRICT_HIDE) { - restrict_offsets.hide = (++restrict_column_offset) * UI_UNIT_X + V2D_SCROLL_WIDTH; - } - if (space_outliner->show_restrict_flags & SO_RESTRICT_SELECT) { - restrict_offsets.select = (++restrict_column_offset) * UI_UNIT_X + V2D_SCROLL_WIDTH; - } - if (space_outliner->outlinevis == SO_VIEW_LAYER && - space_outliner->show_restrict_flags & SO_RESTRICT_ENABLE) { - restrict_offsets.enable = (++restrict_column_offset) * UI_UNIT_X + V2D_SCROLL_WIDTH; - } - - BLI_assert((restrict_column_offset * UI_UNIT_X + V2D_SCROLL_WIDTH) == - outliner_restrict_columns_width(space_outliner)); - - /* Create buttons. */ - uiBut *bt; - - LISTBASE_FOREACH (TreeElement *, te, lb) { - TreeStoreElem *tselem = TREESTORE(te); - RestrictPropertiesActive props_active = props_active_parent; - - if (te->ys + 2 * UI_UNIT_Y >= region->v2d.cur.ymin && te->ys <= region->v2d.cur.ymax) { - if (tselem->type == TSE_R_LAYER && - ELEM(space_outliner->outlinevis, SO_SCENES, SO_VIEW_LAYER)) { - if (space_outliner->show_restrict_flags & SO_RESTRICT_RENDER) { - /* View layer render toggle. */ - ViewLayer *layer = te->directdata; - - bt = uiDefIconButBitS(block, - UI_BTYPE_ICON_TOGGLE_N, - VIEW_LAYER_RENDER, - 0, - ICON_RESTRICT_RENDER_OFF, - (int)(region->v2d.cur.xmax - restrict_offsets.render), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &layer->flag, - 0, - 0, - 0, - 0, - TIP_("Use view layer for rendering")); - UI_but_func_set(bt, restrictbutton_r_lay_fn, tselem->id, NULL); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - UI_but_drawflag_enable(bt, UI_BUT_ICON_REVERSE); - } - } - else if (((tselem->type == TSE_SOME_ID) && (te->idcode == ID_OB)) && - (te->flag & TE_CHILD_NOT_IN_COLLECTION)) { - /* Don't show restrict columns for children that are not directly inside the collection. */ - } - else if ((tselem->type == TSE_SOME_ID) && (te->idcode == ID_OB)) { - PointerRNA ptr; - Object *ob = (Object *)tselem->id; - RNA_id_pointer_create(&ob->id, &ptr); - - if (space_outliner->show_restrict_flags & SO_RESTRICT_HIDE) { - Base *base = (te->directdata) ? (Base *)te->directdata : - BKE_view_layer_base_find(view_layer, ob); - if (base) { - PointerRNA base_ptr; - RNA_pointer_create(&scene->id, &RNA_ObjectBase, base, &base_ptr); - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.hide), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &base_ptr, - props.base_hide_viewport, - -1, - 0, - 0, - 0, - 0, - TIP_("Temporarily hide in viewport\n" - "* Shift to set children")); - UI_but_func_set( - bt, outliner__base_set_flag_recursive_fn, base, (void *)"hide_viewport"); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.base_hide_viewport) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_SELECT) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.select), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &ptr, - props.object_hide_select, - -1, - 0, - 0, - -1, - -1, - TIP_("Disable selection in viewport\n" - "* Shift to set children")); - UI_but_func_set(bt, outliner__object_set_flag_recursive_fn, ob, (char *)"hide_select"); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.object_hide_select) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_VIEWPORT) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.viewport), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &ptr, - props.object_hide_viewport, - -1, - 0, - 0, - -1, - -1, - TIP_("Globally disable in viewports\n" - "* Shift to set children")); - UI_but_func_set(bt, outliner__object_set_flag_recursive_fn, ob, (void *)"hide_viewport"); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.object_hide_viewport) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_RENDER) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.render), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &ptr, - props.object_hide_render, - -1, - 0, - 0, - -1, - -1, - TIP_("Globally disable in renders\n" - "* Shift to set children")); - UI_but_func_set(bt, outliner__object_set_flag_recursive_fn, ob, (char *)"hide_render"); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.object_hide_render) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - } - else if (tselem->type == TSE_CONSTRAINT) { - bConstraint *con = (bConstraint *)te->directdata; - - PointerRNA ptr; - RNA_pointer_create(tselem->id, &RNA_Constraint, con, &ptr); - - if (space_outliner->show_restrict_flags & SO_RESTRICT_HIDE) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.hide), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &ptr, - props.constraint_enable, - -1, - 0, - 0, - -1, - -1, - NULL); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.constraint_enable) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - } - else if (tselem->type == TSE_MODIFIER) { - ModifierData *md = (ModifierData *)te->directdata; - - PointerRNA ptr; - RNA_pointer_create(tselem->id, &RNA_Modifier, md, &ptr); - - if (space_outliner->show_restrict_flags & SO_RESTRICT_VIEWPORT) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.viewport), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &ptr, - props.modifier_show_viewport, - -1, - 0, - 0, - -1, - -1, - NULL); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.modifier_show_viewport) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_RENDER) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.render), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &ptr, - props.modifier_show_render, - -1, - 0, - 0, - -1, - -1, - NULL); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.modifier_show_render) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - } - else if (tselem->type == TSE_POSE_CHANNEL) { - PointerRNA ptr; - bPoseChannel *pchan = (bPoseChannel *)te->directdata; - Bone *bone = pchan->bone; - Object *ob = (Object *)tselem->id; - bArmature *arm = ob->data; - - RNA_pointer_create(&arm->id, &RNA_Bone, bone, &ptr); - - if (space_outliner->show_restrict_flags & SO_RESTRICT_VIEWPORT) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.viewport), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &ptr, - props.bone_hide_viewport, - -1, - 0, - 0, - -1, - -1, - TIP_("Restrict visibility in the 3D View\n" - "* Shift to set children")); - UI_but_func_set(bt, restrictbutton_bone_visibility_fn, bone, NULL); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - UI_but_drawflag_enable(bt, UI_BUT_ICON_REVERSE); - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_SELECT) { - bt = uiDefIconButBitI(block, - UI_BTYPE_ICON_TOGGLE, - BONE_UNSELECTABLE, - 0, - ICON_RESTRICT_SELECT_OFF, - (int)(region->v2d.cur.xmax - restrict_offsets.select), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &(bone->flag), - 0, - 0, - 0, - 0, - TIP_("Restrict selection in the 3D View\n" - "* Shift to set children")); - UI_but_func_set(bt, restrictbutton_bone_select_fn, ob->data, bone); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - UI_but_drawflag_enable(bt, UI_BUT_ICON_REVERSE); - } - } - else if (tselem->type == TSE_EBONE) { - bArmature *arm = (bArmature *)tselem->id; - EditBone *ebone = (EditBone *)te->directdata; - - if (space_outliner->show_restrict_flags & SO_RESTRICT_VIEWPORT) { - bt = uiDefIconButBitI(block, - UI_BTYPE_ICON_TOGGLE, - BONE_HIDDEN_A, - 0, - ICON_RESTRICT_VIEW_OFF, - (int)(region->v2d.cur.xmax - restrict_offsets.viewport), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &(ebone->flag), - 0, - 0, - 0, - 0, - TIP_("Restrict visibility in the 3D View\n" - "* Shift to set children")); - UI_but_func_set(bt, restrictbutton_ebone_visibility_fn, arm, ebone); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - UI_but_drawflag_enable(bt, UI_BUT_ICON_REVERSE); - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_SELECT) { - bt = uiDefIconButBitI(block, - UI_BTYPE_ICON_TOGGLE, - BONE_UNSELECTABLE, - 0, - ICON_RESTRICT_SELECT_OFF, - (int)(region->v2d.cur.xmax - restrict_offsets.select), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &(ebone->flag), - 0, - 0, - 0, - 0, - TIP_("Restrict selection in the 3D View\n" - "* Shift to set children")); - UI_but_func_set(bt, restrictbutton_ebone_select_fn, arm, ebone); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - UI_but_drawflag_enable(bt, UI_BUT_ICON_REVERSE); - } - } - else if (tselem->type == TSE_GP_LAYER) { - ID *id = tselem->id; - bGPDlayer *gpl = (bGPDlayer *)te->directdata; - - if (space_outliner->show_restrict_flags & SO_RESTRICT_HIDE) { - bt = uiDefIconButBitS(block, - UI_BTYPE_ICON_TOGGLE, - GP_LAYER_HIDE, - 0, - ICON_HIDE_OFF, - (int)(region->v2d.cur.xmax - restrict_offsets.hide), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &gpl->flag, - 0, - 0, - 0, - 0, - TIP_("Restrict visibility in the 3D View")); - UI_but_func_set(bt, restrictbutton_gp_layer_flag_fn, id, gpl); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - UI_but_drawflag_enable(bt, UI_BUT_ICON_REVERSE); - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_SELECT) { - bt = uiDefIconButBitS(block, - UI_BTYPE_ICON_TOGGLE, - GP_LAYER_LOCKED, - 0, - ICON_UNLOCKED, - (int)(region->v2d.cur.xmax - restrict_offsets.select), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &gpl->flag, - 0, - 0, - 0, - 0, - TIP_("Restrict editing of strokes and keyframes in this layer")); - UI_but_func_set(bt, restrictbutton_gp_layer_flag_fn, id, gpl); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - } - } - else if (outliner_is_collection_tree_element(te)) { - PointerRNA collection_ptr; - PointerRNA layer_collection_ptr; - - if (outliner_restrict_properties_collection_set( - scene, te, &collection_ptr, &layer_collection_ptr, &props, &props_active)) { - - LayerCollection *layer_collection = (tselem->type == TSE_LAYER_COLLECTION) ? - te->directdata : - NULL; - Collection *collection = outliner_collection_from_tree_element(te); - - if (layer_collection != NULL) { - if (space_outliner->show_restrict_flags & SO_RESTRICT_ENABLE) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax) - restrict_offsets.enable, - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &layer_collection_ptr, - props.layer_collection_exclude, - -1, - 0, - 0, - 0, - 0, - NULL); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_HIDE) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.hide), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &layer_collection_ptr, - props.layer_collection_hide_viewport, - -1, - 0, - 0, - 0, - 0, - TIP_("Temporarily hide in viewport\n" - "* Ctrl to isolate collection\n" - "* Shift to set inside collections and objects")); - UI_but_func_set(bt, - view_layer__layer_collection_set_flag_recursive_fn, - layer_collection, - (char *)"hide_viewport"); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.layer_collection_hide_viewport) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_HOLDOUT) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.holdout), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &layer_collection_ptr, - props.layer_collection_holdout, - -1, - 0, - 0, - 0, - 0, - TIP_("Mask out objects in collection from view layer\n" - "* Ctrl to isolate collection\n" - "* Shift to set inside collections")); - UI_but_func_set(bt, - view_layer__layer_collection_set_flag_recursive_fn, - layer_collection, - (char *)"holdout"); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.layer_collection_holdout) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_INDIRECT_ONLY) { - bt = uiDefIconButR_prop( - block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.indirect_only), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &layer_collection_ptr, - props.layer_collection_indirect_only, - -1, - 0, - 0, - 0, - 0, - TIP_("Objects in collection only contribute indirectly (through shadows and " - "reflections) in the view layer\n" - "* Ctrl to isolate collection\n" - "* Shift to set inside collections")); - UI_but_func_set(bt, - view_layer__layer_collection_set_flag_recursive_fn, - layer_collection, - (char *)"indirect_only"); - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (props_active.layer_collection_holdout || - !props_active.layer_collection_indirect_only) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_VIEWPORT) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.viewport), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &collection_ptr, - props.collection_hide_viewport, - -1, - 0, - 0, - 0, - 0, - TIP_("Globally disable in viewports\n" - "* Ctrl to isolate collection\n" - "* Shift to set inside collections and objects")); - if (layer_collection != NULL) { - UI_but_func_set(bt, - view_layer__collection_set_flag_recursive_fn, - layer_collection, - (char *)"hide_viewport"); - } - else { - UI_but_func_set(bt, - scenes__collection_set_flag_recursive_fn, - collection, - (char *)"hide_viewport"); - } - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.collection_hide_viewport) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_RENDER) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.render), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &collection_ptr, - props.collection_hide_render, - -1, - 0, - 0, - 0, - 0, - TIP_("Globally disable in renders\n" - "* Ctrl to isolate collection\n" - "* Shift to set inside collections and objects")); - if (layer_collection != NULL) { - UI_but_func_set(bt, - view_layer__collection_set_flag_recursive_fn, - layer_collection, - (char *)"hide_render"); - } - else { - UI_but_func_set( - bt, scenes__collection_set_flag_recursive_fn, collection, (char *)"hide_render"); - } - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.collection_hide_render) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - - if (space_outliner->show_restrict_flags & SO_RESTRICT_SELECT) { - bt = uiDefIconButR_prop(block, - UI_BTYPE_ICON_TOGGLE, - 0, - 0, - (int)(region->v2d.cur.xmax - restrict_offsets.select), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &collection_ptr, - props.collection_hide_select, - -1, - 0, - 0, - 0, - 0, - TIP_("Disable selection in viewport\n" - "* Ctrl to isolate collection\n" - "* Shift to set inside collections and objects")); - if (layer_collection != NULL) { - UI_but_func_set(bt, - view_layer__collection_set_flag_recursive_fn, - layer_collection, - (char *)"hide_select"); - } - else { - UI_but_func_set( - bt, scenes__collection_set_flag_recursive_fn, collection, (char *)"hide_select"); - } - UI_but_flag_enable(bt, UI_BUT_DRAG_LOCK); - if (!props_active.collection_hide_select) { - UI_but_flag_enable(bt, UI_BUT_INACTIVE); - } - } - } - } - } - else if (outliner_is_collection_tree_element(te)) { - PointerRNA collection_ptr; - PointerRNA layer_collection_ptr; - outliner_restrict_properties_collection_set( - scene, te, &collection_ptr, &layer_collection_ptr, &props, &props_active); - } - - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_draw_restrictbuts( - block, scene, view_layer, region, space_outliner, &te->subtree, props_active); - } - } -} - -static void outliner_draw_userbuts(uiBlock *block, - ARegion *region, - SpaceOutliner *space_outliner, - ListBase *lb) -{ - - LISTBASE_FOREACH (TreeElement *, te, lb) { - TreeStoreElem *tselem = TREESTORE(te); - if (te->ys + 2 * UI_UNIT_Y >= region->v2d.cur.ymin && te->ys <= region->v2d.cur.ymax) { - if (tselem->type == TSE_SOME_ID) { - uiBut *bt; - ID *id = tselem->id; - const char *tip = NULL; - char buf[16] = ""; - int but_flag = UI_BUT_DRAG_LOCK; - - if (ID_IS_LINKED(id)) { - but_flag |= UI_BUT_DISABLED; - } - - BLI_str_format_int_grouped(buf, id->us); - bt = uiDefBut(block, - UI_BTYPE_BUT, - 1, - buf, - (int)(region->v2d.cur.xmax - OL_TOG_USER_BUTS_USERS), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - NULL, - 0.0, - 0.0, - 0, - 0, - TIP_("Number of users of this data-block")); - UI_but_flag_enable(bt, but_flag); - - if (id->flag & LIB_FAKEUSER) { - tip = TIP_("Data-block will be retained using a fake user"); - } - else { - tip = TIP_("Data-block has no users and will be deleted"); - } - bt = uiDefIconButBitS(block, - UI_BTYPE_ICON_TOGGLE, - LIB_FAKEUSER, - 1, - ICON_FAKE_USER_OFF, - (int)(region->v2d.cur.xmax - OL_TOG_USER_BUTS_STATUS), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - &id->flag, - 0, - 0, - 0, - 0, - tip); - UI_but_func_set(bt, restrictbutton_id_user_toggle, id, NULL); - UI_but_flag_enable(bt, but_flag); - } - } - - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_draw_userbuts(block, region, space_outliner, &te->subtree); - } - } -} - -static bool outliner_draw_overrides_buts(uiBlock *block, - ARegion *region, - SpaceOutliner *space_outliner, - ListBase *lb, - const bool is_open) -{ - bool any_item_has_warnings = false; - - LISTBASE_FOREACH (TreeElement *, te, lb) { - bool item_has_warnings = false; - const bool do_draw = (te->ys + 2 * UI_UNIT_Y >= region->v2d.cur.ymin && - te->ys <= region->v2d.cur.ymax); - int but_flag = UI_BUT_DRAG_LOCK; - const char *tip = NULL; - - TreeStoreElem *tselem = TREESTORE(te); - switch (tselem->type) { - case TSE_LIBRARY_OVERRIDE_BASE: { - ID *id = tselem->id; - - if (id->flag & LIB_LIB_OVERRIDE_RESYNC_LEFTOVER) { - item_has_warnings = true; - if (do_draw) { - tip = TIP_( - "This override data-block is not needed anymore, but was detected as user-edited"); - } - } - else if (ID_IS_OVERRIDE_LIBRARY_REAL(id) && ID_REAL_USERS(id) == 0) { - item_has_warnings = true; - if (do_draw) { - tip = TIP_("This override data-block is unused"); - } - } - break; - } - case TSE_LIBRARY_OVERRIDE: { - const bool is_rna_path_valid = (bool)(POINTER_AS_UINT(te->directdata)); - if (!is_rna_path_valid) { - item_has_warnings = true; - if (do_draw) { - tip = TIP_( - "This override property does not exist in current data, it will be removed on " - "next .blend file save"); - } - } - break; - } - default: - break; - } - - const bool any_child_has_warnings = outliner_draw_overrides_buts( - block, - region, - space_outliner, - &te->subtree, - is_open && TSELEM_OPEN(tselem, space_outliner)); - - if (do_draw && - (item_has_warnings || (any_child_has_warnings && !TSELEM_OPEN(tselem, space_outliner)))) { - if (tip == NULL) { - tip = TIP_("Some sub-items require attention"); - } - uiBut *bt = uiDefIconBut(block, - UI_BTYPE_BUT, - 1, - ICON_ERROR, - (int)(region->v2d.cur.xmax - OL_TOG_USER_BUTS_STATUS), - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - NULL, - 0.0, - 0.0, - 0.0, - 0.0, - tip); - UI_but_flag_enable(bt, but_flag); - } - any_item_has_warnings = any_item_has_warnings || item_has_warnings || any_child_has_warnings; - } - - return any_item_has_warnings; -} - -static void outliner_draw_rnacols(ARegion *region, int sizex) -{ - View2D *v2d = ®ion->v2d; - - float miny = v2d->cur.ymin; - if (miny < v2d->tot.ymin) { - miny = v2d->tot.ymin; - } - - GPU_line_width(1.0f); - - uint pos = GPU_vertformat_attr_add(immVertexFormat(), "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT); - immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR); - immUniformThemeColorShadeAlpha(TH_BACK, -15, -200); - - immBegin(GPU_PRIM_LINES, 4); - - immVertex2f(pos, sizex, v2d->cur.ymax); - immVertex2f(pos, sizex, miny); - - immVertex2f(pos, sizex + OL_RNA_COL_SIZEX, v2d->cur.ymax); - immVertex2f(pos, sizex + OL_RNA_COL_SIZEX, miny); - - immEnd(); - - immUnbindProgram(); -} - -static void outliner_draw_rnabuts( - uiBlock *block, ARegion *region, SpaceOutliner *space_outliner, int sizex, ListBase *lb) -{ - PointerRNA *ptr; - PropertyRNA *prop; - - LISTBASE_FOREACH (TreeElement *, te, lb) { - TreeStoreElem *tselem = TREESTORE(te); - if (te->ys + 2 * UI_UNIT_Y >= region->v2d.cur.ymin && te->ys <= region->v2d.cur.ymax) { - if (tselem->type == TSE_RNA_PROPERTY) { - ptr = &te->rnaptr; - prop = te->directdata; - - if (!TSELEM_OPEN(tselem, space_outliner)) { - if (RNA_property_type(prop) == PROP_POINTER) { - uiBut *but = uiDefAutoButR(block, - ptr, - prop, - -1, - "", - ICON_NONE, - sizex, - te->ys, - OL_RNA_COL_SIZEX, - UI_UNIT_Y - 1); - UI_but_flag_enable(but, UI_BUT_DISABLED); - } - else if (RNA_property_type(prop) == PROP_ENUM) { - uiDefAutoButR(block, - ptr, - prop, - -1, - NULL, - ICON_NONE, - sizex, - te->ys, - OL_RNA_COL_SIZEX, - UI_UNIT_Y - 1); - } - else { - uiDefAutoButR(block, - ptr, - prop, - -1, - "", - ICON_NONE, - sizex, - te->ys, - OL_RNA_COL_SIZEX, - UI_UNIT_Y - 1); - } - } - } - else if (tselem->type == TSE_RNA_ARRAY_ELEM) { - ptr = &te->rnaptr; - prop = te->directdata; - - uiDefAutoButR(block, - ptr, - prop, - te->index, - "", - ICON_NONE, - sizex, - te->ys, - OL_RNA_COL_SIZEX, - UI_UNIT_Y - 1); - } - } - - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_draw_rnabuts(block, region, space_outliner, sizex, &te->subtree); - } - } -} - -static void outliner_buttons(const bContext *C, - uiBlock *block, - ARegion *region, - const float restrict_column_width, - TreeElement *te) -{ - uiBut *bt; - TreeStoreElem *tselem; - int spx, dx, len; - - tselem = TREESTORE(te); - - BLI_assert(tselem->flag & TSE_TEXTBUT); - /* If we add support to rename Sequence, need change this. */ - - if (tselem->type == TSE_EBONE) { - len = sizeof(((EditBone *)0)->name); - } - else if (tselem->type == TSE_MODIFIER) { - len = sizeof(((ModifierData *)0)->name); - } - else if (tselem->id && GS(tselem->id->name) == ID_LI) { - len = sizeof(((Library *)0)->filepath); - } - else { - len = MAX_ID_NAME - 2; - } - - spx = te->xs + 1.8f * UI_UNIT_X; - dx = region->v2d.cur.xmax - (spx + restrict_column_width + 0.2f * UI_UNIT_X); - - bt = uiDefBut(block, - UI_BTYPE_TEXT, - OL_NAMEBUTTON, - "", - spx, - te->ys, - dx, - UI_UNIT_Y - 1, - (void *)te->name, - 1.0, - (float)len, - 0, - 0, - ""); - UI_but_func_rename_set(bt, namebutton_fn, tselem); - - /* Returns false if button got removed. */ - if (false == UI_but_active_only(C, region, block, bt)) { - tselem->flag &= ~TSE_TEXTBUT; - - /* Bad! (notifier within draw) without this, we don't get a refresh. */ - WM_event_add_notifier(C, NC_SPACE | ND_SPACE_OUTLINER, NULL); - } -} - -static void outliner_mode_toggle_fn(bContext *C, void *tselem_poin, void *UNUSED(arg2)) -{ - SpaceOutliner *space_outliner = CTX_wm_space_outliner(C); - TreeStoreElem *tselem = (TreeStoreElem *)tselem_poin; - TreeViewContext tvc; - outliner_viewcontext_init(C, &tvc); - - TreeElement *te = outliner_find_tree_element(&space_outliner->tree, tselem); - if (!te) { - return; - } - - /* Check that the item is actually an object. */ - BLI_assert(tselem->id != NULL && GS(tselem->id->name) == ID_OB); - - Object *ob = (Object *)tselem->id; - const bool object_data_shared = (ob->data == tvc.obact->data); - - wmWindow *win = CTX_wm_window(C); - const bool do_extend = win->eventstate->ctrl != 0 && !object_data_shared; - outliner_item_mode_toggle(C, &tvc, te, do_extend); -} - -/* Draw icons for adding and removing objects from the current interaction mode. */ -static void outliner_draw_mode_column_toggle(uiBlock *block, - TreeViewContext *tvc, - TreeElement *te, - TreeStoreElem *tselem, - const bool lock_object_modes) -{ - if ((tselem->type != TSE_SOME_ID) || (te->idcode != ID_OB)) { - return; - } - - Object *ob = (Object *)tselem->id; - Object *ob_active = tvc->obact; - - /* Not all objects support particle systems. */ - if (ob_active->mode == OB_MODE_PARTICLE_EDIT && !psys_get_current(ob)) { - return; - } - - /* Only for objects with the same type. */ - if (ob->type != ob_active->type) { - return; - } - - bool draw_active_icon = ob->mode == ob_active->mode; - - /* When not locking object modes, objects can remain in non-object modes. For modes that do not - * allow multi-object editing, these other objects should still show be viewed as not in the - * mode. Otherwise multiple objects show the same mode icon in the outliner even though only - * one object is actually editable in the mode. */ - if (!lock_object_modes && ob != ob_active && !(tvc->ob_edit || tvc->ob_pose)) { - draw_active_icon = false; - } - - const bool object_data_shared = (ob->data == ob_active->data); - draw_active_icon = draw_active_icon || object_data_shared; - - int icon; - const char *tip; - if (draw_active_icon) { - icon = UI_icon_from_object_mode(ob_active->mode); - tip = object_data_shared ? TIP_("Change the object in the current mode") : - TIP_("Remove from the current mode"); - } - else { - icon = ICON_DOT; - tip = TIP_( - "Change the object in the current mode\n" - "* Ctrl to add to the current mode"); - } - UI_block_emboss_set(block, UI_EMBOSS_NONE_OR_STATUS); - uiBut *but = uiDefIconBut(block, - UI_BTYPE_ICON_TOGGLE, - 0, - icon, - 0, - te->ys, - UI_UNIT_X, - UI_UNIT_Y, - NULL, - 0.0, - 0.0, - 0.0, - 0.0, - tip); - UI_but_func_set(but, outliner_mode_toggle_fn, tselem, NULL); - UI_but_flag_enable(but, UI_BUT_DRAG_LOCK); - /* Mode toggling handles its own undo state because undo steps need to be grouped. */ - UI_but_flag_disable(but, UI_BUT_UNDO); - - if (ID_IS_LINKED(&ob->id)) { - UI_but_disable(but, TIP_("Can't edit external library data")); - } -} - -static void outliner_draw_mode_column(const bContext *C, - uiBlock *block, - TreeViewContext *tvc, - SpaceOutliner *space_outliner, - ListBase *tree) -{ - TreeStoreElem *tselem; - const bool lock_object_modes = tvc->scene->toolsettings->object_flag & SCE_OBJECT_MODE_LOCK; - - LISTBASE_FOREACH (TreeElement *, te, tree) { - tselem = TREESTORE(te); - - if (tvc->obact && tvc->obact->mode != OB_MODE_OBJECT) { - outliner_draw_mode_column_toggle(block, tvc, te, tselem, lock_object_modes); - } - - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_draw_mode_column(C, block, tvc, space_outliner, &te->subtree); - } - } -} - -/* Returns `true` if some warning was drawn for that element or one of its sub-elements (if it is - * not open). */ -static bool outliner_draw_warning_tree_element(uiBlock *block, - SpaceOutliner *space_outliner, - TreeElement *te, - TreeStoreElem *tselem, - const bool use_mode_column, - const int te_ys) -{ - if ((te->flag & TE_HAS_WARNING) == 0) { - /* If given element has no warning, recusively try to display the first sub-elements' warning. - */ - if (!TSELEM_OPEN(tselem, space_outliner)) { - LISTBASE_FOREACH (TreeElement *, sub_te, &te->subtree) { - TreeStoreElem *sub_tselem = TREESTORE(sub_te); - - if (outliner_draw_warning_tree_element( - block, space_outliner, sub_te, sub_tselem, use_mode_column, te_ys)) { - return true; - } - } - } - return false; - } - - int icon = ICON_NONE; - const char *tip = ""; - const bool has_warning = outliner_element_warnings_get(te, &icon, &tip); - BLI_assert(has_warning); - UNUSED_VARS_NDEBUG(has_warning); - - /* Move the warnings a unit left in view layer mode. */ - const short mode_column_offset = (use_mode_column && (space_outliner->outlinevis == SO_SCENES)) ? - UI_UNIT_X : - 0; - - UI_block_emboss_set(block, UI_EMBOSS_NONE_OR_STATUS); - uiBut *but = uiDefIconBut(block, - UI_BTYPE_ICON_TOGGLE, - 0, - icon, - mode_column_offset, - te_ys, - UI_UNIT_X, - UI_UNIT_Y, - NULL, - 0.0, - 0.0, - 0.0, - 0.0, - tip); - /* No need for undo here, this is a pure info widget. */ - UI_but_flag_disable(but, UI_BUT_UNDO); - - return true; -} - -static void outliner_draw_warning_column(const bContext *C, - uiBlock *block, - SpaceOutliner *space_outliner, - const bool use_mode_column, - ListBase *tree) -{ - LISTBASE_FOREACH (TreeElement *, te, tree) { - TreeStoreElem *tselem = TREESTORE(te); - - outliner_draw_warning_tree_element(block, space_outliner, te, tselem, use_mode_column, te->ys); - - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_draw_warning_column(C, block, space_outliner, use_mode_column, &te->subtree); - } - } -} - -/* ****************************************************** */ -/* Normal Drawing... */ - -TreeElementIcon tree_element_get_icon(TreeStoreElem *tselem, TreeElement *te) -{ - TreeElementIcon data = {0}; - - if (tselem->type != TSE_SOME_ID) { - switch (tselem->type) { - case TSE_ANIM_DATA: - data.icon = ICON_ANIM_DATA; /* XXX */ - break; - case TSE_NLA: - data.icon = ICON_NLA; - break; - case TSE_NLA_TRACK: - data.icon = ICON_NLA; /* XXX */ - break; - case TSE_NLA_ACTION: - data.icon = ICON_ACTION; - break; - case TSE_DRIVER_BASE: - data.icon = ICON_DRIVER; - break; - case TSE_DEFGROUP_BASE: - data.icon = ICON_GROUP_VERTEX; - break; - case TSE_DEFGROUP: - data.icon = ICON_GROUP_VERTEX; - break; - case TSE_BONE: - case TSE_EBONE: - data.icon = ICON_BONE_DATA; - break; - case TSE_CONSTRAINT_BASE: - data.icon = ICON_CONSTRAINT; - data.drag_id = tselem->id; - break; - case TSE_CONSTRAINT: { - bConstraint *con = te->directdata; - data.drag_id = tselem->id; - switch ((eBConstraint_Types)con->type) { - case CONSTRAINT_TYPE_CAMERASOLVER: - data.icon = ICON_CON_CAMERASOLVER; - break; - case CONSTRAINT_TYPE_FOLLOWTRACK: - data.icon = ICON_CON_FOLLOWTRACK; - break; - case CONSTRAINT_TYPE_OBJECTSOLVER: - data.icon = ICON_CON_OBJECTSOLVER; - break; - case CONSTRAINT_TYPE_LOCLIKE: - data.icon = ICON_CON_LOCLIKE; - break; - case CONSTRAINT_TYPE_ROTLIKE: - data.icon = ICON_CON_ROTLIKE; - break; - case CONSTRAINT_TYPE_SIZELIKE: - data.icon = ICON_CON_SIZELIKE; - break; - case CONSTRAINT_TYPE_TRANSLIKE: - data.icon = ICON_CON_TRANSLIKE; - break; - case CONSTRAINT_TYPE_DISTLIMIT: - data.icon = ICON_CON_DISTLIMIT; - break; - case CONSTRAINT_TYPE_LOCLIMIT: - data.icon = ICON_CON_LOCLIMIT; - break; - case CONSTRAINT_TYPE_ROTLIMIT: - data.icon = ICON_CON_ROTLIMIT; - break; - case CONSTRAINT_TYPE_SIZELIMIT: - data.icon = ICON_CON_SIZELIMIT; - break; - case CONSTRAINT_TYPE_SAMEVOL: - data.icon = ICON_CON_SAMEVOL; - break; - case CONSTRAINT_TYPE_TRANSFORM: - data.icon = ICON_CON_TRANSFORM; - break; - case CONSTRAINT_TYPE_TRANSFORM_CACHE: - data.icon = ICON_CON_TRANSFORM_CACHE; - break; - case CONSTRAINT_TYPE_CLAMPTO: - data.icon = ICON_CON_CLAMPTO; - break; - case CONSTRAINT_TYPE_DAMPTRACK: - data.icon = ICON_CON_TRACKTO; - break; - case CONSTRAINT_TYPE_KINEMATIC: - data.icon = ICON_CON_KINEMATIC; - break; - case CONSTRAINT_TYPE_LOCKTRACK: - data.icon = ICON_CON_LOCKTRACK; - break; - case CONSTRAINT_TYPE_SPLINEIK: - data.icon = ICON_CON_SPLINEIK; - break; - case CONSTRAINT_TYPE_STRETCHTO: - data.icon = ICON_CON_STRETCHTO; - break; - case CONSTRAINT_TYPE_TRACKTO: - data.icon = ICON_CON_TRACKTO; - break; - case CONSTRAINT_TYPE_ACTION: - data.icon = ICON_CON_ACTION; - break; - case CONSTRAINT_TYPE_ARMATURE: - data.icon = ICON_CON_ARMATURE; - break; - case CONSTRAINT_TYPE_CHILDOF: - data.icon = ICON_CON_CHILDOF; - break; - case CONSTRAINT_TYPE_MINMAX: - data.icon = ICON_CON_FLOOR; - break; - case CONSTRAINT_TYPE_FOLLOWPATH: - data.icon = ICON_CON_FOLLOWPATH; - break; - case CONSTRAINT_TYPE_PIVOT: - data.icon = ICON_CON_PIVOT; - break; - case CONSTRAINT_TYPE_SHRINKWRAP: - data.icon = ICON_CON_SHRINKWRAP; - break; - - default: - data.icon = ICON_DOT; - break; - } - break; - } - case TSE_MODIFIER_BASE: - data.icon = ICON_MODIFIER_DATA; - data.drag_id = tselem->id; - break; - case TSE_LIBRARY_OVERRIDE_BASE: - case TSE_LIBRARY_OVERRIDE: - data.icon = ICON_LIBRARY_DATA_OVERRIDE; - break; - case TSE_LINKED_OB: - data.icon = ICON_OBJECT_DATA; - break; - case TSE_LINKED_PSYS: - data.icon = ICON_PARTICLES; - break; - case TSE_MODIFIER: { - Object *ob = (Object *)tselem->id; - data.drag_id = tselem->id; - - if (ob->type != OB_GPENCIL) { - ModifierData *md = BLI_findlink(&ob->modifiers, tselem->nr); - const ModifierTypeInfo *modifier_type = BKE_modifier_get_info(md->type); - if (modifier_type != NULL) { - data.icon = modifier_type->icon; - } - else { - data.icon = ICON_DOT; - } - } - else { - /* grease pencil modifiers */ - GpencilModifierData *md = BLI_findlink(&ob->greasepencil_modifiers, tselem->nr); - switch ((GpencilModifierType)md->type) { - case eGpencilModifierType_Noise: - data.icon = ICON_MOD_NOISE; - break; - case eGpencilModifierType_Subdiv: - data.icon = ICON_MOD_SUBSURF; - break; - case eGpencilModifierType_Thick: - data.icon = ICON_MOD_THICKNESS; - break; - case eGpencilModifierType_Tint: - data.icon = ICON_MOD_TINT; - break; - case eGpencilModifierType_Array: - data.icon = ICON_MOD_ARRAY; - break; - case eGpencilModifierType_Build: - data.icon = ICON_MOD_BUILD; - break; - case eGpencilModifierType_Opacity: - data.icon = ICON_MOD_MASK; - break; - case eGpencilModifierType_Color: - data.icon = ICON_MOD_HUE_SATURATION; - break; - case eGpencilModifierType_Lattice: - data.icon = ICON_MOD_LATTICE; - break; - case eGpencilModifierType_Mirror: - data.icon = ICON_MOD_MIRROR; - break; - case eGpencilModifierType_Simplify: - data.icon = ICON_MOD_SIMPLIFY; - break; - case eGpencilModifierType_Smooth: - data.icon = ICON_MOD_SMOOTH; - break; - case eGpencilModifierType_Hook: - data.icon = ICON_HOOK; - break; - case eGpencilModifierType_Offset: - data.icon = ICON_MOD_OFFSET; - break; - case eGpencilModifierType_Armature: - data.icon = ICON_MOD_ARMATURE; - break; - case eGpencilModifierType_Multiply: - data.icon = ICON_GP_MULTIFRAME_EDITING; - break; - case eGpencilModifierType_Time: - data.icon = ICON_MOD_TIME; - break; - case eGpencilModifierType_Texture: - data.icon = ICON_TEXTURE; - break; - case eGpencilModifierType_WeightProximity: - data.icon = ICON_MOD_VERTEX_WEIGHT; - break; - case eGpencilModifierType_WeightAngle: - data.icon = ICON_MOD_VERTEX_WEIGHT; - break; - case eGpencilModifierType_Shrinkwrap: - data.icon = ICON_MOD_SHRINKWRAP; - break; - - /* Default */ - default: - data.icon = ICON_DOT; - break; - } - } - break; - } - case TSE_POSE_BASE: - data.icon = ICON_ARMATURE_DATA; - break; - case TSE_POSE_CHANNEL: - data.icon = ICON_BONE_DATA; - break; - case TSE_PROXY: - data.icon = ICON_GHOST_ENABLED; - break; - case TSE_R_LAYER_BASE: - data.icon = ICON_RENDERLAYERS; - break; - case TSE_SCENE_OBJECTS_BASE: - data.icon = ICON_OUTLINER_OB_GROUP_INSTANCE; - break; - case TSE_R_LAYER: - data.icon = ICON_RENDER_RESULT; - break; - case TSE_POSEGRP_BASE: - case TSE_POSEGRP: - data.icon = ICON_GROUP_BONE; - break; - case TSE_SEQUENCE: - switch (te->idcode) { - case SEQ_TYPE_SCENE: - data.icon = ICON_SCENE_DATA; - break; - case SEQ_TYPE_MOVIECLIP: - data.icon = ICON_TRACKER; - break; - case SEQ_TYPE_MASK: - data.icon = ICON_MOD_MASK; - break; - case SEQ_TYPE_MOVIE: - data.icon = ICON_FILE_MOVIE; - break; - case SEQ_TYPE_SOUND_RAM: - data.icon = ICON_SOUND; - break; - case SEQ_TYPE_IMAGE: - data.icon = ICON_FILE_IMAGE; - break; - case SEQ_TYPE_COLOR: - case SEQ_TYPE_ADJUSTMENT: - data.icon = ICON_COLOR; - break; - case SEQ_TYPE_TEXT: - data.icon = ICON_FONT_DATA; - break; - case SEQ_TYPE_ADD: - case SEQ_TYPE_SUB: - case SEQ_TYPE_MUL: - case SEQ_TYPE_OVERDROP: - case SEQ_TYPE_ALPHAOVER: - case SEQ_TYPE_ALPHAUNDER: - case SEQ_TYPE_COLORMIX: - case SEQ_TYPE_MULTICAM: - case SEQ_TYPE_TRANSFORM: - case SEQ_TYPE_SPEED: - case SEQ_TYPE_GLOW: - case SEQ_TYPE_GAUSSIAN_BLUR: - data.icon = ICON_SHADERFX; - break; - case SEQ_TYPE_CROSS: - case SEQ_TYPE_GAMCROSS: - case SEQ_TYPE_WIPE: - data.icon = ICON_ARROW_LEFTRIGHT; - break; - case SEQ_TYPE_META: - data.icon = ICON_SEQ_STRIP_META; - break; - default: - data.icon = ICON_DOT; - break; - } - break; - case TSE_SEQ_STRIP: - data.icon = ICON_LIBRARY_DATA_DIRECT; - break; - case TSE_SEQUENCE_DUP: - data.icon = ICON_SEQ_STRIP_DUPLICATE; - break; - case TSE_RNA_STRUCT: - if (RNA_struct_is_ID(te->rnaptr.type)) { - data.drag_id = (ID *)te->rnaptr.data; - data.icon = RNA_struct_ui_icon(te->rnaptr.type); - } - else { - data.icon = RNA_struct_ui_icon(te->rnaptr.type); - } - break; - case TSE_LAYER_COLLECTION: - case TSE_SCENE_COLLECTION_BASE: - case TSE_VIEW_COLLECTION_BASE: { - Collection *collection = outliner_collection_from_tree_element(te); - if (collection && !(collection->flag & COLLECTION_IS_MASTER)) { - data.drag_id = tselem->id; - data.drag_parent = (data.drag_id && te->parent) ? TREESTORE(te->parent)->id : NULL; - } - - data.icon = ICON_OUTLINER_COLLECTION; - break; - } - case TSE_GP_LAYER: { - data.icon = ICON_OUTLINER_DATA_GP_LAYER; - break; - } - case TSE_GPENCIL_EFFECT_BASE: - case TSE_GPENCIL_EFFECT: - data.drag_id = tselem->id; - data.icon = ICON_SHADERFX; - break; - default: - data.icon = ICON_DOT; - break; - } - } - else if (tselem->id) { - data.drag_id = tselem->id; - data.drag_parent = (data.drag_id && te->parent) ? TREESTORE(te->parent)->id : NULL; - - if (GS(tselem->id->name) == ID_OB) { - Object *ob = (Object *)tselem->id; - switch (ob->type) { - case OB_LAMP: - data.icon = ICON_OUTLINER_OB_LIGHT; - break; - case OB_MESH: - data.icon = ICON_OUTLINER_OB_MESH; - break; - case OB_CAMERA: - data.icon = ICON_OUTLINER_OB_CAMERA; - break; - case OB_CURVE: - data.icon = ICON_OUTLINER_OB_CURVE; - break; - case OB_MBALL: - data.icon = ICON_OUTLINER_OB_META; - break; - case OB_LATTICE: - data.icon = ICON_OUTLINER_OB_LATTICE; - break; - case OB_ARMATURE: - data.icon = ICON_OUTLINER_OB_ARMATURE; - break; - case OB_FONT: - data.icon = ICON_OUTLINER_OB_FONT; - break; - case OB_SURF: - data.icon = ICON_OUTLINER_OB_SURFACE; - break; - case OB_SPEAKER: - data.icon = ICON_OUTLINER_OB_SPEAKER; - break; - case OB_LIGHTPROBE: - data.icon = ICON_OUTLINER_OB_LIGHTPROBE; - break; - case OB_HAIR: - data.icon = ICON_OUTLINER_OB_HAIR; - break; - case OB_POINTCLOUD: - data.icon = ICON_OUTLINER_OB_POINTCLOUD; - break; - case OB_VOLUME: - data.icon = ICON_OUTLINER_OB_VOLUME; - break; - case OB_EMPTY: - if (ob->instance_collection && (ob->transflag & OB_DUPLICOLLECTION)) { - data.icon = ICON_OUTLINER_OB_GROUP_INSTANCE; - } - else if (ob->empty_drawtype == OB_EMPTY_IMAGE) { - data.icon = ICON_OUTLINER_OB_IMAGE; - } - else if (ob->pd && ob->pd->forcefield) { - data.icon = ICON_OUTLINER_OB_FORCE_FIELD; - } - else { - data.icon = ICON_OUTLINER_OB_EMPTY; - } - break; - case OB_GPENCIL: - data.icon = ICON_OUTLINER_OB_GREASEPENCIL; - break; - } - } - else { - /* TODO(sergey): Casting to short here just to handle ID_NLA which is - * NOT inside of IDType enum. - */ - switch ((short)GS(tselem->id->name)) { - case ID_SCE: - data.icon = ICON_SCENE_DATA; - break; - case ID_ME: - data.icon = ICON_OUTLINER_DATA_MESH; - break; - case ID_CU: - data.icon = ICON_OUTLINER_DATA_CURVE; - break; - case ID_MB: - data.icon = ICON_OUTLINER_DATA_META; - break; - case ID_LT: - data.icon = ICON_OUTLINER_DATA_LATTICE; - break; - case ID_LA: { - Light *la = (Light *)tselem->id; - switch (la->type) { - case LA_LOCAL: - data.icon = ICON_LIGHT_POINT; - break; - case LA_SUN: - data.icon = ICON_LIGHT_SUN; - break; - case LA_SPOT: - data.icon = ICON_LIGHT_SPOT; - break; - case LA_AREA: - data.icon = ICON_LIGHT_AREA; - break; - default: - data.icon = ICON_OUTLINER_DATA_LIGHT; - break; - } - break; - } - case ID_MA: - data.icon = ICON_MATERIAL_DATA; - break; - case ID_TE: - data.icon = ICON_TEXTURE_DATA; - break; - case ID_IM: - data.icon = ICON_IMAGE_DATA; - break; - case ID_SPK: - case ID_SO: - data.icon = ICON_OUTLINER_DATA_SPEAKER; - break; - case ID_AR: - data.icon = ICON_OUTLINER_DATA_ARMATURE; - break; - case ID_CA: - data.icon = ICON_OUTLINER_DATA_CAMERA; - break; - case ID_KE: - data.icon = ICON_SHAPEKEY_DATA; - break; - case ID_WO: - data.icon = ICON_WORLD_DATA; - break; - case ID_AC: - data.icon = ICON_ACTION; - break; - case ID_NLA: - data.icon = ICON_NLA; - break; - case ID_TXT: { - Text *text = (Text *)tselem->id; - if (text->filepath == NULL || (text->flags & TXT_ISMEM)) { - data.icon = ICON_FILE_TEXT; - } - else { - /* Helps distinguish text-based formats like the file-browser does. */ - data.icon = ED_file_extension_icon(text->filepath); - } - break; - } - case ID_GR: - data.icon = ICON_OUTLINER_COLLECTION; - break; - case ID_HA: - data.icon = ICON_OUTLINER_DATA_HAIR; - break; - case ID_PT: - data.icon = ICON_OUTLINER_DATA_POINTCLOUD; - break; - case ID_VO: - data.icon = ICON_OUTLINER_DATA_VOLUME; - break; - case ID_LI: - if (tselem->id->tag & LIB_TAG_MISSING) { - data.icon = ICON_LIBRARY_DATA_BROKEN; - } - else if (((Library *)tselem->id)->parent) { - data.icon = ICON_LIBRARY_DATA_INDIRECT; - } - else { - data.icon = ICON_LIBRARY_DATA_DIRECT; - } - break; - case ID_LS: - data.icon = ICON_LINE_DATA; - break; - case ID_GD: - data.icon = ICON_OUTLINER_DATA_GREASEPENCIL; - break; - case ID_LP: { - LightProbe *lp = (LightProbe *)tselem->id; - switch (lp->type) { - case LIGHTPROBE_TYPE_CUBE: - data.icon = ICON_LIGHTPROBE_CUBEMAP; - break; - case LIGHTPROBE_TYPE_PLANAR: - data.icon = ICON_LIGHTPROBE_PLANAR; - break; - case LIGHTPROBE_TYPE_GRID: - data.icon = ICON_LIGHTPROBE_GRID; - break; - default: - data.icon = ICON_LIGHTPROBE_CUBEMAP; - break; - } - break; - } - case ID_BR: - data.icon = ICON_BRUSH_DATA; - break; - case ID_SCR: - case ID_WS: - data.icon = ICON_WORKSPACE; - break; - case ID_MSK: - data.icon = ICON_MOD_MASK; - break; - case ID_MC: - data.icon = ICON_SEQUENCE; - break; - case ID_PC: - data.icon = ICON_CURVE_BEZCURVE; - break; - case ID_SIM: - /* TODO: Use correct icon. */ - data.icon = ICON_PHYSICS; - break; - default: - break; - } - } - } - - return data; -} - -static void tselem_draw_icon(uiBlock *block, - int xmax, - float x, - float y, - TreeStoreElem *tselem, - TreeElement *te, - float alpha, - const bool is_clickable) -{ - TreeElementIcon data = tree_element_get_icon(tselem, te); - if (data.icon == 0) { - return; - } - - const bool is_collection = outliner_is_collection_tree_element(te); - - /* Collection colors and icons covered by restrict buttons. */ - if (!is_clickable || x >= xmax || is_collection) { - /* Placement of icons, copied from interface_widgets.c */ - float aspect = (0.8f * UI_UNIT_Y) / ICON_DEFAULT_HEIGHT; - x += 2.0f * aspect; - y += 2.0f * aspect; - - if (is_collection) { - Collection *collection = outliner_collection_from_tree_element(te); - if (collection->color_tag != COLLECTION_COLOR_NONE) { - bTheme *btheme = UI_GetTheme(); - UI_icon_draw_ex(x, - y, - data.icon, - U.inv_dpi_fac, - alpha, - 0.0f, - btheme->collection_color[collection->color_tag].color, - true); - return; - } - } - - /* Reduce alpha to match icon buttons */ - alpha *= 0.8f; - - /* Restrict column clip. it has been coded by simply overdrawing, doesn't work for buttons. */ - uchar color[4]; - if (UI_icon_get_theme_color(data.icon, color)) { - UI_icon_draw_ex(x, y, data.icon, U.inv_dpi_fac, alpha, 0.0f, color, true); - } - else { - UI_icon_draw_ex(x, y, data.icon, U.inv_dpi_fac, alpha, 0.0f, NULL, false); - } - } - else { - uiDefIconBut(block, - UI_BTYPE_LABEL, - 0, - data.icon, - x, - y, - UI_UNIT_X, - UI_UNIT_Y, - NULL, - 0.0, - 0.0, - 1.0, - alpha, - (data.drag_id && ID_IS_LINKED(data.drag_id)) ? data.drag_id->lib->filepath : ""); - } -} - -/** - * For icon-only children of a collapsed tree, - * Draw small number over the icon to show how many items of this type are displayed. - */ -static void outliner_draw_iconrow_number(const uiFontStyle *fstyle, - int offsx, - int ys, - const int num_elements) -{ - const float color[4] = {0.0f, 0.0f, 0.0f, 1.0f}; - float ufac = 0.25f * UI_UNIT_X; - float offset_x = (float)offsx + UI_UNIT_X * 0.35f; - - UI_draw_roundbox_corner_set(UI_CNR_ALL); - UI_draw_roundbox_aa( - &(const rctf){ - .xmin = offset_x + ufac, - .xmax = offset_x + UI_UNIT_X - ufac, - .ymin = (float)ys - UI_UNIT_Y * 0.2f + ufac, - .ymax = (float)ys - UI_UNIT_Y * 0.2f + UI_UNIT_Y - ufac, - }, - true, - (float)UI_UNIT_Y / 2.0f - ufac, - color); - - /* Now the numbers. */ - uchar text_col[4]; - - UI_GetThemeColor3ubv(TH_TEXT_HI, text_col); - text_col[3] = 255; - - uiFontStyle fstyle_small = *fstyle; - fstyle_small.points *= 0.8f; - - /* We treat +99 as 4 digits to make sure the (eyeballed) alignment looks nice. */ - int num_digits = 4; - char number_text[4] = "+99\0"; - if (num_elements < 100) { - BLI_snprintf(number_text, sizeof(number_text), "%d", num_elements); - num_digits = num_elements < 10 ? 1 : 2; - } - UI_fontstyle_draw_simple(&fstyle_small, - (offset_x + ufac + UI_UNIT_X * (2 - num_digits) * 0.12f), - (float)ys - UI_UNIT_Y * 0.095f + ufac, - number_text, - text_col); - UI_fontstyle_set(fstyle); - GPU_blend(GPU_BLEND_ALPHA); /* Roundbox and text drawing disables. */ -} - -static void outliner_icon_background_colors(float icon_color[4], float icon_border[4]) -{ - float text[4]; - UI_GetThemeColor4fv(TH_TEXT, text); - - copy_v3_v3(icon_color, text); - icon_color[3] = 0.4f; - copy_v3_v3(icon_border, text); - icon_border[3] = 0.2f; -} - -/* Draw a rounded rectangle behind icons of active elements. */ -static void outliner_draw_active_indicator(const float minx, - const float miny, - const float maxx, - const float maxy, - const float icon_color[4], - const float icon_border[4]) -{ - const float ufac = UI_UNIT_X / 20.0f; - const float radius = UI_UNIT_Y / 4.0f; - - UI_draw_roundbox_corner_set(UI_CNR_ALL); - UI_draw_roundbox_aa( - &(const rctf){ - .xmin = minx, - .xmax = maxx, - .ymin = miny + ufac, - .ymax = maxy - ufac, - }, - true, - radius, - icon_color); - UI_draw_roundbox_aa( - &(const rctf){ - .xmin = minx, - .xmax = maxx, - .ymin = miny + ufac, - .ymax = maxy - ufac, - }, - false, - radius, - icon_border); - GPU_blend(GPU_BLEND_ALPHA); /* Roundbox disables. */ -} - -static void outliner_draw_iconrow_doit(uiBlock *block, - TreeElement *te, - const uiFontStyle *fstyle, - int xmax, - int *offsx, - int ys, - float alpha_fac, - const eOLDrawState active, - const int num_elements) -{ - TreeStoreElem *tselem = TREESTORE(te); - - if (active != OL_DRAWSEL_NONE) { - float icon_color[4], icon_border[4]; - outliner_icon_background_colors(icon_color, icon_border); - if (active == OL_DRAWSEL_ACTIVE) { - UI_GetThemeColor4fv(TH_EDITED_OBJECT, icon_color); - icon_border[3] = 0.3f; - } - - outliner_draw_active_indicator((float)*offsx, - (float)ys, - (float)*offsx + UI_UNIT_X, - (float)ys + UI_UNIT_Y, - icon_color, - icon_border); - } - - if (tselem->flag & TSE_HIGHLIGHTED_ICON) { - alpha_fac += 0.5; - } - tselem_draw_icon(block, xmax, (float)*offsx, (float)ys, tselem, te, alpha_fac, false); - te->xs = *offsx; - te->ys = ys; - te->xend = (short)*offsx + UI_UNIT_X; - - if (num_elements > 1) { - outliner_draw_iconrow_number(fstyle, *offsx, ys, num_elements); - te->flag |= TE_ICONROW_MERGED; - } - else { - te->flag |= TE_ICONROW; - } - - (*offsx) += UI_UNIT_X; -} - -int tree_element_id_type_to_index(TreeElement *te) -{ - TreeStoreElem *tselem = TREESTORE(te); - - const int id_index = (tselem->type == TSE_SOME_ID) ? BKE_idtype_idcode_to_index(te->idcode) : - INDEX_ID_GR; - if (id_index < INDEX_ID_OB) { - return id_index; - } - if (id_index == INDEX_ID_OB) { - const Object *ob = (Object *)tselem->id; - return INDEX_ID_OB + ob->type; - } - return id_index + OB_TYPE_MAX; -} - -typedef struct MergedIconRow { - eOLDrawState active[INDEX_ID_MAX + OB_TYPE_MAX]; - int num_elements[INDEX_ID_MAX + OB_TYPE_MAX]; - TreeElement *tree_element[INDEX_ID_MAX + OB_TYPE_MAX]; -} MergedIconRow; - -static void outliner_draw_iconrow(bContext *C, - uiBlock *block, - const uiFontStyle *fstyle, - const TreeViewContext *tvc, - SpaceOutliner *space_outliner, - ListBase *lb, - int level, - int xmax, - int *offsx, - int ys, - float alpha_fac, - MergedIconRow *merged) -{ - eOLDrawState active = OL_DRAWSEL_NONE; - - LISTBASE_FOREACH (TreeElement *, te, lb) { - TreeStoreElem *tselem = TREESTORE(te); - te->flag &= ~(TE_ICONROW | TE_ICONROW_MERGED); - - /* object hierarchy always, further constrained on level */ - if ((level < 1) || ((tselem->type == TSE_SOME_ID) && (te->idcode == ID_OB)) || - ELEM(tselem->type, TSE_BONE, TSE_EBONE, TSE_POSE_CHANNEL)) { - /* active blocks get white circle */ - if (tselem->type == TSE_SOME_ID) { - if (te->idcode == ID_OB) { - active = (tvc->obact == (Object *)tselem->id) ? OL_DRAWSEL_NORMAL : OL_DRAWSEL_NONE; - } - else if (is_object_data_in_editmode(tselem->id, tvc->obact)) { - active = OL_DRAWSEL_ACTIVE; - } - else { - active = tree_element_active_state_get(tvc, te, tselem); - } - } - else { - active = tree_element_type_active_state_get(C, tvc, te, tselem); - } - - if (!ELEM(tselem->type, - TSE_ID_BASE, - TSE_SOME_ID, - TSE_LAYER_COLLECTION, - TSE_R_LAYER, - TSE_GP_LAYER, - TSE_LIBRARY_OVERRIDE_BASE, - TSE_LIBRARY_OVERRIDE, - TSE_BONE, - TSE_EBONE, - TSE_POSE_CHANNEL, - TSE_POSEGRP, - TSE_DEFGROUP)) { - outliner_draw_iconrow_doit(block, te, fstyle, xmax, offsx, ys, alpha_fac, active, 1); - } - else { - const int index = tree_element_id_type_to_index(te); - merged->num_elements[index]++; - if ((merged->tree_element[index] == NULL) || (active > merged->active[index])) { - merged->tree_element[index] = te; - } - merged->active[index] = MAX2(active, merged->active[index]); - } - } - - /* this tree element always has same amount of branches, so don't draw */ - if (tselem->type != TSE_R_LAYER) { - outliner_draw_iconrow(C, - block, - fstyle, - tvc, - space_outliner, - &te->subtree, - level + 1, - xmax, - offsx, - ys, - alpha_fac, - merged); - } - } - - if (level == 0) { - for (int i = 0; i < INDEX_ID_MAX; i++) { - const int num_subtypes = (i == INDEX_ID_OB) ? OB_TYPE_MAX : 1; - /* See tree_element_id_type_to_index for the index logic. */ - int index_base = i; - if (i > INDEX_ID_OB) { - index_base += OB_TYPE_MAX; - } - for (int j = 0; j < num_subtypes; j++) { - const int index = index_base + j; - if (merged->num_elements[index] != 0) { - outliner_draw_iconrow_doit(block, - merged->tree_element[index], - fstyle, - xmax, - offsx, - ys, - alpha_fac, - merged->active[index], - merged->num_elements[index]); - } - } - } - } -} - -/* closed tree element */ -static void outliner_set_coord_tree_element(TreeElement *te, int startx, int starty) -{ - /* closed items may be displayed in row of parent, don't change their coordinate! */ - if ((te->flag & TE_ICONROW) == 0 && (te->flag & TE_ICONROW_MERGED) == 0) { - te->xs = 0; - te->ys = 0; - te->xend = 0; - } - - LISTBASE_FOREACH (TreeElement *, ten, &te->subtree) { - outliner_set_coord_tree_element(ten, startx + UI_UNIT_X, starty); - } -} - -static bool element_should_draw_faded(const TreeViewContext *tvc, - const TreeElement *te, - const TreeStoreElem *tselem) -{ - if (tselem->type == TSE_SOME_ID) { - switch (te->idcode) { - case ID_OB: { - const Object *ob = (const Object *)tselem->id; - /* Lookup in view layer is logically const as it only checks a cache. */ - const Base *base = (te->directdata) ? (const Base *)te->directdata : - BKE_view_layer_base_find( - (ViewLayer *)tvc->view_layer, (Object *)ob); - const bool is_visible = (base != NULL) && (base->flag & BASE_VISIBLE_VIEWLAYER); - if (!is_visible) { - return true; - } - } - } - } - switch (tselem->type) { - case TSE_LAYER_COLLECTION: { - const LayerCollection *layer_collection = (const LayerCollection *)te->directdata; - const bool is_visible = layer_collection->runtime_flag & LAYER_COLLECTION_VISIBLE_VIEW_LAYER; - const bool is_excluded = layer_collection->flag & LAYER_COLLECTION_EXCLUDE; - return !is_visible || is_excluded; - } - } - - if (te->flag & TE_CHILD_NOT_IN_COLLECTION) { - return true; - } - - return false; -} - -static void outliner_draw_tree_element(bContext *C, - uiBlock *block, - const uiFontStyle *fstyle, - const TreeViewContext *tvc, - ARegion *region, - SpaceOutliner *space_outliner, - TreeElement *te, - bool draw_grayed_out, - int startx, - int *starty, - const float restrict_column_width, - TreeElement **te_edit) -{ - TreeStoreElem *tselem = TREESTORE(te); - float ufac = UI_UNIT_X / 20.0f; - int offsx = 0; - eOLDrawState active = OL_DRAWSEL_NONE; - uchar text_color[4]; - UI_GetThemeColor4ubv(TH_TEXT, text_color); - float icon_bgcolor[4], icon_border[4]; - outliner_icon_background_colors(icon_bgcolor, icon_border); - - if (*starty + 2 * UI_UNIT_Y >= region->v2d.cur.ymin && *starty <= region->v2d.cur.ymax) { - const float alpha_fac = element_should_draw_faded(tvc, te, tselem) ? 0.5f : 1.0f; - int xmax = region->v2d.cur.xmax; - - if ((tselem->flag & TSE_TEXTBUT) && (*te_edit == NULL)) { - *te_edit = te; - } - - /* Icons can be UI buts, we don't want it to overlap with restrict. */ - if (restrict_column_width > 0) { - xmax -= restrict_column_width + UI_UNIT_X; - } - - GPU_blend(GPU_BLEND_ALPHA); - - /* Colors for active/selected data. */ - if (tselem->type == TSE_SOME_ID) { - if (te->idcode == ID_OB) { - Object *ob = (Object *)tselem->id; - Base *base = (te->directdata) ? (Base *)te->directdata : - BKE_view_layer_base_find(tvc->view_layer, ob); - const bool is_selected = (base != NULL) && ((base->flag & BASE_SELECTED) != 0); - - if (ob == tvc->obact) { - active = OL_DRAWSEL_ACTIVE; - } - - if (is_selected) { - if (ob == tvc->obact) { - /* Active selected object. */ - UI_GetThemeColor3ubv(TH_ACTIVE_OBJECT, text_color); - text_color[3] = 255; - } - else { - /* Other selected objects. */ - UI_GetThemeColor3ubv(TH_SELECTED_OBJECT, text_color); - text_color[3] = 255; - } - } - } - else if (is_object_data_in_editmode(tselem->id, tvc->obact)) { - /* Objects being edited. */ - UI_GetThemeColor4fv(TH_EDITED_OBJECT, icon_bgcolor); - icon_border[3] = 0.3f; - active = OL_DRAWSEL_ACTIVE; - } - else { - if (tree_element_active_state_get(tvc, te, tselem)) { - /* Active items like camera or material. */ - icon_bgcolor[3] = 0.2f; - active = OL_DRAWSEL_ACTIVE; - } - } - } - else { - active = tree_element_type_active_state_get(C, tvc, te, tselem); - } - - /* Active circle. */ - if (active != OL_DRAWSEL_NONE) { - outliner_draw_active_indicator((float)startx + offsx + UI_UNIT_X, - (float)*starty, - (float)startx + offsx + 2.0f * UI_UNIT_X, - (float)*starty + UI_UNIT_Y, - icon_bgcolor, - icon_border); - - te->flag |= TE_ACTIVE; /* For lookup in display hierarchies. */ - } - - if (tselem->type == TSE_VIEW_COLLECTION_BASE) { - /* Scene collection in view layer can't expand/collapse. */ - } - else if (te->subtree.first || ((tselem->type == TSE_SOME_ID) && (te->idcode == ID_SCE)) || - (te->flag & TE_LAZY_CLOSED)) { - /* Open/close icon, only when sub-levels, except for scene. */ - int icon_x = startx; - - /* Icons a bit higher. */ - if (TSELEM_OPEN(tselem, space_outliner)) { - UI_icon_draw_alpha((float)icon_x + 2 * ufac, - (float)*starty + 1 * ufac, - ICON_DISCLOSURE_TRI_DOWN, - alpha_fac); - } - else { - UI_icon_draw_alpha((float)icon_x + 2 * ufac, - (float)*starty + 1 * ufac, - ICON_DISCLOSURE_TRI_RIGHT, - alpha_fac); - } - } - offsx += UI_UNIT_X; - - /* Data-type icon. */ - if (!(ELEM(tselem->type, TSE_RNA_PROPERTY, TSE_RNA_ARRAY_ELEM, TSE_ID_BASE))) { - tselem_draw_icon(block, - xmax, - (float)startx + offsx, - (float)*starty, - tselem, - te, - (tselem->flag & TSE_HIGHLIGHTED_ICON) ? alpha_fac + 0.5f : alpha_fac, - true); - offsx += UI_UNIT_X + 4 * ufac; - } - else { - offsx += 2 * ufac; - } - - if (ELEM(tselem->type, TSE_SOME_ID, TSE_LAYER_COLLECTION) || - ((tselem->type == TSE_RNA_STRUCT) && RNA_struct_is_ID(te->rnaptr.type))) { - const BIFIconID lib_icon = UI_icon_from_library(tselem->id); - if (lib_icon != ICON_NONE) { - UI_icon_draw_alpha( - (float)startx + offsx + 2 * ufac, (float)*starty + 2 * ufac, lib_icon, alpha_fac); - offsx += UI_UNIT_X + 4 * ufac; - } - } - GPU_blend(GPU_BLEND_NONE); - - /* Name. */ - if ((tselem->flag & TSE_TEXTBUT) == 0) { - if (ELEM(tselem->type, TSE_RNA_PROPERTY, TSE_RNA_ARRAY_ELEM)) { - UI_GetThemeColorBlend3ubv(TH_BACK, TH_TEXT, 0.75f, text_color); - text_color[3] = 255; - } - text_color[3] *= alpha_fac; - UI_fontstyle_draw_simple(fstyle, startx + offsx, *starty + 5 * ufac, te->name, text_color); - } - - offsx += (int)(UI_UNIT_X + UI_fontstyle_string_width(fstyle, te->name)); - - /* Closed item, we draw the icons, not when it's a scene, or master-server list though. */ - if (!TSELEM_OPEN(tselem, space_outliner)) { - if (te->subtree.first) { - if ((tselem->type == TSE_SOME_ID) && (te->idcode == ID_SCE)) { - /* Pass. */ - } - /* this tree element always has same amount of branches, so don't draw */ - else if (tselem->type != TSE_R_LAYER) { - int tempx = startx + offsx; - - GPU_blend(GPU_BLEND_ALPHA); - - MergedIconRow merged = {{0}}; - outliner_draw_iconrow(C, - block, - fstyle, - tvc, - space_outliner, - &te->subtree, - 0, - xmax, - &tempx, - *starty, - alpha_fac, - &merged); - - GPU_blend(GPU_BLEND_NONE); - } - } - } - } - /* Store coord and continue, we need coordinates for elements outside view too. */ - te->xs = startx; - te->ys = *starty; - te->xend = startx + offsx; - - if (TSELEM_OPEN(tselem, space_outliner)) { - *starty -= UI_UNIT_Y; - - LISTBASE_FOREACH (TreeElement *, ten, &te->subtree) { - /* Check if element needs to be drawn grayed out, but also gray out - * children of a grayed out parent (pass on draw_grayed_out to children). */ - bool draw_children_grayed_out = draw_grayed_out || (ten->flag & TE_DRAGGING); - outliner_draw_tree_element(C, - block, - fstyle, - tvc, - region, - space_outliner, - ten, - draw_children_grayed_out, - startx + UI_UNIT_X, - starty, - restrict_column_width, - te_edit); - } - } - else { - LISTBASE_FOREACH (TreeElement *, ten, &te->subtree) { - outliner_set_coord_tree_element(ten, startx, *starty); - } - - *starty -= UI_UNIT_Y; - } -} - -static bool subtree_contains_object(ListBase *lb) -{ - LISTBASE_FOREACH (TreeElement *, te, lb) { - TreeStoreElem *tselem = TREESTORE(te); - if ((tselem->type == TSE_SOME_ID) && (te->idcode == ID_OB)) { - return true; - } - } - return false; -} - -static void outliner_draw_hierarchy_line( - const uint pos, const int x, const int y1, const int y2, const bool draw_dashed) -{ - /* Small vertical padding. */ - const short line_padding = UI_UNIT_Y / 4.0f; - - /* >= is 1.0 for undashed lines. */ - immUniform1f("dash_factor", draw_dashed ? 0.5f : 1.0f); - - immBegin(GPU_PRIM_LINES, 2); - /* Intentionally draw from top to bottom, so collapsing a child item doesn't make the dashes - * appear to move. */ - immVertex2f(pos, x, y2 + line_padding); - immVertex2f(pos, x, y1 - line_padding); - immEnd(); -} - -static void outliner_draw_hierarchy_lines_recursive(uint pos, - SpaceOutliner *space_outliner, - ListBase *lb, - int startx, - const uchar col[4], - bool draw_grayed_out, - int *starty) -{ - bTheme *btheme = UI_GetTheme(); - int y = *starty; - - /* Draw vertical lines between collections */ - bool draw_hierarchy_line; - bool is_object_line; - LISTBASE_FOREACH (TreeElement *, te, lb) { - TreeStoreElem *tselem = TREESTORE(te); - draw_hierarchy_line = false; - is_object_line = false; - *starty -= UI_UNIT_Y; - short color_tag = COLLECTION_COLOR_NONE; - - /* Only draw hierarchy lines for expanded collections and objects with children. */ - if (TSELEM_OPEN(tselem, space_outliner) && !BLI_listbase_is_empty(&te->subtree)) { - if (tselem->type == TSE_LAYER_COLLECTION) { - draw_hierarchy_line = true; - - Collection *collection = outliner_collection_from_tree_element(te); - color_tag = collection->color_tag; - - y = *starty; - } - else if ((tselem->type == TSE_SOME_ID) && (te->idcode == ID_OB)) { - if (subtree_contains_object(&te->subtree)) { - draw_hierarchy_line = true; - is_object_line = true; - y = *starty; - } - } - - outliner_draw_hierarchy_lines_recursive( - pos, space_outliner, &te->subtree, startx + UI_UNIT_X, col, draw_grayed_out, starty); - } - - if (draw_hierarchy_line) { - if (color_tag != COLLECTION_COLOR_NONE) { - immUniformColor4ubv(btheme->collection_color[color_tag].color); - } - else { - immUniformColor4ubv(col); - } - - outliner_draw_hierarchy_line(pos, startx, y, *starty, is_object_line); - } - } -} - -static void outliner_draw_hierarchy_lines(SpaceOutliner *space_outliner, - ListBase *lb, - int startx, - int *starty) -{ - GPUVertFormat *format = immVertexFormat(); - uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT); - uchar col[4]; - - immBindBuiltinProgram(GPU_SHADER_2D_LINE_DASHED_UNIFORM_COLOR); - - float viewport_size[4]; - GPU_viewport_size_get_f(viewport_size); - immUniform2f("viewport_size", viewport_size[2] / UI_DPI_FAC, viewport_size[3] / UI_DPI_FAC); - immUniform1i("colors_len", 0); /* "simple" mode */ - immUniform1f("dash_width", 8.0f); - UI_GetThemeColorBlend3ubv(TH_BACK, TH_TEXT, 0.4f, col); - col[3] = 255; - - GPU_line_width(1.0f); - GPU_blend(GPU_BLEND_ALPHA); - outliner_draw_hierarchy_lines_recursive(pos, space_outliner, lb, startx, col, false, starty); - GPU_blend(GPU_BLEND_NONE); - - immUnbindProgram(); -} - -static void outliner_draw_struct_marks(ARegion *region, - SpaceOutliner *space_outliner, - ListBase *lb, - int *starty) -{ - LISTBASE_FOREACH (TreeElement *, te, lb) { - TreeStoreElem *tselem = TREESTORE(te); - - /* Selection status. */ - if (TSELEM_OPEN(tselem, space_outliner)) { - if (tselem->type == TSE_RNA_STRUCT) { - GPUVertFormat *format = immVertexFormat(); - uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_I32, 2, GPU_FETCH_INT_TO_FLOAT); - immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR); - immThemeColorShadeAlpha(TH_BACK, -15, -200); - immRecti(pos, 0, *starty + 1, (int)region->v2d.cur.xmax, *starty + UI_UNIT_Y - 1); - immUnbindProgram(); - } - } - - *starty -= UI_UNIT_Y; - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_draw_struct_marks(region, space_outliner, &te->subtree, starty); - if (tselem->type == TSE_RNA_STRUCT) { - GPUVertFormat *format = immVertexFormat(); - uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT); - immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR); - immThemeColorShadeAlpha(TH_BACK, -15, -200); - - immBegin(GPU_PRIM_LINES, 2); - immVertex2f(pos, 0, (float)*starty + UI_UNIT_Y); - immVertex2f(pos, region->v2d.cur.xmax, (float)*starty + UI_UNIT_Y); - immEnd(); - - immUnbindProgram(); - } - } - } -} - -static void outliner_draw_highlights_recursive(uint pos, - const ARegion *region, - const SpaceOutliner *space_outliner, - const ListBase *lb, - const float col_selection[4], - const float col_active[4], - const float col_highlight[4], - const float col_searchmatch[4], - int start_x, - int *io_start_y) -{ - const bool is_searching = (SEARCHING_OUTLINER(space_outliner) || - (space_outliner->outlinevis == SO_DATA_API && - space_outliner->search_string[0] != 0)); - - LISTBASE_FOREACH (TreeElement *, te, lb) { - const TreeStoreElem *tselem = TREESTORE(te); - const int start_y = *io_start_y; - - /* Selection status. */ - if ((tselem->flag & TSE_ACTIVE) && (tselem->flag & TSE_SELECTED)) { - immUniformColor4fv(col_active); - immRecti(pos, 0, start_y, (int)region->v2d.cur.xmax, start_y + UI_UNIT_Y); - } - else if (tselem->flag & TSE_SELECTED) { - immUniformColor4fv(col_selection); - immRecti(pos, 0, start_y, (int)region->v2d.cur.xmax, start_y + UI_UNIT_Y); - } - - /* Highlights. */ - if (tselem->flag & (TSE_DRAG_ANY | TSE_HIGHLIGHTED | TSE_SEARCHMATCH)) { - const int end_x = (int)region->v2d.cur.xmax; - - if (tselem->flag & TSE_DRAG_ANY) { - /* Drag and drop highlight. */ - float col[4]; - UI_GetThemeColorShade4fv(TH_BACK, -40, col); - - if (tselem->flag & TSE_DRAG_BEFORE) { - immUniformColor4fv(col); - immRecti(pos, - start_x, - start_y + UI_UNIT_Y - U.pixelsize, - end_x, - start_y + UI_UNIT_Y + U.pixelsize); - } - else if (tselem->flag & TSE_DRAG_AFTER) { - immUniformColor4fv(col); - immRecti(pos, start_x, start_y - U.pixelsize, end_x, start_y + U.pixelsize); - } - else { - immUniformColor3fvAlpha(col, col[3] * 0.5f); - immRecti(pos, start_x, start_y, end_x, start_y + UI_UNIT_Y); - } - } - else { - if (is_searching && (tselem->flag & TSE_SEARCHMATCH)) { - /* Search match highlights. We don't expand items when searching in the data-blocks, - * but we still want to highlight any filter matches. */ - immUniformColor4fv(col_searchmatch); - immRecti(pos, start_x, start_y, end_x, start_y + UI_UNIT_Y); - } - else if (tselem->flag & TSE_HIGHLIGHTED) { - /* Mouse hover highlight. */ - immUniformColor4fv(col_highlight); - immRecti(pos, 0, start_y, end_x, start_y + UI_UNIT_Y); - } - } - } - - *io_start_y -= UI_UNIT_Y; - if (TSELEM_OPEN(tselem, space_outliner)) { - outliner_draw_highlights_recursive(pos, - region, - space_outliner, - &te->subtree, - col_selection, - col_active, - col_highlight, - col_searchmatch, - start_x + UI_UNIT_X, - io_start_y); - } - } -} - -static void outliner_draw_highlights(ARegion *region, - SpaceOutliner *space_outliner, - int startx, - int *starty) -{ - const float col_highlight[4] = {1.0f, 1.0f, 1.0f, 0.13f}; - float col_selection[4], col_active[4], col_searchmatch[4]; - - UI_GetThemeColor3fv(TH_SELECT_HIGHLIGHT, col_selection); - col_selection[3] = 1.0f; /* No alpha. */ - UI_GetThemeColor3fv(TH_SELECT_ACTIVE, col_active); - col_active[3] = 1.0f; /* No alpha. */ - UI_GetThemeColor4fv(TH_MATCH, col_searchmatch); - col_searchmatch[3] = 0.5f; - - GPU_blend(GPU_BLEND_ALPHA); - GPUVertFormat *format = immVertexFormat(); - uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_I32, 2, GPU_FETCH_INT_TO_FLOAT); - immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR); - outliner_draw_highlights_recursive(pos, - region, - space_outliner, - &space_outliner->tree, - col_selection, - col_active, - col_highlight, - col_searchmatch, - startx, - starty); - immUnbindProgram(); - GPU_blend(GPU_BLEND_NONE); -} - -static void outliner_draw_tree(bContext *C, - uiBlock *block, - const TreeViewContext *tvc, - ARegion *region, - SpaceOutliner *space_outliner, - const float restrict_column_width, - const bool use_mode_column, - const bool use_warning_column, - TreeElement **te_edit) -{ - const uiFontStyle *fstyle = UI_FSTYLE_WIDGET; - int starty, startx; - - /* Move the tree a unit left in view layer mode */ - short columns_offset = (use_mode_column && (space_outliner->outlinevis == SO_SCENES)) ? - UI_UNIT_X : - 0; - if (!use_mode_column && (space_outliner->outlinevis == SO_VIEW_LAYER)) { - columns_offset -= UI_UNIT_X; - } - - if (use_warning_column) { - columns_offset += UI_UNIT_X; - } - - GPU_blend(GPU_BLEND_ALPHA); /* Only once. */ - - if (space_outliner->outlinevis == SO_DATA_API) { - /* struct marks */ - starty = (int)region->v2d.tot.ymax - UI_UNIT_Y - OL_Y_OFFSET; - outliner_draw_struct_marks(region, space_outliner, &space_outliner->tree, &starty); - } - - /* Draw highlights before hierarchy. */ - starty = (int)region->v2d.tot.ymax - UI_UNIT_Y - OL_Y_OFFSET; - startx = 0; - outliner_draw_highlights(region, space_outliner, startx, &starty); - - /* Set scissor so tree elements or lines can't overlap restriction icons. */ - int scissor[4] = {0}; - if (restrict_column_width > 0.0f) { - int mask_x = BLI_rcti_size_x(®ion->v2d.mask) - (int)restrict_column_width + 1; - CLAMP_MIN(mask_x, 0); - - GPU_scissor_get(scissor); - GPU_scissor(0, 0, mask_x, region->winy); - } - - /* Draw hierarchy lines for collections and object children. */ - starty = (int)region->v2d.tot.ymax - OL_Y_OFFSET; - startx = columns_offset + UI_UNIT_X / 2 - (U.pixelsize + 1) / 2; - outliner_draw_hierarchy_lines(space_outliner, &space_outliner->tree, startx, &starty); - - /* Items themselves. */ - starty = (int)region->v2d.tot.ymax - UI_UNIT_Y - OL_Y_OFFSET; - startx = columns_offset; - LISTBASE_FOREACH (TreeElement *, te, &space_outliner->tree) { - outliner_draw_tree_element(C, - block, - fstyle, - tvc, - region, - space_outliner, - te, - (te->flag & TE_DRAGGING) != 0, - startx, - &starty, - restrict_column_width, - te_edit); - } - - if (restrict_column_width > 0.0f) { - /* Reset scissor. */ - GPU_scissor(UNPACK4(scissor)); - } -} - -static void outliner_back(ARegion *region) -{ - int ystart; - - ystart = (int)region->v2d.tot.ymax; - ystart = UI_UNIT_Y * (ystart / (UI_UNIT_Y)) - OL_Y_OFFSET; - - GPUVertFormat *format = immVertexFormat(); - uint pos = GPU_vertformat_attr_add(format, "pos", GPU_COMP_F32, 2, GPU_FETCH_FLOAT); - - immBindBuiltinProgram(GPU_SHADER_2D_UNIFORM_COLOR); - - float col_alternating[4]; - UI_GetThemeColor4fv(TH_ROW_ALTERNATE, col_alternating); - immUniformThemeColorBlend(TH_BACK, TH_ROW_ALTERNATE, col_alternating[3]); - - const float x1 = 0.0f, x2 = region->v2d.cur.xmax; - float y1 = ystart, y2; - int tot = (int)floor(ystart - region->v2d.cur.ymin + 2 * UI_UNIT_Y) / (2 * UI_UNIT_Y); - - if (tot > 0) { - immBegin(GPU_PRIM_TRIS, 6 * tot); - while (tot--) { - y1 -= 2 * UI_UNIT_Y; - y2 = y1 + UI_UNIT_Y; - immVertex2f(pos, x1, y1); - immVertex2f(pos, x2, y1); - immVertex2f(pos, x2, y2); - - immVertex2f(pos, x1, y1); - immVertex2f(pos, x2, y2); - immVertex2f(pos, x1, y2); - } - immEnd(); - } - immUnbindProgram(); -} - -static int outliner_data_api_buttons_start_x(int max_tree_width) -{ - return max_ii(OL_RNA_COLX, max_tree_width + OL_RNA_COL_SPACEX); -} - -static int outliner_width(SpaceOutliner *space_outliner, - int max_tree_width, - float restrict_column_width) -{ - if (space_outliner->outlinevis == SO_DATA_API) { - return outliner_data_api_buttons_start_x(max_tree_width) + OL_RNA_COL_SIZEX + 10 * UI_DPI_FAC; - } - return max_tree_width + restrict_column_width; -} - -static void outliner_update_viewable_area(ARegion *region, - SpaceOutliner *space_outliner, - int tree_width, - int tree_height, - float restrict_column_width) -{ - int sizex = outliner_width(space_outliner, tree_width, restrict_column_width); - int sizey = tree_height; - - /* Extend size to allow for horizontal scrollbar and extra offset. */ - sizey += V2D_SCROLL_HEIGHT + OL_Y_OFFSET; - - UI_view2d_totRect_set(®ion->v2d, sizex, sizey); -} - -/* ****************************************************** */ -/* Main Entry-point - Draw contents of Outliner editor */ - -void draw_outliner(const bContext *C) -{ - Main *mainvar = CTX_data_main(C); - ARegion *region = CTX_wm_region(C); - View2D *v2d = ®ion->v2d; - SpaceOutliner *space_outliner = CTX_wm_space_outliner(C); - uiBlock *block; - TreeElement *te_edit = NULL; - - TreeViewContext tvc; - outliner_viewcontext_init(C, &tvc); - - outliner_build_tree(mainvar, tvc.scene, tvc.view_layer, space_outliner, region); /* Always. */ - - /* If global sync select is dirty, flag other outliners. */ - if (ED_outliner_select_sync_is_dirty(C)) { - ED_outliner_select_sync_flag_outliners(C); - } - - /* Sync selection state from view layer. */ - if (!ELEM(space_outliner->outlinevis, - SO_LIBRARIES, - SO_OVERRIDES_LIBRARY, - SO_DATA_API, - SO_ID_ORPHANS) && - space_outliner->flag & SO_SYNC_SELECT) { - outliner_sync_selection(C, space_outliner); - } - - /* Force display to pixel coords. */ - v2d->flag |= (V2D_PIXELOFS_X | V2D_PIXELOFS_Y); - /* Set matrix for 2D-view controls. */ - UI_view2d_view_ortho(v2d); - - /* Only show mode column in View Layers and Scenes view. */ - const bool use_mode_column = (space_outliner->flag & SO_MODE_COLUMN) && - (ELEM(space_outliner->outlinevis, SO_VIEW_LAYER, SO_SCENES)); - - const bool use_warning_column = - ELEM(space_outliner->outlinevis, SO_LIBRARIES, SO_OVERRIDES_LIBRARY) && - outliner_tree_display_warnings_poll(space_outliner->runtime->tree_display); - - /* Draw outliner stuff (background, hierarchy lines and names). */ - const float restrict_column_width = outliner_restrict_columns_width(space_outliner); - outliner_back(region); - block = UI_block_begin(C, region, __func__, UI_EMBOSS); - outliner_draw_tree((bContext *)C, - block, - &tvc, - region, - space_outliner, - restrict_column_width, - use_mode_column, - use_warning_column, - &te_edit); - - /* Compute outliner dimensions after it has been drawn. */ - int tree_width, tree_height; - outliner_tree_dimensions(space_outliner, &tree_width, &tree_height); - - /* Default to no emboss for outliner UI. */ - UI_block_emboss_set(block, UI_EMBOSS_NONE_OR_STATUS); - - if (space_outliner->outlinevis == SO_DATA_API) { - int buttons_start_x = outliner_data_api_buttons_start_x(tree_width); - /* draw rna buttons */ - outliner_draw_rnacols(region, buttons_start_x); - - UI_block_emboss_set(block, UI_EMBOSS); - outliner_draw_rnabuts(block, region, space_outliner, buttons_start_x, &space_outliner->tree); - UI_block_emboss_set(block, UI_EMBOSS_NONE_OR_STATUS); - } - else if (space_outliner->outlinevis == SO_ID_ORPHANS) { - /* draw user toggle columns */ - outliner_draw_userbuts(block, region, space_outliner, &space_outliner->tree); - } - else if (space_outliner->outlinevis == SO_OVERRIDES_LIBRARY) { - /* Draw overrides status columns. */ - outliner_draw_overrides_buts(block, region, space_outliner, &space_outliner->tree, true); - } - else if (restrict_column_width > 0.0f) { - /* draw restriction columns */ - RestrictPropertiesActive props_active; - memset(&props_active, 1, sizeof(RestrictPropertiesActive)); - outliner_draw_restrictbuts(block, - tvc.scene, - tvc.view_layer, - region, - space_outliner, - &space_outliner->tree, - props_active); - } - - /* Draw mode icons */ - if (use_mode_column) { - outliner_draw_mode_column(C, block, &tvc, space_outliner, &space_outliner->tree); - } - - /* Draw warning icons */ - if (use_warning_column) { - outliner_draw_warning_column(C, block, space_outliner, use_mode_column, &space_outliner->tree); - } - - UI_block_emboss_set(block, UI_EMBOSS); - - /* Draw edit buttons if necessary. */ - if (te_edit) { - outliner_buttons(C, block, region, restrict_column_width, te_edit); - } - - UI_block_end(C, block); - UI_block_draw(C, block); - - /* Update total viewable region. */ - outliner_update_viewable_area( - region, space_outliner, tree_width, tree_height, restrict_column_width); -} |