Files
test/source/blender/blenkernel/intern/nla.cc
Damien Picard 13d4829b7b I18n: Disambiguate "Meta" in the context of NLA's meta strips
Use the "Action" context for this, if in any language it's different
from a sequencer meta strip, which already uses "Sequence".

Reported by Ye Gui in #43295.
2025-09-29 11:07:03 +02:00

2830 lines
84 KiB
C++

/* SPDX-FileCopyrightText: 2009 Blender Authors, Joshua Leung. All rights reserved.
*
* SPDX-License-Identifier: GPL-2.0-or-later */
/** \file
* \ingroup bke
*/
#include <cfloat>
#include <cmath>
#include <cstddef>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include "CLG_log.h"
#include "MEM_guardedalloc.h"
#include "BLI_ghash.h"
#include "BLI_listbase.h"
#include "BLI_string.h"
#include "BLI_string_utf8.h"
#include "BLI_string_utils.hh"
#include "BLI_utildefines.h"
#include "BLT_translation.hh"
#include "DNA_anim_types.h"
#include "DNA_scene_types.h"
#include "DNA_speaker_types.h"
#include "BKE_action.hh"
#include "BKE_anim_data.hh"
#include "BKE_fcurve.hh"
#include "BKE_global.hh"
#include "BKE_lib_id.hh"
#include "BKE_lib_query.hh"
#include "BKE_main.hh"
#include "BKE_nla.hh"
#include "BKE_sound.h"
#include "BLO_read_write.hh"
#include "RNA_access.hh"
#include "RNA_prototypes.hh"
#include "ANIM_nla.hh"
#include "nla_private.h"
static CLG_LogRef LOG = {"anim.nla"};
using namespace blender;
/**
* Find the active track and strip.
*
* The active strip may or may not be on the active track.
*/
static void nla_tweakmode_find_active(const ListBase /*NlaTrack*/ *nla_tracks,
NlaTrack **r_track_of_active_strip,
NlaStrip **r_active_strip);
/* *************************************************** */
/* Data Management */
/* Freeing ------------------------------------------- */
void BKE_nlastrip_free(NlaStrip *strip, const bool do_id_user)
{
NlaStrip *cs, *csn;
/* sanity checks */
if (strip == nullptr) {
return;
}
/* free child-strips */
for (cs = static_cast<NlaStrip *>(strip->strips.first); cs; cs = csn) {
csn = cs->next;
BKE_nlastrip_remove_and_free(&strip->strips, cs, do_id_user);
}
/* remove reference to action */
if (strip->act != nullptr && do_id_user) {
id_us_min(&strip->act->id);
}
/* free own F-Curves */
BKE_fcurves_free(&strip->fcurves);
/* free own F-Modifiers */
free_fmodifiers(&strip->modifiers);
/* free the strip itself */
MEM_freeN(strip);
}
void BKE_nlatrack_free(NlaTrack *nlt, const bool do_id_user)
{
NlaStrip *strip, *stripn;
/* sanity checks */
if (nlt == nullptr) {
return;
}
/* free strips */
for (strip = static_cast<NlaStrip *>(nlt->strips.first); strip; strip = stripn) {
stripn = strip->next;
BKE_nlastrip_remove_and_free(&nlt->strips, strip, do_id_user);
}
/* free NLA track itself now */
MEM_freeN(nlt);
}
void BKE_nla_tracks_free(ListBase *tracks, bool do_id_user)
{
NlaTrack *nlt, *nltn;
/* sanity checks */
if (ELEM(nullptr, tracks, tracks->first)) {
return;
}
/* free tracks one by one */
for (nlt = static_cast<NlaTrack *>(tracks->first); nlt; nlt = nltn) {
nltn = nlt->next;
BKE_nlatrack_remove_and_free(tracks, nlt, do_id_user);
}
/* clear the list's pointers to be safe */
BLI_listbase_clear(tracks);
}
/* Copying ------------------------------------------- */
NlaStrip *BKE_nlastrip_copy(Main *bmain,
NlaStrip *strip,
const bool use_same_action,
const int flag)
{
NlaStrip *strip_d;
NlaStrip *cs_d;
const bool do_id_user = (flag & LIB_ID_CREATE_NO_USER_REFCOUNT) == 0;
/* sanity check */
if (strip == nullptr) {
return nullptr;
}
/* make a copy */
strip_d = static_cast<NlaStrip *>(MEM_dupallocN(strip));
strip_d->next = strip_d->prev = nullptr;
/* handle action */
if (strip_d->act) {
if (use_same_action) {
if (do_id_user) {
/* increase user-count of action */
id_us_plus(&strip_d->act->id);
}
}
else {
/* use a copy of the action instead (user count shouldn't have changed yet) */
BKE_id_copy_ex(bmain, &strip_d->act->id, (ID **)&strip_d->act, flag);
}
}
/* copy F-Curves and modifiers */
BKE_fcurves_copy(&strip_d->fcurves, &strip->fcurves);
copy_fmodifiers(&strip_d->modifiers, &strip->modifiers);
/* make a copy of all the child-strips, one at a time */
BLI_listbase_clear(&strip_d->strips);
LISTBASE_FOREACH (NlaStrip *, cs, &strip->strips) {
cs_d = BKE_nlastrip_copy(bmain, cs, use_same_action, flag);
BLI_addtail(&strip_d->strips, cs_d);
}
/* return the strip */
return strip_d;
}
NlaTrack *BKE_nlatrack_copy(Main *bmain,
NlaTrack *nlt,
const bool use_same_actions,
const int flag)
{
NlaStrip *strip_d;
NlaTrack *nlt_d;
/* sanity check */
if (nlt == nullptr) {
return nullptr;
}
/* make a copy */
nlt_d = static_cast<NlaTrack *>(MEM_dupallocN(nlt));
nlt_d->next = nlt_d->prev = nullptr;
/* make a copy of all the strips, one at a time */
BLI_listbase_clear(&nlt_d->strips);
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
strip_d = BKE_nlastrip_copy(bmain, strip, use_same_actions, flag);
BLI_addtail(&nlt_d->strips, strip_d);
}
/* return the copy */
return nlt_d;
}
void BKE_nla_tracks_copy(Main *bmain, ListBase *dst, const ListBase *src, const int flag)
{
NlaTrack *nlt_d;
/* sanity checks */
if (ELEM(nullptr, dst, src)) {
return;
}
/* clear out the destination list first for precautions... */
BLI_listbase_clear(dst);
/* copy each NLA-track, one at a time */
LISTBASE_FOREACH (NlaTrack *, nlt, src) {
/* make a copy, and add the copy to the destination list */
/* XXX: we need to fix this sometime. */
nlt_d = BKE_nlatrack_copy(bmain, nlt, true, flag);
BLI_addtail(dst, nlt_d);
}
}
/**
* Find `active_strip` in `strips_source`, then return the strip with the same
* index from `strips_dest`.
*/
static NlaStrip *find_active_strip_from_listbase(const NlaStrip *active_strip,
const ListBase /*NlaStrip*/ *strips_source,
const ListBase /*NlaStrip*/ *strips_dest)
{
BLI_assert_msg(BLI_listbase_count(strips_source) == BLI_listbase_count(strips_dest),
"Expecting the same number of source and destination strips");
NlaStrip *strip_dest = static_cast<NlaStrip *>(strips_dest->first);
LISTBASE_FOREACH (const NlaStrip *, strip_source, strips_source) {
if (strip_dest == nullptr) {
/* The tracks are assumed to have an equal number of strips, but this is
* not the case. Not sure when this might happen, but it's better to not
* crash. */
break;
}
if (strip_source == active_strip) {
return strip_dest;
}
const bool src_is_meta = strip_source->type == NLASTRIP_TYPE_META;
const bool dst_is_meta = strip_dest->type == NLASTRIP_TYPE_META;
BLI_assert_msg(src_is_meta == dst_is_meta,
"Expecting topology of source and destination strips to be equal");
if (src_is_meta && dst_is_meta) {
NlaStrip *found_in_meta = find_active_strip_from_listbase(
active_strip, &strip_source->strips, &strip_dest->strips);
if (found_in_meta != nullptr) {
return found_in_meta;
}
}
strip_dest = strip_dest->next;
}
return nullptr;
}
/* Set adt_dest->actstrip to the strip with the same index as
* adt_source->actstrip. Note that this always sets `adt_dest->actstrip`; sets
* to nullptr when `adt_source->actstrip` cannot be found. */
static void update_active_strip(AnimData *adt_dest,
NlaTrack *track_dest,
const AnimData *adt_source,
const NlaTrack *track_source)
{
BLI_assert(BLI_listbase_count(&track_source->strips) == BLI_listbase_count(&track_dest->strips));
NlaStrip *active_strip = find_active_strip_from_listbase(
adt_source->actstrip, &track_source->strips, &track_dest->strips);
adt_dest->actstrip = active_strip;
}
/* Set adt_dest->act_track to the track with the same index as adt_source->act_track. */
static void update_active_track(AnimData *adt_dest, const AnimData *adt_source)
{
adt_dest->act_track = nullptr;
adt_dest->actstrip = nullptr;
if (adt_source->act_track == nullptr && adt_source->actstrip == nullptr) {
return;
}
BLI_assert(BLI_listbase_count(&adt_source->nla_tracks) ==
BLI_listbase_count(&adt_dest->nla_tracks));
NlaTrack *track_dest = static_cast<NlaTrack *>(adt_dest->nla_tracks.first);
LISTBASE_FOREACH (NlaTrack *, track_source, &adt_source->nla_tracks) {
if (track_source == adt_source->act_track) {
adt_dest->act_track = track_dest;
}
/* Only search for the active strip if it hasn't been found yet. */
if (adt_dest->actstrip == nullptr && adt_source->actstrip != nullptr) {
update_active_strip(adt_dest, track_dest, adt_source, track_source);
}
track_dest = track_dest->next;
}
#ifndef NDEBUG
{
const bool source_has_actstrip = adt_source->actstrip != nullptr;
const bool dest_has_actstrip = adt_dest->actstrip != nullptr;
BLI_assert_msg(source_has_actstrip == dest_has_actstrip,
"Active strip did not copy correctly");
}
#endif
}
void BKE_nla_tracks_copy_from_adt(Main *bmain,
AnimData *adt_dest,
const AnimData *adt_source,
const int flag)
{
adt_dest->act_track = nullptr;
adt_dest->actstrip = nullptr;
BKE_nla_tracks_copy(bmain, &adt_dest->nla_tracks, &adt_source->nla_tracks, flag);
update_active_track(adt_dest, adt_source);
}
/* Adding ------------------------------------------- */
NlaTrack *BKE_nlatrack_new()
{
/* allocate new track */
NlaTrack *nlt = MEM_callocN<NlaTrack>("NlaTrack");
/* set settings requiring the track to not be part of the stack yet */
nlt->flag = NLATRACK_SELECTED | NLATRACK_OVERRIDELIBRARY_LOCAL;
return nlt;
}
void BKE_nlatrack_insert_before(ListBase *nla_tracks,
NlaTrack *next,
NlaTrack *new_track,
bool is_liboverride)
{
if (is_liboverride) {
/* Currently, all library override tracks are assumed to be grouped together at the start of
* the list. Non overridden must be placed after last library track. */
if (next != nullptr && (next->flag & NLATRACK_OVERRIDELIBRARY_LOCAL) == 0) {
BKE_nlatrack_insert_after(nla_tracks, next, new_track, is_liboverride);
return;
}
}
BLI_insertlinkbefore(nla_tracks, next, new_track);
new_track->index = BLI_findindex(nla_tracks, new_track);
/* Must have unique name, but we need to seed this. */
STRNCPY_UTF8(new_track->name, "NlaTrack");
BLI_uniquename(nla_tracks,
new_track,
DATA_("NlaTrack"),
'.',
offsetof(NlaTrack, name),
sizeof(new_track->name));
}
void BKE_nlatrack_insert_after(ListBase *nla_tracks,
NlaTrack *prev,
NlaTrack *new_track,
const bool is_liboverride)
{
BLI_assert(nla_tracks);
BLI_assert(new_track);
/* If nullptr, then caller intends to insert a new head. But, tracks are not allowed to be
* placed before library overrides. So it must inserted after the last override. */
if (prev == nullptr) {
NlaTrack *first_track = (NlaTrack *)nla_tracks->first;
if (first_track != nullptr && (first_track->flag & NLATRACK_OVERRIDELIBRARY_LOCAL) == 0) {
prev = first_track;
}
}
/* In liboverride case, we only add local tracks after all those coming from the linked data,
* so we need to find the first local track. */
if (is_liboverride && prev != nullptr && (prev->flag & NLATRACK_OVERRIDELIBRARY_LOCAL) == 0) {
NlaTrack *first_local = prev->next;
for (; first_local != nullptr && (first_local->flag & NLATRACK_OVERRIDELIBRARY_LOCAL) == 0;
first_local = first_local->next)
{
}
prev = first_local != nullptr ? first_local->prev : prev;
}
/* Add track to stack, and make it the active one. */
BLI_insertlinkafter(nla_tracks, prev, new_track);
new_track->index = BLI_findindex(nla_tracks, new_track);
/* must have unique name, but we need to seed this */
BLI_uniquename(nla_tracks,
new_track,
DATA_("NlaTrack"),
'.',
offsetof(NlaTrack, name),
sizeof(new_track->name));
}
NlaTrack *BKE_nlatrack_new_before(ListBase *nla_tracks, NlaTrack *next, bool is_liboverride)
{
NlaTrack *new_track = BKE_nlatrack_new();
BKE_nlatrack_insert_before(nla_tracks, next, new_track, is_liboverride);
return new_track;
}
NlaTrack *BKE_nlatrack_new_after(ListBase *nla_tracks, NlaTrack *prev, bool is_liboverride)
{
NlaTrack *new_track = BKE_nlatrack_new();
BKE_nlatrack_insert_after(nla_tracks, prev, new_track, is_liboverride);
return new_track;
}
NlaTrack *BKE_nlatrack_new_head(ListBase *nla_tracks, bool is_liboverride)
{
return BKE_nlatrack_new_before(nla_tracks, (NlaTrack *)nla_tracks->first, is_liboverride);
}
NlaTrack *BKE_nlatrack_new_tail(ListBase *nla_tracks, const bool is_liboverride)
{
return BKE_nlatrack_new_after(nla_tracks, (NlaTrack *)nla_tracks->last, is_liboverride);
}
float BKE_nla_clip_length_get_nonzero(const NlaStrip *strip)
{
if (strip->actend <= strip->actstart) {
return 1.0f;
}
return strip->actend - strip->actstart;
}
void BKE_nla_clip_length_ensure_nonzero(const float *actstart, float *r_actend)
{
if (*r_actend <= *actstart) {
*r_actend = *actstart + 1.0f;
}
}
/**
* Create a new strip for the given Action, auto-choosing a suitable Slot.
*
* This does _not_ sync the Action length. This way the caller can determine whether to do that
* immediately after, or whether an explicit slot should be chosen first.
*/
static NlaStrip *nlastrip_new(bAction *act, ID &animated_id)
{
using namespace blender::animrig;
NlaStrip *strip;
/* sanity checks */
if (act == nullptr) {
return nullptr;
}
/* allocate new strip */
strip = MEM_callocN<NlaStrip>("NlaStrip");
/* generic settings
* - selected flag to highlight this to the user
* - (XXX) disabled Auto-Blends, as this was often causing some unwanted effects
*/
strip->flag = NLASTRIP_FLAG_SELECT | NLASTRIP_FLAG_SYNC_LENGTH;
/* Disable sync for actions with a manual frame range, since it only syncs to range anyway. */
if (act->flag & ACT_FRAME_RANGE) {
strip->flag &= ~NLASTRIP_FLAG_SYNC_LENGTH;
}
/* Enable cyclic time for known cyclic actions. */
Action &action = act->wrap();
if (action.is_cyclic()) {
strip->flag |= NLASTRIP_FLAG_USR_TIME_CYCLIC;
}
/* Assign the Action, and automatically choose a suitable slot. The caller can change the slot to
* something more specific later, if necessary. */
nla::assign_action(*strip, action, animated_id);
/* strip should be referenced as-is */
strip->scale = 1.0f;
strip->repeat = 1.0f;
return strip;
}
/**
* Same as #BKE_nlastrip_recalculate_bounds_sync_action() except that it doesn't
* take any previous values into account.
*/
static void nlastrip_set_initial_length(NlaStrip *strip)
{
const float2 frame_range = strip->act->wrap().get_frame_range_of_slot(strip->action_slot_handle);
strip->actstart = frame_range[0];
strip->actend = frame_range[1];
BKE_nla_clip_length_ensure_nonzero(&strip->actstart, &strip->actend);
strip->start = strip->actstart;
strip->end = strip->actend;
}
NlaStrip *BKE_nlastrip_new(bAction *act, ID &animated_id)
{
NlaStrip *strip = nlastrip_new(act, animated_id);
if (!strip) {
return nullptr;
}
nlastrip_set_initial_length(strip);
return strip;
}
NlaStrip *BKE_nlastrip_new_for_slot(bAction *act,
blender::animrig::slot_handle_t slot_handle,
ID &animated_id)
{
using namespace blender::animrig;
NlaStrip *strip = BKE_nlastrip_new(act, animated_id);
if (!strip) {
return nullptr;
}
const ActionSlotAssignmentResult result = nla::assign_action_slot_handle(
*strip, slot_handle, animated_id);
switch (result) {
case ActionSlotAssignmentResult::OK:
break;
case ActionSlotAssignmentResult::SlotNotFromAction:
case ActionSlotAssignmentResult::MissingAction:
BLI_assert_unreachable();
[[fallthrough]];
case ActionSlotAssignmentResult::SlotNotSuitable:
BKE_nlastrip_free(strip, true);
return nullptr;
}
nlastrip_set_initial_length(strip);
return strip;
}
NlaStrip *BKE_nlastack_add_strip(const OwnedAnimData owned_adt, const bool is_liboverride)
{
NlaStrip *strip;
NlaTrack *nlt;
AnimData *adt = &owned_adt.adt;
/* sanity checks */
if (ELEM(nullptr, adt, adt->action)) {
return nullptr;
}
/* create a new NLA strip */
strip = BKE_nlastrip_new_for_slot(adt->action, adt->slot_handle, owned_adt.owner_id);
if (strip == nullptr) {
return nullptr;
}
/* firstly try adding strip to last track, but if that fails, add to a new track */
if (BKE_nlatrack_add_strip(
static_cast<NlaTrack *>(adt->nla_tracks.last), strip, is_liboverride) == 0)
{
/* trying to add to the last track failed (no track or no space),
* so add a new track to the stack, and add to that...
*/
nlt = BKE_nlatrack_new_tail(&adt->nla_tracks, is_liboverride);
BKE_nlatrack_set_active(&adt->nla_tracks, nlt);
BKE_nlatrack_add_strip(nlt, strip, is_liboverride);
STRNCPY_UTF8(nlt->name, adt->action->id.name + 2);
}
/* automatically name it too */
BKE_nlastrip_validate_name(adt, strip);
/* returns the strip added */
return strip;
}
NlaStrip *BKE_nla_add_soundstrip(Main *bmain, Scene *scene, Speaker *speaker)
{
NlaStrip *strip = MEM_callocN<NlaStrip>("NlaSoundStrip");
/* if speaker has a sound, set the strip length to the length of the sound,
* otherwise default to length of 10 frames
*/
#ifdef WITH_AUDASPACE
if (speaker->sound) {
SoundInfo info;
if (BKE_sound_info_get(bmain, speaker->sound, &info)) {
strip->end = float(ceil(double(info.length) * scene->frames_per_second()));
}
}
else
#endif
{
strip->end = 10.0f;
/* quiet compiler warnings */
UNUSED_VARS(bmain, scene, speaker);
}
/* general settings */
strip->type = NLASTRIP_TYPE_SOUND;
strip->flag = NLASTRIP_FLAG_SELECT;
strip->extendmode = NLASTRIP_EXTEND_NOTHING; /* nothing to extend... */
/* strip should be referenced as-is */
strip->scale = 1.0f;
strip->repeat = 1.0f;
/* return this strip */
return strip;
}
void BKE_nla_strip_foreach_id(NlaStrip *strip, LibraryForeachIDData *data)
{
BKE_LIB_FOREACHID_PROCESS_IDSUPER(data, strip->act, IDWALK_CB_USER);
LISTBASE_FOREACH (FCurve *, fcu, &strip->fcurves) {
BKE_LIB_FOREACHID_PROCESS_FUNCTION_CALL(data, BKE_fcurve_foreach_id(fcu, data));
}
LISTBASE_FOREACH (NlaStrip *, substrip, &strip->strips) {
BKE_LIB_FOREACHID_PROCESS_FUNCTION_CALL(data, BKE_nla_strip_foreach_id(substrip, data));
}
}
/* Removing ------------------------------------------ */
void BKE_nlatrack_remove(ListBase *tracks, NlaTrack *nlt)
{
BLI_assert(tracks);
BLI_remlink(tracks, nlt);
}
void BKE_nlatrack_remove_and_free(ListBase *tracks, NlaTrack *nlt, bool do_id_user)
{
BKE_nlatrack_remove(tracks, nlt);
BKE_nlatrack_free(nlt, do_id_user);
}
bool BKE_nlatrack_is_enabled(const AnimData &adt, const NlaTrack &nlt)
{
if (adt.flag & ADT_NLA_SOLO_TRACK) {
return (nlt.flag & NLATRACK_SOLO);
}
return !(nlt.flag & NLATRACK_MUTED);
}
/* *************************************************** */
/* NLA Evaluation <-> Editing Stuff */
/* Strip Mapping ------------------------------------- */
/* non clipped mapping for strip-time <-> global time (for Action-Clips)
* invert = convert action-strip time to global time
*/
static float nlastrip_get_frame_actionclip(NlaStrip *strip, float cframe, short mode)
{
float scale;
// float repeat; // UNUSED
/* get number of repeats */
if (IS_EQF(strip->repeat, 0.0f)) {
strip->repeat = 1.0f;
}
// repeat = strip->repeat; /* UNUSED */
/* scaling */
if (IS_EQF(strip->scale, 0.0f)) {
strip->scale = 1.0f;
}
/* Scale must be positive - we've got a special flag for reversing. */
scale = fabsf(strip->scale);
/* length of referenced action */
const float actlength = BKE_nla_clip_length_get_nonzero(strip);
/* reversed = play strip backwards */
if (strip->flag & NLASTRIP_FLAG_REVERSE) {
/* FIXME: this won't work right with Graph Editor? */
if (mode == NLATIME_CONVERT_MAP) {
return strip->end - scale * (cframe - strip->actstart);
}
if (mode == NLATIME_CONVERT_UNMAP) {
return (strip->end + (strip->actstart * scale - cframe)) / scale;
}
/* if (mode == NLATIME_CONVERT_EVAL) */
if (IS_EQF(float(cframe), strip->end) && IS_EQF(strip->repeat, floorf(strip->repeat))) {
/* This case prevents the motion snapping back to the first frame at the end of the strip
* by catching the case where repeats is a whole number, which means that the end of the
* strip could also be interpreted as the end of the start of a repeat. */
return strip->actstart;
}
/* - The `fmod(..., actlength * scale)` is needed to get the repeats working.
* - The `/ scale` is needed to ensure that scaling influences the timing within the repeat.
*/
return strip->actend - fmodf(cframe - strip->start, actlength * scale) / scale;
}
if (mode == NLATIME_CONVERT_MAP) {
return strip->start + scale * (cframe - strip->actstart);
}
if (mode == NLATIME_CONVERT_UNMAP) {
return strip->actstart + (cframe - strip->start) / scale;
}
/* if (mode == NLATIME_CONVERT_EVAL) */
if (IS_EQF(cframe, strip->end) && IS_EQF(strip->repeat, floorf(strip->repeat))) {
/* This case prevents the motion snapping back to the first frame at the end of the strip
* by catching the case where repeats is a whole number, which means that the end of the
* strip could also be interpreted as the end of the start of a repeat. */
return strip->actend;
}
/* - The `fmod(..., actlength * scale)` is needed to get the repeats working.
* - The `/ scale` is needed to ensure that scaling influences the timing within the repeat.
*/
return strip->actstart + fmodf(cframe - strip->start, actlength * scale) / scale;
}
/* non clipped mapping for strip-time <-> global time (for Transitions)
* invert = convert action-strip time to global time
*/
static float nlastrip_get_frame_transition(NlaStrip *strip, float cframe, short mode)
{
float length;
/* length of strip */
length = strip->end - strip->start;
/* reversed = play strip backwards */
if (strip->flag & NLASTRIP_FLAG_REVERSE) {
if (mode == NLATIME_CONVERT_MAP) {
return strip->end - (length * cframe);
}
return (strip->end - cframe) / length;
}
if (mode == NLATIME_CONVERT_MAP) {
return (length * cframe) + strip->start;
}
return (cframe - strip->start) / length;
}
float nlastrip_get_frame(NlaStrip *strip, float cframe, short mode)
{
switch (strip->type) {
case NLASTRIP_TYPE_META: /* Meta - for now, does the same as transition
* (is really just an empty container). */
case NLASTRIP_TYPE_TRANSITION: /* transition */
return nlastrip_get_frame_transition(strip, cframe, mode);
case NLASTRIP_TYPE_CLIP: /* action-clip (default) */
default:
return nlastrip_get_frame_actionclip(strip, cframe, mode);
}
}
float BKE_nla_tweakedit_remap(AnimData *adt, const float cframe, const eNlaTime_ConvertModes mode)
{
NlaStrip *strip;
/* Sanity checks:
* - Obviously we've got to have some starting data.
* - When not in tweak-mode, the active Action does not have any scaling applied :)
* - When in tweak-mode, if the no-mapping flag is set, do not map.
*/
if ((adt == nullptr) || (adt->flag & ADT_NLA_EDIT_ON) == 0 || (adt->flag & ADT_NLA_EDIT_NOMAP)) {
return cframe;
}
/* if the active-strip info has been stored already, access this, otherwise look this up
* and store for (very probable) future usage
*/
if (adt->act_track == nullptr) {
if (adt->actstrip) {
adt->act_track = BKE_nlatrack_find_tweaked(adt);
}
else {
adt->act_track = BKE_nlatrack_find_active(&adt->nla_tracks);
}
}
if (adt->actstrip == nullptr) {
adt->actstrip = BKE_nlastrip_find_active(adt->act_track);
}
strip = adt->actstrip;
/* Sanity checks:
* - In rare cases, we may not be able to find this strip for some reason (internal error)
* - For now, if the user has defined a curve to control the time, this correction cannot be
* performed reliably.
*/
if ((strip == nullptr) || (strip->flag & NLASTRIP_FLAG_USR_TIME)) {
return cframe;
}
/* perform the correction now... */
return nlastrip_get_frame(strip, cframe, mode);
}
/* *************************************************** */
/* NLA API */
/* List of Strips ------------------------------------ */
/* (these functions are used for NLA-Tracks and also for nested/meta-strips) */
bool BKE_nlastrips_has_space(ListBase *strips, float start, float end)
{
/* sanity checks */
if ((strips == nullptr) || IS_EQF(start, end)) {
return false;
}
if (start > end) {
puts("BKE_nlastrips_has_space() error... start and end arguments swapped");
std::swap(start, end);
}
/* loop over NLA strips checking for any overlaps with this area... */
LISTBASE_FOREACH (NlaStrip *, strip, strips) {
/* if start frame of strip is past the target end-frame, that means that
* we've gone past the window we need to check for, so things are fine
*/
if (strip->start >= end) {
return true;
}
/* if the end of the strip is greater than either of the boundaries, the range
* must fall within the extents of the strip
*/
if ((strip->end > start) || (strip->end > end)) {
return false;
}
}
/* if we are still here, we haven't encountered any overlapping strips */
return true;
}
void BKE_nlastrips_sort_strips(ListBase *strips)
{
ListBase tmp = {nullptr, nullptr};
NlaStrip *strip, *stripn;
/* sanity checks */
if (ELEM(nullptr, strips, strips->first)) {
return;
}
/* we simply perform insertion sort on this list, since it is assumed that per track,
* there are only likely to be at most 5-10 strips
*/
for (strip = static_cast<NlaStrip *>(strips->first); strip; strip = stripn) {
short not_added = 1;
stripn = strip->next;
/* remove this strip from the list, and add it to the new list, searching from the end of
* the list, assuming that the lists are in order
*/
BLI_remlink(strips, strip);
LISTBASE_FOREACH_BACKWARD (NlaStrip *, sstrip, &tmp) {
/* check if add after */
if (sstrip->start <= strip->start) {
BLI_insertlinkafter(&tmp, sstrip, strip);
not_added = 0;
break;
}
}
/* add before first? */
if (not_added) {
BLI_addhead(&tmp, strip);
}
}
/* reassign the start and end points of the strips */
strips->first = tmp.first;
strips->last = tmp.last;
}
void BKE_nlastrips_add_strip_unsafe(ListBase *strips, NlaStrip *strip)
{
bool not_added = true;
/* sanity checks */
BLI_assert(!ELEM(nullptr, strips, strip));
/* find the right place to add the strip to the nominated track */
LISTBASE_FOREACH (NlaStrip *, ns, strips) {
/* if current strip occurs after the new strip, add it before */
if (ns->start >= strip->start) {
BLI_insertlinkbefore(strips, ns, strip);
not_added = false;
break;
}
}
if (not_added) {
/* just add to the end of the list of the strips then... */
BLI_addtail(strips, strip);
}
}
bool BKE_nlastrips_add_strip(ListBase *strips, NlaStrip *strip)
{
if (ELEM(nullptr, strips, strip)) {
return false;
}
if (!BKE_nlastrips_has_space(strips, strip->start, strip->end)) {
return false;
}
BKE_nlastrips_add_strip_unsafe(strips, strip);
return true;
}
/* Meta-Strips ------------------------------------ */
void BKE_nlastrips_make_metas(ListBase *strips, bool is_temp)
{
NlaStrip *mstrip = nullptr;
NlaStrip *strip, *stripn;
/* sanity checks */
if (ELEM(nullptr, strips, strips->first)) {
return;
}
/* group all continuous chains of selected strips into meta-strips */
for (strip = static_cast<NlaStrip *>(strips->first); strip; strip = stripn) {
stripn = strip->next;
if (strip->flag & NLASTRIP_FLAG_SELECT) {
/* if there is an existing meta-strip, add this strip to it, otherwise, create a new one */
if (mstrip == nullptr) {
/* add a new meta-strip, and add it before the current strip that it will replace... */
mstrip = MEM_callocN<NlaStrip>("Meta-NlaStrip");
mstrip->type = NLASTRIP_TYPE_META;
BLI_insertlinkbefore(strips, strip, mstrip);
/* set flags */
mstrip->flag = NLASTRIP_FLAG_SELECT;
/* set temp flag if appropriate (i.e. for transform-type editing) */
if (is_temp) {
mstrip->flag |= NLASTRIP_FLAG_TEMP_META;
}
/* set default repeat/scale values to prevent warnings */
mstrip->repeat = mstrip->scale = 1.0f;
/* make its start frame be set to the start frame of the current strip */
mstrip->start = strip->start;
}
/* remove the selected strips from the track, and add to the meta */
BLI_remlink(strips, strip);
BLI_addtail(&mstrip->strips, strip);
/* expand the meta's dimensions to include the newly added strip- i.e. its last frame */
mstrip->end = strip->end;
}
else {
/* current strip wasn't selected, so the end of 'island' of selected strips has been
* reached, so stop adding strips to the current meta.
*/
mstrip = nullptr;
}
}
}
void BKE_nlastrips_clear_metastrip(ListBase *strips, NlaStrip *strip)
{
NlaStrip *cs, *csn;
/* sanity check */
if (ELEM(nullptr, strips, strip)) {
return;
}
/* move each one of the meta-strip's children before the meta-strip
* in the list of strips after unlinking them from the meta-strip
*/
for (cs = static_cast<NlaStrip *>(strip->strips.first); cs; cs = csn) {
csn = cs->next;
BLI_remlink(&strip->strips, cs);
BLI_insertlinkbefore(strips, strip, cs);
}
/* free the meta-strip now */
BKE_nlastrip_remove_and_free(strips, strip, true);
}
void BKE_nlastrips_clear_metas(ListBase *strips, bool only_sel, bool only_temp)
{
NlaStrip *strip, *stripn;
/* sanity checks */
if (ELEM(nullptr, strips, strips->first)) {
return;
}
/* remove meta-strips fitting the criteria of the arguments */
for (strip = static_cast<NlaStrip *>(strips->first); strip; strip = stripn) {
stripn = strip->next;
/* check if strip is a meta-strip */
if (strip->type == NLASTRIP_TYPE_META) {
/* if check if selection and 'temporary-only' considerations are met */
if ((!only_sel) || (strip->flag & NLASTRIP_FLAG_SELECT)) {
if ((!only_temp) || (strip->flag & NLASTRIP_FLAG_TEMP_META)) {
BKE_nlastrips_clear_metastrip(strips, strip);
}
}
}
}
}
bool BKE_nlameta_add_strip(NlaStrip *mstrip, NlaStrip *strip)
{
/* sanity checks */
if (ELEM(nullptr, mstrip, strip)) {
return false;
}
/* firstly, check if the meta-strip has space for this */
if (BKE_nlastrips_has_space(&mstrip->strips, strip->start, strip->end) == 0) {
return false;
}
/* check if this would need to be added to the ends of the meta,
* and subsequently, if the neighboring strips allow us enough room
*/
if (strip->start < mstrip->start) {
/* check if strip to the left (if it exists) ends before the
* start of the strip we're trying to add
*/
if ((mstrip->prev == nullptr) || (mstrip->prev->end <= strip->start)) {
/* add strip to start of meta's list, and expand dimensions */
BLI_addhead(&mstrip->strips, strip);
mstrip->start = strip->start;
return true;
}
/* failed... no room before */
return false;
}
if (strip->end > mstrip->end) {
/* check if strip to the right (if it exists) starts before the
* end of the strip we're trying to add
*/
if ((mstrip->next == nullptr) || (mstrip->next->start >= strip->end)) {
/* add strip to end of meta's list, and expand dimensions */
BLI_addtail(&mstrip->strips, strip);
mstrip->end = strip->end;
return true;
}
/* failed... no room after */
return false;
}
/* just try to add to the meta-strip (no dimension changes needed) */
return BKE_nlastrips_add_strip(&mstrip->strips, strip);
}
void BKE_nlameta_flush_transforms(NlaStrip *mstrip)
{
float oStart, oEnd, offset;
float oLen, nLen;
short scaleChanged = 0;
/* sanity checks
* - strip must exist
* - strip must be a meta-strip with some contents
*/
if (ELEM(nullptr, mstrip, mstrip->strips.first)) {
return;
}
if (mstrip->type != NLASTRIP_TYPE_META) {
return;
}
/* get the original start/end points, and calculate the start-frame offset
* - these are simply the start/end frames of the child strips,
* since we assume they weren't transformed yet
*/
oStart = ((NlaStrip *)mstrip->strips.first)->start;
oEnd = ((NlaStrip *)mstrip->strips.last)->end;
offset = mstrip->start - oStart;
/* check if scale changed */
oLen = oEnd - oStart;
nLen = mstrip->end - mstrip->start;
scaleChanged = !IS_EQF(oLen, nLen);
/* optimization:
* don't flush if nothing changed yet
* TODO: maybe we need a flag to say always flush?
*/
if (IS_EQF(oStart, mstrip->start) && IS_EQF(oEnd, mstrip->end) && !scaleChanged) {
return;
}
/* for each child-strip, calculate new start/end points based on this new info */
LISTBASE_FOREACH (NlaStrip *, strip, &mstrip->strips) {
if (scaleChanged) {
float p1, p2;
if (oLen) {
/* Compute positions of endpoints relative to old extents of strip. */
p1 = (strip->start - oStart) / oLen;
p2 = (strip->end - oStart) / oLen;
}
else {
/* WORKAROUND: in theory, a strip should never be zero length. However,
* zero-length strips are nevertheless showing up here (see issue #113552).
* This is a stop-gap fix to handle that and prevent a divide by zero. A
* proper fix will need to track down and fix the source(s) of these
* zero-length strips. */
p1 = 0.0f;
p2 = 1.0f;
}
/* Apply new strip endpoints using the proportions,
* then wait for second pass to flush scale properly. */
strip->start = (p1 * nLen) + mstrip->start;
strip->end = (p2 * nLen) + mstrip->start;
/* Recompute the playback scale, given the new start & end frame of the strip. */
const double action_len = strip->actend - strip->actstart;
const double repeated_len = action_len * strip->repeat;
const double strip_len = strip->end - strip->start;
strip->scale = strip_len / repeated_len;
}
else {
/* just apply the changes in offset to both ends of the strip */
strip->start += offset;
strip->end += offset;
}
}
/* apply a second pass over child strips, to finish up unfinished business */
LISTBASE_FOREACH (NlaStrip *, strip, &mstrip->strips) {
/* only if scale changed, need to perform RNA updates */
if (scaleChanged) {
/* use RNA updates to compute scale properly */
PointerRNA ptr = RNA_pointer_create_discrete(nullptr, &RNA_NlaStrip, strip);
RNA_float_set(&ptr, "frame_start", strip->start);
RNA_float_set(&ptr, "frame_end", strip->end);
}
/* finally, make sure the strip's children (if it is a meta-itself), get updated */
BKE_nlameta_flush_transforms(strip);
}
}
/* NLA-Tracks ---------------------------------------- */
NlaTrack *BKE_nlatrack_find_active(ListBase *tracks)
{
/* sanity check */
if (ELEM(nullptr, tracks, tracks->first)) {
return nullptr;
}
/* try to find the first active track */
LISTBASE_FOREACH (NlaTrack *, nlt, tracks) {
if (nlt->flag & NLATRACK_ACTIVE) {
return nlt;
}
}
/* none found */
return nullptr;
}
NlaTrack *BKE_nlatrack_find_tweaked(AnimData *adt)
{
/* sanity check */
if (adt == nullptr) {
return nullptr;
}
/* Since the track itself gets disabled, we want the first disabled... */
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
if (nlt->flag & (NLATRACK_ACTIVE | NLATRACK_DISABLED)) {
/* For good measure, make sure that strip actually exists there */
if (BLI_findindex(&nlt->strips, adt->actstrip) != -1) {
return nlt;
}
if (G.debug & G_DEBUG) {
printf("%s: Active strip (%p, %s) not in NLA track found (%p, %s)\n",
__func__,
adt->actstrip,
(adt->actstrip) ? adt->actstrip->name : "<None>",
nlt,
nlt->name);
}
}
}
/* Not found! */
return nullptr;
}
void BKE_nlatrack_solo_toggle(AnimData *adt, NlaTrack *nlt)
{
/* sanity check */
if (ELEM(nullptr, adt, adt->nla_tracks.first)) {
return;
}
/* firstly, make sure 'solo' flag for all tracks is disabled */
LISTBASE_FOREACH (NlaTrack *, nt, &adt->nla_tracks) {
if (nt != nlt) {
nt->flag &= ~NLATRACK_SOLO;
}
}
/* now, enable 'solo' for the given track if appropriate */
if (nlt) {
/* toggle solo status */
nlt->flag ^= NLATRACK_SOLO;
/* set or clear solo-status on AnimData */
if (nlt->flag & NLATRACK_SOLO) {
adt->flag |= ADT_NLA_SOLO_TRACK;
}
else {
adt->flag &= ~ADT_NLA_SOLO_TRACK;
}
}
else {
adt->flag &= ~ADT_NLA_SOLO_TRACK;
}
}
void BKE_nlatrack_set_active(ListBase *tracks, NlaTrack *nlt_a)
{
/* sanity check */
if (ELEM(nullptr, tracks, tracks->first)) {
return;
}
/* deactivate all the rest */
LISTBASE_FOREACH (NlaTrack *, nlt, tracks) {
nlt->flag &= ~NLATRACK_ACTIVE;
}
/* set the given one as the active one */
if (nlt_a) {
nlt_a->flag |= NLATRACK_ACTIVE;
}
}
bool BKE_nlatrack_has_space(NlaTrack *nlt, float start, float end)
{
/* sanity checks
* - track must exist
* - track must be editable
* - bounds cannot be equal (0-length is nasty)
*/
if ((nlt == nullptr) || (nlt->flag & NLATRACK_PROTECTED) || IS_EQF(start, end)) {
return false;
}
if (start > end) {
puts("BKE_nlatrack_has_space() error... start and end arguments swapped");
std::swap(start, end);
}
/* check if there's any space left in the track for a strip of the given length */
return BKE_nlastrips_has_space(&nlt->strips, start, end);
}
bool BKE_nlatrack_has_strips(ListBase *tracks)
{
/* sanity checks */
if (BLI_listbase_is_empty(tracks)) {
return false;
}
/* Check each track for NLA strips. */
LISTBASE_FOREACH (NlaTrack *, track, tracks) {
if (BLI_listbase_count(&track->strips) > 0) {
return true;
}
}
/* none found */
return false;
}
void BKE_nlatrack_sort_strips(NlaTrack *nlt)
{
/* sanity checks */
if (ELEM(nullptr, nlt, nlt->strips.first)) {
return;
}
/* sort the strips with a more generic function */
BKE_nlastrips_sort_strips(&nlt->strips);
}
bool BKE_nlatrack_add_strip(NlaTrack *nlt, NlaStrip *strip, const bool is_liboverride)
{
/* sanity checks */
if (ELEM(nullptr, nlt, strip)) {
return false;
}
/*
* Do not allow adding strips if this track is locked, or not a local one in liboverride case.
*/
if (nlt->flag & NLATRACK_PROTECTED ||
(is_liboverride && (nlt->flag & NLATRACK_OVERRIDELIBRARY_LOCAL) == 0))
{
return false;
}
/* try to add the strip to the track using a more generic function */
return BKE_nlastrips_add_strip(&nlt->strips, strip);
}
void BKE_nlatrack_remove_strip(NlaTrack *track, NlaStrip *strip)
{
BLI_assert(track);
BKE_nlastrip_remove(&track->strips, strip);
}
bool BKE_nlatrack_get_bounds(NlaTrack *nlt, float bounds[2])
{
NlaStrip *strip;
/* initialize bounds */
if (bounds) {
bounds[0] = bounds[1] = 0.0f;
}
else {
return false;
}
/* sanity checks */
if (ELEM(nullptr, nlt, nlt->strips.first)) {
return false;
}
/* lower bound is first strip's start frame */
strip = static_cast<NlaStrip *>(nlt->strips.first);
bounds[0] = strip->start;
/* upper bound is last strip's end frame */
strip = static_cast<NlaStrip *>(nlt->strips.last);
bounds[1] = strip->end;
/* done */
return true;
}
bool BKE_nlatrack_is_nonlocal_in_liboverride(const ID *id, const NlaTrack *nlt)
{
return (ID_IS_OVERRIDE_LIBRARY(id) &&
(nlt == nullptr || (nlt->flag & NLATRACK_OVERRIDELIBRARY_LOCAL) == 0));
}
/* NLA Strips -------------------------------------- */
static NlaStrip *nlastrip_find_active(ListBase /*NlaStrip*/ *strips)
{
LISTBASE_FOREACH (NlaStrip *, strip, strips) {
if (strip->flag & NLASTRIP_FLAG_ACTIVE) {
return strip;
}
if (strip->type != NLASTRIP_TYPE_META) {
continue;
}
NlaStrip *inner_active = nlastrip_find_active(&strip->strips);
if (inner_active != nullptr) {
return inner_active;
}
}
return nullptr;
}
float BKE_nlastrip_compute_frame_from_previous_strip(NlaStrip *strip)
{
float limit_prev = MINAFRAMEF;
/* Find the previous end frame, with a special case if the previous strip was a transition : */
if (strip->prev) {
if (strip->prev->type == NLASTRIP_TYPE_TRANSITION) {
limit_prev = strip->prev->start + NLASTRIP_MIN_LEN_THRESH;
}
else {
limit_prev = strip->prev->end;
}
}
return limit_prev;
}
float BKE_nlastrip_compute_frame_to_next_strip(NlaStrip *strip)
{
float limit_next = MAXFRAMEF;
/* Find the next begin frame, with a special case if the next strip's a transition : */
if (strip->next) {
if (strip->next->type == NLASTRIP_TYPE_TRANSITION) {
limit_next = strip->next->end - NLASTRIP_MIN_LEN_THRESH;
}
else {
limit_next = strip->next->start;
}
}
return limit_next;
}
NlaStrip *BKE_nlastrip_next_in_track(NlaStrip *strip, bool skip_transitions)
{
NlaStrip *next = strip->next;
while (next != nullptr) {
if (skip_transitions && (next->type == NLASTRIP_TYPE_TRANSITION)) {
next = next->next;
}
else {
return next;
}
}
return nullptr;
}
NlaStrip *BKE_nlastrip_prev_in_track(NlaStrip *strip, bool skip_transitions)
{
NlaStrip *prev = strip->prev;
while (prev != nullptr) {
if (skip_transitions && (prev->type == NLASTRIP_TYPE_TRANSITION)) {
prev = prev->prev;
}
else {
return prev;
}
}
return nullptr;
}
NlaStrip *BKE_nlastrip_find_active(NlaTrack *nlt)
{
if (nlt == nullptr) {
return nullptr;
}
return nlastrip_find_active(&nlt->strips);
}
void BKE_nlastrip_remove(ListBase *strips, NlaStrip *strip)
{
BLI_assert(strips);
BLI_remlink(strips, strip);
}
void BKE_nlastrip_remove_and_free(ListBase *strips, NlaStrip *strip, const bool do_id_user)
{
BKE_nlastrip_remove(strips, strip);
BKE_nlastrip_free(strip, do_id_user);
}
void BKE_nlastrip_set_active(AnimData *adt, NlaStrip *strip)
{
/* sanity checks */
if (adt == nullptr) {
return;
}
/* Loop over tracks, deactivating. */
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
LISTBASE_FOREACH (NlaStrip *, nls, &nlt->strips) {
if (nls != strip) {
nls->flag &= ~NLASTRIP_FLAG_ACTIVE;
}
else {
nls->flag |= NLASTRIP_FLAG_ACTIVE;
}
}
}
}
static NlaStrip *nlastrip_find_by_name(ListBase /*NlaStrip*/ *strips, const char *name)
{
LISTBASE_FOREACH (NlaStrip *, strip, strips) {
if (STREQ(strip->name, name)) {
return strip;
}
if (strip->type != NLASTRIP_TYPE_META) {
continue;
}
NlaStrip *inner_strip = nlastrip_find_by_name(&strip->strips, name);
if (inner_strip != nullptr) {
return inner_strip;
}
}
return nullptr;
}
NlaStrip *BKE_nlastrip_find_by_name(NlaTrack *nlt, const char *name)
{
return nlastrip_find_by_name(&nlt->strips, name);
}
bool BKE_nlastrip_within_bounds(NlaStrip *strip, float min, float max)
{
const float stripLen = (strip) ? strip->end - strip->start : 0.0f;
const float boundsLen = fabsf(max - min);
/* sanity checks */
if ((strip == nullptr) || IS_EQF(stripLen, 0.0f) || IS_EQF(boundsLen, 0.0f)) {
return false;
}
/* only ok if at least part of the strip is within the bounding window
* - first 2 cases cover when the strip length is less than the bounding area
* - second 2 cases cover when the strip length is greater than the bounding area
*/
if ((stripLen < boundsLen) &&
!(IN_RANGE(strip->start, min, max) || IN_RANGE(strip->end, min, max)))
{
return false;
}
if ((stripLen > boundsLen) &&
!(IN_RANGE(min, strip->start, strip->end) || IN_RANGE(max, strip->start, strip->end)))
{
return false;
}
/* should be ok! */
return true;
}
float BKE_nlastrip_distance_to_frame(const NlaStrip *strip, const float timeline_frame)
{
if (timeline_frame < strip->start) {
return strip->start - timeline_frame;
}
if (strip->end < timeline_frame) {
return timeline_frame - strip->end;
}
return 0.0f;
}
/* Ensure that strip doesn't overlap those around it after resizing
* by offsetting those which follow. */
static void nlastrip_fix_resize_overlaps(NlaStrip *strip)
{
/* next strips - do this first, since we're often just getting longer */
if (strip->next) {
NlaStrip *nls = strip->next;
float offset = 0.0f;
if (nls->type == NLASTRIP_TYPE_TRANSITION) {
/* transition strips should grow/shrink to accommodate the resized strip,
* but if the strip's bounds now exceed the transition, we're forced to
* offset everything to maintain the balance
*/
if (strip->end <= nls->start) {
/* grow the transition to fill the void */
nls->start = strip->end;
}
else if (strip->end < nls->end) {
/* shrink the transition to give the strip room */
nls->start = strip->end;
}
else {
/* Shrink transition down to 1 frame long (so that it can still be found),
* then offset everything else by the remaining deficit to give the strip room. */
nls->start = nls->end - 1.0f;
/* XXX: review whether preventing fractional values is good here... */
offset = ceilf(strip->end - nls->start);
/* apply necessary offset to ensure that the strip has enough space */
for (; nls; nls = nls->next) {
nls->start += offset;
nls->end += offset;
}
}
}
else if (strip->end > nls->start) {
/* NOTE: need to ensure we don't have a fractional frame offset, even if that leaves a gap,
* otherwise it will be very hard to get rid of later
*/
offset = ceilf(strip->end - nls->start);
/* apply to times of all strips in this direction */
for (; nls; nls = nls->next) {
nls->start += offset;
nls->end += offset;
}
}
}
/* previous strips - same routine as before */
/* NOTE: when strip bounds are recalculated, this is not considered! */
if (strip->prev) {
NlaStrip *nls = strip->prev;
float offset = 0.0f;
if (nls->type == NLASTRIP_TYPE_TRANSITION) {
/* transition strips should grow/shrink to accommodate the resized strip,
* but if the strip's bounds now exceed the transition, we're forced to
* offset everything to maintain the balance
*/
if (strip->start >= nls->end) {
/* grow the transition to fill the void */
nls->end = strip->start;
}
else if (strip->start > nls->start) {
/* shrink the transition to give the strip room */
nls->end = strip->start;
}
else {
/* Shrink transition down to 1 frame long (so that it can still be found),
* then offset everything else by the remaining deficit to give the strip room. */
nls->end = nls->start + 1.0f;
/* XXX: review whether preventing fractional values is good here... */
offset = ceilf(nls->end - strip->start);
/* apply necessary offset to ensure that the strip has enough space */
for (; nls; nls = nls->prev) {
nls->start -= offset;
nls->end -= offset;
}
}
}
else if (strip->start < nls->end) {
/* NOTE: need to ensure we don't have a fractional frame offset, even if that leaves a gap,
* otherwise it will be very hard to get rid of later
*/
offset = ceilf(nls->end - strip->start);
/* apply to times of all strips in this direction */
for (; nls; nls = nls->prev) {
nls->start -= offset;
nls->end -= offset;
}
}
}
}
void BKE_nlastrip_recalculate_bounds_sync_action(NlaStrip *strip)
{
float prev_actstart;
if (strip == nullptr || strip->type != NLASTRIP_TYPE_CLIP) {
return;
}
prev_actstart = strip->actstart;
const float2 frame_range = strip->act->wrap().get_frame_range_of_slot(strip->action_slot_handle);
strip->actstart = frame_range[0];
strip->actend = frame_range[1];
BKE_nla_clip_length_ensure_nonzero(&strip->actstart, &strip->actend);
/* Set start such that key's do not visually move, to preserve the overall animation result. */
strip->start += (strip->actstart - prev_actstart) * strip->scale;
BKE_nlastrip_recalculate_bounds(strip);
}
void BKE_nlastrip_recalculate_bounds(NlaStrip *strip)
{
float mapping;
/* sanity checks
* - must have a strip
* - can only be done for action clips
*/
if ((strip == nullptr) || (strip->type != NLASTRIP_TYPE_CLIP)) {
return;
}
/* calculate new length factors */
const float actlen = BKE_nla_clip_length_get_nonzero(strip);
mapping = strip->scale * strip->repeat;
/* adjust endpoint of strip in response to this */
if (IS_EQF(mapping, 0.0f) == 0) {
strip->end = (actlen * mapping) + strip->start;
}
/* make sure we don't overlap our neighbors */
nlastrip_fix_resize_overlaps(strip);
}
void BKE_nlastrip_recalculate_blend(NlaStrip *strip)
{
/* check if values need to be re-calculated. */
if (strip->blendin == 0 && strip->blendout == 0) {
return;
}
const double strip_len = strip->end - strip->start;
double blend_in = strip->blendin;
double blend_out = strip->blendout;
double blend_in_max = strip_len - blend_out;
CLAMP_MIN(blend_in_max, 0);
/* blend-out is limited to the length of the strip. */
CLAMP(blend_in, 0, blend_in_max);
CLAMP(blend_out, 0, strip_len - blend_in);
strip->blendin = blend_in;
strip->blendout = blend_out;
}
/* Animated Strips ------------------------------------------- */
bool BKE_nlatrack_has_animated_strips(NlaTrack *nlt)
{
/* sanity checks */
if (ELEM(nullptr, nlt, nlt->strips.first)) {
return false;
}
/* check each strip for F-Curves only (don't care about whether the flags are set) */
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
if (strip->fcurves.first) {
return true;
}
}
/* none found */
return false;
}
bool BKE_nlatracks_have_animated_strips(ListBase *tracks)
{
/* sanity checks */
if (ELEM(nullptr, tracks, tracks->first)) {
return false;
}
/* check each track, stopping on the first hit */
LISTBASE_FOREACH (NlaTrack *, nlt, tracks) {
if (BKE_nlatrack_has_animated_strips(nlt)) {
return true;
}
}
/* none found */
return false;
}
void BKE_nlastrip_validate_fcurves(NlaStrip *strip)
{
FCurve *fcu;
/* sanity checks */
if (strip == nullptr) {
return;
}
/* if controlling influence... */
if (strip->flag & NLASTRIP_FLAG_USR_INFLUENCE) {
/* try to get F-Curve */
fcu = BKE_fcurve_find(&strip->fcurves, "influence", 0);
/* add one if not found */
if (fcu == nullptr) {
/* make new F-Curve */
fcu = BKE_fcurve_create();
BLI_addtail(&strip->fcurves, fcu);
/* set default flags */
fcu->flag = (FCURVE_VISIBLE | FCURVE_SELECTED);
fcu->auto_smoothing = U.auto_smoothing_new;
/* store path - make copy, and store that */
fcu->rna_path = BLI_strdupn("influence", 9);
/* insert keyframe to ensure current value stays on first refresh */
fcu->bezt = MEM_callocN<BezTriple>("nlastrip influence bezt");
fcu->totvert = 1;
fcu->bezt->vec[1][0] = strip->start;
fcu->bezt->vec[1][1] = strip->influence;
/* Respect User Preferences for default interpolation and handles. */
fcu->bezt->h1 = fcu->bezt->h2 = U.keyhandles_new;
fcu->bezt->ipo = U.ipo_new;
}
}
/* if controlling time... */
if (strip->flag & NLASTRIP_FLAG_USR_TIME) {
/* try to get F-Curve */
fcu = BKE_fcurve_find(&strip->fcurves, "strip_time", 0);
/* add one if not found */
if (fcu == nullptr) {
/* make new F-Curve */
fcu = BKE_fcurve_create();
BLI_addtail(&strip->fcurves, fcu);
/* set default flags */
fcu->flag = (FCURVE_VISIBLE | FCURVE_SELECTED);
fcu->auto_smoothing = U.auto_smoothing_new;
/* store path - make copy, and store that */
fcu->rna_path = BLI_strdupn("strip_time", 10);
/* TODO: insert a few keyframes to ensure default behavior? */
}
}
}
bool BKE_nlastrip_controlcurve_remove(NlaStrip *strip, FCurve *fcurve)
{
if (STREQ(fcurve->rna_path, "strip_time")) {
strip->flag &= ~NLASTRIP_FLAG_USR_TIME;
}
else if (STREQ(fcurve->rna_path, "influence")) {
strip->flag &= ~NLASTRIP_FLAG_USR_INFLUENCE;
}
else {
return false;
}
BLI_remlink(&strip->fcurves, fcurve);
BKE_fcurve_free(fcurve);
return true;
}
bool BKE_nlastrip_has_curves_for_property(const PointerRNA *ptr, const PropertyRNA *prop)
{
/* sanity checks */
if (ELEM(nullptr, ptr, prop)) {
return false;
}
/* 1) Must be NLA strip */
if (ptr->type == &RNA_NlaStrip) {
/* 2) Must be one of the predefined properties */
static PropertyRNA *prop_influence = nullptr;
static PropertyRNA *prop_time = nullptr;
static bool needs_init = true;
/* Init the properties on first use */
if (needs_init) {
prop_influence = RNA_struct_type_find_property(&RNA_NlaStrip, "influence");
prop_time = RNA_struct_type_find_property(&RNA_NlaStrip, "strip_time");
needs_init = false;
}
/* Check if match */
if (ELEM(prop, prop_influence, prop_time)) {
return true;
}
}
/* No criteria met */
return false;
}
/* Sanity Validation ------------------------------------ */
void BKE_nlastrip_validate_name(AnimData *adt, NlaStrip *strip)
{
GHash *gh;
/* sanity checks */
if (ELEM(nullptr, adt, strip)) {
return;
}
/* give strip a default name if none already */
if (strip->name[0] == 0) {
switch (strip->type) {
case NLASTRIP_TYPE_CLIP: /* act-clip */
STRNCPY_UTF8(strip->name, (strip->act) ? (strip->act->id.name + 2) : DATA_("<No Action>"));
break;
case NLASTRIP_TYPE_TRANSITION: /* transition */
STRNCPY_UTF8(strip->name, DATA_("Transition"));
break;
case NLASTRIP_TYPE_META: /* meta */
STRNCPY_UTF8(strip->name, CTX_DATA_(BLT_I18NCONTEXT_ID_ACTION, "Meta"));
break;
default:
STRNCPY_UTF8(strip->name, DATA_("NLA Strip"));
break;
}
}
/* build a hash-table of all the strips in the tracks
* - this is easier than iterating over all the tracks+strips hierarchy every time
* (and probably faster)
*/
gh = BLI_ghash_str_new("nlastrip_validate_name gh");
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
LISTBASE_FOREACH (NlaStrip *, tstrip, &nlt->strips) {
/* don't add the strip of interest */
if (tstrip == strip) {
continue;
}
/* Use the name of the strip as the key, and the strip as the value,
* since we're mostly interested in the keys. */
BLI_ghash_insert(gh, tstrip->name, tstrip);
}
}
/* If the hash-table has a match for this name, try other names...
* - In an extreme case, it might not be able to find a name,
* but then everything else in Blender would fail too :).
*/
BLI_uniquename_cb(
[&](const blender::StringRefNull check_name) {
return BLI_ghash_haskey(gh, check_name.c_str());
},
DATA_("NlaStrip"),
'.',
strip->name,
sizeof(strip->name));
/* free the hash... */
BLI_ghash_free(gh, nullptr, nullptr);
}
/* ---- */
/* Get strips which overlap the given one at the start/end of its range
* - strip: strip that we're finding overlaps for
* - track: nla-track that the overlapping strips should be found from
* - start, end: frames for the offending endpoints
*/
static void nlastrip_get_endpoint_overlaps(NlaStrip *strip,
NlaTrack *track,
float **start,
float **end)
{
/* find strips that overlap over the start/end of the given strip,
* but which don't cover the entire length
*/
/* TODO: this scheme could get quite slow for doing this on many strips... */
LISTBASE_FOREACH (NlaStrip *, nls, &track->strips) {
/* Check if strip overlaps (extends over or exactly on)
* the entire range of the strip we're validating. */
if ((nls->start <= strip->start) && (nls->end >= strip->end)) {
*start = nullptr;
*end = nullptr;
return;
}
/* check if strip doesn't even occur anywhere near... */
if (nls->end < strip->start) {
continue; /* skip checking this strip... not worthy of mention */
}
if (nls->start > strip->end) {
return; /* the range we're after has already passed */
}
/* if this strip is not part of an island of continuous strips, it can be used
* - this check needs to be done for each end of the strip we try and use...
*/
if ((nls->next == nullptr) || IS_EQF(nls->next->start, nls->end) == 0) {
if ((nls->end > strip->start) && (nls->end < strip->end)) {
*start = &nls->end;
}
}
if ((nls->prev == nullptr) || IS_EQF(nls->prev->end, nls->start) == 0) {
if ((nls->start < strip->end) && (nls->start > strip->start)) {
*end = &nls->start;
}
}
}
}
/* Determine auto-blending for the given strip */
static void BKE_nlastrip_validate_autoblends(NlaTrack *nlt, NlaStrip *nls)
{
float *ps = nullptr, *pe = nullptr;
float *ns = nullptr, *ne = nullptr;
/* sanity checks */
if (ELEM(nullptr, nls, nlt)) {
return;
}
if ((nlt->prev == nullptr) && (nlt->next == nullptr)) {
return;
}
if ((nls->flag & NLASTRIP_FLAG_AUTO_BLENDS) == 0) {
return;
}
/* get test ranges */
if (nlt->prev) {
nlastrip_get_endpoint_overlaps(nls, nlt->prev, &ps, &pe);
}
if (nlt->next) {
nlastrip_get_endpoint_overlaps(nls, nlt->next, &ns, &ne);
}
/* set overlaps for this strip
* - don't use the values obtained though if the end in question
* is directly followed/preceded by another strip, forming an
* 'island' of continuous strips
*/
if ((ps || ns) && ((nls->prev == nullptr) || IS_EQF(nls->prev->end, nls->start) == 0)) {
/* start overlaps - pick the largest overlap */
if (((ps && ns) && (*ps > *ns)) || (ps)) {
nls->blendin = *ps - nls->start;
}
else {
nls->blendin = *ns - nls->start;
}
}
else { /* no overlap allowed/needed */
nls->blendin = 0.0f;
}
if ((pe || ne) && ((nls->next == nullptr) || IS_EQF(nls->next->start, nls->end) == 0)) {
/* end overlaps - pick the largest overlap */
if (((pe && ne) && (*pe > *ne)) || (pe)) {
nls->blendout = nls->end - *pe;
}
else {
nls->blendout = nls->end - *ne;
}
}
else { /* no overlap allowed/needed */
nls->blendout = 0.0f;
}
}
/**
* Ensure every transition's start/end properly set.
* Strip will be removed / freed if it doesn't fit (invalid).
* Return value indicates if passed strip is valid/fixed or invalid/removed.
*/
static bool nlastrip_validate_transition_start_end(ListBase *strips, NlaStrip *strip)
{
if (!(strip->type == NLASTRIP_TYPE_TRANSITION)) {
return true;
}
if (strip->prev) {
strip->start = strip->prev->end;
}
if (strip->next) {
strip->end = strip->next->start;
}
if (strip->start >= strip->end || strip->prev == nullptr || strip->next == nullptr) {
BKE_nlastrip_remove_and_free(strips, strip, true);
return false;
}
return true;
}
void BKE_nla_validate_state(AnimData *adt)
{
/* sanity checks */
if (ELEM(nullptr, adt, adt->nla_tracks.first)) {
return;
}
/* Adjust blending values for auto-blending,
* and also do an initial pass to find the earliest strip. */
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
LISTBASE_FOREACH_MUTABLE (NlaStrip *, strip, &nlt->strips) {
if (!nlastrip_validate_transition_start_end(&nlt->strips, strip)) {
printf(
"While moving NLA strips, a transition strip could no longer be applied to the new "
"positions and was removed.\n");
continue;
}
/* auto-blending first */
BKE_nlastrip_validate_autoblends(nlt, strip);
BKE_nlastrip_recalculate_blend(strip);
}
}
}
/* Action Stashing -------------------------------------- */
/* name of stashed tracks - the translation stuff is included here to save extra work */
#define STASH_TRACK_NAME DATA_("[Action Stash]")
bool BKE_nla_action_slot_is_stashed(AnimData *adt,
bAction *act,
const blender::animrig::slot_handle_t slot_handle)
{
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
if (strstr(nlt->name, STASH_TRACK_NAME)) {
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
if (strip->act == act && strip->action_slot_handle == slot_handle) {
return true;
}
}
}
}
return false;
}
bool BKE_nla_action_stash(const OwnedAnimData owned_adt, const bool is_liboverride)
{
NlaTrack *prev_track = nullptr;
NlaTrack *nlt;
NlaStrip *strip;
AnimData *adt = &owned_adt.adt;
/* sanity check */
if (ELEM(nullptr, adt, adt->action)) {
CLOG_ERROR(&LOG, "Invalid argument - %p %p", adt, adt->action);
return false;
}
/* do not add if it is already stashed */
if (BKE_nla_action_slot_is_stashed(adt, adt->action, adt->slot_handle)) {
return false;
}
/* create a new track, and add this immediately above the previous stashing track */
for (prev_track = static_cast<NlaTrack *>(adt->nla_tracks.last); prev_track;
prev_track = prev_track->prev)
{
if (strstr(prev_track->name, STASH_TRACK_NAME)) {
break;
}
}
nlt = BKE_nlatrack_new_after(&adt->nla_tracks, prev_track, is_liboverride);
BKE_nlatrack_set_active(&adt->nla_tracks, nlt);
BLI_assert(nlt != nullptr);
/* We need to ensure that if there wasn't any previous instance,
* it must go to be bottom of the stack. */
if (prev_track == nullptr) {
BLI_remlink(&adt->nla_tracks, nlt);
BLI_addhead(&adt->nla_tracks, nlt);
}
STRNCPY_UTF8(nlt->name, STASH_TRACK_NAME);
BLI_uniquename(
&adt->nla_tracks, nlt, STASH_TRACK_NAME, '.', offsetof(NlaTrack, name), sizeof(nlt->name));
/* add the action as a strip in this new track
* NOTE: a new user is created here
*/
strip = BKE_nlastrip_new_for_slot(adt->action, adt->slot_handle, owned_adt.owner_id);
BLI_assert(strip != nullptr);
BKE_nlatrack_add_strip(nlt, strip, is_liboverride);
BKE_nlastrip_validate_name(adt, strip);
/* mark the stash track and strip so that they doesn't disturb the stack animation,
* and are unlikely to draw attention to itself (or be accidentally bumped around)
*
* NOTE: this must be done *after* adding the strip to the track, or else
* the strip locking will prevent the strip from getting added
*/
nlt->flag |= (NLATRACK_MUTED | NLATRACK_PROTECTED);
strip->flag &= ~(NLASTRIP_FLAG_SELECT | NLASTRIP_FLAG_ACTIVE);
/* also mark the strip for auto syncing the length, so that the strips accurately
* reflect the length of the action
* XXX: we could do with some extra flags here to prevent repeats/scaling options from working!
*/
strip->flag |= NLASTRIP_FLAG_SYNC_LENGTH;
/* succeeded */
return true;
}
/* Core Tools ------------------------------------------- */
void BKE_nla_action_pushdown(const OwnedAnimData owned_adt, const bool is_liboverride)
{
NlaStrip *strip;
AnimData *adt = &owned_adt.adt;
/* sanity checks */
/* TODO: need to report the error for this */
if (ELEM(nullptr, adt, adt->action)) {
return;
}
/* Add a new NLA strip to the track, which references the active action + slot. */
strip = BKE_nlastack_add_strip(owned_adt, is_liboverride);
if (strip == nullptr) {
return;
}
/* Clear reference to action now that we've pushed it onto the stack. */
const bool unassign_ok = animrig::unassign_action(owned_adt.owner_id);
BLI_assert_msg(unassign_ok,
"Expecting un-assigning an action to always work when pushing down an NLA strip");
UNUSED_VARS_NDEBUG(unassign_ok);
/* copy current "action blending" settings from adt to the strip,
* as it was keyframed with these settings, so omitting them will
* change the effect [#54233]. */
strip->blendmode = adt->act_blendmode;
strip->influence = adt->act_influence;
strip->extendmode = adt->act_extendmode;
if (adt->act_influence < 1.0f) {
/* enable "user-controlled" influence (which will insert a default keyframe)
* so that the influence doesn't get lost on the new update
*
* NOTE: An alternative way would have been to instead hack the influence
* to not get always get reset to full strength if NLASTRIP_FLAG_USR_INFLUENCE
* is disabled but auto-blending isn't being used. However, that approach
* is a bit hacky/hard to discover, and may cause backwards compatibility issues,
* so it's better to just do it this way.
*/
strip->flag |= NLASTRIP_FLAG_USR_INFLUENCE;
BKE_nlastrip_validate_fcurves(strip);
}
/* make strip the active one... */
BKE_nlastrip_set_active(adt, strip);
}
static void nla_tweakmode_find_active(const ListBase /*NlaTrack*/ *nla_tracks,
NlaTrack **r_track_of_active_strip,
NlaStrip **r_active_strip)
{
NlaTrack *activeTrack = nullptr;
NlaStrip *activeStrip = nullptr;
/* go over the tracks, finding the active one, and its active strip
* - if we cannot find both, then there's nothing to do
*/
LISTBASE_FOREACH (NlaTrack *, nlt, nla_tracks) {
/* check if active */
if (nlt->flag & NLATRACK_ACTIVE) {
/* store reference to this active track */
activeTrack = nlt;
/* now try to find active strip */
activeStrip = BKE_nlastrip_find_active(nlt);
break;
}
}
/* There are situations where we may have multiple strips selected and we want to enter
* tweak-mode on all of those at once. Usually in those cases,
* it will usually just be a single strip per AnimData.
* In such cases, compromise and take the last selected track and/or last selected strip,
* #28468.
*/
if (activeTrack == nullptr) {
/* try last selected track for active strip */
LISTBASE_FOREACH_BACKWARD (NlaTrack *, nlt, nla_tracks) {
if (nlt->flag & NLATRACK_SELECTED) {
/* assume this is the active track */
activeTrack = nlt;
/* try to find active strip */
activeStrip = BKE_nlastrip_find_active(nlt);
break;
}
}
}
if ((activeTrack) && (activeStrip == nullptr)) {
/* No active strip in active or last selected track;
* compromise for first selected (assuming only single). */
LISTBASE_FOREACH (NlaStrip *, strip, &activeTrack->strips) {
if (strip->flag & (NLASTRIP_FLAG_SELECT | NLASTRIP_FLAG_ACTIVE)) {
activeStrip = strip;
break;
}
}
}
*r_track_of_active_strip = activeTrack;
*r_active_strip = activeStrip;
}
bool BKE_nla_tweakmode_enter(const OwnedAnimData owned_adt)
{
NlaTrack *activeTrack = nullptr;
NlaStrip *activeStrip = nullptr;
AnimData &adt = owned_adt.adt;
/* verify that data is valid */
if (ELEM(nullptr, adt.nla_tracks.first)) {
return false;
}
/* If block is already in tweak-mode, just leave, but we should report
* that this block is in tweak-mode (as our return-code). */
if (adt.flag & ADT_NLA_EDIT_ON) {
return true;
}
nla_tweakmode_find_active(&adt.nla_tracks, &activeTrack, &activeStrip);
if (ELEM(nullptr, activeTrack, activeStrip, activeStrip->act)) {
if (G.debug & G_DEBUG) {
printf("NLA tweak-mode enter - neither active requirement found\n");
printf("\tactiveTrack = %p, activeStrip = %p\n", (void *)activeTrack, (void *)activeStrip);
}
return false;
}
/* Go over all the tracks, tagging each strip that uses the same
* action as the active strip, but leaving everything else alone.
*/
LISTBASE_FOREACH (NlaTrack *, nlt, &adt.nla_tracks) {
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
if (strip->act == activeStrip->act) {
strip->flag |= NLASTRIP_FLAG_TWEAKUSER;
}
else {
strip->flag &= ~NLASTRIP_FLAG_TWEAKUSER;
}
}
}
/* Untag tweaked track. This leads to non tweaked actions being drawn differently than the
* tweaked action. */
activeStrip->flag &= ~NLASTRIP_FLAG_TWEAKUSER;
/* go over all the tracks after AND INCLUDING the active one, tagging them as being disabled
* - the active track needs to also be tagged, otherwise, it'll overlap with the tweaks going
* on.
*/
activeTrack->flag |= NLATRACK_DISABLED;
if ((adt.flag & ADT_NLA_EVAL_UPPER_TRACKS) == 0) {
for (NlaTrack *nlt = activeTrack->next; nlt; nlt = nlt->next) {
nlt->flag |= NLATRACK_DISABLED;
}
}
/* Remember which Action + Slot was previously assigned. This will be stored in adt.tmpact and
* related properties further down. This doesn't manipulate `adt.tmpact` quite yet, so that the
* assignment of the tweaked NLA strip's Action can happen normally (we're not in tweak mode
* yet). */
bAction *prev_action = adt.action;
const animrig::slot_handle_t prev_slot_handle = adt.slot_handle;
if (activeStrip->act) {
animrig::Action &strip_action = activeStrip->act->wrap();
if (strip_action.is_action_layered()) {
animrig::Slot *strip_slot = strip_action.slot_for_handle(activeStrip->action_slot_handle);
if (animrig::assign_action_and_slot(&strip_action, strip_slot, owned_adt.owner_id) !=
animrig::ActionSlotAssignmentResult::OK)
{
printf("NLA tweak-mode enter - could not assign slot %s\n",
strip_slot ? strip_slot->identifier : "-unassigned-");
/* There is one other reason this could fail: when already in NLA tweak mode. But since
* we're here in the code, the ADT_NLA_EDIT_ON flag is not yet set, and thus that shouldn't
* be the case.
*
* Because this ADT is not in tweak mode, it means that the Action assignment will have
* succeeded (I know, too much coupling here, would be better to have another
* SlotAssignmentResult value for this). */
}
}
else {
adt.action = activeStrip->act;
id_us_plus(&adt.action->id);
}
}
else {
/* This is a strange situation to be in, as every *tweak-able* NLA strip should have an Action.
*/
BLI_assert_unreachable();
const bool unassign_ok = animrig::unassign_action(owned_adt);
BLI_assert_msg(unassign_ok,
"Expecting un-assigning the Action to work (while entering NLA tweak mode)");
UNUSED_VARS_NDEBUG(unassign_ok);
}
/* Actually set those properties that make this 'NLA Tweak Mode'. */
const animrig::ActionSlotAssignmentResult result = animrig::assign_tmpaction_and_slot_handle(
prev_action, prev_slot_handle, owned_adt);
BLI_assert_msg(
result == animrig::ActionSlotAssignmentResult::OK,
"Expecting the Action+Slot of an NLA strip to be suitable for direct assignment as well");
UNUSED_VARS_NDEBUG(result);
adt.act_track = activeTrack;
adt.actstrip = activeStrip;
adt.flag |= ADT_NLA_EDIT_ON;
return true;
}
static bool is_nla_in_tweakmode(AnimData *adt)
{
return adt && (adt->flag & ADT_NLA_EDIT_ON);
}
static void nla_tweakmode_exit_sync_strip_lengths(AnimData *adt)
{
NlaStrip *active_strip = adt->actstrip;
if (!active_strip || !active_strip->act) {
return;
}
/* sync the length of the user-strip with the new state of the action
* but only if the user has explicitly asked for this to happen
* (see #34645 for things to be careful about)
*/
if (active_strip->flag & NLASTRIP_FLAG_SYNC_LENGTH) {
BKE_nlastrip_recalculate_bounds_sync_action(active_strip);
}
/* Sync strip extents of strips using the same Action as the tweaked strip. */
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
if ((strip->flag & NLASTRIP_FLAG_SYNC_LENGTH) && active_strip->act == strip->act) {
BKE_nlastrip_recalculate_bounds_sync_action(strip);
}
}
}
}
/**
* Perform that part of the "exit NLA tweak mode" that does not need to follow
* any pointers to other data-blocks.
*/
static void nla_tweakmode_exit_nofollowptr(AnimData *adt)
{
BKE_nla_tweakmode_clear_flags(adt);
adt->action = adt->tmpact;
adt->slot_handle = adt->tmp_slot_handle;
adt->tmpact = nullptr;
adt->tmp_slot_handle = animrig::Slot::unassigned;
STRNCPY_UTF8(adt->last_slot_identifier, adt->tmp_last_slot_identifier);
adt->act_track = nullptr;
adt->actstrip = nullptr;
}
void BKE_nla_tweakmode_exit_nofollowptr(AnimData *adt)
{
if (!is_nla_in_tweakmode(adt)) {
return;
}
nla_tweakmode_exit_nofollowptr(adt);
}
void BKE_nla_tweakmode_exit(const OwnedAnimData owned_adt)
{
if (!is_nla_in_tweakmode(&owned_adt.adt)) {
return;
}
if (owned_adt.adt.action) {
/* When a strip has no slot assigned, it can still enter tweak mode. Inserting a key will then
* create a slot. When exiting tweak mode, this slot has to be assigned to the strip.
* At this moment in time, the adt->action is still the one being tweaked. */
NlaStrip *active_strip = owned_adt.adt.actstrip;
if (active_strip && active_strip->action_slot_handle != owned_adt.adt.slot_handle) {
const animrig::ActionSlotAssignmentResult result = animrig::nla::assign_action_slot_handle(
*active_strip, owned_adt.adt.slot_handle, owned_adt.owner_id);
BLI_assert_msg(result == animrig::ActionSlotAssignmentResult::OK,
"When exiting tweak mode, syncing the tweaked Action slot should work");
UNUSED_VARS_NDEBUG(result);
}
/* The Action will be replaced with adt->tmpact, and thus needs to be unassigned first. */
/* The high-level function animrig::unassign_action() will check whether NLA tweak mode is
* enabled, and if so, refuse to work (and rightfully so). However, exiting tweak mode is not
* just setting a flag (see BKE_animdata_action_editable(), it checks for the tmpact pointer as
* well). Because of that, here we call the low-level generic assignment function, to
* circumvent that check and unconditionally unassign the tweaked Action. */
const bool unassign_ok = animrig::generic_assign_action(owned_adt.owner_id,
nullptr,
owned_adt.adt.action,
owned_adt.adt.slot_handle,
owned_adt.adt.last_slot_identifier);
BLI_assert_msg(unassign_ok,
"When exiting tweak mode, unassigning the tweaked Action should work");
UNUSED_VARS_NDEBUG(unassign_ok);
}
nla_tweakmode_exit_sync_strip_lengths(&owned_adt.adt);
nla_tweakmode_exit_nofollowptr(&owned_adt.adt);
/* nla_tweakmode_exit_nofollowptr() does not follow any pointers, and thus cannot update the slot
* user map. So it has to be done here now. This is safe to do here, as slot->users_add()
* gracefully handles duplicates. */
if (owned_adt.adt.action && owned_adt.adt.slot_handle != animrig::Slot::unassigned) {
animrig::Action &action = owned_adt.adt.action->wrap();
BLI_assert_msg(action.is_action_layered(),
"when a slot is assigned, the action should layered");
animrig::Slot *slot = action.slot_for_handle(owned_adt.adt.slot_handle);
if (slot) {
slot->users_add(owned_adt.owner_id);
}
}
}
void BKE_nla_tweakmode_clear_flags(AnimData *adt)
{
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
nlt->flag &= ~NLATRACK_DISABLED;
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
strip->flag &= ~NLASTRIP_FLAG_TWEAKUSER;
}
}
adt->flag &= ~ADT_NLA_EDIT_ON;
}
void BKE_nla_debug_print_flags(AnimData *adt, ID *owner_id)
{
if (!adt) {
adt = BKE_animdata_from_id(owner_id);
}
printf("\033[38;5;214mNLA state");
if (owner_id) {
printf(" for %s", owner_id->name);
}
printf("\033[0m\n");
if (!adt) {
printf(" - ADT is nil!\n");
return;
}
printf(" - ADT flags:");
if (adt->flag & ADT_NLA_SOLO_TRACK) {
printf(" SOLO_TRACK");
}
if (adt->flag & ADT_NLA_EVAL_OFF) {
printf(" EVAL_OFF");
}
if (adt->flag & ADT_NLA_EDIT_ON) {
printf(" EDIT_ON");
}
if (adt->flag & ADT_NLA_EDIT_NOMAP) {
printf(" EDIT_NOMAP");
}
if (adt->flag & ADT_NLA_SKEYS_COLLAPSED) {
printf(" SKEYS_COLLAPSED");
}
if (adt->flag & ADT_NLA_EVAL_UPPER_TRACKS) {
printf(" EVAL_UPPER_TRACKS");
}
if ((adt->flag & (ADT_NLA_SOLO_TRACK | ADT_NLA_EVAL_OFF | ADT_NLA_EDIT_ON | ADT_NLA_EDIT_NOMAP |
ADT_NLA_SKEYS_COLLAPSED | ADT_NLA_EVAL_UPPER_TRACKS)) == 0)
{
printf(" -");
}
printf("\n");
if (BLI_listbase_is_empty(&adt->nla_tracks)) {
printf(" - No tracks\n");
return;
}
printf(" - Active track: %s (#%d)\n",
adt->act_track ? adt->act_track->name : "-nil-",
adt->act_track ? adt->act_track->index : 0);
printf(" - Active strip: %s\n", adt->actstrip ? adt->actstrip->name : "-nil-");
LISTBASE_FOREACH (NlaTrack *, nlt, &adt->nla_tracks) {
printf(" - Track #%d %s: ", nlt->index, nlt->name);
if (nlt->flag & NLATRACK_ACTIVE) {
printf("ACTIVE ");
}
if (nlt->flag & NLATRACK_SELECTED) {
printf("SELECTED ");
}
if (nlt->flag & NLATRACK_MUTED) {
printf("MUTED ");
}
if (nlt->flag & NLATRACK_SOLO) {
printf("SOLO ");
}
if (nlt->flag & NLATRACK_PROTECTED) {
printf("PROTECTED ");
}
if (nlt->flag & NLATRACK_DISABLED) {
printf("DISABLED ");
}
if (nlt->flag & NLATRACK_TEMPORARILY_ADDED) {
printf("TEMPORARILY_ADDED ");
}
if (nlt->flag & NLATRACK_OVERRIDELIBRARY_LOCAL) {
printf("OVERRIDELIBRARY_LOCAL ");
}
printf("\n");
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
printf(" - Strip %s: ", strip->name);
if (strip->flag & NLASTRIP_FLAG_ACTIVE) {
printf("ACTIVE ");
}
if (strip->flag & NLASTRIP_FLAG_SELECT) {
printf("SELECT ");
}
if (strip->flag & NLASTRIP_FLAG_TWEAKUSER) {
printf("TWEAKUSER ");
}
if (strip->flag & NLASTRIP_FLAG_USR_INFLUENCE) {
printf("USR_INFLUENCE ");
}
if (strip->flag & NLASTRIP_FLAG_USR_TIME) {
printf("USR_TIME ");
}
if (strip->flag & NLASTRIP_FLAG_USR_TIME_CYCLIC) {
printf("USR_TIME_CYCLIC ");
}
if (strip->flag & NLASTRIP_FLAG_SYNC_LENGTH) {
printf("SYNC_LENGTH ");
}
if (strip->flag & NLASTRIP_FLAG_AUTO_BLENDS) {
printf("AUTO_BLENDS ");
}
if (strip->flag & NLASTRIP_FLAG_REVERSE) {
printf("REVERSE ");
}
if (strip->flag & NLASTRIP_FLAG_MUTED) {
printf("MUTED ");
}
if (strip->flag & NLASTRIP_FLAG_INVALID_LOCATION) {
printf("INVALID_LOCATION ");
}
if (strip->flag & NLASTRIP_FLAG_NO_TIME_MAP) {
printf("NO_TIME_MAP ");
}
if (strip->flag & NLASTRIP_FLAG_TEMP_META) {
printf("TEMP_META ");
}
if (strip->flag & NLASTRIP_FLAG_EDIT_TOUCHED) {
printf("EDIT_TOUCHED ");
}
printf("\n");
}
}
}
static void blend_write_nla_strips(BlendWriter *writer, ListBase *strips)
{
BLO_write_struct_list(writer, NlaStrip, strips);
LISTBASE_FOREACH (NlaStrip *, strip, strips) {
/* write the strip's F-Curves and modifiers */
BKE_fcurve_blend_write_listbase(writer, &strip->fcurves);
BKE_fmodifiers_blend_write(writer, &strip->modifiers);
/* write the strip's children */
blend_write_nla_strips(writer, &strip->strips);
}
}
static void blend_data_read_nla_strips(BlendDataReader *reader, ListBase *strips)
{
LISTBASE_FOREACH (NlaStrip *, strip, strips) {
/* strip's child strips */
BLO_read_struct_list(reader, NlaStrip, &strip->strips);
blend_data_read_nla_strips(reader, &strip->strips);
/* strip's F-Curves */
BLO_read_struct_list(reader, FCurve, &strip->fcurves);
BKE_fcurve_blend_read_data_listbase(reader, &strip->fcurves);
/* strip's F-Modifiers */
BLO_read_struct_list(reader, FModifier, &strip->modifiers);
BKE_fmodifiers_blend_read_data(reader, &strip->modifiers, nullptr);
}
}
void BKE_nla_blend_write(BlendWriter *writer, ListBase *tracks)
{
/* write all the tracks */
LISTBASE_FOREACH (NlaTrack *, nlt, tracks) {
/* write the track first */
BLO_write_struct(writer, NlaTrack, nlt);
/* write the track's strips */
blend_write_nla_strips(writer, &nlt->strips);
}
}
void BKE_nla_blend_read_data(BlendDataReader *reader, ID *id_owner, ListBase *tracks)
{
LISTBASE_FOREACH (NlaTrack *, nlt, tracks) {
/* If linking from a library, clear 'local' library override flag. */
if (ID_IS_LINKED(id_owner)) {
nlt->flag &= ~NLATRACK_OVERRIDELIBRARY_LOCAL;
}
/* relink list of strips */
BLO_read_struct_list(reader, NlaStrip, &nlt->strips);
/* relink strip data */
blend_data_read_nla_strips(reader, &nlt->strips);
}
}
void BKE_nla_liboverride_post_process(ID *id, AnimData *adt)
{
/* TODO(Sybren): replace these two parameters with an OwnedAnimData struct. */
BLI_assert(id);
BLI_assert(adt);
/* The 'id' parameter is unused as it's not necessary here, but still can be useful when
* debugging. And with the NLA complexity the way it is, debugging comfort is kinda nice. */
UNUSED_VARS_NDEBUG(id);
const bool is_tweak_mode = (adt->flag & ADT_NLA_EDIT_ON);
const bool has_tracks = !BLI_listbase_is_empty(&adt->nla_tracks);
if (!has_tracks) {
if (is_tweak_mode) {
/* No tracks, so it's impossible to actually be in tweak mode. */
BKE_nla_tweakmode_exit({*id, *adt});
}
return;
}
if (!is_tweak_mode) {
/* Library-linked data should already have been cleared of NLA Tweak Mode flags, and the
* overrides also didn't set tweak mode enabled. Assuming here that the override-added
* tracks/strips are consistent with the override's tweak mode flag. */
return;
}
/* In tweak mode, with tracks, so ensure that the active track/strip pointers are correct. Since
* these pointers may come from a library, but the override may have added other tracks and
* strips (one of which is in tweak mode), always look up the current pointer values. */
nla_tweakmode_find_active(&adt->nla_tracks, &adt->act_track, &adt->actstrip);
if (!adt->act_track || !adt->actstrip) {
/* Could not find the active track/strip, so better to exit tweak mode. */
BKE_nla_tweakmode_exit({*id, *adt});
}
}
static bool visit_strip(NlaStrip *strip, blender::FunctionRef<bool(NlaStrip *)> callback)
{
if (!callback(strip)) {
return false;
}
/* Recurse into sub-strips. */
LISTBASE_FOREACH (NlaStrip *, sub_strip, &strip->strips) {
if (!visit_strip(sub_strip, callback)) {
return false;
}
}
return true;
}
namespace blender::bke::nla {
bool foreach_strip(ID *id, blender::FunctionRef<bool(NlaStrip *)> callback)
{
const AnimData *adt = BKE_animdata_from_id(id);
if (!adt) {
/* Having no NLA trivially means that we've looped through all the strips. */
return true;
}
return foreach_strip_adt(*adt, callback);
}
bool foreach_strip_adt(const AnimData &adt, blender::FunctionRef<bool(NlaStrip *)> callback)
{
LISTBASE_FOREACH (NlaTrack *, nlt, &adt.nla_tracks) {
LISTBASE_FOREACH (NlaStrip *, strip, &nlt->strips) {
if (!visit_strip(strip, callback)) {
return false;
}
}
}
return true;
}
} // namespace blender::bke::nla