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

    struct EyeBrowEstimation {
        enum class EyeBrowState {
            Neutral = 0,
            Raised,
            Squinting,
            Frowning
        };

        float neutralScore;        
        float raisedScore;         
        float squintingScore;      
        float frowningScore;       
        EyeBrowState eyeBrowState; 
    };

    struct IEyeBrowEstimator : IRefCounted {
        virtual Result<FSDKError>
        estimate(
            const Image& warp,
            EyeBrowEstimation& estimation) const noexcept = 0;

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

        virtual Result<FSDKError>
        validate(
            Span<const Image> warps,
            Span<Result<FSDKError>> errors) const noexcept = 0;
    };
} // namespace fsdk