@polygonjs/polygonjs
Version:
node-based WebGL 3D engine https://polygonjs.com
57 lines (56 loc) • 1.95 kB
JavaScript
;
import { Float32BufferAttribute, Vector3 } from "three";
import { BufferGeometry } from "three";
import { TypedSopNode } from "./_Base";
import { ObjectType } from "../../../core/geometry/Constant";
import { NodeParamsConfig, ParamConfig } from "../utils/params/ParamsConfig";
class LineSopParamsConfig extends NodeParamsConfig {
constructor() {
super(...arguments);
/** @param length of the line */
this.length = ParamConfig.FLOAT(1, { range: [0, 10] });
/** @param number of points */
this.pointsCount = ParamConfig.INTEGER(2, {
range: [2, 100],
rangeLocked: [true, false]
});
/** @param start position of the line */
this.origin = ParamConfig.VECTOR3([0, 0, 0]);
/** @param direction of the line */
this.direction = ParamConfig.VECTOR3([0, 1, 0]);
}
}
const ParamsConfig = new LineSopParamsConfig();
export class LineSopNode extends TypedSopNode {
constructor() {
super(...arguments);
this.paramsConfig = ParamsConfig;
this._lastPt = new Vector3();
this._current = new Vector3();
}
static type() {
return "line";
}
initializeNode() {
}
cook() {
const pointsCount = Math.max(2, this.pv.pointsCount);
const positions = new Array(pointsCount * 3);
const indices = new Array(pointsCount);
this._lastPt.copy(this.pv.direction).normalize().multiplyScalar(this.pv.length);
for (let i = 0; i < pointsCount; i++) {
const i_n = i / (pointsCount - 1);
this._current.copy(this._lastPt).multiplyScalar(i_n);
this._current.add(this.pv.origin);
this._current.toArray(positions, i * 3);
if (i > 0) {
indices[(i - 1) * 2] = i - 1;
indices[(i - 1) * 2 + 1] = i;
}
}
const geometry = new BufferGeometry();
geometry.setAttribute("position", new Float32BufferAttribute(positions, 3));
geometry.setIndex(indices);
this.setGeometry(geometry, ObjectType.LINE_SEGMENTS);
}
}