/*
 *  Copyright (c) 2019 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include "modules/rtp_rtcp/source/absolute_capture_time_interpolator.h"

#include <cstdint>
#include <cstdlib>
#include <optional>
#include <span>

#include "api/rtp_headers.h"
#include "api/units/time_delta.h"
#include "api/units/timestamp.h"
#include "rtc_base/checks.h"
#include "rtc_base/synchronization/mutex.h"
#include "system_wrappers/include/clock.h"
#include "system_wrappers/include/metrics.h"
#include "system_wrappers/include/ntp_time.h"

namespace webrtc {

AbsoluteCaptureTimeInterpolator::AbsoluteCaptureTimeInterpolator(Clock* clock)
    : clock_(clock) {}

uint32_t AbsoluteCaptureTimeInterpolator::GetSource(
    uint32_t ssrc,
    std::span<const uint32_t> csrcs) {
  if (csrcs.empty()) {
    return ssrc;
  }

  return csrcs[0];
}

std::optional<AbsoluteCaptureTime>
AbsoluteCaptureTimeInterpolator::OnReceivePacket(
    uint32_t source,
    uint32_t rtp_timestamp,
    int rtp_clock_frequency_hz,
    const std::optional<AbsoluteCaptureTime>& received_extension) {
  const Timestamp receive_time = clock_->CurrentTime();
  if (!first_packet_time_) {
    first_packet_time_ = receive_time;
  }

  MutexLock lock(&mutex_);

  if (received_extension == std::nullopt) {
    if (!ShouldInterpolateExtension(receive_time, source, rtp_timestamp,
                                    rtp_clock_frequency_hz)) {
      last_receive_time_ = Timestamp::MinusInfinity();
      return std::nullopt;
    }

    return AbsoluteCaptureTime{
        .absolute_capture_timestamp = InterpolateAbsoluteCaptureTimestamp(
            rtp_timestamp, rtp_clock_frequency_hz, last_rtp_timestamp_,
            last_received_extension_.absolute_capture_timestamp),
        .estimated_capture_clock_offset =
            last_received_extension_.estimated_capture_clock_offset,
    };
  } else {
    last_source_ = source;
    last_rtp_timestamp_ = rtp_timestamp;
    last_rtp_clock_frequency_hz_ = rtp_clock_frequency_hz;
    last_received_extension_ = *received_extension;

    last_receive_time_ = receive_time;
    // Record statistics on the abs-capture-time extension
    if (!first_extension_time_) {
      RTC_HISTOGRAM_COUNTS_1M("WebRTC.Call.AbsCapture.ExtensionWait",
                              (receive_time - *first_packet_time_).ms());
      first_extension_time_ = receive_time;
    }
    int64_t ntp_delta =
        uint64_t{clock_->ConvertTimestampToNtpTime(receive_time)} -
        received_extension->absolute_capture_timestamp;
    TimeDelta capture_delta = TimeDelta::Micros(Q32x32ToInt64Us(ntp_delta));
    RTC_HISTOGRAM_COUNTS_1G("WebRTC.Call.AbsCapture.Delta",
                            abs(capture_delta.us()));
    if (previous_capture_delta_) {
      RTC_HISTOGRAM_COUNTS_1G(
          "WebRTC.Call.AbsCapture.DeltaDeviation",
          abs((capture_delta - *previous_capture_delta_).us()));
    }
    previous_capture_delta_ = capture_delta;
    if (received_extension->estimated_capture_clock_offset) {
      if (!first_offset_time_) {
        RTC_HISTOGRAM_COUNTS_1M("WebRTC.Call.AbsCapture.OffsetWait",
                                (receive_time - *first_packet_time_).ms());
        first_offset_time_ = receive_time;
      }
      TimeDelta offset_as_delta = TimeDelta::Micros(
          Q32x32ToInt64Us(*received_extension->estimated_capture_clock_offset));
      RTC_HISTOGRAM_COUNTS_1G("WebRTC.Call.AbsCapture.Offset",
                              abs(offset_as_delta.us()));
      if (previous_offset_as_delta_) {
        RTC_HISTOGRAM_COUNTS_1G(
            "WebRTC.Call.AbsCapture.OffsetDeviation",
            abs((offset_as_delta - *previous_offset_as_delta_).us()));
      }
      previous_offset_as_delta_ = offset_as_delta;
    }
    return received_extension;
  }
}

uint64_t AbsoluteCaptureTimeInterpolator::InterpolateAbsoluteCaptureTimestamp(
    uint32_t rtp_timestamp,
    int rtp_clock_frequency_hz,
    uint32_t last_rtp_timestamp,
    uint64_t last_absolute_capture_timestamp) {
  RTC_DCHECK_GT(rtp_clock_frequency_hz, 0);

  return last_absolute_capture_timestamp +
         static_cast<int64_t>(uint64_t{rtp_timestamp - last_rtp_timestamp}
                              << 32) /
             rtp_clock_frequency_hz;
}

bool AbsoluteCaptureTimeInterpolator::ShouldInterpolateExtension(
    Timestamp receive_time,
    uint32_t source,
    uint32_t /* rtp_timestamp */,
    int rtp_clock_frequency_hz) const {
  // Shouldn't if the last received extension is not eligible for interpolation,
  // in particular if we don't have a previously received extension stored.
  if (receive_time - last_receive_time_ > kInterpolationMaxInterval) {
    return false;
  }

  // Shouldn't if the source has changed.
  if (last_source_ != source) {
    return false;
  }

  // Shouldn't if the RTP clock frequency has changed.
  if (last_rtp_clock_frequency_hz_ != rtp_clock_frequency_hz) {
    return false;
  }

  // Shouldn't if the RTP clock frequency is invalid.
  if (rtp_clock_frequency_hz <= 0) {
    return false;
  }

  return true;
}

}  // namespace webrtc