123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241 |
- import {
- Box3,
- Float32BufferAttribute,
- InstancedBufferGeometry,
- InstancedInterleavedBuffer,
- InterleavedBufferAttribute,
- Sphere,
- Vector3,
- WireframeGeometry
- } from 'three';
- const _box = new Box3();
- const _vector = new Vector3();
- class LineSegmentsGeometry extends InstancedBufferGeometry {
- constructor() {
- super();
- this.isLineSegmentsGeometry = true;
- this.type = 'LineSegmentsGeometry';
- const positions = [ - 1, 2, 0, 1, 2, 0, - 1, 1, 0, 1, 1, 0, - 1, 0, 0, 1, 0, 0, - 1, - 1, 0, 1, - 1, 0 ];
- const uvs = [ - 1, 2, 1, 2, - 1, 1, 1, 1, - 1, - 1, 1, - 1, - 1, - 2, 1, - 2 ];
- const index = [ 0, 2, 1, 2, 3, 1, 2, 4, 3, 4, 5, 3, 4, 6, 5, 6, 7, 5 ];
- this.setIndex( index );
- this.setAttribute( 'position', new Float32BufferAttribute( positions, 3 ) );
- this.setAttribute( 'uv', new Float32BufferAttribute( uvs, 2 ) );
- }
- applyMatrix4( matrix ) {
- const start = this.attributes.instanceStart;
- const end = this.attributes.instanceEnd;
- if ( start !== undefined ) {
- start.applyMatrix4( matrix );
- end.applyMatrix4( matrix );
- start.needsUpdate = true;
- }
- if ( this.boundingBox !== null ) {
- this.computeBoundingBox();
- }
- if ( this.boundingSphere !== null ) {
- this.computeBoundingSphere();
- }
- return this;
- }
- setPositions( array ) {
- let lineSegments;
- if ( array instanceof Float32Array ) {
- lineSegments = array;
- } else if ( Array.isArray( array ) ) {
- lineSegments = new Float32Array( array );
- }
- const instanceBuffer = new InstancedInterleavedBuffer( lineSegments, 6, 1 ); // xyz, xyz
- this.setAttribute( 'instanceStart', new InterleavedBufferAttribute( instanceBuffer, 3, 0 ) ); // xyz
- this.setAttribute( 'instanceEnd', new InterleavedBufferAttribute( instanceBuffer, 3, 3 ) ); // xyz
- //
- this.computeBoundingBox();
- this.computeBoundingSphere();
- return this;
- }
- setColors( array ) {
- let colors;
- if ( array instanceof Float32Array ) {
- colors = array;
- } else if ( Array.isArray( array ) ) {
- colors = new Float32Array( array );
- }
- const instanceColorBuffer = new InstancedInterleavedBuffer( colors, 6, 1 ); // rgb, rgb
- this.setAttribute( 'instanceColorStart', new InterleavedBufferAttribute( instanceColorBuffer, 3, 0 ) ); // rgb
- this.setAttribute( 'instanceColorEnd', new InterleavedBufferAttribute( instanceColorBuffer, 3, 3 ) ); // rgb
- return this;
- }
- fromWireframeGeometry( geometry ) {
- this.setPositions( geometry.attributes.position.array );
- return this;
- }
- fromEdgesGeometry( geometry ) {
- this.setPositions( geometry.attributes.position.array );
- return this;
- }
- fromMesh( mesh ) {
- this.fromWireframeGeometry( new WireframeGeometry( mesh.geometry ) );
- // set colors, maybe
- return this;
- }
- fromLineSegments( lineSegments ) {
- const geometry = lineSegments.geometry;
- this.setPositions( geometry.attributes.position.array ); // assumes non-indexed
- // set colors, maybe
- return this;
- }
- computeBoundingBox() {
- if ( this.boundingBox === null ) {
- this.boundingBox = new Box3();
- }
- const start = this.attributes.instanceStart;
- const end = this.attributes.instanceEnd;
- if ( start !== undefined && end !== undefined ) {
- this.boundingBox.setFromBufferAttribute( start );
- _box.setFromBufferAttribute( end );
- this.boundingBox.union( _box );
- }
- }
- computeBoundingSphere() {
- if ( this.boundingSphere === null ) {
- this.boundingSphere = new Sphere();
- }
- if ( this.boundingBox === null ) {
- this.computeBoundingBox();
- }
- const start = this.attributes.instanceStart;
- const end = this.attributes.instanceEnd;
- if ( start !== undefined && end !== undefined ) {
- const center = this.boundingSphere.center;
- this.boundingBox.getCenter( center );
- let maxRadiusSq = 0;
- for ( let i = 0, il = start.count; i < il; i ++ ) {
- _vector.fromBufferAttribute( start, i );
- maxRadiusSq = Math.max( maxRadiusSq, center.distanceToSquared( _vector ) );
- _vector.fromBufferAttribute( end, i );
- maxRadiusSq = Math.max( maxRadiusSq, center.distanceToSquared( _vector ) );
- }
- this.boundingSphere.radius = Math.sqrt( maxRadiusSq );
- if ( isNaN( this.boundingSphere.radius ) ) {
- console.error( 'THREE.LineSegmentsGeometry.computeBoundingSphere(): Computed radius is NaN. The instanced position data is likely to have NaN values.', this );
- }
- }
- }
- toJSON() {
- // todo
- }
- applyMatrix( matrix ) {
- console.warn( 'THREE.LineSegmentsGeometry: applyMatrix() has been renamed to applyMatrix4().' );
- return this.applyMatrix4( matrix );
- }
- }
- export { LineSegmentsGeometry };
|