| from app.utils import safe_float, safe_int |
|
|
| FEATURE_COLUMNS = [ |
| "delta_mean_motion","delta_inclination","delta_eccentricity","delta_raan","delta_bstar", |
| "launch_year_gap","same_object_type","same_shell","shell_density_proxy","close_approach_proxy", |
| "persistence_proxy","recurrence_count","trend_delta_score","score_volatility_proxy", |
| "graph_degree_sum","graph_common_neighbors","graph_jaccard","graph_local_density", |
| ] |
|
|
| def normalize_object(raw): |
| name = raw.get("OBJECT_NAME") or raw.get("object_name") or "UNKNOWN" |
| norad = raw.get("NORAD_CAT_ID") or raw.get("norad_cat_id") |
| intl = raw.get("OBJECT_ID") or raw.get("object_id") or "" |
| launch_year = int(intl[:4]) if len(intl) >= 4 and intl[:4].isdigit() else None |
| return { |
| "object_id": str(norad or name), |
| "norad_cat_id": safe_int(norad, 0) or None, |
| "object_name": name, |
| "object_type": raw.get("OBJECT_TYPE") or raw.get("object_type"), |
| "mean_motion": safe_float(raw.get("MEAN_MOTION") or raw.get("mean_motion")), |
| "inclination": safe_float(raw.get("INCLINATION") or raw.get("inclination")), |
| "eccentricity": safe_float(raw.get("ECCENTRICITY") or raw.get("eccentricity")), |
| "raan": safe_float(raw.get("RA_OF_ASC_NODE") or raw.get("raan")), |
| "bstar": safe_float(raw.get("BSTAR") or raw.get("bstar")), |
| "launch_year": launch_year, |
| } |
|
|
| def orbital_shell_key(obj): |
| mm = safe_float(obj.get("mean_motion")) |
| inc = safe_float(obj.get("inclination")) |
| ecc = safe_float(obj.get("eccentricity")) |
| return f"mm:{int(mm)}|inc:{int(inc//5)*5}|ecc:{int(ecc*1000)//10}" |
|
|
| def base_pair_features(a, b): |
| mm1, mm2 = safe_float(a.get("mean_motion")), safe_float(b.get("mean_motion")) |
| inc1, inc2 = safe_float(a.get("inclination")), safe_float(b.get("inclination")) |
| ecc1, ecc2 = safe_float(a.get("eccentricity")), safe_float(b.get("eccentricity")) |
| raan1, raan2 = safe_float(a.get("raan")), safe_float(b.get("raan")) |
| b1, b2 = safe_float(a.get("bstar")), safe_float(b.get("bstar")) |
| ly1, ly2 = safe_int(a.get("launch_year")), safe_int(b.get("launch_year")) |
| same_type = 1 if (a.get("object_type") or "") == (b.get("object_type") or "") else 0 |
| same_shell = 1 if orbital_shell_key(a) == orbital_shell_key(b) else 0 |
| delta_mm = abs(mm1 - mm2) |
| delta_inc = abs(inc1 - inc2) |
| delta_ecc = abs(ecc1 - ecc2) |
| delta_raan = abs(raan1 - raan2) |
| delta_bstar = abs(b1 - b2) |
| launch_gap = abs(ly1 - ly2) if ly1 and ly2 else 25 |
| shell_density_proxy = max(0.0, 10.0 - delta_mm) + max(0.0, 8.0 - delta_inc / 2.0) |
| close_approach_proxy = 1.0 / (1.0 + delta_mm + delta_inc / 10.0 + delta_ecc * 50.0 + delta_raan / 60.0) |
| persistence_proxy = 1.0 if same_shell else 0.25 |
| return { |
| "delta_mean_motion": delta_mm, |
| "delta_inclination": delta_inc, |
| "delta_eccentricity": delta_ecc, |
| "delta_raan": delta_raan, |
| "delta_bstar": delta_bstar, |
| "launch_year_gap": float(launch_gap), |
| "same_object_type": float(same_type), |
| "same_shell": float(same_shell), |
| "shell_density_proxy": float(shell_density_proxy), |
| "close_approach_proxy": float(close_approach_proxy), |
| "persistence_proxy": float(persistence_proxy), |
| } |
|
|
| def combine_features(a, b, trend, graph): |
| return {**base_pair_features(a, b), **trend, **graph} |
|
|
