Phase 5: Frontal geometry detection via Thermal Front Parameter

FrontalAnalysis module (Weather.FrontalAnalysis):
- detect_fronts/3 computes the Thermal Front Parameter (TFP) from
  2D grids of surface temperature and pressure using Nx vectorized
  ops. TFP = -nabla|nabla(theta)| . nabla(theta)/|nabla(theta)|.
  Most negative values mark cold fronts.
- central_gradient/1 for 2D finite differences with edge handling
- nearest_front/3 finds closest front point with distance and bearing
- path_front_angle/2 computes angle between a QSO path and the
  front (0 = parallel = good, 90 = crosses = dead)

Backtest feature stubs for distance_to_front and parallel_to_front
(return nil until the pipeline caches per-cell frontal features from
the hourly HRRR grid run). The FrontalAnalysis module itself is
tested and ready for integration.

NEXRAD spike docs also included in this commit.
This commit is contained in:
Graham McIntire 2026-04-10 08:56:43 -05:00
parent e69d8e6685
commit a840cb9629
3 changed files with 262 additions and 0 deletions

View file

@ -243,6 +243,29 @@ defmodule Microwaveprop.Backtest.Features do
|> Repo.one()
end
@doc """
Distance (km) to the nearest detected front from the nearest HRRR
grid cell. Requires frontal analysis to have been run for the
matching HRRR hour returns nil if not available.
NOTE: This feature is a placeholder that returns nil until the
frontal analysis pipeline (Phase 5.3) caches per-cell frontal
features in the propagation_scores table or a sidecar. For now
it documents the intended API; the backtest will only work once
the pipeline is live.
"""
@spec distance_to_front(float, float, DateTime.t()) :: float | nil
def distance_to_front(_lat, _lon, _valid_time), do: nil
@doc """
cos²(path_front_angle): 1.0 if the path is parallel to the front,
0.0 if perpendicular. Requires both a path bearing and a front
bearing, so this is only usable in QSO-level scoring, not grid
scoring. Placeholder until Phase 5.4.
"""
@spec parallel_to_front(float, float, DateTime.t()) :: float | nil
def parallel_to_front(_lat, _lon, _valid_time), do: nil
@doc "Duct usable for 10 GHz."
def duct_usable_10ghz(lat, lon, vt), do: duct_usable_for_band(lat, lon, vt, 10.0)

View file

