FaceAttributesEstimator.h

Go to the documentation of this file.

#ifndef __PBIO_API__PBIO__FACE_ATTRIBUTES_ESTIMATOR_H_
#define __PBIO_API__PBIO__FACE_ATTRIBUTES_ESTIMATOR_H_


#include "ComplexObject.h"
#include "RawSample.h"
#include "SmartPtr.h"


namespace pbio
{

class FacerecService;

class FaceAttributesEstimator : public ComplexObject
{
public:

    typedef LightSmartPtr<FaceAttributesEstimator>::tPtr Ptr;


    struct EyeStateScore{

        enum EyeState{
            NOT_COMPUTED = 0,
            CLOSED = 1,
            OPENED = 2
        } eye_state;

        float score;

        EyeStateScore() : eye_state( NOT_COMPUTED ), score( -1.f ) {}
    };


    struct Attribute
    {
        bool verdict;

        float score;

        enum MaskAttribute{
            NOT_COMPUTED = 0,
            NO_MASK = 1,
            HAS_MASK = 2
        } mask_attribute;

        Attribute() : verdict( false ), score( -1.0f ), mask_attribute( NOT_COMPUTED ) {}

        EyeStateScore left_eye_state, right_eye_state;
    };

    Attribute estimate(const pbio::RawSample &sample) const;

private:

    FaceAttributesEstimator(
        const DHPtr &dll_handle,
        void* impl);

    friend class FacerecService;
    friend class object_with_ref_counter<FaceAttributesEstimator>;
};

}  // pbio namespace




namespace pbio
{


inline
FaceAttributesEstimator::FaceAttributesEstimator(
    const DHPtr &dll_handle,
    void* impl):
ComplexObject(dll_handle, impl)
{
    // nothing else
}


inline
FaceAttributesEstimator::Attribute FaceAttributesEstimator::estimate(const pbio::RawSample &sample) const
{
    Attribute result;

    std::ostringstream name_task;
    pbio::stl_wraps::WrapOStreamImpl name_task_wrap(name_task);

    void* exception = NULL;

    _dll_handle->FaceAttributesEstimator_getTaskName(
        _impl,
        &name_task_wrap,
        pbio::stl_wraps::WrapOStream::write_func,
        &exception);

    checkException(exception, *_dll_handle);

    if (name_task.str() == "masked_face" || name_task.str() == "masked_face_v2")
    {
        int32_t verdict;

        void* exception = NULL;

        _dll_handle->FaceAttributesEstimator_estimateMaskedFace(
            _impl,
            (pbio::facerec::RawSampleImpl const*) sample._impl,
            &verdict,
            &result.score,
            &exception);

        checkException(exception, *_dll_handle);

        result.verdict = verdict;

        result.mask_attribute = verdict ? Attribute::MaskAttribute::HAS_MASK: Attribute::MaskAttribute::NO_MASK;
    }else
    if (name_task.str() == "eyes_openness" || name_task.str() == "eyes_openness_v2" )
    {
        EyeStateScore left_eye_state, right_eye_state;
        int32_t left_eye_verdict, right_eye_verdict;

        void* exception = NULL;

        _dll_handle->FaceAttributesEstimator_estimateEyesOpenness(
            _impl,
            (pbio::facerec::RawSampleImpl const*) sample._impl,
            &left_eye_verdict,
            &right_eye_verdict,
            &result.left_eye_state.score,
            &result.right_eye_state.score,
            &exception);

        checkException(exception, *_dll_handle);

        if (result.left_eye_state.score == -1.f)
            result.left_eye_state.eye_state = EyeStateScore::NOT_COMPUTED;
        else
            result.left_eye_state.eye_state = left_eye_verdict ? EyeStateScore::OPENED: EyeStateScore::CLOSED;

        if (result.right_eye_state.score == -1.f)
            result.right_eye_state.eye_state = EyeStateScore::NOT_COMPUTED;
        else
            result.right_eye_state.eye_state = right_eye_verdict ? EyeStateScore::OPENED: EyeStateScore::CLOSED;
    }else
        PBI0x3dfb4fe3Assert(0xec9eb983, false, "FaceAttributesEstimator: unknown name_task: '" + name_task.str() + "'");


    return result;
}


}  // pbio namespace

#endif  // __PBIO_API__PBIO__FACE_ATTRIBUTES_ESTIMATOR_H_