IRedEyeEstimator.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(IRedEyeEstimator);
#endif

    enum class RedEyeStatus : uint8_t {
        NonRed, 
        Red,    
    };

    struct RedEyeEstimation {
        struct RedEyeAttributes {
            RedEyeStatus status; 
            float score;         
        };

        RedEyeAttributes leftEye;  
        RedEyeAttributes rightEye; 
    };

    struct IRedEyeEstimator : IRefCounted {
        virtual Result<FSDKError> estimate(
            const Image& warp,
            const Landmarks5& landmarks5Transformed,
            RedEyeEstimation& estimation) const noexcept = 0;

        virtual Result<FSDKError> estimate(
            Span<const Image> warps,
            Span<const Landmarks5> landmarks5Transformed,
            Span<RedEyeEstimation> estimations) const noexcept = 0;

        virtual Result<FSDKError> validate(
            Span<const Image> warps,
            Span<const Landmarks5> landmarks5Transformed,
            Span<Result<FSDKError>> errors) const noexcept = 0;
    };

} // namespace fsdk