import numpy as np
import pyuasrisk

# Define boundaries: [South_lat, West_lon, MinAlt_m, North_lat, East_lon, MaxAlt_m]
xyz_bounds = np.array([50.9, -1.5, 0, 50.95, -1.4, 100])
xy_res = 200  # Resolution in meters for X (longitude-equivalent) and Y (latitude-equivalent)
z_res = 40    # Resolution in meters for Z axis (altitude)

voxel_grid = pyuasrisk.VoxelGrid(xyz_bounds, xy_res, z_res)

print(f"Voxel grid dimensions (num_cells_lat, num_cells_lon, num_cells_alt): {voxel_grid.size}")
print(f"Initial layers: {voxel_grid.layers}") # May be empty or have defaults

# Add a custom layer (data should be Fortran-ordered for C++ compatibility)
layer_data = np.full(voxel_grid.size, 5.0, dtype=np.float32, order="F")
voxel_grid.add("my_custom_layer", layer_data)
print(f"Layers after adding: {voxel_grid.layers}")
idx_to_check = np.array([0, 0, 0], dtype=int)
print(f"Value at index {idx_to_check} in 'my_custom_layer': {voxel_grid.at('my_custom_layer', idx_to_check)}")

# Coordinate conversion
world_coords = np.array([-1.45, 50.925, 50]) # Lon, Lat, Alt
local_idx = voxel_grid.world2Local(world_coords)
print(f"World {world_coords} -> Local Index {local_idx}")
retrieved_world_coords = voxel_grid.local2World(local_idx)
print(f"Local Index {local_idx} -> World {retrieved_world_coords}")
assert np.allclose(world_coords, retrieved_world_coords, atol=xy_res), "Coordinate conversion issue"

import numpy as np
import pyuasrisk

aircraft_state = pyuasrisk.AircraftStateModel()
aircraft_state.position = np.array([-1.45, 50.925, 120])  # Lon, Lat, Alt (meters)
aircraft_state.velocity = np.array([20, 5, 0])  # Velocity vector (vx, vy, vz) m/s in a local frame

# mass(kg), width(m), length(m), failure_prob (per flight hour)
aircraft_model = pyuasrisk.AircraftModel(mass=25, width=2, length=1.5, failureProb=2.5e-4)
aircraft_model.state = aircraft_state

# Add descent models (parameters are specific to each model)
aircraft_model.addGlideDescentModel(glide_speed=20, glide_ratio=10) # speed(m/s), ratio(-)
aircraft_model.addBallisticDescentModel(frontal_area=0.5, drag_coeff=0.8) # area(m^2), drag_coeff(-)
# aircraft_model.addParachuteDescentModel(parachute_area=3.0, parachute_drag_coeff=1.0, deployment_time=2.0)

print(f"Aircraft Mass: {aircraft_model.mass} kg")
print(f"Descent Models: {aircraft_model.descentNames}")

import numpy as np
import pyuasrisk

# Boundaries and resolution
xyz_bounds = np.array([50.9, -1.5, 0, 50.95, -1.4, 120])
xy_res, z_res = 100, 30
center_coords = np.array([-1.45, 50.925, 60]) # Lon, Lat, Alt

# Aircraft model (from example 2)
aircraft_state = pyuasrisk.AircraftStateModel()
aircraft_state.position = center_coords # Position for which risk is evaluated if failure occurs there
aircraft_state.velocity = np.array([15, 0, 0])
am = pyuasrisk.AircraftModel(10, 1, 1, 1e-4)
am.state = aircraft_state
am.addGlideDescentModel(18, 8)

# Initialize GroundRiskVoxelGrid
grvg = pyuasrisk.GroundRiskVoxelGrid(xyz_bounds, xy_res, z_res, am)

print("Evaluating ground risk grid...")
grvg.eval() # Populates "Ground Risk" and other layers
print("Evaluation complete.")

risk_val = grvg.atPosition("Ground Risk", center_coords)
print(f"Ground Risk at {center_coords}: {risk_val}")
# print(f"Mean Ground Risk: {grvg.get('Ground Risk').mean()}")

import numpy as np
import pyuasrisk

xyz_bounds = np.array([50.9, -1.5, 0, 50.95, -1.4, 120])
xy_res, z_res = 100, 30
failure_point = np.array([-1.45, 50.925, 100]) # Lon, Lat, Alt of failure
heading_at_failure = 90 # Degrees from North (East)

# Aircraft model (from example 2)
aircraft_state = pyuasrisk.AircraftStateModel() # State here is for aircraft params, not failure point
am = pyuasrisk.AircraftModel(10, 1, 1, 1e-4)
am.state = aircraft_state # Nominal state
am.addGlideDescentModel(18, 8)

igrvg = pyuasrisk.IncrementalGroundRiskVoxelGrid(xyz_bounds, xy_res, z_res, am)

