package com.netsdk.demo.customize.faceReconEx;

import com.netsdk.demo.util.CaseMenu;
import com.netsdk.demo.util.PaintJPanel;
import com.netsdk.lib.NetSDKLib;
import com.netsdk.lib.ToolKits;
import com.netsdk.lib.Utils;
import com.netsdk.lib.enumeration.EM_PERSON_FEATURE_ERRCODE;
import com.netsdk.lib.enumeration.ENUMERROR;
import com.sun.jna.Memory;
import com.sun.jna.ptr.IntByReference;

import javax.imageio.ImageIO;
import javax.swing.*;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Scanner;

import static com.netsdk.lib.NetSDKLib.EM_CERTIFICATE_TYPE.CERTIFICATE_TYPE_IC;
import static com.netsdk.lib.Utils.getOsPrefix;

/**
 * @author ： 47040
 * @since ： Created in 2020/7/20 16:28
 */
public class FaceReconExDemo {

  // The constant net sdk
  public static final NetSDKLib netsdk = NetSDKLib.NETSDK_INSTANCE;

  // The constant config sdk.
  public static NetSDKLib configsdk = NetSDKLib.CONFIG_INSTANCE;

  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 NetSDKLib.NET_DEVICEINFO_Ex deviceInfo = new NetSDKLib.NET_DEVICEINFO_Ex(); // 设备信息

  private NetSDKLib.LLong m_hLoginHandle = new NetSDKLib.LLong(0); // 登录句柄

  /** login with high level 高安全级别登陆 */
  public void loginWithHighLevel() {

    NetSDKLib.NET_IN_LOGIN_WITH_HIGHLEVEL_SECURITY pstlnParam =
        new NetSDKLib.NET_IN_LOGIN_WITH_HIGHLEVEL_SECURITY() {
          {
            szIP = m_strIpAddr.getBytes();
            nPort = m_nPort;
            szUserName = m_strUser.getBytes();
            szPassword = m_strPassword.getBytes();
          }
        }; // 输入结构体参数
    NetSDKLib.NET_OUT_LOGIN_WITH_HIGHLEVEL_SECURITY pstOutParam =
        new NetSDKLib.NET_OUT_LOGIN_WITH_HIGHLEVEL_SECURITY(); // 输结构体参数

    // 写入sdk
    m_hLoginHandle = netsdk.CLIENT_LoginWithHighLevelSecurity(pstlnParam, pstOutParam);

    if (m_hLoginHandle.longValue() == 0) {
      System.err.printf(
          "Login Device[%s] Port[%d]Failed. %s\n",
          m_strIpAddr, m_nPort, netsdk.CLIENT_GetLastError());
    } else {
      deviceInfo = pstOutParam.stuDeviceInfo; // 获取设备信息
      System.out.println("Login Success");
      System.out.println("Device Address：" + m_strIpAddr);
      System.out.println("设备包含：" + deviceInfo.byChanNum + "个通道");
    }
  }

  /** logout 退出 */
  public void logOut() {
    if (m_hLoginHandle.longValue() != 0) {
      netsdk.CLIENT_Logout(m_hLoginHandle);
      System.out.println("LogOut Success");
    }
  }

  //////////////////////////////////////// 订阅事件/退订 ///////////////////////////////////
  ////////////////////////////////////////////////////////////////////////////////////////
  ////////////////////////////////////////////////////////////////////////////////////////

  private NetSDKLib.LLong m_hAttachHandle = new NetSDKLib.LLong(0); // 订阅相关

  private final NetSDKLib.fAnalyzerDataCallBack analyzerDataCB =
      FaceReconAnalyzerDataCallBack.getSingleInstance();

  private int channel = 0;

  public void setChannelID() {
    System.out.println("请输入通道，从0开始计数，-1表示全部");
    Scanner sc = new Scanner(System.in);
    this.channel = sc.nextInt();
  }