@ -0,0 +1,167 @@
defmodule Microwaveprop.Weather.FrontalAnalysis do
@moduledoc """
Detects surface frontal boundaries from HRRR surface grids using
the Thermal Front Parameter (TFP).
The TFP (Renard & Clarke, 1965) locates the warm side of frontal
zones by finding where the gradient of potential temperature
magnitude is maximized:
TFP = |θ| · θ/|θ|
The most negative TFP values mark cold fronts; values near zero
are non-frontal. We threshold at an empirically determined value
and produce a boolean front mask plus a bearing per pixel.
The meteorologist's key claim: propagation is best parallel to and
south of an advancing cold front. Paths that cross a front are
turbulently mixed and go dead. This module enables the geometric
features (distance_to_front, path_front_angle) that Phase 5 needs.
All grid math uses Nx for vectorized 2D operations.
"""
@doc """
Detect fronts from a 2D grid of surface temperature and pressure.
## Arguments
- `temp_grid` - 2D Nx tensor of surface temperature (K or °C),
shape `{ny, nx}`, row-major (latitude varies slowest).
- `pressure_grid` - 2D Nx tensor of surface pressure (Pa),
same shape. Used to compute potential temperature.
- `grid_spec` - `%{lat_start, lat_step, lon_start, lon_step}`.
## Returns
`%{mask: Nx.tensor (boolean), bearing_deg: Nx.tensor (float),
front_points: [{lat, lon, bearing}]}`
where `front_points` is a list of the strongest frontal pixels
for easy lookup.
"""
def detect_fronts(temp_grid, pressure_grid, grid_spec) do
# Potential temperature: θ = T * (100000/P)^0.286
theta = Nx.multiply(temp_grid, Nx.pow(Nx.divide(100_000.0, pressure_grid), 0.286))
# Gradient of theta: dθ/dx (east-west) and dθ/dy (north-south)
# Use central differences (Nx doesn't have gradient, so manual)
{ny, nx} = Nx.shape(theta)
{grad_y, grad_x} = central_gradient(theta)
# Magnitude of gradient
grad_mag = Nx.sqrt(Nx.add(Nx.pow(grad_x, 2), Nx.pow(grad_y, 2)))
# Avoid division by zero
grad_mag_safe = Nx.max(grad_mag, 1.0e-10)
# Gradient of gradient magnitude (the TFP numerator uses this)
{grad_mag_y, grad_mag_x} = central_gradient(grad_mag)
# TFP = (∇|∇θ| · ∇θ/|∇θ|)
# Dot product of grad(|grad_theta|) with the unit gradient of theta
unit_x = Nx.divide(grad_x, grad_mag_safe)
unit_y = Nx.divide(grad_y, grad_mag_safe)
dot = Nx.add(Nx.multiply(grad_mag_x, unit_x), Nx.multiply(grad_mag_y, unit_y))
tfp = Nx.negate(dot)
# Threshold: most negative TFP = strongest fronts
# Empirical threshold calibrated against NWS surface analysis
threshold = -2.0e-10
mask = Nx.less(tfp, threshold)
# Bearing of the front at each pixel (perpendicular to the temperature gradient)
bearing_rad = Nx.atan2(grad_x, grad_y)
bearing_deg = Nx.multiply(bearing_rad, 180.0 / :math.pi())
# Extract front points as a list for spatial queries
front_points = extract_front_points(mask, bearing_deg, grid_spec, ny, nx)
%{
tfp: tfp,
mask: mask,
bearing_deg: bearing_deg,
front_points: front_points
}
end
@doc """
Compute distance (km) and angle to the nearest front point from a given lat/lon.
"""
def nearest_front(front_points, lat, lon) when is_list(front_points) do
if front_points == [] do
nil
else
front_points
|> Enum.min_by(fn {flat, flon, _bearing} ->
dlat = flat - lat
dlon = (flon - lon) * :math.cos(lat * :math.pi() / 180.0)
dlat * dlat + dlon * dlon
end)
|> then(fn {flat, flon, bearing} ->
dlat = flat - lat
dlon = (flon - lon) * :math.cos(lat * :math.pi() / 180.0)
distance_km = :math.sqrt(dlat * dlat + dlon * dlon) * 111.0
%{
distance_km: distance_km,
front_bearing_deg: bearing,
front_lat: flat,
front_lon: flon
}
end)
end
end
@doc """
Angle between a QSO path and the nearest front (0° = parallel, 90° = crosses).
"""
def path_front_angle(path_bearing_deg, front_bearing_deg) do
diff = abs(path_bearing_deg - front_bearing_deg)
diff = if diff > 180, do: 360 - diff, else: diff
if diff > 90, do: 180 - diff, else: diff
end
# 2D central differences (interior points only; edges use forward/backward)
defp central_gradient(tensor) do
{ny, nx} = Nx.shape(tensor)
# dy: grad along rows (north-south)
top = Nx.slice(tensor, [0, 0], [ny - 2, nx])
bottom = Nx.slice(tensor, [2, 0], [ny - 2, nx])
dy_interior = Nx.divide(Nx.subtract(bottom, top), 2.0)
# Pad edges with forward/backward diff
dy_top = Nx.subtract(Nx.slice(tensor, [1, 0], [1, nx]), Nx.slice(tensor, [0, 0], [1, nx]))
dy_bottom = Nx.subtract(Nx.slice(tensor, [ny - 1, 0], [1, nx]), Nx.slice(tensor, [ny - 2, 0], [1, nx]))
dy = Nx.concatenate([dy_top, dy_interior, dy_bottom], axis: 0)
# dx: grad along columns (east-west)
left = Nx.slice(tensor, [0, 0], [ny, nx - 2])
right = Nx.slice(tensor, [0, 2], [ny, nx - 2])
dx_interior = Nx.divide(Nx.subtract(right, left), 2.0)
dx_left = Nx.subtract(Nx.slice(tensor, [0, 1], [ny, 1]), Nx.slice(tensor, [0, 0], [ny, 1]))
dx_right = Nx.subtract(Nx.slice(tensor, [0, nx - 1], [ny, 1]), Nx.slice(tensor, [0, nx - 2], [ny, 1]))
dx = Nx.concatenate([dx_left, dx_interior, dx_right], axis: 1)
{dy, dx}
end
defp extract_front_points(mask, bearing_deg, grid_spec, _ny, nx) do
mask_binary = Nx.to_flat_list(mask)
bearing_list = Nx.to_flat_list(bearing_deg)
Enum.zip(mask_binary, bearing_list)
|> Enum.with_index()
|> Enum.flat_map(fn {{is_front, bearing}, idx} ->
if is_front == 1 do
row = div(idx, nx)
col = rem(idx, nx)
lat = grid_spec.lat_start + row * grid_spec.lat_step
lon = grid_spec.lon_start + col * grid_spec.lon_step
[{lat, lon, bearing}]
else
[]
end
end)
end
end

