package com.netsdk.demo.customize.heatmap;
|
|
import java.awt.image.BufferedImage;
|
import java.io.ByteArrayInputStream;
|
import java.io.ByteArrayOutputStream;
|
import java.io.File;
|
import java.io.IOException;
|
|
import javax.imageio.ImageIO;
|
import javax.swing.JOptionPane;
|
|
import com.netsdk.demo.util.CaseMenu;
|
import com.netsdk.lib.HeatMapLib;
|
import com.netsdk.lib.NetSDKLib;
|
import com.netsdk.lib.ToolKits;
|
import com.netsdk.lib.HeatMapLib.*;
|
import com.netsdk.lib.NetSDKLib.*;
|
import com.sun.jna.Memory;
|
import com.sun.jna.Pointer;
|
import com.sun.jna.ptr.IntByReference;
|
|
/**
|
* 生成热度图Demo，没有界面的，通过调用转换库
|
*/
|
public class HeatMapEx2 {
|
|
static NetSDKLib NetSdk = NetSDKLib.NETSDK_INSTANCE;
|
static NetSDKLib ConfigSdk = NetSDKLib.CONFIG_INSTANCE;
|
static HeatMapLib HeatMapSdk = HeatMapLib.HEATMAP_INSTANCE;
|
|
//登陆参数
|
private String m_strIp = "172.29.4.55";
|
private Integer m_nPort = new Integer("37777");
|
private String m_strUser = "admin";
|
private String m_strPassword = "admin123";
|
|
//设备信息
|
private NetSDKLib.NET_DEVICEINFO_Ex m_stDeviceInfo = new NetSDKLib.NET_DEVICEINFO_Ex();
|
|
private static LLong m_hLoginHandle = new LLong(0); //登陆句柄
|
|
private DisConnect disConnect = new DisConnect(); //设备断线通知回调
|
private HaveReConnect haveReConnect = new HaveReConnect(); //网络连接恢复
|
|
public static GRAY_MAP grayData = null;
|
|
//设备断线回调: 通过 CLIENT_Init 设置该回调函数，当设备出现断线时，SDK会调用该函数
|
public class DisConnect implements NetSDKLib.fDisConnect {
|
public void invoke(LLong m_hLoginHandle, String pchDVRIP, int nDVRPort, Pointer dwUser) {
|
System.out.printf("Device[%s] Port[%d] DisConnect!\n", pchDVRIP, nDVRPort);
|
}
|
}
|
|
//网络连接恢复，设备重连成功回调
|
// 通过 CLIENT_SetAutoReconnect 设置该回调函数，当已断线的设备重连成功时，SDK会调用该函数
|
public class HaveReConnect implements NetSDKLib.fHaveReConnect {
|
public void invoke(LLong m_hLoginHandle, String pchDVRIP, int nDVRPort, Pointer dwUser) {
|
System.out.printf("ReConnect Device[%s] Port[%d]\n", pchDVRIP, nDVRPort);
|
}
|
}
|
|
public void EndTest()
|
{
|
System.out.println("End Test");
|
if( m_hLoginHandle.longValue() != 0)
|
{
|
NetSdk.CLIENT_Logout(m_hLoginHandle);
|
}
|
System.out.println("See You...");
|
|
NetSdk.CLIENT_Cleanup();
|
System.exit(0);
|
}
|
|
public void InitTest()
|
{
|
//初始化SDK库
|
NetSdk.CLIENT_Init(disConnect, null);
|
|
// 设置断线重连回调接口，设置过断线重连成功回调函数后，当设备出现断线情况，SDK内部会自动进行重连操作
|
// 此操作为可选操作，但建议用户进行设置
|
NetSdk.CLIENT_SetAutoReconnect(haveReConnect, null);
|
|
//设置登录超时时间和尝试次数，可选
|
int waitTime = 5000; //登录请求响应超时时间设置为5S
|
int tryTimes = 3; //登录时尝试建立链接3次
|
NetSdk.CLIENT_SetConnectTime(waitTime, tryTimes);
|
|
// 设置更多网络参数，NET_PARAM的nWaittime，nConnectTryNum成员与CLIENT_SetConnectTime
|
// 接口设置的登录设备超时时间和尝试次数意义相同,可选
|
NetSDKLib.NET_PARAM netParam = new NetSDKLib.NET_PARAM();
|
netParam.nConnectTime = 10000; //登录时尝试建立链接的超时时间
|
NetSdk.CLIENT_SetNetworkParam(netParam);
|
|
// 打开日志，可选
|
NetSDKLib.LOG_SET_PRINT_INFO setLog = new NetSDKLib.LOG_SET_PRINT_INFO();
|
|
File path = new File(".");
|
String logPath = path.getAbsoluteFile().getParent() + File.separator + "sdklog";
|
|
File file = new File(logPath);
|
if (!file.exists()) {
|
file.mkdir();
|
}
|
|
logPath = file + File.separator + "123456789.log";
|
|
System.out.println(logPath);
|
setLog.bSetFilePath = 1;
|
System.arraycopy(logPath.getBytes(), 0, setLog.szLogFilePath, 0, logPath.getBytes().length);
|
|
for (int i = 0; i < setLog.szLogFilePath.length; i++) {
|
System.out.print((char)setLog.szLogFilePath[i] + " ");
|
}
|
System.out.println();
|
setLog.bSetPrintStrategy = 1;
|
setLog.nPrintStrategy = 0;
|
boolean bLogopen = NetSdk.CLIENT_LogOpen(setLog);
|
if (!bLogopen) {
|
System.err.println("Failed to open NetSDK log !!!");
|
}
|
|
// 向设备登入
|
int nSpecCap = 0;
|
Pointer pCapParam = null;
|
IntByReference nError = new IntByReference(0);
|
m_hLoginHandle = NetSdk.CLIENT_LoginEx2(m_strIp, m_nPort, m_strUser ,
|
m_strPassword ,nSpecCap,pCapParam, m_stDeviceInfo,nError);
|
|
if(m_hLoginHandle.longValue() != 0) {
|
System.out.printf("Login Device[%s] Port[%d]Success!\n" , m_strIp , m_nPort);
|
}
|
else {
|
System.out.printf("Login Device[%s] Port[%d]Fail.Last Error[0x%x]\n" , m_strIp , m_nPort , NetSdk.CLIENT_GetLastError());
|
// EndTest();
|
}
|
|
}
|
|
// 远程抓图
|
public void snapPicture(int chn)
|
{
|
//设置抓图回调函数， 图片主要在m_SnapReceiveCB中返回
|
NetSdk.CLIENT_SetSnapRevCallBack(fSnapReceiveCB.getInstance(), null);
|
|
// 发送抓图命令给前端设备，抓图的信息
|
NetSDKLib.SNAP_PARAMS stuSnapParams = new NetSDKLib.SNAP_PARAMS() ;
|
stuSnapParams.Channel = chn; //抓图通道
|
stuSnapParams.mode = 0; //表示请求一帧
|
stuSnapParams.Quality = 3;
|
stuSnapParams.InterSnap = 5;
|
stuSnapParams.CmdSerial = 100; // 请求序列号，有效值范围 0~65535，超过范围会被截断为
|
|
IntByReference reserved = new IntByReference(0);
|
|
if (false == NetSdk.CLIENT_SnapPictureEx(m_hLoginHandle, stuSnapParams, reserved)) {
|
System.err.println("CLIENT_SnapPictureEx Failed!" + ToolKits.getErrorCode());
|
return;
|
} else {
|
System.out.println("CLIENT_SnapPictureEx success");
|
}
|
}
|
|
public static class fSnapReceiveCB implements NetSDKLib.fSnapRev{
|
private fSnapReceiveCB() {}
|
|
private static class fSnapReceiveCBHolder {
|
private static fSnapReceiveCB instance = new fSnapReceiveCB();
|
}
|
|
private static fSnapReceiveCB getInstance() {
|
return fSnapReceiveCBHolder.instance;
|
}
|
public void invoke( LLong lLoginID, Pointer pBuf, int RevLen, int EncodeType, int CmdSerial, Pointer dwUser)
|
{
|
////////////////////// 读取抓图的数据(格式JPG) ///////////////////////////
|
System.out.println("正在获取背景图数据...");
|
|
// jpg格式的Buf
|
byte[] buf = pBuf.getByteArray(0, RevLen);
|
|
ByteArrayInputStream byteArrInput = new ByteArrayInputStream(buf);
|
try {
|
BufferedImage bufferedImage = ImageIO.read(byteArrInput);
|
if(bufferedImage == null) {
|
return;
|
}
|
|
ByteArrayOutputStream byteArrOutput = new ByteArrayOutputStream();
|
try {
|
// 此方式转换出来的位深度为 24
|
ImageIO.write(bufferedImage, "bmp", byteArrOutput);
|
} catch (IOException e1) {
|
e1.printStackTrace();
|
}
|
|
////////////////// 获取背景图的缓存、宽高 /////////////
|
byte[] buffer = byteArrOutput.toByteArray(); // bmp格式的Buf
|
int width = bufferedImage.getWidth();
|
int height = bufferedImage.getHeight();
|
|
new MyThread(buffer, width, height).start();
|
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
|
|
private static class MyThread extends Thread {
|
byte[] bBackBuf;;
|
int nBackWidth;
|
int nBackHeight;
|
public MyThread(byte[] buffer, int width, int height) {
|
this.bBackBuf = buffer;
|
this.nBackWidth = width;
|
this.nBackHeight = height;
|
}
|
|
@Override
|
public void run() {
|
System.out.println("正在查询灰度图数据...");
|
////////////////// 查询灰度图数据 //////////////////////////
|
if(queryHeatMap(0, "2022-05-06 14:00:0", "2022-05-06 17:00:00")) {
|
|
System.out.println("正在生成热度图...");
|
/**
|
* 入参
|
*/
|
HEATMAP_IMAGE_IN stIn = new HEATMAP_IMAGE_IN();
|
/////////////////////// 灰度图, 不带头 ////////////////////////
|
// 获取到的灰度数据的位深度是8， 位深度为8时，需要加 1024
|
|
// 灰度图大小，不带头
|
int nGrayLen = 0;
|
|
// 位深度
|
int nBit = grayData.getGrayBufLen() / ( grayData.getGrayWidth() * grayData.getGrayHeight() ) * 8;
|
|
if(nBit == 8) {
|
nGrayLen = grayData.getGrayBufLen() * 8/8;
|
} else {
|
nGrayLen = grayData.getGrayWidth() * grayData.getGrayHeight() * nBit / 8;
|
}
|
|
// 申请内存
|
stIn.stuGrayBmpInfo.pBuffer = new Memory(nGrayLen);
|
stIn.stuGrayBmpInfo.pBuffer.clear(nGrayLen);
|
|
// 灰度图赋值
|
stIn.stuGrayBmpInfo.pBuffer = grayData.getGratBufData();
|
stIn.stuGrayBmpInfo.nWidth = grayData.getGrayWidth();
|
stIn.stuGrayBmpInfo.nHeight = grayData.getGrayHeight();
|
stIn.stuGrayBmpInfo.nBitCount = nBit;
|
stIn.stuGrayBmpInfo.nDirection = 0;
|
|
////////////////////////// 背景图, 带头 //////////////////////////////
|
// 转换出来的背景图位深度是24
|
// 54是头
|
int nBackPicLen = nBackWidth * nBackHeight * 24/8 + 54;
|
|
stIn.stuBkBmpInfo.pBuffer = new Memory(nBackPicLen);
|
stIn.stuBkBmpInfo.pBuffer.clear(nBackPicLen);
|
|
stIn.stuBkBmpInfo.pBuffer.write(0, bBackBuf, 0, nBackPicLen);
|
stIn.stuBkBmpInfo.nWidth = nBackWidth;
|
stIn.stuBkBmpInfo.nHeight = nBackHeight;
|
stIn.stuBkBmpInfo.nBitCount = 24;
|
stIn.stuBkBmpInfo.nDirection = 1;
|
|
/**
|
* 出参
|
*/
|
//////////////////////// 热度图, 带头 //////////////////////////
|
HEATMAP_IMAGE_Out stOut = new HEATMAP_IMAGE_Out();
|
stOut.pBuffer = new Memory(nBackPicLen);
|
stOut.pBuffer.clear(nBackPicLen);
|
|
stOut.nPicSize = nBackPicLen; // 热度图的大小与背景图相同
|
stOut.fOpacity = 0.2F; // 透明度
|
|
// 生成热度图
|
if(HeatMapSdk.CreateHeatMap(stIn, stOut)) {
|
File currentPath = new File(".");
|
String strFileSnapPic = currentPath.getAbsoluteFile().getParent() + "\\" + "HeatMap.jpg";
|
|
byte[] buf = stOut.pBuffer.getByteArray(0, nBackPicLen);
|
ByteArrayInputStream bInputStream = new ByteArrayInputStream(buf);
|
|
try {
|
BufferedImage bufferedImage = ImageIO.read(bInputStream);
|
if(bufferedImage == null) {
|
System.err.println("bufferedImage == null");
|
return;
|
}
|
|
// 写文件，生成图片
|
ImageIO.write(bufferedImage, "jpg", new File(strFileSnapPic));
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
System.out.println("已生成热度图...");
|
} else {
|
System.err.println("失败！！！");
|
}
|
}
|
}
|
}
|
|
// 查询热度图
|
public static boolean queryHeatMap(int chn, String startTime, String endTime) {
|
// 开始时间
|
String[] startStr = startTime.split(" ");
|
String[] startTimeStr1 = startStr[0].split("-");
|
String[] startTimeStr2 = startStr[1].split(":");
|
|
// 结束时间
|
String[] endStr = endTime.split(" ");
|
String[] endTimeStr1 = endStr[0].split("-");
|
String[] endTimeStr2 = endStr[1].split(":");
|
|
|
NET_QUERY_HEAT_MAP heatMap = new NET_QUERY_HEAT_MAP();
|
heatMap.stuIn.nChannel = chn; // 通道
|
|
// 开始时间
|
heatMap.stuIn.stuBegin.dwYear = Integer.parseInt(startTimeStr1[0]);
|
heatMap.stuIn.stuBegin.dwMonth = Integer.parseInt(startTimeStr1[1]);
|
heatMap.stuIn.stuBegin.dwDay = Integer.parseInt(startTimeStr1[2]);
|
heatMap.stuIn.stuBegin.dwHour = Integer.parseInt(startTimeStr2[0]);
|
heatMap.stuIn.stuBegin.dwMinute = Integer.parseInt(startTimeStr2[1]);
|
heatMap.stuIn.stuBegin.dwSecond = Integer.parseInt(startTimeStr2[2]);
|
|
// 结束时间
|
heatMap.stuIn.stuEnd.dwYear = Integer.parseInt(endTimeStr1[0]);
|
heatMap.stuIn.stuEnd.dwMonth = Integer.parseInt(endTimeStr1[1]);
|
heatMap.stuIn.stuEnd.dwDay = Integer.parseInt(endTimeStr1[2]);
|
heatMap.stuIn.stuEnd.dwHour = Integer.parseInt(endTimeStr2[0]);
|
heatMap.stuIn.stuEnd.dwMinute = Integer.parseInt(endTimeStr2[1]);
|
heatMap.stuIn.stuEnd.dwSecond = Integer.parseInt(endTimeStr2[2]);
|
|
// 指针申请内存
|
int size = 4000 * 4000;
|
heatMap.stuOut.pBufData = new Memory(size);
|
heatMap.stuOut.pBufData.clear(size);
|
|
heatMap.stuOut.nBufLen = size;
|
|
heatMap.stuOut.emDataType = 1; // 1-灰度数据 2-原始数据
|
|
IntByReference pRetLen = new IntByReference(0);
|
|
heatMap.write();
|
boolean bRet = NetSdk.CLIENT_QueryDevState(m_hLoginHandle, NetSDKLib.NET_DEVSTATE_GET_HEAT_MAP,
|
heatMap.getPointer(), heatMap.size(), pRetLen, 5000);
|
heatMap.read();
|
if(bRet) {
|
grayData = new GRAY_MAP();
|
grayData.setGratBufData(heatMap.stuOut.pBufData);
|
grayData.setGrayBufLen(heatMap.stuOut.nBufRet);
|
grayData.setGrayWidth(heatMap.stuOut.nWidth);
|
grayData.setGrayHeight(heatMap.stuOut.nHeight);
|
} else {
|
JOptionPane.showMessageDialog(null, "查到热度图失败， 错误码 ：" + ToolKits.getErrorCode());
|
System.err.println("Query HeatMap Failed!" + ToolKits.getErrorCode());
|
return false;
|
}
|
return true;
|
}
|
|
|
// 灰度图数据
|
static class GRAY_MAP {
|
public Pointer pGratBufData;
|
public int nGrayBufLen;
|
public int nGrayWidth;
|
public int nGrayHeight;
|
|
public Pointer getGratBufData() {
|
return pGratBufData;
|
}
|
public void setGratBufData(Pointer pGratBufData) {
|
this.pGratBufData = pGratBufData;
|
}
|
public int getGrayBufLen() {
|
return nGrayBufLen;
|
}
|
public void setGrayBufLen(int nGrayBufLen) {
|
this.nGrayBufLen = nGrayBufLen;
|
}
|
public int getGrayWidth() {
|
return nGrayWidth;
|
}
|
public void setGrayWidth(int nGrayWidth) {
|
this.nGrayWidth = nGrayWidth;
|
}
|
public int getGrayHeight() {
|
return nGrayHeight;
|
}
|
public void setGrayHeight(int nGrayHeight) {
|
this.nGrayHeight = nGrayHeight;
|
}
|
|
|
}
|
|
public void CreateHeatMap() {
|
snapPicture(0);
|
}
|
|
|
public void RunTest()
|
{
|
System.out.println("Run Test");
|
CaseMenu menu = new CaseMenu();
|
menu.addItem(new CaseMenu.Item(this , "获取热度图" , "CreateHeatMap"));
|
menu.run();
|
}
|
|
public static void main(String[]args)
|
{
|
HeatMapEx2 demo = new HeatMapEx2();
|
|
demo.InitTest();
|
demo.RunTest();
|
demo.EndTest();
|
|
|
}
|
}
|
