12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697 |
- import Node from './Node.js';
- import OperatorNode from '../math/OperatorNode.js';
- class VarNode extends Node {
- constructor( node, name = null ) {
- super();
- this.node = node;
- this.name = name;
- }
- op( op, ...params ) {
- this.node = new OperatorNode( op, this.node, ...params );
- return this;
- }
- assign( ...params ) {
- return this.op( '=', ...params );
- }
- add( ...params ) {
- return this.op( '+', ...params );
- }
- sub( ...params ) {
- return this.op( '-', ...params );
- }
- mul( ...params ) {
- return this.op( '*', ...params );
- }
- div( ...params ) {
- return this.op( '/', ...params );
- }
- getHash( builder ) {
- return this.name || super.getHash( builder );
- }
- getNodeType( builder ) {
- return this.node.getNodeType( builder );
- }
- generate( builder ) {
- const node = this.node;
- const name = this.name;
- if ( name === null && node.isTempNode === true ) {
- return node.build( builder );
- }
- const type = builder.getVectorType( this.getNodeType( builder ) );
- const snippet = node.build( builder, type );
- const nodeVar = builder.getVarFromNode( this, type );
- if ( name !== null ) {
- nodeVar.name = name;
- }
- const propertyName = builder.getPropertyName( nodeVar );
- builder.addFlowCode( `${propertyName} = ${snippet}` );
- return propertyName;
- }
- }
- export default VarNode;
|