  public void AttachEventRealLoadPic() {

    this.DetachEventRealLoadPic(); // 先退订，设备不会对重复订阅作校验，重复订阅后会有重复的事件返回

    int bNeedPicture = 1; // 需要图片

    m_hAttachHandle =
        netsdk.CLIENT_RealLoadPictureEx(
            m_hLoginHandle,
            channel,
            NetSDKLib.EVENT_IVS_ALL,
            bNeedPicture,
            analyzerDataCB,
            null,
            null);
    if (m_hAttachHandle.longValue() != 0) {
      System.out.printf("Chn[%d] CLIENT_RealLoadPictureEx Success\n", channel);
    } else {
      System.out.printf(
          "Ch[%d] CLIENT_RealLoadPictureEx Failed!LastError = %s\n",
          channel, ToolKits.getErrorCode());
    }
  }

  /** 停止侦听智能事件 */
  public void DetachEventRealLoadPic() {
    if (m_hAttachHandle.longValue() != 0) {
      netsdk.CLIENT_StopLoadPic(m_hAttachHandle);
    }
  }

  private String faceUid;

  public void addPerson() throws IOException {
    // 人脸图片
    BufferedImage jpg = ImageIO.read(new File("D:/新建文件夹/psc2-large.jpg"));
    ByteArrayOutputStream byteArrOutput = new ByteArrayOutputStream();
    ImageIO.write(jpg, "jpg", byteArrOutput);
    ////////////////// 获取背景图的缓存、宽高 /////////////
    byte[] buffer = byteArrOutput.toByteArray(); // jpg格式的Buf
    Memory memory = new Memory(buffer.length);
    memory.write(0, buffer, 0, buffer.length);
    faceUid =
    		addTargetRecognitionDB(
            "1",
            buffer.length,
            jpg.getWidth(),
            jpg.getHeight(),
            memory,
            "person1",
            (byte) 1,
            Arrays.asList("2021", "04", "28").toArray(new String[] {}),
            "浙江",
            "杭州",
            "1",
            (byte) 1);
  }
  /**
   * 添加人员信息(即注册人脸)
   *
   * @param groupId 组ID(人脸库ID)
   * @param nPicBufLen 图片大小
   * @param width 图片宽
   * @param height 图片高
   * @param memory 保存图片的缓存
   * @param personName 人员名称
   * @param bySex 性别
   * @param birthday 生日(年月日数组)
   * @param province 省份
   * @param id 证件编号
   * @param byIdType 证件类型
   * @return 新增人脸的uid(uid由设备生成,具备唯一性)
   */
  public String addTargetRecognitionDB(
      String groupId,
      int nPicBufLen,
      int width,
      int height,
      Memory memory,
      String personName,
      byte bySex,
      String[] birthday,
      String province,
      String city,
      String id,
      byte byIdType) {
    // 入参
    NetSDKLib.NET_IN_OPERATE_FACERECONGNITIONDB stuIn =
        new NetSDKLib.NET_IN_OPERATE_FACERECONGNITIONDB();
    stuIn.emOperateType = NetSDKLib.EM_OPERATE_FACERECONGNITIONDB_TYPE.NET_FACERECONGNITIONDB_ADD;

    ///////// 使用人员扩展信息 //////////
    stuIn.bUsePersonInfoEx = 1;

    // 组ID设置
    System.arraycopy(
        groupId.getBytes(), 0, stuIn.stPersonInfoEx.szGroupID, 0, groupId.getBytes().length);

    // 生日设置
    stuIn.stPersonInfoEx.wYear = (short) Integer.parseInt(birthday[0]);
    stuIn.stPersonInfoEx.byMonth = (byte) Integer.parseInt(birthday[1]);
    stuIn.stPersonInfoEx.byDay = (byte) Integer.parseInt(birthday[2]);

    // 性别,1-男,2-女,作为查询条件时,此参数填0,则表示此参数无效
    stuIn.stPersonInfoEx.bySex = bySex;

    // 人员名字
    try {
      System.arraycopy(
          personName.getBytes(encode),
          0,
          stuIn.stPersonInfoEx.szPersonName,
          0,
          personName.getBytes(encode).length);
    } catch (UnsupportedEncodingException e) {
      e.printStackTrace();
    }

    // 证件类型
    stuIn.stPersonInfoEx.byIDType = byIdType;

    // 人员唯一标识(证件号码,工号,或其他编号)
    System.arraycopy(id.getBytes(), 0, stuIn.stPersonInfoEx.szID, 0, id.getBytes().length);

    // 国际,符合ISO3166规范
    System.arraycopy("CN".getBytes(), 0, stuIn.stPersonInfoEx.szCountry, 0, "CN".getBytes().length);

    // 省份
    try {
      System.arraycopy(
          province.getBytes(encode),
          0,
          stuIn.stPersonInfoEx.szProvince,
          0,
          province.getBytes(encode).length);
    } catch (UnsupportedEncodingException e) {
      e.printStackTrace();
    }

    // 城市
    try {
      System.arraycopy(
          city.getBytes(encode), 0, stuIn.stPersonInfoEx.szCity, 0, city.getBytes(encode).length);
    } catch (UnsupportedEncodingException e) {
      e.printStackTrace();
    }

    // 图片张数、大小、宽、高、缓存设置
    if (memory != null) {
      stuIn.stPersonInfoEx.wFacePicNum = 1; // 图片张数
      stuIn.stPersonInfoEx.szFacePicInfo[0].dwFileLenth = nPicBufLen;
      stuIn.stPersonInfoEx.szFacePicInfo[0].dwOffSet = 0;
      stuIn.stPersonInfoEx.szFacePicInfo[0].wWidth = (short) width;
      stuIn.stPersonInfoEx.szFacePicInfo[0].wHeight = (short) height;

      stuIn.nBufferLen = nPicBufLen;
      stuIn.pBuffer = memory;
    }

    // 出参
    NetSDKLib.NET_OUT_OPERATE_FACERECONGNITIONDB stuOut =
        new NetSDKLib.NET_OUT_OPERATE_FACERECONGNITIONDB();

    stuIn.write();
    if (netsdk.CLIENT_OperateFaceRecognitionDB(m_hLoginHandle, stuIn, stuOut, 3000)) {
      stuOut.read();
      System.out.println(
          "添加人员成功!,szUID: "
              + new String(stuOut.szUID, Charset.forName(Utils.getPlatformEncode())).trim());
    } else {
      System.out.println("添加人员信息失败,失败原因: " + ENUMERROR.getErrorMessage());
      return "-1";
    }
    stuIn.read();
    return new String(stuOut.szUID, Charset.forName(Utils.getPlatformEncode())).trim();
  }

