crs_matrix.h 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  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. #ifndef CERES_PUBLIC_CRS_MATRIX_H_
  31. #define CERES_PUBLIC_CRS_MATRIX_H_
  32. #include <vector>
  33. #include "ceres/internal/disable_warnings.h"
  34. #include "ceres/internal/export.h"
  35. namespace ceres {
  36. // A compressed row sparse matrix used primarily for communicating the
  37. // Jacobian matrix to the user.
  38. struct CERES_EXPORT CRSMatrix {
  39. CRSMatrix() = default;
  40. int num_rows{0};
  41. int num_cols{0};
  42. // A compressed row matrix stores its contents in three arrays,
  43. // rows, cols and values.
  44. //
  45. // rows is a num_rows + 1 sized array that points into the cols and
  46. // values array. For each row i:
  47. //
  48. // cols[rows[i]] ... cols[rows[i + 1] - 1] are the indices of the
  49. // non-zero columns of row i.
  50. //
  51. // values[rows[i]] .. values[rows[i + 1] - 1] are the values of the
  52. // corresponding entries.
  53. //
  54. // cols and values contain as many entries as there are non-zeros in
  55. // the matrix.
  56. //
  57. // e.g, consider the 3x4 sparse matrix
  58. //
  59. // [ 0 10 0 4 ]
  60. // [ 0 2 -3 2 ]
  61. // [ 1 2 0 0 ]
  62. //
  63. // The three arrays will be:
  64. //
  65. //
  66. // -row0- ---row1--- -row2-
  67. // rows = [ 0, 2, 5, 7]
  68. // cols = [ 1, 3, 1, 2, 3, 0, 1]
  69. // values = [10, 4, 2, -3, 2, 1, 2]
  70. std::vector<int> cols;
  71. std::vector<int> rows;
  72. std::vector<double> values;
  73. };
  74. } // namespace ceres
  75. #include "ceres/internal/reenable_warnings.h"
  76. #endif // CERES_PUBLIC_CRS_MATRIX_H_