include/fsdk/Estimators/IGazeEstimator.h

#pragma once

#include <fsdk/IObject.h>
#include <fsdk/FSDKError.h>
#include <fsdk/Optional.h>
#include <fsdk/Types.h>
#include <fsdk/Types/SensorType.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