  public void queryStatus() {
    if (Integer.parseInt(faceUid) > 0) {
      searchByPicture(faceUid, "1", 0);
    } else {
      System.out.println("uid 无效，请重试");
    }
  }
  /**
   * 查询状态
   *
   * @param uid 人员uid
   * @param channel 通道号(查询注册库与通道无关,0，-1均可)
   * @param groudId 人脸库的id
   */
  public void searchByPicture(String uid, String groudId, int channel) {
    // IVSS设备，查询条件只有  stInStartFind.stPerson 里的参数有效
    NetSDKLib.NET_IN_STARTFIND_FACERECONGNITION stInStartFind =
        new NetSDKLib.NET_IN_STARTFIND_FACERECONGNITION();
    NetSDKLib.LLong findHandle;
    int nToken;
    // 通道号
    stInStartFind.nChannelID = channel;
    stInStartFind.bPersonExEnable = 1; // 人员信息查询条件是否有效, 并使用扩展结构体
    stInStartFind.stFilterInfo.nUIDNum = 1;
    // uid 64*32,每个uid占32字节,最多64个uid
    byte[] szUid = uid.getBytes(Charset.forName(Utils.getPlatformEncode()));
    System.arraycopy(szUid, 0, stInStartFind.stFilterInfo.szUIDs, 0 * 32, szUid.length);
    // System.arraycopy(szUid, 0, stInStartFind.stPersonInfoEx.szUID, 0, szUid.length);

    // System.arraycopy(szGroudId, 0, stInStartFind.stPersonInfoEx.szGroupID, 0, szGroudId.length);
    stInStartFind.stFilterInfo.nGroupIdNum = 1;
    byte[] szGroupId = groudId.getBytes(Charset.forName(Utils.getPlatformEncode()));
    System.arraycopy(
        szGroupId, 0, stInStartFind.stFilterInfo.szGroupIdArr[0].szGroupId, 0, szGroupId.length);
    // 查询注册库
    stInStartFind.stFilterInfo.szRange[0] = NetSDKLib.EM_FACE_DB_TYPE.NET_FACE_DB_TYPE_BLACKLIST;
    stInStartFind.stFilterInfo.emFaceType =
        NetSDKLib.EM_FACERECOGNITION_FACE_TYPE.EM_FACERECOGNITION_FACE_TYPE_ALL;

    // 让设备根据查询条件整理结果集
    NetSDKLib.NET_OUT_STARTFIND_FACERECONGNITION stOutParam =
        new NetSDKLib.NET_OUT_STARTFIND_FACERECONGNITION();
    stInStartFind.write();
    stOutParam.write();
    if (netsdk.CLIENT_StartFindFaceRecognition(m_hLoginHandle, stInStartFind, stOutParam, 2000)) {
      findHandle = stOutParam.lFindHandle;
      if (stOutParam.nTotalCount == 0) {
        System.out.println("查询结果为空,请检查查询条件");
        return;
      }
      if (stOutParam.nTotalCount == -1) { // -1表示总条数未生成,要推迟获取, 使用CLIENT_AttachFaceFindState接口状态
        nToken = stOutParam.nToken;
        // 入参
        NetSDKLib.NET_IN_FACE_FIND_STATE pstInParam = new NetSDKLib.NET_IN_FACE_FIND_STATE();
        pstInParam.nTokenNum = 1;
        pstInParam.nTokens = new IntByReference(nToken); // 查询令牌
        pstInParam.cbFaceFindState = DefaultFaceFindStateCallback.getInstance();

        // 出参
        NetSDKLib.NET_OUT_FACE_FIND_STATE pstOutParam = new NetSDKLib.NET_OUT_FACE_FIND_STATE();
        pstInParam.write();
        NetSDKLib.LLong attachFaceHandle =
            netsdk.CLIENT_AttachFaceFindState(m_hLoginHandle, pstInParam, pstOutParam, 4000);
        pstInParam.read();
        if (attachFaceHandle.longValue() != 0) {
          System.out.println("AttachFaceFindState Succeed!");
        }
        // 等待进度完成
        try {
          while (DefaultFaceFindStateCallback.getInstance()
                  .getProgress(m_hLoginHandle.longValue(), attachFaceHandle.longValue(), nToken)
              < 100) {}
          // 进度到100
          doFindSearchByPicture(findHandle, 10);
        } catch (InterruptedException e) {
          e.printStackTrace();
        }
      } else {
        doFindSearchByPicture(findHandle, 10);
      }
    } else {
      System.out.println(
          "CLIENT_StartFindFaceRecognition Failed, Error:" + ToolKits.getErrorCode());
    }
  }

