123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899 |
- 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;
- if ( node.isTempNode === true ) {
- return node.build( builder );
- }
- const name = this.name;
- 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;
- }
- }
- VarNode.prototype.isVarNode = true;
- export default VarNode;
|