1 files changed, 10 insertions, 0 deletions
diff --git a/modules/audio_coding/neteq/tools/packet_source.h b/modules/audio_coding/neteq/tools/packet_source.h
index 669bc14e..ab9ef83e 100644
--- a/modules/audio_coding/neteq/tools/packet_source.h
+++ b/modules/audio_coding/neteq/tools/packet_source.h
@@ -11,7 +11,10 @@
 #ifndef WEBRTC_MODULES_AUDIO_CODING_NETEQ_TOOLS_PACKET_SOURCE_H_
 #define WEBRTC_MODULES_AUDIO_CODING_NETEQ_TOOLS_PACKET_SOURCE_H_
 
+#include <bitset>
+
 #include "webrtc/base/constructormagic.h"
+#include "webrtc/typedefs.h"
 
 namespace webrtc {
 namespace test {
@@ -28,6 +31,13 @@ class PacketSource {
   // depleted, or if an error occurred.
   virtual Packet* NextPacket() = 0;
 
+  virtual void FilterOutPayloadType(uint8_t payload_type) {
+    filter_.set(payload_type, true);
+  }
+
+ protected:
+  std::bitset<128> filter_;  // Payload type is 7 bits in the RFC.
+
  private:
   DISALLOW_COPY_AND_ASSIGN(PacketSource);
 };