$search
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 <stdlib.h> 00040 00041 using namespace spline_smoother; 00042 00043 static double getRandomNumber(double min, double max) 00044 { 00045 return ((double)rand() / RAND_MAX)*(max-min) + min; 00046 } 00047 00048 TEST(TestSplines, testQuinticCoefficients) 00049 { 00050 // seed the random number generator: 00051 srand(0); 00052 00053 // generate random boundary conditions: 00054 double bc[6]; 00055 for (int i=0; i<6; i++) 00056 bc[i] = getRandomNumber(-1.0, 1.0); 00057 00058 // and a random time: 00059 double time = getRandomNumber(0.5,10.0); 00060 00061 // get the spline coefficients: 00062 std::vector<double> coeffs; 00063 getQuinticSplineCoefficients(bc[0], bc[1], bc[2], bc[3], bc[4], bc[5], time, coeffs); 00064 00065 // now sample the spline at t=0 and t=time to cross-check 00066 double test_bc[6]; 00067 sampleQuinticSpline(coeffs, 0, test_bc[0], test_bc[1], test_bc[2]); 00068 sampleQuinticSpline(coeffs, time, test_bc[3], test_bc[4], test_bc[5]); 00069 00070 double tolerance=1e-10; 00071 00072 EXPECT_NEAR(bc[0], test_bc[0], tolerance); 00073 EXPECT_NEAR(bc[1], test_bc[1], tolerance); 00074 EXPECT_NEAR(bc[2], test_bc[2], tolerance); 00075 EXPECT_NEAR(bc[3], test_bc[3], tolerance); 00076 EXPECT_NEAR(bc[4], test_bc[4], tolerance); 00077 EXPECT_NEAR(bc[5], test_bc[5], tolerance); 00078 } 00079 00080 TEST(TestSplines, testCubicCoefficients) 00081 { 00082 // seed the random number generator: 00083 srand(1); 00084 00085 // generate random boundary conditions: 00086 double bc[4]; 00087 for (int i=0; i<4; i++) 00088 bc[i] = getRandomNumber(-1.0, 1.0); 00089 00090 // and a random time: 00091 double time = getRandomNumber(0.5,10.0); 00092 00093 // get the spline coefficients: 00094 std::vector<double> coeffs; 00095 getCubicSplineCoefficients(bc[0], bc[1], bc[2], bc[3], time, coeffs); 00096 00097 // now sample the spline at t=0 and t=time to cross-check 00098 double test_bc[4]; 00099 double dummy; 00100 sampleCubicSpline(coeffs, 0, test_bc[0], test_bc[1], dummy); 00101 sampleCubicSpline(coeffs, time, test_bc[2], test_bc[3], dummy); 00102 00103 double tolerance=1e-10; 00104 00105 EXPECT_NEAR(bc[0], test_bc[0], tolerance); 00106 EXPECT_NEAR(bc[1], test_bc[1], tolerance); 00107 EXPECT_NEAR(bc[2], test_bc[2], tolerance); 00108 EXPECT_NEAR(bc[3], test_bc[3], tolerance); 00109 } 00110 00111 int main(int argc, char** argv) 00112 { 00113 testing::InitGoogleTest(&argc, argv); 00114 return RUN_ALL_TESTS(); 00115 }