src/media/cast/transport/rtp_sender/rtp_sender.h

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 // This file contains the interface to the cast RTP sender.
   6
   7 #ifndef MEDIA_CAST_TRANSPORT_RTP_SENDER_RTP_SENDER_H_
   8 #define MEDIA_CAST_TRANSPORT_RTP_SENDER_RTP_SENDER_H_
   9
  10 #include <map>
  11 #include <set>
  12
  13 #include "base/memory/scoped_ptr.h"
  14 #include "base/time/tick_clock.h"
  15 #include "base/time/time.h"
  16 #include "base/memory/weak_ptr.h"
  17 #include "media/cast/cast_config.h"
  18 #include "media/cast/cast_environment.h"
  19 #include "media/cast/transport/cast_transport_defines.h"
  20 #include "media/cast/transport/cast_transport_sender.h"
  21 #include "media/cast/transport/pacing/paced_sender.h"
  22 #include "media/cast/transport/rtp_sender/packet_storage/packet_storage.h"
  23 #include "media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer.h"
  24
  25 namespace media {
  26 namespace cast {
  27
  28 namespace transport {
  29
  30 // This object is only called from the main cast thread.
  31 // This class handles splitting encoded audio and video frames into packets and
  32 // add an RTP header to each packet. The sent packets are stored until they are
  33 // acknowledged by the remote peer or timed out.
  34 class RtpSender {
  35  public:
  36   RtpSender(
  37       base::TickClock* clock,
  38       const scoped_refptr<base::SingleThreadTaskRunner>& transport_task_runner,
  39       PacedSender* const transport);
  40
  41   ~RtpSender();
  42
  43   // Initialize audio stack. Audio must be initialized prior to sending encoded
  44   // audio frames. Returns false if configuration is invalid.
  45   bool InitializeAudio(const CastTransportAudioConfig& config);
  46
  47   // Initialize video stack. Video must be initialized prior to sending encoded
  48   // video frames. Returns false if configuration is invalid.
  49   bool InitializeVideo(const CastTransportVideoConfig& config);
  50
  51   // The video_frame objects ownership is handled by the main cast thread.
  52   void IncomingEncodedVideoFrame(const EncodedVideoFrame* video_frame,
  53                                  const base::TimeTicks& capture_time);
  54
  55   // The audio_frame objects ownership is handled by the main cast thread.
  56   void IncomingEncodedAudioFrame(const EncodedAudioFrame* audio_frame,
  57                                  const base::TimeTicks& recorded_time);
  58
  59   void ResendPackets(const MissingFramesAndPacketsMap& missing_packets);
  60
  61   // Set the callback on which RTP statistics data will be returned. Calling
  62   // this function would start a timer that would schedule the callback in
  63   // a constant interval.
  64   void SubscribeRtpStatsCallback(const CastTransportRtpStatistics& callback);
  65
  66  private:
  67   void ScheduleNextStatsReport();
  68   void RtpStatistics();
  69   void UpdateSequenceNumber(Packet* packet);
  70
  71   base::TickClock* clock_;  // Not owned by this class.
  72   RtpPacketizerConfig config_;
  73   scoped_ptr<RtpPacketizer> packetizer_;
  74   scoped_ptr<PacketStorage> storage_;
  75   PacedSender* const transport_;
  76   CastTransportRtpStatistics stats_callback_;
  77   scoped_refptr<base::SingleThreadTaskRunner> transport_task_runner_;
  78
  79   // NOTE: Weak pointers must be invalidated before all other member variables.
  80   base::WeakPtrFactory<RtpSender> weak_factory_;
  81
  82   DISALLOW_COPY_AND_ASSIGN(RtpSender);
  83 };
  84
  85 }  // namespace transport
  86 }  // namespace cast
  87 }  // namespace media
  88
  89 #endif  // MEDIA_CAST_TRANSPORT_RTP_SENDER_RTP_SENDER_H_