Helper method to extract rainfall data specifically

Author: ConnectedSystems

src/Climate.jl

@@ -22,6 +22,21 @@ function subcatchment_data(node::NetworkNode, climate::Climate)::DataFrame
 end
 
 
+"""
+    rainfall_data(node::NetworkNode, climate::Climate)::DataFrame
+
+Extract rainfall data for a given node.
+"""
+function rainfall_data(node::NetworkNode, climate::Climate)::DataFrame
+    data = climate.climate_data
+    rain_col = filter(x -> occursin(node.name, x) 
+                            & occursin(climate.rainfall_id, x), 
+                            names(data))[1]
+
+    return data[:, rain_col]
+end
+
+
 """
     climate_values(node::NetworkNode, climate::Climate, timestep::Union{Nothing, Int}=nothing)