c7a33a62a2
Instead of relying on them being included indirectly. Pull Request: https://projects.blender.org/blender/blender/pulls/134406
104 lines
3.1 KiB
C++
104 lines
3.1 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup animrig
|
|
*
|
|
* \brief Internal C++ functions to deal with Actions, Slots, and their runtime data.
|
|
*/
|
|
|
|
#include "BKE_anim_data.hh"
|
|
#include "BKE_global.hh"
|
|
#include "BKE_lib_query.hh"
|
|
#include "BKE_main.hh"
|
|
#include "BKE_nla.hh"
|
|
#include "BKE_node.hh"
|
|
|
|
#include "BLI_listbase.h"
|
|
#include "BLI_set.hh"
|
|
|
|
#include "ANIM_action.hh"
|
|
#include "ANIM_action_iterators.hh"
|
|
|
|
#include "action_runtime.hh"
|
|
|
|
namespace blender::animrig::internal {
|
|
|
|
void rebuild_slot_user_cache(Main &bmain)
|
|
{
|
|
/* Loop over all Actions and clear their slots' user cache. */
|
|
LISTBASE_FOREACH (bAction *, dna_action, &bmain.actions) {
|
|
Action &action = dna_action->wrap();
|
|
for (Slot *slot : action.slots()) {
|
|
BLI_assert_msg(slot->runtime, "Slot::runtime should always be allocated");
|
|
slot->runtime->users.clear();
|
|
}
|
|
}
|
|
|
|
/* Mark all Slots as clear. This is a bit of a lie, because the code below still has to run.
|
|
* However, this is a necessity to make the `slot.users_add(*id)` call work without triggering
|
|
* an infinite recursion.
|
|
*
|
|
* The alternative would be to go around the `slot.users_add()` function and access the
|
|
* runtime directly, but this is IMO a bit cleaner. */
|
|
bmain.is_action_slot_to_id_map_dirty = false;
|
|
|
|
/* Visit any ID to see which Action+Slot it is using. Returns whether the ID
|
|
* was visited for the first time. */
|
|
Set<ID *> visited_ids;
|
|
auto visit_id = [&visited_ids](ID *id) -> bool {
|
|
BLI_assert(id);
|
|
|
|
if (!visited_ids.add(id)) {
|
|
return false;
|
|
}
|
|
|
|
foreach_action_slot_use(*id, [&](const Action &action, slot_handle_t slot_handle) {
|
|
const Slot *slot = action.slot_for_handle(slot_handle);
|
|
if (!slot) {
|
|
return true;
|
|
}
|
|
/* Constant cast because the `foreach` produces const Actions, and I (Sybren)
|
|
* didn't want to make a non-const duplicate. */
|
|
const_cast<Slot *>(slot)->users_add(*id);
|
|
return true;
|
|
});
|
|
|
|
return true;
|
|
};
|
|
|
|
/* Loop over all IDs to cache their slot usage. */
|
|
ListBase *ids_of_idtype;
|
|
ID *id;
|
|
FOREACH_MAIN_LISTBASE_BEGIN (&bmain, ids_of_idtype) {
|
|
/* Check whether this ID type can be animated. If not, just skip all IDs of this type. */
|
|
id = static_cast<ID *>(ids_of_idtype->first);
|
|
if (!id || !id_type_can_have_animdata(GS(id->name))) {
|
|
continue;
|
|
}
|
|
|
|
FOREACH_MAIN_LISTBASE_ID_BEGIN (ids_of_idtype, id) {
|
|
BLI_assert(id_can_have_animdata(id));
|
|
|
|
/* Process the ID itself. */
|
|
if (!visit_id(id)) {
|
|
continue;
|
|
}
|
|
|
|
/* Process embedded IDs, as these are not listed in bmain, but still can
|
|
* have their own Action+Slot. Unfortunately there is no generic looper
|
|
* for embedded IDs. At this moment the only animatable embedded ID is a
|
|
* node tree. */
|
|
bNodeTree *node_tree = bke::node_tree_from_id(id);
|
|
if (node_tree) {
|
|
visit_id(&node_tree->id);
|
|
}
|
|
}
|
|
FOREACH_MAIN_LISTBASE_ID_END;
|
|
}
|
|
FOREACH_MAIN_LISTBASE_END;
|
|
}
|
|
|
|
} // namespace blender::animrig::internal
|