aos/network/testing_time_converter.cc

#include "aos/network/testing_time_converter.h"

#include <chrono>
#include <deque>
#include <optional>
#include <tuple>

#include "aos/events/event_scheduler.h"
#include "aos/network/multinode_timestamp_filter.h"
#include "aos/time/time.h"

namespace aos::message_bridge {

namespace chrono = std::chrono;

TestingTimeConverter ::TestingTimeConverter(size_t node_count)
    : InterpolatedTimeConverter(node_count),
      last_monotonic_(node_count, logger::BootTimestamp::epoch()) {
  CHECK_GE(node_count, 1u);
}

TestingTimeConverter::~TestingTimeConverter() {
  if (at_end_) {
    auto next_timestamp = NextTimestamp();
    CHECK(next_timestamp.has_value()) << ": Unexpected error";
    CHECK(!next_timestamp.value().has_value())
        << ": At the end but there is more data.";
  }
}

void TestingTimeConverter::StartEqual() {
  CHECK(first_);
  first_ = false;
  ts_.emplace_back(std::make_tuple(last_distributed_, last_monotonic_));
}

chrono::nanoseconds TestingTimeConverter::AddMonotonic(
    std::vector<monotonic_clock::duration> times) {
  CHECK_EQ(times.size(), last_monotonic_.size());
  for (size_t i = 0; i < times.size(); ++i) {
    CHECK_GT(times[i].count(), 0);
    last_monotonic_[i].time += times[i];
  }
  chrono::nanoseconds dt(0);
  if (!first_) {
    dt = *std::max_element(times.begin(), times.end());
    last_distributed_ += dt;
  } else {
    first_ = false;
  }
  ts_.emplace_back(std::make_tuple(last_distributed_, last_monotonic_));
  return dt;
}

chrono::nanoseconds TestingTimeConverter::AddMonotonic(
    std::vector<logger::BootTimestamp> times) {
  CHECK_EQ(times.size(), last_monotonic_.size());
  chrono::nanoseconds dt(0);
  if (!first_) {
    CHECK_EQ(times[0].boot, last_monotonic_[0].boot);
    dt = times[0].time - last_monotonic_[0].time;
    for (size_t i = 0; i < times.size(); ++i) {
      CHECK_GT(times[i], last_monotonic_[i]);
      dt = std::max(dt, times[i].time - times[0].time);
    }
    last_distributed_ += dt;
    last_monotonic_ = times;
  } else {
    first_ = false;
    last_monotonic_ = times;
  }
  ts_.emplace_back(std::make_tuple(last_distributed_, std::move(times)));
  return dt;
}

void TestingTimeConverter::RebootAt(size_t node_index,
                                    distributed_clock::time_point t) {
  CHECK(!first_);
  const chrono::nanoseconds dt = t - last_distributed_;

  for (size_t i = 0; i < last_monotonic_.size(); ++i) {
    last_monotonic_[i].time += dt;
  }

  ++last_monotonic_[node_index].boot;
  last_monotonic_[node_index].time = monotonic_clock::epoch();

  last_distributed_ = t;
  ts_.emplace_back(std::make_tuple(last_distributed_, last_monotonic_));
}

void TestingTimeConverter::AddNextTimestamp(
    distributed_clock::time_point time,
    std::vector<logger::BootTimestamp> times) {
  CHECK_EQ(times.size(), last_monotonic_.size());
  if (!first_) {
    CHECK_GT(time, last_distributed_);
    for (size_t i = 0; i < times.size(); ++i) {
      CHECK_GT(times[i], last_monotonic_[i]);
    }
  } else {
    first_ = false;
  }
  last_distributed_ = time;
  last_monotonic_ = times;

  ts_.emplace_back(std::make_tuple(time, std::move(times)));
}

std::optional<std::optional<std::tuple<distributed_clock::time_point,
                                       std::vector<logger::BootTimestamp>>>>
TestingTimeConverter::NextTimestamp() {
  CHECK(!first_) << ": Tried to pull a timestamp before one was added.  This "
                    "is unlikely to be what you want.";
  if (ts_.empty()) {
    std::optional<std::optional<std::tuple<distributed_clock::time_point,
                                           std::vector<logger::BootTimestamp>>>>
        result;
    result.emplace(std::nullopt);
    return result;
  }
  auto result = ts_.front();
  ts_.pop_front();
  return result;
}

}  // namespace aos::message_bridge