package com.netsdk.demo.customize;

import com.netsdk.demo.util.CaseMenu;
import com.netsdk.lib.NetSDKLib;
import com.netsdk.lib.ToolKits;
import com.netsdk.lib.callback.impl.DefaultHaveReconnectCallBack;
import com.netsdk.lib.structure.*;
import com.netsdk.lib.structure.NET_TIME_EX;
import com.sun.jna.Pointer;

import java.io.*;
import java.text.SimpleDateFormat;
import java.util.Date;

import static com.netsdk.lib.NetSDKLib.*;
import static com.netsdk.lib.Utils.getOsPrefix;

public class SameObjectSearchDemo {

    // SDk对象初始化
    public static final NetSDKLib netsdk = NetSDKLib.NETSDK_INSTANCE;
    public static final NetSDKLib configsdk = NetSDKLib.CONFIG_INSTANCE;

    /**
     * 选择通道
     */
    private int channelId = -1;// 逻辑通道
    // 判断是否初始化
    private static boolean bInit = false;
    // 判断log是否打开
    private static boolean bLogOpen = false;
    // 设备信息
    private NetSDKLib.NET_DEVICEINFO_Ex deviceInfo = new NetSDKLib.NET_DEVICEINFO_Ex();
    // 登录句柄
    private NetSDKLib.LLong loginHandle = new NetSDKLib.LLong(0);
    // 智能事件订阅句柄
    private NetSDKLib.LLong m_attachHandle = new NetSDKLib.LLong(0);

    //实现断线重连静态单例
    public static class DefaultDisconnectCallback implements NetSDKLib.fDisConnect {
        private static volatile DefaultDisconnectCallback INSTANCE;

        private DefaultDisconnectCallback() {}

        public static DefaultDisconnectCallback getINSTANCE() {
            if (INSTANCE == null) {
                synchronized (DefaultDisconnectCallback.class){
                    if (INSTANCE == null){
                        INSTANCE = new DefaultDisconnectCallback();
                    }
                }
            }
            return INSTANCE;
        }

        @Override
        public void invoke(NetSDKLib.LLong lLoginID, String pchDVRIP, int nDVRPort, Pointer dwUser) {
            System.out.printf("Device[%s] Port[%d] DisConnected!\n", pchDVRIP, nDVRPort);
        }
    }

    // 回调函数需要是静态的，防止被系统回收
    // 断线回调
    private static NetSDKLib.fDisConnect disConnectCB = MultiTalkDevDemo.DefaultDisconnectCallback.getINSTANCE();
    // 重连回调
    private static NetSDKLib.fHaveReConnect haveReConnectCB = DefaultHaveReconnectCallBack.getINSTANCE();

