file.cc 3.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  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: keir@google.com (Keir Mierle)
  30. //
  31. // Really simple file IO.
  32. #include "ceres/file.h"
  33. #include <cstdio>
  34. #include <string>
  35. #include "glog/logging.h"
  36. namespace ceres::internal {
  37. void WriteStringToFileOrDie(const std::string& data,
  38. const std::string& filename) {
  39. FILE* file_descriptor = fopen(filename.c_str(), "wb");
  40. if (!file_descriptor) {
  41. LOG(FATAL) << "Couldn't write to file: " << filename;
  42. }
  43. fwrite(data.c_str(), 1, data.size(), file_descriptor);
  44. fclose(file_descriptor);
  45. }
  46. void ReadFileToStringOrDie(const std::string& filename, std::string* data) {
  47. FILE* file_descriptor = fopen(filename.c_str(), "r");
  48. if (!file_descriptor) {
  49. LOG(FATAL) << "Couldn't read file: " << filename;
  50. }
  51. // Resize the input buffer appropriately.
  52. fseek(file_descriptor, 0L, SEEK_END);
  53. int64_t num_bytes = ftell(file_descriptor);
  54. data->resize(num_bytes);
  55. // Read the data.
  56. fseek(file_descriptor, 0L, SEEK_SET);
  57. int64_t num_read =
  58. fread(&((*data)[0]), sizeof((*data)[0]), num_bytes, file_descriptor);
  59. if (num_read != num_bytes) {
  60. LOG(FATAL) << "Couldn't read all of " << filename
  61. << "expected bytes: " << num_bytes * sizeof((*data)[0])
  62. << "actual bytes: " << num_read;
  63. }
  64. fclose(file_descriptor);
  65. }
  66. std::string JoinPath(const std::string& dirname, const std::string& basename) {
  67. #ifdef _WIN32
  68. static const char separator = '\\';
  69. #else
  70. static const char separator = '/';
  71. #endif // _WIN32
  72. if ((!basename.empty() && basename[0] == separator) || dirname.empty()) {
  73. return basename;
  74. } else if (dirname[dirname.size() - 1] == separator) {
  75. return dirname + basename;
  76. } else {
  77. return dirname + std::string(&separator, 1) + basename;
  78. }
  79. }
  80. } // namespace ceres::internal