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 fsdk::Result<fsdk::FSDKError> estimate(
            const fsdk::Image& warp,
            const fsdk::Landmarks5& landmarks5Transformed,
            fsdk::GazeEstimation& eyeAngles) const noexcept = 0;

        virtual fsdk::Result<fsdk::FSDKError> estimate(
            const fsdk::Span<const fsdk::Image> warps,
            const fsdk::Span<const fsdk::Landmarks5> landmarks5Transformed,
            fsdk::Span<fsdk::GazeEstimation> eyeAngles) const noexcept = 0;

        virtual fsdk::Point2i getFaceCenter(const fsdk::Landmarks5& landmarks5) const noexcept = 0;
    };

} // namespace fsdk