myvtm/Module/mod_livenessdetection/YituLiveSDK.h

#ifndef YITU_LIVE_SDK_H_
#define YITU_LIVE_SDK_H_

#pragma once

namespace YituLiveSDK {

#define FUNCTION_RTN_SUCC       0

#define CAPTURE_TYPE_PASSIVE                    0
#define CAPTURE_TYPE_ACTIVE_WITHOUT_LIVENESS    1
#define CAPTURE_TYPE_ACTIVE_WITH_LIVENESS       2

#define STATUS_HeadLeft  0
#define STATUS_HeadRight  1
#define STATUS_HeadUp  2
#define STATUS_HeadDown  3
#define STATUS_HalfFace  4
#define STATUS_HasSunGlasses  5
#define STATUS_HasMask  6
#define STATUS_HasHat  7
#define STATUS_FrontFace  8
#define STATUS_TrackBreak  9
#define STATUS_Normal  10
#define STATUS_Checking_Live   11  

#define STATUS_NoFace  99

#define LIVENESSTYPE_HACK  0
#define LIVENESSTYPE_LIVE  1
#define LIVENESSTYPE_NOT_SURE  2

	struct Yitu_InitParam
	{
		char* szPath;                       //path for models
		HWND hwnd;                          //hwnd to receive the messages
		UINT uMsgLivenessDone;              //message id for liveness checking done
		UINT uMsgImageInfo;                 //message id for liveness status change
		UINT uMsgEchoCamera;                //message id for camera echo change
	};

	struct Yitu_ImageData
	{
		BITMAP* pImage;  //image data
		int nCameraID;  //identify
		double dTimeStamp;//seconds from 1970.1.1, in ms precision
		int nFrameID;      //frame id, independent for each camera
	};

	int YituInit(const char *pszDLLPath, Yitu_InitParam* param);
	int YituUninit();
	int YituStartCapture(int nType);
	int YituStopCapture();
	int YituAppendFrame(Yitu_ImageData* pData);
	int YituGetImage(BITMAP* pbitmap);
	int YituHasHackInSession(bool* bHasHack);
	int YituGetLiveImageCount(int* nLiveCount);
	int YituGetLiveImage(int nLiveImage,BITMAP* pbitmap,int* nLivenessType);

	//structure and functions for debug
	struct Yitu_TrackData
	{
		int mCameraId;
		int mTrackId;

		int mLastFrameId;
		RECT rectFace;

		double mEyeCloseScore;
		double mEyeOpenScore;

		bool mHasEyeClose;
		bool mHasEyeOpen;
		double mFrontFaceScore;

		double dblFanpaiScore;
		int nFanpaiType;
	};

	int YituGetTrackCount(int nCameraID,int* nCount);
	int YituGetTrackData(int nCameraID,int nIndex,Yitu_TrackData* pData);
}

#endif