/// /// module BABYLON { var maxSimultaneousLights = 4; class WaterMaterialDefines extends MaterialDefines { public BUMP = false; public REFLECTION = false; public CLIPPLANE = false; public ALPHATEST = false; public POINTSIZE = false; public FOG = false; public LIGHT0 = false; public LIGHT1 = false; public LIGHT2 = false; public LIGHT3 = false; public SPOTLIGHT0 = false; public SPOTLIGHT1 = false; public SPOTLIGHT2 = false; public SPOTLIGHT3 = false; public HEMILIGHT0 = false; public HEMILIGHT1 = false; public HEMILIGHT2 = false; public HEMILIGHT3 = false; public DIRLIGHT0 = false; public DIRLIGHT1 = false; public DIRLIGHT2 = false; public DIRLIGHT3 = false; public POINTLIGHT0 = false; public POINTLIGHT1 = false; public POINTLIGHT2 = false; public POINTLIGHT3 = false; public SHADOW0 = false; public SHADOW1 = false; public SHADOW2 = false; public SHADOW3 = false; public SHADOWS = false; public SHADOWVSM0 = false; public SHADOWVSM1 = false; public SHADOWVSM2 = false; public SHADOWVSM3 = false; public SHADOWPCF0 = false; public SHADOWPCF1 = false; public SHADOWPCF2 = false; public SHADOWPCF3 = false; public NORMAL = false; public UV1 = false; public UV2 = false; public VERTEXCOLOR = false; public VERTEXALPHA = false; public BONES = false; public BONES4 = false; public BonesPerMesh = 0; public INSTANCES = false; public SPECULARTERM = false; constructor() { super(); this._keys = Object.keys(this); } } export class WaterMaterial extends Material { /* * Public members */ @serializeAsTexture() public bumpTexture: BaseTexture; @serializeAsColor3() public diffuseColor = new Color3(1, 1, 1); @serializeAsColor3() public specularColor = new Color3(0, 0, 0); @serialize() public specularPower = 64; @serialize() public disableLighting = false; /** * @param {number}: Represents the wind force */ @serialize() public windForce: number = 6; /** * @param {Vector2}: The direction of the wind in the plane (X, Z) */ @serializeAsVector2() public windDirection: Vector2 = new Vector2(0, 1); /** * @param {number}: Wave height, represents the height of the waves */ @serialize() public waveHeight: number = 0.4; /** * @param {number}: Bump height, represents the bump height related to the bump map */ @serialize() public bumpHeight: number = 0.4; /** * @param {number}: The water color blended with the reflection and refraction samplers */ @serializeAsColor3() public waterColor: Color3 = new Color3(0.1, 0.1, 0.6); /** * @param {number}: The blend factor related to the water color */ @serialize() public colorBlendFactor: number = 0.2; /** * @param {number}: Represents the maximum length of a wave */ @serialize() public waveLength: number = 0.1; /** * @param {number}: Defines the waves speed */ @serialize() public waveSpeed: number = 1.0; /* * Private members */ private _mesh: AbstractMesh = null; private _refractionRTT: RenderTargetTexture; private _reflectionRTT: RenderTargetTexture; private _material: ShaderMaterial; private _reflectionTransform: Matrix = Matrix.Zero(); private _lastTime: number = 0; private _scaledDiffuse = new Color3(); private _scaledSpecular = new Color3(); private _renderId: number; private _defines = new WaterMaterialDefines(); private _cachedDefines = new WaterMaterialDefines(); /** * Constructor */ constructor(name: string, scene: Scene, public renderTargetSize: Vector2 = new Vector2(512, 512)) { super(name, scene); // Create render targets this._createRenderTargets(scene, renderTargetSize); } // Get / Set public get refractionTexture(): RenderTargetTexture { return this._refractionRTT; } public get reflectionTexture(): RenderTargetTexture { return this._reflectionRTT; } // Methods public addToRenderList(node: any): void { this._refractionRTT.renderList.push(node); this._reflectionRTT.renderList.push(node); } public enableRenderTargets(enable: boolean): void { var refreshRate = enable ? 1 : 0; this._refractionRTT.refreshRate = refreshRate; this._reflectionRTT.refreshRate = refreshRate; } public needAlphaBlending(): boolean { return (this.alpha < 1.0); } public needAlphaTesting(): boolean { return false; } public getAlphaTestTexture(): BaseTexture { return null; } private _checkCache(scene: Scene, mesh?: AbstractMesh, useInstances?: boolean): boolean { if (!mesh) { return true; } if (this._defines.INSTANCES !== useInstances) { return false; } if (mesh._materialDefines && mesh._materialDefines.isEqual(this._defines)) { return true; } return false; } public isReady(mesh?: AbstractMesh, useInstances?: boolean): boolean { if (this.checkReadyOnlyOnce) { if (this._wasPreviouslyReady) { return true; } } var scene = this.getScene(); if (!this.checkReadyOnEveryCall) { if (this._renderId === scene.getRenderId()) { if (this._checkCache(scene, mesh, useInstances)) { return true; } } } var engine = scene.getEngine(); var needNormals = false; var needUVs = false; this._defines.reset(); // Textures if (scene.texturesEnabled) { if (this.bumpTexture && StandardMaterial.BumpTextureEnabled) { if (!this.bumpTexture.isReady()) { return false; } else { needUVs = true; this._defines.BUMP = true; } } if (StandardMaterial.ReflectionTextureEnabled) { this._defines.REFLECTION = true; } } // Effect if (scene.clipPlane) { this._defines.CLIPPLANE = true; } if (engine.getAlphaTesting()) { this._defines.ALPHATEST = true; } // Point size if (this.pointsCloud || scene.forcePointsCloud) { this._defines.POINTSIZE = true; } // Fog if (scene.fogEnabled && mesh && mesh.applyFog && scene.fogMode !== Scene.FOGMODE_NONE && this.fogEnabled) { this._defines.FOG = true; } var lightIndex = 0; if (scene.lightsEnabled && !this.disableLighting) { for (var index = 0; index < scene.lights.length; index++) { var light = scene.lights[index]; if (!light.isEnabled()) { continue; } // Excluded check if (light._excludedMeshesIds.length > 0) { for (var excludedIndex = 0; excludedIndex < light._excludedMeshesIds.length; excludedIndex++) { var excludedMesh = scene.getMeshByID(light._excludedMeshesIds[excludedIndex]); if (excludedMesh) { light.excludedMeshes.push(excludedMesh); } } light._excludedMeshesIds = []; } // Included check if (light._includedOnlyMeshesIds.length > 0) { for (var includedOnlyIndex = 0; includedOnlyIndex < light._includedOnlyMeshesIds.length; includedOnlyIndex++) { var includedOnlyMesh = scene.getMeshByID(light._includedOnlyMeshesIds[includedOnlyIndex]); if (includedOnlyMesh) { light.includedOnlyMeshes.push(includedOnlyMesh); } } light._includedOnlyMeshesIds = []; } if (!light.canAffectMesh(mesh)) { continue; } needNormals = true; this._defines["LIGHT" + lightIndex] = true; var type; if (light instanceof SpotLight) { type = "SPOTLIGHT" + lightIndex; } else if (light instanceof HemisphericLight) { type = "HEMILIGHT" + lightIndex; } else if (light instanceof PointLight) { type = "POINTLIGHT" + lightIndex; } else { type = "DIRLIGHT" + lightIndex; } this._defines[type] = true; // Specular if (!light.specular.equalsFloats(0, 0, 0)) { this._defines.SPECULARTERM = true; } // Shadows if (scene.shadowsEnabled) { var shadowGenerator = light.getShadowGenerator(); if (mesh && mesh.receiveShadows && shadowGenerator) { this._defines["SHADOW" + lightIndex] = true; this._defines.SHADOWS = true; if (shadowGenerator.useVarianceShadowMap || shadowGenerator.useBlurVarianceShadowMap) { this._defines["SHADOWVSM" + lightIndex] = true; } if (shadowGenerator.usePoissonSampling) { this._defines["SHADOWPCF" + lightIndex] = true; } } } lightIndex++; if (lightIndex === maxSimultaneousLights) break; } } // Attribs if (mesh) { if (needNormals && mesh.isVerticesDataPresent(VertexBuffer.NormalKind)) { this._defines.NORMAL = true; } if (needUVs) { if (mesh.isVerticesDataPresent(VertexBuffer.UVKind)) { this._defines.UV1 = true; } if (mesh.isVerticesDataPresent(VertexBuffer.UV2Kind)) { this._defines.UV2 = true; } } if (mesh.useVertexColors && mesh.isVerticesDataPresent(VertexBuffer.ColorKind)) { this._defines.VERTEXCOLOR = true; if (mesh.hasVertexAlpha) { this._defines.VERTEXALPHA = true; } } if (mesh.useBones && mesh.computeBonesUsingShaders) { this._defines.BONES = true; this._defines.BonesPerMesh = (mesh.skeleton.bones.length + 1); this._defines.BONES4 = true; } // Instances if (useInstances) { this._defines.INSTANCES = true; } } this._mesh = mesh; // Get correct effect if (!this._defines.isEqual(this._cachedDefines)) { this._defines.cloneTo(this._cachedDefines); scene.resetCachedMaterial(); // Fallbacks var fallbacks = new EffectFallbacks(); if (this._defines.FOG) { fallbacks.addFallback(1, "FOG"); } for (lightIndex = 0; lightIndex < maxSimultaneousLights; lightIndex++) { if (!this._defines["LIGHT" + lightIndex]) { continue; } if (lightIndex > 0) { fallbacks.addFallback(lightIndex, "LIGHT" + lightIndex); } if (this._defines["SHADOW" + lightIndex]) { fallbacks.addFallback(0, "SHADOW" + lightIndex); } if (this._defines["SHADOWPCF" + lightIndex]) { fallbacks.addFallback(0, "SHADOWPCF" + lightIndex); } if (this._defines["SHADOWVSM" + lightIndex]) { fallbacks.addFallback(0, "SHADOWVSM" + lightIndex); } } if (this._defines.BONES4) { fallbacks.addFallback(0, "BONES4"); } //Attributes var attribs = [VertexBuffer.PositionKind]; if (this._defines.NORMAL) { attribs.push(VertexBuffer.NormalKind); } if (this._defines.UV1) { attribs.push(VertexBuffer.UVKind); } if (this._defines.UV2) { attribs.push(VertexBuffer.UV2Kind); } if (this._defines.VERTEXCOLOR) { attribs.push(VertexBuffer.ColorKind); } if (this._defines.BONES) { attribs.push(VertexBuffer.MatricesIndicesKind); attribs.push(VertexBuffer.MatricesWeightsKind); } if (this._defines.INSTANCES) { attribs.push("world0"); attribs.push("world1"); attribs.push("world2"); attribs.push("world3"); } // Legacy browser patch var shaderName = "water"; var join = this._defines.toString(); this._effect = scene.getEngine().createEffect(shaderName, attribs, ["world", "view", "viewProjection", "vEyePosition", "vLightsType", "vDiffuseColor", "vSpecularColor", "vLightData0", "vLightDiffuse0", "vLightSpecular0", "vLightDirection0", "vLightGround0", "lightMatrix0", "vLightData1", "vLightDiffuse1", "vLightSpecular1", "vLightDirection1", "vLightGround1", "lightMatrix1", "vLightData2", "vLightDiffuse2", "vLightSpecular2", "vLightDirection2", "vLightGround2", "lightMatrix2", "vLightData3", "vLightDiffuse3", "vLightSpecular3", "vLightDirection3", "vLightGround3", "lightMatrix3", "vFogInfos", "vFogColor", "pointSize", "vNormalInfos", "mBones", "vClipPlane", "normalMatrix", "shadowsInfo0", "shadowsInfo1", "shadowsInfo2", "shadowsInfo3", // Water "worldReflectionViewProjection", "windDirection", "waveLength", "time", "windForce", "cameraPosition", "bumpHeight", "waveHeight", "waterColor", "colorBlendFactor", "waveSpeed" ], ["normalSampler", "shadowSampler0", "shadowSampler1", "shadowSampler2", "shadowSampler3", // Water "refractionSampler", "reflectionSampler" ], join, fallbacks, this.onCompiled, this.onError); } if (!this._effect.isReady()) { return false; } this._renderId = scene.getRenderId(); this._wasPreviouslyReady = true; if (mesh) { if (!mesh._materialDefines) { mesh._materialDefines = new WaterMaterialDefines(); } this._defines.cloneTo(mesh._materialDefines); } return true; } public bindOnlyWorldMatrix(world: Matrix): void { this._effect.setMatrix("world", world); } public bind(world: Matrix, mesh?: Mesh): void { var scene = this.getScene(); // Matrices this.bindOnlyWorldMatrix(world); this._effect.setMatrix("viewProjection", scene.getTransformMatrix()); // Bones if (mesh && mesh.useBones && mesh.computeBonesUsingShaders) { this._effect.setMatrices("mBones", mesh.skeleton.getTransformMatrices(mesh)); } if (scene.getCachedMaterial() !== this) { // Textures if (this.bumpTexture && StandardMaterial.BumpTextureEnabled) { this._effect.setTexture("normalSampler", this.bumpTexture); this._effect.setFloat2("vNormalInfos", this.bumpTexture.coordinatesIndex, this.bumpTexture.level); this._effect.setMatrix("normalMatrix", this.bumpTexture.getTextureMatrix()); } // Clip plane if (scene.clipPlane) { var clipPlane = scene.clipPlane; this._effect.setFloat4("vClipPlane", clipPlane.normal.x, clipPlane.normal.y, clipPlane.normal.z, clipPlane.d); } // Point size if (this.pointsCloud) { this._effect.setFloat("pointSize", this.pointSize); } this._effect.setVector3("vEyePosition", scene._mirroredCameraPosition ? scene._mirroredCameraPosition : scene.activeCamera.position); } this._effect.setColor4("vDiffuseColor", this._scaledDiffuse, this.alpha * mesh.visibility); if (this._defines.SPECULARTERM) { this._effect.setColor4("vSpecularColor", this.specularColor, this.specularPower); } if (scene.lightsEnabled && !this.disableLighting) { var lightIndex = 0; for (var index = 0; index < scene.lights.length; index++) { var light = scene.lights[index]; if (!light.isEnabled()) { continue; } if (!light.canAffectMesh(mesh)) { continue; } if (light instanceof PointLight) { // Point Light light.transferToEffect(this._effect, "vLightData" + lightIndex); } else if (light instanceof DirectionalLight) { // Directional Light light.transferToEffect(this._effect, "vLightData" + lightIndex); } else if (light instanceof SpotLight) { // Spot Light light.transferToEffect(this._effect, "vLightData" + lightIndex, "vLightDirection" + lightIndex); } else if (light instanceof HemisphericLight) { // Hemispheric Light light.transferToEffect(this._effect, "vLightData" + lightIndex, "vLightGround" + lightIndex); } light.diffuse.scaleToRef(light.intensity, this._scaledDiffuse); this._effect.setColor4("vLightDiffuse" + lightIndex, this._scaledDiffuse, light.range); if (this._defines.SPECULARTERM) { light.specular.scaleToRef(light.intensity, this._scaledSpecular); this._effect.setColor3("vLightSpecular" + lightIndex, this._scaledSpecular); } // Shadows if (scene.shadowsEnabled) { var shadowGenerator = light.getShadowGenerator(); if (mesh.receiveShadows && shadowGenerator) { this._effect.setMatrix("lightMatrix" + lightIndex, shadowGenerator.getTransformMatrix()); this._effect.setTexture("shadowSampler" + lightIndex, shadowGenerator.getShadowMapForRendering()); this._effect.setFloat3("shadowsInfo" + lightIndex, shadowGenerator.getDarkness(), shadowGenerator.getShadowMap().getSize().width, shadowGenerator.bias); } } lightIndex++; if (lightIndex === maxSimultaneousLights) break; } } // View if (scene.fogEnabled && mesh.applyFog && scene.fogMode !== Scene.FOGMODE_NONE) { this._effect.setMatrix("view", scene.getViewMatrix()); } // Fog if (scene.fogEnabled && mesh.applyFog && scene.fogMode !== Scene.FOGMODE_NONE) { this._effect.setFloat4("vFogInfos", scene.fogMode, scene.fogStart, scene.fogEnd, scene.fogDensity); this._effect.setColor3("vFogColor", scene.fogColor); } // Water if (StandardMaterial.ReflectionTextureEnabled) { this._effect.setTexture("refractionSampler", this._refractionRTT); this._effect.setTexture("reflectionSampler", this._reflectionRTT); } var wrvp = this._mesh.getWorldMatrix().multiply(this._reflectionTransform).multiply(scene.getProjectionMatrix()); this._lastTime += scene.getEngine().getDeltaTime(); this._effect.setMatrix("worldReflectionViewProjection", wrvp); this._effect.setVector2("windDirection", this.windDirection); this._effect.setFloat("waveLength", this.waveLength); this._effect.setFloat("time", this._lastTime / 100000); this._effect.setFloat("windForce", this.windForce); this._effect.setFloat("waveHeight", this.waveHeight); this._effect.setFloat("bumpHeight", this.bumpHeight); this._effect.setColor4("waterColor", this.waterColor, 1.0); this._effect.setFloat("colorBlendFactor", this.colorBlendFactor); this._effect.setFloat("waveSpeed", this.waveSpeed); super.bind(world, mesh); } private _createRenderTargets(scene: Scene, renderTargetSize: Vector2): void { // Render targets this._refractionRTT = new RenderTargetTexture(name + "_refraction", {width: renderTargetSize.x, height: renderTargetSize.y}, scene, false, true); this._reflectionRTT = new RenderTargetTexture(name + "_reflection", {width: renderTargetSize.x, height: renderTargetSize.y}, scene, false, true); scene.customRenderTargets.push(this._refractionRTT); scene.customRenderTargets.push(this._reflectionRTT); var isVisible: boolean; var clipPlane = null; var savedViewMatrix; var mirrorMatrix = Matrix.Zero(); this._refractionRTT.onBeforeRender = () => { if (this._mesh) { isVisible = this._mesh.isVisible; this._mesh.isVisible = false; } // Clip plane clipPlane = scene.clipPlane; var positiony = this._mesh ? this._mesh.position.y : 0.0; scene.clipPlane = Plane.FromPositionAndNormal(new Vector3(0, positiony + 0.05, 0), new Vector3(0, 1, 0)); }; this._refractionRTT.onAfterRender = () => { if (this._mesh) { this._mesh.isVisible = isVisible; } // Clip plane scene.clipPlane = clipPlane; }; this._reflectionRTT.onBeforeRender = () => { if (this._mesh) { isVisible = this._mesh.isVisible; this._mesh.isVisible = false; } // Clip plane clipPlane = scene.clipPlane; var positiony = this._mesh ? this._mesh.position.y : 0.0; scene.clipPlane = Plane.FromPositionAndNormal(new Vector3(0, positiony - 0.05, 0), new Vector3(0, -1, 0)); // Transform Matrix.ReflectionToRef(scene.clipPlane, mirrorMatrix); savedViewMatrix = scene.getViewMatrix(); mirrorMatrix.multiplyToRef(savedViewMatrix, this._reflectionTransform); scene.setTransformMatrix(this._reflectionTransform, scene.getProjectionMatrix()); scene.getEngine().cullBackFaces = false; scene._mirroredCameraPosition = Vector3.TransformCoordinates(scene.activeCamera.position, mirrorMatrix); }; this._reflectionRTT.onAfterRender = () => { if (this._mesh) { this._mesh.isVisible = isVisible; } // Clip plane scene.clipPlane = clipPlane; // Transform scene.setTransformMatrix(savedViewMatrix, scene.getProjectionMatrix()); scene.getEngine().cullBackFaces = true; scene._mirroredCameraPosition = null; }; } public getAnimatables(): IAnimatable[] { var results = []; if (this.bumpTexture && this.bumpTexture.animations && this.bumpTexture.animations.length > 0) { results.push(this.bumpTexture); } if (this._reflectionRTT && this._reflectionRTT.animations && this._reflectionRTT.animations.length > 0) { results.push(this._reflectionRTT); } if (this._refractionRTT && this._refractionRTT.animations && this._refractionRTT.animations.length > 0) { results.push(this._refractionRTT); } return results; } public dispose(forceDisposeEffect?: boolean): void { if (this.bumpTexture) { this.bumpTexture.dispose(); } var index = this.getScene().customRenderTargets.indexOf(this._refractionRTT); if (index != -1){ this.getScene().customRenderTargets.splice(index, 1); } index = -1; index = this.getScene().customRenderTargets.indexOf(this._reflectionRTT); if (index != -1){ this.getScene().customRenderTargets.splice(index, 1); } if (this._reflectionRTT) { this._reflectionRTT.dispose(); } if (this._refractionRTT) { this._refractionRTT.dispose(); } super.dispose(forceDisposeEffect); } public clone(name: string): WaterMaterial { return SerializationHelper.Clone(() => new WaterMaterial(name, this.getScene()), this); } public serialize(): any { var serializationObject = SerializationHelper.Serialize(this); serializationObject.customType = "BABYLON.WaterMaterial"; return serializationObject; } // Statics public static Parse(source: any, scene: Scene, rootUrl: string): WaterMaterial { return SerializationHelper.Parse(() => new WaterMaterial(source.name, scene), source, scene, rootUrl); } public static CreateDefaultMesh(name: string, scene: Scene): Mesh { var mesh = Mesh.CreateGround(name, 512, 512, 32, scene, false); return mesh; } } }