IPortraitStyle.h

#pragma once

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

#include <fsdk/Types/Image.h>
#include <fsdk/Types/Result.h>
#include <fsdk/Types/Span.h>

namespace fsdk {

#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IPortraitStyleEstimator);
#endif

    enum class PortraitStyleStatus : uint8_t {
        NonPortrait = 0,       
        Portrait = 1,          
        HiddenShoulders = 2    
    };

    struct PortraitStyleEstimation {
        PortraitStyleStatus status; 
        float nonPortraitScore;             
        float portraitScore;                
        float hiddenShouldersScore;         
    };

    struct IPortraitStyleEstimator : IRefCounted {

        virtual Result<FSDKError>
        estimate(
            const Image& image,
            const Detection& detection,
            PortraitStyleEstimation& estimation) const noexcept = 0;

        virtual Result<FSDKError>
        estimate(
            Span<const Image> images,
            Span<const Detection> detections,
            Span<PortraitStyleEstimation> estimations) const noexcept = 0;

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

} // namespace fsdk