snowboy-detect.h

#pragma once
#include <memory>
#include <string>
 
namespace snowboy {
    namespace testing {
        class Inspector;
    }
    struct WaveHeader;
    class PipelineDetect;
    struct PipelineVad;
    class PipelinePersonalEnroll;
    class PipelineTemplateCut;
    struct MatrixBase;
 
    class SnowboyDetect {
        friend class testing::Inspector;
 
    public:
        SnowboyDetect(const std::string& resource_filename,
                      const std::string& model_str);
 
        bool Reset();
 
        int RunDetection(const std::string& data, bool is_end = false);
 
        int RunDetection(const float* const data,
                         const int array_length, bool is_end = false);
 
        int RunDetection(const int16_t* const data,
                         const int array_length, bool is_end = false);
 
        int RunDetection(const int32_t* const data,
                         const int array_length, bool is_end = false);
 
        void SetSensitivity(const std::string& sensitivity_str);
 
        void SetHighSensitivity(const std::string& high_sensitivity_str);
 
        std::string GetSensitivity() const;
 
        void SetAudioGain(const float audio_gain);
 
        void UpdateModel() const;
 
        int NumHotwords() const;
 
        void ApplyFrontend(const bool apply_frontend);
 
        int SampleRate() const;
 
        int NumChannels() const;
 
        int BitsPerSample() const;
 
        ~SnowboyDetect();
 
    private:
        std::unique_ptr<WaveHeader> wave_header_;
        std::unique_ptr<PipelineDetect> detect_pipeline_;
    };
 
    class SnowboyVad {
        friend class testing::Inspector;
 
    public:
        SnowboyVad(const std::string& resource_filename);
 
        bool Reset();
 
        int RunVad(const std::string& data, bool is_end = false);
 
        int RunVad(const float* const data,
                   const int array_length, bool is_end = false);
 
        int RunVad(const int16_t* const data,
                   const int array_length, bool is_end = false);
 
        int RunVad(const int32_t* const data,
                   const int array_length, bool is_end = false);
 
        void SetAudioGain(const float audio_gain);
 
        void ApplyFrontend(const bool apply_frontend);
 
        int SampleRate() const;
 
        int NumChannels() const;
 
        int BitsPerSample() const;
 
        ~SnowboyVad();
 
    private:
        std::unique_ptr<WaveHeader> wave_header_;
        std::unique_ptr<PipelineVad> vad_pipeline_;
    };
 
    class SnowboyPersonalEnroll {
        friend class testing::Inspector;
 
    public:
        SnowboyPersonalEnroll(const std::string& resource_filename, const std::string& model_filename);
 
        int RunEnrollment(const std::string& data);
 
        int RunEnrollment(const float* const data, const int array_length);
 
        int RunEnrollment(const int16_t* const data, const int array_length);
 
        int RunEnrollment(const int32_t* const data, const int array_length);
 
        bool Reset();
 
        int GetNumTemplates() const;
 
        int SampleRate() const;
 
        int NumChannels() const;
 
        int BitsPerSample() const;
 
        ~SnowboyPersonalEnroll();
 
    private:
        int RunEnrollment(const MatrixBase& mat);
        std::unique_ptr<WaveHeader> wave_header_;
        std::unique_ptr<PipelinePersonalEnroll> enroll_pipeline_;
    };
 
    class SnowboyTemplateCut {
        friend class testing::Inspector;
 
    public:
        SnowboyTemplateCut(const std::string& resource_filename);
 
        std::string CutTemplate(const std::string& data);
 
        int CutTemplate(const float* const data, const int array_length, float* const data_out, int* array_length_out);
 
        int CutTemplate(const int16_t* const data, const int array_length, int16_t* const data_out, int* array_length_out);
 
        int CutTemplate(const int32_t* const data, const int array_length, int32_t* const data_out, int* array_length_out);
 
        bool Reset();
 
        int SampleRate() const;
 
        int NumChannels() const;
 
        int BitsPerSample() const;
 
        ~SnowboyTemplateCut();
 
    private:
        std::unique_ptr<WaveHeader> wave_header_;
        std::unique_ptr<PipelineTemplateCut> cut_pipeline_;
    };
 
} // namespace snowboy