package com.netsdk.demo.customize.heatmap;
|
|
import com.netsdk.demo.util.CaseMenu;
|
import com.netsdk.lib.NetSDKLib;
|
import com.netsdk.lib.ToolKits;
|
import com.netsdk.lib.callback.impl.DefaultDisconnectCallback;
|
import com.netsdk.lib.callback.impl.DefaultHaveReconnectCallBack;
|
import com.netsdk.lib.callback.impl.DefaultVideoStatHeatMapCallBack;
|
import com.netsdk.lib.enumeration.EM_HEATMAP_TYPE;
|
import com.netsdk.lib.enumeration.ENUMERROR;
|
import com.netsdk.lib.structure.NET_IN_GET_HEATMAPS_INFO;
|
import com.netsdk.lib.structure.NET_OUT_GET_HEATMAPS_INFO;
|
import com.netsdk.module.BaseModule;
|
import com.netsdk.module.HeatMapModule;
|
import com.netsdk.module.entity.BmpInfo;
|
import com.netsdk.module.entity.DeviceInfo;
|
import com.netsdk.module.entity.HeatMapGrayData;
|
import com.sun.jna.Memory;
|
import com.sun.jna.Pointer;
|
import com.sun.jna.ptr.IntByReference;
|
|
import javax.imageio.ImageIO;
|
import java.awt.image.BufferedImage;
|
import java.io.*;
|
|
/**
|
* @author 47081
|
* @version 1.0
|
* @description 热度图数据合成到背景图上
|
* @date 2020/9/25
|
*/
|
public class HeatMapDemo {
|
|
private NetSDKLib netSdk = NetSDKLib.NETSDK_INSTANCE;
|
/** 二次封装的基础模块,包含初始化,登录,登出 */
|
private BaseModule baseModule;
|
/** 二次封装的热度图模块，包含订阅，获取，退订等 */
|
private HeatMapModule heatMapModule;
|
|
private long loginHandler;
|
private long attachHeatMapHandler;
|
/** 这个static只是为了方便在抓图回调里获取灰度图数据 */
|
public static int token;
|
|
private boolean isSetSnapCallBack = false;
|
|
public HeatMapDemo() {
|
this.baseModule = new BaseModule(netSdk);
|
this.heatMapModule = new HeatMapModule(netSdk);
|
}
|
|
/**
|
* sdk初始化
|
*
|
* @return
|
*/
|
public boolean init() {
|
return baseModule.init(
|
DefaultDisconnectCallback.getINSTANCE(), DefaultHaveReconnectCallBack.getINSTANCE(), true);
|
}
|
|
/**
|
* 登录设备
|
*
|
* @param ip 设备ip
|
* @param port 设备端口
|
* @param userName 用户名
|
* @param password 密码
|
* @return
|
*/
|
public boolean login(String ip, int port, String userName, String password) {
|
DeviceInfo info = baseModule.login(ip, port, userName, password);
|
if (info == null || info.getLoginHandler() == 0) {
|
loginHandler = 0;
|
return false;
|
}
|
loginHandler = info.getLoginHandler();
|
System.out.println("Login success! LoginHandler is " + loginHandler);
|
return true;
|
}
|
|
/**
|
* sdk清理资源
|
*
|
* @return
|
*/
|
public void clean() {
|
baseModule.clean();
|
}
|
|
/** 订阅热度图数据 */
|
public boolean attachHeatMap() {
|
attachHeatMapHandler =
|
heatMapModule.attach(
|
loginHandler, 0, DefaultVideoStatHeatMapCallBack.getInstance(), null, 3000);
|
System.err.println("订阅：" + ToolKits.getErrorCode() + " 错误msg：" + ENUMERROR.getErrorMessage());
|
return attachHeatMapHandler != 0;
|
}
|
|
/**
|
* 退订热度图数据
|
*
|
* @return
|
*/
|
public boolean detach() {
|
return heatMapModule.detach(attachHeatMapHandler);
|
}
|
|
/**
|
* 获取热度图数据
|
*
|
* @return
|
*/
|
public boolean get() {
|
token =
|
heatMapModule.get(
|
attachHeatMapHandler,
|
0,
|
"2020/9/27/19/0/0",
|
"2020/9/27/20/00/00",
|
EM_HEATMAP_TYPE.EM_HEATMAP_HUMANSTATISTICS,
|
3000);
|
return token != 0;
|
}
|
|
/** 热度图合成到背景图上 */
|
public void mergeBmpWithFile() {
|
// 读取bmp文件
|
try {
|
BufferedImage bmp = ImageIO.read(new File("D:/1.bmp"));
|
ByteArrayOutputStream byteArrOutput = new ByteArrayOutputStream();
|
// 此方式转换出来的位深度是24位
|
ImageIO.write(bmp, "bmp", byteArrOutput);
|
////////////////// 获取背景图的缓存、宽高 /////////////
|
byte[] buffer = byteArrOutput.toByteArray(); // bmp格式的Buf
|
int width = bmp.getWidth();
|
int height = bmp.getHeight();
|
int bitCount = bmp.getColorModel().getPixelSize();
|
// 54是头
|
int nBackPicLen = width * height * 24 / 8 + 54;
|
BmpInfo backInfo = new BmpInfo(buffer, width, height, bitCount, 1, nBackPicLen);
|
HeatMapGrayData grayData = DefaultVideoStatHeatMapCallBack.grayDatas.get(token);
|
BmpInfo grayInfo =
|
new BmpInfo(
|
grayData.getData(),
|
grayData.getWidth(),
|
grayData.getHeight(),
|
grayData.getnBit(),
|
0,
|
grayData.getLength());
|
heatMapModule.createHeatMap(grayInfo, backInfo, "D:/3.bmp", 0.5f);
|
byteArrOutput.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
|
/** 远程抓图叠加热度图 */
|
public void mergeBmpWithSnap() {
|
if (!isSetSnapCallBack) {
|
// 设置抓图回调
|
netSdk.CLIENT_SetSnapRevCallBack(SnapPicHeatMapCallBack.getInstance(), null);
|
}
|
// 发送抓图命令给前端设备，抓图的信息
|
NetSDKLib.SNAP_PARAMS stuSnapParams = new NetSDKLib.SNAP_PARAMS();
|
// 抓图通道
|
stuSnapParams.Channel = 0;
|
// 表示请求一帧
|
stuSnapParams.mode = 0;
|
stuSnapParams.Quality = 3;
|
stuSnapParams.InterSnap = 5;
|
// 请求序列号，有效值范围 0~65535，超过范围会被截断为unsigned short
|
stuSnapParams.CmdSerial = 100;
|
|
IntByReference reserved = new IntByReference(0);
|
// 热度图叠加到背景图在抓图回调SnapPicHeatMapCallBack.java里完成
|
if (!netSdk.CLIENT_SnapPictureEx(new NetSDKLib.LLong(loginHandler), stuSnapParams, reserved)) {
|
System.out.println("snap picture failed. the error is " + ENUMERROR.getErrorMessage());
|
} else {
|
System.out.println("snap picture success.");
|
}
|
}
|
|
/**
|
* byte to int
|
*
|
* @param b bmp
|
* @return
|
* @throws IOException
|
*/
|
private int byte2Int(byte[] b) throws IOException {
|
int num = (b[3] & 0xff << 24) | (b[2] & 0xff) << 16 | (b[1] & 0xff) << 8 | (b[0] & 0xff);
|
return num;
|
}
|
|
/**
|
* 登出设备
|
*
|
* @return
|
*/
|
public boolean logout() {
|
boolean result = baseModule.logout(loginHandler);
|
if (!result) {
|
System.out.println("logout failed. the error is " + ENUMERROR.getErrorMessage());
|
}
|
return result;
|
}
|
|
public void runTest() {
|
CaseMenu menu = new CaseMenu();
|
menu.addItem(new CaseMenu.Item(this, "订阅热度图", "attachHeatMap"));
|
menu.addItem(new CaseMenu.Item(this, "获取热度图数据", "get"));
|
menu.addItem(new CaseMenu.Item(this, "热度图叠加背景图", "mergeBmpWithFile"));
|
menu.addItem(new CaseMenu.Item(this, "远程抓图叠加热度图", "mergeBmpWithSnap"));
|
menu.addItem(new CaseMenu.Item(this, "退订热度图", "detach"));
|
menu.addItem(new CaseMenu.Item(this, "热度图温度", "getHeatMapDirectlyTest"));
|
menu.run();
|
}
|
|
int channel = 0;
|
|
public void getHeatMapDirectlyTest() {
|
// 获取热成像温度的入参
|
NET_IN_GET_HEATMAPS_INFO inParam = new NET_IN_GET_HEATMAPS_INFO();
|
inParam.nChannel = channel;
|
NET_OUT_GET_HEATMAPS_INFO outParam = new NET_OUT_GET_HEATMAPS_INFO();
|
/** dwMaxDataBufLen: The allocated memory should be larger. */
|
outParam.dwMaxDataBufLen = 1024 * 1024;
|
outParam.pbDataBuf = new Memory(outParam.dwMaxDataBufLen);
|
outParam.pbDataBuf.clear(outParam.dwMaxDataBufLen);
|
Pointer pInParam = new Memory(inParam.dwSize);
|
ToolKits.SetStructDataToPointer(inParam, pInParam, 0);
|
Pointer pOutParam = new Memory(outParam.size());
|
ToolKits.SetStructDataToPointer(outParam, pOutParam, 0);
|
boolean result =
|
netSdk.CLIENT_GetHeatMapsDirectly(
|
new NetSDKLib.LLong(loginHandler), pInParam, pOutParam, 3000);
|
if (result) {
|
System.out.println("GetHeatMapsDirectly success！");
|
ToolKits.GetPointerData(pOutParam, outParam);
|
outParam.read();
|
System.out.println(outParam.dwRetDataBufLen);
|
System.out.println(outParam);
|
|
} else {
|
System.err.println(
|
"GetHeatMapsDirectly false!"
|
+ ENUMERROR.getErrorMessage()
|
+ " SDK error type is:"
|
+ ENUMERROR.getENUMError());
|
}
|
}
|
|
public static void main(String[] args) {
|
String ipsr = "172.11.2.5";
|
// 设备ip 172.32.102.46 、172.32.102.57
|
String ip = "172.32.102.88";
|
int port = 37777;
|
String username = "admin";
|
String password = "admin123";
|
HeatMapDemo demo = new HeatMapDemo();
|
demo.init();
|
if (demo.login(ip, port, username, password)) {
|
/**
|
* 热度图调用顺序： 1.订阅数据, 2.获取数据 3. 设备传上来热度图数据,在回调中获取到数据,进行业务处理 4.业务处理完成，退订数据
|
*
|
* <p>demo使用顺序: main方法启动,登录设备: 输入1订阅热度图数据 输入2获取热度图数据,查看控制台输出是否有热度图数据
|
* 输入3可将热度图数据叠加到.bmp文件上(demo中是D:/1.bmp)，输出一个新的bmp文件 输入4可远程抓一张图片,将热度图数据叠加到抓图上,输出一个新的bmp文件 输入5退订
|
* 输入6热度图温度 输入0结束demo
|
*/
|
demo.runTest();
|
demo.logout();
|
}
|
demo.clean();
|
}
|
}
|
