Make the UI show free disk space, and a rough estimation of for how much longer we...

[nageru] / disk_space_estimator.cpp

diff --git a/disk_space_estimator.cpp b/disk_space_estimator.cpp
new file mode 100644 (file)
index 0000000..1cb7d6f
--- /dev/null
+++ b/disk_space_estimator.cpp
@@ -0,0 +1,54 @@
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/vfs.h>
+#include <unistd.h>
+
+#include "disk_space_estimator.h"
+#include "timebase.h"
+
+DiskSpaceEstimator::DiskSpaceEstimator(DiskSpaceEstimator::callback_t callback)
+       : callback(callback)
+{
+}
+
+void DiskSpaceEstimator::report_write(const std::string &filename, uint64_t pts)
+{
+       if (filename != last_filename) {
+               last_filename = filename;
+               measure_points.clear();
+       }
+
+       // Reject points that are out-of-order (happens with B-frames).
+       if (!measure_points.empty() && pts < measure_points.back().pts) {
+               return;
+       }
+
+       // Remove too old points.
+       while (measure_points.size() > 1 && measure_points.front().pts + window_length < pts) {
+               measure_points.pop_front();
+       }
+
+       struct stat st;
+       if (stat(filename.c_str(), &st) == -1) {
+               perror(filename.c_str());
+               return;
+       }
+
+       struct statfs fst;
+       if (statfs(filename.c_str(), &fst) == -1) {
+               perror(filename.c_str());
+               return;
+       }
+
+       if (!measure_points.empty()) {
+               off_t free_bytes = off_t(fst.f_bavail) * fst.f_frsize;
+               double bytes_per_second = double(st.st_size - measure_points.front().size) /
+                       (pts - measure_points.front().pts) * TIMEBASE;
+               double seconds_left = free_bytes / bytes_per_second;
+               callback(free_bytes, seconds_left);
+       }
+
+       measure_points.push_back({ pts, st.st_size });
+}
+
+DiskSpaceEstimator *global_disk_space_estimator = nullptr;  // Created in MainWindow::MainWindow().