IBestShotQualityEstimator.h

#pragma once
#include <fsdk/Estimators/IHeadPoseEstimator.h>

#include <fsdk/IObject.h>
#include <fsdk/FSDKError.h>
#include <fsdk/Optional.h>
#include <fsdk/Types.h>
#include <fsdk/vlc/future.h>
#include <vector>

namespace fsdk {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IBestShotQualityEstimator);
#endif

    struct IBestShotQualityEstimator : IRefCounted {
        enum EstimationRequest {
            estimateAGS         = 1 << 0, 
            estimateHeadPose    = 1 << 1, 
            estimateAll         = 0xffff  
        };

        struct EstimationResult {
            Optional<HeadPoseEstimation> headPose;  
            Optional<float> ags;                    
        };

        virtual Result<FSDKError> estimate(
            const Image& image,
            const Detection& detection,
            const EstimationRequest request,
            EstimationResult& result) const noexcept = 0;

        virtual Result<FSDKError> estimate(
            Span<const Image> images,
            Span<const Detection> detections,
            const EstimationRequest request,
            Span<EstimationResult> results) const noexcept = 0;

        using EstimationBatch = std::vector<EstimationResult>;
        using EstimationBatchFuture = vlc::future<EstimationBatch>;
        virtual EstimationBatchFuture estimateAsync(
            Span<const Image> images,
            Span<const Detection> detections,
            const EstimationRequest request) const = 0;

        virtual Result<FSDKError>
        validate(
            Span<const Image> images,
            Span<const Detection> detections,
            const EstimationRequest request,
            Span<Result<FSDKError>> errors) const noexcept = 0;
    };

    /*
        Implementation details.
    */
    inline IBestShotQualityEstimator::EstimationRequest
    operator|(
        IBestShotQualityEstimator::EstimationRequest first,
        IBestShotQualityEstimator::EstimationRequest second
    ) {
        return static_cast<IBestShotQualityEstimator::EstimationRequest>(
            static_cast<int>(first) | static_cast<int>(second)
        );
    }
} // namespace fsdk