void midpoint_integrate_impl(
     const StateSpace& space,
     const StateSpaceSystem& sys,
     const typename pp::topology_traits<StateSpace>::point_type& start_point,
     typename pp::topology_traits<StateSpace>::point_type& end_point,
     const InputTrajectory& u_traj,
     double start_time,
     double end_time,
     double time_step) {
   typedef typename pp::topology_traits<StateSpace>::point_type PointType;
   typedef typename pp::topology_traits<StateSpace>::point_difference_type PointDiffType;
   
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::const_waypoint_descriptor InputWaypoint;
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::point_type InputType;
   std::pair< InputWaypoint, InputType> u_wp = u_traj.get_waypoint_at_time(start_time);
   
   PointDiffType dp = sys.get_state_derivative(space, start_point, u_wp.second.pt, start_time);
   double t = start_time;
   end_point = start_point;
   
   while(((time_step > 0.0) && (t < end_time)) || 
         ((time_step < 0.0) && (t > end_time))) {
     PointType mid_p = space.adjust(end_point, (0.5 * time_step) * dp);
     t += 0.5 * time_step;
     u_wp = u_traj.move_time_diff_from(u_wp, 0.5 * time_step);
     dp = sys.get_state_derivative(space, mid_p, u_wp.second.pt, t);
     
     end_point = space.adjust(end_point, time_step * dp);
     t += 0.5 * time_step;
     u_wp = u_traj.move_time_diff_from(u_wp, 0.5 * time_step);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
   };
 };
 void runge_kutta4_integrate_impl(
     const StateSpace& space,
     const StateSpaceSystem& sys,
     const typename pp::topology_traits<StateSpace>::point_type& start_point,
     typename pp::topology_traits<StateSpace>::point_type& end_point,
     const InputTrajectory& u_traj,
     double start_time,
     double end_time,
     double time_step) {
   typedef typename pp::topology_traits<StateSpace>::point_type PointType;
   typedef typename pp::topology_traits<StateSpace>::point_difference_type PointDiffType;
   
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::const_waypoint_descriptor InputWaypoint;
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::point_type InputType;
   std::pair< InputWaypoint, InputType> u_wp = u_traj.get_waypoint_at_time(start_time);
   
   PointDiffType dp = sys.get_state_derivative(space, start_point, u_wp.second.pt, start_time);
   double t = start_time;
   end_point = start_point;
   
   while(((time_step > 0.0) && (t < end_time)) || 
         ((time_step < 0.0) && (t > end_time))) {
   
     PointType w = end_point;
     PointDiffType k1 = time_step * dp;
     end_point = space.adjust(end_point, 0.5 * k1);
     
     t += time_step * 0.5;
     u_wp = u_traj.move_time_diff_from(u_wp, 0.5 * time_step);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k2 = time_step * dp;
     end_point = space.adjust(w, 0.5 * k2);
     
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k3 = time_step * dp;
     end_point = space.adjust(w, k3);
     
     t += time_step * 0.5;
     u_wp = u_traj.move_time_diff_from(u_wp, 0.5 * time_step);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     end_point = space.adjust(end_point, (1.0 / 6.0) * k1 + (2.0 / 6.0) * k2 + (time_step / 6.0) * dp - (2.0/3.0) * k3);
     
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
   };
 };
 void dormand_prince45_integrate_impl(
     const StateSpace& space,
     const StateSpaceSystem& sys,
     const typename pp::topology_traits<StateSpace>::point_type& start_point,
     typename pp::topology_traits<StateSpace>::point_type& end_point,
     const InputTrajectory& u_traj,
     double start_time,
     double end_time,
     double time_step,
     double tolerance,
     double min_step,
     double max_step) {
   using std::fabs;
   using std::pow;
   using ReaK::to_vect;
   typedef typename pp::topology_traits<StateSpace>::point_type PointType;
   typedef typename pp::topology_traits<StateSpace>::point_difference_type PointDiffType;
   
   if ((time_step == 0.0) ||
       ((time_step > 0.0) && (start_time > end_time)) ||
       ((time_step < 0.0) && (end_time > start_time)) ||
       (tolerance <= 0.0) ||
       (min_step > max_step))
     throw impossible_integration(start_time, end_time, time_step);
   
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::const_waypoint_descriptor InputWaypoint;
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::point_type InputType;
   std::pair< InputWaypoint, InputType> u_wp = u_traj.get_waypoint_at_time(start_time);
   
   PointDiffType dp = sys.get_state_derivative(space, start_point, u_wp.second.pt, start_time);
   double t = start_time;
   end_point = start_point;
   
   while(((time_step > 0.0) && (t < end_time)) || 
         ((time_step < 0.0) && (t > end_time))) {
     
     PointType prevY = end_point;
     PointDiffType k1 = time_step * dp;
     end_point = space.adjust(end_point, 0.2 * k1);
     
     t += time_step / 5.0;
     u_wp = u_traj.move_time_diff_from(u_wp, time_step / 5.0);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k2 = time_step * dp;
     end_point = space.adjust(prevY, (3.0 / 40.0) * k1 + (9.0 / 40.0) * k2);
     
     t += time_step / 10.0;
     u_wp = u_traj.move_time_diff_from(u_wp, time_step / 10.0);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k3 = time_step * dp;
     end_point = space.adjust(prevY, (44.0 / 45.0) * k1 - (56.0 / 15.0) * k2 + (32.0 / 9.0) * k3);
     
     t += time_step / 2.0;
     u_wp = u_traj.move_time_diff_from(u_wp, time_step / 2.0);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k4 = time_step * dp;
     end_point = space.adjust(prevY, (19372.0 / 6561.0) * k1 - (25360.0 / 2187.0) * k2 + (64448.0 / 6561.0) * k3 - (212.0 / 729.0) * k4);
     
     t += 4.0 * time_step / 45.0;
     u_wp = u_traj.move_time_diff_from(u_wp, 4.0 * time_step / 45.0);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k5 = time_step * dp;
     end_point = space.adjust(prevY, (9017.0 / 3168.0) * k1 - (355.0 / 33.0) * k2 - (46732.0 / 5247.0) * k3 + (49.0 / 176.0) * k4 - (5103.0 / 18656.0) * k5);
     
     t += time_step / 9.0;
     u_wp = u_traj.move_time_diff_from(u_wp, time_step / 9.0);
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k6 = time_step * dp;
     end_point = space.adjust(prevY, (35.0 / 384.0) * k1 + (500.0 / 1113.0) * k3 + (125.0 / 192.0) * k4 - (2187.0 / 6784.0) * k5 + (11.0 / 84.0) * k6);
     
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     PointDiffType k7 = time_step * dp;
     
     vect_n<double> err_vect = to_vect<double>((5170.0 / 57600.0) * k1 + (7571.0 / 16695.0) * k3 + (393.0 / 640.0) * k4 - (92097.0 / 339200.0) * k5 + (187.0 / 2100.0) * k6 - (39.0 / 40.0) * k7);
     double Rmax = 0.0;
     std::size_t worst_DOF = 0;
     for(std::size_t i = 0; i < err_vect.size(); ++i) {
       double R = fabs(err_vect[i] / time_step);
       if(R > Rmax) {
         Rmax = R;
         worst_DOF = i;
       };
     };
     
     if(Rmax > tolerance) {
       if(fabs(time_step) <= min_step)
         throw untolerable_integration(tolerance, Rmax, worst_DOF, time_step, t);
       
       t -= time_step;
       u_wp = u_traj.move_time_diff_from(u_wp, -time_step);
       end_point = prevY;
       dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
       double R = 0.84 * pow(tolerance / Rmax, 0.25);
       if(R < 0.1)
         time_step *= 0.1;
       else
         time_step *= R;
     } else {
       end_point = space.adjust(prevY, (5170.0 / 57600.0) * k1 + (7571.0 / 16695.0) * k3 + (393.0 / 640.0) * k4 - (92097.0 / 339200.0) * k5 + (187.0 / 2100.0) * k6 + (1.0 / 40.0) * k7);
       dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
       
       double R = 0.84 * pow(tolerance / Rmax, 0.25);
       if(R >= 4.0)
         time_step *= 4.0;
       else if(R > 1.0)
         time_step *= R;
       //if(((time_step > 0.0) && (t + time_step > end_time)) || ((time_step < 0.0) && (t + time_step < end_time)))
         //time_step = end_time - t;
     };
     
     if(fabs(time_step) < min_step)
       time_step *= fabs(min_step / time_step);
     if(fabs(time_step) > max_step)
       time_step *= fabs(max_step / time_step);
   };
 };