strike_prob = igrvg.getPositionPointStrikeProbability(failure_point, heading_at_failure)
fatality_prob = igrvg.getPositionPointFatalityProbability(failure_point, heading_at_failure)

print(f"For failure at {failure_point}, heading {heading_at_failure} deg:")
print(f"  Strike Probability (on ground): {strike_prob}")
print(f"  Fatality Probability (on ground): {fatality_prob}")

import numpy as np
import pyuasrisk
import os

xyz_bounds = np.array([50.9, -1.5, 0, 50.95, -1.4, 3000]) # Up to ~10,000 ft
xy_res, z_res = 500, 150
center_coords_air = np.array([-1.45, 50.925, 1500]) # Lon, Lat, Alt

# Create dummy data files for the example
data_dir = "temp_uasrisk_data"
os.makedirs(data_dir, exist_ok=True)
traj_file = os.path.join(data_dir, "dummy_traj.csv")
airspace_file = os.path.join(data_dir, "dummy_airspace.aip")

with open(traj_file, "w") as f:
    f.write("icao24,callsign,origin_country,time_position,last_contact,longitude,latitude,baro_altitude,on_ground,velocity,true_track,vertical_rate,sensors,geo_altitude,squawk,spi,position_source\n")
    # Example track point:
    # f.write("ICAO123,FLT001,Country,1678886400,1678886400,-1.45,50.92,1500,false,100,90,0,,1500,1234,false,0\n")
with open(airspace_file, "w") as f:
    f.write("<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n<OPENAIP VERSION=\"0\">\n<AIRSPACES>\n</AIRSPACES>\n</OPENAIP>\n")

try:
    arvg = pyuasrisk.AirRiskVoxelGrid(xyz_bounds, xy_res, z_res, traj_file, airspace_file)
    print("Evaluating air risk grid...")
    arvg.eval() # Processes data files and populates "Air Risk" layer
    print("Evaluation complete.")
    risk_val = arvg.atPosition("Air Risk", center_coords_air)
    print(f"Air Risk at {center_coords_air}: {risk_val}")
    # print(f"Mean Air Risk: {arvg.get('Air Risk').mean()}")
except Exception as e:
    print(f"Air risk calculation failed: {e}. Ensure data files are valid or dependencies met.")
finally:
    print(f"Note: Dummy data files are at {data_dir}")
    # Consider cleaning up dummy files: os.remove(traj_file); os.remove(airspace_file); os.rmdir(data_dir)

import numpy as np
import pyuasrisk
import os

xyz_bounds = np.array([50.9, -1.5, 0, 50.95, -1.4, 3000])
xy_res, z_res = 500, 150
eval_coords = np.array([-1.45, 50.925, 1500]) # Lon, Lat, Alt

# Aircraft model (from example 2)
aircraft_state = pyuasrisk.AircraftStateModel()
am = pyuasrisk.AircraftModel(25, 2, 1.5, 2.5e-4)
am.state = aircraft_state
am.addGlideDescentModel(20,10)

# Data files (ensure they exist, e.g., from example 5)
data_dir = "temp_uasrisk_data" # Make sure this dir and files exist
traj_file = os.path.join(data_dir, "dummy_traj.csv")
airspace_file = os.path.join(data_dir, "dummy_airspace.aip")
if not (os.path.exists(traj_file) and os.path.exists(airspace_file)):
    print(f"Warning: Data files for combined risk not found in {data_dir}. Creating dummies.")
    os.makedirs(data_dir, exist_ok=True)
    with open(traj_file, "w") as f: f.write("icao24,callsign,origin_country,time_position,last_contact,longitude,latitude,baro_altitude,on_ground,velocity,true_track,vertical_rate,sensors,geo_altitude,squawk,spi,position_source\n")
    with open(airspace_file, "w") as f: f.write("<?xml version=\"1.0\"?><OPENAIP></OPENAIP>\n")


try:
    rvg = pyuasrisk.RiskVoxelGrid(xyz_bounds, xy_res, z_res, traj_file, airspace_file, am)
    print("Evaluating combined risk grid (Air + Ground)...")
    rvg.eval() # Evaluates all components
    print("Evaluation complete.")

    total_risk = rvg.atPosition("Total Risk", eval_coords)
    ground_comp = rvg.atPosition("Ground Risk", eval_coords)
    air_comp = rvg.atPosition("Air Risk", eval_coords)
    print(f"At {eval_coords}:")
    print(f"  Total Risk: {total_risk}")
    print(f"  Ground Risk component: {ground_comp}")
    print(f"  Air Risk component: {air_comp}")
    # print(f"Mean Total Risk: {rvg.get('Total Risk').mean()}")
except Exception as e:
    print(f"Combined risk calculation failed: {e}. Check data, aircraft model, and dependencies.")
finally:
    print(f"Note: Dummy data files may have been used from {data_dir}")