87 lines
2.4 KiB
JavaScript
87 lines
2.4 KiB
JavaScript
import {
|
|
InstancedInterleavedBuffer,
|
|
InterleavedBufferAttribute,
|
|
Line2NodeMaterial,
|
|
Mesh,
|
|
Vector3
|
|
} from 'three/webgpu';
|
|
|
|
import { LineSegmentsGeometry } from '../../lines/LineSegmentsGeometry.js';
|
|
|
|
const _start = new Vector3();
|
|
const _end = new Vector3();
|
|
|
|
/**
|
|
* A class for creating wireframes based on wide lines.
|
|
*
|
|
* This module can only be used with {@link WebGPURenderer}. When using {@link WebGLRenderer},
|
|
* import the class from `lines/Wireframe.js`.
|
|
*
|
|
* @augments Mesh
|
|
* @three_import import { Wireframe } from 'three/addons/lines/webgpu/Wireframe.js';
|
|
*/
|
|
class Wireframe extends Mesh {
|
|
|
|
/**
|
|
* Constructs a new wireframe.
|
|
*
|
|
* @param {LineSegmentsGeometry} [geometry] - The line geometry.
|
|
* @param {Line2NodeMaterial} [material] - The line material.
|
|
*/
|
|
constructor( geometry = new LineSegmentsGeometry(), material = new Line2NodeMaterial( { color: Math.random() * 0xffffff } ) ) {
|
|
|
|
super( geometry, material );
|
|
|
|
/**
|
|
* This flag can be used for type testing.
|
|
*
|
|
* @type {boolean}
|
|
* @readonly
|
|
* @default true
|
|
*/
|
|
this.isWireframe = true;
|
|
|
|
this.type = 'Wireframe';
|
|
|
|
}
|
|
|
|
/**
|
|
* Computes an array of distance values which are necessary for rendering dashed lines.
|
|
* For each vertex in the geometry, the method calculates the cumulative length from the
|
|
* current point to the very beginning of the line.
|
|
*
|
|
* @return {Wireframe} A reference to this instance.
|
|
*/
|
|
computeLineDistances() {
|
|
|
|
// for backwards-compatibility, but could be a method of LineSegmentsGeometry...
|
|
|
|
const geometry = this.geometry;
|
|
|
|
const instanceStart = geometry.attributes.instanceStart;
|
|
const instanceEnd = geometry.attributes.instanceEnd;
|
|
const lineDistances = new Float32Array( 2 * instanceStart.count );
|
|
|
|
for ( let i = 0, j = 0, l = instanceStart.count; i < l; i ++, j += 2 ) {
|
|
|
|
_start.fromBufferAttribute( instanceStart, i );
|
|
_end.fromBufferAttribute( instanceEnd, i );
|
|
|
|
lineDistances[ j ] = ( j === 0 ) ? 0 : lineDistances[ j - 1 ];
|
|
lineDistances[ j + 1 ] = lineDistances[ j ] + _start.distanceTo( _end );
|
|
|
|
}
|
|
|
|
const instanceDistanceBuffer = new InstancedInterleavedBuffer( lineDistances, 2, 1 ); // d0, d1
|
|
|
|
geometry.setAttribute( 'instanceDistanceStart', new InterleavedBufferAttribute( instanceDistanceBuffer, 1, 0 ) ); // d0
|
|
geometry.setAttribute( 'instanceDistanceEnd', new InterleavedBufferAttribute( instanceDistanceBuffer, 1, 1 ) ); // d1
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export { Wireframe };
|