added vegetation

api/app/services/link.py

@@ -1,19 +1,44 @@
+from sqlalchemy.orm import Session
+
+from app.core.diffraction import deygout
+from app.core.fresnel import los_analysis
 from app.core.geo import GeoPoint, sample_path
 from app.core.propagation import manual_link_budget
 from app.models.link import LinkBudgetRequest, LinkBudgetResponse
+from app.services.terrain import surface_profile_from_points
 from app.services.vegetation import vegetation_loss_along
 
 
-def link_budget(request: LinkBudgetRequest) -> LinkBudgetResponse:
+def link_budget(request: LinkBudgetRequest, db: Session | None = None) -> LinkBudgetResponse:
     if request.model != "manual":
         raise NotImplementedError(f"{request.model} link model is implemented in a later stage")
 
     tx = GeoPoint(lat=request.tx.lat, lon=request.tx.lon)
     rx = GeoPoint(lat=request.rx.lat, lon=request.rx.lon)
     points = sample_path(tx, rx, 256)
+    freq_hz = request.frequency_mhz * 1_000_000
+    surface_profile = surface_profile_from_points(
+        points,
+        include_buildings=request.include_buildings,
+        include_canopy=False,
+        db=db,
+    )
+    los_result = los_analysis(
+        surface_profile,
+        tx_height_agl=request.tx.height_agl,
+        rx_height_agl=request.rx.height_agl,
+        freq_hz=freq_hz,
+        k=request.k_factor,
+    )
+    diffraction_db = deygout(
+        surface_profile,
+        tx_height_agl=request.tx.height_agl,
+        rx_height_agl=request.rx.height_agl,
+        freq_hz=freq_hz,
+    )
     vegetation_db = vegetation_loss_along(
         points,
-        freq_hz=request.frequency_mhz * 1_000_000,
+        freq_hz=freq_hz,
         include_vegetation=request.include_vegetation,
     )
     budget = manual_link_budget(
@@ -24,6 +49,8 @@ def link_budget(request: LinkBudgetRequest) -> LinkBudgetResponse:
         rx_gain_dbi=request.rx.gain_dbi,
         sensitivity_dbm=request.rx.sensitivity_dbm,
         frequency_mhz=request.frequency_mhz,
+        diffraction_db=diffraction_db,
         vegetation_db=vegetation_db,
+        fresnel_clear=los_result.los_clear,
     )
     return LinkBudgetResponse(**budget.__dict__)