Clock.h

/*
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 *
 * Clock.h
 * Provides the TimeInterval and TimeStamp classes.
 * Copyright (C) 2005 Simon Newton
 *
 * The struct timeval can represent both absolute time and time intervals.
 * We define our own wrapper classes that:
 * - hide some of the platform differences, like the fact windows doesn't
 * provide timeradd and timersub.
 * - Reduces bugs by using the compiler to check if the value was supposed
 * to be an interval or absolute time. For example, passing an absolute
 * time instead of an Interval to RegisterTimeout would be bad.
 */

#ifndef INCLUDE_OLA_CLOCK_H_
#define INCLUDE_OLA_CLOCK_H_

#include <ola/base/Macro.h>
#include <stdint.h>
#include <sys/time.h>

#include <iomanip>
#include <ostream>
#include <sstream>
#include <string>

namespace ola {

static const int USEC_IN_SECONDS = 1000000;
static const int ONE_THOUSAND = 1000;

class BaseTimeVal {
 public:
 // Constructors
 BaseTimeVal() { timerclear(&m_tv); }
 BaseTimeVal(int32_t sec, int32_t usec);

 explicit BaseTimeVal(const struct timeval &timestamp) { m_tv = timestamp; }
 explicit BaseTimeVal(const struct timespec &timestamp) { Set(timestamp); }
 explicit BaseTimeVal(int64_t interval_useconds) { Set(interval_useconds); }

 BaseTimeVal(const BaseTimeVal &other) : m_tv(other.m_tv) {}

 // Assignable
 BaseTimeVal& operator=(const BaseTimeVal& other);
 BaseTimeVal& operator=(const struct timeval &tv);
 BaseTimeVal& operator=(const struct timespec &ts);

 // Comparables
 bool operator==(const BaseTimeVal &other) const;
 bool operator!=(const BaseTimeVal &other) const;
 bool operator>(const BaseTimeVal &other) const;
 bool operator>=(const BaseTimeVal &other) const;
 bool operator<(const BaseTimeVal &other) const;
 bool operator<=(const BaseTimeVal &other) const;

 // Arithmetic
 BaseTimeVal& operator+=(const BaseTimeVal& other);
 BaseTimeVal &operator-=(const BaseTimeVal &other);
 const BaseTimeVal operator+(const BaseTimeVal &interval) const;
 const BaseTimeVal operator-(const BaseTimeVal &other) const;
 BaseTimeVal operator*(unsigned int i) const;

 // Various other methods.
 bool IsSet() const;
 void AsTimeval(struct timeval *tv) const;

 time_t Seconds() const { return m_tv.tv_sec; }
 int32_t MicroSeconds() const { return static_cast<int32_t>(m_tv.tv_usec); }

 int64_t InMilliSeconds() const;

 int64_t AsInt() const;

 std::string ToString() const;

 private:
 struct timeval m_tv;

 void TimerAdd(const struct timeval &tv1, const struct timeval &tv2,
 struct timeval *result) const;

 void TimerSub(const struct timeval &tv1, const struct timeval &tv2,
 struct timeval *result) const;

 void Set(int64_t interval_useconds);

 void Set(const struct timespec &ts);
};

class TimeInterval {
 public:
 // Constructors
 TimeInterval() {}
 TimeInterval(int32_t sec, int32_t usec) : m_interval(sec, usec) {}
 explicit TimeInterval(int64_t usec) : m_interval(usec) {}

 TimeInterval(const TimeInterval &other) : m_interval(other.m_interval) {}

 // Assignable
 TimeInterval& operator=(const TimeInterval& other);

 // Comparables
 bool operator==(const TimeInterval &other) const;
 bool operator!=(const TimeInterval &other) const;
 bool operator>(const TimeInterval &other) const;
 bool operator>=(const TimeInterval &other) const;
 bool operator<(const TimeInterval &other) const;
 bool operator<=(const TimeInterval &other) const;

