array_utils.cc 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. // Ceres Solver - A fast non-linear least squares minimizer
  2. // Copyright 2023 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: sameeragarwal@google.com (Sameer Agarwal)
  30. #include "ceres/array_utils.h"
  31. #include <algorithm>
  32. #include <cmath>
  33. #include <cstddef>
  34. #include <string>
  35. #include <vector>
  36. #include "ceres/stringprintf.h"
  37. #include "ceres/types.h"
  38. namespace ceres::internal {
  39. bool IsArrayValid(const int64_t size, const double* x) {
  40. if (x != nullptr) {
  41. for (int64_t i = 0; i < size; ++i) {
  42. if (!std::isfinite(x[i]) || (x[i] == kImpossibleValue)) {
  43. return false;
  44. }
  45. }
  46. }
  47. return true;
  48. }
  49. int64_t FindInvalidValue(const int64_t size, const double* x) {
  50. if (x == nullptr) {
  51. return size;
  52. }
  53. for (int64_t i = 0; i < size; ++i) {
  54. if (!std::isfinite(x[i]) || (x[i] == kImpossibleValue)) {
  55. return i;
  56. }
  57. }
  58. return size;
  59. }
  60. void InvalidateArray(const int64_t size, double* x) {
  61. if (x != nullptr) {
  62. for (int64_t i = 0; i < size; ++i) {
  63. x[i] = kImpossibleValue;
  64. }
  65. }
  66. }
  67. void AppendArrayToString(const int64_t size,
  68. const double* x,
  69. std::string* result) {
  70. for (int64_t i = 0; i < size; ++i) {
  71. if (x == nullptr) {
  72. StringAppendF(result, "Not Computed ");
  73. } else {
  74. if (x[i] == kImpossibleValue) {
  75. StringAppendF(result, "Uninitialized ");
  76. } else {
  77. StringAppendF(result, "%12g ", x[i]);
  78. }
  79. }
  80. }
  81. }
  82. void MapValuesToContiguousRange(const int64_t size, int* array) {
  83. std::vector<int> unique_values(array, array + size);
  84. std::sort(unique_values.begin(), unique_values.end());
  85. unique_values.erase(std::unique(unique_values.begin(), unique_values.end()),
  86. unique_values.end());
  87. for (int64_t i = 0; i < size; ++i) {
  88. array[i] =
  89. std::lower_bound(unique_values.begin(), unique_values.end(), array[i]) -
  90. unique_values.begin();
  91. }
  92. }
  93. } // namespace ceres::internal