@polygonjs/polygonjs/src/engine/operations/sop/UvTransform.ts

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

import {BaseSopOperation} from './_Base';
import {CoreGroup} from '../../../core/geometry/Group';
import {SopType} from '../../poly/registers/nodes/types/Sop';
import {BufferAttribute, Vector2} from 'three';
import {InputCloneMode} from '../../poly/InputCloneMode';
import {DefaultOperationParams} from '../../../core/operations/_Base';

interface UvTransformSopParams extends DefaultOperationParams {
	attribName: string;
	t: Vector2;
	s: Vector2;
	pivot: Vector2;
}

export class UvTransformSopOperation extends BaseSopOperation {
	static override readonly DEFAULT_PARAMS: UvTransformSopParams = {
		attribName: 'uv',
		t: new Vector2(0, 0),
		s: new Vector2(1, 1),
		pivot: new Vector2(0, 0),
	};
	static override readonly INPUT_CLONED_STATE = InputCloneMode.FROM_NODE;
	static override type(): Readonly<SopType.UV_TRANSFORM> {
		return SopType.UV_TRANSFORM;
	}

	override cook(input_contents: CoreGroup[], params: UvTransformSopParams): CoreGroup {
		const objects = input_contents[0].threejsObjectsWithGeo();
		for (let object of objects) {
			const geometry = object.geometry;
			const attrib = geometry.getAttribute(params.attribName) as BufferAttribute;
			const array = attrib.array;
			const ptsCount = array.length / 2;
			for (let i = 0; i < ptsCount; i++) {
				array[i * 2 + 0] = params.t.x + params.pivot.x + params.s.x * (array[i * 2 + 0] - params.pivot.x);
				array[i * 2 + 1] = params.t.y + params.pivot.y + params.s.y * (array[i * 2 + 1] - params.pivot.y);
			}
			attrib.needsUpdate = true;
		}
		return input_contents[0];
	}
}