IBestShotMobile.h

#pragma once

#include <bsmobile/BestShotMobileDefs.h>

#include <fsdk/FaceEngine.h>
#include <tsdk/ITrackEngine.h>

namespace mobile {

    enum class LivenessType {
        None = 0,       
        Online = 1,     
        Offline = 2,    
        Offline_OSL = 3 
    };

    enum class LivenessState {
        Alive,             
        Fake,              
        None,              
        NotReady,          
        BadHeadPose,       
        BadQuality,        
        FaceNotFound,      
        FaceTooClose,      
        FaceCloseToBorder, 
        FaceTooSmall,      
        TooManyFaces,      
        Timeout,           
        CriticalError      
    };

    enum class BestShotState {
        Ok,          
        BadQuality,  
        BadHeadPose, 
        Error        
    };

    struct BestShotInfo {
        BestShotState state{};

        fsdk::Image image;
        fsdk::Detection detection;
        fsdk::Landmarks5 landmarks;

        fsdk::HeadPoseEstimation headPoseEstimation{};
        float agsEstimation{};

        tsdk::FrameId frameIndex{};
        tsdk::TrackId trackId{};
    };

    struct IBestShotMobileObserver {
        BESTSHOT_MOBILE_API virtual void bestShot(const BestShotInfo& bestShotInfo) = 0;

        BESTSHOT_MOBILE_API virtual void
        liveness(LivenessState livenessState, const BestShotInfo& bestShotInfo) = 0;

        BESTSHOT_MOBILE_API virtual void trackEnd(const tsdk::TrackId& trackId) = 0;
    };

    struct IBestShotMobile {

        virtual ~IBestShotMobile() = default;

        BESTSHOT_MOBILE_API virtual bool pushFrame(const fsdk::Image& frame, uint32_t frameId) = 0;

        BESTSHOT_MOBILE_API virtual void setBestShotMobileObserver(IBestShotMobileObserver* observer) = 0;

        BESTSHOT_MOBILE_API virtual void join() = 0;

        BESTSHOT_MOBILE_API virtual LivenessType getLivenessType() = 0;
    };

    IBestShotMobile* createBestShotMobile(
        fsdk::FaceEngineType* faceEngine,
        tsdk::ITrackEngine* trackEngine,
        fsdk::ISettingsProvider* settings);

    IBestShotMobile* createBestShotMobile(
        fsdk::FaceEngineType* faceEngine,
        tsdk::ITrackEngine* trackEngine,
        const char* configPath);

} // namespace mobile