Update 2024 YR4 notebook

B612-Asteroid-Institute · Feb 25, 2025 · 36da4b6 · 36da4b6
1 parent a45d707
commit 36da4b6
Showing 1 changed file with 19 additions and 8 deletions.
diff --git a/examples/2024_yr4_impact_risk.ipynb b/examples/2024_yr4_impact_risk.ipynb
@@ -58,17 +58,26 @@
    "outputs": [],
    "source": [
     "# Now we initialize our propagator, and pass it to our calculate_impacts function\n",
-    "from adam_core.dynamics.impacts import calculate_impacts\n",
+    "from adam_core.dynamics.impacts import calculate_impacts, CollisionConditions\n",
     "from adam_assist import ASSISTPropagator\n",
     "\n",
     "propagator = ASSISTPropagator()\n",
     "\n",
+    "# Define the collision conditions including a potential impact on the Moon\n",
+    "conditions = CollisionConditions.from_kwargs(\n",
+    "    condition_id=[\"Earth\", \"Moon\"],\n",
+    "    collision_object_name=[\"Earth\", \"Moon\"],\n",
+    "    collision_distance=[6420, 1740],\n",
+    "    stopping_condition=[True, True],\n",
+    ")\n",
+    "\n",
     "variants, impacts = calculate_impacts(\n",
     "    orbit,\n",
     "    days_until_thirty_days_after_impact[0].as_py(),\n",
     "    propagator,\n",
-    "    num_samples=10000,\n",
+    "    num_samples=50000,\n",
     "    processes=10, # Multiprocessing speeds things up if you have the CPUs\n",
+    "    conditions=conditions,\n",
     ")\n",
     "\n",
     "print(variants, impacts)"
@@ -83,8 +92,8 @@
     "# Now we can summarize the returns, a simple ratio in our case\n",
     "# with only 1 orbit considered\n",
     "from adam_core.dynamics.impacts import calculate_impact_probabilities\n",
-    "ip = calculate_impact_probabilities(variants, impacts)\n",
-    "print(ip.to_dataframe())"
+    "ip = calculate_impact_probabilities(variants, impacts, conditions=conditions)\n",
+    "ip.to_dataframe()"
    ]
   },
   {
@@ -114,13 +123,13 @@
    "source": [
     "from adam_core.dynamics.plots import generate_impact_visualization_data\n",
     "\n",
-    "propagated_best_fit_orbit, propagated_variants = generate_impact_visualization_data(\n",
+    "propagation_times, propagated_best_fit_orbit, propagated_variants = generate_impact_visualization_data(\n",
     "    orbit,\n",
     "    variants,\n",
     "    impacts,\n",
     "    propagator,\n",
     "    time_step=5,\n",
-    "    time_range=120,\n",
+    "    time_range=60,\n",
     "    max_processes=None\n",
     ")"
    ]
@@ -134,11 +143,13 @@
     "from adam_core.dynamics.plots import plot_impact_simulation\n",
     "\n",
     "fig = plot_impact_simulation(\n",
+    "    propagation_times,\n",
     "    propagated_best_fit_orbit, \n",
     "    propagated_variants, \n",
     "    impacts, \n",
-    "    grid=True, \n",
     "    title=\"2024 YR4 Impact Simulation\",\n",
+    "    downsample_non_impactors=True,\n",
+    "    downsample_size=0.02\n",
     ")\n",
     "fig.show()"
    ]
@@ -167,7 +178,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.7"
+   "version": "3.11.5"
   }
  },
  "nbformat": 4,