2
0

ReflectNode.js 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  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. builder.material.addFragmentNode( 'vec3 reflectVec = inverseTransformDirection( reflect( -normalize( vViewPosition ), normal ), viewMatrix );' );
  26. result = 'reflectVec';
  27. break;
  28. case THREE.ReflectNode.CUBE:
  29. var reflectVec = new THREE.ReflectNode( THREE.ReflectNode.VECTOR ).build( builder, 'v3' );
  30. builder.material.addFragmentNode( 'vec3 reflectCubeVec = vec3( -1.0 * ' + reflectVec + '.x, ' + reflectVec + '.yz );' );
  31. result = 'reflectCubeVec';
  32. break;
  33. case THREE.ReflectNode.SPHERE:
  34. var reflectVec = new THREE.ReflectNode( THREE.ReflectNode.VECTOR ).build( builder, 'v3' );
  35. builder.material.addFragmentNode( 'vec2 reflectSphereVec = normalize((viewMatrix * vec4(' + reflectVec + ', 0.0 )).xyz + vec3(0.0,0.0,1.0)).xy * 0.5 + 0.5;' );
  36. result = 'reflectSphereVec';
  37. break;
  38. }
  39. return builder.format( result, this.getType( this.type ), output );
  40. };
  41. THREE.ReflectNode.prototype.toJSON = function ( meta ) {
  42. var data = this.getJSONNode( meta );
  43. if ( ! data ) {
  44. data = this.createJSONNode( meta );
  45. data.scope = this.scope;
  46. }
  47. return data;
  48. };