mirror/rebound-collisions
1
0
Fork 0
mirror of https://github.com/Findus23/rebound-collisions.git synced 2024-09-19 15:53:48 +02:00
Code Releases Activity

better heartbeat checks

Browse source
This commit is contained in:
Lukas Winkler 2021-08-20 18:00:04 +02:00
parent 233da94f10
commit 479699bab3
Signed by: lukas
GPG key ID: 54DE4D798D244853
1 changed files with 36 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
heartbeat/heartbeat.c
View file

@ -11,9 +11,11 @@ struct hb_event {
};
struct hb_event hb_escapes[500];
struct hb_event hb_wide_orbits[500];
struct hb_event hb_sun_collisions[500];
int hb_escape_index = 0;
int hb_wide_orbit_index = 0;
int hb_sun_collision_index = 0;
int needs_synchronize = 0;
@ -25,13 +27,24 @@ void init_logfile(char *filename) {
logfile = f;
}
double elliptical_orbit_velocity(struct reb_simulation *sim, double m0, double m1, double a, double r)
// Computes the orbital velocity on a bound orbit with semi-major axis 'a' of masses 'm0' and 'm1' at distance 'r'.
// uses Vis-Viva equation
{
double v_sqr = sim->G * (m0 + m1) * (2.0 / r - 1.0 / a);
return sqrt(v_sqr);
}
void heartbeat(struct reb_simulation *sim) {
if ((sim->steps_done % 10) == 0) {
if ((sim->steps_done % 100) == 0) {
const struct reb_particle *const particles = sim->particles;
int N = sim->N - sim->N_var;
for (int i = 1; i < N; i++) { // skip sun
struct reb_particle p = particles[i];
double distance_squared = p.x * p.x + p.y * p.y + p.z * p.z;
struct reb_orbit tmp_orbit = reb_tools_particle_to_orbit(sim->G, p, sim->particles[0]);
double perihelion_dist = tmp_orbit.a * (1.0 - tmp_orbit.e);
if (distance_squared > max_distance_from_sun_squared) {
printf("remove %u at t=%f (max)\n", p.hash, sim->t);
reb_remove_by_hash(sim, p.hash, 1);
@ -41,7 +54,8 @@ void heartbeat(struct reb_simulation *sim) {
hb_escape_index++;
needs_synchronize = 1;
} else if (distance_squared < min_distance_from_sun_squared) {
} else if (distance_squared < min_distance_from_sun_squared ||
perihelion_dist * perihelion_dist < min_distance_from_sun_squared) {
printf("remove %u at t=%f (min)\n", p.hash, sim->t);
double mass = p.m;
reb_remove_by_hash(sim, p.hash, 1);
@ -55,7 +69,18 @@ void heartbeat(struct reb_simulation *sim) {
hb_sun_collision_index++;
needs_synchronize = 1;
} else if (perihelion_dist > 11.) {
// remove bodies if their perihel distance is above 11AU
printf("remove %u at t=%f (max)\n", p.hash, sim->t);
reb_remove_by_hash(sim, p.hash, 1);
hb_wide_orbits[hb_wide_orbit_index].hash = p.hash;
hb_wide_orbits[hb_wide_orbit_index].time = sim->t;
hb_wide_orbits[hb_wide_orbit_index].new = 1;
hb_escape_index++;
needs_synchronize = 1;
}
if (needs_synchronize) {
printf("distance: %f\n", sqrt(distance_squared));
needs_synchronize = 0;
@ -64,6 +89,15 @@ void heartbeat(struct reb_simulation *sim) {
reb_integrator_synchronize(sim);
sim->ri_mercurius.recalculate_coordinates_this_timestep = 1;
sim->ri_mercurius.recalculate_dcrit_this_timestep = 1;
} else {
double perihelion_vel = elliptical_orbit_velocity(
sim,
sim->particles[0].m, sim->particles[i].m,
tmp_orbit.a, perihelion_dist);
double T_eff = 2.0 * M_PI * perihelion_dist / perihelion_vel;
if (T_eff < sim->dt * 20) {
printf("Warning: effective orbital period too low (%f < %f)\n", T_eff, sim->dt * 20);
}
}
}
}