roche/
lib.rs

1use pyo3::prelude::*;
2
3pub mod blink;
4pub use blink::blink;
5
6pub mod constants;
7
8pub mod disc_eclipse;
9pub use disc_eclipse::disc_eclipse;
10
11pub mod errors;
12
13pub mod face;
14pub use face::*;
15
16pub mod fblink;
17pub use fblink::*;
18
19pub mod ingress_egress;
20pub use ingress_egress::ingress_egress;
21
22pub mod jacobi;
23pub use jacobi::*;
24
25pub mod lobes;
26pub use lobes::lobe1;
27pub use lobes::lobe2;
28
29pub mod phases;
30pub use phases::*;
31
32pub mod planck;
33pub use planck::*;
34
35pub mod point;
36pub use point::Point;
37
38pub mod pot_min;
39pub use pot_min::*;
40
41pub mod potential;
42pub use potential::*;
43
44pub mod ref_sphere;
45pub use ref_sphere::*;
46
47pub mod roche_context;
48pub use roche_context::RocheContext;
49
50pub mod set_earth;
51pub use set_earth::*;
52
53pub mod solve_triads;
54pub use solve_triads::findi;
55
56pub mod sphere_eclipse;
57pub use sphere_eclipse::*;
58
59pub mod star_eclipse;
60pub use star_eclipse::star_eclipse;
61
62pub mod stream_physics;
63pub use stream_physics::*;
64
65pub mod vec3;
66pub use vec3::Vec3;
67
68pub mod vel_transform;
69pub use vel_transform::vel_transform;
70
71pub mod vstream_physics;
72pub use vstream_physics::*;
73
74pub mod x_lagrange;
75pub use x_lagrange::*;
76
77pub mod zeta_rlobe_eggleton;
78pub use zeta_rlobe_eggleton::*;
79
80
81#[pyclass(from_py_object, eq, eq_int)]
82#[derive(Debug, PartialEq, Eq, Clone, Copy)]
83pub enum Star {
84    Primary = 1,
85    Secondary = 2,
86}
87
88pub type Etype = Vec<(f64, f64)>;
89
90// Python module
91#[pymodule]
92fn roche(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
93    m.add_class::<point::Point>()?;
94    m.add_class::<Star>()?;
95    m.add_class::<vec3::Vec3>()?;
96    m.add_function(wrap_pyfunction!(face::face, m)?)?;
97    m.add_function(wrap_pyfunction!(fblink::fblink, m)?)?;
98    m.add_function(wrap_pyfunction!(ingress_egress::ingress_egress_wrapper, m)?)?;
99    m.add_function(wrap_pyfunction!(jacobi::jacobi, m)?)?;
100    m.add_function(wrap_pyfunction!(lobes::lobe1, m)?)?;
101    m.add_function(wrap_pyfunction!(lobes::lobe2, m)?)?;
102    m.add_function(wrap_pyfunction!(lobes::vlobe1, m)?)?;
103    m.add_function(wrap_pyfunction!(lobes::vlobe2, m)?)?;
104    m.add_function(wrap_pyfunction!(phases::wdradius, m)?)?;
105    m.add_function(wrap_pyfunction!(phases::wdphases, m)?)?;
106    m.add_function(wrap_pyfunction!(phases::bsphases, m)?)?;
107    m.add_function(wrap_pyfunction!(planck::planck, m)?)?;
108    m.add_function(wrap_pyfunction!(planck::dplanck, m)?)?;
109    m.add_function(wrap_pyfunction!(planck::dlpdlt, m)?)?;
110    m.add_function(wrap_pyfunction!(potential::rpot, m)?)?;
111    m.add_function(wrap_pyfunction!(potential::rpot1, m)?)?;
112    m.add_function(wrap_pyfunction!(potential::rpot2, m)?)?;
113    m.add_function(wrap_pyfunction!(potential::drpot, m)?)?;
114    m.add_function(wrap_pyfunction!(potential::drpot1, m)?)?;
115    m.add_function(wrap_pyfunction!(potential::drpot2, m)?)?;
116    m.add_function(wrap_pyfunction!(potential::rpot_val, m)?)?;
117    m.add_function(wrap_pyfunction!(potential::rpot_val_grad, m)?)?;
118    m.add_function(wrap_pyfunction!(ref_sphere::ref_sphere, m)?)?;
119    m.add_function(wrap_pyfunction!(set_earth::set_earth_iangle, m)?)?;
120    m.add_function(wrap_pyfunction!(set_earth::set_earth, m)?)?;
121    m.add_function(wrap_pyfunction!(sphere_eclipse::sphere_eclipse_wrapper, m)?)?;
122    m.add_function(wrap_pyfunction!(sphere_eclipse::sphere_eclipse_vector_wrapper, m)?)?;
123    m.add_function(wrap_pyfunction!(solve_triads::findi, m)?)?;
124    m.add_function(wrap_pyfunction!(solve_triads::findq, m)?)?;
125    m.add_function(wrap_pyfunction!(solve_triads::findphi, m)?)?;
126    m.add_function(wrap_pyfunction!(stream_physics::stradv_wrapper, m)?)?;
127    m.add_function(wrap_pyfunction!(stream_physics::rocacc, m)?)?;
128    m.add_function(wrap_pyfunction!(stream_physics::strinit, m)?)?;
129    m.add_function(wrap_pyfunction!(stream_physics::stream, m)?)?;
130    m.add_function(wrap_pyfunction!(stream_physics::streamr, m)?)?;
131    m.add_function(wrap_pyfunction!(stream_physics::strmnx_wrapper, m)?)?;
132    m.add_function(wrap_pyfunction!(stream_physics::brightspot_position, m)?)?;
133    m.add_function(wrap_pyfunction!(vstream_physics::vstream_reg, m)?)?;
134    m.add_function(wrap_pyfunction!(x_lagrange::x_l1, m)?)?;
135    m.add_function(wrap_pyfunction!(x_lagrange::x_l1_1, m)?)?;
136    m.add_function(wrap_pyfunction!(x_lagrange::x_l1_2, m)?)?;
137    m.add_function(wrap_pyfunction!(x_lagrange::x_l2, m)?)?;
138    m.add_function(wrap_pyfunction!(x_lagrange::x_l3, m)?)?;
139    m.add_function(wrap_pyfunction!(zeta_rlobe_eggleton::zeta_rlobe_eggleton, m)?)?;
140    m.add_function(wrap_pyfunction!(zeta_rlobe_eggleton::dzetadq_rlobe_eggleton, m)?)?;
141    Ok(())
142}
roche/lib.rs

roche/
lib.rs
