include/fsdk/Estimators/IEthnicityEstimator.h

#pragma once

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

namespace fsdk {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IEthnicityEstimator);
#endif

    struct IEthnicityEstimator : IRefCounted {
        enum class Ethnicities {
            Black = 0,
            Indian,
            Asian,
            White,
            Arab,
            Pamir,
            Kavkaz,
            Other
        };

        struct EstimationResult {
            inline float getEthnicityScore(Ethnicities ethnicity) const;

            Ethnicities predominant;
            float scores[static_cast<int>(Ethnicities::Other)];
        };

        virtual Result<FSDKError> estimate(const Image& warp, EstimationResult& estimation) const noexcept = 0;

        virtual Result<FSDKError>
        estimate(Span<const Image> warps, Span<EstimationResult> estimations) const noexcept = 0;
    };

    float
    IEthnicityEstimator::EstimationResult::getEthnicityScore(IEthnicityEstimator::Ethnicities ethnicity) const {
        if(static_cast<uint32_t>(ethnicity) >= static_cast<uint32_t>(IEthnicityEstimator::Ethnicities::Other)) {
            return 0;
        }
        return scores[static_cast<uint32_t>(ethnicity)];
    }

} // namespace fsdk