Computes the GPS segment list for a map-matched trajectory and the corresponding map polyline

compute_gps_segment.py

def compute_gps_segments(unique_df: pd.DataFrame,
                         polyline: np.ndarray) -> List[GpsSegment]:
    traj = Trajectory(lat=unique_df["match_latitude"].to_numpy(),
                      lon=unique_df["match_longitude"].to_numpy(),
                      time=unique_df["time_stamp"].to_numpy() / 1000.0,
                      ids=unique_df["signal_id"].to_numpy())

    map_tree = traj.merge(map_lat=polyline[:, 0],  # Latitude
                          map_lon=polyline[:, 1])  # Longitude

    gps_segments = map_tree.get_segments()
    return gps_segments