ReflectNode.js 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. /**
  2. * @author sunag / http://www.sunag.com.br/
  3. */
  4. THREE.ReflectNode = function ( scope ) {
  5. THREE.TempNode.call( this, 'v3', { unique: true } );
  6. this.scope = scope || THREE.ReflectNode.CUBE;
  7. };
  8. THREE.ReflectNode.CUBE = 'cube';
  9. THREE.ReflectNode.SPHERE = 'sphere';
  10. THREE.ReflectNode.VECTOR = 'vector';
  11. THREE.ReflectNode.prototype = Object.create( THREE.TempNode.prototype );
  12. THREE.ReflectNode.prototype.constructor = THREE.ReflectNode;
  13. THREE.ReflectNode.prototype.nodeType = "Reflect";
  14. THREE.ReflectNode.prototype.getType = function ( builder ) {
  15. switch ( this.scope ) {
  16. case THREE.ReflectNode.SPHERE:
  17. return 'v2';
  18. }
  19. return this.type;
  20. };
  21. THREE.ReflectNode.prototype.generate = function ( builder, output ) {
  22. var result;
  23. switch ( this.scope ) {
  24. case THREE.ReflectNode.VECTOR:
  25. //vec3 reflectVec = reflect( -geometry.viewDir, geometry.normal )
  26. //builder.material.addFragmentNode( 'vec3 reflectVec = inverseTransformDirection( reflect( -geometry.viewDir, geometry.normal ), viewMatrix );' );
  27. builder.material.addFragmentNode( 'vec3 reflectVec = inverseTransformDirection( reflect( -normalize( vViewPosition ), normal ), viewMatrix );' );
  28. result = 'reflectVec';
  29. break;
  30. case THREE.ReflectNode.CUBE:
  31. var reflectVec = new THREE.ReflectNode( THREE.ReflectNode.VECTOR ).build( builder, 'v3' );
  32. //flipEnvMap
  33. builder.material.addFragmentNode( 'vec3 reflectCubeVec = vec3( -1.0 * ' + reflectVec + '.x, ' + reflectVec + '.yz );' );
  34. result = 'reflectCubeVec';
  35. break;
  36. case THREE.ReflectNode.SPHERE:
  37. var reflectVec = new THREE.ReflectNode( THREE.ReflectNode.VECTOR ).build( builder, 'v3' );
  38. builder.material.addFragmentNode( 'vec2 reflectSphereVec = normalize((viewMatrix * vec4(' + reflectVec + ', 0.0 )).xyz + vec3(0.0,0.0,1.0)).xy * 0.5 + 0.5;' );
  39. result = 'reflectSphereVec';
  40. break;
  41. }
  42. return builder.format( result, this.getType( this.type ), output );
  43. };
  44. THREE.ReflectNode.prototype.toJSON = function ( meta ) {
  45. var data = this.getJSONNode( meta );
  46. if ( ! data ) {
  47. data = this.createJSONNode( meta );
  48. data.scope = this.scope;
  49. }
  50. return data;
  51. };