Home | History | Annotate | Download | only in neteq 
      1 /*
      2  *  Copyright (c) 2012 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_PACKET_BUFFER_H_
     12 #define WEBRTC_MODULES_AUDIO_CODING_NETEQ_PACKET_BUFFER_H_
     13 
     14 #include "webrtc/base/constructormagic.h"
     15 #include "webrtc/modules/audio_coding/neteq/packet.h"
     16 #include "webrtc/typedefs.h"
     17 
     18 namespace webrtc {
     19 
     20 // Forward declaration.
     21 class DecoderDatabase;
     22 
     23 // This is the actual buffer holding the packets before decoding.
     24 class PacketBuffer {
     25  public:
     26   enum BufferReturnCodes {
     27     kOK = 0,
     28     kFlushed,
     29     kNotFound,
     30     kBufferEmpty,
     31     kInvalidPacket,
     32     kInvalidPointer
     33   };
     34 
     35   // Constructor creates a buffer which can hold a maximum of
     36   // |max_number_of_packets| packets.
     37   PacketBuffer(size_t max_number_of_packets);
     38 
     39   // Deletes all packets in the buffer before destroying the buffer.
     40   virtual ~PacketBuffer();
     41 
     42   // Flushes the buffer and deletes all packets in it.
     43   virtual void Flush();
     44 
     45   // Returns true for an empty buffer.
     46   virtual bool Empty() const { return buffer_.empty(); }
     47 
     48   // Inserts |packet| into the buffer. The buffer will take over ownership of
     49   // the packet object.
     50   // Returns PacketBuffer::kOK on success, PacketBuffer::kFlushed if the buffer
     51   // was flushed due to overfilling.
     52   virtual int InsertPacket(Packet* packet);
     53 
     54   // Inserts a list of packets into the buffer. The buffer will take over
     55   // ownership of the packet objects.
     56   // Returns PacketBuffer::kOK if all packets were inserted successfully.
     57   // If the buffer was flushed due to overfilling, only a subset of the list is
     58   // inserted, and PacketBuffer::kFlushed is returned.
     59   // The last three parameters are included for legacy compatibility.
     60   // TODO(hlundin): Redesign to not use current_*_payload_type and
     61   // decoder_database.
     62   virtual int InsertPacketList(PacketList* packet_list,
     63                                const DecoderDatabase& decoder_database,
     64                                uint8_t* current_rtp_payload_type,
     65                                uint8_t* current_cng_rtp_payload_type);
     66 
     67   // Gets the timestamp for the first packet in the buffer and writes it to the
     68   // output variable |next_timestamp|.
     69   // Returns PacketBuffer::kBufferEmpty if the buffer is empty,
     70   // PacketBuffer::kOK otherwise.
     71   virtual int NextTimestamp(uint32_t* next_timestamp) const;
     72 
     73   // Gets the timestamp for the first packet in the buffer with a timestamp no
     74   // lower than the input limit |timestamp|. The result is written to the output
     75   // variable |next_timestamp|.
     76   // Returns PacketBuffer::kBufferEmpty if the buffer is empty,
     77   // PacketBuffer::kOK otherwise.
     78   virtual int NextHigherTimestamp(uint32_t timestamp,
     79                                   uint32_t* next_timestamp) const;
     80 
     81   // Returns a (constant) pointer the RTP header of the first packet in the
     82   // buffer. Returns NULL if the buffer is empty.
     83   virtual const RTPHeader* NextRtpHeader() const;
     84 
     85   // Extracts the first packet in the buffer and returns a pointer to it.
     86   // Returns NULL if the buffer is empty. The caller is responsible for deleting
     87   // the packet.
     88   // Subsequent packets with the same timestamp as the one extracted will be
     89   // discarded and properly deleted. The number of discarded packets will be
     90   // written to the output variable |discard_count|.
     91   virtual Packet* GetNextPacket(int* discard_count);
     92 
     93   // Discards the first packet in the buffer. The packet is deleted.
     94   // Returns PacketBuffer::kBufferEmpty if the buffer is empty,
     95   // PacketBuffer::kOK otherwise.
     96   virtual int DiscardNextPacket();
     97 
     98   // Discards all packets that are (strictly) older than |timestamp_limit|.
     99   // Returns number of packets discarded.
    100   virtual int DiscardOldPackets(uint32_t timestamp_limit);
    101 
    102   // Returns the number of packets in the buffer, including duplicates and
    103   // redundant packets.
    104   virtual int NumPacketsInBuffer() const {
    105     return static_cast<int>(buffer_.size());
    106   }
    107 
    108   // Returns the number of samples in the buffer, including samples carried in
    109   // duplicate and redundant packets.
    110   virtual int NumSamplesInBuffer(DecoderDatabase* decoder_database,
    111                                  int last_decoded_length) const;
    112 
    113   // Increase the waiting time counter for every packet in the buffer by |inc|.
    114   // The default value for |inc| is 1.
    115   virtual void IncrementWaitingTimes(int inc = 1);
    116 
    117   virtual void BufferStat(int* num_packets, int* max_num_packets) const;
    118 
    119   // Static method that properly deletes the first packet, and its payload
    120   // array, in |packet_list|. Returns false if |packet_list| already was empty,
    121   // otherwise true.
    122   static bool DeleteFirstPacket(PacketList* packet_list);
    123 
    124   // Static method that properly deletes all packets, and their payload arrays,
    125   // in |packet_list|.
    126   static void DeleteAllPackets(PacketList* packet_list);
    127 
    128  private:
    129   size_t max_number_of_packets_;
    130   PacketList buffer_;
    131   DISALLOW_COPY_AND_ASSIGN(PacketBuffer);
    132 };
    133 
    134 }  // namespace webrtc
    135 #endif  // WEBRTC_MODULES_AUDIO_CODING_NETEQ_PACKET_BUFFER_H_
    136