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 fsdk::Image& childWarp, 
            ChildEstimation& childEstimation
        ) const noexcept = 0;
        
        virtual Result<FSDKError>
        estimate(
            const fsdk::Image childWarps[],
            const uint32_t batchSize,
            ChildEstimation childEstimations[]) const noexcept = 0;
    };

} // namespace fsdk