JoinNode.js 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. import { TempNode } from '../core/TempNode.js';
  2. import { NodeUtils } from '../core/NodeUtils.js';
  3. var inputs = NodeUtils.elements;
  4. function JoinNode( x, y, z, w ) {
  5. TempNode.call( this, 'f' );
  6. this.x = x;
  7. this.y = y;
  8. this.z = z;
  9. this.w = w;
  10. }
  11. JoinNode.prototype = Object.create( TempNode.prototype );
  12. JoinNode.prototype.constructor = JoinNode;
  13. JoinNode.prototype.nodeType = 'Join';
  14. JoinNode.prototype.getNumElements = function () {
  15. var i = inputs.length;
  16. while ( i -- ) {
  17. if ( this[ inputs[ i ] ] !== undefined ) {
  18. ++ i;
  19. break;
  20. }
  21. }
  22. return Math.max( i, 2 );
  23. };
  24. JoinNode.prototype.getType = function ( builder ) {
  25. return builder.getTypeFromLength( this.getNumElements() );
  26. };
  27. JoinNode.prototype.generate = function ( builder, output ) {
  28. var type = this.getType( builder ),
  29. length = this.getNumElements(),
  30. outputs = [];
  31. for ( var i = 0; i < length; i ++ ) {
  32. var elm = this[ inputs[ i ] ];
  33. outputs.push( elm ? elm.build( builder, 'f' ) : '0.0' );
  34. }
  35. var code = ( length > 1 ? builder.getConstructorFromLength( length ) : '' ) + '( ' + outputs.join( ', ' ) + ' )';
  36. return builder.format( code, type, output );
  37. };
  38. JoinNode.prototype.copy = function ( source ) {
  39. TempNode.prototype.copy.call( this, source );
  40. for ( var prop in source.inputs ) {
  41. this[ prop ] = source.inputs[ prop ];
  42. }
  43. return this;
  44. };
  45. JoinNode.prototype.toJSON = function ( meta ) {
  46. var data = this.getJSONNode( meta );
  47. if ( ! data ) {
  48. data = this.createJSONNode( meta );
  49. data.inputs = {};
  50. var length = this.getNumElements();
  51. for ( var i = 0; i < length; i ++ ) {
  52. var elm = this[ inputs[ i ] ];
  53. if ( elm ) {
  54. data.inputs[ inputs[ i ] ] = elm.toJSON( meta ).uuid;
  55. }
  56. }
  57. }
  58. return data;
  59. };
  60. export { JoinNode };