gedai.core

Pre-process ADS-B data and create a pandas DataFrame.

gedai.core.assign_to_flight(func: Callable[[DataFrame, ...], DataFrame]) → Callable[[...], Flight]

Decorator that allows a DataFrame-transforming function to be applied directly to a Flight object, and automatically updates the Flight’s data with any new columns returned by the function.

The decorated function must: - Take a pandas DataFrame (typically flight.data) as its first argument. - Return a DataFrame that includes any new columns to be assigned.

When used, the decorated function can be called with a Flight object as its first argument, and the resulting Flight will have the new columns added via .assign().

Parameters:

func – A function that modifies a DataFrame and returns it with additional columns to assign.

Returns:

A callable that accepts a Flight object and returns a new Flight with the updated data.

gedai.core.calculate_distance(obj: DataFrame) → DataFrame

gedai.core.calculate_distance(obj: Flight) → Flight

Add distance column to a DataFrame of Flight.

Parameters:

obj (Union[pd.DataFrame, Flight]) – Flight data

Returns:

Object with added “distance” column [km]

Return type:

Union[pd.DataFrame, Flight]