prop/lib/microwaveprop/weather/weather_layers.ex

123 lines
3.3 KiB
Elixir

defmodule Microwaveprop.Weather.WeatherLayers do
@moduledoc "Derives scalar weather map layers from HRRR profile data."
alias Microwaveprop.Weather.SoundingParams
@doc """
Derives a map of scalar weather fields from a row containing HRRR profile data.
Expects a map with keys: `:surface_temp_c`, `:surface_dewpoint_c`,
`:surface_pressure_mb`, `:surface_refractivity`, `:profile`, `:duct_characteristics`.
Returns a map with derived fields: `:surface_rh`, `:surface_refractivity`,
`:temp_850mb`, `:dewpoint_850mb`, `:lapse_rate`, `:inversion_strength`,
`:inversion_base_m`, `:duct_base_m`, `:duct_strength`.
"""
def derive(row) do
sorted = sort_profile(row.profile)
%{
surface_rh: compute_rh(row.surface_temp_c, row.surface_dewpoint_c),
surface_refractivity: row.surface_refractivity,
temp_850mb: level_value(sorted, 850, "tmpc"),
dewpoint_850mb: level_value(sorted, 850, "dwpc"),
lapse_rate: compute_lapse_rate(sorted),
inversion_strength: inversion_strength(sorted),
inversion_base_m: inversion_base_m(sorted),
duct_base_m: duct_field(row.duct_characteristics, "base"),
duct_strength: duct_field(row.duct_characteristics, "strength")
}
end
defp sort_profile(nil), do: []
defp sort_profile([]), do: []
defp sort_profile(profile) do
profile
|> Enum.filter(fn p -> p["pres"] != nil and p["tmpc"] != nil and p["hght"] != nil end)
|> Enum.sort_by(fn p -> p["pres"] end, :desc)
end
defp compute_rh(t_c, td_c) do
100.0 * SoundingParams.sat_vap_pres(td_c) / SoundingParams.sat_vap_pres(t_c)
end
defp level_value([], _target_pres, _key), do: nil
defp level_value(sorted, target_pres, key) do
nearest = Enum.min_by(sorted, fn p -> abs(p["pres"] - target_pres) end)
if abs(nearest["pres"] - target_pres) <= 25.0 do
nearest[key]
end
end
defp compute_lapse_rate([]), do: nil
defp compute_lapse_rate(sorted) when length(sorted) < 2, do: nil
defp compute_lapse_rate(sorted) do
surface = hd(sorted)
top = List.last(sorted)
dh_km = (top["hght"] - surface["hght"]) / 1000.0
if dh_km > 0 do
(surface["tmpc"] - top["tmpc"]) / dh_km
end
end
defp inversion_strength([]), do: 0.0
defp inversion_strength(sorted) do
sorted
|> Enum.chunk_every(2, 1, :discard)
|> Enum.reduce(0.0, fn [lower, upper], max_str ->
dt = upper["tmpc"] - lower["tmpc"]
if dt > max_str do
dt
else
max_str
end
end)
end
defp inversion_base_m([]), do: nil
defp inversion_base_m(sorted) do
sfc_hght = hd(sorted)["hght"]
result =
sorted
|> Enum.chunk_every(2, 1, :discard)
|> Enum.reduce({0.0, nil}, fn [lower, upper], {max_str, _base} = acc ->
dt = upper["tmpc"] - lower["tmpc"]
if dt > 0 and dt > max_str do
{dt, lower["hght"] - sfc_hght}
else
acc
end
end)
case result do
{str, base} when str > 0 -> base
_ -> nil
end
end
defp duct_field(nil, _key), do: nil
defp duct_field([], _key), do: nil
defp duct_field(ducts, "base") do
ducts
|> Enum.map(fn d -> d["base"] end)
|> Enum.min()
end
defp duct_field(ducts, "strength") do
ducts
|> Enum.map(fn d -> d["strength"] end)
|> Enum.max()
end
end