three 基础库
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

149 lines
3.0 KiB

import { TempNode } from '../core/TempNode.js';
import { FunctionNode } from '../core/FunctionNode.js';
import { FunctionCallNode } from '../core/FunctionCallNode.js';
import { FloatNode } from '../inputs/FloatNode.js';
import { UVNode } from '../accessors/UVNode.js';
const NOISE_COMMON_SRC = `
vec3 mod289( vec3 x ) {
return x - floor( x * ( 1.0 / 289.0 ) ) * 289.0;
}
vec4 mod289( vec4 x ) {
return x - floor( x * ( 1.0 / 289.0 ) ) * 289.0;
}
vec4 permute( vec4 x ) {
return mod289( ( ( x * 34.0 ) + 1.0 ) * x );
}
vec4 taylorInvSqrt( vec4 r ) {
return 1.79284291400159 - 0.85373472095314 * r;
}
vec2 fade( vec2 t ) {
return t * t * t * ( t * ( t * 6.0 - 15.0 ) + 10.0 );
}
vec3 fade( vec3 t ) {
return t * t * t * ( t * ( t * 6.0 - 15.0 ) + 10.0 );
}
`.trim();
const NOISE2D_SRC = `
float noise2d( vec2 P, float amplitude, float pivot ) {
vec4 Pi = floor(P.xyxy) + vec4(0.0, 0.0, 1.0, 1.0);
vec4 Pf = fract(P.xyxy) - vec4(0.0, 0.0, 1.0, 1.0);
Pi = mod289(Pi); // To avoid truncation effects in permutation
vec4 ix = Pi.xzxz;
vec4 iy = Pi.yyww;
vec4 fx = Pf.xzxz;
vec4 fy = Pf.yyww;
vec4 i = permute(permute(ix) + iy);
vec4 gx = fract(i * (1.0 / 41.0)) * 2.0 - 1.0 ;
vec4 gy = abs(gx) - 0.5 ;
vec4 tx = floor(gx + 0.5);
gx = gx - tx;
vec2 g00 = vec2(gx.x,gy.x);
vec2 g10 = vec2(gx.y,gy.y);
vec2 g01 = vec2(gx.z,gy.z);
vec2 g11 = vec2(gx.w,gy.w);
vec4 norm = taylorInvSqrt(vec4(dot(g00, g00), dot(g01, g01), dot(g10, g10), dot(g11, g11)));
g00 *= norm.x;
g01 *= norm.y;
g10 *= norm.z;
g11 *= norm.w;
float n00 = dot(g00, vec2(fx.x, fy.x));
float n10 = dot(g10, vec2(fx.y, fy.y));
float n01 = dot(g01, vec2(fx.z, fy.z));
float n11 = dot(g11, vec2(fx.w, fy.w));
vec2 fade_xy = fade(Pf.xy);
vec2 n_x = mix(vec2(n00, n01), vec2(n10, n11), fade_xy.x);
float n_xy = mix(n_x.x, n_x.y, fade_xy.y);
return 2.3 * n_xy * amplitude + pivot;
}
`.trim();
class Noise2DNode extends TempNode {
constructor( uv = new UVNode(), amplitude = new FloatNode( 1.0 ), pivot = new FloatNode( 0.0 ) ) {
super( 'f' );
this.uv = uv;
this.amplitude = amplitude;
this.pivot = pivot;
}
generate( builder, output ) {
const noise2d = new FunctionCallNode( Noise2DNode.Nodes.noise2d, [ this.uv, this.amplitude, this.pivot ] );
return builder.format( noise2d.generate( builder, output ), this.getType( builder ), output );
}
copy( source ) {
super.copy( source );
this.uv = source.uv;
this.amplitude = source.amplitude;
this.pivot = source.pivot;
}
toJSON( meta ) {
let data = this.getJSONNode( meta );
if ( ! data ) {
data = this.createJSONNode( meta );
data.uv = this.uv.toJSON( meta ).uuid;
data.amplitude = this.amplitude.toJSON( meta ).uuid;
data.pivot = this.pivot.toJSON( meta ).uuid;
}
return data;
}
}
Noise2DNode.prototype.nodeType = 'Noise2D';
Noise2DNode.Nodes = ( function () {
const noiseCommon = new FunctionNode( NOISE_COMMON_SRC );
const noise2d = new FunctionNode( NOISE2D_SRC );
noise2d.includes = [ noiseCommon ];
return { noiseCommon, noise2d };
} )();
export { Noise2DNode };