mirror of
https://github.com/tengge1/ShadowEditor.git
synced 2026-01-25 15:08:11 +00:00
141 lines
2.4 KiB
JavaScript
141 lines
2.4 KiB
JavaScript
/**
|
|
* @author sunag / http://www.sunag.com.br/
|
|
*/
|
|
|
|
import { TempNode } from '../core/TempNode.js';
|
|
|
|
function Math2Node( a, b, method ) {
|
|
|
|
TempNode.call( this );
|
|
|
|
this.a = a;
|
|
this.b = b;
|
|
|
|
this.method = method;
|
|
|
|
}
|
|
|
|
Math2Node.MIN = 'min';
|
|
Math2Node.MAX = 'max';
|
|
Math2Node.MOD = 'mod';
|
|
Math2Node.STEP = 'step';
|
|
Math2Node.REFLECT = 'reflect';
|
|
Math2Node.DISTANCE = 'distance';
|
|
Math2Node.DOT = 'dot';
|
|
Math2Node.CROSS = 'cross';
|
|
Math2Node.POW = 'pow';
|
|
|
|
Math2Node.prototype = Object.create( TempNode.prototype );
|
|
Math2Node.prototype.constructor = Math2Node;
|
|
Math2Node.prototype.nodeType = "Math2";
|
|
|
|
Math2Node.prototype.getInputType = function ( builder ) {
|
|
|
|
// use the greater length vector
|
|
|
|
if ( builder.getTypeLength( this.b.getType( builder ) ) > builder.getTypeLength( this.a.getType( builder ) ) ) {
|
|
|
|
return this.b.getType( builder );
|
|
|
|
}
|
|
|
|
return this.a.getType( builder );
|
|
|
|
};
|
|
|
|
Math2Node.prototype.getType = function ( builder ) {
|
|
|
|
switch ( this.method ) {
|
|
|
|
case Math2Node.DISTANCE:
|
|
case Math2Node.DOT:
|
|
|
|
return 'f';
|
|
|
|
case Math2Node.CROSS:
|
|
|
|
return 'v3';
|
|
|
|
}
|
|
|
|
return this.getInputType( builder );
|
|
|
|
};
|
|
|
|
Math2Node.prototype.generate = function ( builder, output ) {
|
|
|
|
var a, b,
|
|
type = this.getInputType( builder ),
|
|
al = builder.getTypeLength( this.a.getType( builder ) ),
|
|
bl = builder.getTypeLength( this.b.getType( builder ) );
|
|
|
|
// optimzer
|
|
|
|
switch ( this.method ) {
|
|
|
|
case Math2Node.CROSS:
|
|
|
|
a = this.a.build( builder, 'v3' );
|
|
b = this.b.build( builder, 'v3' );
|
|
|
|
break;
|
|
|
|
case Math2Node.STEP:
|
|
|
|
a = this.a.build( builder, al === 1 ? 'f' : type );
|
|
b = this.b.build( builder, type );
|
|
|
|
break;
|
|
|
|
case Math2Node.MIN:
|
|
case Math2Node.MAX:
|
|
case Math2Node.MOD:
|
|
|
|
a = this.a.build( builder, type );
|
|
b = this.b.build( builder, bl === 1 ? 'f' : type );
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
a = this.a.build( builder, type );
|
|
b = this.b.build( builder, type );
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return builder.format( this.method + '( ' + a + ', ' + b + ' )', this.getType( builder ), output );
|
|
|
|
};
|
|
|
|
Math2Node.prototype.copy = function ( source ) {
|
|
|
|
TempNode.prototype.copy.call( this, source );
|
|
|
|
this.a = source.a;
|
|
this.b = source.b;
|
|
this.method = source.method;
|
|
|
|
};
|
|
|
|
Math2Node.prototype.toJSON = function ( meta ) {
|
|
|
|
var 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.method = this.method;
|
|
|
|
}
|
|
|
|
return data;
|
|
|
|
};
|
|
|
|
export { Math2Node };
|