Home | History | Annotate | Download | only in cng
      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_CODECS_CNG_AUDIO_ENCODER_CNG_H_
     12 #define WEBRTC_MODULES_AUDIO_CODING_CODECS_CNG_AUDIO_ENCODER_CNG_H_
     13 
     14 #include <vector>
     15 
     16 #include "webrtc/base/scoped_ptr.h"
     17 #include "webrtc/common_audio/vad/include/vad.h"
     18 #include "webrtc/modules/audio_coding/codecs/audio_encoder.h"
     19 #include "webrtc/modules/audio_coding/codecs/cng/webrtc_cng.h"
     20 
     21 namespace webrtc {
     22 
     23 // Deleter for use with scoped_ptr.
     24 struct CngInstDeleter {
     25   void operator()(CNG_enc_inst* ptr) const { WebRtcCng_FreeEnc(ptr); }
     26 };
     27 
     28 class Vad;
     29 
     30 class AudioEncoderCng final : public AudioEncoder {
     31  public:
     32   struct Config {
     33     bool IsOk() const;
     34 
     35     size_t num_channels = 1;
     36     int payload_type = 13;
     37     // Caller keeps ownership of the AudioEncoder object.
     38     AudioEncoder* speech_encoder = nullptr;
     39     Vad::Aggressiveness vad_mode = Vad::kVadNormal;
     40     int sid_frame_interval_ms = 100;
     41     int num_cng_coefficients = 8;
     42     // The Vad pointer is mainly for testing. If a NULL pointer is passed, the
     43     // AudioEncoderCng creates (and destroys) a Vad object internally. If an
     44     // object is passed, the AudioEncoderCng assumes ownership of the Vad
     45     // object.
     46     Vad* vad = nullptr;
     47   };
     48 
     49   explicit AudioEncoderCng(const Config& config);
     50   ~AudioEncoderCng() override;
     51 
     52   size_t MaxEncodedBytes() const override;
     53   int SampleRateHz() const override;
     54   size_t NumChannels() const override;
     55   int RtpTimestampRateHz() const override;
     56   size_t Num10MsFramesInNextPacket() const override;
     57   size_t Max10MsFramesInAPacket() const override;
     58   int GetTargetBitrate() const override;
     59   EncodedInfo EncodeInternal(uint32_t rtp_timestamp,
     60                              rtc::ArrayView<const int16_t> audio,
     61                              size_t max_encoded_bytes,
     62                              uint8_t* encoded) override;
     63   void Reset() override;
     64   bool SetFec(bool enable) override;
     65   bool SetDtx(bool enable) override;
     66   bool SetApplication(Application application) override;
     67   void SetMaxPlaybackRate(int frequency_hz) override;
     68   void SetProjectedPacketLossRate(double fraction) override;
     69   void SetTargetBitrate(int target_bps) override;
     70 
     71  private:
     72   EncodedInfo EncodePassive(size_t frames_to_encode,
     73                             size_t max_encoded_bytes,
     74                             uint8_t* encoded);
     75   EncodedInfo EncodeActive(size_t frames_to_encode,
     76                            size_t max_encoded_bytes,
     77                            uint8_t* encoded);
     78   size_t SamplesPer10msFrame() const;
     79 
     80   AudioEncoder* speech_encoder_;
     81   const int cng_payload_type_;
     82   const int num_cng_coefficients_;
     83   const int sid_frame_interval_ms_;
     84   std::vector<int16_t> speech_buffer_;
     85   std::vector<uint32_t> rtp_timestamps_;
     86   bool last_frame_active_;
     87   rtc::scoped_ptr<Vad> vad_;
     88   rtc::scoped_ptr<CNG_enc_inst, CngInstDeleter> cng_inst_;
     89 
     90   RTC_DISALLOW_COPY_AND_ASSIGN(AudioEncoderCng);
     91 };
     92 
     93 }  // namespace webrtc
     94 
     95 #endif  // WEBRTC_MODULES_AUDIO_CODING_CODECS_CNG_AUDIO_ENCODER_CNG_H_
     96