matrixMultiplicationBlock.ts 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  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 output component
  34. */
  35. public get output(): NodeMaterialConnectionPoint {
  36. return this._outputs[0];
  37. }
  38. /**
  39. * Gets the current class name
  40. * @returns the class name
  41. */
  42. public getClassName() {
  43. return "MatrixMultiplicationBlock";
  44. }
  45. protected _buildBlock(state: NodeMaterialBuildState) {
  46. super._buildBlock(state);
  47. let output = this._outputs[0];
  48. state.compilationString += this._declareOutput(output, state) + ` = ${this.left.associatedVariableName} * ${this.right.associatedVariableName};\r\n`;
  49. return this;
  50. }
  51. }