ICrowdEstimator.h

#pragma once

#include <fsdk/Def.h>
#include <fsdk/FSDKError.h>
#include <fsdk/IRefCounted.h>
#include <fsdk/Types/Image.h>
#include <fsdk/Types/Ref.h>
#include <fsdk/Types/Result.h>
#include <fsdk/Types/ResultValue.h>
#include <fsdk/Types/Span.h>
#include <fsdk/IPointBatch.h>

namespace fsdk {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(ICrowdEstimator);
#endif

    enum CrowdEstimatorType {
        CET_DEFAULT = 0,    
        CET_SINGLE_NET = 1, 
        CET_TWO_NETS = 2,   
        CET_COUNT
    };

    struct CrowdEstimation {
        size_t count;          
        IPointBatchPtr points; 
    };

    struct ICrowdEstimator : IRefCounted {
        enum EstimationRequest {
            estimateHeadCount = 0,               
            estimateHeadCountAndCoords = 1 << 0, 
        };

        virtual Result<FSDKError> estimate(
            const Image& image,
            const Rect& rect,
            CrowdEstimation& estimation,
            EstimationRequest request = EstimationRequest::estimateHeadCount) const noexcept = 0;

        virtual Result<FSDKError> estimate(
            Span<const Image> images,
            Span<const Rect> rects,
            Span<CrowdEstimation> estimations,
            EstimationRequest request = EstimationRequest::estimateHeadCount) const noexcept = 0;

        virtual Result<FSDKError> validate(
            Span<const Image> images,
            Span<const Rect> rects,
            Span<Result<FSDKError>> errors,
            EstimationRequest request = EstimationRequest::estimateHeadCount) const noexcept = 0;

        virtual CrowdEstimatorType getType() const noexcept = 0;
    };

    inline ICrowdEstimator::EstimationRequest
    operator|(ICrowdEstimator::EstimationRequest a, ICrowdEstimator::EstimationRequest b) {
        return static_cast<ICrowdEstimator::EstimationRequest>(static_cast<int>(a) | static_cast<int>(b));
    }

} // namespace fsdk