Basic post-rain and faders

Data Files/MWSE/mods/tew/Vapourmist/services/mist.lua

@@ -1,54 +1,102 @@
 local shader = require("tew.Vapourmist.components.shader")
+local util = require("tew.Vapourmist.components.util")
+local debugLog = util.debugLog
+local config = require("tew.Vapourmist.config")
 
-local MIN_DISTANCE = 8192 * 1.5
 local MAX_DISTANCE = 8192 * 3
 local MAX_DEPTH = 8192 / 10
-local DENSITY = 10
+local MAX_DENSITY = 10
+local STEPS = 50
+local mistDensity = 0
 local WtC = tes3.worldController.weatherController
+local WorldC = tes3.worldController
 
-local fogId = "tew_mist"
+local TIMER_DURATION = 0.1
+local FADE_DURATION = 0.05
+
+local mistDeployed = false
+
+local FOG_ID = "tew_mist"
+
+local toWeather, postRainMist
+
+local wetWeathers = {
+    ["Rain"] = true,
+    ["Thunderstorm"] = true
+}
 
 local mist = {}
 
 ---@type mwseTimer
-local FOG_TIMER
+local FOG_TIMER, FADE_IN_TIMER, FADE_OUT_TIMER, FADE_OUT_REMOVE_TIMER
 
 ---@type fogParams
 local fogParams = {
     color = tes3vector3.new(),
     center = tes3vector3.new(),
     radius = tes3vector3.new(MAX_DISTANCE, MAX_DISTANCE, MAX_DEPTH),
-    density = DENSITY,
+    density = mistDensity,
 }
 
+local function isAvailable(weather, gameHour)
+    local weatherName = weather.name
+
+    if config.blockedMist[weatherName] then return false end
+
+    return
+    ((
+        (gameHour > WtC.sunriseHour - 1 and gameHour < WtC.sunriseHour + 1.5)
+    or
+        (gameHour >= WtC.sunsetHour - 0.4 and gameHour < WtC.sunsetHour + 2))
+    and not
+        wetWeathers[weatherName])
+    or
+    (
+        config.mistyWeathers[weatherName]
+    )
+end
+
 local function getCloudColourMix(fogComp, skyComp)
-	return math.lerp(fogComp, skyComp, 0.2)
+    return math.lerp(fogComp, skyComp, 0.2)
 end
 
 local function getDarkerColour(comp)
-	return math.clamp(math.lerp(comp, 0.0, 0.1), 0.03, 0.9)
+    return math.clamp(math.lerp(comp, 0.0, 0.1), 0.03, 0.9)
 end
 
 -- Calculate output colours from current fog colour --
 local function getOutputValues()
-	local currentFogColor = WtC.currentFogColor:copy()
-	local currentSkyColor = WtC.currentSkyColor:copy()
-	local weatherColour = {
-		r = getCloudColourMix(currentFogColor.r, currentSkyColor.r),
-		g = getCloudColourMix(currentFogColor.g, currentSkyColor.g),
-		b = getCloudColourMix(currentFogColor.b, currentSkyColor.b)
-	}
+    local currentFogColor = WtC.currentFogColor:copy()
+    local currentSkyColor = WtC.currentSkyColor:copy()
+    local weatherColour = {
+        r = getCloudColourMix(currentFogColor.r, currentSkyColor.r),
+        g = getCloudColourMix(currentFogColor.g, currentSkyColor.g),
+        b = getCloudColourMix(currentFogColor.b, currentSkyColor.b)
+    }
+
+    return tes3vector3.new(
+    getDarkerColour(weatherColour.r),
+    getDarkerColour(weatherColour.g),
+    getDarkerColour(weatherColour.b)
+)
+end
 
-	return tes3vector3.new(
-        getDarkerColour(weatherColour.r),
-        getDarkerColour(weatherColour.g),
-        getDarkerColour(weatherColour.b)
-    )
+local function fadeIn()
+    if mistDensity >= MAX_DENSITY then return end
+    mistDensity = mistDensity + MAX_DENSITY/STEPS
+    debug.log(tostring(mistDensity))
+end
+
+local function fadeOut()
+    if mistDensity <= 0 then return end
+    mistDensity = mistDensity - MAX_DENSITY/STEPS
 end
 
