@polygonjs/polygonjs/src/engine/nodes/anim/Position.ts

node-based WebGL 3D engine https://polygonjs.com

/**
 * Sets how the position of an animation
 *
 *
 */
import {TypedAnimNode} from './_Base';
import {TimelineBuilder} from '../../../core/animation/TimelineBuilder';
import {
	AnimationPosition,
	AnimationPositionMode,
	AnimationPositionRelativeTo,
	ANIMATION_POSITION_MODES,
	ANIMATION_POSITION_RELATIVE_TOS,
} from '../../../core/animation/Position';

import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
class PositionAnimParamsConfig extends NodeParamsConfig {
	/** @param sets the mode of the position. It can either be relative or absolute */
	mode = ParamConfig.INTEGER(0, {
		menu: {
			entries: ANIMATION_POSITION_MODES.map((name, value) => {
				return {name, value};
			}),
		},
	});
	/** @param if sets to relative, sets if it is relative to the start or end */
	relativeTo = ParamConfig.INTEGER(0, {
		menu: {
			entries: ANIMATION_POSITION_RELATIVE_TOS.map((name, value) => {
				return {name, value};
			}),
		},
	});
	/** @param offset */
	offset = ParamConfig.FLOAT(0);
}
const ParamsConfig = new PositionAnimParamsConfig();

export class PositionAnimNode extends TypedAnimNode<PositionAnimParamsConfig> {
	override paramsConfig = ParamsConfig;
	static override type() {
		return 'position';
	}

	override initializeNode() {
		this.io.inputs.setCount(0, 1);
	}

	setMode(mode: AnimationPositionMode) {
		this.p.mode.set(ANIMATION_POSITION_MODES.indexOf(mode));
	}
	setRelativeTo(relativeTo: AnimationPositionRelativeTo) {
		this.p.relativeTo.set(ANIMATION_POSITION_RELATIVE_TOS.indexOf(relativeTo));
	}

	override cook(input_contents: TimelineBuilder[]) {
		const timeline_builder = input_contents[0] || new TimelineBuilder();

		const position = new AnimationPosition();
		position.setMode(ANIMATION_POSITION_MODES[this.pv.mode]);
		position.setRelativeTo(ANIMATION_POSITION_RELATIVE_TOS[this.pv.relativeTo]);
		position.setOffset(this.pv.offset);
		timeline_builder.setPosition(position);

		this.setTimelineBuilder(timeline_builder);
	}
}