    //订阅句柄
    NetSDKLib.LLong lAttachHandle = new NetSDKLib.LLong(0);
    // 编码格式
    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";
        }
    }

    /**
     * 按照指定格式，获取当前时间
     */
    public static String getDate() {
        SimpleDateFormat simpleDate = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        return simpleDate.format(new Date()).replaceAll("[^0-9]", "-");
    }

    /**
     * 初始化SDK库
     */
    public static boolean initSdk() {
        bInit = netsdk.CLIENT_Init(disConnectCB, null);// 进程启动时，初始化一次
        if (!bInit) {
            System.out.println("Initialize SDK failed");
            return false;
        }
        // 配置日志
        enableLog();
        // 设置断线重连回调接口, 此操作为可选操作，但建议用户进行设置
        netsdk.CLIENT_SetAutoReconnect(haveReConnectCB, null);
        // 设置登录超时时间和尝试次数，可选
        // 登录请求响应超时时间设置为3S
        int waitTime = 3000; //单位为ms
        // 登录时尝试建立链接 1 次
        int tryTimes = 1;
        // 设置连接设备超时时间和尝试次数
        netsdk.CLIENT_SetConnectTime(waitTime, tryTimes);
        // 设置更多网络参数， NET_PARAM 的nWaittime ， nConnectTryNum 成员与 CLIENT_SetConnectTime
        // 接口设置的登录设备超时时间和尝试次数意义相同,可选
        NetSDKLib.NET_PARAM netParam = new NetSDKLib.NET_PARAM();
        // 登录时尝试建立链接的超时时间
        netParam.nConnectTime = 10000;
        // 设置子连接的超时时间
        netParam.nGetConnInfoTime = 3000;
        //设置登陆网络环境
        netsdk.CLIENT_SetNetworkParam(netParam);
        return true;
    }

    /**
     * 打开 sdk log
     */
    private static void enableLog() {
        // SDK全局日志打印信息
        NetSDKLib.LOG_SET_PRINT_INFO setLog = new NetSDKLib.LOG_SET_PRINT_INFO();
        //设置日志路径
        File path = new File("sdklog/");
        //判断日志路径是否存在，若不存在则创建
        if (!path.exists()){
            path.mkdir();
        }
        // 这里的log保存地址依据实际情况自己调整
        String logPath = path.getAbsoluteFile().getParent() + "\\sdklog\\" + "sdklog" + getDate() + ".log";
        //日志输出策略,0:输出到文件(默认); 1:输出到窗口,
        setLog.nPrintStrategy = 0;
        //是否重设日志路径, 取值0否 ,1是
        setLog.bSetFilePath = 1;
        //日志路径(默认"./sdk_log/sdk_log.log")
        byte[] szLogFilePath = setLog.szLogFilePath;
        //自定义log保存地址，将数据logPath数据copy到LOG_SET_PRINT_INFO-->szLogFilePath变量中
        System.arraycopy(logPath.getBytes(), 0, szLogFilePath, 0, logPath.getBytes().length);
        //是否重设日志打印输出策略 取值0否 ,1是
        setLog.bSetPrintStrategy = 1;
        // 打开日志功能
        bLogOpen = netsdk.CLIENT_LogOpen(setLog);
        if (!bLogOpen){
            System.err.println("Failed to open NetSDK log "+ ToolKits.getErrorCode());
        }else {
            System.out.println("Success to open NetSDK log ");
        }
    }

    /**
     * 高安全登录
     */
    public void loginWithHighLevel() {
        // 输入结构体参数
        NetSDKLib.NET_IN_LOGIN_WITH_HIGHLEVEL_SECURITY pstlnParam = new NetSDKLib.NET_IN_LOGIN_WITH_HIGHLEVEL_SECURITY() {
            {
                // IP
                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();
        // 高安全级别登陆
        loginHandle = netsdk.CLIENT_LoginWithHighLevelSecurity(pstlnParam, pstOutParam);
        if (loginHandle.longValue() == 0) {  //登陆失败
            System.err.printf("Login Device[%s] Port[%d]Failed. %s\n", m_strIpAddr, m_nPort, ToolKits.getErrorCode());

        } else {
            //登陆成功
            // 获取设备信息
            deviceInfo = pstOutParam.stuDeviceInfo;
            System.out.printf("Login Success Device Address[%s] 设备包含[%d]个通道 \n",m_strIpAddr,deviceInfo.byChanNum);
        }
    }

    /**
     * 停止侦听智能事件
     */
    public void DetachEventRealLoadPic() {
        if (m_attachHandle.longValue() != 0) {
            netsdk.CLIENT_StopLoadPic(m_attachHandle);
            System.out.println("CLIENT_StopLoadPic Success");
        }
    }

    /**
     * 退出
     */
    public void logOut() {
        //判断是否已登录
        if (loginHandle.longValue() != 0) {
            netsdk.CLIENT_Logout(loginHandle);
            System.out.println("LogOut Success");
        }
    }

    /**
     * 清理sdk环境并退出
     */
    public static void cleanAndExit() {
        //判断log是否打开
        if (bLogOpen) {
            // 关闭sdk日志打印
            netsdk.CLIENT_LogClose();
        }
        // 进程关闭时，调用一次
        netsdk.CLIENT_Cleanup();
        System.exit(0);
    }

    /**
     * 初始化测试
     */
    public void initTest() {
        initSdk();

        this.loginWithHighLevel();
    }

    /**
     * 结束测试
     */
    public void endTest() {
        System.out.println("End Test");
        this.logOut(); // 登出设备
        System.out.println("See You...");
        cleanAndExit(); // 清理资源并退出
    }

    public void runTest()
    {
        System.out.println("Run Test");
        CaseMenu menu = new CaseMenu();
        menu.addItem(new CaseMenu.Item(this, "订阅按图索骥物品事件", "AttachEventRealLoadPic"));
        menu.addItem(new CaseMenu.Item(this, "退订按图索骥物品事件", "DetachEventRealLoadPic"));
        menu.run();
    }


    /**
     * 订阅智能任务
     */
    public void AttachEventRealLoadPic() {
        // 先退订，设备不会对重复订阅作校验，重复订阅后会有重复的事件返回
        this.DetachEventRealLoadPic();
        // 需要图片
        int bNeedPicture = 1;
        //EVENT_IVS_ALL 表示订阅所有智能事件
        m_attachHandle = netsdk.CLIENT_RealLoadPictureEx(loginHandle, channelId, EVENT_IVS_ALL, bNeedPicture, AnalyzerDataCB.getInstance(), null, null);
        /**
         * // EVENT_IVS_SAME_OBJECT_SEARCH_DETECT 按图索骥物品检测事件
         * // EVENT_IVS_SAME_OBJECT_SEARCH_COUNT 按图索骥物品计数事件
        **/
        if (m_attachHandle.longValue() != 0) {
            System.out.printf("Chn[%d] CLIENT_RealLoadPictureEx Success\n", channelId);
        } else {
            System.out.printf("Ch[%d] CLIENT_RealLoadPictureEx Failed!LastError = %s\n", channelId,
                    ToolKits.getErrorCode());
        }
    }


    /**
     * 报警事件（智能）回调
     */
    private static class AnalyzerDataCB implements NetSDKLib.fAnalyzerDataCallBack {
        private final File picturePath;
        private static AnalyzerDataCB instance;

        private AnalyzerDataCB() {
            picturePath = new File("./AnalyzerPicture/");
            if (!picturePath.exists()) {
                picturePath.mkdirs();
            }
        }

        public static AnalyzerDataCB getInstance() {
            if (instance == null) {
                synchronized (AnalyzerDataCB.class) {
                    if (instance == null) {
                        instance = new AnalyzerDataCB();
                    }
                }
            }
            return instance;
        }

        @Override
        public int invoke(NetSDKLib.LLong lAnalyzerHandle, int dwAlarmType, Pointer pAlarmInfo, Pointer pBuffer, int dwBufSize,
                          Pointer dwUser, int nSequence, Pointer reserved) {
            if (lAnalyzerHandle == null || lAnalyzerHandle.longValue() == 0) {
                return -1;
            }

            switch (dwAlarmType) {
                case EVENT_IVS_SAME_OBJECT_SEARCH_DETECT : {// 按图索骥物品检测事件(对应 NET_DEV_EVENT_SAME_OBJECT_SEARCH_DETECT_INFO)
                    System.out.println("按图索骥物品检测事件");
                    NET_DEV_EVENT_SAME_OBJECT_SEARCH_DETECT_INFO msg = new NET_DEV_EVENT_SAME_OBJECT_SEARCH_DETECT_INFO();
                    ToolKits.GetPointerData(pAlarmInfo, msg);

                    // 通道号
                    System.out.println("nChannelID:" + msg.nChannelID);

                    // 0:脉冲 1:开始 2:停止
                    System.out.println("nAction:" + msg.nAction);

                    /**
                     事件发生的时间
                     */
                    NET_TIME_EX stuUTC = msg.stuUTC;
                    System.out.println("stuUTC:" + stuUTC.toStringTime());
                    /**
                     事件编号，用来唯一标志一个事件
                     */
                    int nEventID = msg.nEventID;
                    System.out.println("nEventID:" + nEventID);

                    /**
                     事件名称
                     */
                    byte[] szName = msg.szName;
                    try {
                        System.out.println("szName:" + new String(szName,encode));
                    } catch (UnsupportedEncodingException e) {
                        e.printStackTrace();
                    }

                    /**
                     * 智能事件所属大类
                     */
                    System.out.println("szClass:" + new String(msg.szClass));

                    /**
                     * 相对事件时间戳,单位毫秒
                     */
                    System.out.println("dnPTS:" + msg.dbPTS);

                    /**
                     * 事件组ID，同一辆车抓拍过程内GroupID相同
                     */
                    System.out.println("nGroupID:" + msg.nGroupID);

                    /**
                     * CountInGroup一个事件组内应有的抓拍张数
                     */
                    System.out.println("nCountInGroup:" + msg.nCountInGroup);

                    /**
                     * 一个事件组内的抓拍序号，从1开始
                     */
                    System.out.println("nIndexInGroup:" + msg.nIndexInGroup);

                    /**
                     * 物体信息列表数组有效个数
                     */
                    System.out.println("nObjectsNum:" + msg.nObjectsNum);

                    /**
                     * 物体信息列表
                     */
//                    for(int i = 0;i < msg.stuObjects.length;i++){
//                        System.out.println("stuObjects[" + i +"].szText:" + new String(msg.stuObjects[i].szText));// 物体上相关的带0结束符文本,比如车牌,集装箱号等等
//                        System.out.println("stuObjects[" + i +"].wBrandYear:" + msg.stuObjects[i].wBrandYear);// 车辆品牌年款 需要通过映射表得到真正的年款 映射表详见开发手册
//                    }

                    try {
                        System.out.println("stuObjects[" + 0 +"].szText:" + new String(msg.stuObjects[0].szText,encode));// 物体上相关的带0结束符文本,比如车牌,集装箱号等等
                        System.out.println("stuObjects[" + 0 +"].wBrandYear:" +  msg.stuObjects[0].wBrandYear);// 车辆品牌年款 需要通过映射表得到真正的年款 映射表详见开发手册
                    }catch (UnsupportedEncodingException e){
                        e.printStackTrace();
                    }
                    /**
                     * 算法名称
                     */
                    byte[] szAlgName = msg.szAlgName;
                    try {
                        System.out.println("szAlgName:" + new String(szAlgName, encode));
                    }catch (UnsupportedEncodingException e){
                        e.printStackTrace();
                    }
                    /**
                     * 物品类型
                     */
                    byte[] szObjectType = msg.szObjectType;
                    try {
                        System.out.println("szObjectType:" + new String(szObjectType, encode));
                    }catch (UnsupportedEncodingException e){
                        e.printStackTrace();
                    }

                    /**
                     * 全景图
                     */
                    System.out.println("stuImageInfo.nIndexInData:" + msg.stuImageInfo.nIndexInData);// 在上传图片数据中的图片序号

                    /**
                     * 自定义报警ID
                     */
                    System.out.println("nAlarmId:" + msg.nAlarmId);

                    /**
                     * ObjectType物品类型id
                     */
                    System.out.println("nObjectTypeID:" + msg.nObjectTypeID);

                    /**
                     * 规则使用的按图索骥算法id
                     */
                    System.out.println("nAlgId:" + msg.nAlgId);
                    System.out.println("--------------------------");
                    break;
                }
                case EVENT_IVS_SAME_OBJECT_SEARCH_COUNT : {// 按图索骥物品计数事件(对应 NET_DEV_EVENT_SAME_OBJECT_SEARCH_COUNT_INFO)
                    System.out.println("按图索骥物品计数事件");
                    NET_DEV_EVENT_SAME_OBJECT_SEARCH_COUNT_INFO msg = new NET_DEV_EVENT_SAME_OBJECT_SEARCH_COUNT_INFO();
                    ToolKits.GetPointerData(pAlarmInfo, msg);

                    /**
                     * 通道号
                     */
                    System.out.println("nChannelID:"+ msg.nChannelID);

                    /**
                     * 0:脉冲,1:开始, 2:停止
                     */
                    System.out.println("nAction:"+ msg.nAction);

                    /**
                     事件发生的时间
                     */
                    NET_TIME_EX stuUTC = msg.stuUTC;
                    System.out.println("stuUTC:"+ stuUTC.toStringTime());

                    /**
                     事件名称
                     */
                    byte[] szName = msg.szName;
                    try {
                        System.out.println("szName:"+ new String(szName,encode));
                    } catch (UnsupportedEncodingException e) {
                        e.printStackTrace();
                    }

                    /**
                     * 智能事件所属大类
                     */
                    try {
                        System.out.println("szClass:" + new String(msg.szClass,encode));
                    }catch (UnsupportedEncodingException e){
                        e.printStackTrace();
                    }

                    /**
                     * 相对事件时间戳,单位毫秒
                     */
                    System.out.println("dnPTS:" + msg.dbPTS);

                    /**
                     * / 事件时间毫秒数
                     */
                    System.out.println("nUTCMS:" + msg.nUTCMS);

                    /**
                     事件编号，用来唯一标志一个事件
                     */
                    int nEventID = msg.nEventID;
                    System.out.println("nEventID:"+ nEventID);

                    /**
                     * GroupID事件组ID，同一物体抓拍过程内GroupID相同
                     */
                    System.out.println("nGroupID:" + msg.nGroupID);

                    /**
                     * GroupID事件组ID，同一物体抓拍过程内GroupID相同
                     */
                    System.out.println("nGroupID:" + msg.nGroupID);

                    /**
                     * CountInGroup一个事件组内的抓拍张数
                     */
                    System.out.println("nCountInGroup:" + msg.nCountInGroup);

                    /**
                     * IndexInGroup一个事件组内的抓拍序号，从1开始
                     */
                    System.out.println("nIndexInGroup:" + msg.nIndexInGroup);

                    /**
                     * 物体信息列表实际数量
                     */
                    System.out.println("nObjectsNum:" + msg.nObjectsNum);

                    /**
                     * 物体信息列表，最大支持64个物体
                     */
//                    for(int i = 0;i < msg.stObjects.length;i++){
//                        System.out.println("stObjects[" + i +"].szText:" +msg.stObjects[i].szText);//物体上相关的带0结束符文本,比如车牌,集装箱号等等
//                        System.out.println("stObjects[" + i +"].szSubText:" +msg.stObjects[i].szSubText);//"ObjectType"为"Vehicle"或者"Logo"时,表示车标下的某一车系,比如奥迪A6L,由于车系较多,SDK实现时透传此字段,设备如实填写。
//                    }
                    try {
                        System.out.println("stObjects[" + 0 +"].szText:" + new String(msg.stObjects[0].szText,encode));//物体上相关的带0结束符文本,比如车牌,集装箱号等等
                        System.out.println("stObjects[" + 0 +"].szSubText:" + new String(msg.stObjects[0].szSubText,encode));//"ObjectType"为"Vehicle"或者"Logo"时,表示车标下的某一车系,比如奥迪A6L,由于车系较多,SDK实现时透传此字段,设备如实填写。
                    }catch (UnsupportedEncodingException e){
                        e.printStackTrace();
                    }


                    /**
                     * 算法名称
                     */
                    try {
                        System.out.println("szAlgName:" + new String(msg.szAlgName,encode));
                    }catch (UnsupportedEncodingException e){
                        e.printStackTrace();
                    }


                    /**
                     * 物品类型
                     */
                    try {
                        System.out.println("szObjectType:" + new String(msg.szObjectType,encode));
                    }catch (UnsupportedEncodingException e){
                        e.printStackTrace();
                    }


                    /**
                     * 自定义报警ID
                     */
                    System.out.println("nAlarmId:" + msg.nAlarmId);

                    /**
                     * 物体个数
                     */
                    System.out.println("nCount:" + msg.nCount);
                    System.out.println("--------------------------");
                    break;
                }
                default:
                    System.out.println("其他事件--------------------"+ dwAlarmType);
                    break;
            }
            return 0;
        }
    }


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


    public static void main(String[] args) {
        SameObjectSearchDemo demo=new SameObjectSearchDemo();
        demo.initTest();
        demo.runTest();
        demo.endTest();
    }
}