311 lines
11 KiB
Java
311 lines
11 KiB
Java
|
package com.whdc.service.impl;
|
|||
|
|
|
|||
|
|
import com.sun.jna.Pointer;
|
|||
|
|
import com.sun.jna.ptr.IntByReference;
|
|||
|
|
import com.whdc.alarm.NVSSDK2;
|
|||
|
|
import com.whdc.alarm.NetClient2;
|
|||
|
|
import com.whdc.entity.AlarmRecord;
|
|||
|
|
import lombok.extern.slf4j.Slf4j;
|
|||
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|||
|
|
import org.springframework.beans.factory.annotation.Value;
|
|||
|
|
import org.springframework.stereotype.Service;
|
|||
|
|
|
|||
|
|
import javax.annotation.PostConstruct;
|
|||
|
|
import javax.annotation.PreDestroy;
|
|||
|
|
import java.io.File;
|
|||
|
|
import java.time.LocalDateTime;
|
|||
|
|
import java.util.concurrent.ExecutorService;
|
|||
|
|
import java.util.concurrent.Executors;
|
|||
|
|
|
|||
|
|
@Slf4j
|
|||
|
|
@Service
|
|||
|
|
public class VehicleDetectionService {
|
|||
|
|
|
|||
|
|
@Value("${app.image-save-path}")
|
|||
|
|
private String imageSavePath;
|
|||
|
|
|
|||
|
|
@Autowired
|
|||
|
|
private AlarmRecordService alarmRecordService;
|
|||
|
|
|
|||
|
|
private int m_iLogonID = -1;
|
|||
|
|
private int m_iConnectID = -1;
|
|||
|
|
private boolean isRunning = false;
|
|||
|
|
private ExecutorService executorService;
|
|||
|
|
|
|||
|
|
private String[] strCarPlateColor = new String[NVSSDK2.MAX_CAR_PLATE_COLOR];
|
|||
|
|
private String[] strCarColor = new String[NVSSDK2.MAX_CAR_COLOR];
|
|||
|
|
|
|||
|
|
// @PostConstruct
|
|||
|
|
public void init() {
|
|||
|
|
executorService = Executors.newSingleThreadExecutor();
|
|||
|
|
initValueString();
|
|||
|
|
createImageDirectory();
|
|||
|
|
startDetection();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
@PreDestroy
|
|||
|
|
public void destroy() {
|
|||
|
|
stopDetection();
|
|||
|
|
if (executorService != null) {
|
|||
|
|
executorService.shutdown();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void initValueString() {
|
|||
|
|
strCarPlateColor[0] = "unknown";
|
|||
|
|
strCarPlateColor[1] = "white on blue";
|
|||
|
|
strCarPlateColor[2] = "black on yellow";
|
|||
|
|
strCarPlateColor[3] = "Black on white";
|
|||
|
|
strCarPlateColor[4] = "black and white";
|
|||
|
|
strCarPlateColor[5] = "white on green";
|
|||
|
|
|
|||
|
|
strCarColor[0] = "white";
|
|||
|
|
strCarColor[1] = "red";
|
|||
|
|
strCarColor[2] = "yellow";
|
|||
|
|
// ... 其他颜色定义
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void createImageDirectory() {
|
|||
|
|
File dir = new File(imageSavePath);
|
|||
|
|
if (!dir.exists()) {
|
|||
|
|
if (dir.mkdirs()) {
|
|||
|
|
System.out.println("创建图片保存目录成功: " + imageSavePath);
|
|||
|
|
} else {
|
|||
|
|
System.err.println("创建图片保存目录失败: " + imageSavePath);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void startDetection() {
|
|||
|
|
if (isRunning) {
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
executorService.execute(() -> {
|
|||
|
|
try {
|
|||
|
|
isRunning = true;
|
|||
|
|
System.out.println("开始车辆检测...");
|
|||
|
|
|
|||
|
|
// SDK初始化
|
|||
|
|
if (sdkInit() != 0) {
|
|||
|
|
System.err.println("SDK初始化失败");
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设备登录（这里需要根据实际情况配置）
|
|||
|
|
if (logonDevice("192.168.1.100", "admin", "password", 3000) != 0) {
|
|||
|
|
System.err.println("设备登录失败");
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 开始抓拍
|
|||
|
|
startSnap();
|
|||
|
|
|
|||
|
|
// 保持运行
|
|||
|
|
while (isRunning) {
|
|||
|
|
try {
|
|||
|
|
Thread.sleep(1000);
|
|||
|
|
} catch (InterruptedException e) {
|
|||
|
|
Thread.currentThread().interrupt();
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
System.err.println("车辆检测服务异常: " + e.getMessage());
|
|||
|
|
e.printStackTrace();
|
|||
|
|
} finally {
|
|||
|
|
stopDetection();
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void stopDetection() {
|
|||
|
|
isRunning = false;
|
|||
|
|
if (m_iConnectID != -1) {
|
|||
|
|
NetClient2.StopRecv(m_iConnectID);
|
|||
|
|
m_iConnectID = -1;
|
|||
|
|
}
|
|||
|
|
if (m_iLogonID != -1) {
|
|||
|
|
NetClient2.Logoff(m_iLogonID);
|
|||
|
|
m_iLogonID = -1;
|
|||
|
|
}
|
|||
|
|
NetClient2.Cleanup();
|
|||
|
|
System.out.println("车辆检测服务已停止");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private int sdkInit() {
|
|||
|
|
NVSSDK2.SDK_VERSION ver = new NVSSDK2.SDK_VERSION();
|
|||
|
|
int iRet = NetClient2.GetVersion(ver);
|
|||
|
|
System.out.println("[SDK_VERSION]" + ver.m_cVerInfo);
|
|||
|
|
|
|||
|
|
// 设置回调函数
|
|||
|
|
iRet = NetClient2.SetNotifyFunction(
|
|||
|
|
new NVSSDK2.MAIN_NOTIFY() {
|
|||
|
|
public void MainNotify(int iLogonID, int wParam, Pointer lParam, Pointer noitfyUserData) {
|
|||
|
|
// 处理主通知
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
new NVSSDK2.ALARM_NOTIFY() {
|
|||
|
|
public void AlarmNotify(int iLogonID, int iChannel, int iAlarmState, int iAlarmType, Pointer pUserData) {
|
|||
|
|
// 处理告警通知
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
new NVSSDK2.PARACHANGE_NOTIFY() {
|
|||
|
|
public void ParaChangeNotify(int iLogonID, int iChannel, int paraType, Pointer para, Pointer noitfyUserData) {
|
|||
|
|
// 处理参数变化通知
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
System.out.println("SetNotifyFunction(" + iRet + ")");
|
|||
|
|
|
|||
|
|
iRet = NetClient2.Startup();
|
|||
|
|
System.out.println("Startup(" + iRet + ")");
|
|||
|
|
|
|||
|
|
return iRet;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private int logonDevice(String ip, String username, String password, int port) {
|
|||
|
|
m_iLogonID = NetClient2.Logon("", ip, username, password, "", port);
|
|||
|
|
|
|||
|
|
// 等待登录成功
|
|||
|
|
int retryCount = 0;
|
|||
|
|
while (retryCount < 30) {
|
|||
|
|
int logonStatus = NetClient2.GetLogonStatus(m_iLogonID);
|
|||
|
|
if (logonStatus == NVSSDK2.LOGON_SUCCESS) {
|
|||
|
|
log.info("设备登录成功: {}", ip);
|
|||
|
|
return 0;
|
|||
|
|
} else if (logonStatus == NVSSDK2.LOGON_FAILED) {
|
|||
|
|
System.err.println("设备登录失败: " + ip);
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
Thread.sleep(1000);
|
|||
|
|
} catch (InterruptedException e) {
|
|||
|
|
Thread.currentThread().interrupt();
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
retryCount++;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
System.err.println("设备登录超时: " + ip);
|
|||
|
|
return -1;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private int startSnap() {
|
|||
|
|
NVSSDK2.NetPicPara tNetPicParam = new NVSSDK2.NetPicPara();
|
|||
|
|
tNetPicParam.iStructLen = tNetPicParam.size();
|
|||
|
|
tNetPicParam.iChannelNo = 0;
|
|||
|
|
tNetPicParam.cbkPicStreamNotify = new NVSSDK2.NET_PICSTREAM_NOTIFY() {
|
|||
|
|
public int PicDataNotify(int ulID, int lCommand, Pointer tInfo, int iLen, Pointer lpUserData) {
|
|||
|
|
return handlePicData(ulID, lCommand, tInfo, iLen, lpUserData);
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
tNetPicParam.pvUser = null;
|
|||
|
|
|
|||
|
|
IntByReference pConnectID = new IntByReference();
|
|||
|
|
int iRet = NetClient2.StartRecvNetPicStream(m_iLogonID, tNetPicParam, tNetPicParam.size(), pConnectID);
|
|||
|
|
if (iRet < 0) {
|
|||
|
|
m_iConnectID = -1;
|
|||
|
|
System.out.println("StartRecvNetPicStream Failed!");
|
|||
|
|
return -1;
|
|||
|
|
} else {
|
|||
|
|
m_iConnectID = pConnectID.getValue();
|
|||
|
|
System.out.println("StartRecvNetPicStream Success! ConnectID(" + m_iConnectID + ")");
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private int handlePicData(int ulID, int lCommand, Pointer tInfo, int iLen, Pointer lpUserData) {
|
|||
|
|
if (lCommand != NVSSDK2.NET_PICSTREAM_CMD_ITS) {
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
NVSSDK2.ItsPicStream tItsPicStream = new NVSSDK2.ItsPicStream();
|
|||
|
|
tItsPicStream.write();
|
|||
|
|
Pointer pItsBuffer = tItsPicStream.getPointer();
|
|||
|
|
byte[] RecvBuffer = tInfo.getByteArray(0, iLen);
|
|||
|
|
int iCopySize = Math.min(tItsPicStream.size(), iLen);
|
|||
|
|
pItsBuffer.write(0, RecvBuffer, 0, iCopySize);
|
|||
|
|
tItsPicStream.read();
|
|||
|
|
|
|||
|
|
String strIP = new String(tItsPicStream.cCameraIP).trim();
|
|||
|
|
String strPlate = new String(tItsPicStream.cPlate).trim();
|
|||
|
|
|
|||
|
|
// 检测到车辆，生成告警
|
|||
|
|
if (!strPlate.isEmpty() && !strPlate.equals("未知")) {
|
|||
|
|
log.info("检测到车辆闯入: {}, IP: {}", strPlate, strIP);
|
|||
|
|
|
|||
|
|
// 保存图片
|
|||
|
|
String imageFileName = saveVehicleImage(tItsPicStream);
|
|||
|
|
|
|||
|
|
// 创建告警记录
|
|||
|
|
createAlarmRecord(tItsPicStream, imageFileName);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
System.err.println("处理图片数据异常: " + e.getMessage());
|
|||
|
|
e.printStackTrace();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private String saveVehicleImage(NVSSDK2.ItsPicStream tItsPicStream) {
|
|||
|
|
String fileName = "vehicle_" + System.currentTimeMillis() + "_" +
|
|||
|
|
new String(tItsPicStream.cPlate).trim() + ".jpg";
|
|||
|
|
String filePath = imageSavePath + File.separator + fileName;
|
|||
|
|
|
|||
|
|
// 这里简化图片保存逻辑，实际需要根据SDK的图片数据结构来保存
|
|||
|
|
try {
|
|||
|
|
// 实际实现需要根据tItsPicStream中的图片数据来保存
|
|||
|
|
log.info("保存车辆图片: " + filePath);
|
|||
|
|
return fileName;
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
System.err.println("保存车辆图片失败: " + e.getMessage());
|
|||
|
|
return "";
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void createAlarmRecord(NVSSDK2.ItsPicStream tItsPicStream, String imageFileName) {
|
|||
|
|
try {
|
|||
|
|
AlarmRecord alarm = new AlarmRecord();
|
|||
|
|
alarm.setCameraIP(new String(tItsPicStream.cCameraIP).trim());
|
|||
|
|
alarm.setLicensePlate(new String(tItsPicStream.cPlate).trim());
|
|||
|
|
alarm.setChannelID(tItsPicStream.iChannelID);
|
|||
|
|
alarm.setSpeed((double) tItsPicStream.fSpeed);
|
|||
|
|
alarm.setAlarmType(tItsPicStream.iAlarmType);
|
|||
|
|
alarm.setAlarmCode(new String(tItsPicStream.cAlarmCode).trim());
|
|||
|
|
alarm.setImagePath(imageFileName);
|
|||
|
|
alarm.setAlarmTime(LocalDateTime.now());
|
|||
|
|
|
|||
|
|
// 设置车辆颜色
|
|||
|
|
if (tItsPicStream.iCarColor >= 0 && tItsPicStream.iCarColor < strCarColor.length) {
|
|||
|
|
alarm.setCarColor(strCarColor[tItsPicStream.iCarColor]);
|
|||
|
|
} else {
|
|||
|
|
alarm.setCarColor("unknown");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 设置车牌颜色
|
|||
|
|
if (tItsPicStream.iPlateColor >= 0 && tItsPicStream.iPlateColor < strCarPlateColor.length) {
|
|||
|
|
alarm.setPlateColor(strCarPlateColor[tItsPicStream.iPlateColor]);
|
|||
|
|
} else {
|
|||
|
|
alarm.setPlateColor("unknown");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 保存到数据库
|
|||
|
|
alarmRecordService.saveAlarmRecord(alarm);
|
|||
|
|
log.info("告警记录保存成功: {}", alarm.getLicensePlate());
|
|||
|
|
|
|||
|
|
} catch (Exception e) {
|
|||
|
|
log.error("创建告警记录失败: {}", e.getMessage());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public boolean isRunning() {
|
|||
|
|
return isRunning;
|
|||
|
|
}
|
|||
|
|
}
|