DepthLivenessEstimator.h

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

#ifndef __PBIO_API__PBIO__DEPTH_LIVENESS_ESTIMATOR_H_
#define __PBIO_API__PBIO__DEPTH_LIVENESS_ESTIMATOR_H_

#include <sstream>
#include <vector>


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


namespace pbio
{

class FacerecService;

class DepthLivenessEstimator : public ComplexObject
{
public:

    typedef LightSmartPtr<DepthLivenessEstimator>::tPtr Ptr;

    enum Liveness
    {
        NOT_ENOUGH_DATA = 0, 

        REAL = 1,

        FAKE = 2,

        NOT_COMPUTED    = 3
    };


    Liveness estimateLiveness(
        const pbio::RawSample& sample,
        const pbio::DepthMapRaw depth_map);

private:

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

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

}  // pbio namespace




namespace pbio
{


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


inline
DepthLivenessEstimator::Liveness DepthLivenessEstimator::estimateLiveness(
    const pbio::RawSample& sample,
    const pbio::DepthMapRaw depth_map)
{
    void* exception = NULL;

    const int result = _dll_handle->DepthLivenessEstimator_estimateLiveness(
        _impl,
        (pbio::facerec::RawSampleImpl const*) sample._impl,
        depth_map.depth_map_rows,
        depth_map.depth_map_cols,
        depth_map.depth_map_2_image_offset_y,
        depth_map.depth_map_2_image_offset_x,
        depth_map.depth_map_2_image_scale_x,
        depth_map.depth_map_2_image_scale_y,
        depth_map.horizontal_fov,
        depth_map.vertical_fov,
        depth_map.depth_unit_in_millimeters,
        depth_map.depth_data,
        depth_map.depth_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__DEPTH_LIVENESS_ESTIMATOR_H_