Add floodplain discharge to inflow reservoirs and lakes

docs/src/changelog.md

@@ -17,6 +17,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
   lake storage. The variable `actevap` has also been added to the reservoir module.
 - The `set_states` function for model type `sbm` with local inertial routing for river and
   land component.
+- Inflow to reservoir and lake locations for local inertial routing with floodplain routing,
+  the floodplain discharge was not added to the inflow of these locations.
 
 ### Changed
 - Stop exposing scalar variables through BMI. The `BMI.get_value_ptr` function was

src/flow.jl

@@ -688,24 +688,23 @@ function shallowwater_river_update(
     # local inertial model (fixed h).
     for v in eachindex(sw.reservoir_index)
         i = sw.reservoir_index[v]
-        update(
-            sw.reservoir,
-            v,
-            sum_at(sw.q0, links_at_node.src[i]) + inflow_wb[i] + sw.inflow_wb[i],
-            Δt,
-        )
+
+        q_in = sum_at(sw.q0, links_at_node.src[i])
+        if !isnothing(sw.floodplain)
+            q_in = q_in + sum_at(sw.floodplain.q0, links_at_node.src[i])
+        end
+        update(sw.reservoir, v, q_in + inflow_wb[i] + sw.inflow_wb[i], Δt)
         sw.q[i] = sw.reservoir.outflow[v]
         sw.q_av[i] += sw.q[i] * Δt
     end
     for v in eachindex(sw.lake_index)
         i = sw.lake_index[v]
-        update(
-            sw.lake,
-            v,
-            sum_at(sw.q0, links_at_node.src[i]) + inflow_wb[i] + sw.inflow_wb[i],
-            doy,
-            Δt,
-        )
+
+        q_in = sum_at(sw.q0, links_at_node.src[i])
+        if !isnothing(sw.floodplain)
+            q_in = q_in + sum_at(sw.floodplain.q0, links_at_node.src[i])
+        end
+        update(sw.lake, v, q_in + inflow_wb[i] + sw.inflow_wb[i], doy, Δt)
         sw.q[i] = sw.lake.outflow[v]
         sw.q_av[i] += sw.q[i] * Δt
     end