123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216 |
- /*
- * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
- *
- * Use of this source code is governed by a BSD-style license
- * that can be found in the LICENSE file in the root of the source
- * tree. An additional intellectual property rights grant can be found
- * in the file PATENTS. All contributing project authors may
- * be found in the AUTHORS file in the root of the source tree.
- */
- #ifndef MODULES_AUDIO_CODING_NETEQ_DECODER_DATABASE_H_
- #define MODULES_AUDIO_CODING_NETEQ_DECODER_DATABASE_H_
- #include <map>
- #include <memory>
- #include <string>
- #include "api/audio_codecs/audio_decoder_factory.h"
- #include "api/audio_codecs/audio_format.h"
- #include "api/scoped_refptr.h"
- #include "modules/audio_coding/codecs/cng/webrtc_cng.h"
- #include "modules/audio_coding/neteq/packet.h"
- #include "rtc_base/constructor_magic.h"
- namespace webrtc {
- class DecoderDatabase {
- public:
- enum DatabaseReturnCodes {
- kOK = 0,
- kInvalidRtpPayloadType = -1,
- kCodecNotSupported = -2,
- kInvalidSampleRate = -3,
- kDecoderExists = -4,
- kDecoderNotFound = -5,
- kInvalidPointer = -6
- };
- // Class that stores decoder info in the database.
- class DecoderInfo {
- public:
- DecoderInfo(const SdpAudioFormat& audio_format,
- absl::optional<AudioCodecPairId> codec_pair_id,
- AudioDecoderFactory* factory,
- const std::string& codec_name);
- explicit DecoderInfo(const SdpAudioFormat& audio_format,
- absl::optional<AudioCodecPairId> codec_pair_id,
- AudioDecoderFactory* factory = nullptr);
- DecoderInfo(DecoderInfo&&);
- ~DecoderInfo();
- // Get the AudioDecoder object, creating it first if necessary.
- AudioDecoder* GetDecoder() const;
- // Delete the AudioDecoder object, unless it's external. (This means we can
- // always recreate it later if we need it.)
- void DropDecoder() const { decoder_.reset(); }
- int SampleRateHz() const {
- if (IsDtmf()) {
- // DTMF has a 1:1 mapping between clock rate and sample rate.
- return audio_format_.clockrate_hz;
- }
- const AudioDecoder* decoder = GetDecoder();
- RTC_DCHECK_EQ(1, !!decoder + !!cng_decoder_);
- return decoder ? decoder->SampleRateHz() : cng_decoder_->sample_rate_hz;
- }
- const SdpAudioFormat& GetFormat() const { return audio_format_; }
- // Returns true if the decoder's format is comfort noise.
- bool IsComfortNoise() const {
- RTC_DCHECK_EQ(!!cng_decoder_, subtype_ == Subtype::kComfortNoise);
- return subtype_ == Subtype::kComfortNoise;
- }
- // Returns true if the decoder's format is DTMF.
- bool IsDtmf() const { return subtype_ == Subtype::kDtmf; }
- // Returns true if the decoder's format is RED.
- bool IsRed() const { return subtype_ == Subtype::kRed; }
- // Returns true if the decoder's format is named |name|.
- bool IsType(const char* name) const;
- // Returns true if the decoder's format is named |name|.
- bool IsType(const std::string& name) const;
- const std::string& get_name() const { return name_; }
- private:
- // TODO(ossu): |name_| is kept here while we retain the old external
- // decoder interface. Remove this once using an
- // AudioDecoderFactory has supplanted the old functionality.
- const std::string name_;
- const SdpAudioFormat audio_format_;
- const absl::optional<AudioCodecPairId> codec_pair_id_;
- AudioDecoderFactory* const factory_;
- mutable std::unique_ptr<AudioDecoder> decoder_;
- // Set iff this is a comfort noise decoder.
- struct CngDecoder {
- static absl::optional<CngDecoder> Create(const SdpAudioFormat& format);
- int sample_rate_hz;
- };
- const absl::optional<CngDecoder> cng_decoder_;
- enum class Subtype : int8_t { kNormal, kComfortNoise, kDtmf, kRed };
- static Subtype SubtypeFromFormat(const SdpAudioFormat& format);
- const Subtype subtype_;
- };
- // Maximum value for 8 bits, and an invalid RTP payload type (since it is
- // only 7 bits).
- static const uint8_t kRtpPayloadTypeError = 0xFF;
- DecoderDatabase(
- const rtc::scoped_refptr<AudioDecoderFactory>& decoder_factory,
- absl::optional<AudioCodecPairId> codec_pair_id);
- virtual ~DecoderDatabase();
- // Returns true if the database is empty.
- virtual bool Empty() const;
- // Returns the number of decoders registered in the database.
- virtual int Size() const;
- // Resets the database, erasing all registered payload types, and deleting
- // any AudioDecoder objects that were not externally created and inserted
- // using InsertExternal().
- virtual void Reset();
- // Replaces the existing set of decoders with the given set. Returns the
- // payload types that were reassigned or removed while doing so.
- virtual std::vector<int> SetCodecs(
- const std::map<int, SdpAudioFormat>& codecs);
- // Registers a decoder for the given payload type. Returns kOK on success;
- // otherwise an error code.
- virtual int RegisterPayload(int rtp_payload_type,
- const SdpAudioFormat& audio_format);
- // Removes the entry for |rtp_payload_type| from the database.
- // Returns kDecoderNotFound or kOK depending on the outcome of the operation.
- virtual int Remove(uint8_t rtp_payload_type);
- // Remove all entries.
- virtual void RemoveAll();
- // Returns a pointer to the DecoderInfo struct for |rtp_payload_type|. If
- // no decoder is registered with that |rtp_payload_type|, NULL is returned.
- virtual const DecoderInfo* GetDecoderInfo(uint8_t rtp_payload_type) const;
- // Sets the active decoder to be |rtp_payload_type|. If this call results in a
- // change of active decoder, |new_decoder| is set to true. The previous active
- // decoder's AudioDecoder object is deleted.
- virtual int SetActiveDecoder(uint8_t rtp_payload_type, bool* new_decoder);
- // Returns the current active decoder, or NULL if no active decoder exists.
- virtual AudioDecoder* GetActiveDecoder() const;
- // Sets the active comfort noise decoder to be |rtp_payload_type|. If this
- // call results in a change of active comfort noise decoder, the previous
- // active decoder's AudioDecoder object is deleted.
- virtual int SetActiveCngDecoder(uint8_t rtp_payload_type);
- // Returns the current active comfort noise decoder, or NULL if no active
- // comfort noise decoder exists.
- virtual ComfortNoiseDecoder* GetActiveCngDecoder() const;
- // The following are utility methods: they will look up DecoderInfo through
- // GetDecoderInfo and call the respective method on that info object, if it
- // exists.
- // Returns a pointer to the AudioDecoder object associated with
- // |rtp_payload_type|, or NULL if none is registered. If the AudioDecoder
- // object does not exist for that decoder, the object is created.
- AudioDecoder* GetDecoder(uint8_t rtp_payload_type) const;
- // Returns if |rtp_payload_type| is registered with a format named |name|.
- bool IsType(uint8_t rtp_payload_type, const char* name) const;
- // Returns if |rtp_payload_type| is registered with a format named |name|.
- bool IsType(uint8_t rtp_payload_type, const std::string& name) const;
- // Returns true if |rtp_payload_type| is registered as comfort noise.
- bool IsComfortNoise(uint8_t rtp_payload_type) const;
- // Returns true if |rtp_payload_type| is registered as DTMF.
- bool IsDtmf(uint8_t rtp_payload_type) const;
- // Returns true if |rtp_payload_type| is registered as RED.
- bool IsRed(uint8_t rtp_payload_type) const;
- // Returns kOK if all packets in |packet_list| carry payload types that are
- // registered in the database. Otherwise, returns kDecoderNotFound.
- int CheckPayloadTypes(const PacketList& packet_list) const;
- private:
- typedef std::map<uint8_t, DecoderInfo> DecoderMap;
- DecoderMap decoders_;
- int active_decoder_type_;
- int active_cng_decoder_type_;
- mutable std::unique_ptr<ComfortNoiseDecoder> active_cng_decoder_;
- rtc::scoped_refptr<AudioDecoderFactory> decoder_factory_;
- const absl::optional<AudioCodecPairId> codec_pair_id_;
- RTC_DISALLOW_COPY_AND_ASSIGN(DecoderDatabase);
- };
- } // namespace webrtc
- #endif // MODULES_AUDIO_CODING_NETEQ_DECODER_DATABASE_H_
|