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

    struct ChildEstimation {
        float childScore = 0.0f; 
        bool isChild = false;         
    };

    struct IChildEstimator : IRefCounted {
        virtual Result<FSDKError> estimate(
            const Image& childWarp, 
            ChildEstimation& childEstimation
        ) const noexcept = 0;
        
        virtual Result<FSDKError>
        estimate(
            Span<const Image> childWarps,
            Span<ChildEstimation> childEstimations) const noexcept = 0;

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

} // namespace fsdk