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