ExtendedMaterialNode.js 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. import MaterialNode from './MaterialNode.js';
  2. import { materialReference } from './MaterialReferenceNode.js';
  3. import { normalView } from './NormalNode.js';
  4. import { normalMap } from '../display/NormalMapNode.js';
  5. import { bumpMap } from '../display/BumpMapNode.js';
  6. import { addNodeClass } from '../core/Node.js';
  7. import { nodeImmutable } from '../shadernode/ShaderNode.js';
  8. class ExtendedMaterialNode extends MaterialNode {
  9. constructor( scope ) {
  10. super( scope );
  11. }
  12. getNodeType( builder ) {
  13. const scope = this.scope;
  14. let type = null;
  15. if ( scope === ExtendedMaterialNode.NORMAL || scope === ExtendedMaterialNode.CLEARCOAT_NORMAL ) {
  16. type = 'vec3';
  17. }
  18. return type || super.getNodeType( builder );
  19. }
  20. setup( builder ) {
  21. const material = builder.material;
  22. const scope = this.scope;
  23. let node = null;
  24. if ( scope === ExtendedMaterialNode.NORMAL ) {
  25. if ( material.normalMap ) {
  26. node = normalMap( this.getTexture( 'normalMap' ), materialReference( 'normalScale', 'vec2' ) );
  27. } else if ( material.bumpMap ) {
  28. // @TODO: Replace material.bumpMap to this.getTexture( 'bumpMap' )
  29. node = bumpMap( material.bumpMap, materialReference( 'bumpScale', 'float' ) );
  30. } else {
  31. node = normalView;
  32. }
  33. } else if ( scope === ExtendedMaterialNode.CLEARCOAT_NORMAL ) {
  34. node = material.clearcoatNormalMap ? normalMap( this.getTexture( 'clearcoatNormalMap' ), materialReference( 'clearcoatNormalScale', 'vec2' ) ) : normalView;
  35. }
  36. return node || super.setup( builder );
  37. }
  38. }
  39. ExtendedMaterialNode.NORMAL = 'normal';
  40. ExtendedMaterialNode.CLEARCOAT_NORMAL = 'clearcoatNormal';
  41. export default ExtendedMaterialNode;
  42. export const materialNormal = nodeImmutable( ExtendedMaterialNode, ExtendedMaterialNode.NORMAL );
  43. export const materialClearcoatNormal = nodeImmutable( ExtendedMaterialNode, ExtendedMaterialNode.CLEARCOAT_NORMAL );
  44. addNodeClass( 'ExtendedMaterialNode', ExtendedMaterialNode );