matrixMultiplicationBlock.ts 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. import { NodeMaterialBlock } from '../nodeMaterialBlock';
  2. import { NodeMaterialBlockConnectionPointTypes } from '../nodeMaterialBlockConnectionPointTypes';
  3. import { NodeMaterialBuildState } from '../nodeMaterialBuildState';
  4. import { NodeMaterialBlockTargets } from '../nodeMaterialBlockTargets';
  5. import { NodeMaterialConnectionPoint } from '../nodeMaterialBlockConnectionPoint';
  6. /**
  7. * Block used to multiply two matrices
  8. */
  9. export class MatrixMultiplicationBlock extends NodeMaterialBlock {
  10. /**
  11. * Creates a new MatrixMultiplicationBlock
  12. * @param name defines the block name
  13. */
  14. public constructor(name: string) {
  15. super(name, NodeMaterialBlockTargets.Vertex);
  16. this.registerInput("left", NodeMaterialBlockConnectionPointTypes.Matrix);
  17. this.registerInput("right", NodeMaterialBlockConnectionPointTypes.Matrix);
  18. this.registerOutput("output", NodeMaterialBlockConnectionPointTypes.Matrix);
  19. }
  20. /**
  21. * Gets the left operand
  22. */
  23. public get left(): NodeMaterialConnectionPoint {
  24. return this._inputs[0];
  25. }
  26. /**
  27. * Gets the right operand
  28. */
  29. public get right(): NodeMaterialConnectionPoint {
  30. return this._inputs[1];
  31. }
  32. /**
  33. * Gets the current class name
  34. * @returns the class name
  35. */
  36. public getClassName() {
  37. return "MatrixMultiplicationBlock";
  38. }
  39. protected _buildBlock(state: NodeMaterialBuildState) {
  40. super._buildBlock(state);
  41. let output = this._outputs[0];
  42. state.compilationString += this._declareOutput(output, state) + ` = ${this.left.associatedVariableName} * ${this.right.associatedVariableName};\r\n`;
  43. return this;
  44. }
  45. }