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

    struct GazeEstimation {
        float yaw;      
        float pitch;    
    };

    struct IGazeEstimator: IRefCounted {
        virtual Result<FSDKError> estimate(
            const Image& warp,
            const Landmarks5& landmarks5Transformed,
            GazeEstimation& eyeAngles) const noexcept = 0;

        virtual Result<FSDKError> estimate(
            Span<const Image> warps,
            Span<const Landmarks5> landmarks5Transformed,
            Span<GazeEstimation> eyeAngles) const noexcept = 0;
        
        virtual Result<FSDKError>
        validate(
            Span<const Image> warps,
            Span<const Landmarks5> landmarks5Transformed,
            Span<Result<FSDKError>> errors) const noexcept = 0;
        
        virtual Point2i getFaceCenter(const Landmarks5& landmarks5) const noexcept = 0;
    };

} // namespace fsdk