package com.iailab.netsdk.demo.module;

import com.iailab.netsdk.lib.NetSDKLib;
import com.iailab.netsdk.lib.ToolKits;
import com.iailab.netsdk.lib.NetSDKLib.*;

import com.sun.jna.Memory;
import com.sun.jna.ptr.IntByReference;

/**
 * 以图搜图接口实现，跟查询人脸库里的人员信息的查找接口是一样的，入参和实现有区别
 * 目前只支持IVSS
 */
public class SearchByPictureModule {
	// 查询密令
	public static int nToken = 0;
	
	private static LLong m_FindHandle = null;
	
	// 订阅句柄
	private static LLong attachFaceHandle = new LLong(0);
	
	/**
	 * 按条件查询目标识别结果
	 * @param memory 	  图片缓存
	 * @param startTime  起始时间, 历史库需要时间，人脸库不需要时间
	 * @param endTime    结束时间, 历史库需要时间，人脸库不需要时间
	 * @param isHistory  是否是历史库， true-历史库； false-人脸库
	 * @param nChn       通道号， 历史库需要通道号，人脸库不需要通道号
	 * @param similary   相似度
	 * @return 查询到的所有人员个数  
	 */
	public static int startFindPerson(Memory memory,
									  String startTime, 
									  String endTime,
									  boolean isHistory,
									  int nChn,
									  String similary) {	
		
		m_FindHandle = null;
		nToken = 0;
		int nTotalCount = 0;  
		
		/*
		 * 入参, IVVS设备，查询条件只有  stuInStartFind.stPerson 里的参数有效
		 */
		NET_IN_STARTFIND_FACERECONGNITION stuIn = new NET_IN_STARTFIND_FACERECONGNITION();

		// 人员信息查询条件是否有效, 并使用扩展结构体
		stuIn.bPersonExEnable = 1; 
		
		// 图片信息
		if(memory != null) {
			stuIn.pBuffer = memory;
			stuIn.nBufferLen = (int)memory.size();
			stuIn.stPersonInfoEx.wFacePicNum = 1;
			stuIn.stPersonInfoEx.szFacePicInfo[0].dwOffSet = 0;
			stuIn.stPersonInfoEx.szFacePicInfo[0].dwFileLenth = (int)memory.size();
		}
		
		// 相似度
		if(!similary.isEmpty()) {
			stuIn.stMatchOptions.nSimilarity = Integer.parseInt(similary);
		}
		
		stuIn.stFilterInfo.nGroupIdNum = 0;
		stuIn.stFilterInfo.nRangeNum = 1;
		
		if(isHistory) {  // 历史库
			// 通道号
			stuIn.nChannelID = nChn; 
			stuIn.stFilterInfo.szRange[0] = EM_FACE_DB_TYPE.NET_FACE_DB_TYPE_HISTORY;  // 待查询数据库类型，设备只支持一个
			// 开始时间
			String[] startTimeStr = startTime.split("-");
			stuIn.stFilterInfo.stStartTime.dwYear = Integer.parseInt(startTimeStr[0]);
			stuIn.stFilterInfo.stStartTime.dwMonth = Integer.parseInt(startTimeStr[1]);
			stuIn.stFilterInfo.stStartTime.dwDay = Integer.parseInt(startTimeStr[2]);
			stuIn.stFilterInfo.stStartTime.dwHour= 0;
			stuIn.stFilterInfo.stStartTime.dwMinute= 0;
			stuIn.stFilterInfo.stStartTime.dwSecond= 0;
			// 结束时间
			String[] endTimeStr = endTime.split("-");
			stuIn.stFilterInfo.stEndTime.dwYear = Integer.parseInt(endTimeStr[0]);
			stuIn.stFilterInfo.stEndTime.dwMonth = Integer.parseInt(endTimeStr[1]);
			stuIn.stFilterInfo.stEndTime.dwDay = Integer.parseInt(endTimeStr[2]);
			stuIn.stFilterInfo.stEndTime.dwHour=23;
			stuIn.stFilterInfo.stEndTime.dwMinute=59;
			stuIn.stFilterInfo.stEndTime.dwSecond=59;
			stuIn.stFilterInfo.emFaceType = EM_FACERECOGNITION_FACE_TYPE.EM_FACERECOGNITION_FACE_TYPE_ALL;
		} else {        // 人脸库
			stuIn.stFilterInfo.szRange[0] = EM_FACE_DB_TYPE.NET_FACE_DB_TYPE_BLACKLIST;  // 待查询数据库类型，设备只支持一个
		}
	
	    /*
	     * 出参
	     */
	    NET_OUT_STARTFIND_FACERECONGNITION stuOut = new NET_OUT_STARTFIND_FACERECONGNITION();
	    stuIn.write();
	    stuOut.write();
	    if(LoginModule.netsdk.CLIENT_StartFindFaceRecognition(LoginModule.m_hLoginHandle, stuIn,  stuOut, 4000)) {        
	    	m_FindHandle = stuOut.lFindHandle;
	    	nTotalCount = stuOut.nTotalCount;
	    	nToken = stuOut.nToken;
	    } else {
	        System.out.println("CLIENT_StartFindFaceRecognition Failed, Error:" + ToolKits.getErrorCodePrint());
	    }
	    
	    return nTotalCount;
	}
	
