Cast: Update simulator tool with more inputs.

media/cast/test/utility/udp_proxy.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef MEDIA_CAST_TEST_UTILITY_UDP_PROXY_H_
 #define MEDIA_CAST_TEST_UTILITY_UDP_PROXY_H_
 
 #include <vector>
 
 #include "base/basictypes.h"
+#include "base/memory/linked_ptr.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
 #include "base/single_thread_task_runner.h"
 #include "media/cast/transport/cast_transport_config.h"
 #include "net/base/ip_endpoint.h"
+#include "third_party/mt19937ar/mt19937ar.h"
 
 namespace net {
 class NetLog;
 };
 
 namespace base {
 class TickClock;
 };
 
 namespace media {
@@ skipping 10 matching lines @@
       const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
       base::TickClock* clock);
   virtual void AppendToPipe(scoped_ptr<PacketPipe> pipe);
  protected:
   scoped_ptr<PacketPipe> pipe_;
   // Allows injection of fake task runner for testing.
   scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
   base::TickClock* clock_;
 };
 
+// Implements a Interrupted Poisson Process for packet delivery.
+// The process has 2 states: ON and OFF, the rate of switching between
+// these two states are defined.
+// When in ON state packets are sent according to a defined rate.
+// When in OFF state packets are not sent.
+// The rate above is the average rate of a poisson distribution.
+class InterruptedPoissonProcess {
+ public:
+  InterruptedPoissonProcess(
+      const std::vector<double>& average_rates,
+      double coef_burstiness,
+      double coef_variance,
+      uint32 rand_seed);
+
+  scoped_ptr<PacketPipe> NewBuffer(size_t size);
+
+ private:
+  class InternalBuffer;
+
+  // |task_runner| is the executor of the IO thread.
+  // |clock| is the system clock.
+  void InitOnIOThread(
+      const scoped_refptr<base::SingleThreadTaskRunner>& task_runner,
+      base::TickClock* clock);
+
+  base::TimeDelta NextEvent(double rate);
+  double RandDouble();
+  void ComputeRates();
+  void UpdateRates();
+  void SwitchOff();
+  void SwitchOn();
+  void SendPacket();
+
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
+  base::TickClock* clock_;
+  const std::vector<double> average_rates_;
+  const double coef_burstiness_;
+  const double coef_variance_;
+  int rate_index_;
+
+  // The following rates are per milliseconds.
+  double send_rate_;
+  double switch_off_rate_;
+  double switch_on_rate_;
+  bool on_state_;
+
+  std::vector<base::WeakPtr<InternalBuffer> > send_buffers_;
+
+  // Fast pseudo random number generator.
+  MersenneTwister mt_rand_;
+
+  base::WeakPtrFactory<InterruptedPoissonProcess> weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(InterruptedPoissonProcess);
+};
+
 // A UDPProxy will set up a UDP socket and bind to |local_port|.
 // Packets send to that port will be forwarded to |destination|.
 // Packets send from |destination| to |local_port| will be returned
 // to whoever sent a packet to |local_port| last. (Not counting packets
 // from |destination|.) The UDPProxy will run a separate thread to
 // do the forwarding of packets, and will keep doing so until destroyed.
 // You can insert delays and packet drops by supplying a PacketPipe.
 // The PacketPipes may also be NULL if you just want to forward packets.
 class UDPProxy {
  public:
@@ skipping 51 matching lines @@
 // This method builds a stack of PacketPipes to emulate a crappy wifi network.
 // ~2mbit, 20% packet loss, ~40ms latency and packets can get reordered.
 // 300ms drouputs every ~2 seconds.
 scoped_ptr<PacketPipe> EvilNetwork();
 
 }  // namespace test
 }  // namespace cast
 }  // namespace media
 
 #endif  // MEDIA_CAST_TEST_UTILITY_UDP_PROXY_H_