-local function updateDensity(dist)
-    local f = math.clamp(dist, MIN_DISTANCE, MAX_DISTANCE)
-    f = math.remap(f, MIN_DISTANCE, MAX_DISTANCE, 15.0, 0.0)
+local function updateMist()
+    debugLog("Updating mist")
+    if tes3.player.cell.isInterior then
+        return
+    end
 
     local playerPos = tes3.mobilePlayer.position:copy()
 
@@ -57,46 +105,108 @@ local function updateDensity(dist)
         (playerPos.y),
         0
     )
-
-    fogParams.density = f
+    debug.log(tostring(mistDensity))
+    fogParams.density = mistDensity
     fogParams.center = mistCenter
     fogParams.color = getOutputValues()
-
-    shader.createOrUpdateFog(fogId, fogParams)
+    shader.createOrUpdateFog(FOG_ID, fogParams)
 end
 
-local function update()
-    if tes3.player.cell.isInterior then
-        return
-    end
-
-	local currDist = 0
-
-    if currDist <= MAX_DISTANCE then
-        updateDensity(currDist)
-    end
-
-end
-event.register(tes3.event.loaded, function()
-    FOG_TIMER = timer.start({
+function mist.onLoaded()
+    FOG_TIMER = timer.start{
         iterations = -1,
         duration = 0.01,
-        callback = update,
+        callback = updateMist,
         type = timer.game,
-        data = {},
-    })
-end)
-
---- @param e cellChangedEventData
-function mist.onCellChanged(e)
-    local dist = 5000
-    if dist > MAX_DISTANCE or e.cell.isInterior then
-        shader.deleteFog(fogId)
+        persist = false
+    }
+
+    timer.start{
+		duration = TIMER_DURATION,
+		callback = mist.conditionCheck,
+		iterations = -1,
+		type = timer.game,
+		persist = false
+	}
+    mist.conditionCheck()
+end
+
+function mist.onWeatherChanged(e)
+    local fromWeather = e.from.name
+    toWeather = e.to.name
+	if wetWeathers[fromWeather] and config.blockedMist[toWeather] ~= true then
+		debugLog("Adding post-rain mist.")
+
+		-- Slight offset so it makes sense --
+		timer.start {
+			type = timer.game,
+			iterations = 1,
+			duration = 0.3,
+			callback = function()
+                postRainMist = true
+                updateMist()
+                FOG_TIMER:resume()
+            end
+		}
+
+        timer.start {
+			type = timer.game,
+			iterations = 1,
+			duration = 1,
+			callback = function()
+                postRainMist = false
+            end
+		}
+	end
+end
+
+function mist.conditionCheck()
+    local cell = tes3.getPlayerCell()
+    if not cell.isOrBehavesAsExterior then
         FOG_TIMER:pause()
+        shader.deleteFog(FOG_ID)
+    end
+
+    toWeather = WtC.nextWeather or WtC.currentWeather
+    local gameHour = WorldC.hour.value
+
+    if isAvailable(toWeather, gameHour) or postRainMist then
+        if not mistDeployed then
+            debugLog("Mist available.")
+            updateMist()
+            FOG_TIMER:resume()
+            FADE_OUT_TIMER:cancel()
+            FADE_OUT_REMOVE_TIMER:cancel()
+            FADE_IN_TIMER = timer.start{
+                duration = FADE_DURATION,
+                callback = fadeIn,
+                iterations = STEPS,
+                type = timer.game,
+                persist = false
+            }
+            mistDeployed = true
+        end
     else
-        update()
-        shader.createOrUpdateFog(fogId, fogParams)
-        FOG_TIMER:resume()
+        debugLog("Mist not available.")
+        FADE_IN_TIMER:cancel()
+        FADE_OUT_TIMER = timer.start{
+            duration = FADE_DURATION,
+            callback = fadeOut,
+            iterations = STEPS,
+            type = timer.game,
+            persist = false
+        }
+        FADE_OUT_REMOVE_TIMER = timer.start{
+            duration = (FADE_DURATION*STEPS) + FADE_DURATION,
+            callback = function()
+                mistDeployed = false
+                FOG_TIMER:pause()
+                shader.deleteFog(FOG_ID)
+            end,
+            iterations = 1,
+            type = timer.game,
+            persist = false
+        }
     end
 end