More Simulation Framework features

webrtc/modules/remote_bitrate_estimator/test/packet_sender.h

Index: webrtc/modules/remote_bitrate_estimator/test/packet_sender.h
diff --git a/webrtc/modules/remote_bitrate_estimator/test/packet_sender.h b/webrtc/modules/remote_bitrate_estimator/test/packet_sender.h
index 2e690c83770c7c6b0c2e9f8b951b957e2319e9bb..487dc30ed45128bf27ada4a99095235fadd415b1 100644
--- a/webrtc/modules/remote_bitrate_estimator/test/packet_sender.h
+++ b/webrtc/modules/remote_bitrate_estimator/test/packet_sender.h
@@ -40,6 +40,7 @@ class PacketSender : public PacketProcessor {
   // output of the estimators is sampled and therefore the baseline files may
   // have to be regenerated.
   virtual int GetFeedbackIntervalMs() const = 0;
+  virtual void set_choke_filter(ChokeFilter* choke_filter) {}
   void SetSenderTimestamps(Packets* in_out);
 
  protected:
@@ -63,11 +64,15 @@ class VideoSender : public PacketSender, public BitrateObserver {
                         uint8_t fraction_lost,
                         int64_t rtt) override;
 
+  void Pause();
+  void Resume();
+
  protected:
   void ProcessFeedbackAndGeneratePackets(int64_t time_ms,
                                          std::list<FeedbackPacket*>* feedbacks,
                                          Packets* generated);
 
+  bool running_;
   VideoSource* source_;
   rtc::scoped_ptr<BweSender> bwe_;
   int64_t start_of_run_ms_;
@@ -112,21 +117,16 @@ class PacedVideoSender : public VideoSender, public PacedSender::Callback {
 
 class TcpSender : public PacketSender {
  public:
-  TcpSender(PacketProcessorListener* listener, int flow_id, int64_t offset_ms)
-      : PacketSender(listener, flow_id),
-        cwnd_(10),
-        ssthresh_(std::numeric_limits<int>::max()),
-        ack_received_(false),
-        last_acked_seq_num_(0),
-        next_sequence_number_(0),
-        offset_ms_(offset_ms),
-        last_reduction_time_ms_(-1),
-        last_rtt_ms_(0) {}
-
+  TcpSender(PacketProcessorListener* listener, int flow_id, int64_t offset_ms);
+  TcpSender(PacketProcessorListener* listener,
+            int flow_id,
+            int64_t offset_ms,
+            int send_limit_bytes);
   virtual ~TcpSender() {}
 
   void RunFor(int64_t time_ms, Packets* in_out) override;
   int GetFeedbackIntervalMs() const override { return 10; }
+  void set_choke_filter(ChokeFilter* choke_filter) override;
 
  private:
   struct InFlight {
@@ -163,6 +163,13 @@ class TcpSender : public PacketSender {
   int64_t offset_ms_;
   int64_t last_reduction_time_ms_;
   int64_t last_rtt_ms_;
+  int total_sent_bytes_;
+  int send_limit_bytes_;  // Initialized by default as kNoLimit.
+  bool running_;          // Initialized by default as true.
+  ChokeFilter* choke_filter_;
+  int64_t last_generated_packets_ms_;
+  uint32_t estimated_kbps_;
+  size_t num_recent_sent_packets_;
 };
 }  // namespace bwe
 }  // namespace testing