1 #include "shared/disk_space_estimator.h"
6 #include <sys/statfs.h>
8 #include "shared/metrics.h"
9 #include "shared/timebase.h"
13 DiskSpaceEstimator::DiskSpaceEstimator(DiskSpaceEstimator::callback_t callback)
16 global_metrics.add("disk_free_bytes", &metric_disk_free_bytes, Metrics::TYPE_GAUGE);
19 void DiskSpaceEstimator::report_write(const string &filename, off_t bytes, uint64_t pts)
22 report_write_internal(filename, total_size, pts);
25 void DiskSpaceEstimator::report_append(const string &filename, uint64_t pts)
27 if (filename != last_filename) {
28 last_filename = filename;
29 measure_points.clear();
33 if (stat(filename.c_str(), &st) == -1) {
34 perror(filename.c_str());
38 report_write_internal(filename, st.st_size, pts);
41 void DiskSpaceEstimator::report_write_internal(const string &filename, off_t file_size, uint64_t pts)
43 if (measure_points.empty()) {
44 first_pts_this_file = pts;
47 // Reject points that are out-of-order (happens with B-frames).
48 if (!measure_points.empty() && pts <= measure_points.back().pts) {
52 // Remove too old points.
53 while (measure_points.size() > 1 && measure_points.front().pts + window_length < pts) {
54 measure_points.pop_front();
58 if (statfs(filename.c_str(), &fst) == -1) {
59 perror(filename.c_str());
63 off_t free_bytes = off_t(fst.f_bavail) * fst.f_frsize;
64 metric_disk_free_bytes = free_bytes;
66 if (!measure_points.empty()) {
67 double bytes_per_second = double(file_size - measure_points.front().size) /
68 (pts - measure_points.front().pts) * TIMEBASE;
69 double seconds_left = free_bytes / bytes_per_second;
71 // Only report every second, since updating the UI can be expensive.
72 if (last_pts_reported == 0 || pts - last_pts_reported >= TIMEBASE) {
73 callback(free_bytes, seconds_left, double(pts - first_pts_this_file) / TIMEBASE);
74 last_pts_reported = pts;
78 measure_points.push_back({ pts, file_size });
81 DiskSpaceEstimator *global_disk_space_estimator = nullptr; // Created in MainWindow::MainWindow().