  /**
   * 以图搜图
   *
   * @param longHandle CLIENT_StartFindFaceRecognition 接口返回的查询句柄
   * @param count 查询的个数
   */
  public void doFindSearchByPicture(NetSDKLib.LLong longHandle, int count) {
    int doNextCount = 0;
    // 分页查找数据
    NetSDKLib.NET_IN_DOFIND_FACERECONGNITION stFindIn =
        new NetSDKLib.NET_IN_DOFIND_FACERECONGNITION();
    stFindIn.lFindHandle = longHandle;
    stFindIn.nCount = 10; // 当前想查询的记录条数
    stFindIn.nBeginNum = 0; // 每次递增

    NetSDKLib.NET_OUT_DOFIND_FACERECONGNITION stFindOut =
        new NetSDKLib.NET_OUT_DOFIND_FACERECONGNITION();
    stFindOut.bUseCandidatesEx = 1; // 是否使用候选对象扩展结构体
    for (int i = 0; i < count; i++) {
      stFindOut.stuCandidatesEx[i].stPersonInfo.szFacePicInfo[0].nFilePathLen = 256;
      stFindOut.stuCandidatesEx[i].stPersonInfo.szFacePicInfo[0].pszFilePath = new Memory(256);
    }

    do {
      stFindIn.write();
      stFindOut.write();
      if (netsdk.CLIENT_DoFindFaceRecognition(stFindIn, stFindOut, 1000)) {
        System.out.printf("Record Number [%d]\n", stFindOut.nCadidateExNum);

        if (stFindOut.nCadidateExNum == 0) {
          System.out.println("没有查询到相关数据");
          break;
        }
        for (int i = 0; i < stFindOut.nCadidateExNum; i++) {
          int index = i + doNextCount * count; // 查询的总个数 - 1, 从0开始
          // 建模状态
          /** 0:未知 1:建模失败,可能是图片不符合要求,需要换图片 2:有可用的特征值 3:正在计算特征值 4:已建模，但算法升级导致数据不可用，需要重新建模 */
          int status = stFindOut.stuCandidatesEx[i].stPersonInfo.emFeatureState;
          System.out.println("建模状态: " + status);
          // 如果建模失败,打印建模失败原因
          if (status == 1) {
            System.out.println(
                "失败原因: "
                    + EM_PERSON_FEATURE_ERRCODE.getErrorMessage(
                        stFindOut.stuCandidatesEx[i].stPersonInfo.emFeatureErrCode));
          }
        }
      } else {
        System.out.println(
            "CLIENT_DoFindFaceRecognition Failed, Error:" + ENUMERROR.getErrorMessage());
        break;
      }

      if (stFindOut.nCadidateNum < stFindIn.nCount) {
        System.out.println("没有更多数据,结束查询");
        break;
      } else {
        stFindIn.nBeginNum += count;
        doNextCount++;
      }
    } while (true);
    netsdk.CLIENT_StopFindFaceRecognition(longHandle);
  }
  /////////////////////////////////////// 简易控制台 ///////////////////////////////////////
  /////////////////////////////////////////////////////////////////////////////////////////
  /////////////////////////////////////////////////////////////////////////////////////////

