verify_stream.h

Go to the documentation of this file.

/* 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;
    };
}