import { TempNode } from '../core/TempNode.js';
import { FloatNode } from '../inputs/FloatNode.js';
import { FunctionNode } from '../core/FunctionNode.js';
import { NormalNode } from '../accessors/NormalNode.js';
import { PositionNode } from '../accessors/PositionNode.js';

class BumpMapNode extends TempNode {

	constructor( value, scale ) {

		super( 'v3' );

		this.value = value;
		this.scale = scale || new FloatNode( 1 );

		this.toNormalMap = false;

	}

	generate( builder, output ) {

		if ( builder.isShader( 'fragment' ) ) {

			if ( this.toNormalMap ) {

				const bumpToNormal = builder.include( BumpMapNode.Nodes.bumpToNormal );

				return builder.format( bumpToNormal + '( ' + this.value.build( builder, 'sampler2D' ) + ', ' +
					this.value.uv.build( builder, 'v2' ) + ', ' +
					this.scale.build( builder, 'f' ) + ' )', this.getType( builder ), output );

			} else {

				const derivativeHeight = builder.include( BumpMapNode.Nodes.dHdxy_fwd ),
					perturbNormalArb = builder.include( BumpMapNode.Nodes.perturbNormalArb );

				this.normal = this.normal || new NormalNode();
				this.position = this.position || new PositionNode( PositionNode.VIEW );

				const derivativeHeightCode = derivativeHeight + '( ' + this.value.build( builder, 'sampler2D' ) + ', ' +
					this.value.uv.build( builder, 'v2' ) + ', ' +
					this.scale.build( builder, 'f' ) + ' )';

				return builder.format( perturbNormalArb + '( -' + this.position.build( builder, 'v3' ) + ', ' +
					this.normal.build( builder, 'v3' ) + ', ' +
					derivativeHeightCode + ' )', this.getType( builder ), output );

			}

		} else {

			console.warn( 'THREE.BumpMapNode is not compatible with ' + builder.shader + ' shader.' );

			return builder.format( 'vec3( 0.0 )', this.getType( builder ), output );

		}

	}

	copy( source ) {

		super.copy( source );

		this.value = source.value;
		this.scale = source.scale;

		return this;

	}

	toJSON( meta ) {

		let data = this.getJSONNode( meta );

		if ( ! data ) {

			data = this.createJSONNode( meta );

			data.value = this.value.toJSON( meta ).uuid;
			data.scale = this.scale.toJSON( meta ).uuid;

		}

		return data;

	}

}

BumpMapNode.Nodes = ( function () {

	const dHdxy_fwd = new FunctionNode( [

		// Bump Mapping Unparametrized Surfaces on the GPU by Morten S. Mikkelsen
		// http://api.unrealengine.com/attachments/Engine/Rendering/LightingAndShadows/BumpMappingWithoutTangentSpace/mm_sfgrad_bump.pdf

		// Evaluate the derivative of the height w.r.t. screen-space using forward differencing (listing 2)

		'vec2 dHdxy_fwd( sampler2D bumpMap, vec2 vUv, float bumpScale ) {',

		// Workaround for Adreno 3XX dFd*( vec3 ) bug. See #9988

		'	vec2 dSTdx = dFdx( vUv );',
		'	vec2 dSTdy = dFdy( vUv );',

		'	float Hll = bumpScale * texture2D( bumpMap, vUv ).x;',
		'	float dBx = bumpScale * texture2D( bumpMap, vUv + dSTdx ).x - Hll;',
		'	float dBy = bumpScale * texture2D( bumpMap, vUv + dSTdy ).x - Hll;',

		'	return vec2( dBx, dBy );',

		'}'

	].join( '\n' ), null, { derivatives: true } );

	const perturbNormalArb = new FunctionNode( [

		'vec3 perturbNormalArb( vec3 surf_pos, vec3 surf_norm, vec2 dHdxy ) {',

		// Workaround for Adreno 3XX dFd*( vec3 ) bug. See #9988

		'	vec3 vSigmaX = vec3( dFdx( surf_pos.x ), dFdx( surf_pos.y ), dFdx( surf_pos.z ) );',
		'	vec3 vSigmaY = vec3( dFdy( surf_pos.x ), dFdy( surf_pos.y ), dFdy( surf_pos.z ) );',
		'	vec3 vN = surf_norm;', // normalized

		'	vec3 R1 = cross( vSigmaY, vN );',
		'	vec3 R2 = cross( vN, vSigmaX );',

		'	float fDet = dot( vSigmaX, R1 );',

		'	fDet *= ( float( gl_FrontFacing ) * 2.0 - 1.0 );',

		'	vec3 vGrad = sign( fDet ) * ( dHdxy.x * R1 + dHdxy.y * R2 );',

		'	return normalize( abs( fDet ) * surf_norm - vGrad );',

		'}'

	].join( '\n' ), [ dHdxy_fwd ], { derivatives: true } );

	const bumpToNormal = new FunctionNode( [
		'vec3 bumpToNormal( sampler2D bumpMap, vec2 uv, float scale ) {',

		'	vec2 dSTdx = dFdx( uv );',
		'	vec2 dSTdy = dFdy( uv );',

		'	float Hll = texture2D( bumpMap, uv ).x;',
		'	float dBx = texture2D( bumpMap, uv + dSTdx ).x - Hll;',
		'	float dBy = texture2D( bumpMap, uv + dSTdy ).x - Hll;',

		'	return vec3( .5 - ( dBx * scale ), .5 - ( dBy * scale ), 1.0 );',

		'}'
	].join( '\n' ), null, { derivatives: true } );

	return {
		dHdxy_fwd: dHdxy_fwd,
		perturbNormalArb: perturbNormalArb,
		bumpToNormal: bumpToNormal
	};

} )();

BumpMapNode.prototype.nodeType = 'BumpMap';
BumpMapNode.prototype.hashProperties = [ 'toNormalMap' ];

export { BumpMapNode };