57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
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, 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=freq_hz,
|
|
include_vegetation=request.include_vegetation,
|
|
)
|
|
budget = manual_link_budget(
|
|
tx=tx,
|
|
rx=rx,
|
|
tx_power_dbm=request.tx.power_dbm,
|
|
tx_gain_dbi=request.tx.gain_dbi,
|
|
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__)
|