include/fsdk/Estimators/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