325 lines
10 KiB
C++
325 lines
10 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup bke
|
|
*/
|
|
|
|
#include <cstring>
|
|
|
|
#include "BLI_array.h"
|
|
|
|
#include "BKE_collection.h"
|
|
#include "BKE_customdata.hh"
|
|
#include "BKE_editmesh.hh"
|
|
#include "BKE_layer.h"
|
|
|
|
#include "DNA_ID.h"
|
|
#include "DNA_layer_types.h"
|
|
#include "DNA_mesh_types.h"
|
|
#include "DNA_object_types.h"
|
|
#include "DNA_scene_types.h"
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Selected Object Array
|
|
* \{ */
|
|
|
|
Object **BKE_view_layer_array_selected_objects_params(ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len,
|
|
const ObjectsInViewLayerParams *params)
|
|
{
|
|
if (params->no_dup_data) {
|
|
FOREACH_SELECTED_OBJECT_BEGIN (view_layer, v3d, ob_iter) {
|
|
ID *id = static_cast<ID *>(ob_iter->data);
|
|
if (id) {
|
|
id->tag |= LIB_TAG_DOIT;
|
|
}
|
|
}
|
|
FOREACH_SELECTED_OBJECT_END;
|
|
}
|
|
|
|
Object **object_array = nullptr;
|
|
BLI_array_declare(object_array);
|
|
|
|
FOREACH_SELECTED_OBJECT_BEGIN (view_layer, v3d, ob_iter) {
|
|
if (params->filter_fn) {
|
|
if (!params->filter_fn(ob_iter, params->filter_userdata)) {
|
|
continue;
|
|
}
|
|
}
|
|
|
|
if (params->no_dup_data) {
|
|
ID *id = static_cast<ID *>(ob_iter->data);
|
|
if (id) {
|
|
if (id->tag & LIB_TAG_DOIT) {
|
|
id->tag &= ~LIB_TAG_DOIT;
|
|
}
|
|
else {
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
|
|
BLI_array_append(object_array, ob_iter);
|
|
}
|
|
FOREACH_SELECTED_OBJECT_END;
|
|
|
|
if (object_array != nullptr) {
|
|
BLI_array_trim(object_array);
|
|
}
|
|
else {
|
|
/* We always need a valid allocation (prevent crash on free). */
|
|
object_array = static_cast<Object **>(MEM_mallocN(0, __func__));
|
|
}
|
|
*r_len = BLI_array_len(object_array);
|
|
return object_array;
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Objects in Mode Array
|
|
* \{ */
|
|
|
|
Base **BKE_view_layer_array_from_bases_in_mode_params(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len,
|
|
const ObjectsInModeParams *params)
|
|
{
|
|
if (params->no_dup_data) {
|
|
FOREACH_BASE_IN_MODE_BEGIN (scene, view_layer, v3d, -1, params->object_mode, base_iter) {
|
|
ID *id = static_cast<ID *>(base_iter->object->data);
|
|
if (id) {
|
|
id->tag |= LIB_TAG_DOIT;
|
|
}
|
|
}
|
|
FOREACH_BASE_IN_MODE_END;
|
|
}
|
|
|
|
Base **base_array = nullptr;
|
|
BLI_array_declare(base_array);
|
|
|
|
FOREACH_BASE_IN_MODE_BEGIN (scene, view_layer, v3d, -1, params->object_mode, base_iter) {
|
|
if (params->filter_fn) {
|
|
if (!params->filter_fn(base_iter->object, params->filter_userdata)) {
|
|
continue;
|
|
}
|
|
}
|
|
if (params->no_dup_data) {
|
|
ID *id = static_cast<ID *>(base_iter->object->data);
|
|
if (id) {
|
|
if (id->tag & LIB_TAG_DOIT) {
|
|
id->tag &= ~LIB_TAG_DOIT;
|
|
}
|
|
else {
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
BLI_array_append(base_array, base_iter);
|
|
}
|
|
FOREACH_BASE_IN_MODE_END;
|
|
|
|
/* We always need a valid allocation (prevent crash on free). */
|
|
if (base_array != nullptr) {
|
|
BLI_array_trim(base_array);
|
|
}
|
|
else {
|
|
base_array = static_cast<Base **>(MEM_mallocN(0, __func__));
|
|
}
|
|
*r_len = BLI_array_len(base_array);
|
|
return base_array;
|
|
}
|
|
|
|
Object **BKE_view_layer_array_from_objects_in_mode_params(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len,
|
|
const ObjectsInModeParams *params)
|
|
{
|
|
Base **base_array = BKE_view_layer_array_from_bases_in_mode_params(
|
|
scene, view_layer, v3d, r_len, params);
|
|
if (base_array != nullptr) {
|
|
for (uint i = 0; i < *r_len; i++) {
|
|
((Object **)base_array)[i] = base_array[i]->object;
|
|
}
|
|
}
|
|
return (Object **)base_array;
|
|
}
|
|
|
|
Object **BKE_view_layer_array_from_objects_in_edit_mode(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len)
|
|
{
|
|
ObjectsInModeParams params = {0};
|
|
params.object_mode = OB_MODE_EDIT;
|
|
return BKE_view_layer_array_from_objects_in_mode_params(scene, view_layer, v3d, r_len, ¶ms);
|
|
}
|
|
|
|
Base **BKE_view_layer_array_from_bases_in_edit_mode(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len)
|
|
{
|
|
ObjectsInModeParams params = {0};
|
|
params.object_mode = OB_MODE_EDIT;
|
|
return BKE_view_layer_array_from_bases_in_mode_params(scene, view_layer, v3d, r_len, ¶ms);
|
|
}
|
|
|
|
Object **BKE_view_layer_array_from_objects_in_edit_mode_unique_data(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len)
|
|
{
|
|
ObjectsInModeParams params = {0};
|
|
params.object_mode = OB_MODE_EDIT;
|
|
params.no_dup_data = true;
|
|
return BKE_view_layer_array_from_objects_in_mode_params(scene, view_layer, v3d, r_len, ¶ms);
|
|
}
|
|
|
|
Base **BKE_view_layer_array_from_bases_in_edit_mode_unique_data(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len)
|
|
{
|
|
ObjectsInModeParams params = {0};
|
|
params.object_mode = OB_MODE_EDIT;
|
|
params.no_dup_data = true;
|
|
return BKE_view_layer_array_from_bases_in_mode_params(scene, view_layer, v3d, r_len, ¶ms);
|
|
}
|
|
|
|
Object **BKE_view_layer_array_from_objects_in_edit_mode_unique_data_with_uvs(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len)
|
|
{
|
|
ObjectsInModeParams params = {0};
|
|
params.object_mode = OB_MODE_EDIT;
|
|
params.no_dup_data = true;
|
|
params.filter_fn = BKE_view_layer_filter_edit_mesh_has_uvs;
|
|
return BKE_view_layer_array_from_objects_in_mode_params(scene, view_layer, v3d, r_len, ¶ms);
|
|
}
|
|
|
|
Object **BKE_view_layer_array_from_objects_in_mode_unique_data(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d,
|
|
uint *r_len,
|
|
const eObjectMode mode)
|
|
{
|
|
ObjectsInModeParams params = {0};
|
|
params.object_mode = mode;
|
|
params.no_dup_data = true;
|
|
return BKE_view_layer_array_from_objects_in_mode_params(scene, view_layer, v3d, r_len, ¶ms);
|
|
}
|
|
|
|
ListBase *BKE_view_layer_object_bases_get(ViewLayer *view_layer)
|
|
{
|
|
BLI_assert_msg((view_layer->flag & VIEW_LAYER_OUT_OF_SYNC) == 0,
|
|
"Object Bases out of sync, invoke BKE_view_layer_synced_ensure.");
|
|
return &view_layer->object_bases;
|
|
}
|
|
|
|
Base *BKE_view_layer_active_base_get(ViewLayer *view_layer)
|
|
{
|
|
BLI_assert_msg((view_layer->flag & VIEW_LAYER_OUT_OF_SYNC) == 0,
|
|
"Active Base out of sync, invoke BKE_view_layer_synced_ensure.");
|
|
return view_layer->basact;
|
|
}
|
|
|
|
LayerCollection *BKE_view_layer_active_collection_get(ViewLayer *view_layer)
|
|
{
|
|
BLI_assert_msg((view_layer->flag & VIEW_LAYER_OUT_OF_SYNC) == 0,
|
|
"Active Collection out of sync, invoke BKE_view_layer_synced_ensure.");
|
|
return view_layer->active_collection;
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Filter Functions
|
|
* \{ */
|
|
|
|
bool BKE_view_layer_filter_edit_mesh_has_uvs(const Object *ob, void * /*user_data*/)
|
|
{
|
|
if (ob->type == OB_MESH) {
|
|
const Mesh *me = static_cast<const Mesh *>(ob->data);
|
|
const BMEditMesh *em = me->edit_mesh;
|
|
if (em != nullptr) {
|
|
if (CustomData_has_layer(&em->bm->ldata, CD_PROP_FLOAT2)) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool BKE_view_layer_filter_edit_mesh_has_edges(const Object *ob, void * /*user_data*/)
|
|
{
|
|
if (ob->type == OB_MESH) {
|
|
const Mesh *me = static_cast<const Mesh *>(ob->data);
|
|
const BMEditMesh *em = me->edit_mesh;
|
|
if (em != nullptr) {
|
|
if (em->bm->totedge != 0) {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Object *BKE_view_layer_non_active_selected_object(const Scene *scene,
|
|
ViewLayer *view_layer,
|
|
const View3D *v3d)
|
|
{
|
|
BKE_view_layer_synced_ensure(scene, view_layer);
|
|
Object *ob_active = BKE_view_layer_active_object_get(view_layer);
|
|
Object *ob_result = nullptr;
|
|
FOREACH_SELECTED_OBJECT_BEGIN (view_layer, v3d, ob_iter) {
|
|
if (ob_iter == ob_active) {
|
|
continue;
|
|
}
|
|
|
|
if (ob_result == nullptr) {
|
|
ob_result = ob_iter;
|
|
}
|
|
else {
|
|
ob_result = nullptr;
|
|
break;
|
|
}
|
|
}
|
|
FOREACH_SELECTED_OBJECT_END;
|
|
return ob_result;
|
|
}
|
|
|
|
/** \} */
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
/** \name Active object accessors.
|
|
* \{ */
|
|
|
|
Object *BKE_view_layer_active_object_get(const ViewLayer *view_layer)
|
|
{
|
|
Base *base = BKE_view_layer_active_base_get((ViewLayer *)view_layer);
|
|
return base ? base->object : nullptr;
|
|
}
|
|
|
|
Object *BKE_view_layer_edit_object_get(const ViewLayer *view_layer)
|
|
{
|
|
Object *ob = BKE_view_layer_active_object_get(view_layer);
|
|
if (ob == nullptr) {
|
|
return nullptr;
|
|
}
|
|
if (!(ob->mode & OB_MODE_EDIT)) {
|
|
return nullptr;
|
|
}
|
|
return ob;
|
|
}
|
|
|
|
/** \} */
|