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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

use rebound_bind as rb;

pub enum S {
    Null,
    Default,
    Custom(unsafe extern "C" fn(r: *mut rb::reb_simulation, i: u32, j: u32) -> i32),
}

impl From<S> for Option<unsafe extern "C" fn(r: *mut rb::reb_simulation, i: u32, j: u32) -> i32> {
    fn from(value: S) -> Self {
        match value {
            S::Null => None,
            S::Default => Some(rb::reb_integrator_trace_switch_default),
            S::Custom(f) => Some(f),
        }
    }
}

pub enum SPeri {
    Null,
    Default,
    None,
    Custom(unsafe extern "C" fn(r: *mut rb::reb_simulation, j: u32) -> i32),
}

impl From<SPeri> for Option<unsafe extern "C" fn(r: *mut rb::reb_simulation, j: u32) -> i32> {
    fn from(value: SPeri) -> Self {
        match value {
            SPeri::Null => None,
            SPeri::Default => Some(rb::reb_integrator_trace_switch_peri_default),
            SPeri::None => Some(rb::reb_integrator_trace_switch_peri_none),
            SPeri::Custom(f) => Some(f),
        }
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum PeriMode {
    PartialBs = rb::reb_integrator_trace_REB_TRACE_PERI_PARTIAL_BS as isize,
    FullBs = rb::reb_integrator_trace_REB_TRACE_PERI_FULL_BS as isize,
    FullIas15 = rb::reb_integrator_trace_REB_TRACE_PERI_FULL_IAS15 as isize,
}

impl From<PeriMode> for rb::reb_integrator_trace__bindgen_ty_1 {
    fn from(value: PeriMode) -> Self {
        match value {
            PeriMode::PartialBs => rb::reb_integrator_trace_REB_TRACE_PERI_PARTIAL_BS,
            PeriMode::FullBs => rb::reb_integrator_trace_REB_TRACE_PERI_FULL_BS,
            PeriMode::FullIas15 => rb::reb_integrator_trace_REB_TRACE_PERI_FULL_IAS15,
        }
    }
}

impl PeriMode {
    fn from_raw(value: rb::reb_integrator_trace__bindgen_ty_1) -> Option<Self> {
        match value {
            rb::reb_integrator_trace_REB_TRACE_PERI_PARTIAL_BS => Some(Self::PartialBs),
            rb::reb_integrator_trace_REB_TRACE_PERI_FULL_BS => Some(Self::FullBs),
            rb::reb_integrator_trace_REB_TRACE_PERI_FULL_IAS15 => Some(Self::FullIas15),
            _ => None,
        }
    }
}

#[macro_export]
#[doc(hidden)]
macro_rules! _reb_integrator_trace_s_c_fn {
    ($safe_fn:path) => {{
        unsafe extern "C" fn wrapper(r: *mut $crate::bind::reb_simulation, i: u32, j: u32) -> i32 {
            let r_safe = unsafe { &mut *r };
            $safe_fn(r_safe, i, j)
        }

        wrapper
    }};
}

pub use _reb_integrator_trace_s_c_fn as trace_s_c_fn;

#[macro_export]
#[doc(hidden)]
macro_rules! _reb_integrator_trace_s_peri_c_fn {
    ($safe_fn:path) => {{
        unsafe extern "C" fn wrapper(r: *mut $crate::bind::reb_simulation, j: u32) -> i32 {
            let r_safe = unsafe { &mut *r };
            $safe_fn(r_safe, j)
        }

        wrapper
    }};
}

pub use _reb_integrator_trace_s_peri_c_fn as trace_s_peri_c_fn;

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

impl<'a> IntegratorTrace<'a> {
    pub fn set_s(self, s: S) -> Self {
        unsafe {
            (*self.inner).S = s.into();
        }
        self
    }

    pub fn set_s_peri(self, s_peri: SPeri) -> Self {
        unsafe {
            (*self.inner).S_peri = s_peri.into();
        }
        self
    }

    pub fn set_peri_mode(self, peri_mode: PeriMode) -> Self {
        unsafe {
            (*self.inner).peri_mode = peri_mode.into();
        }
        self
    }

    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_peri_crit_eta(self, peri_crit_eta: f64) -> Self {
        unsafe {
            (*self.inner).peri_crit_eta = peri_crit_eta;
        }
        self
    }

    pub fn peri_mode(&self) -> Option<PeriMode> {
        unsafe { PeriMode::from_raw((*self.inner).peri_mode) }
    }

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

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