three
Version:
JavaScript 3D library
157 lines (87 loc) • 3.27 kB
JavaScript
import NodeMaterial from './NodeMaterial.js';
import { attribute } from '../../nodes/core/AttributeNode.js';
import { cameraProjectionMatrix } from '../../nodes/accessors/Camera.js';
import { materialColor, materialOpacity, materialPointWidth } from '../../nodes/accessors/MaterialNode.js'; // or should this be a property, instead?
import { modelViewMatrix } from '../../nodes/accessors/ModelNode.js';
import { positionGeometry } from '../../nodes/accessors/Position.js';
import { smoothstep, lengthSq } from '../../nodes/math/MathNode.js';
import { Fn, vec4, float } from '../../nodes/tsl/TSLBase.js';
import { uv } from '../../nodes/accessors/UV.js';
import { viewport } from '../../nodes/display/ScreenNode.js';
import { PointsMaterial } from '../PointsMaterial.js';
const _defaultValues = /*@__PURE__*/ new PointsMaterial();
class InstancedPointsNodeMaterial extends NodeMaterial {
static get type() {
return 'InstancedPointsNodeMaterial';
}
constructor( params = {} ) {
super();
this.lights = false;
this.useAlphaToCoverage = true;
this.useColor = params.vertexColors;
this.pointWidth = 1;
this.pointColorNode = null;
this.pointWidthNode = null;
this.setDefaultValues( _defaultValues );
this.setValues( params );
}
setup( builder ) {
this.setupShaders( builder );
super.setup( builder );
}
setupShaders( { renderer } ) {
const useAlphaToCoverage = this.alphaToCoverage;
const useColor = this.useColor;
this.vertexNode = Fn( () => {
const instancePosition = attribute( 'instancePosition' ).xyz;
// camera space
const mvPos = vec4( modelViewMatrix.mul( vec4( instancePosition, 1.0 ) ) );
const aspect = viewport.z.div( viewport.w );
// clip space
const clipPos = cameraProjectionMatrix.mul( mvPos );
// offset in ndc space
const offset = positionGeometry.xy.toVar();
offset.mulAssign( this.pointWidthNode ? this.pointWidthNode : materialPointWidth );
offset.assign( offset.div( viewport.z ) );
offset.y.assign( offset.y.mul( aspect ) );
// back to clip space
offset.assign( offset.mul( clipPos.w ) );
//clipPos.xy += offset;
clipPos.addAssign( vec4( offset, 0, 0 ) );
return clipPos;
} )();
this.fragmentNode = Fn( () => {
const alpha = float( 1 ).toVar();
const len2 = lengthSq( uv().mul( 2 ).sub( 1 ) );
if ( useAlphaToCoverage && renderer.samples > 1 ) {
const dlen = float( len2.fwidth() ).toVar();
alpha.assign( smoothstep( dlen.oneMinus(), dlen.add( 1 ), len2 ).oneMinus() );
} else {
len2.greaterThan( 1.0 ).discard();
}
let pointColorNode;
if ( this.pointColorNode ) {
pointColorNode = this.pointColorNode;
} else {
if ( useColor ) {
const instanceColor = attribute( 'instanceColor' );
pointColorNode = instanceColor.mul( materialColor );
} else {
pointColorNode = materialColor;
}
}
alpha.mulAssign( materialOpacity );
return vec4( pointColorNode, alpha );
} )();
}
get alphaToCoverage() {
return this.useAlphaToCoverage;
}
set alphaToCoverage( value ) {
if ( this.useAlphaToCoverage !== value ) {
this.useAlphaToCoverage = value;
this.needsUpdate = true;
}
}
}
export default InstancedPointsNodeMaterial;