three.js/examples/js/nodes/BuilderNode.js

/**
 * @author sunag / http://www.sunag.com.br/
 */

THREE.BuilderNode = function( material ) {

	this.material = material;

	this.caches = [];

	this.parsing = false;
	this.optimize = true;

	this.addCache();

};

THREE.BuilderNode.prototype = {
	constructor: THREE.BuilderNode,

	addCache : function( name, requires ) {

		this.caches.push( {
			name : name || '',
			requires : requires || {}
		} );

		return this.updateCache();

	},

	removeCache : function() {

		this.caches.pop();

		return this.updateCache();

	},

	isCache : function( name ) {

		var i = this.caches.length;

		while ( i -- ) {

			if ( this.caches[ i ].name == name ) return true;

		}

		return false;

	},

	updateCache : function() {

		var cache = this.caches[ this.caches.length - 1 ];

		this.cache = cache.name;
		this.requires = cache.requires;

		return this;

	},

	require : function( name, node ) {

		this.requires[ name ] = node;

		return this;

	},

	include : function( func ) {

		this.material.include( this.shader, func );

		return this;

	},

	colorToVector : function( color ) {

		return color.replace( 'r', 'x' ).replace( 'g', 'y' ).replace( 'b', 'z' ).replace( 'a', 'w' );

	},

	getFormatConstructor : function( len ) {

		return THREE.BuilderNode.constructors[ len - 1 ];

	},

	getFormatName : function( format ) {

		return format.replace( /c/g, 'v3' ).replace( /fv1|iv1/g, 'v1' );

	},

	isFormatMatrix : function( format ) {

		return /^m/.test( format );

	},

	getFormatLength : function( format ) {

		return parseInt( this.getFormatName( format ).substr( 1 ) );

	},

	getFormatByLength : function( len ) {

		if ( len == 1 ) return 'fv1';

		return 'v' + len;

	},

	format : function( code, from, to ) {

		var format = this.getFormatName( from + '=' + to );

		switch ( format ) {
			case 'v1=v2': return 'vec2(' + code + ')';
			case 'v1=v3': return 'vec3(' + code + ')';
			case 'v1=v4': return 'vec4(' + code + ')';

			case 'v2=v1': return code + '.x';
			case 'v2=v3': return 'vec3(' + code + ',0.0)';
			case 'v2=v4': return 'vec4(' + code + ',0.0,1.0)';

			case 'v3=v1': return code + '.x';
			case 'v3=v2': return code + '.xy';
			case 'v3=v4': return 'vec4(' + code + ',1.0)';

			case 'v4=v1': return code + '.x';
			case 'v4=v2': return code + '.xy';
			case 'v4=v3': return code + '.xyz';
		}

		return code;

	},

	getTypeByFormat : function( format ) {

		return THREE.BuilderNode.type[ format ];

	},

	getUuid : function( uuid, useCache ) {

		useCache = useCache !== undefined ? useCache : true;

		if ( useCache && this.cache ) uuid = this.cache + '-' + uuid;

		return uuid;

	},

	getElementByIndex : function( index ) {

		return THREE.BuilderNode.elements[ index ];

	},

	getIndexByElement : function( elm ) {

		return THREE.BuilderNode.elements.indexOf( elm );

	},

	isShader : function( shader ) {

		return this.shader == shader;

	},

	setShader : function( shader ) {

		this.shader = shader;

		return this;

	}
};

THREE.BuilderNode.type = {
	'float' : 'fv1',
	vec2 : 'v2',
	vec3 : 'v3',
	vec4 : 'v4',
	mat4 : 'v4'
};

THREE.BuilderNode.constructors = [
	'',
	'vec2',
	'vec3',
	'vec4'
];

THREE.BuilderNode.elements = [
	'x',
	'y',
	'z',
	'w'
];