voice_recognition_layer.h

Go to the documentation of this file.

#ifndef VOICE_RECOGNITION_LAYER_H
#define VOICE_RECOGNITION_LAYER_H
 
#include "../security_layer.h"
 
class QAudioInput;
class QProgressBar;
class QLabel;
class TranslationManager;

class VoiceRecognitionLayer final : public SecurityLayer {
    Q_OBJECT
 
public:
    explicit VoiceRecognitionLayer(QWidget *parent = nullptr);

    ~VoiceRecognitionLayer() override;

    void Initialize() override;

    void Reset() override;
 
private slots:
    void ProcessAudioInput();

    void UpdateProgress();

    void FinishRecognition();
 
private:
    void StartRecording();

    void StopRecording();

    void UpdateAudioVisualizer(const QByteArray &data);

    bool IsSpeaking(float audio_level) const;

    QLabel *audio_visualizer_label_;
    QProgressBar *recognition_progress_;
    QTimer *progress_timer_;
    QTimer *recognition_timer_;
    QTimer *audio_process_timer_;

    QAudioInput *audio_input_;
    QIODevice *audio_device_;

    QByteArray audio_buffer_;
    QVector<float> visualizer_data_;
    bool is_recording_;

    float noise_threshold_;
    bool is_speaking_;
    int silence_counter_;
    float current_audio_level_;
    TranslationManager *translator_;
};
 
#endif // VOICE_RECOGNITION_LAYER_H