package com.netsdk.demo.customize.faceReconEx;
|
|
import com.netsdk.demo.util.QueueGeneration;
|
import com.netsdk.lib.NetSDKLib;
|
import com.sun.jna.Pointer;
|
|
import java.io.File;
|
import java.io.UnsupportedEncodingException;
|
import java.util.UUID;
|
import java.util.concurrent.LinkedBlockingDeque;
|
|
import static com.netsdk.demo.util.StructFieldChooser.GetSelectedSingleFieldValue;
|
import static com.netsdk.lib.Utils.getOsPrefix;
|
|
/**
|
* @author ： 47040
|
* @since ： Created in 2020/7/20 20:10
|
*/
|
public class FaceReconAnalyzerDataCallBack implements NetSDKLib.fAnalyzerDataCallBack {
|
|
//////////////////////////////// 单例////////////////////////////////
|
|
private static FaceReconAnalyzerDataCallBack singleInstance;
|
|
public static FaceReconAnalyzerDataCallBack getSingleInstance() {
|
if (singleInstance == null) {
|
singleInstance = new FaceReconAnalyzerDataCallBack();
|
}
|
return singleInstance;
|
}
|
|
//////////////////////////////// 系统编码 ////////////////////////////////
|
|
public static String encode;
|
|
static {
|
String osPrefix = getOsPrefix();
|
if (osPrefix.toLowerCase().startsWith("win32-amd64")) {
|
encode = "GBK";
|
} else if (osPrefix.toLowerCase().startsWith("linux-amd64")) {
|
encode = "UTF-8";
|
}
|
}
|
|
//////////////////////////////// 静态池 ////////////////////////////////
|
// 用阻塞队列模拟一个静态对象池
|
|
// 设置一个队列模拟静态池，容量看情况改，越大可同时处理的事件就越多，占用内存也越多
|
private final static int MAX_TASK_COUNT = 10; // 队列容量
|
private final static LinkedBlockingDeque<NetSDKLib.DEV_EVENT_FACERECOGNITION_INFO> faceReconPool = new LinkedBlockingDeque<>(MAX_TASK_COUNT);
|
|
static {
|
// 初始化队列
|
for (int i = 0; i < MAX_TASK_COUNT; i++) {
|
faceReconPool.offer(new NetSDKLib.DEV_EVENT_FACERECOGNITION_INFO());
|
}
|
}
|
|
//////////////////////////////// 图片文件夹 ////////////////////////////////
|
|
private static final String imageSaveFolder = "faceReconEx/"; // 图片保存路径
|
|
static {
|
// 创建图片文件夹
|
File path = new File(imageSaveFolder);
|
if (!path.exists()) {
|
if (!path.mkdir()) {
|
System.err.println("文件夹创建失败，无法保存图片");
|
}
|
}
|
}
|
|
//////////////////////////////// 图片文件夹 ////////////////////////////////
|
// 两个线程池阻塞队列，用于模拟并发业务
|
|
private final QueueGeneration savePicService = new QueueGeneration(); // 保存图片异步队列
|
|
private final QueueGeneration customerService = new QueueGeneration(); // 业务操作异步队列
|
|
|
public FaceReconAnalyzerDataCallBack() {
|
savePicService.init(); // 图片任务队列启动
|
customerService.init(); // 业务任务队列启动
|
}
|
|
/**
|
* 智能分析回调函数
|
* 请特别注意: 此回调函数仅用于获取事件数据，如果要调其他 sdk 接口或执行相对耗时的业务操作请另开线程，务必不要阻塞本函数
|
* 在本函数内直接调其他 sdk 接口可能会引发底层死锁；长时间阻塞本函数会导致 sdk 心跳包无法正常发送造成设备掉线
|
*
|
* @param lAnalyzerHandle 订阅句柄，必然和订阅时获取的句柄相同
|
* @param dwAlarmType 事件类型枚举
|
* @param pAlarmInfo 事件数据指针
|
* @param pBuffer 图片数据指针
|
* @param dwBufSize 图片数据长度
|
* @param dwUser 用户自定义信息 不建议使用
|
* @param nSequence 表示上传的相同图片情况，0表示是第一次出现，为2表示最后一次出现或仅出现一次，为1表示此次之后还有
|
* @param reserved 表示当前回调数据的状态, 为0表示当前数据为实时数据，为1表示当前回调数据是离线数据，为2时表示离线数据传送结束
|
* @return 0 正常返回
|
*/
|
@Override
|
public int invoke(NetSDKLib.LLong lAnalyzerHandle, int dwAlarmType, Pointer pAlarmInfo, Pointer pBuffer, int dwBufSize, Pointer dwUser, int nSequence, Pointer reserved) {
|
switch (dwAlarmType) {
|
case NetSDKLib.EVENT_IVS_FACERECOGNITION: {
|
handleTargetRecognition(pAlarmInfo, pBuffer, dwBufSize, lAnalyzerHandle);
|
break;
|
}
|
default:
|
System.out.printf("Get Other Event 0x%x\n", dwAlarmType);
|
break;
|
}
|
return 0;
|
}
|
|
/**
|
* 人像抓拍
|
*
|
* @param pBuffer 指针内容
|
* @param dwBufSize 指针内容大小
|
* @param lAnalyzerHandle 布防返回结果
|
*/
|
public void handleTargetRecognition(Pointer pAlarmInfo, Pointer pBuffer, int dwBufSize, NetSDKLib.LLong lAnalyzerHandle) {
|
if (pAlarmInfo == Pointer.NULL || pBuffer == Pointer.NULL || dwBufSize <= 0) {
|
return;
|
}
|
|
NetSDKLib.DEV_EVENT_FACERECOGNITION_INFO msg = null;
|
NetSDKLib.NET_TIME_EX UTC = null; // 抓拍时间
|
int channel = -1; // 事件通道
|
NetSDKLib.NET_PIC_INFO facePicInfo = null; // 人像数据
|
byte[] serialUUID = null; // 唯一记录ID
|
NetSDKLib.NET_PIC_INFO backgroundPic = null; // 大图数据
|
NetSDKLib.NET_FACE_DATA faceInfo = null; // 人脸数据
|
NetSDKLib.CANDIDATE_INFOEX suitableCandidateInfo = null; // 最佳候选人
|
|
//////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////// 从指针获取数据 //////////////////////////////////
|
//////////////////////////////////////////////////////////////////////////////////
|
|
try {
|
msg = faceReconPool.take(); // 从静态池取
|
// 抓拍时间
|
UTC = (NetSDKLib.NET_TIME_EX) GetSelectedSingleFieldValue("UTC", msg, pAlarmInfo);
|
// 通道号
|
channel = (int) GetSelectedSingleFieldValue("nChannelID", msg, pAlarmInfo);
|
// 唯一记录ID
|
serialUUID = (byte[]) GetSelectedSingleFieldValue("szSerialUUID", msg, pAlarmInfo);
|
// 人像数据
|
facePicInfo = (NetSDKLib.NET_PIC_INFO) GetSelectedSingleFieldValue("stuObject.stPicInfo", msg, pAlarmInfo);
|
// 大图数据
|
backgroundPic = (NetSDKLib.NET_PIC_INFO) GetSelectedSingleFieldValue("stuGlobalScenePicInfo", msg, pAlarmInfo);
|
// 人脸数据
|
faceInfo = (NetSDKLib.NET_FACE_DATA) GetSelectedSingleFieldValue("stuFaceData", msg, pAlarmInfo);
|
// 最佳候选人数据
|
int nCandidateNum = (int) GetSelectedSingleFieldValue("nRetCandidatesExNum", msg, pAlarmInfo); // 读取有效候选人数量
|
if (nCandidateNum != 0) {
|
for (int i = 0; i < nCandidateNum; i++) { // 长度不为0，证明存在候选人
|
msg.stuCandidatesEx[i] = (NetSDKLib.CANDIDATE_INFOEX) GetSelectedSingleFieldValue(String.format("stuCandidatesEx[%d]", i), msg, pAlarmInfo);
|
}
|
// 遍历找出相似度最大的那个人
|
int index = 0;
|
int bySimilarity = 0;
|
for (int j = 0; j < nCandidateNum; j++) {
|
if (msg.stuCandidatesEx[j].bySimilarity > bySimilarity) {
|
index = j;
|
bySimilarity = msg.stuCandidatesEx[j].bySimilarity;
|
}
|
}
|
suitableCandidateInfo = msg.stuCandidatesEx[index];
|
}
|
} catch (InterruptedException e) {
|
System.err.println("静态池错误: " + e.getLocalizedMessage());
|
} catch (Exception e) {
|
System.err.println("数据获取异常：" + e.getLocalizedMessage());
|
} finally {
|
if (msg != null) {
|
faceReconPool.offer(msg); // 重新放回静态池
|
}
|
}
|
|
//////////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////// 提取数据及业务操作 //////////////////////////////////
|
/////////////////////////////////////////////////////////////////////////////////////
|
|
try {
|
// 订阅句柄
|
Long analyzerHandle = lAnalyzerHandle.longValue();
|
// 抓拍时间
|
String eventTime = UTC == null ? "" : UTC.toStringTime();
|
|
String uuid = UUID.randomUUID().toString();
|
// --> 人像数据 人脸图片信息(如果需要其他数据请从 facePicInfo 内获取)
|
String facePicPath = "";
|
if (facePicInfo != null && facePicInfo.dwFileLenth > 0) {
|
facePicPath = imageSaveFolder + eventTime.trim().replaceAll("[^0-9]", "-") + "-" + uuid + "-facePic.jpg";
|
savePicService.addEvent(new SavePicHandler(pBuffer, facePicInfo.dwOffSet, facePicInfo.dwFileLenth, facePicPath)); // 新开线程保存图片
|
}
|
|
// --> 大图数据 操作背景图片(如果需要其他数据请从 backgroundPic 内获取)
|
String backPicPath = "";
|
if (backgroundPic != null && backgroundPic.dwFileLenth > 0) {
|
backPicPath = imageSaveFolder + eventTime.trim().replaceAll("[^0-9]", "-") + "-" + uuid + "-scenePic.jpg";
|
savePicService.addEvent(new SavePicHandler(pBuffer, backgroundPic.dwOffSet, backgroundPic.dwFileLenth, backPicPath)); // 新开线程保存图片
|
}
|
|
// --> 人脸数据 (如果需要其他数据请从 faceInfo 内获取)
|
int emSex = 0; // 性别枚举
|
int emMask = 0; // 口罩枚举
|
int age = 0; // 年龄
|
String emGlasses = ""; // 是否带眼镜
|
if (faceInfo != null) {
|
emSex = faceInfo.emSex;
|
emMask = faceInfo.emMask;
|
age = faceInfo.nAge;
|
for (int i = 0; i < faceInfo.nFeatureValidNum; i++) {
|
if (faceInfo.emFeature[i] == 1) {
|
emGlasses = "戴眼镜";
|
} else if (faceInfo.emFeature[i] == 10) {
|
emGlasses = "没戴眼镜";
|
} else if (faceInfo.emFeature[i] == 14) {
|
emGlasses = "太阳眼镜";
|
}
|
}
|
}
|
|
String uniqueSerialUUID = "";
|
if (serialUUID != null)
|
uniqueSerialUUID = new String(serialUUID, encode).trim();
|
|
// --> 最佳候选人数据 (如果需要其他数据请从 suitableCandidateInfo 内获取)
|
String szUID = ""; // User ID
|
String szID = ""; // User szID
|
String szPersonName = ""; // User name
|
String faceDbId = ""; // DB ID
|
int similarity = 0; // similarity
|
if (suitableCandidateInfo != null) {
|
szUID = new String(suitableCandidateInfo.stPersonInfo.szUID, encode).trim();
|
szID = new String(suitableCandidateInfo.stPersonInfo.szID, encode).trim();
|
szPersonName = new String(suitableCandidateInfo.stPersonInfo.szPersonName, encode).trim();
|
faceDbId = new String(suitableCandidateInfo.stPersonInfo.szGroupID).trim();
|
similarity = suitableCandidateInfo.bySimilarity;
|
|
}
|
|
// 打印看一下
|
System.out.println("\n\n<<—————————— EVENT_IVS ——————————>>\n" +
|
" 订阅句柄 lAnalyzerHandle:" + analyzerHandle + "\n" +
|
" 时间事件 Time : " + eventTime + "\n" +
|
" 通道 Channel : " + channel + "\n" +
|
" 事件识别ID SerialUUID : " + uniqueSerialUUID + "\n" +
|
" 性别 emSex : " + emSex + "\n" +
|
" 年龄 age : " + age + "\n" +
|
" 戴眼镜 emGlasses : " + emGlasses + "\n" +
|
" 戴口罩 emMask : " + emMask + "\n" +
|
" 候选人ID User ID : " + szUID + "\n" +
|
" 候选人证件 User szID : " + szID + "\n" +
|
" 候选人姓名 User name : " + szPersonName + "\n" +
|
" 注册库ID DB ID : " + faceDbId + "\n" +
|
" 相似度 similarity : " + similarity
|
);
|
|
// 耗时的业务操作请另开线程
|
CustomerServiceModel model = new CustomerServiceModel(
|
analyzerHandle,
|
eventTime, channel,
|
facePicPath, backPicPath,
|
emSex, emMask, age, emGlasses,
|
szUID, szID, szPersonName, faceDbId, similarity
|
);
|
customerService.addEvent(new CustomerServiceHandler(model));
|
} catch (
|
UnsupportedEncodingException e) {
|
System.err.println("编码错误: " + e.getLocalizedMessage());
|
}
|
}
|
}
|
