68 lines
1.9 KiB
JavaScript
68 lines
1.9 KiB
JavaScript
import LightingModel from '../core/LightingModel.js';
|
|
import F_Schlick from './BSDF/F_Schlick.js';
|
|
import BRDF_Lambert from './BSDF/BRDF_Lambert.js';
|
|
import { diffuseColor } from '../core/PropertyNode.js';
|
|
import { transformedNormalView } from '../accessors/NormalNode.js';
|
|
import { materialSpecularStrength } from '../accessors/MaterialNode.js';
|
|
import { shininess, specularColor } from '../core/PropertyNode.js';
|
|
import { positionViewDirection } from '../accessors/PositionNode.js';
|
|
import { tslFn, float } from '../shadernode/ShaderNode.js';
|
|
|
|
const G_BlinnPhong_Implicit = () => float( 0.25 );
|
|
|
|
const D_BlinnPhong = tslFn( ( { dotNH } ) => {
|
|
|
|
return shininess.mul( float( 0.5 ) ).add( 1.0 ).mul( float( 1 / Math.PI ) ).mul( dotNH.pow( shininess ) );
|
|
|
|
} );
|
|
|
|
const BRDF_BlinnPhong = tslFn( ( { lightDirection } ) => {
|
|
|
|
const halfDir = lightDirection.add( positionViewDirection ).normalize();
|
|
|
|
const dotNH = transformedNormalView.dot( halfDir ).clamp();
|
|
const dotVH = positionViewDirection.dot( halfDir ).clamp();
|
|
|
|
const F = F_Schlick( { f0: specularColor, f90: 1.0, dotVH } );
|
|
const G = G_BlinnPhong_Implicit();
|
|
const D = D_BlinnPhong( { dotNH } );
|
|
|
|
return F.mul( G ).mul( D );
|
|
|
|
} );
|
|
|
|
class PhongLightingModel extends LightingModel {
|
|
|
|
constructor( specular = true ) {
|
|
|
|
super();
|
|
|
|
this.specular = specular;
|
|
|
|
}
|
|
|
|
direct( { lightDirection, lightColor, reflectedLight } ) {
|
|
|
|
const dotNL = transformedNormalView.dot( lightDirection ).clamp();
|
|
const irradiance = dotNL.mul( lightColor );
|
|
|
|
reflectedLight.directDiffuse.addAssign( irradiance.mul( BRDF_Lambert( { diffuseColor: diffuseColor.rgb } ) ) );
|
|
|
|
if ( this.specular === true ) {
|
|
|
|
reflectedLight.directSpecular.addAssign( irradiance.mul( BRDF_BlinnPhong( { lightDirection } ) ).mul( materialSpecularStrength ) );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
indirectDiffuse( { irradiance, reflectedLight } ) {
|
|
|
|
reflectedLight.indirectDiffuse.addAssign( irradiance.mul( BRDF_Lambert( { diffuseColor } ) ) );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default PhongLightingModel;
|