	/**
	 * 查找目标识别结果
	 * @param beginNum 查询起始序号
	 * @param nCount  当前想查询的记录条数
	 * @return 返回的人员信息数组
	 */
	public static CANDIDATE_INFOEX[] doFindNextPerson(int beginNum, int nCount) {
    	/*
    	 *入参
    	 */
        NetSDKLib.NET_IN_DOFIND_FACERECONGNITION  stuIn = new NetSDKLib.NET_IN_DOFIND_FACERECONGNITION();
        stuIn.lFindHandle = m_FindHandle;
        stuIn.nCount      = nCount;  	 // 当前想查询的记录条数
        stuIn.nBeginNum   = beginNum;     // 查询起始序号
        
        /*
         * 出参
         */
        NetSDKLib.NET_OUT_DOFIND_FACERECONGNITION stuOut = new NetSDKLib.NET_OUT_DOFIND_FACERECONGNITION();;	
        stuOut.bUseCandidatesEx = 1;				// 是否使用候选对象扩展结构体
        
        // 必须申请内存，每次查询几个，必须至少申请几个，最大申请20个
        for(int i = 0; i < nCount; i++) {
            stuOut.stuCandidatesEx[i].stPersonInfo.szFacePicInfo[0].nFilePathLen = 256;
            stuOut.stuCandidatesEx[i].stPersonInfo.szFacePicInfo[0].pszFilePath = new Memory(256);
        }
        
    	stuIn.write();
    	stuOut.write();
        if(LoginModule.netsdk.CLIENT_DoFindFaceRecognition(stuIn, stuOut, 4000)) {
        	stuIn.read();
        	stuOut.read();
        	
        	if(stuOut.nCadidateExNum == 0) {
        		return null;
        	}
        	
        	// 获取到的信息
        	CANDIDATE_INFOEX[]	stuCandidatesEx = new CANDIDATE_INFOEX[stuOut.nCadidateExNum];
        	for(int i = 0; i < stuOut.nCadidateExNum; i++) {
        		stuCandidatesEx[i] = new CANDIDATE_INFOEX();
        		stuCandidatesEx[i] = stuOut.stuCandidatesEx[i];
        	}
        	
        	return stuCandidatesEx;
        } else {
        	System.out.println("CLIENT_DoFindFaceRecognition Failed, Error:" + ToolKits.getErrorCodePrint());
        }
    
        return null;
	}
	
	/**
	 * 结束查询
	 */
	public static boolean doFindClosePerson() {
		boolean bRet = false;
		if(m_FindHandle.longValue() != 0) {
			bRet = LoginModule.netsdk.CLIENT_StopFindFaceRecognition(m_FindHandle);
		}
		return bRet;
	}
	
	/**
	 * 订阅人脸查询状态
	 * @param faceFindStateCb 人脸状态回调函数
	 * @return
	 */
	public static boolean attachFaceFindState(fFaceFindState faceFindStateCb) {
		/*
		 * 入参
		 */
		NET_IN_FACE_FIND_STATE stuIn = new NET_IN_FACE_FIND_STATE();
		stuIn.nTokenNum = 1;   
		stuIn.nTokens = new IntByReference(nToken);  // 查询令牌
		stuIn.cbFaceFindState = faceFindStateCb;
		
		/*
		 * 出参
		 */
		NET_OUT_FACE_FIND_STATE stuOut = new NET_OUT_FACE_FIND_STATE();
		
		stuIn.write();
		attachFaceHandle = LoginModule.netsdk.CLIENT_AttachFaceFindState(LoginModule.m_hLoginHandle, stuIn, stuOut, 4000);
		stuIn.read();
		
		if(attachFaceHandle.longValue() != 0) {
			System.out.println("AttachFaceFindState Succeed!");
			return true;
		}
		
		return false;
	}
	
	/**
	 * 关闭订阅
	 */
	public static void detachFaceFindState() {
		if(attachFaceHandle.longValue() != 0) {		
			LoginModule.netsdk.CLIENT_DetachFaceFindState(attachFaceHandle);
			attachFaceHandle.setValue(0);
		}
	}
}