add Davis's suggestion

Co-authored-by: Davis E. King <[email protected]>
davisking · Feb 29, 2024 · f51b1f2 · f51b1f2
1 parent fbd6e19
commit f51b1f2
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/dlib/image_loader/jxl_loader.cpp b/dlib/image_loader/jxl_loader.cpp
@@ -17,10 +17,9 @@ namespace dlib
     {
         std::ifstream stream(filename, std::ios::binary);
         stream.exceptions(std::ifstream::failbit | std::ifstream::badbit | std::ifstream::eofbit);
-        stream.seekg(0, std::ios_base::end);
-        std::vector<unsigned char> buffer(stream.tellg());
-        stream.seekg(0);
-        stream.read(reinterpret_cast<char*>(buffer.data()), buffer.size());
+        std::vector<unsigned char> buffer;
+        vectorstream temp(buffer);
+        temp << stream.rdbuf();
         return buffer;
     }