IHumanDetector.h

#pragma once

#include <fsdk/FSDKError.h>
#include <fsdk/IHumanDetectionBatch.h>
#include <fsdk/IObject.h>
#include <fsdk/Types/Human.h>

#include <fsdk/vlc/future.h>

namespace fsdk {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IHumanDetector);
#endif

    enum HumanDetectionType {
        HDT_BOX = 0,     
        HDT_ALL = 0xffff 
    };

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

    struct IHumanDetector : IRefCounted {
        virtual ResultValue<FSDKError, IHumanDetectionBatchPtr> detect(
            Span<const Image> images,
            Span<const Rect> rects,
            uint32_t detectionPerImageNum,
            HumanDetectionType type = HDT_BOX) const noexcept = 0;

        virtual ResultValue<FSDKError, Human>
        detectOne(const Image& image, const Rect& rect, HumanDetectionType type = HDT_BOX) const noexcept = 0;

        virtual ResultValue<FSDKError, Human> redetectOne(
            const Image& image,
            const Detection& detection,
            HumanDetectionType type = HDT_BOX) const noexcept = 0;

        virtual ResultValue<FSDKError, IHumanDetectionBatchPtr> redetect(
            Span<const Image> images,
            Ref<IHumanDetectionBatch> detectionBatch,
            HumanDetectionType type = HDT_BOX) const noexcept = 0;

        virtual ResultValue<FSDKError, IHumanDetectionBatchPtr> redetect(
            Span<const Image> images,
            Span<Span<const Detection>> detections,
            HumanDetectionType type = HDT_BOX) const noexcept = 0;

        virtual Result<FSDKError> validate(
            Span<const Image> images,
            Span<const Rect> rects,
            uint32_t detectionPerImageNum,
            Span<Result<FSDKError>> errors) const noexcept = 0;

        virtual Result<FSDKError> validate(
            Span<const Image> images,
            Ref<IHumanDetectionBatch> detectionBatch,
            Span<Result<FSDKError>> errors) const noexcept = 0;

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

        using HumanBatchFuture = vlc::future<IHumanDetectionBatchPtr>;

        virtual HumanBatchFuture detectAsync(
            Span<const Image> images,
            Span<const Rect> rectangles,
            uint32_t detectionPerImageNum,
            HumanDetectionType type = HDT_BOX) const = 0;

        virtual HumanBatchFuture redetectAsync(
            Span<const Image> images,
            Ref<IHumanDetectionBatch> detectionBatch,
            HumanDetectionType type = HDT_BOX) const = 0;

        virtual HumanBatchFuture redetectAsync(
            Span<const Image> images,
            Span<Span<const Detection>> detections,
            HumanDetectionType type = HDT_BOX) const = 0;
    };

} // namespace fsdk