guoyao-jiankang/public/static/threejs135/jsm/nodes/math/CondNode.js

import { TempNode } from '../core/TempNode.js';

class CondNode extends TempNode {

	constructor( a, b, op, ifNode, elseNode ) {

		super();

		this.a = a;
		this.b = b;

		this.op = op;

		this.ifNode = ifNode;
		this.elseNode = elseNode;

	}

	getType( builder ) {

		if ( this.ifNode ) {

			const ifType = this.ifNode.getType( builder );
			const elseType = this.elseNode.getType( builder );

			if ( builder.getTypeLength( elseType ) > builder.getTypeLength( ifType ) ) {

				return elseType;

			}

			return ifType;

		}

		return 'b';

	}

	getCondType( builder ) {

		if ( builder.getTypeLength( this.b.getType( builder ) ) > builder.getTypeLength( this.a.getType( builder ) ) ) {

			return this.b.getType( builder );

		}

		return this.a.getType( builder );

	}

	generate( builder, output ) {

		const type = this.getType( builder ),
			condType = this.getCondType( builder ),
			a = this.a.build( builder, condType ),
			b = this.b.build( builder, condType );

		let code;

		if ( this.ifNode ) {

			const ifCode = this.ifNode.build( builder, type ),
				elseCode = this.elseNode.build( builder, type );

			code = '( ' + [ a, this.op, b, '?', ifCode, ':', elseCode ].join( ' ' ) + ' )';

		} else {

			code = '( ' + a + ' ' + this.op + ' ' + b + ' )';

		}

		return builder.format( code, this.getType( builder ), output );

	}

	copy( source ) {

		super.copy( source );

		this.a = source.a;
		this.b = source.b;

		this.op = source.op;

		this.ifNode = source.ifNode;
		this.elseNode = source.elseNode;

		return this;

	}

	toJSON( meta ) {

		let data = this.getJSONNode( meta );

		if ( ! data ) {

			data = this.createJSONNode( meta );

			data.a = this.a.toJSON( meta ).uuid;
			data.b = this.b.toJSON( meta ).uuid;

			data.op = this.op;

			if ( data.ifNode ) data.ifNode = this.ifNode.toJSON( meta ).uuid;
			if ( data.elseNode ) data.elseNode = this.elseNode.toJSON( meta ).uuid;

		}

		return data;

	}

}

CondNode.EQUAL = '==';
CondNode.NOT_EQUAL = '!=';
CondNode.GREATER = '>';
CondNode.GREATER_EQUAL = '>=';
CondNode.LESS = '<';
CondNode.LESS_EQUAL = '<=';
CondNode.AND = '&&';
CondNode.OR = '||';

CondNode.prototype.nodeType = 'Cond';

export { CondNode };