Convert src/runtime/motion/spring.ts to JavaScript

pull/8569/head
S. Elliott Johnson 3 years ago
parent e798760aa0
commit 4aa44b7298

@ -1,15 +1,13 @@
import { Readable, writable } from '../store'; import { writable } from '../store';
import { loop, now, Task } from '../internal'; import { loop, now } from '../internal';
import { is_date } from './utils'; import { is_date } from './utils';
/** @param {TickContext<T>} ctx
interface TickContext<T> { * @param {T} last_value
inv_mass: number; * @param {T} current_value
dt: number; * @param {T} target_value
opts: Spring<T>; * @returns {T}
settled: boolean; */
} function tick_spring(ctx, last_value, current_value, target_value) {
function tick_spring<T>(ctx: TickContext<T>, last_value: T, current_value: T, target_value: T): T {
if (typeof current_value === 'number' || is_date(current_value)) { if (typeof current_value === 'number' || is_date(current_value)) {
// @ts-ignore // @ts-ignore
const delta = target_value - current_value; const delta = target_value - current_value;
@ -19,20 +17,20 @@ function tick_spring<T>(ctx: TickContext<T>, last_value: T, current_value: T, ta
const damper = ctx.opts.damping * velocity; const damper = ctx.opts.damping * velocity;
const acceleration = (spring - damper) * ctx.inv_mass; const acceleration = (spring - damper) * ctx.inv_mass;
const d = (velocity + acceleration) * ctx.dt; const d = (velocity + acceleration) * ctx.dt;
if (Math.abs(d) < ctx.opts.precision && Math.abs(delta) < ctx.opts.precision) { if (Math.abs(d) < ctx.opts.precision && Math.abs(delta) < ctx.opts.precision) {
return target_value; // settled return target_value; // settled
} else { }
else {
ctx.settled = false; // signal loop to keep ticking ctx.settled = false; // signal loop to keep ticking
// @ts-ignore // @ts-ignore
return is_date(current_value) ? new Date(current_value.getTime() + d) : current_value + d; return is_date(current_value) ? new Date(current_value.getTime() + d) : current_value + d;
} }
} else if (Array.isArray(current_value)) { }
else if (Array.isArray(current_value)) {
// @ts-ignore // @ts-ignore
return current_value.map((_, i) => return current_value.map((_, i) => tick_spring(ctx, last_value[i], current_value[i], target_value[i]));
tick_spring(ctx, last_value[i], current_value[i], target_value[i]) }
); else if (typeof current_value === 'object') {
} else if (typeof current_value === 'object') {
const next_value = {}; const next_value = {};
for (const k in current_value) { for (const k in current_value) {
// @ts-ignore // @ts-ignore
@ -40,109 +38,120 @@ function tick_spring<T>(ctx: TickContext<T>, last_value: T, current_value: T, ta
} }
// @ts-ignore // @ts-ignore
return next_value; return next_value;
} else { }
else {
throw new Error(`Cannot spring ${typeof current_value} values`); throw new Error(`Cannot spring ${typeof current_value} values`);
} }
} }
/** @param {T} value
interface SpringOpts { * @param {SpringOpts} opts
stiffness?: number; * @returns {import("/Users/elliottjohnson/dev/sveltejs/svelte/spring.ts-to-jsdoc").Spring<T>}
damping?: number; */
precision?: number; export function spring(value, opts = {}) {
}
interface SpringUpdateOpts {
hard?: any;
soft?: string | number | boolean;
}
type Updater<T> = (target_value: T, value: T) => T;
export interface Spring<T> extends Readable<T> {
set: (new_value: T, opts?: SpringUpdateOpts) => Promise<void>;
update: (fn: Updater<T>, opts?: SpringUpdateOpts) => Promise<void>;
precision: number;
damping: number;
stiffness: number;
}
export function spring<T = any>(value?: T, opts: SpringOpts = {}): Spring<T> {
const store = writable(value); const store = writable(value);
const { stiffness = 0.15, damping = 0.8, precision = 0.01 } = opts; const { stiffness = 0.15, damping = 0.8, precision = 0.01 } = opts;
/** @type {number} */
let last_time: number; let last_time;
let task: Task; /** @type {Task} */
let current_token: object; let task;
let last_value: T = value; /** @type {object} */
let target_value: T = value; let current_token;
/** @type {T} */
let last_value = value;
/** @type {T} */
let target_value = value;
let inv_mass = 1; let inv_mass = 1;
let inv_mass_recovery_rate = 0; let inv_mass_recovery_rate = 0;
let cancel_task = false; let cancel_task = false;
/** @param {T} new_value
function set(new_value: T, opts: SpringUpdateOpts = {}): Promise<void> { * @param {SpringUpdateOpts} opts
* @returns {Promise<void>}
*/
function set(new_value, opts = {}) {
target_value = new_value; target_value = new_value;
const token = (current_token = {}); const token = (current_token = {});
if (value == null || opts.hard || (spring.stiffness >= 1 && spring.damping >= 1)) { if (value == null || opts.hard || (spring.stiffness >= 1 && spring.damping >= 1)) {
cancel_task = true; // cancel any running animation cancel_task = true; // cancel any running animation
last_time = now(); last_time = now();
last_value = new_value; last_value = new_value;
store.set((value = target_value)); store.set((value = target_value));
return Promise.resolve(); return Promise.resolve();
} else if (opts.soft) { }
else if (opts.soft) {
const rate = opts.soft === true ? 0.5 : +opts.soft; const rate = opts.soft === true ? 0.5 : +opts.soft;
inv_mass_recovery_rate = 1 / (rate * 60); inv_mass_recovery_rate = 1 / (rate * 60);
inv_mass = 0; // infinite mass, unaffected by spring forces inv_mass = 0; // infinite mass, unaffected by spring forces
} }
if (!task) { if (!task) {
last_time = now(); last_time = now();
cancel_task = false; cancel_task = false;
task = loop((now) => { task = loop((now) => {
if (cancel_task) { if (cancel_task) {
cancel_task = false; cancel_task = false;
task = null; task = null;
return false; return false;
} }
inv_mass = Math.min(inv_mass + inv_mass_recovery_rate, 1); inv_mass = Math.min(inv_mass + inv_mass_recovery_rate, 1);
const ctx = {
const ctx: TickContext<T> = {
inv_mass, inv_mass,
opts: spring, opts: spring,
settled: true, // tick_spring may signal false settled: true,
dt: ((now - last_time) * 60) / 1000 dt: ((now - last_time) * 60) / 1000
}; };
const next_value = tick_spring(ctx, last_value, value, target_value); const next_value = tick_spring(ctx, last_value, value, target_value);
last_time = now; last_time = now;
last_value = value; last_value = value;
store.set((value = next_value)); store.set((value = next_value));
if (ctx.settled) { if (ctx.settled) {
task = null; task = null;
} }
return !ctx.settled; return !ctx.settled;
}); });
} }
return new Promise((fulfil) => { return new Promise((fulfil) => {
task.promise.then(() => { task.promise.then(() => {
if (token === current_token) fulfil(); if (token === current_token)
fulfil();
}); });
}); });
} }
/** @type {Spring<T>} */
const spring: Spring<T> = { const spring = {
set, set,
update: (fn, opts: SpringUpdateOpts) => set(fn(target_value, value), opts), update: (fn, opts) => set(fn(target_value, value), opts),
subscribe: store.subscribe, subscribe: store.subscribe,
stiffness, stiffness,
damping, damping,
precision precision
}; };
return spring; return spring;
} }
/**
* @typedef {(target_value: T, value: T) => T} Updater
* @template T
*/
/** @typedef {Object} TickContext
* @property {number} inv_mass
* @property {number} dt
* @property {Spring<T>} opts
* @property {boolean} settled
*/
/** @typedef {Object} SpringOpts
* @property {number} [stiffness]
* @property {number} [damping]
* @property {number} [precision]
*/
/** @typedef {Object} SpringUpdateOpts
* @property {any} [hard]
* @property {string|number|boolean} [soft]
*/
/** @typedef {Object} Spring
* @property {(new_value:T,opts?:SpringUpdateOpts)=>Promise<void>} set
* @property {(fn:Updater<T>,opts?:SpringUpdateOpts)=>Promise<void>} update
* @property {number} precision
* @property {number} damping
* @property {number} stiffness
*/
Loading…
Cancel
Save