  // 初始化测试
  public void InitTest() {

    FaceReconExUtils.Init(); // 初始化SDK库
    this.loginWithHighLevel(); // 高安全登录
  }

  // 加载测试内容
  public void RunTest() {
    CaseMenu menu = new CaseMenu();
    menu.addItem(new CaseMenu.Item(this, "选择通道", "setChannelID"));
    menu.addItem(new CaseMenu.Item(this, "订阅任务", "AttachEventRealLoadPic"));
    menu.addItem(new CaseMenu.Item(this, "退订任务", "DetachEventRealLoadPic"));
    menu.addItem(new CaseMenu.Item(this, "添加人脸", "addPerson"));
    menu.addItem(new CaseMenu.Item(this, "查询人员建模状态", "queryStatus"));
    menu.run();
  }

  // 结束测试
  public void EndTest() {
    System.out.println("End Test");
    this.logOut(); // 退出
    System.out.println("See You...");

    FaceReconExUtils.cleanAndExit(); // 清理资源并退出
  }

  /////////////// 配置登陆地址，端口，用户名，密码  ////////////////////////
  //    private String m_strIpAddr = "172.23.12.29";  // 模拟器
  private String m_strIpAddr = "172.12.245.68"; // 模拟器
  private int m_nPort = 37777;
  private String m_strUser = "admin";
  private String m_strPassword = "admin123";
  //////////////////////////////////////////////////////////////////////

  public static void main(String[] args) {
    FaceReconExDemo demo = new FaceReconExDemo();
    if (args.length == 4) {
      demo.m_strIpAddr = args[0];
      demo.m_nPort = Integer.parseInt(args[1]);
      demo.m_strUser = args[2];
      demo.m_strPassword = args[3];
    }
    demo.InitTest();
    if (demo.m_hLoginHandle.longValue() != 0) {
      demo.RunTest();
    }
    demo.EndTest();
  }
}