package com.netsdk.demo.customize; import com.netsdk.demo.customize.faceReconEx.DefaultFaceFindStateCallback; import com.netsdk.demo.util.CaseMenu; import com.netsdk.lib.NetSDKLib; import com.netsdk.lib.ToolKits; import com.netsdk.lib.enumeration.EM_PERSON_FEATURE_ERRCODE; import com.netsdk.lib.enumeration.ENUMERROR; import com.netsdk.lib.structure.*; import com.netsdk.lib.utils.Initialization; import com.sun.jna.Memory; import com.sun.jna.Pointer; import com.sun.jna.ptr.IntByReference; import java.awt.*; public class MultiFaceDemo extends Initialization { static NetSDKLib netsdkApi = NetSDKLib.NETSDK_INSTANCE; static NetSDKLib configApi = NetSDKLib.CONFIG_INSTANCE; private static NetSDKLib.LLong findHandle = new NetSDKLib.LLong(0); private static NetSDKLib.LLong attachFaceHandle = new NetSDKLib.LLong(0); private static int nToken; public class MultiFaceDetectState implements NetSDKLib.fMultiFaceDetectState{ @Override public void invoke(NetSDKLib.LLong lAttachHandle, Pointer pstStates, Pointer dwUser) { //todo } } // private final NetSDKLib.fMultiFaceDetectStateEx = new public class IntPoint extends NetSDKLib.SdkStructure { public int value; public IntPoint() { } public IntPoint(int value) { this.value = value; } } /** * 以图搜图 * * @param longHandle CLIENT_StartFindFaceRecognition 接口返回的查询句柄 * @param count 查询的个数 */ public void doFindSearchByPictureEx(NetSDKLib.LLong longHandle, int count) { int doNextCount = 0; // 分页查找数据 NET_IN_DOFIND_FACERECONGNITION_EX stFindIn = new NET_IN_DOFIND_FACERECONGNITION_EX(); stFindIn.lFindHandle = longHandle; stFindIn.nCount = 10; // 当前想查询的记录条数 stFindIn.nBeginNum = 0; // 每次递增 NET_OUT_DOFIND_FACERECONGNITION_EX stFindOut = new NET_OUT_DOFIND_FACERECONGNITION_EX(); stFindOut.nCadidateEx2Num = count; // stFindOut.bUseCandidatesEx = 1; // 是否使用候选对象扩展结构体 NET_CANDIDATE_INFOEX2[] tmp = new NET_CANDIDATE_INFOEX2[count]; for(int i = 0; i < count; i ++) { tmp[i] = new NET_CANDIDATE_INFOEX2(); } stFindOut.nBufferLen = tmp[0].size()*count; stFindOut.pBuffer = new Memory(tmp[0].size() * count); stFindOut.pBuffer.clear(tmp[0].size() * count); for (int i = 0; i < count; i++) { tmp[i].stuCandidatesEx.stPersonInfo.szFacePicInfo[0].nFilePathLen = 256; tmp[i].stuCandidatesEx.stPersonInfo.szFacePicInfo[0].pszFilePath = new Memory(256); } stFindOut.pstuCandidatesEx2 = new Memory(tmp[0].size() * count); // Pointer初始化 stFindOut.pstuCandidatesEx2.clear(tmp[0].size() * count); ToolKits.SetStructArrToPointerData(tmp, stFindOut.pstuCandidatesEx2); // 将数组内存拷贝给 Pointer do { stFindIn.write(); stFindOut.write(); if (netsdkApi.CLIENT_DoFindFaceRecognitionEx(stFindIn.getPointer(), stFindOut.getPointer(), 1000)) { System.out.println( "CLIENT_DoFindFaceRecognitionEx Succeed"); stFindOut.read(); ToolKits.GetPointerDataToStructArr(stFindOut.pstuCandidatesEx2, tmp); System.out.printf("Record Number [%d]\n", stFindOut.nRetCadidateEx2Num); if (stFindOut.nRetCadidateEx2Num == 0) { System.out.println("没有查询到相关数据"); break; } for (int i = 0; i < stFindOut.nRetCadidateEx2Num; i++) { int index = i + doNextCount * count; // 查询的总个数 - 1, 从0开始 // 建模状态 /** 0:未知 1:建模失败,可能是图片不符合要求,需要换图片 2:有可用的特征值 3:正在计算特征值 4:已建模，但算法升级导致数据不可用，需要重新建模 */ int status = tmp[i].stuCandidatesEx.stPersonInfo.emFeatureState; System.out.println("建模状态: " + status); // 如果建模失败,打印建模失败原因 if (status == 1) { System.out.println( "失败原因: " + EM_PERSON_FEATURE_ERRCODE.getErrorMessage( tmp[i].stuCandidatesEx.stPersonInfo.emFeatureErrCode)); } } } else { System.out.println( "CLIENT_DoFindFaceRecognitionEx Failed, Error:" + ENUMERROR.getErrorMessage()); break; } if (stFindOut.nRetCadidateEx2Num < stFindIn.nCount) { System.out.println("没有更多数据,结束查询"); break; } else { stFindIn.nBeginNum += count; doNextCount++; } } while (true); netsdkApi.CLIENT_StopFindFaceRecognition(longHandle); } /** * 以图搜图 * * @param startTime 开始时间 * @param endTime 结束时间 * @param chn 通道号 * @param similary 相似度 * @param memory 图片缓存 * @param nPicLen 图片大小 */ public void searchByPictureMultiVer(String startTime, String endTime, int chn, String similary, Memory memory, int nPicLen) { String[] startTimeStr = startTime.split("-"); String[] endTimeStr = endTime.split("-"); NET_IN_STARTMULTIFIND_FACERECONGNITION_EX stInStartFind = new NET_IN_STARTMULTIFIND_FACERECONGNITION_EX(); // 通道号 stInStartFind.nChannelCount = 5; IntPoint[] intPoints = new IntPoint[5]; for (int i = 0; i < 5; i++) { intPoints[i] = new IntPoint(i); } stInStartFind.pChannelID = new Memory(intPoints[0].size() * stInStartFind.nChannelCount); stInStartFind.pChannelID.clear(intPoints[0].size() * stInStartFind.nChannelCount); ToolKits.SetStructArrToPointerData(intPoints, stInStartFind.pChannelID); stInStartFind.bPersonEx2Enable = 1; // 人员信息查询条件是否有效, 并使用扩展结构体 stInStartFind.emObjectType = NetSDKLib.EM_OBJECT_TYPE.EM_OBJECT_TYPE_FACE; // 图片信息 if (memory != null && nPicLen > 0) { System.out.println("detect image"); String picPath = "D:\\tool\\work\\4.jpg"; byte[] picBuf = ToolKits.readPictureToByteArray(picPath); // ToolKits.savePicture(picBuf, "D:\\tool\\work\\wcnm.jpg"); stInStartFind.pBuffer = new Memory(picBuf.length); stInStartFind.pBuffer.write(0,picBuf,0,picBuf.length); stInStartFind.nBufferLen = nPicLen; stInStartFind.bPersonEx2Enable = 1; // 人员信息查询条件是否有效, 并使用扩展结构体 stInStartFind.stPersonInfoEx2.nFacePicNumEx = 1; stInStartFind.stPersonInfoEx2.stuFacePicInfoEx[0].dwOffSet = 0; stInStartFind.stPersonInfoEx2.stuFacePicInfoEx[0].dwFileLenth = nPicLen; // stInStartFind.stPersonInfoEx2.bPersonExEnable = 1; // stInStartFind.stPersonInfoEx2.nFacePicNumEx = 1; // stInStartFind.stPersonInfoEx2.stPersonInfoEx. // stInStartFind.stPersonInfoEx2.stPersonInfoEx.szFacePicInfo[0].dwOffSet = 0; // stInStartFind.stPersonInfoEx2.stPersonInfoEx.szFacePicInfo[0].dwFileLenth = nPicLen; } // 相似度 if (!similary.equals("")) { stInStartFind.stMatchOptions.nSimilarity = Integer.parseInt(similary); stInStartFind.stMatchOptions.nMaxCandidate = 5; } stInStartFind.stFilterInfo.nGroupIdNum = 0; stInStartFind.stFilterInfo.nRangeNum = 1; stInStartFind.stFilterInfo.szRange[0] = NetSDKLib.EM_FACE_DB_TYPE.NET_FACE_DB_TYPE_HISTORY; stInStartFind.stFilterInfo.stStartTime.dwYear = Integer.parseInt(startTimeStr[0]); stInStartFind.stFilterInfo.stStartTime.dwMonth = Integer.parseInt(startTimeStr[1]); stInStartFind.stFilterInfo.stStartTime.dwDay = Integer.parseInt(startTimeStr[2]); stInStartFind.stFilterInfo.stEndTime.dwYear = Integer.parseInt(endTimeStr[0]); stInStartFind.stFilterInfo.stEndTime.dwMonth = Integer.parseInt(endTimeStr[1]); stInStartFind.stFilterInfo.stEndTime.dwDay = Integer.parseInt(endTimeStr[2]); stInStartFind.stFilterInfo.emFaceType = NetSDKLib.EM_FACERECOGNITION_FACE_TYPE.EM_FACERECOGNITION_FACE_TYPE_ALL; //让设备根据查询条件整理结果集 NET_OUT_STARTMULTIFIND_FACERECONGNITION_EX stOutParam = new NET_OUT_STARTMULTIFIND_FACERECONGNITION_EX(); stInStartFind.write(); stOutParam.write(); if (netsdkApi.CLIENT_StartMultiFindFaceRecognitionEx(loginHandle, stInStartFind.getPointer(), stOutParam.getPointer(), 2000)) { stOutParam.read(); System.out.println("CLIENT_StartMultiFindFaceRecognitionEx success, count = " + stOutParam.nTotalCount); findHandle = stOutParam.lFindHandle; if(stOutParam.nTotalCount > 0){ doFindSearchByPictureEx(findHandle, 10); } if (stOutParam.nTotalCount == -1) { // -1表示总条数未生成,要推迟获取, 使用CLIENT_AttachFaceFindState接口状态 nToken = stOutParam.nToken; // 入参 NET_IN_MULTIFACE_DETECT_STATE pstInParam = new NET_IN_MULTIFACE_DETECT_STATE(); pstInParam.nTokensNum = 1; pstInParam.nTokens[0] = nToken; // 查询令牌 pstInParam.cbMultiFaceDetectState = new MultiFaceDetectState(); // 出参 NET_OUT_MULTIFACE_DETECT_STATE pstOutParam = new NET_OUT_MULTIFACE_DETECT_STATE(); pstInParam.write(); attachFaceHandle = netsdkApi.CLIENT_AttachDetectMultiFaceState(loginHandle, pstInParam.getPointer(), pstOutParam.getPointer(), 4000); pstInParam.read(); if (attachFaceHandle.longValue() != 0) { System.out.println("AttachFaceFindState Succeed!"); } else { System.out.println( "AttachFaceFindState Failed, Error:" + ToolKits.getErrorCode()); return; } NET_IN_FACE_RECOGNITION_DETECT_MULTI_FACE_INFO stIn = new NET_IN_FACE_RECOGNITION_DETECT_MULTI_FACE_INFO(); stIn.nBigPicNum = 1; String picPath = "D:\\tool\\work\\4.jpg"; byte[] picBuf = ToolKits.readPictureToByteArray(picPath); stIn.stuBigPicInfo[0].dwFileLenth = picBuf.length; stIn.nBufferLen = picBuf.length; stIn.pBuffer = new Memory(picBuf.length); stIn.pBuffer.write(0, picBuf, 0, picBuf.length); stIn.nToken = nToken; NET_OUT_FACE_RECOGNITION_DETECT_MULTI_FACE_INFO stOut = new NET_OUT_FACE_RECOGNITION_DETECT_MULTI_FACE_INFO(); stIn.write(); stOut.write(); boolean bRet = netsdkApi.CLIENT_FaceRecognitionDetectMultiFace(loginHandle, stIn.getPointer(), stOut.getPointer(), 3000); if (bRet) { System.out.println("CLIENT_FaceRecognitionDetectMultiFace success"); } else { System.out.println("CLIENT_FaceRecognitionDetectMultiFace failed"); System.out.println("error = " + ToolKits.getErrorCode()); } } } else { System.out.println( "CLIENT_StartMultiFindFaceRecognitionEx Failed, Error:" + ToolKits.getErrorCode()); } } public void tryMethod(){ String picPath = "D:\\tool\\work\\4.jpg"; byte[] picBuf = ToolKits.readPictureToByteArray(picPath); Memory memory = new Memory(picBuf.length); memory.write(0,picBuf,0,picBuf.length); searchByPictureMultiVer("2022-11-20","2022-11-27",0,"10",memory,picBuf.length); // doFindSearchByPictureEx(loginHandle, 10); } public void RunTest() { System.out.println("Run Test"); CaseMenu menu = new CaseMenu(); menu.addItem((new CaseMenu.Item(this, "tryMethod", "tryMethod"))); menu.run(); } public static void main(String[] args) { MultiFaceDemo multiFaceDemo = new MultiFaceDemo(); InitTest("172.25.239.136", 37777, "admin", "Admin123"); multiFaceDemo.RunTest(); LoginOut(); } }