DataSpeed.h

// -*- indent-tabs-mode: nil -*-

#ifndef __ARC_DATASPEED_H__
#define __ARC_DATASPEED_H__

#include <string>

#include <arc/Logger.h>

#define DATASPEED_AVERAGING_PERIOD 60

namespace Arc {


  class DataSpeed {
  private:
    time_t first_time;
    time_t last_time;
    time_t last_activity_time;
    unsigned long long int N;
    unsigned long long int Nall;
    unsigned long long int Nmax;
    time_t first_speed_failure;
    time_t last_printed;

    time_t T;
    time_t min_speed_time;
    time_t max_inactivity_time;
    unsigned long long int min_speed;
    unsigned long long int min_average_speed;
    bool be_verbose;
    std::string verbose_prefix;

    bool min_speed_failed;
    bool min_average_speed_failed;
    bool max_inactivity_time_failed;

    bool disabled;
    static Logger logger;

  public:

    typedef void (*show_progress_t)(FILE *o, const char *s, unsigned int t,
                                    unsigned long long int all,
                                    unsigned long long int max,
                                    double instant, double average);
  private:
    show_progress_t show_progress;
    void print_statistics(FILE *o, time_t t);
  public:

    DataSpeed(time_t base = DATASPEED_AVERAGING_PERIOD);

    DataSpeed(unsigned long long int min_speed, time_t min_speed_time,
              unsigned long long int min_average_speed,
              time_t max_inactivity_time,
              time_t base = DATASPEED_AVERAGING_PERIOD);
    ~DataSpeed();
    void verbose(bool val);
    void verbose(const std::string& prefix);
    bool verbose();

    void set_min_speed(unsigned long long int min_speed,
                       time_t min_speed_time);

    void set_min_average_speed(unsigned long long int min_average_speed);

    void set_max_inactivity_time(time_t max_inactivity_time);
    time_t get_max_inactivity_time() { return max_inactivity_time; };
    void set_base(time_t base_ = DATASPEED_AVERAGING_PERIOD);
    void set_max_data(unsigned long long int max = 0);

    void set_progress_indicator(show_progress_t func = NULL);
    void reset();

    bool transfer(unsigned long long int n = 0);
    void hold(bool disable);
    bool min_speed_failure() {
      return min_speed_failed;
    }
    bool min_average_speed_failure() {
      return min_average_speed_failed;
    }
    bool max_inactivity_time_failure() {
      return max_inactivity_time_failed;
    }
    unsigned long long int transferred_size() {
      return Nall;
    }
  };

} // namespace Arc

#endif // __ARC_DATASPEED_H__