@polygonjs/polygonjs/src/core/geometry/modules/cad/CadCoreEdge.ts

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

import {CoreEntity} from '../../CoreEntity';
import {TopoDS_Shape, TopoDS_Edge} from './CadCommon';
import {Vector2, Vector3, Vector4} from 'three';
import {AttribValue, NumericAttribValue} from '../../../../types/GlobalTypes';
import {Attribute, CoreAttribute} from '../../Attribute';
import {CadLoaderSync} from './CadLoaderSync';
import {traverseVertices} from './CadTraverse';
import {ObjectBuilder, CoreObjectType} from '../../ObjectContent';
import {AttribClass} from '../../Constant';

export class CadCoreEdge extends CoreEntity {
	constructor(protected _shape: TopoDS_Shape, protected _edge: TopoDS_Edge, _index: number) {
		super(_shape, _index);
	}
	edge() {
		return this._edge;
	}
	geometry() {
		return this._shape;
	}
	builder<T extends CoreObjectType>(): ObjectBuilder<T> | undefined {
		return undefined;
	}

	setAttribValue(attribName: string, attribValue: NumericAttribValue | string) {}
	attribValue(attribName: string, target?: Vector2 | Vector3 | Vector4): AttribValue | undefined {
		if (attribName === Attribute.POINT_INDEX) {
			return this._index;
		} else {
			const remapedName = CoreAttribute.remapName(attribName);
			if (remapedName == Attribute.POSITION && target instanceof Vector3) {
				this.position(target);
			}
			// TODO: use vertices center for position?
			// this._edge.
			return this._index;
		}
	}
	stringAttribValue(attribName: string): string | null {
		return '';
	}
	position(target: Vector3): Vector3 {
		const oc = CadLoaderSync.oc();
		let verticesCount = 0;
		target.set(0, 0, 0);
		traverseVertices(oc, this._edge, (vertex, i) => {
			const point = oc.BRep_Tool.Pnt(vertex);
			target.x += point.X();
			target.y += point.Y();
			target.z += point.Z();
			verticesCount++;
		});
		target.divideScalar(verticesCount);
		return target;
	}
	//
	//
	// RELATED ENTITIES
	//
	//
	relatedEntities(attribClass: AttribClass): CoreEntity[] {
		return [];
	}
}