00001 /********************************************************************* 00002 * Software License Agreement (BSD License) 00003 * 00004 * Copyright (c) 2009, Willow Garage, Inc. 00005 * All rights reserved. 00006 * 00007 * Redistribution and use in source and binary forms, with or without 00008 * modification, are permitted provided that the following conditions 00009 * are met: 00010 * 00011 * * Redistributions of source code must retain the above copyright 00012 * notice, this list of conditions and the following disclaimer. 00013 * * Redistributions in binary form must reproduce the above 00014 * copyright notice, this list of conditions and the following 00015 * disclaimer in the documentation and/or other materials provided 00016 * with the distribution. 00017 * * Neither the name of the Willow Garage nor the names of its 00018 * contributors may be used to endorse or promote products derived 00019 * from this software without specific prior written permission. 00020 * 00021 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00022 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 00023 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 00024 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 00025 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 00026 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 00027 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 00028 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 00029 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 00030 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 00031 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 00032 * POSSIBILITY OF SUCH DAMAGE. 00033 *********************************************************************/ 00034 00037 #include <gtest/gtest.h> 00038 #include <spline_smoother/splines.h> 00039 #include <spline_smoother/clamped_cubic_spline_smoother.h> 00040 #include <spline_smoother/numerical_differentiation_spline_smoother.h> 00041 #include <arm_navigation_msgs/FilterJointTrajectory.h> 00042 #include <stdlib.h> 00043 00044 using namespace spline_smoother; 00045 00046 TEST(TestClampedCubicSplineSmoother, TestZeroPositionsSmall) 00047 { 00048 int length = ClampedCubicSplineSmoother<arm_navigation_msgs::FilterJointTrajectory>::MAX_TRIDIAGONAL_SOLVER_ELEMENTS - 2; 00049 00050 arm_navigation_msgs::FilterJointTrajectory wpt; 00051 arm_navigation_msgs::FilterJointTrajectory wpt_out; 00052 wpt.request.trajectory.points.resize(length); 00053 wpt.request.trajectory.joint_names.resize(1); 00054 wpt.request.trajectory.joint_names[0] = std::string("test"); 00055 for (int i=0; i<length; i++) 00056 { 00057 wpt.request.trajectory.points[i].positions.resize(1); 00058 wpt.request.trajectory.points[i].accelerations.resize(1); 00059 wpt.request.trajectory.points[i].velocities.resize(1); 00060 wpt.request.trajectory.points[i].positions[0] = 0.0; 00061 wpt.request.trajectory.points[i].velocities[0] = 0.0; 00062 wpt.request.trajectory.points[i].accelerations[0] = 0.0; 00063 wpt.request.trajectory.points[i].time_from_start = ros::Duration(i); 00064 } 00065 00066 ClampedCubicSplineSmoother<arm_navigation_msgs::FilterJointTrajectory> ccss; 00067 ccss.smooth(wpt, wpt_out); 00068 00069 // verify that velocities are 0: 00070 for (int i=0; i<length; i++) 00071 { 00072 EXPECT_NEAR(wpt_out.request.trajectory.points[i].velocities[0], 0.0, 1e-8); 00073 } 00074 } 00075 00076 TEST(TestClampedCubicSplineSmoother, TestStraightLineSmall) 00077 { 00078 int length = ClampedCubicSplineSmoother<arm_navigation_msgs::FilterJointTrajectory>::MAX_TRIDIAGONAL_SOLVER_ELEMENTS-2; 00079 00080 arm_navigation_msgs::FilterJointTrajectory wpt; 00081 arm_navigation_msgs::FilterJointTrajectory wpt_out; 00082 wpt.request.trajectory.points.resize(length); 00083 wpt.request.trajectory.joint_names.resize(1); 00084 wpt.request.trajectory.joint_names[0] = std::string("test"); 00085 for (int i=0; i<length; i++) 00086 { 00087 wpt.request.trajectory.points[i].positions.resize(1); 00088 wpt.request.trajectory.points[i].accelerations.resize(1); 00089 wpt.request.trajectory.points[i].velocities.resize(1); 00090 wpt.request.trajectory.points[i].positions[0] = i; 00091 wpt.request.trajectory.points[i].velocities[0] = 1.0; 00092 wpt.request.trajectory.points[i].accelerations[0] = 0.0; 00093 wpt.request.trajectory.points[i].time_from_start = ros::Duration(i); 00094 } 00095 00096 ClampedCubicSplineSmoother<arm_navigation_msgs::FilterJointTrajectory> ccss; 00097 ccss.smooth(wpt, wpt_out); 00098 00099 // verify that velocities are still 1: 00100 for (int i=0; i<length; i++) 00101 { 00102 EXPECT_NEAR(wpt.request.trajectory.points[i].velocities[0], 1.0, 1e-8); 00103 } 00104 }