git.sesse.net Git - nageru/blob - disk_space_estimator.h

   1 #ifndef _DISK_SPACE_ESTIMATOR_H
   2 #define _DISK_SPACE_ESTIMATOR_H
   3
   4 // A class responsible for measuring how much disk there is left when we
   5 // store our video to disk, and how much recording time that equates to.
   6 // It gets callbacks from the Mux writing the stream to disk (which also
   7 // knows which filesystem the file is going to), makes its calculations,
   8 // and calls back to the MainWindow, which shows it to the user.
   9 //
  10 // The bitrate is measured over a simple 30-second sliding window.
  11
  12 #include "timebase.h"
  13
  14 #include <deque>
  15 #include <functional>
  16 #include <stdint.h>
  17 #include <string>
  18 #include <sys/types.h>
  19
  20 class DiskSpaceEstimator {
  21 public:
  22         typedef std::function<void(off_t free_bytes, double estimated_seconds_left)> callback_t;
  23         DiskSpaceEstimator(callback_t callback);
  24
  25         // Report that a video frame with the given pts has just been written
  26         // to the given file, so the estimator should stat the file and see
  27         // by big it is. (The file is assumed to hold only that single frame,
  28         // unlike in Nageru, where it is a growing file.)
  29         //
  30         // If the filename changed since last time, the estimation is reset.
  31         // <pts> is taken to be in TIMEBASE units (see timebase.h).
  32         void report_write(const std::string &filename, uint64_t pts);
  33
  34 private:
  35         static constexpr int64_t window_length = 30 * TIMEBASE;
  36
  37         callback_t callback;
  38
  39         struct MeasurePoint {
  40                 uint64_t pts;
  41                 off_t size;
  42         };
  43         std::deque<MeasurePoint> measure_points;
  44         uint64_t last_pts_reported = 0;
  45         off_t total_size = 0;
  46 };
  47
  48 extern DiskSpaceEstimator *global_disk_space_estimator;
  49
  50 #endif  // !defined(_DISK_SPACE_ESTIMATOR_H)