Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/flo-bit/shapecraft. flo-bit.dev/shapecraft
Something went wrong. Try again.
13 kB · 407 lines
JavaScript
at commit 049ef598
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408import TempNode from '../core/TempNode.js';import { sub, mul, div } from './OperatorNode.js';import { addMethodChaining, nodeObject, nodeProxy, float, vec2, vec3, vec4, Fn } from '../tsl/TSLCore.js';
class MathNode extends TempNode {
static get type() {
return 'MathNode';
}
constructor( method, aNode, bNode = null, cNode = null ) {
super();
this.method = method;
this.aNode = aNode; this.bNode = bNode; this.cNode = cNode;
}
getInputType( builder ) {
const aType = this.aNode.getNodeType( builder ); const bType = this.bNode ? this.bNode.getNodeType( builder ) : null; const cType = this.cNode ? this.cNode.getNodeType( builder ) : null;
const aLen = builder.isMatrix( aType ) ? 0 : builder.getTypeLength( aType ); const bLen = builder.isMatrix( bType ) ? 0 : builder.getTypeLength( bType ); const cLen = builder.isMatrix( cType ) ? 0 : builder.getTypeLength( cType );
if ( aLen > bLen && aLen > cLen ) {
return aType;
} else if ( bLen > cLen ) {
return bType;
} else if ( cLen > aLen ) {
return cType;
}
return aType;
}
getNodeType( builder ) {
const method = this.method;
if ( method === MathNode.LENGTH || method === MathNode.DISTANCE || method === MathNode.DOT ) {
return 'float';
} else if ( method === MathNode.CROSS ) {
return 'vec3';
} else if ( method === MathNode.ALL ) {
return 'bool';
} else if ( method === MathNode.EQUALS ) {
return builder.changeComponentType( this.aNode.getNodeType( builder ), 'bool' );
} else if ( method === MathNode.MOD ) {
return this.aNode.getNodeType( builder );
} else {
return this.getInputType( builder );
}
}
generate( builder, output ) {
const method = this.method;
const type = this.getNodeType( builder ); const inputType = this.getInputType( builder );
const a = this.aNode; const b = this.bNode; const c = this.cNode;
const isWebGL = builder.renderer.isWebGLRenderer === true;
if ( method === MathNode.TRANSFORM_DIRECTION ) {
// dir can be either a direction vector or a normal vector // upper-left 3x3 of matrix is assumed to be orthogonal
let tA = a; let tB = b;
if ( builder.isMatrix( tA.getNodeType( builder ) ) ) {
tB = vec4( vec3( tB ), 0.0 );
} else {
tA = vec4( vec3( tA ), 0.0 );
}
const mulNode = mul( tA, tB ).xyz;
return normalize( mulNode ).build( builder, output );
} else if ( method === MathNode.NEGATE ) {
return builder.format( '( - ' + a.build( builder, inputType ) + ' )', type, output );
} else if ( method === MathNode.ONE_MINUS ) {
return sub( 1.0, a ).build( builder, output );
} else if ( method === MathNode.RECIPROCAL ) {
return div( 1.0, a ).build( builder, output );
} else if ( method === MathNode.DIFFERENCE ) {
return abs( sub( a, b ) ).build( builder, output );
} else {
const params = [];
if ( method === MathNode.CROSS || method === MathNode.MOD ) {
params.push( a.build( builder, type ), b.build( builder, type ) );
} else if ( isWebGL && method === MathNode.STEP ) {
params.push( a.build( builder, builder.getTypeLength( a.getNodeType( builder ) ) === 1 ? 'float' : inputType ), b.build( builder, inputType ) );
} else if ( ( isWebGL && ( method === MathNode.MIN || method === MathNode.MAX ) ) || method === MathNode.MOD ) {
params.push( a.build( builder, inputType ), b.build( builder, builder.getTypeLength( b.getNodeType( builder ) ) === 1 ? 'float' : inputType ) );
} else if ( method === MathNode.REFRACT ) {
params.push( a.build( builder, inputType ), b.build( builder, inputType ), c.build( builder, 'float' ) );
} else if ( method === MathNode.MIX ) {
params.push( a.build( builder, inputType ), b.build( builder, inputType ), c.build( builder, builder.getTypeLength( c.getNodeType( builder ) ) === 1 ? 'float' : inputType ) );
} else {
params.push( a.build( builder, inputType ) ); if ( b !== null ) params.push( b.build( builder, inputType ) ); if ( c !== null ) params.push( c.build( builder, inputType ) );
}
return builder.format( `${ builder.getMethod( method, type ) }( ${params.join( ', ' )} )`, type, output );
}
}
serialize( data ) {
super.serialize( data );
data.method = this.method;
}
deserialize( data ) {
super.deserialize( data );
this.method = data.method;
}
}
// 1 input
MathNode.ALL = 'all';MathNode.ANY = 'any';MathNode.EQUALS = 'equals';
MathNode.RADIANS = 'radians';MathNode.DEGREES = 'degrees';MathNode.EXP = 'exp';MathNode.EXP2 = 'exp2';MathNode.LOG = 'log';MathNode.LOG2 = 'log2';MathNode.SQRT = 'sqrt';MathNode.INVERSE_SQRT = 'inversesqrt';MathNode.FLOOR = 'floor';MathNode.CEIL = 'ceil';MathNode.NORMALIZE = 'normalize';MathNode.FRACT = 'fract';MathNode.SIN = 'sin';MathNode.COS = 'cos';MathNode.TAN = 'tan';MathNode.ASIN = 'asin';MathNode.ACOS = 'acos';MathNode.ATAN = 'atan';MathNode.ABS = 'abs';MathNode.SIGN = 'sign';MathNode.LENGTH = 'length';MathNode.NEGATE = 'negate';MathNode.ONE_MINUS = 'oneMinus';MathNode.DFDX = 'dFdx';MathNode.DFDY = 'dFdy';MathNode.ROUND = 'round';MathNode.RECIPROCAL = 'reciprocal';MathNode.TRUNC = 'trunc';MathNode.FWIDTH = 'fwidth';MathNode.BITCAST = 'bitcast';MathNode.TRANSPOSE = 'transpose';
// 2 inputs
MathNode.ATAN2 = 'atan2';MathNode.MIN = 'min';MathNode.MAX = 'max';MathNode.MOD = 'mod';MathNode.STEP = 'step';MathNode.REFLECT = 'reflect';MathNode.DISTANCE = 'distance';MathNode.DIFFERENCE = 'difference';MathNode.DOT = 'dot';MathNode.CROSS = 'cross';MathNode.POW = 'pow';MathNode.TRANSFORM_DIRECTION = 'transformDirection';
// 3 inputs
MathNode.MIX = 'mix';MathNode.CLAMP = 'clamp';MathNode.REFRACT = 'refract';MathNode.SMOOTHSTEP = 'smoothstep';MathNode.FACEFORWARD = 'faceforward';
export default MathNode;
export const EPSILON = /*@__PURE__*/ float( 1e-6 );export const INFINITY = /*@__PURE__*/ float( 1e6 );export const PI = /*@__PURE__*/ float( Math.PI );export const PI2 = /*@__PURE__*/ float( Math.PI * 2 );
export const all = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ALL );export const any = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ANY );export const equals = /*@__PURE__*/ nodeProxy( MathNode, MathNode.EQUALS );
export const radians = /*@__PURE__*/ nodeProxy( MathNode, MathNode.RADIANS );export const degrees = /*@__PURE__*/ nodeProxy( MathNode, MathNode.DEGREES );export const exp = /*@__PURE__*/ nodeProxy( MathNode, MathNode.EXP );export const exp2 = /*@__PURE__*/ nodeProxy( MathNode, MathNode.EXP2 );export const log = /*@__PURE__*/ nodeProxy( MathNode, MathNode.LOG );export const log2 = /*@__PURE__*/ nodeProxy( MathNode, MathNode.LOG2 );export const sqrt = /*@__PURE__*/ nodeProxy( MathNode, MathNode.SQRT );export const inverseSqrt = /*@__PURE__*/ nodeProxy( MathNode, MathNode.INVERSE_SQRT );export const floor = /*@__PURE__*/ nodeProxy( MathNode, MathNode.FLOOR );export const ceil = /*@__PURE__*/ nodeProxy( MathNode, MathNode.CEIL );export const normalize = /*@__PURE__*/ nodeProxy( MathNode, MathNode.NORMALIZE );export const fract = /*@__PURE__*/ nodeProxy( MathNode, MathNode.FRACT );export const sin = /*@__PURE__*/ nodeProxy( MathNode, MathNode.SIN );export const cos = /*@__PURE__*/ nodeProxy( MathNode, MathNode.COS );export const tan = /*@__PURE__*/ nodeProxy( MathNode, MathNode.TAN );export const asin = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ASIN );export const acos = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ACOS );export const atan = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ATAN );export const abs = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ABS );export const sign = /*@__PURE__*/ nodeProxy( MathNode, MathNode.SIGN );export const length = /*@__PURE__*/ nodeProxy( MathNode, MathNode.LENGTH );export const negate = /*@__PURE__*/ nodeProxy( MathNode, MathNode.NEGATE );export const oneMinus = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ONE_MINUS );export const dFdx = /*@__PURE__*/ nodeProxy( MathNode, MathNode.DFDX );export const dFdy = /*@__PURE__*/ nodeProxy( MathNode, MathNode.DFDY );export const round = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ROUND );export const reciprocal = /*@__PURE__*/ nodeProxy( MathNode, MathNode.RECIPROCAL );export const trunc = /*@__PURE__*/ nodeProxy( MathNode, MathNode.TRUNC );export const fwidth = /*@__PURE__*/ nodeProxy( MathNode, MathNode.FWIDTH );export const bitcast = /*@__PURE__*/ nodeProxy( MathNode, MathNode.BITCAST );export const transpose = /*@__PURE__*/ nodeProxy( MathNode, MathNode.TRANSPOSE );
export const atan2 = /*@__PURE__*/ nodeProxy( MathNode, MathNode.ATAN2 );export const min = /*@__PURE__*/ nodeProxy( MathNode, MathNode.MIN );export const max = /*@__PURE__*/ nodeProxy( MathNode, MathNode.MAX );export const mod = /*@__PURE__*/ nodeProxy( MathNode, MathNode.MOD );export const step = /*@__PURE__*/ nodeProxy( MathNode, MathNode.STEP );export const reflect = /*@__PURE__*/ nodeProxy( MathNode, MathNode.REFLECT );export const distance = /*@__PURE__*/ nodeProxy( MathNode, MathNode.DISTANCE );export const difference = /*@__PURE__*/ nodeProxy( MathNode, MathNode.DIFFERENCE );export const dot = /*@__PURE__*/ nodeProxy( MathNode, MathNode.DOT );export const cross = /*@__PURE__*/ nodeProxy( MathNode, MathNode.CROSS );export const pow = /*@__PURE__*/ nodeProxy( MathNode, MathNode.POW );export const pow2 = /*@__PURE__*/ nodeProxy( MathNode, MathNode.POW, 2 );export const pow3 = /*@__PURE__*/ nodeProxy( MathNode, MathNode.POW, 3 );export const pow4 = /*@__PURE__*/ nodeProxy( MathNode, MathNode.POW, 4 );export const transformDirection = /*@__PURE__*/ nodeProxy( MathNode, MathNode.TRANSFORM_DIRECTION );
export const cbrt = ( a ) => mul( sign( a ), pow( abs( a ), 1.0 / 3.0 ) );export const lengthSq = ( a ) => dot( a, a );export const mix = /*@__PURE__*/ nodeProxy( MathNode, MathNode.MIX );export const clamp = ( value, low = 0, high = 1 ) => nodeObject( new MathNode( MathNode.CLAMP, nodeObject( value ), nodeObject( low ), nodeObject( high ) ) );export const saturate = ( value ) => clamp( value );export const refract = /*@__PURE__*/ nodeProxy( MathNode, MathNode.REFRACT );export const smoothstep = /*@__PURE__*/ nodeProxy( MathNode, MathNode.SMOOTHSTEP );export const faceForward = /*@__PURE__*/ nodeProxy( MathNode, MathNode.FACEFORWARD );
export const rand = /*@__PURE__*/ Fn( ( [ uv ] ) => {
const a = 12.9898, b = 78.233, c = 43758.5453; const dt = dot( uv.xy, vec2( a, b ) ), sn = mod( dt, PI );
return fract( sin( sn ).mul( c ) );
} );
export const mixElement = ( t, e1, e2 ) => mix( e1, e2, t );export const smoothstepElement = ( x, low, high ) => smoothstep( low, high, x );
addMethodChaining( 'all', all );addMethodChaining( 'any', any );addMethodChaining( 'equals', equals );
addMethodChaining( 'radians', radians );addMethodChaining( 'degrees', degrees );addMethodChaining( 'exp', exp );addMethodChaining( 'exp2', exp2 );addMethodChaining( 'log', log );addMethodChaining( 'log2', log2 );addMethodChaining( 'sqrt', sqrt );addMethodChaining( 'inverseSqrt', inverseSqrt );addMethodChaining( 'floor', floor );addMethodChaining( 'ceil', ceil );addMethodChaining( 'normalize', normalize );addMethodChaining( 'fract', fract );addMethodChaining( 'sin', sin );addMethodChaining( 'cos', cos );addMethodChaining( 'tan', tan );addMethodChaining( 'asin', asin );addMethodChaining( 'acos', acos );addMethodChaining( 'atan', atan );addMethodChaining( 'abs', abs );addMethodChaining( 'sign', sign );addMethodChaining( 'length', length );addMethodChaining( 'lengthSq', lengthSq );addMethodChaining( 'negate', negate );addMethodChaining( 'oneMinus', oneMinus );addMethodChaining( 'dFdx', dFdx );addMethodChaining( 'dFdy', dFdy );addMethodChaining( 'round', round );addMethodChaining( 'reciprocal', reciprocal );addMethodChaining( 'trunc', trunc );addMethodChaining( 'fwidth', fwidth );addMethodChaining( 'atan2', atan2 );addMethodChaining( 'min', min );addMethodChaining( 'max', max );addMethodChaining( 'mod', mod );addMethodChaining( 'step', step );addMethodChaining( 'reflect', reflect );addMethodChaining( 'distance', distance );addMethodChaining( 'dot', dot );addMethodChaining( 'cross', cross );addMethodChaining( 'pow', pow );addMethodChaining( 'pow2', pow2 );addMethodChaining( 'pow3', pow3 );addMethodChaining( 'pow4', pow4 );addMethodChaining( 'transformDirection', transformDirection );addMethodChaining( 'mix', mixElement );addMethodChaining( 'clamp', clamp );addMethodChaining( 'refract', refract );addMethodChaining( 'smoothstep', smoothstepElement );addMethodChaining( 'faceForward', faceForward );addMethodChaining( 'difference', difference );addMethodChaining( 'saturate', saturate );addMethodChaining( 'cbrt', cbrt );addMethodChaining( 'transpose', transpose );addMethodChaining( 'rand', rand );