rebound-rs 4.6.0-alpha.1

Rust wrapper for the REBOUND N-body simulation library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

use rebound_bind as rb;

pub enum L {
    Null,
    Mercurius,
    C4,
    C5,
    Infinity,
    Custom(unsafe extern "C" fn(r: *const rb::reb_simulation, d: f64, dcrit: f64) -> f64),
}

impl From<L>
    for Option<unsafe extern "C" fn(r: *const rb::reb_simulation, d: f64, dcrit: f64) -> f64>
{
    fn from(l: L) -> Self {
        match l {
            L::Null => None,
            L::Mercurius => Some(rb::reb_integrator_mercurius_L_mercury),
            L::C4 => Some(rb::reb_integrator_mercurius_L_C4),
            L::C5 => Some(rb::reb_integrator_mercurius_L_C5),
            L::Infinity => Some(rb::reb_integrator_mercurius_L_infinity),
            L::Custom(f) => Some(f),
        }
    }
}

#[macro_export]
#[doc(hidden)]
macro_rules! _reb_integrator_mercurius_c_fn {
    ($safe_fn:path) => {{
        unsafe extern "C" fn wrapper(
            r: *const $crate::bind::reb_simulation,
            d: f64,
            dcrit: f64,
        ) -> f64 {
            // SAFETY: REBOUND invokes this callback with the active simulation pointer.
            let r_safe = unsafe { &mut *r };
            $safe_fn(r_safe, d, dcrit)
        }

        wrapper
    }};
}

pub use _reb_integrator_mercurius_c_fn as mercurius_c_fn;

pub struct IntegratorMercurius<'a> {
    pub(crate) inner: *mut rb::reb_integrator_mercurius,
    pub(crate) _marker: core::marker::PhantomData<&'a mut rb::reb_simulation>,
}

impl<'a> IntegratorMercurius<'a> {
    pub fn set_r_crit_hill(self, r_crit_hill: f64) -> Self {
        unsafe {
            (*self.inner).r_crit_hill = r_crit_hill;
        }
        self
    }

    pub fn set_recalculate_coordinates_this_timestep(
        self,
        recalculate_coordinates_this_timestep: u32,
    ) -> Self {
        unsafe {
            (*self.inner).recalculate_coordinates_this_timestep =
                recalculate_coordinates_this_timestep;
        }
        self
    }

    pub fn set_recalculate_r_crit_this_timestep(
        self,
        recalculate_r_crit_this_timestep: u32,
    ) -> Self {
        unsafe {
            (*self.inner).recalculate_r_crit_this_timestep = recalculate_r_crit_this_timestep;
        }
        self
    }

    pub fn set_safe_mode(self, safe_mode: u32) -> Self {
        unsafe {
            (*self.inner).safe_mode = safe_mode;
        }
        self
    }

    pub fn set_l(self, l: L) -> Self {
        unsafe {
            (*self.inner).L = l.into();
        }
        self
    }

    pub fn r_crit_hill(&self) -> f64 {
        unsafe { (*self.inner).r_crit_hill }
    }

    pub fn recalculate_coordinates_this_timestep(&self) -> u32 {
        unsafe { (*self.inner).recalculate_coordinates_this_timestep }
    }

    pub fn recalculate_r_crit_this_timestep(&self) -> u32 {
        unsafe { (*self.inner).recalculate_r_crit_this_timestep }
    }

    pub fn safe_mode(&self) -> u32 {
        unsafe { (*self.inner).safe_mode }
    }
}