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.
 
 
 

116 lines
2.8 KiB

import { InputNode } from '../core/InputNode.js';
import { UVNode } from '../accessors/UVNode.js';
import { ColorSpaceNode } from '../utils/ColorSpaceNode.js';
import { ExpressionNode } from '../core/ExpressionNode.js';
class TextureNode extends InputNode {
constructor( value, uv, bias, project ) {
super( 'v4', { shared: true } );
this.value = value;
this.uv = uv || new UVNode();
this.bias = bias;
this.project = project !== undefined ? project : false;
}
getTexture( builder, output ) {
return super.generate( builder, output, this.value.uuid, 't' );
}
generate( builder, output ) {
if ( output === 'sampler2D' ) {
return this.getTexture( builder, output );
}
const tex = this.getTexture( builder, output ),
uv = this.uv.build( builder, this.project ? 'v4' : 'v2' );
let bias = this.bias ? this.bias.build( builder, 'f' ) : undefined;
if ( bias === undefined && builder.context.bias ) {
bias = builder.context.bias.setTexture( this ).build( builder, 'f' );
}
let method, code;
if ( this.project ) method = 'texture2DProj';
else method = bias ? 'tex2DBias' : 'tex2D';
if ( bias ) code = method + '( ' + tex + ', ' + uv + ', ' + bias + ' )';
else code = method + '( ' + tex + ', ' + uv + ' )';
// add a custom context for fix incompatibility with the core
// include ColorSpace function only for vertex shader (in fragment shader color space functions is added automatically by core)
// this should be removed in the future
// context.include is used to include or not functions if used FunctionNode
// context.ignoreCache =: not create temp variables nodeT0..9 to optimize the code
const context = { include: builder.isShader( 'vertex' ), ignoreCache: true };
const outputType = this.getType( builder );
builder.addContext( context );
this.colorSpace = this.colorSpace || new ColorSpaceNode( new ExpressionNode( '', outputType ) );
this.colorSpace.fromDecoding( builder.getTextureEncodingFromMap( this.value ) );
this.colorSpace.input.parse( code );
code = this.colorSpace.build( builder, outputType );
// end custom context
builder.removeContext();
return builder.format( code, outputType, output );
}
copy( source ) {
super.copy( source );
if ( source.value ) this.value = source.value;
this.uv = source.uv;
if ( source.bias ) this.bias = source.bias;
if ( source.project !== undefined ) this.project = source.project;
return this;
}
toJSON( meta ) {
let data = this.getJSONNode( meta );
if ( ! data ) {
data = this.createJSONNode( meta );
if ( this.value ) data.value = this.value.uuid;
data.uv = this.uv.toJSON( meta ).uuid;
data.project = this.project;
if ( this.bias ) data.bias = this.bias.toJSON( meta ).uuid;
}
return data;
}
}
TextureNode.prototype.nodeType = 'Texture';
export { TextureNode };