(PointSetIO) Check successful write (#51)

educelab · Oct 31, 2023 · efc8bf2 · efc8bf2
1 parent 63aecd9
commit efc8bf2
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/core/include/vc/core/io/PointSetIO.hpp b/core/include/vc/core/io/PointSetIO.hpp
@@ -480,6 +480,13 @@ class PointSetIO
             }
             outfile << std::endl;
         }
+
+        outfile.flush();
+        outfile.close();
+        if (outfile.fail()) {
+            auto msg = "failure writing file '" + path.string() + "'";
+            throw IOException(msg);
+        }
     }
 
     /** @brief Write a binary PointSet */
@@ -499,6 +506,13 @@ class PointSetIO
             auto nbytes = T::channels * sizeof(typename T::value_type);
             outfile.write(reinterpret_cast<const char*>(p.val), nbytes);
         }
+
+        outfile.flush();
+        outfile.close();
+        if (outfile.fail()) {
+            auto msg = "failure writing file '" + path.string() + "'";
+            throw IOException(msg);
+        }
     }
 
     /** @brief Write an ASCII OrderedPointSet */
@@ -519,6 +533,13 @@ class PointSetIO
             }
             outfile << std::endl;
         }
+
+        outfile.flush();
+        outfile.close();
+        if (outfile.fail()) {
+            auto msg = "failure writing file '" + path.string() + "'";
+            throw IOException(msg);
+        }
     }
 
     /** @brief Write a binary OrderedPointSet */
@@ -538,6 +559,13 @@ class PointSetIO
             auto nbytes = T::channels * sizeof(typename T::value_type);
             outfile.write(reinterpret_cast<const char*>(p.val), nbytes);
         }
+
+        outfile.flush();
+        outfile.close();
+        if (outfile.fail()) {
+            auto msg = "failure writing file '" + path.string() + "'";
+            throw IOException(msg);
+        }
     }
     /**@}*/
 };