IMouthEstimator.h

#pragma once
 
#include <fsdk/IObject.h>
#include <fsdk/FSDKError.h>
#include <fsdk/Optional.h>
#include <fsdk/Types.h>
 
namespace fsdk {
 
#ifndef DOXYGEN_SHOULD_SKIP_THIS
    DECLARE_SMARTPTR(IMouthEstimator);
#endif
 
    enum class SmileType {
        None,      
        SmileLips, 
        SmileOpen  
    };
 
    struct SmileTypeScores {
        float smileLips; 
        float smileOpen; 
    };
 
    struct MouthEstimation {
        float opened;    
        float smile;     
        float occluded;  
        bool isOpened;   
        bool isSmiling;  
        bool isOccluded; 
    };
 
    struct MouthEstimationExtended {
        float opened;    
        float smile;     
        float occluded;  
        bool isOpened;   
        bool isSmiling;  
        bool isOccluded; 
 
        SmileTypeScores smileTypeScores; 
        SmileType smileType;             
    };
 
    struct IMouthEstimator : IRefCounted {
        virtual Result<FSDKError> estimate(const Image& warp, MouthEstimation& estimation) const noexcept = 0;
 
        virtual Result<FSDKError>
        estimate(Span<const Image> warps, Span<MouthEstimation> mouthEstimations) const noexcept = 0;
 
        virtual Result<FSDKError>
        estimate(const Image& warp, MouthEstimationExtended& estimation) const noexcept = 0;
 
        virtual Result<FSDKError>
        estimate(Span<const Image> warps, Span<MouthEstimationExtended> mouthEstimations) const noexcept = 0;
 
        virtual Result<FSDKError>
        validate(Span<const Image> warps, Span<Result<FSDKError>> errors) const noexcept = 0;
    };
 
} // namespace fsdk