Home | History | Annotate | Download | only in tools
      1 /*
      2  *  Copyright (c) 2014 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_AUDIO_CODING_NETEQ_TOOLS_PACKET_SOURCE_H_
     12 #define WEBRTC_MODULES_AUDIO_CODING_NETEQ_TOOLS_PACKET_SOURCE_H_
     13 
     14 #include <bitset>
     15 
     16 #include "webrtc/base/constructormagic.h"
     17 #include "webrtc/modules/audio_coding/neteq/tools/packet.h"
     18 #include "webrtc/typedefs.h"
     19 
     20 namespace webrtc {
     21 namespace test {
     22 
     23 // Interface class for an object delivering RTP packets to test applications.
     24 class PacketSource {
     25  public:
     26   PacketSource() : use_ssrc_filter_(false), ssrc_(0) {}
     27   virtual ~PacketSource() {}
     28 
     29   // Returns a pointer to the next packet. Returns NULL if the source is
     30   // depleted, or if an error occurred.
     31   virtual Packet* NextPacket() = 0;
     32 
     33   virtual void FilterOutPayloadType(uint8_t payload_type) {
     34     filter_.set(payload_type, true);
     35   }
     36 
     37   virtual void SelectSsrc(uint32_t ssrc) {
     38     use_ssrc_filter_ = true;
     39     ssrc_ = ssrc;
     40   }
     41 
     42  protected:
     43   std::bitset<128> filter_;  // Payload type is 7 bits in the RFC.
     44   // If SSRC filtering discards all packet that do not match the SSRC.
     45   bool use_ssrc_filter_;  // True when SSRC filtering is active.
     46   uint32_t ssrc_;  // The selected SSRC. All other SSRCs will be discarded.
     47 
     48  private:
     49   RTC_DISALLOW_COPY_AND_ASSIGN(PacketSource);
     50 };
     51 
     52 }  // namespace test
     53 }  // namespace webrtc
     54 #endif  // WEBRTC_MODULES_AUDIO_CODING_NETEQ_TOOLS_PACKET_SOURCE_H_
     55