Home | History | Annotate | Download | only in audio_processing 
      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_PROCESSING_VOICE_DETECTION_IMPL_H_
     12 #define WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_
     13 
     14 #include "webrtc/modules/audio_processing/include/audio_processing.h"
     15 #include "webrtc/modules/audio_processing/processing_component.h"
     16 
     17 namespace webrtc {
     18 
     19 class AudioBuffer;
     20 class CriticalSectionWrapper;
     21 
     22 class VoiceDetectionImpl : public VoiceDetection,
     23                            public ProcessingComponent {
     24  public:
     25   VoiceDetectionImpl(const AudioProcessing* apm, CriticalSectionWrapper* crit);
     26   virtual ~VoiceDetectionImpl();
     27 
     28   int ProcessCaptureAudio(AudioBuffer* audio);
     29 
     30   // VoiceDetection implementation.
     31   virtual bool is_enabled() const OVERRIDE;
     32 
     33   // ProcessingComponent implementation.
     34   virtual int Initialize() OVERRIDE;
     35 
     36  private:
     37   // VoiceDetection implementation.
     38   virtual int Enable(bool enable) OVERRIDE;
     39   virtual int set_stream_has_voice(bool has_voice) OVERRIDE;
     40   virtual bool stream_has_voice() const OVERRIDE;
     41   virtual int set_likelihood(Likelihood likelihood) OVERRIDE;
     42   virtual Likelihood likelihood() const OVERRIDE;
     43   virtual int set_frame_size_ms(int size) OVERRIDE;
     44   virtual int frame_size_ms() const OVERRIDE;
     45 
     46   // ProcessingComponent implementation.
     47   virtual void* CreateHandle() const OVERRIDE;
     48   virtual int InitializeHandle(void* handle) const OVERRIDE;
     49   virtual int ConfigureHandle(void* handle) const OVERRIDE;
     50   virtual void DestroyHandle(void* handle) const OVERRIDE;
     51   virtual int num_handles_required() const OVERRIDE;
     52   virtual int GetHandleError(void* handle) const OVERRIDE;
     53 
     54   const AudioProcessing* apm_;
     55   CriticalSectionWrapper* crit_;
     56   bool stream_has_voice_;
     57   bool using_external_vad_;
     58   Likelihood likelihood_;
     59   int frame_size_ms_;
     60   int frame_size_samples_;
     61 };
     62 }  // namespace webrtc
     63 
     64 #endif  // WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_
     65