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