nyx_space/md/trajectory/
traj_it.rs

1/*
2    Nyx, blazing fast astrodynamics
3    Copyright (C) 2018-onwards Christopher Rabotin <christopher.rabotin@gmail.com>
4
5    This program is free software: you can redistribute it and/or modify
6    it under the terms of the GNU Affero General Public License as published
7    by the Free Software Foundation, either version 3 of the License, or
8    (at your option) any later version.
9
10    This program is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13    GNU Affero General Public License for more details.
14
15    You should have received a copy of the GNU Affero General Public License
16    along with this program.  If not, see <https://www.gnu.org/licenses/>.
17*/
18
19use super::{Interpolatable, Traj};
20use crate::linalg::allocator::Allocator;
21use crate::linalg::DefaultAllocator;
22use crate::time::TimeSeries;
23
24pub struct TrajIterator<'a, S: Interpolatable>
25where
26    DefaultAllocator: Allocator<S::VecLength> + Allocator<S::Size> + Allocator<S::Size, S::Size>,
27{
28    pub time_series: TimeSeries,
29    /// A shared pointer to the original trajectory.
30    pub traj: &'a Traj<S>,
31}
32
33impl<S: Interpolatable> Iterator for TrajIterator<'_, S>
34where
35    DefaultAllocator: Allocator<S::VecLength> + Allocator<S::Size> + Allocator<S::Size, S::Size>,
36{
37    type Item = S;
38
39    fn next(&mut self) -> Option<Self::Item> {
40        match self.time_series.next() {
41            Some(next_epoch) => match self.traj.at(next_epoch) {
42                Ok(item) => Some(item),
43                Err(e) => {
44                    if next_epoch >= self.traj.first().epoch()
45                        && next_epoch <= self.traj.last().epoch()
46                    {
47                        let msg = format!(
48                            "{e} out of bounds in {}! Please submit bug report with exported traj",
49                            self.traj
50                        );
51                        if log_enabled!(log::Level::Error) {
52                            error!("{msg}");
53                        } else {
54                            eprintln!("{msg}");
55                        };
56                    }
57                    None
58                }
59            },
60            None => None,
61        }
62    }
63}
nyx_space/md/trajectory/traj_it.rs

nyx_space/md/trajectory/
traj_it.rs
