IRLivenessEstimator.h

Go to the documentation of this file.

#ifndef __PBIO_API__PBIO__IR_LIVENESS_ESTIMATOR_H_
#define __PBIO_API__PBIO__IR_LIVENESS_ESTIMATOR_H_

#include <sstream>
#include <vector>


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


namespace pbio
{

class FacerecService;

class IRLivenessEstimator : public ComplexObject
{
public:

    typedef LightSmartPtr<IRLivenessEstimator>::tPtr Ptr;

    enum Liveness
    {
        NOT_ENOUGH_DATA = 0, 

        REAL = 1,

        FAKE = 2,

        NOT_COMPUTED    = 3
    };


    Liveness estimateLiveness(
        const pbio::RawSample& sample,
        const pbio::IRFrameRaw ir_frame);

private:

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

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

}  // pbio namespace




namespace pbio
{


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


inline
IRLivenessEstimator::Liveness IRLivenessEstimator::estimateLiveness(
    const pbio::RawSample& sample,
    const pbio::IRFrameRaw ir_frame)
{
    void* exception = NULL;

    const int result = _dll_handle->IRLivenessEstimator_estimateLiveness(
        _impl,
        (pbio::facerec::RawSampleImpl const*) sample._impl,
        ir_frame.ir_frame_rows,
        ir_frame.ir_frame_cols,
        ir_frame.ir_frame_2_image_offset_x,
        ir_frame.ir_frame_2_image_offset_y,
        ir_frame.ir_frame_2_image_scale_x,
        ir_frame.ir_frame_2_image_scale_y,
        ir_frame.ir_horizontal_fov,
        ir_frame.ir_vertical_fov,
        ir_frame.ir_frame_data,
        ir_frame.ir_data_stride_in_bytes,
        &exception);

    checkException(exception, *_dll_handle);

    if(result == 1)
        return REAL;

    if(result == 2)
        return FAKE;

    return NOT_ENOUGH_DATA;
}


}  // pbio namespace

#endif  // __PBIO_API__PBIO__IR_LIVENESS_ESTIMATOR_H_