package com.netsdk.demo.customize;

import com.netsdk.demo.BaseDemo;
import com.netsdk.demo.util.CaseMenu;
import com.netsdk.lib.NetSDKLib;
import com.netsdk.lib.ToolKits;
import com.netsdk.lib.Utils;
import com.netsdk.lib.callback.impl.DefaultAnalyseTaskResultCallBack;
import com.netsdk.lib.enumeration.EM_DATA_SOURCE_TYPE;
import com.netsdk.module.AnalyseTaskModule;
import com.netsdk.module.entity.AddAnalyseTaskResult;
import com.netsdk.module.entity.PushAnalysePictureInfo;
import com.sun.jna.Memory;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;

import static com.netsdk.lib.NetSDKLib.EVENT_IVS_FACEANALYSIS;

/**
 * @author 47081
 * @version 1.0
 * @description 图片智能分析demo, 1:1,1:n
 * @date 2020/10/21
 */
public class PictureAnalyseDemo extends BaseDemo {
    private final AnalyseTaskModule analyseTaskModule;
    //任务id
    private int taskId;

    public PictureAnalyseDemo() {
        super();
        analyseTaskModule = new AnalyseTaskModule(getNetSdkApi());
    }

    /**
     * 1:1,图片比对
     */
    public void matchPicture() {
        try {
            int result = analyseTaskModule.matchImage(getLoginHandler(), new FileInputStream(new File("D:/shanyu.jpg")), new FileInputStream(new File("D:/test.jpg")), 3000);
            System.out.println(result);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 添加智能分析任务
     */
    public void addTask() {
        NetSDKLib.NET_PUSH_PICFILE_INFO inParam = new NetSDKLib.NET_PUSH_PICFILE_INFO();
        //立刻启动
        inParam.emStartRule = NetSDKLib.EM_ANALYSE_TASK_START_RULE.EM_ANALYSE_TASK_START_NOW;
        //配置分析规则
        //分析规则条数
        inParam.stuRuleInfo.nRuleCount = 1;
        //大类类型,目标检测/目标识别
        inParam.stuRuleInfo.stuRuleInfos[0].emClassType = NetSDKLib.EM_SCENE_CLASS_TYPE.EM_SENCE_CLASS_FACERECOGNITION;
        //规则类型,人脸分析EVENT_IVS_FACEANALYSIS
        inParam.stuRuleInfo.stuRuleInfos[0].dwRuleType = EVENT_IVS_FACEANALYSIS;

        //检测物体类型个数
        inParam.stuRuleInfo.stuRuleInfos[0].nObjectTypeNum = 1;
        //检测物体类型列表,检测人脸
        inParam.stuRuleInfo.stuRuleInfos[0].emObjectTypes[0] = NetSDKLib.EM_ANALYSE_OBJECT_TYPE.EM_ANALYSE_OBJECT_TYPE_HUMANFACE;
        //规则配置
        NetSDKLib.NET_FACEANALYSIS_RULE_INFO info = new NetSDKLib.NET_FACEANALYSIS_RULE_INFO();

        //检测区顶点数
        info.nDetectRegionPoint = 4;
        info.stuDetectRegion[0].nX = 0;
        info.stuDetectRegion[0].nY = 0;
        info.stuDetectRegion[1].nX = 0;
        info.stuDetectRegion[1].nY = 8192;
        info.stuDetectRegion[2].nX = 8192;
        info.stuDetectRegion[2].nY = 8192;
        info.stuDetectRegion[3].nX = 8192;
        info.stuDetectRegion[3].nY = 0;
        //灵敏度,1-10
        info.nSensitivity = 2;
        //布控组
        info.nLinkGroupNum = 1;
        //布控组启用
        info.stuLinkGroup[0].bEnable = true;
        //人脸库id
        System.arraycopy("1".getBytes(Charset.forName(Utils.getPlatformEncode())), 0, info.stuLinkGroup[0].szGroupID, 0, "1".getBytes(Charset.forName(Utils.getPlatformEncode())).length);
        //陌生人布控模式,未启用
        info.stuStrangerMode.bEnable = false;
        //启用相似度
        info.stuLinkGroup[0].bShowPlate = true;
        info.stuLinkGroup[0].bySimilarity = 20;
        //尺寸过滤
        info.bSizeFileter = false;
        //是否开启人脸属性识别
        info.bFeatureEnable = false;
        info.nFaceFeatureNum = 0;
        info.bFeatureFilter = false;
        info.nMinQuality = 50;
        inParam.stuRuleInfo.stuRuleInfos[0].pReserved = new Memory(info.size());
        ToolKits.SetStructDataToPointer(info, inParam.stuRuleInfo.stuRuleInfos[0].pReserved, 0);
        AddAnalyseTaskResult result = analyseTaskModule.addAnalyseTask(getLoginHandler(), EM_DATA_SOURCE_TYPE.EM_DATA_SOURCE_PUSH_PICFILE, inParam, 3000);
        if (result.isResult()) {
            taskId = result.getTaskId();
        }
    }

    public void attachAnalyseTask() {
        setAttachHandler(analyseTaskModule.attachAnalyseTaskResult(getLoginHandler(), new int[]{taskId}, new int[]{}, true, DefaultAnalyseTaskResultCallBack.getSingleInstance(), 3000, null));
    }

    public void startAnalyseTask() {
        analyseTaskModule.startAnalyseTask(getLoginHandler(), taskId, 3000);
    }

    /**
     * 主动推送图片
     */
    public void pushPicture() {
        List<PushAnalysePictureInfo> infos = new ArrayList<>();
        //推送的图片最好是jg,100k以下
        infos.add(new PushAnalysePictureInfo("D:/liyang(3).jpg", "test"));
        infos.add(new PushAnalysePictureInfo("D:/test.jpg", "test1"));
        infos.add(new PushAnalysePictureInfo("D:/shanyu.jpg", "test2"));
        analyseTaskModule.pushAnalysePicture(getLoginHandler(), taskId, 1, infos, 3000);
    }

    public void dettachAnalyseTask() {
        analyseTaskModule.detachAnalyseTaskResult(getLoginHandler());
    }

    public void removeTask() {
        analyseTaskModule.removeAnalyseTask(getLoginHandler(), taskId, 3000);
    }

    /**
     * demo调用顺序
     * 1.添加目标识别任务
     * 2.订阅智能分析任务
     * 3.推送图片
     * 等待回调事件上报
     * 4.取消订阅
     * 5.删除任务
     * 6.1:1
     *
     * @param args
     */
    public static void main(String[] args) {
        String ip = "172.12.245.69";
        int port = 37777;
        String username = "admin";
        String password = "admin123";
        PictureAnalyseDemo demo = new PictureAnalyseDemo();
        demo.init();

        if (demo.login(ip, port, username, password)) {
            //登录成功后
            demo.addItem(new CaseMenu.Item(demo, "1:1", "matchPicture"));
            demo.addItem(new CaseMenu.Item(demo, "添加目标识别任务", "addTask"));
            demo.addItem(new CaseMenu.Item(demo, "订阅智能分析任务", "attachAnalyseTask"));
            //demo.addItem(new CaseMenu.Item(demo, "启动智能分析任务", "startAnalyseTask"));
            demo.addItem(new CaseMenu.Item(demo, "推送图片", "pushPicture"));
            demo.addItem(new CaseMenu.Item(demo, "取消订阅", "dettachAnalyseTask"));
            demo.addItem(new CaseMenu.Item(demo, "移除任务", "removeTask"));
            demo.run();
            demo.logout();
        }
        demo.clean();
    }

}