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

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

/**
 * Applies a CAD revolution operation
 *
 *
 */
import {CADSopNode} from './_BaseCAD';
import {step} from '../../../core/geometry/modules/cad/CadConstant';
import {NodeParamsConfig, ParamConfig} from '../utils/params/ParamsConfig';
import {CoreCadType} from '../../../core/geometry/modules/cad/CadCoreType';
import {CadGeometryType, cadGeometryTypeFromShape, TopoDS_Edge} from '../../../core/geometry/modules/cad/CadCommon';
import {CadObject} from '../../../core/geometry/modules/cad/CadObject';
import {
	cadAxis,
	// cadPlaneXY
} from '../../../core/geometry/modules/cad/CadMath';
import {CoreGroup} from '../../../core/geometry/Group';
import {SopType} from '../../poly/registers/nodes/types/Sop';
import {curveDataFromEdge} from '../../../core/geometry/modules/cad/toObject3D/CadEdge';
import {traverseEdges} from '../../../core/geometry/modules/cad/CadTraverse';
import {CadLoaderSync} from '../../../core/geometry/modules/cad/CadLoaderSync';
// import {withCadException} from '../../../core/geometry/modules/cad/CadExceptionHandler';

class CADRevolutionSopParamsConfig extends NodeParamsConfig {
	/** @param axis */
	axis = ParamConfig.VECTOR3([0, 1, 0]);
	/** @param closed */
	closed = ParamConfig.BOOLEAN(true);
	/** @param thetaMin */
	// thetaMin = ParamConfig.FLOAT(0, {
	// 	range: [0, 2 * Math.PI],
	// 	rangeLocked: [false, false],
	// 	step,
	// 	visibleIf: {closed: false},
	// });
	/** @param thetaMax */
	// thetaMax = ParamConfig.FLOAT(`2*$PI`, {
	// 	range: [0, 2 * Math.PI],
	// 	rangeLocked: [false, false],
	// 	step,
	// 	visibleIf: {closed: false},
	// });
	/** @param phi */
	phi = ParamConfig.FLOAT(`2*$PI`, {
		range: [0, 2 * Math.PI],
		rangeLocked: [true, true],
		step,
		visibleIf: {closed: false},
	});
}
const ParamsConfig = new CADRevolutionSopParamsConfig();

export class CADRevolutionSopNode extends CADSopNode<CADRevolutionSopParamsConfig> {
	override paramsConfig = ParamsConfig;
	static override type() {
		return SopType.CAD_REVOLUTION;
	}
	protected override initializeNode() {
		this.io.inputs.setCount(1);
	}

	override cook(inputCoreGroups: CoreGroup[]) {
		// TODO: if input is 2d, convert to 3d
		// provide axis, angle, min, max
		const oc = CadLoaderSync.oc();
		const coreGroup0 = inputCoreGroups[0];
		const newObjects: CadObject<CadGeometryType>[] = [];
		const axis = cadAxis(this.pv.axis);

		// const thetaMin = Math.min(this.pv.thetaMin, this.pv.thetaMax);
		// const thetaMax = Math.max(this.pv.thetaMin, this.pv.thetaMax);

		const createRevolution = (edge: TopoDS_Edge) => {
			const handle = curveDataFromEdge(oc, edge).curveHandle;
			const api = new oc.BRepPrimAPI_MakeRevolution_6(axis, handle, this.pv.phi);
			const newShape = api.Shape();
			api.delete();
			const type = cadGeometryTypeFromShape(oc, newShape);
			if (type) {
				newObjects.push(new CadObject(newShape, type));
			} else {
				console.log('no type', newShape);
			}
		};

		const inputObjects = coreGroup0.cadObjects();
		if (inputObjects) {
			for (const inputObject of inputObjects) {
				const type = inputObject.type;
				switch (type) {
					case CadGeometryType.CURVE_2D: {
						// if (CoreCadType.isGeom2dCurve(inputObject)) {
						// 	const curve2D = inputObject.object();
						// 	// const api = new oc.GeomAPI()
						// 	console.log('A');
						// 	const handle2D = new oc.Handle_Geom2d_Curve_2(curve2D);
						// 	console.log('B');
						// 	const handle = oc.GeomAPI.To3d(handle2D, cadPlaneXY());
						// 	console.log('C');
						// 	const api = new oc.BRepLib_MakeEdge_24(handle);
						// 	console.log('D');
						// 	const edge = api.Edge();
						// 	console.log('E');
						// 	withCadException(oc, () => {
						// 		const result = oc.BRepLib.BuildCurves3d_2(edge);
						// 		if (result) {
						// 			createRevolution(edge);
						// 		}
						// 	});
						// 	console.log('F');
						// }
						// break;
					}
					case CadGeometryType.EDGE: {
						if (CoreCadType.isEdge(inputObject)) {
							createRevolution(inputObject.cadGeometry());
						}
						break;
					}
					case CadGeometryType.WIRE: {
						if (CoreCadType.isWire(inputObject)) {
							traverseEdges(oc, inputObject.cadGeometry(), (edge) => {
								createRevolution(edge);
							});
						}
						break;
					}
				}
			}
		}

		this.setCADObjects(newObjects);
	}
}