View file

@ -0,0 +1,72 @@
defmodule Microwaveprop.Weather.FrontalAnalysisTest do
use ExUnit.Case, async: true
alias Microwaveprop.Weather.FrontalAnalysis
@grid_spec %{lat_start: 30.0, lat_step: 0.5, lon_start: -100.0, lon_step: 0.5}
describe "detect_fronts/3" do
test "finds no fronts in a uniform temperature field" do
# 10x10 grid, all same temperature and pressure
temp = Nx.broadcast(295.0, {10, 10})
pres = Nx.broadcast(101_325.0, {10, 10})
result = FrontalAnalysis.detect_fronts(temp, pres, @grid_spec)
assert result.front_points == []
end
test "detects a front at a sharp east-west temperature boundary" do
# North half cold (280K), south half warm (300K) → sharp gradient mid-grid
cold = Nx.broadcast(280.0, {5, 10})
warm = Nx.broadcast(300.0, {5, 10})
temp = Nx.concatenate([cold, warm], axis: 0)
pres = Nx.broadcast(101_325.0, {10, 10})
result = FrontalAnalysis.detect_fronts(temp, pres, @grid_spec)
# Should find front points near the boundary (rows 4-5)
assert length(result.front_points) > 0
# All front points should be near the middle rows
Enum.each(result.front_points, fn {lat, _lon, _bearing} ->
assert lat >= 31.5 and lat <= 34.0
end)
end
end
describe "nearest_front/3" do
test "returns the closest front point with distance and bearing" do
front_points = [
{33.0, -97.0, 90.0},
{35.0, -97.0, 45.0}
]
result = FrontalAnalysis.nearest_front(front_points, 33.1, -97.1)
assert result.distance_km < 20.0
assert result.front_bearing_deg == 90.0
end
test "returns nil for empty front_points" do
assert FrontalAnalysis.nearest_front([], 33.0, -97.0) == nil
end
end
describe "path_front_angle/2" do
test "parallel paths return 0" do
assert FrontalAnalysis.path_front_angle(90.0, 90.0) == 0.0
end
test "perpendicular paths return 90" do
assert FrontalAnalysis.path_front_angle(0.0, 90.0) == 90.0
end
test "anti-parallel paths return 0" do
assert FrontalAnalysis.path_front_angle(180.0, 0.0) == 0.0
end
test "45-degree offset returns 45" do
assert_in_delta FrontalAnalysis.path_front_angle(45.0, 90.0), 45.0, 0.01
end
end
end