@polygonjs/polygonjs/src/core/softBody/SoftBodyControllerRegister.ts

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

import {Object3D} from 'three';
import {CoreGraphNodeId} from '../graph/CoreGraph';
import {CoreSoftBodyAttribute, SoftBodyIdAttribute} from './SoftBodyAttribute';
import {SoftBodyController} from './SoftBodyController';
import type {PolyScene} from '../../engine/scene/PolyScene';
import type {TetSoftBodySolverSopNode} from '../../engine/nodes/sop/TetSoftBodySolver';
import {SoftBody} from './SoftBody';
import {TetEmbed} from './Common';
import {coreObjectClassFactory} from '../geometry/CoreObjectFactory';

const controllers: WeakMap<Object3D, SoftBodyController> = new WeakMap();
const softBodies: WeakMap<Object3D, SoftBody> = new WeakMap();

interface CreateOrFindSoftBodyControllerOptions {
	tetEmbed: TetEmbed;
	threejsObjectInSceneTree: Object3D;
}

export function createOrFindSoftBodyController(
	scene: PolyScene,
	node: TetSoftBodySolverSopNode,
	options: CreateOrFindSoftBodyControllerOptions
) {
	const {tetEmbed, threejsObjectInSceneTree} = options;
	const {highResObject} = tetEmbed;
	// update tetEmbed before passing it to the other functions
	if (highResObject) {
		tetEmbed.highResObject = threejsObjectInSceneTree;
		tetEmbed.threejsObjectInSceneTree = threejsObjectInSceneTree;
	} else {
		tetEmbed.lowResObject = threejsObjectInSceneTree;
		tetEmbed.threejsObjectInSceneTree = threejsObjectInSceneTree;
	}
	//
	let controller = controllers.get(threejsObjectInSceneTree);
	if (!controller) {
		controller = new SoftBodyController(scene, {
			node,
		});
		controllers.set(threejsObjectInSceneTree, controller);
		const {softBody} = createOrFindSoftBody(node, tetEmbed);
		if (softBody) {
			controller.setSoftBody(softBody);
		} else {
			console.warn('no softbody found');
		}
	}

	return {controller};
}
export function createOrFindSoftBody(node: TetSoftBodySolverSopNode, tetEmbed: TetEmbed) {
	const {threejsObjectInSceneTree} = tetEmbed;
	if (!threejsObjectInSceneTree) {
		throw 'createOrFindSoftBody: threejsObjectInSceneTree is null';
	}
	let softBody = softBodies.get(threejsObjectInSceneTree);
	if (!softBody) {
		const highResSkinningLookupSpacing = CoreSoftBodyAttribute.getHighResSkinningLookupSpacing(tetEmbed.tetObject);
		const highResSkinningLookupPadding = CoreSoftBodyAttribute.getHighResSkinningLookupPadding(tetEmbed.tetObject);

		softBody = new SoftBody({
			node,
			tetEmbed,
			highResSkinning: {
				lookup: {
					spacing: highResSkinningLookupSpacing,
					padding: highResSkinningLookupPadding,
				},
			},
		});
		softBodies.set(threejsObjectInSceneTree, softBody);
	}

	return {softBody};
}
export function softBodyControllerNodeIdFromObject(softBodyObject: Object3D) {
	const nodeId = coreObjectClassFactory(softBodyObject).attribValue(
		softBodyObject,
		SoftBodyIdAttribute.SOLVER_NODE
	) as CoreGraphNodeId | undefined;
	return nodeId;
}

export function softBodyControllerFromObject(softBodyObject: Object3D) {
	// const nodeId = CoreObject.attribValue(clothObject, ClothIdAttribute.OBJECT) as CoreGraphNodeId | undefined;
	// if (nodeId == null) {
	// 	return;
	// }
	return controllers.get(softBodyObject);
}
export function softBodyFromObject(softBodyObject: Object3D) {
	// const nodeId = CoreObject.attribValue(clothObject, ClothIdAttribute.OBJECT) as CoreGraphNodeId | undefined;
	// if (nodeId == null) {
	// 	return;
	// }
	return softBodies.get(softBodyObject);
}