mirror of
https://github.com/Findus23/rebound-collisions.git
synced 2024-09-19 15:53:48 +02:00
better heartbeat checks
This commit is contained in:
parent
233da94f10
commit
479699bab3
1 changed files with 36 additions and 2 deletions
|
@ -11,9 +11,11 @@ struct hb_event {
|
||||||
};
|
};
|
||||||
|
|
||||||
struct hb_event hb_escapes[500];
|
struct hb_event hb_escapes[500];
|
||||||
|
struct hb_event hb_wide_orbits[500];
|
||||||
struct hb_event hb_sun_collisions[500];
|
struct hb_event hb_sun_collisions[500];
|
||||||
|
|
||||||
int hb_escape_index = 0;
|
int hb_escape_index = 0;
|
||||||
|
int hb_wide_orbit_index = 0;
|
||||||
int hb_sun_collision_index = 0;
|
int hb_sun_collision_index = 0;
|
||||||
|
|
||||||
int needs_synchronize = 0;
|
int needs_synchronize = 0;
|
||||||
|
@ -25,13 +27,24 @@ void init_logfile(char *filename) {
|
||||||
logfile = f;
|
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) {
|
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;
|
const struct reb_particle *const particles = sim->particles;
|
||||||
int N = sim->N - sim->N_var;
|
int N = sim->N - sim->N_var;
|
||||||
for (int i = 1; i < N; i++) { // skip sun
|
for (int i = 1; i < N; i++) { // skip sun
|
||||||
struct reb_particle p = particles[i];
|
struct reb_particle p = particles[i];
|
||||||
double distance_squared = p.x * p.x + p.y * p.y + p.z * p.z;
|
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) {
|
if (distance_squared > max_distance_from_sun_squared) {
|
||||||
printf("remove %u at t=%f (max)\n", p.hash, sim->t);
|
printf("remove %u at t=%f (max)\n", p.hash, sim->t);
|
||||||
reb_remove_by_hash(sim, p.hash, 1);
|
reb_remove_by_hash(sim, p.hash, 1);
|
||||||
|
@ -41,7 +54,8 @@ void heartbeat(struct reb_simulation *sim) {
|
||||||
|
|
||||||
hb_escape_index++;
|
hb_escape_index++;
|
||||||
needs_synchronize = 1;
|
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);
|
printf("remove %u at t=%f (min)\n", p.hash, sim->t);
|
||||||
double mass = p.m;
|
double mass = p.m;
|
||||||
reb_remove_by_hash(sim, p.hash, 1);
|
reb_remove_by_hash(sim, p.hash, 1);
|
||||||
|
@ -55,7 +69,18 @@ void heartbeat(struct reb_simulation *sim) {
|
||||||
|
|
||||||
hb_sun_collision_index++;
|
hb_sun_collision_index++;
|
||||||
needs_synchronize = 1;
|
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) {
|
if (needs_synchronize) {
|
||||||
printf("distance: %f\n", sqrt(distance_squared));
|
printf("distance: %f\n", sqrt(distance_squared));
|
||||||
needs_synchronize = 0;
|
needs_synchronize = 0;
|
||||||
|
@ -64,6 +89,15 @@ void heartbeat(struct reb_simulation *sim) {
|
||||||
reb_integrator_synchronize(sim);
|
reb_integrator_synchronize(sim);
|
||||||
sim->ri_mercurius.recalculate_coordinates_this_timestep = 1;
|
sim->ri_mercurius.recalculate_coordinates_this_timestep = 1;
|
||||||
sim->ri_mercurius.recalculate_dcrit_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);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue