Liveness2DEstimator.h

См. документацию.

#ifndef __PBIO_API__PBIO__LIVENESS_2D_ESTIMATOR_H_
#define __PBIO_API__PBIO__LIVENESS_2D_ESTIMATOR_H_

#include <sstream>
#include <vector>


#include "ComplexObject.h"
#include "Error.h"
#include "RawSample.h"
#include "SmartPtr.h"
#include "stl_wraps_impls/WrapOStreamImpl.h"


namespace pbio
{

class FacerecService;

class Liveness2DEstimator : public ComplexObject
{
public:

    typedef LightSmartPtr<Liveness2DEstimator>::tPtr Ptr;

    enum Liveness
    {
        NOT_ENOUGH_DATA = 0, 

        REAL = 1,

        FAKE = 2,

        NOT_COMPUTED = 3,

        FACE_NOT_FULLY_FRAMED = 4,

        MULTIPLE_FACE_FRAMED = 5,

        FACE_TURNED_RIGHT = 6,

        FACE_TURNED_LEFT = 7,

        FACE_TURNED_UP = 8,

        FACE_TURNED_DOWN = 9,

        BAD_IMAGE_LIGHTING = 10,

        BAD_IMAGE_NOISE = 11,

        BAD_IMAGE_BLUR = 12,

        BAD_IMAGE_FLARE = 13,
    };

    struct LivenessAndScore
    {
        Liveness liveness;

        float score;
    };


    Liveness estimateLiveness(const pbio::RawSample& sample);


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

private:

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

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

}  // pbio namespace




namespace pbio
{


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


inline
Liveness2DEstimator::Liveness Liveness2DEstimator::estimateLiveness(const pbio::RawSample& sample)
{
    void* exception = NULL;
    int32_t verdict;
    float score;

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

    checkException(exception, *_dll_handle);

    return static_cast<Liveness>(verdict);
}


inline
Liveness2DEstimator::LivenessAndScore Liveness2DEstimator::estimate(const pbio::RawSample& sample)
{
    void* exception = NULL;
    int32_t verdict;
    float score;

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

    checkException(exception, *_dll_handle);

    LivenessAndScore result;
    result.liveness = static_cast<Liveness>(verdict);
    result.score = score;

    return result;
}


}  // pbio namespace

#endif  // __PBIO_API__PBIO__LIVENESS_2D_ESTIMATOR_H_