whispercpptranscriber.h

Go to the documentation of this file.

#pragma once
 
#include "config.h"
#include "transcription/modelpackage.h"
#include "transcription/transcriptionengine.h"
#include "transcription/transcriptiontypes.h"
 
#include <atomic>
#include <memory>
 
struct whisper_context;
struct whisper_state;
class WhisperCppModelHandle;
 
class WhisperCppTranscriber final : public TranscriptionSession
{
public:
    [[nodiscard]] static std::shared_ptr<const TranscriptionModelHandle>
    loadModelHandle(const TranscriberConfig &config, RuntimeError *error = nullptr);
 
    [[nodiscard]] static std::unique_ptr<TranscriptionSession>
    createSession(TranscriberConfig config, std::shared_ptr<const TranscriptionModelHandle> model);
 
    WhisperCppTranscriber(TranscriberConfig config, std::shared_ptr<const TranscriptionModelHandle> model);
 
    ~WhisperCppTranscriber() override;
 
    WhisperCppTranscriber(const WhisperCppTranscriber &) = delete;
    WhisperCppTranscriber &operator=(const WhisperCppTranscriber &) = delete;
    WhisperCppTranscriber(WhisperCppTranscriber &&) = delete;
    WhisperCppTranscriber &operator=(WhisperCppTranscriber &&) = delete;
 
    [[nodiscard]] static QString backendNameStatic();
 
    [[nodiscard]] static BackendCapabilities capabilitiesStatic();
    [[nodiscard]] static RuntimeDiagnostics diagnosticsStatic();
    [[nodiscard]] QString backendName() const override;
 
    bool warmup(RuntimeError *error = nullptr) override;
 
    [[nodiscard]] TranscriptUpdate pushAudioChunk(const AudioChunk &chunk) override;
 
    [[nodiscard]] TranscriptUpdate finish() override;
 
    [[nodiscard]] TranscriptUpdate cancel() override;
 
private:
    static void freeState(whisper_state *state) noexcept;
 
    bool ensureState(RuntimeError *error = nullptr);
 
    TranscriberConfig m_config;
    std::shared_ptr<const WhisperCppModelHandle> m_model;
    std::unique_ptr<whisper_state, void (*)(whisper_state *)> m_state;
    std::vector<float> m_bufferedSamples;
    std::atomic_bool m_cancelRequested = false;
};
 
[[nodiscard]] std::shared_ptr<const TranscriptionEngine> createWhisperCppTranscriptionEngine(const TranscriberConfig &config);