IPPEEstimator.h

#pragma once

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

#include <algorithm>

namespace fsdk {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IPPEEstimator);
#endif
    
    struct OnePPEEstimation {
        float positive = 0.0f;
        float negative = 0.0f;
        float unknown  = 0.0f;

        enum class PPEState : uint8_t {
            Positive, 
            Negative, 
            Unknown,  
            Count     
        };

        inline PPEState getPredominantState();
    };
    using PPEState = OnePPEEstimation::PPEState;

    struct PPEEstimation {
        OnePPEEstimation helmetEstimation;
        OnePPEEstimation hoodEstimation;
        OnePPEEstimation vestEstimation;
        OnePPEEstimation glovesEstimation;
    };

    struct IPPEEstimator : IRefCounted {
        virtual ResultValue<FSDKError, PPEEstimation>
        estimate(
            const Image& image,
            const Detection& detection) const noexcept = 0;
    };

    PPEState OnePPEEstimation::getPredominantState() {
        float* arr = &this->positive;
        return static_cast<PPEState>( 
            std::distance(arr, std::max_element(arr, arr + static_cast<int>(PPEState::Count)))
        );
    }
} // namespace fsdk