IEyeEstimator.h

#pragma once
 
#include <fsdk/IObject.h>
#include <fsdk/FSDKError.h>
#include <fsdk/Optional.h>
#include <fsdk/Types.h>
 
namespace fsdk {
 
#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IEyeEstimator);
#endif
 
    struct EyesEstimation {
        struct EyeAttributes {
            enum class State : uint8_t {
                Closed,  
                Open,    
                Occluded 
            };
 
            static constexpr uint64_t irisLandmarksCount = 32;  
            static constexpr uint64_t eyelidLandmarksCount = 6; 
 
            using IrisLandmarks = Landmarks<irisLandmarksCount>;
 
            using EyelidLandmarks = Landmarks<eyelidLandmarksCount>;
 
            State state; 
 
            IrisLandmarks iris;     
            EyelidLandmarks eyelid; 
        };
 
        EyeAttributes leftEye;  
        EyeAttributes rightEye; 
    };
 
    struct FSDK_API EyeCropper {
        struct EyesRects {
            Rect leftEyeRect;
            Rect rightEyeRect;
        };
 
        EyesRects cropByLandmarks5(const Image& warp, const Landmarks5& landmarks5);
 
        EyeCropper::EyesRects cropByLandmarks68(const Image& warp, const Landmarks68& landmarks68);
    };
 
    struct IEyeEstimator : IRefCounted {
        virtual Result<FSDKError> estimate(
            const Image& warp,
            const EyeCropper::EyesRects& eyeRects,
            EyesEstimation& eyesEstimation) const noexcept = 0;
 
        virtual Result<FSDKError> estimate(
            Span<const Image> warps,
            Span<const EyeCropper::EyesRects> eyeRects,
            Span<EyesEstimation> eyesEstimations) const noexcept = 0;
 
        virtual Result<FSDKError> validate(
            Span<const Image> warps,
            Span<const EyeCropper::EyesRects> eyeRects,
            Span<Result<FSDKError>> errors) const noexcept = 0;
    };
 
} // namespace fsdk