evaluator_test_utils.cc 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. // Ceres Solver - A fast non-linear least squares minimizer
  2. // Copyright 2015 Google Inc. All rights reserved.
  3. // http://ceres-solver.org/
  4. //
  5. // Redistribution and use in source and binary forms, with or without
  6. // modification, are permitted provided that the following conditions are met:
  7. //
  8. // * Redistributions of source code must retain the above copyright notice,
  9. // this list of conditions and the following disclaimer.
  10. // * Redistributions in binary form must reproduce the above copyright notice,
  11. // this list of conditions and the following disclaimer in the documentation
  12. // and/or other materials provided with the distribution.
  13. // * Neither the name of Google Inc. nor the names of its contributors may be
  14. // used to endorse or promote products derived from this software without
  15. // specific prior written permission.
  16. //
  17. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  18. // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19. // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20. // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  21. // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  22. // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  23. // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  24. // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  25. // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  26. // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  27. // POSSIBILITY OF SUCH DAMAGE.
  28. //
  29. // Author: keir@google.com (Keir Mierle)
  30. // sameeragarwal@google.com (Sameer Agarwal)
  31. #include "ceres/evaluator_test_utils.h"
  32. #include "ceres/internal/eigen.h"
  33. #include "gtest/gtest.h"
  34. namespace ceres::internal {
  35. void CompareEvaluations(int expected_num_rows,
  36. int expected_num_cols,
  37. double expected_cost,
  38. const double* expected_residuals,
  39. const double* expected_gradient,
  40. const double* expected_jacobian,
  41. const double actual_cost,
  42. const double* actual_residuals,
  43. const double* actual_gradient,
  44. const double* actual_jacobian) {
  45. EXPECT_EQ(expected_cost, actual_cost);
  46. if (expected_residuals != nullptr) {
  47. ConstVectorRef expected_residuals_vector(expected_residuals,
  48. expected_num_rows);
  49. ConstVectorRef actual_residuals_vector(actual_residuals, expected_num_rows);
  50. EXPECT_TRUE(
  51. (actual_residuals_vector.array() == expected_residuals_vector.array())
  52. .all())
  53. << "Actual:\n"
  54. << actual_residuals_vector << "\nExpected:\n"
  55. << expected_residuals_vector;
  56. }
  57. if (expected_gradient != nullptr) {
  58. ConstVectorRef expected_gradient_vector(expected_gradient,
  59. expected_num_cols);
  60. ConstVectorRef actual_gradient_vector(actual_gradient, expected_num_cols);
  61. EXPECT_TRUE(
  62. (actual_gradient_vector.array() == expected_gradient_vector.array())
  63. .all())
  64. << "Actual:\n"
  65. << actual_gradient_vector.transpose() << "\nExpected:\n"
  66. << expected_gradient_vector.transpose();
  67. }
  68. if (expected_jacobian != nullptr) {
  69. ConstMatrixRef expected_jacobian_matrix(
  70. expected_jacobian, expected_num_rows, expected_num_cols);
  71. ConstMatrixRef actual_jacobian_matrix(
  72. actual_jacobian, expected_num_rows, expected_num_cols);
  73. EXPECT_TRUE(
  74. (actual_jacobian_matrix.array() == expected_jacobian_matrix.array())
  75. .all())
  76. << "Actual:\n"
  77. << actual_jacobian_matrix << "\nExpected:\n"
  78. << expected_jacobian_matrix;
  79. }
  80. }
  81. } // namespace ceres::internal