rtc_tools/rtc_event_log_visualizer/log_scream_simulation.h

/*
 *  Copyright 2025 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.
 */
#ifndef RTC_TOOLS_RTC_EVENT_LOG_VISUALIZER_LOG_SCREAM_SIMULATION_H_
#define RTC_TOOLS_RTC_EVENT_LOG_VISUALIZER_LOG_SCREAM_SIMULATION_H_

#include <cstdint>
#include <optional>
#include <vector>

#include "api/candidate.h"
#include "api/environment/environment.h"
#include "api/transport/network_types.h"
#include "api/units/data_rate.h"
#include "api/units/data_size.h"
#include "api/units/time_delta.h"
#include "api/units/timestamp.h"
#include "logging/rtc_event_log/events/logged_rtp_rtcp.h"
#include "logging/rtc_event_log/events/rtc_event_ice_candidate_pair_config.h"
#include "logging/rtc_event_log/rtc_event_log_parser.h"
#include "modules/congestion_controller/rtp/transport_feedback_adapter.h"
#include "modules/congestion_controller/scream/scream_v2.h"
#include "rtc_base/bitrate_tracker.h"

namespace webrtc {

class LogScreamSimulation {
 public:
  struct State {
    Timestamp time;

    DataRate target_rate = DataRate::Zero();
    DataRate pacing_rate = DataRate::Zero();
    DataRate send_rate = DataRate::Zero();

    DataSize ref_window = DataSize::Zero();
    DataSize ref_window_i = DataSize::Zero();
    DataSize max_data_in_flight = DataSize::Zero();
    DataSize data_in_flight = DataSize::Zero();

    double queue_delay_dev_norm;
    double l4s_alpha = 0.0;
    double l4s_alpha_v = 0.0;
  };

  struct Config {
    TimeDelta rate_window = TimeDelta::Millis(100);
  };

  LogScreamSimulation(const Config& config, const Environment& env);
  ~LogScreamSimulation() = default;
  void ProcessEventsInLog(const ParsedRtcEventLog& parsed_log_);

  const std::vector<State>& updates() const { return state_; }

 private:
  void ProcessUntil(Timestamp to_time);
  void OnPacketSent(const LoggedPacketInfo& packet);
  void OnTransportFeedback(const LoggedRtcpPacketTransportFeedback& feedback);
  void OnCongestionControlFeedback(
      const LoggedRtcpCongestionControlFeedback& feedback);
  void OnIceConfig(const LoggedIceCandidatePairConfig& candidate);
  void LogState(const TransportPacketsFeedback& msg);

  const Environment env_;
  std::optional<ScreamV2> scream_;

  Timestamp current_time_ = Timestamp::MinusInfinity();
  Timestamp last_process_ = Timestamp::MinusInfinity();
  TransportFeedbackAdapter transport_feedback_;
  BitrateTracker send_rate_tracker_;

  // With RFC 8888, transport sequence numbers are not stored per packet.
  // Instead, we generate one.
  int64_t next_ccfb_packet_id_ = 0;
  std::optional<IceCandidateType> local_candidate_type_;
  std::optional<IceCandidateType> remote_candidate_type_;

  std::vector<State> state_;
};
}  // namespace webrtc

#endif  // RTC_TOOLS_RTC_EVENT_LOG_VISUALIZER_LOG_SCREAM_SIMULATION_H_