Ejemplo n.º 4
0
 void hamming_mod_integrate_impl(
     const StateSpace& space,
     const StateSpaceSystem& sys,
     const typename pp::topology_traits<StateSpace>::point_type& start_point,
     typename pp::topology_traits<StateSpace>::point_type& end_point,
     const InputTrajectory& u_traj,
     double start_time,
     double end_time,
     double time_step) {
   if ((time_step == 0.0) ||
       ((time_step > 0.0) && (start_time > end_time)) ||
       ((time_step < 0.0) && (start_time < end_time)))
     throw impossible_integration(start_time, end_time, time_step);
   
   typedef typename pp::topology_traits<StateSpace>::point_type PointType;
   typedef typename pp::topology_traits<StateSpace>::point_difference_type PointDiffType;
   
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::const_waypoint_descriptor InputWaypoint;
   typedef typename pp::spatial_trajectory_traits<InputTrajectory>::point_type InputType;
   std::pair< InputWaypoint, InputType> u_wp = u_traj.get_waypoint_at_time(start_time);
   
   double back_time = start_time;
   PointDiffType dp = sys.get_state_derivative(space, start_point, u_wp.second.pt, start_time);
   PointType w = start_point;
   PointType Y_n[3];
   PointDiffType Yp_n[3];
   
   for(std::size_t i = 0; i < 3; ++i) {
     PointDiffType k1 = time_step * dp;
     Y_n[i] = space.adjust(w, -0.25 * k1);
     
     back_time -= time_step * 0.25;
     u_wp = u_traj.move_time_diff_from(u_wp, -0.25 * time_step);
     Yp_n[i] = sys.get_state_derivative(space, Y_n[i], u_wp.second.pt, back_time);
     PointDiffType k2 = time_step * Yp_n_1;
     Y_n[i] = space.adjust(Y_n[i], (5.0 / 32.0) * k1 - (9.0 / 32.0) * k2);
     
     back_time -= time_step * 0.125;
     u_wp = u_traj.move_time_diff_from(u_wp, -0.125 * time_step);
     Yp_n[i] = sys.get_state_derivative(space, Y_n[i], u_wp.second.pt, back_time);
     PointDiffType k3 = time_step * Yp_n[i];
     Y_n[i] = space.adjust(Y_n[i], (1250865.0 / 351520.0) * k2 - (276165.0 / 351520.0) * k1 - (1167360.0 / 351520.0) * k3);
     
     back_time -= 57.0 * time_step / 104.0;
     u_wp = u_traj.move_time_diff_from(u_wp, -57.0 / 104.0 * time_step);
     Yp_n[i] = sys.get_state_derivative(space, Y_n[i], u_wp.second.pt, back_time);
     PointDiffType k4 = time_step * Yp_n[i];
     Y_n[i] = space.adjust(w, 8.0 * k2 - (439.0 / 216.0) * k1 - (3680.0 / 513.0) * k3 + (845.0 / 4104.0) * k4);
     
     back_time -= time_step / 13.0;
     u_wp = u_traj.move_time_diff_from(u_wp, -time_step / 13.0);
     Yp_n[i] = sys.get_state_derivative(space, Y_n[i], u_wp.second.pt, back_time);
     PointDiffType k5 = time_step * Yp_n[i];
     Y_n[i] = space.adjust(w, (8.0 / 27.0) * k1 - 2.0 * k2 + (3544.0 / 2565.0) * k3 - (1859.0 / 4104.0) * k4 + (11.0 / 40.0) * k5);
     
     back_time += time_step * 0.5;
     u_wp = u_traj.move_time_diff_from(u_wp, 0.5 * time_step);
     Yp_n[i] = sys.get_state_derivative(space, Y_n[i], u_wp.second.pt, back_time);
     Y_n[i] = space.adjust(w, (-16.0 / 135.0) * k1 - (6656.0 / 12825.0) * k3 - (28561.0 / 56430.0) * k4 + (9.0 / 50.0) * k5 - (2.0 * time_step / 55.0) * Yp_n[i]);
     
     back_time -= time_step * 0.5;
     u_wp = u_traj.move_time_diff_from(u_wp, -0.5 * time_step);
     Yp_n[i] = sys.get_state_derivative(space, Y_n[i], u_wp.second.pt, back_time);
     
     w = Y_n[i];
     dp = Yp_n[i];
   };
   
   PointType C = start_point;
   PointType P_n = start_point; // this makes sure the initial error sweep is zero.
   
   double t = start_time;
   end_point = start_point;
   u_wp = u_traj.get_waypoint_at_time(t);
   dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
   
   while(((time_step > 0.0) && (t < end_time)) || 
         ((time_step < 0.0) && (t > end_time))) {
     
     PointType P = space.adjust(Y_n[2], (time_step * 4.0 / 3.0) * (2.0 * (dp + Yp_n[1]) - Yp_n[0]));
     PointType M = space.adjust(P, (112.0 / 121.0) * space.difference(C, P_n));
     
     t += time_step;
     u_wp = u_traj.move_time_diff_from(u_wp, time_step);
     PointDiffType Mp = sys.get_state_derivative(space, M, u_wp.second.pt, t);
     
     C = space.adjust(end_point, 0.125 * space.difference(end_point, Y_n[1]) + (time_step * 3.0 / 8.0) * (Mp + 2.0 * dp - Yp_n[0]));
     
     Y_n[2] = Y_n[1];
     Y_n[1] = Y_n[0];
     Y_n[0] = end_point;
     end_point = space.adjust(C, (9.0 / 121.0) * space.difference(P, C));
     P_n = P;
     
     Yp_n[1] = Yp_n[0];
     Yp_n[0] = dp;
     dp = sys.get_state_derivative(space, end_point, u_wp.second.pt, t);
     
   };
 };