use copp::diag::CoppError;
use copp::path::{Jet3, Path, sin};
use copp::robot::Robot;
use copp::solver::reach_set2::{
ReachSet2OptionsBuilder, Topp2ProblemBuilder, reach_set2_backward, reach_set2_bidirectional,
};
use std::f64::consts::PI;
fn main() -> Result<(), CoppError> {
let path = Path::from_parametric(
|s: Jet3| {
vec![
sin(2.0 * PI * s + 0.0),
sin(3.0 * PI * s + 0.3),
sin(5.0 * PI * s + 0.7),
]
},
0.0,
1.0,
)?;
let n = 1001;
let s: Vec<f64> = (0..n).map(|j| j as f64 / (n - 1) as f64).collect();
const DIM: usize = 3;
let mut robot = Robot::with_capacity(DIM, n);
let vel_max = vec![1.0; DIM];
let vel_min = vec![-1.0; DIM];
let acc_max = vec![1.0; DIM];
let acc_min = vec![-1.0; DIM];
robot
.with_s(s.as_slice())?
.with_q_from_path_2nd(&path, 0, n)?
.with_axial_velocity((vel_max.as_slice(), n), (vel_min.as_slice(), n), 0)?
.with_axial_acceleration((acc_max.as_slice(), n), (acc_min.as_slice(), n), 0)?;
let idx_s_interval = (0, n - 1); let a_boundary = (0.0, 0.0); let problem = Topp2ProblemBuilder::new(&robot, idx_s_interval, a_boundary).build()?;
let options = ReachSet2OptionsBuilder::new().build()?;
let reach_back = reach_set2_backward(&problem, &options)?;
let reach_bidir = reach_set2_bidirectional(&problem, &options)?;
println!("reach_set2 done.");
println!("dim = {DIM}, N = {n}");
println!(
"backward-only: a_max.len() = {}, a_min.len() = {}",
reach_back.a_max.len(),
reach_back.a_min.len()
);
println!(
"bidirectional: a_max.len() = {}, a_min.len() = {}",
reach_bidir.a_max.len(),
reach_bidir.a_min.len()
);
let k0 = 0;
let km = n / 2;
let k1 = n - 1;
println!(
"bidirectional bounds @k=0/mid/end: [{:.6}, {:.6}], [{:.6}, {:.6}], [{:.6}, {:.6}]",
reach_bidir.a_min[k0],
reach_bidir.a_max[k0],
reach_bidir.a_min[km],
reach_bidir.a_max[km],
reach_bidir.a_min[k1],
reach_bidir.a_max[k1],
);
Ok(())
}