@polygonjs/polygonjs/src/engine/nodes/sop/Tetrahedralize.ts

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

/**
 * Converts an input geometry to tetrahedrons
 *
 *
 *
 */
import {TetSopNode} from './_BaseTet';
import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
import {SopType} from '../../poly/registers/nodes/types/Sop';
import {CoreGroup} from '../../../core/geometry/Group';
import {tetrahedralize} from '../../../core/geometry/modules/tet/utils/tetrahedralize';
import {TetGeometry} from '../../../core/geometry/modules/tet/TetGeometry';
import {MeshWithBVHGeometry, ThreeMeshBVHHelper} from '../../../core/geometry/bvh/ThreeMeshBVHHelper';
import {Mesh, Vector3} from 'three';
import {mergeFaces} from '../../../core/geometry/operation/Fuse';
import {jitterPositions} from '../../../core/geometry/operation/Jitter';

const jitterMult = new Vector3(1, 1, 1);
class TetrahedralizeSopParamsConfig extends NodeParamsConfig {
	fuseDist = ParamConfig.FLOAT(0.001);
	jitter = ParamConfig.FLOAT(0.001);
	innerPointsResolution = ParamConfig.INTEGER(5, {
		range: [0, 10],
	});
	minQuality = ParamConfig.FLOAT(0.25, {
		range: [0, 1],
		rangeLocked: [true, true],
	});
	stepByStep = ParamConfig.BOOLEAN(0, {
		separatorBefore: true,
	});
	step = ParamConfig.INTEGER(-1, {
		range: [-1, 5000],
		rangeLocked: [true, false],
		visibleIf: {stepByStep: 1},
	});
	deleteOutsideTets = ParamConfig.BOOLEAN(1, {
		visibleIf: {stepByStep: 1},
	});
}
const ParamsConfig = new TetrahedralizeSopParamsConfig();

export class TetrahedralizeSopNode extends TetSopNode<TetrahedralizeSopParamsConfig> {
	override paramsConfig = ParamsConfig;
	static override type() {
		return SopType.TETRAHEDRALIZE;
	}

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

	override cook(inputCoreGroups: CoreGroup[]) {
		const coreGroup = inputCoreGroups[0];
		const inputMeshes = coreGroup.threejsObjectsWithGeo() as Mesh[];

		for (const inputMesh of inputMeshes) {
			mergeFaces(inputMesh.geometry, this.pv.fuseDist);
		}
		jitterPositions(coreGroup, {
			amount: this.pv.jitter,
			mult: jitterMult,
			seed: 0,
		});

		const tetGeometries: TetGeometry[] = [];
		for (const inputMesh of inputMeshes) {
			ThreeMeshBVHHelper.assignDefaultBVHIfNone(inputMesh);
			const tetGeometry = tetrahedralize({
				mesh: inputMesh as MeshWithBVHGeometry,
				jitterAmount: this.pv.jitter,
				innerPointsResolution: this.pv.innerPointsResolution,
				minQuality: this.pv.minQuality,
				stage: this.pv.stepByStep ? (this.pv.step >= 0 ? this.pv.step : null) : null,
				deleteOutsideTets: this.pv.stepByStep ? this.pv.deleteOutsideTets : true,
			});
			tetGeometries.push(tetGeometry);
		}

		this.setTetGeometries(tetGeometries);
	}
}