@polygonjs/polygonjs/src/engine/functions/_Add.ts

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

import {Vector2, Vector3, Vector4} from 'three';
import {NamedFunction, NamedFunction2} from './_Base';

export class addNumber extends NamedFunction<Array<number>, Array<string>> {
	static override type() {
		return 'addNumber';
	}
	func(...args: Array<number>): number {
		let first = args[0];
		for (let i = 1; i < args.length; i++) {
			first += args[i];
		}
		return first;
	}
}

type AvailableItem = Vector2 | Vector3 | Vector4;
export class addVector<V extends AvailableItem> extends NamedFunction<Array<V>, Array<string>> {
	static override type() {
		return 'addVector';
	}
	func(...args: Array<V>): V {
		const first: V = args[0];
		for (let i = 1; i < args.length; i++) {
			first.add(args[i] as any);
		}
		return first;
	}
}

export class addVectorNumber<V extends AvailableItem> extends NamedFunction2<[V, number]> {
	static override type() {
		return 'addVectorNumber';
	}
	func(vector: V, number: number): V {
		return vector.addScalar(number) as V;
	}
}