src/third_party/webrtc/modules/rtp_rtcp/source/receive_statistics_impl.h

   1 /*
   2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #ifndef WEBRTC_MODULES_RTP_RTCP_SOURCE_RECEIVE_STATISTICS_IMPL_H_
  12 #define WEBRTC_MODULES_RTP_RTCP_SOURCE_RECEIVE_STATISTICS_IMPL_H_
  13
  14 #include "webrtc/modules/rtp_rtcp/interface/receive_statistics.h"
  15
  16 #include <algorithm>
  17
  18 #include "webrtc/modules/rtp_rtcp/source/bitrate.h"
  19 #include "webrtc/system_wrappers/interface/critical_section_wrapper.h"
  20 #include "webrtc/system_wrappers/interface/scoped_ptr.h"
  21
  22 namespace webrtc {
  23
  24 class CriticalSectionWrapper;
  25
  26 class StreamStatisticianImpl : public StreamStatistician {
  27  public:
  28   StreamStatisticianImpl(Clock* clock,
  29                          RtcpStatisticsCallback* rtcp_callback,
  30                          StreamDataCountersCallback* rtp_callback);
  31   virtual ~StreamStatisticianImpl() {}
  32
  33   virtual bool GetStatistics(RtcpStatistics* statistics, bool reset) OVERRIDE;
  34   virtual void GetDataCounters(uint32_t* bytes_received,
  35                                uint32_t* packets_received) const OVERRIDE;
  36   virtual uint32_t BitrateReceived() const OVERRIDE;
  37   virtual void ResetStatistics() OVERRIDE;
  38   virtual bool IsRetransmitOfOldPacket(const RTPHeader& header,
  39                                        int min_rtt) const OVERRIDE;
  40   virtual bool IsPacketInOrder(uint16_t sequence_number) const OVERRIDE;
  41
  42   void IncomingPacket(const RTPHeader& rtp_header,
  43                       size_t bytes,
  44                       bool retransmitted);
  45   void FecPacketReceived();
  46   void SetMaxReorderingThreshold(int max_reordering_threshold);
  47   void ProcessBitrate();
  48   virtual void LastReceiveTimeNtp(uint32_t* secs, uint32_t* frac) const;
  49
  50  private:
  51   bool InOrderPacketInternal(uint16_t sequence_number) const;
  52   RtcpStatistics CalculateRtcpStatistics();
  53   void UpdateJitter(const RTPHeader& header,
  54                     uint32_t receive_time_secs,
  55                     uint32_t receive_time_frac);
  56   void UpdateCounters(const RTPHeader& rtp_header,
  57                       size_t bytes,
  58                       bool retransmitted);
  59   void NotifyRtpCallback() LOCKS_EXCLUDED(stream_lock_.get());
  60   void NotifyRtcpCallback() LOCKS_EXCLUDED(stream_lock_.get());
  61
  62   Clock* clock_;
  63   scoped_ptr<CriticalSectionWrapper> stream_lock_;
  64   Bitrate incoming_bitrate_;
  65   uint32_t ssrc_;
  66   int max_reordering_threshold_;  // In number of packets or sequence numbers.
  67
  68   // Stats on received RTP packets.
  69   uint32_t jitter_q4_;
  70   uint32_t cumulative_loss_;
  71   uint32_t jitter_q4_transmission_time_offset_;
  72
  73   int64_t last_receive_time_ms_;
  74   uint32_t last_receive_time_secs_;
  75   uint32_t last_receive_time_frac_;
  76   uint32_t last_received_timestamp_;
  77   int32_t last_received_transmission_time_offset_;
  78   uint16_t received_seq_first_;
  79   uint16_t received_seq_max_;
  80   uint16_t received_seq_wraps_;
  81
  82   // Current counter values.
  83   uint16_t received_packet_overhead_;
  84   StreamDataCounters receive_counters_;
  85
  86   // Counter values when we sent the last report.
  87   uint32_t last_report_inorder_packets_;
  88   uint32_t last_report_old_packets_;
  89   uint16_t last_report_seq_max_;
  90   RtcpStatistics last_reported_statistics_;
  91
  92   RtcpStatisticsCallback* const rtcp_callback_;
  93   StreamDataCountersCallback* const rtp_callback_;
  94 };
  95
  96 class ReceiveStatisticsImpl : public ReceiveStatistics,
  97                               public RtcpStatisticsCallback,
  98                               public StreamDataCountersCallback {
  99  public:
 100   explicit ReceiveStatisticsImpl(Clock* clock);
 101
 102   ~ReceiveStatisticsImpl();
 103
 104   // Implement ReceiveStatistics.
 105   virtual void IncomingPacket(const RTPHeader& header,
 106                               size_t bytes,
 107                               bool retransmitted) OVERRIDE;
 108   virtual void FecPacketReceived(uint32_t ssrc) OVERRIDE;
 109   virtual StatisticianMap GetActiveStatisticians() const OVERRIDE;
 110   virtual StreamStatistician* GetStatistician(uint32_t ssrc) const OVERRIDE;
 111   virtual void SetMaxReorderingThreshold(int max_reordering_threshold) OVERRIDE;
 112
 113   // Implement Module.
 114   virtual int32_t Process() OVERRIDE;
 115   virtual int32_t TimeUntilNextProcess() OVERRIDE;
 116
 117   void ChangeSsrc(uint32_t from_ssrc, uint32_t to_ssrc);
 118
 119   virtual void RegisterRtcpStatisticsCallback(RtcpStatisticsCallback* callback)
 120       OVERRIDE;
 121
 122   virtual void RegisterRtpStatisticsCallback(
 123       StreamDataCountersCallback* callback) OVERRIDE;
 124
 125  private:
 126   virtual void StatisticsUpdated(const RtcpStatistics& statistics,
 127                                  uint32_t ssrc) OVERRIDE;
 128   virtual void DataCountersUpdated(const StreamDataCounters& counters,
 129                                    uint32_t ssrc) OVERRIDE;
 130
 131   typedef std::map<uint32_t, StreamStatisticianImpl*> StatisticianImplMap;
 132
 133   Clock* clock_;
 134   scoped_ptr<CriticalSectionWrapper> receive_statistics_lock_;
 135   int64_t last_rate_update_ms_;
 136   StatisticianImplMap statisticians_;
 137
 138   RtcpStatisticsCallback* rtcp_stats_callback_;
 139   StreamDataCountersCallback* rtp_stats_callback_;
 140 };
 141 }  // namespace webrtc
 142 #endif  // WEBRTC_MODULES_RTP_RTCP_SOURCE_RECEIVE_STATISTICS_IMPL_H_