10ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// Ceres Solver - A fast non-linear least squares minimizer 20ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// Copyright 2010, 2011, 2012 Google Inc. All rights reserved. 30ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// http://code.google.com/p/ceres-solver/ 40ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// 50ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// Redistribution and use in source and binary forms, with or without 60ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// modification, are permitted provided that the following conditions are met: 70ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// 80ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// * Redistributions of source code must retain the above copyright notice, 90ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// this list of conditions and the following disclaimer. 100ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// * Redistributions in binary form must reproduce the above copyright notice, 110ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// this list of conditions and the following disclaimer in the documentation 120ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// and/or other materials provided with the distribution. 130ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// * Neither the name of Google Inc. nor the names of its contributors may be 140ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// used to endorse or promote products derived from this software without 150ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// specific prior written permission. 160ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// 170ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 180ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 190ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 200ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 210ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 220ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 230ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 240ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 250ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 260ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 270ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// POSSIBILITY OF SUCH DAMAGE. 280ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// 290ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong// Author: sameeragarwal@google.com (Sameer Agarwal) 300ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 310ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong#include "ceres/autodiff_cost_function.h" 320ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 330ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong#include <cstddef> 340ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 350ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong#include "gtest/gtest.h" 360ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong#include "ceres/cost_function.h" 370ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 380ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kongnamespace ceres { 390ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kongnamespace internal { 400ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 410ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kongclass BinaryScalarCost { 420ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong public: 430ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong explicit BinaryScalarCost(double a): a_(a) {} 440ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong template <typename T> 450ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong bool operator()(const T* const x, const T* const y, 460ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong T* cost) const { 470ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong cost[0] = x[0] * y[0] + x[1] * y[1] - T(a_); 480ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong return true; 490ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong } 500ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong private: 510ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong double a_; 520ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong}; 530ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 540ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus KongTEST(AutodiffCostFunction, BilinearDifferentiationTest) { 550ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong CostFunction* cost_function = 560ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong new AutoDiffCostFunction<BinaryScalarCost, 1, 2, 2>( 570ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong new BinaryScalarCost(1.0)); 580ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 590ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong double** parameters = new double*[2]; 600ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[0] = new double[2]; 610ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[1] = new double[2]; 620ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 630ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[0][0] = 1; 640ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[0][1] = 2; 650ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 660ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[1][0] = 3; 670ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[1][1] = 4; 680ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 690ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong double** jacobians = new double*[2]; 700ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong jacobians[0] = new double[2]; 710ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong jacobians[1] = new double[2]; 720ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 730ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong double residuals = 0.0; 740ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 750ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong cost_function->Evaluate(parameters, &residuals, NULL); 760ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(10.0, residuals); 770ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong cost_function->Evaluate(parameters, &residuals, jacobians); 780ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 790ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(3, jacobians[0][0]); 800ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(4, jacobians[0][1]); 810ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(1, jacobians[1][0]); 820ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(2, jacobians[1][1]); 830ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 840ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] jacobians[0]; 850ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] jacobians[1]; 860ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] parameters[0]; 870ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] parameters[1]; 880ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] jacobians; 890ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] parameters; 900ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete cost_function; 910ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong} 920ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 930ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kongstruct TenParameterCost { 940ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong template <typename T> 950ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong bool operator()(const T* const x0, 960ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x1, 970ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x2, 980ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x3, 990ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x4, 1000ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x5, 1010ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x6, 1020ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x7, 1030ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x8, 1040ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong const T* const x9, 1050ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong T* cost) const { 1060ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong cost[0] = *x0 + *x1 + *x2 + *x3 + *x4 + *x5 + *x6 + *x7 + *x8 + *x9; 1070ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong return true; 1080ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong } 1090ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong}; 1100ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 1110ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus KongTEST(AutodiffCostFunction, ManyParameterAutodiffInstantiates) { 1120ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong CostFunction* cost_function = 1130ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong new AutoDiffCostFunction< 1140ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong TenParameterCost, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1>( 1150ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong new TenParameterCost); 1160ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 1170ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong double** parameters = new double*[10]; 1180ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong double** jacobians = new double*[10]; 1190ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong for (int i = 0; i < 10; ++i) { 1200ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[i] = new double[1]; 1210ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong parameters[i][0] = i; 1220ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong jacobians[i] = new double[1]; 1230ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong } 1240ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 1250ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong double residuals = 0.0; 1260ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 1270ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong cost_function->Evaluate(parameters, &residuals, NULL); 1280ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(45.0, residuals); 1290ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 1300ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong cost_function->Evaluate(parameters, &residuals, jacobians); 1310ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(residuals, 45.0); 1320ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong for (int i = 0; i < 10; ++i) { 1330ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong EXPECT_EQ(1.0, jacobians[i][0]); 1340ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong } 1350ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 1360ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong for (int i = 0; i < 10; ++i) { 1370ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] jacobians[i]; 1380ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] parameters[i]; 1390ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong } 1400ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] jacobians; 1410ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete[] parameters; 1420ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong delete cost_function; 1430ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong} 1440ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong 1450ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong} // namespace internal 1460ae28bd5885b5daa526898fcf7c323dc2c3e1963Angus Kong} // namespace ceres 147