package com.ruoyi.HIKVISION;
|
|
import com.ruoyi.HIKVISION.NetSDK.HCNetSDK;
|
import com.ruoyi.aibrain.domain.AiCamera;
|
import com.ruoyi.common.config.RuoYiConfig;
|
import com.ruoyi.common.constant.Constants;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.sun.jna.Memory;
|
import com.sun.jna.NativeLong;
|
import com.sun.jna.ptr.IntByReference;
|
|
import java.io.*;
|
import java.nio.ByteBuffer;
|
import java.nio.charset.StandardCharsets;
|
|
//import com.ruoyi.project.capture.HCNetSDK;
|
|
public class GetPicFromIPV {
|
private HCNetSDK hCNetSDK = HCNetSDK.INSTANCE;
|
|
Integer lUserID = -1;
|
|
public static void main(String[] args) {
|
final GetPicFromIPV getPicFromIPV = new GetPicFromIPV();
|
AiCamera device = new AiCamera();
|
device.setCameraIp("192.168.1.67");
|
device.setCameraPort("8000");
|
device.setCameraUser("admin");
|
device.setCameraPassword("a1234567");
|
device.setPicture("/profile/upload/2023/06/28/微信图片_20211124142412_20230628101845A001.png");
|
getPicFromIPV.getDVRPic(device, false);
|
|
}
|
|
/**
|
* 抓拍图片
|
*
|
* @param cate 是否走内存
|
* @param dvr 用户信息
|
*/
|
public String getDVRPic(AiCamera dvr, boolean cate) {
|
|
String msg = "200";
|
if (!hCNetSDK.NET_DVR_Init()) {
|
System.out.println("hksdk(抓图)-海康sdk初始化失败!");
|
return "hksdk(抓图)-海康sdk初始化失败!";
|
}
|
// 设备信息
|
HCNetSDK.NET_DVR_DEVICEINFO_V30 devinfo = new HCNetSDK.NET_DVR_DEVICEINFO_V30();
|
HCNetSDK.NET_DVR_USER_LOGIN_INFO m_strLoginInfo = new HCNetSDK.NET_DVR_USER_LOGIN_INFO();//设备登录信息
|
HCNetSDK.NET_DVR_DEVICEINFO_V40 m_strDeviceInfo = new HCNetSDK.NET_DVR_DEVICEINFO_V40();//设备信息
|
//注册设备
|
String m_sDeviceIP = dvr.getCameraIp();
|
String m_sUsername = dvr.getCameraUser();
|
String m_sPassword = dvr.getCameraPassword();
|
m_strLoginInfo.wPort = (short) Integer.parseInt("8000");
|
|
//注册
|
m_strLoginInfo.sDeviceAddress = new byte[HCNetSDK.NET_DVR_DEV_ADDRESS_MAX_LEN];
|
System.arraycopy(m_sDeviceIP.getBytes(), 0, m_strLoginInfo.sDeviceAddress, 0, m_sDeviceIP.length());
|
m_strLoginInfo.sUserName = new byte[HCNetSDK.NET_DVR_LOGIN_USERNAME_MAX_LEN];
|
System.arraycopy(m_sUsername.getBytes(), 0, m_strLoginInfo.sUserName, 0, m_sUsername.length());
|
m_strLoginInfo.sPassword = new byte[HCNetSDK.NET_DVR_LOGIN_PASSWD_MAX_LEN];
|
System.arraycopy(m_sPassword.getBytes(), 0, m_strLoginInfo.sPassword, 0, m_sPassword.length());
|
m_strLoginInfo.bUseAsynLogin = false; //是否异步登录：0- 否，1- 是
|
m_strLoginInfo.write();
|
lUserID = hCNetSDK.NET_DVR_Login_V40(m_strLoginInfo, m_strDeviceInfo);
|
// 返回一个用户编号，同时将设备信息写入devinfo
|
if (lUserID.intValue() < 0) {
|
System.out.println("hksdk(抓图)-设备注册失败,错误码:" + hCNetSDK.NET_DVR_GetLastError());
|
return "hksdk(抓图)-设备注册失败,错误码:" + hCNetSDK.NET_DVR_GetLastError();
|
}
|
HCNetSDK.NET_DVR_WORKSTATE_V30 devwork = new HCNetSDK.NET_DVR_WORKSTATE_V30();
|
if (!hCNetSDK.NET_DVR_GetDVRWorkState_V30(lUserID, devwork)) {
|
// 返回Boolean值，判断是否获取设备能力
|
System.out.println("hksdk(抓图)-返回设备状态失败" + hCNetSDK.NET_DVR_GetLastError());
|
msg = "hksdk(抓图)-返回设备状态失败" + hCNetSDK.NET_DVR_GetLastError();
|
}
|
// 本地资源路径
|
String localPath = RuoYiConfig.getProfile();
|
// 数据库资源地址
|
String path = localPath + StringUtils.substringAfter(dvr.getPicture(), Constants.RESOURCE_PREFIX);
|
// String path = "";
|
// path = imgPath + "-" + channelNo + ".jpeg";
|
// NativeLong chanLong = new NativeLong(channelNo);
|
//内存
|
if (cate) {
|
picCutCate(new NativeLong(lUserID.intValue()), 1, path);
|
} else {
|
//非内存直接保存
|
//图片质量
|
HCNetSDK.NET_DVR_JPEGPARA jpeg = new HCNetSDK.NET_DVR_JPEGPARA();
|
//设置图片分辨率
|
jpeg.wPicSize = 0;
|
//设置图片质量
|
jpeg.wPicQuality = 0;
|
//需要加入通道
|
System.out.println("-----------这里开始封装 NET_DVR_CaptureJPEGPicture---------");
|
boolean is = hCNetSDK.NET_DVR_CaptureJPEGPicture(lUserID, 1, jpeg, path.getBytes(StandardCharsets.UTF_8));
|
System.out.println("-----------抓图工具返回结果----------" + is);
|
if (!is) {
|
System.out.println("hksdk(抓图)-抓取失败,错误码:" + hCNetSDK.NET_DVR_GetLastError());
|
msg = "hksdk(抓图)-抓取失败,错误码:" + hCNetSDK.NET_DVR_GetLastError();
|
}
|
}
|
|
System.out.println("-----------处理完成截图数据----------");
|
//退出登录
|
hCNetSDK.NET_DVR_Logout(lUserID);
|
//释放SDK资源
|
hCNetSDK.NET_DVR_Cleanup();
|
return msg;
|
}
|
|
|
private void picCutCate(NativeLong lUserID, int chanLong, String imgPath) {
|
//图片质量
|
HCNetSDK.NET_DVR_JPEGPARA jpeg = new HCNetSDK.NET_DVR_JPEGPARA();
|
//设置图片分辨率
|
jpeg.wPicSize = 0;
|
//设置图片质量
|
jpeg.wPicQuality = 0;
|
IntByReference a = new IntByReference();
|
//设置图片大小
|
ByteBuffer jpegBuffer = ByteBuffer.allocate(1920 * 1080);
|
byte[] jpegBytes = new byte[1920 * 1080];
|
File file = new File(imgPath);
|
// 抓图到内存，单帧数据捕获并保存成JPEG存放在指定的内存空间中
|
System.out.println("-----------这里开始封装 NET_DVR_CaptureJPEGPicture_NEW---------");
|
// boolean is = hCNetSDK.NET_DVR_CaptureJPEGPicture_NEW(lUserID.intValue(), chanLong, jpeg, jpegBytes, jpegBytes.length, a);
|
boolean is = hCNetSDK.NET_DVR_CaptureJPEGPicture_NEW(lUserID.intValue(), chanLong, jpeg, new Memory(1920*1080), jpegBytes.length, a);
|
// boolean is = hCNetSDK.NET_DVR_CaptureJPEGPicture_NEW(lUserID.intValue(), new NativeLong(chanLong), jpeg, jpegBuffer, 1024*1024, a);
|
System.out.println("-----------这里开始图片存入内存----------" + is);
|
if (is) {
|
/**
|
* 该方式使用内存获取 但是读取有问题无法预览
|
* linux下 可能有问题
|
* */
|
System.out.println("hksdk(抓图)-结果状态值(0表示成功):" + hCNetSDK.NET_DVR_GetLastError());
|
//存储到本地
|
BufferedOutputStream outputStream = null;
|
try {
|
outputStream = new BufferedOutputStream(new FileOutputStream(file));
|
System.out.println(a.getValue());
|
outputStream.write(jpegBytes, 0, a.getValue());
|
//outputStream.write(jpegBytes);
|
outputStream.flush();
|
} catch (FileNotFoundException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
if (outputStream != null) {
|
try {
|
outputStream.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
} else {
|
System.out.println("hksdk(抓图)-抓取失败,错误码:" + hCNetSDK.NET_DVR_GetLastError());
|
}
|
}
|
}
|
