From 54b8ac3695026cc5be1b03ce105bf5a7110283aa Mon Sep 17 00:00:00 2001
From: Lukas Winkler <git@lw1.at>
Date: Mon, 8 Feb 2021 19:03:51 +0100
Subject: [PATCH] calculate innermost semimajor axis

---
 utils.py     | 11 +++++++++++
 water_sim.py | 13 ++++++++++---
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/utils.py b/utils.py
index 6ee6d8a..73b5d71 100644
--- a/utils.py
+++ b/utils.py
@@ -11,10 +11,14 @@ from matplotlib.axes import Axes
 from matplotlib.figure import Figure
 from numpy import linalg
 from rebound import Simulation, Orbit, Particle, OrbitPlot
+from scipy.constants import pi, gravitational_constant
 from setproctitle import setproctitle
 
 from extradata import ExtraData
 
+solar_radius = 6.957e8
+solar_mass = 1.9885e+30
+
 
 def unique_hash() -> c_uint32:
     """
@@ -40,6 +44,13 @@ def innermost_period(sim: Simulation) -> float:
     return orb.P
 
 
+def third_kepler_law(orbital_period: float):
+    return (
+                   gravitational_constant * solar_mass / (4 * pi ** 2)
+                   * orbital_period ** 2
+           ) ** (1 / 3)
+
+
 def total_momentum(sim: Simulation) -> float:
     total = 0
     for p in sim.particles:
diff --git a/water_sim.py b/water_sim.py
index 8c5f385..4dc69de 100644
--- a/water_sim.py
+++ b/water_sim.py
@@ -8,12 +8,13 @@ from sys import argv
 from rebound import Simulation, Particle, NoParticles, Escape, \
     SimulationArchive
 from rebound.simulation import POINTER_REB_SIM, reb_collision
-from scipy.constants import astronomical_unit, mega
+from scipy.constants import astronomical_unit, mega, year
 
 from extradata import ExtraData, ParticleData
 from merge import merge_particles, handle_escape
 from radius_utils import radius
-from utils import unique_hash, filename_from_argv, innermost_period, total_momentum, process_friendlyness, total_mass
+from utils import unique_hash, filename_from_argv, innermost_period, total_momentum, process_friendlyness, total_mass, \
+    third_kepler_law, solar_radius
 
 MIN_TIMESTEP_PER_ORBIT = 20
 PERFECT_MERGING = False
@@ -54,6 +55,12 @@ def main(fn: Path, testrun=False):
         extradata.meta.num_savesteps = num_savesteps
         extradata.meta.perfect_merging = PERFECT_MERGING
 
+        innermost_semimajor_axis = third_kepler_law(
+            orbital_period=sim.dt * year * MIN_TIMESTEP_PER_ORBIT
+        ) / astronomical_unit
+        print(f"innermost semimajor axis is {innermost_semimajor_axis}")
+
+
         initcon = INITCON_FILE.read_text()
         num_embryos = int(re.search(r"Generated (\d+) minor bodies", initcon, re.MULTILINE).group(1))
         num_planetesimals = int(re.search(r"Generated (\d+) small bodies", initcon, re.MULTILINE).group(1))
@@ -92,7 +99,7 @@ def main(fn: Path, testrun=False):
             )
 
             if columns[1] == 0:  # that should not be needed, but nevertheless is
-                part = Particle(m=columns[0], hash=hash)
+                part = Particle(m=columns[0], hash=hash, r=solar_radius / astronomical_unit)
             else:
                 part = Particle(
                     m=columns[0], a=columns[1], e=columns[2],