@polygonjs/polygonjs/src/core/TransformSpace.ts

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

import {Matrix4} from 'three';
import {TypeAssert} from '../engine/poly/Assert';
import {CoreObjectType, isObject3D, ObjectContent} from './geometry/ObjectContent';

export enum ObjectTransformSpace {
	PARENT = 'parent',
	LOCAL = 'local',
}
export enum ObjectTransformMode {
	SET = 'set matrix',
	MULT = 'multiply matrix',
}
export const OBJECT_TRANSFORM_SPACES: ObjectTransformSpace[] = [
	ObjectTransformSpace.PARENT,
	ObjectTransformSpace.LOCAL,
];
export const OBJECT_TRANSFORM_MODES: ObjectTransformMode[] = [ObjectTransformMode.SET, ObjectTransformMode.MULT];
export const OBJECT_TRANSFORM_SPACE_MENU_ENTRIES = [
	{name: 'parent', value: OBJECT_TRANSFORM_SPACES.indexOf(ObjectTransformSpace.PARENT)},
	{name: 'local', value: OBJECT_TRANSFORM_SPACES.indexOf(ObjectTransformSpace.LOCAL)},
];

export function applyTransformWithSpaceToObject(
	object: ObjectContent<CoreObjectType>,
	matrix: Matrix4,
	transformSpace: ObjectTransformSpace,
	transformMode: ObjectTransformMode
) {
	switch (transformSpace) {
		case ObjectTransformSpace.PARENT: {
			if (isObject3D(object)) {
				object.updateMatrix();
				object.applyMatrix4(matrix);
				object.matrix.decompose(object.position, object.quaternion, object.scale);
			} else {
				object.applyMatrix4(matrix);
			}
			return;
		}
		case ObjectTransformSpace.LOCAL: {
			if (isObject3D(object)) {
				object.updateMatrix();
				if (transformMode == ObjectTransformMode.SET) {
					object.matrix.copy(matrix);
				} else {
					object.matrix.multiply(matrix);
				}
				object.matrix.decompose(object.position, object.quaternion, object.scale);
			} else {
				// it should ideally multiply the existing matrix,
				// but I'm not sure how to do that with cad objects for now
				object.applyMatrix4(matrix);
			}
			return;
		}
	}
	TypeAssert.unreachable(transformSpace);
}