package com.dahua.netsdk.module;
|
|
import com.dahua.netsdk.lib.NetSDKLib;
|
import com.dahua.netsdk.lib.enumeration.EM_NEEDED_PIC_RETURN_TYPE;
|
import com.dahua.netsdk.lib.enumeration.EM_OPERATE_FACERECONGNITIONDB_TYPE;
|
import com.dahua.netsdk.lib.enumeration.ENUMERROR;
|
import com.dahua.netsdk.module.entity.PictureInfo;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* @author 47081
|
* @version 1.0
|
* @description 目标识别二次封装模块，一些人脸相关的接口封装
|
* @date 2021/4/26
|
*/
|
public class FaceRecognitionModule extends BaseModule {
|
/**
|
* @param loginHandler 登录句柄
|
* @param personInfoEx 人员信息
|
* @param picture 图片数据
|
* @return
|
*/
|
public boolean addFaceRecognitionDB(
|
long loginHandler, NetSDKLib.FACERECOGNITION_PERSON_INFOEX personInfoEx, byte[] picture) {
|
NetSDKLib.NET_IN_OPERATE_FACERECONGNITIONDB inParam =
|
new NetSDKLib.NET_IN_OPERATE_FACERECONGNITIONDB();
|
if (personInfoEx == null) {
|
System.out.println("人员信息不能为空");
|
return false;
|
}
|
inParam.emOperateType = EM_OPERATE_FACERECONGNITIONDB_TYPE.NET_FACERECONGNITIONDB_ADD.ordinal();
|
// 使用人员拓展信息结构体
|
inParam.bUsePersonInfoEx = 1;
|
inParam.stPersonInfoEx = personInfoEx;
|
if (picture == null || picture.length == 0) {
|
System.out.println("图片信息为空,下发可能失败");
|
} else {
|
PictureInfo info = PictureInfo.generate(picture);
|
if (info.getLength() == 0) {
|
System.out.println("图片信息异常,请检查后重试");
|
return false;
|
}
|
// 写入图片信息
|
inParam.stPersonInfoEx.wFacePicNum = 1;
|
inParam.stPersonInfoEx.szFacePicInfo[0].dwFileLenth = info.getLength();
|
inParam.stPersonInfoEx.szFacePicInfo[0].wHeight = (short) info.getHeight();
|
inParam.stPersonInfoEx.szFacePicInfo[0].wWidth = (short) info.getWidth();
|
// 图片数据
|
inParam.pBuffer = info.getMemory();
|
inParam.nBufferLen = info.getLength();
|
}
|
NetSDKLib.NET_OUT_OPERATE_FACERECONGNITIONDB outParam =
|
new NetSDKLib.NET_OUT_OPERATE_FACERECONGNITIONDB();
|
inParam.write();
|
outParam.write();
|
boolean result =
|
getNetsdkApi()
|
.CLIENT_OperateFaceRecognitionDB(
|
new NetSDKLib.LLong(loginHandler), inParam, outParam, 3000);
|
if (!result) {
|
System.out.println(ENUMERROR.getErrorMessage());
|
}
|
return result;
|
}
|
public long startFindFaceRecognition(long loginHandler, NetSDKLib.FACERECOGNITION_PERSON_INFOEX personInfo,NetSDKLib.NET_FACE_MATCH_OPTIONS faceMatch){
|
NetSDKLib.NET_IN_STARTFIND_FACERECONGNITION inParam=new NetSDKLib.NET_IN_STARTFIND_FACERECONGNITION();
|
inParam.bPersonExEnable=1;
|
return 0L;
|
}
|
/**
|
* 查找人员信息
|
*
|
* @param findHandler 查询句柄,{@link NetSDKLib.NET_OUT_STARTFIND_FACERECONGNITION#lFindHandle}
|
* @param begin 起始查询的条数
|
* @param findNum 本次查询的条数
|
* @param returnType 返回的图片的类型
|
*/
|
public List<NetSDKLib.CANDIDATE_INFOEX> findFaceRecognition(
|
long findHandler, int begin, int findNum, EM_NEEDED_PIC_RETURN_TYPE returnType) {
|
NetSDKLib.NET_IN_DOFIND_FACERECONGNITION inParam =
|
new NetSDKLib.NET_IN_DOFIND_FACERECONGNITION();
|
inParam.lFindHandle = new NetSDKLib.LLong(findHandler);
|
inParam.nBeginNum = begin;
|
inParam.nCount = findNum;
|
|
NetSDKLib.NET_OUT_DOFIND_FACERECONGNITION outParam =
|
new NetSDKLib.NET_OUT_DOFIND_FACERECONGNITION();
|
outParam.bUseCandidatesEx = 1;
|
inParam.write();
|
outParam.write();
|
if (!getNetsdkApi().CLIENT_DoFindFaceRecognition(inParam, outParam, 3000)) {
|
System.out.println("查询失败" + ENUMERROR.getErrorMessage());
|
return new ArrayList<>();
|
}
|
List<NetSDKLib.CANDIDATE_INFOEX> infos = new ArrayList<>();
|
for (int i = 0; i < outParam.nCadidateExNum; i++) {
|
infos.add(outParam.stuCandidatesEx[i]);
|
}
|
return infos;
|
}
|
}
|
