verify_stream.h

/* Copyright 2019 ID R&D Inc. All Rights Reserved. */
 
#pragma once
 
#include <iostream>
#include <memory>
#include <vector>
 
#include <voicesdk/core/config.h>
#include <voicesdk/core/common/intervals.h>
#include <voicesdk/verify/verify.h>
 
namespace voicesdk {
 
    struct VerifyStreamResult {
        const AudioInterval audioInterval;
 
        const VerifyResult verifyResult;
 
        VerifyStreamResult(const AudioInterval& audioInterval_, const VerifyResult& verifyResult_) :
                audioInterval(audioInterval_),
                verifyResult(verifyResult_) {}
 
        friend std::ostream &operator<<(std::ostream& os, const VerifyStreamResult& obj) {
            os << "StreamVerifyResult["
               << "audioInterval: " << obj.audioInterval << ", "
               << "verifyResult:  " << obj.verifyResult  << "]";
            return os;
        }
 
        bool operator==(const VerifyStreamResult& other) const {
            return audioInterval == other.audioInterval && verifyResult == other.verifyResult;
        }
    };
 
 
    class VOICE_SDK_API VoiceVerifyStream {
    public:
        using Ptr = std::shared_ptr<VoiceVerifyStream>;
 
        static Ptr create(
            VoiceTemplateFactory::Ptr voiceTemplateFactory,
            VoiceTemplateMatcher::Ptr voiceTemplateMatcher,
            const std::vector<VoiceTemplate::Ptr>& voiceTemplates,
            size_t sampleRate,
            size_t audioContextLengthSeconds = 10,
            float windowLengthSeconds = 3.0f
        );
 
        virtual void reset() = 0;
 
        virtual void addSamples(const uint8_t *pcm16Bytes, size_t bytesNum) = 0;
 
        virtual void addSamples(const int16_t *pcm16Samples, size_t samplesNum) = 0;
 
        virtual void addSamples(const float *floatSamples, size_t samplesNum) = 0;
 
        virtual void finalize() = 0;
 
        virtual bool hasVerifyResults() const = 0;
 
        virtual std::vector<VerifyStreamResult> getVerifyResult() = 0;
 
        virtual VerifyStreamResult getVerifyResultForOneTemplate() = 0;
 
        virtual ~VoiceVerifyStream() = default;
    };
 
    class VOICE_SDK_API VoiceVerifyStreamOpus {
    public:
        using Ptr = std::shared_ptr<VoiceVerifyStreamOpus>;
 
        static Ptr create(
                VoiceTemplateFactory::Ptr voiceTemplateFactory,
                VoiceTemplateMatcher::Ptr voiceTemplateMatcher,
                VoiceTemplate::Ptr voiceTemplate,
                size_t sampleRate,
                size_t audioContextLengthSeconds = 10
        );
 
        virtual void reset() = 0;
 
        virtual void addPacket(const uint8_t *bytes, size_t bytesNum) = 0;
 
        virtual void finalize() = 0;
 
        virtual bool hasVerifyResults() const = 0;
 
        virtual VerifyStreamResult getVerifyResult() = 0;
 
        virtual ~VoiceVerifyStreamOpus() = default;
    };
}