from sqlalchemy.orm import Session

from app.core.diffraction import deygout
from app.core.fresnel import LosSample, 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.models.terrain import Obstruction
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,
    )
    geometric_obstructions = [
        sample for sample in los_result.obstructions if sample.clearance_m < 0
    ]
    building_obstructions = [
        sample
        for sample in geometric_obstructions
        if sample.obstruction_type == "building"
    ]
    return LinkBudgetResponse(
        **budget.__dict__,
        los_clear=los_result.los_clear,
        geometric_los=los_result.geometric_los,
        first_fresnel_clearance_pct=los_result.first_fresnel_clearance_pct,
        fresnel_violations_count=len(los_result.obstructions),
        geometric_obstructions_count=len(geometric_obstructions),
        building_obstructions_count=len(building_obstructions),
        worst_obstruction=_convert_obstruction(los_result.worst_sample),
    )


def _convert_obstruction(sample: LosSample | None) -> Obstruction | None:
    if sample is None:
        return None
    return Obstruction(
        distance_m=sample.distance_m,
        clearance_m=sample.clearance_m,
        fresnel_radius_m=sample.fresnel_radius_m,
        required_clearance_m=sample.required_clearance_m,
        type=sample.obstruction_type,
    )