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

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

import {BaseSopOperation} from './_Base';
import {CoreGroup} from '../../../core/geometry/Group';
import {Vector3, Box3} from 'three';
import {CoreTransform} from '../../../core/Transform';
import {RoundedBoxGeometry} from 'three/examples/jsm/geometries/RoundedBoxGeometry';
import {InputCloneMode} from '../../../engine/poly/InputCloneMode';
import {DefaultOperationParams} from '../../../core/operations/_Base';
const tmpBox = new Box3();
const tmpSize = new Vector3();
const tmpCenter = new Vector3();
interface RoundedBoxSopParams extends DefaultOperationParams {
	size: number;
	sizes: Vector3;
	divisions: number;
	bevel: number;
	center: Vector3;
}

export class RoundedBoxSopOperation extends BaseSopOperation {
	static override readonly DEFAULT_PARAMS: RoundedBoxSopParams = {
		size: 1,
		sizes: new Vector3(1, 1, 1),
		divisions: 2,
		bevel: 0.1,
		center: new Vector3(0, 0, 0),
	};
	static override readonly INPUT_CLONED_STATE = InputCloneMode.NEVER;
	static override type(): Readonly<'roundedBox'> {
		return 'roundedBox';
	}
	private _coreTransform = new CoreTransform();
	override cook(inputCoreGroups: CoreGroup[], params: RoundedBoxSopParams) {
		const coreGroup = inputCoreGroups[0];
		const geometry = coreGroup ? this._cookWithInput(coreGroup, params) : this._cookWithoutInput(params);

		return this.createCoreGroupFromGeometry(geometry);
	}
	private _cookWithoutInput(params: RoundedBoxSopParams) {
		const {sizes, size} = params;
		const geometry = new RoundedBoxGeometry(
			sizes.x * size,
			sizes.y * size,
			sizes.z * size,
			params.divisions,
			params.bevel
		);
		geometry.translate(params.center.x, params.center.y, params.center.z);
		geometry.computeVertexNormals();
		return geometry;
	}

	private _cookWithInput(coreGroup: CoreGroup, params: RoundedBoxSopParams) {
		coreGroup.boundingBox(tmpBox);
		tmpBox.getSize(tmpSize);
		tmpBox.getCenter(tmpCenter);

		const geometry = new RoundedBoxGeometry(tmpSize.x, tmpSize.y, tmpSize.z, params.divisions, params.bevel);
		const matrix = this._coreTransform.translationMatrix(tmpCenter);
		geometry.applyMatrix4(matrix);
		return geometry;
	}
}