tornavis/source/blender/editors/transform/transform_snap_animation.c

137 lines
3.4 KiB
C

/* SPDX-License-Identifier: GPL-2.0-or-later
* Copyright 2001-2002 NaN Holding BV. All rights reserved. */
/** \file
* \ingroup edtransform
*/
#include "DNA_anim_types.h"
#include "BLI_math.h"
#include "BKE_context.h"
#include "BKE_nla.h"
#include "ED_markers.h"
#include "ED_screen.h"
#include "transform.h"
#include "transform_snap.h"
/* -------------------------------------------------------------------- */
/** \name Snapping in Anim Editors
* \{ */
short getAnimEdit_SnapMode(TransInfo *t)
{
short autosnap = SACTSNAP_OFF;
if (t->spacetype == SPACE_ACTION) {
SpaceAction *saction = (SpaceAction *)t->area->spacedata.first;
if (saction) {
autosnap = saction->autosnap;
}
}
else if (t->spacetype == SPACE_GRAPH) {
if ((t->mode == TFM_TRANSLATION) && transform_snap_is_active(t)) {
return autosnap;
}
SpaceGraph *sipo = (SpaceGraph *)t->area->spacedata.first;
if (sipo) {
autosnap = sipo->autosnap;
}
}
else if (t->spacetype == SPACE_NLA) {
SpaceNla *snla = (SpaceNla *)t->area->spacedata.first;
if (snla) {
autosnap = snla->autosnap;
}
}
else {
autosnap = SACTSNAP_OFF;
}
/* toggle autosnap on/off
* - when toggling on, prefer nearest frame over 1.0 frame increments
*/
if (t->modifiers & MOD_SNAP_INVERT) {
if (autosnap) {
autosnap = SACTSNAP_OFF;
}
else {
autosnap = SACTSNAP_FRAME;
}
}
return autosnap;
}
void snapFrameTransform(TransInfo *t,
const eAnimEdit_AutoSnap autosnap,
const float val_initial,
const float val_final,
float *r_val_final)
{
float deltax = val_final - val_initial;
switch (autosnap) {
case SACTSNAP_FRAME:
*r_val_final = floorf(val_final + 0.5f);
break;
case SACTSNAP_MARKER:
/* Snap to nearest marker. */
/* TODO: need some more careful checks for where data comes from. */
*r_val_final = (float)ED_markers_find_nearest_marker_time(&t->scene->markers, val_final);
break;
case SACTSNAP_SECOND:
case SACTSNAP_TSTEP: {
const Scene *scene = t->scene;
const double secf = FPS;
if (autosnap == SACTSNAP_SECOND) {
*r_val_final = floorf((val_final / secf) + 0.5) * secf;
}
else {
deltax = (float)(floor((deltax / secf) + 0.5) * secf);
*r_val_final = val_initial + deltax;
}
break;
}
case SACTSNAP_STEP:
deltax = floorf(deltax + 0.5f);
*r_val_final = val_initial + deltax;
break;
case SACTSNAP_OFF:
break;
}
}
void transform_snap_anim_flush_data(TransInfo *t,
TransData *td,
const eAnimEdit_AutoSnap autosnap,
float *r_val_final)
{
BLI_assert(autosnap != SACTSNAP_OFF);
float val = td->loc[0];
float ival = td->iloc[0];
AnimData *adt = !ELEM(t->spacetype, SPACE_NLA, SPACE_SEQ) ? td->extra : NULL;
/* Convert frame to nla-action time (if needed) */
if (adt) {
val = BKE_nla_tweakedit_remap(adt, val, NLATIME_CONVERT_MAP);
ival = BKE_nla_tweakedit_remap(adt, ival, NLATIME_CONVERT_MAP);
}
snapFrameTransform(t, autosnap, ival, val, &val);
/* Convert frame out of nla-action time. */
if (adt) {
val = BKE_nla_tweakedit_remap(adt, val, NLATIME_CONVERT_UNMAP);
}
*r_val_final = val;
}
/** \} */