cpureferencetranscriber.h

Go to the documentation of this file.

#pragma once
 
#include "config.h"
#include "transcription/cpureferencemodel.h"
#include "transcription/modelpackage.h"
#include "transcription/transcriptionengine.h"
#include "transcription/transcriptiontypes.h"
 
#include <memory>
 
class CpuReferenceTranscriber 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);
 
    CpuReferenceTranscriber(TranscriberConfig config, std::shared_ptr<const TranscriptionModelHandle> model);
 
    ~CpuReferenceTranscriber() override;
 
    CpuReferenceTranscriber(const CpuReferenceTranscriber &) = delete;
    CpuReferenceTranscriber &operator=(const CpuReferenceTranscriber &) = delete;
    CpuReferenceTranscriber(CpuReferenceTranscriber &&) = delete;
    CpuReferenceTranscriber &operator=(CpuReferenceTranscriber &&) = 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:
    TranscriberConfig m_config;
    std::shared_ptr<const CpuReferenceModelHandle> m_model;
    std::vector<float> m_bufferedSamples;
    bool m_warmedUp = false;
    bool m_cancelRequested = false;
};
 
[[nodiscard]] std::shared_ptr<const TranscriptionEngine> createCpuReferenceTranscriptionEngine(const TranscriberConfig &config);