 // Arithmetic
 TimeInterval& operator+=(const TimeInterval& other);
 TimeInterval operator*(unsigned int i) const;

 // Various other methods.
 bool IsZero() const { return !m_interval.IsSet(); }

 void AsTimeval(struct timeval *tv) const { m_interval.AsTimeval(tv); }

 time_t Seconds() const { return m_interval.Seconds(); }
 int32_t MicroSeconds() const { return m_interval.MicroSeconds(); }

 int64_t InMilliSeconds() const { return m_interval.InMilliSeconds(); }
 int64_t AsInt() const { return m_interval.AsInt(); }

 std::string ToString() const { return m_interval.ToString(); }

 friend std::ostream& operator<< (std::ostream &out,
 const TimeInterval &interval) {
 return out << interval.m_interval.ToString();
 }

 private:
 explicit TimeInterval(const BaseTimeVal &time_val) : m_interval(time_val) {}

 BaseTimeVal m_interval;
 friend class TimeStamp;
};


class TimeStamp {
 public:
 // Constructors
 TimeStamp() {}
 explicit TimeStamp(const struct timeval &timestamp) : m_tv(timestamp) {}
 explicit TimeStamp(const struct timespec &timestamp) : m_tv(timestamp) {}

 TimeStamp(const TimeStamp &other) : m_tv(other.m_tv) {}

 // Assignable
 TimeStamp& operator=(const TimeStamp& other);
 TimeStamp& operator=(const struct timeval &tv);
 TimeStamp& operator=(const struct timespec &ts);

 // Comparables
 bool operator==(const TimeStamp &other) const { return m_tv == other.m_tv; }
 bool operator!=(const TimeStamp &other) const { return m_tv != other.m_tv; }
 bool operator>(const TimeStamp &other) const { return m_tv > other.m_tv; }
 bool operator>=(const TimeStamp &other) const { return m_tv >= other.m_tv; }
 bool operator<(const TimeStamp &other) const { return m_tv < other.m_tv; }
 bool operator<=(const TimeStamp &other) const { return m_tv <= other.m_tv; }

 // Arithmetic
 TimeStamp &operator+=(const TimeInterval &interval);
 TimeStamp &operator-=(const TimeInterval &interval);
 const TimeStamp operator+(const TimeInterval &interval) const;
 const TimeInterval operator-(const TimeStamp &other) const;
 const TimeStamp operator-(const TimeInterval &interval) const;

 // Various other methods.
 bool IsSet() const { return m_tv.IsSet(); }

 time_t Seconds() const { return m_tv.Seconds(); }
 int32_t MicroSeconds() const { return m_tv.MicroSeconds(); }

 std::string ToString() const { return m_tv.ToString(); }

 friend std::ostream& operator<<(std::ostream &out,
 const TimeStamp &timestamp) {
 return out << timestamp.m_tv.ToString();
 }

 private:
 BaseTimeVal m_tv;

 explicit TimeStamp(const BaseTimeVal &time_val) : m_tv(time_val) {}
};

class Clock {
 public:
 Clock() {}
 virtual ~Clock() {}
 virtual void CurrentMonotonicTime(TimeStamp* timestamp) const;

 virtual void CurrentRealTime(TimeStamp* timestamp) const;

 virtual void CurrentTime(TimeStamp* timestamp) const;

 private:
 DISALLOW_COPY_AND_ASSIGN(Clock);
};

class MockClock: public Clock {
 public:
 MockClock() : Clock() {}

 // Advance the time
 void AdvanceTime(const TimeInterval &interval);
 void AdvanceTime(int32_t sec, int32_t usec);

 void CurrentMonotonicTime(TimeStamp *timestamp) const;
 void CurrentRealTime(TimeStamp *timestamp) const;
 void CurrentTime(TimeStamp *timestamp) const;

 private:
 TimeInterval m_offset;
};
} // namespace ola
#endif // INCLUDE_OLA_CLOCK_H_