package com.ruoyi.aibrain.service.impl;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
|
import com.alibaba.fastjson2.JSONArray;
|
import com.alibaba.fastjson2.JSONObject;
|
import com.ruoyi.activeMQ.vo.RuleAreaVO;
|
import com.ruoyi.aibrain.domain.AiCameraRule;
|
import com.ruoyi.aibrain.mapper.AiCameraRuleMapper;
|
import com.ruoyi.common.utils.DateUtils;
|
import com.ruoyi.common.utils.SecurityUtils;
|
import com.ruoyi.system.mapper.AiRuleMapper;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import com.ruoyi.aibrain.mapper.AiCameraMapper;
|
import com.ruoyi.aibrain.domain.AiCamera;
|
import com.ruoyi.aibrain.service.IAiCameraService;
|
|
/**
|
* 设备管理Service业务层处理
|
*
|
* @author wf
|
* @date 2023-03-08
|
*/
|
@Service
|
public class AiCameraServiceImpl implements IAiCameraService
|
{
|
@Autowired
|
private AiCameraMapper aiCameraMapper;
|
|
@Autowired
|
private AiCameraRuleMapper aiCameraRuleMapper;
|
|
@Autowired
|
private AiRuleMapper aiRuleMapper;
|
|
/**
|
* 查询设备管理
|
*
|
* @param id 设备管理主键
|
* @return 设备管理
|
*/
|
@Override
|
public AiCamera selectAiCameraById(Long id)
|
{
|
AiCamera aiCamera = aiCameraMapper.selectAiCameraById(id);
|
List<AiCameraRule> rules = aiCameraRuleMapper.selectAiCameraRuleListByCameraId(id);
|
for (AiCameraRule rule : rules){
|
String left = "[";
|
String center = "[";
|
String right = "[";
|
JSONArray array = JSONArray.parseArray(rule.getRuleArea());
|
List<RuleAreaVO> areas = new ArrayList<>();
|
if (array != null){
|
for (int i = 0; i < array.size(); i++) {
|
if(rule.getRuleId() == 1){
|
if(i >= 0 && i < 4){
|
left = left + array.getJSONObject(i) + ",";
|
}
|
if(i >= 4 && i < 8){
|
center = center + array.getJSONObject(i)+ ",";
|
}
|
if(i >= 8){
|
right = right + array.getJSONObject(i)+ ",";
|
}
|
}
|
JSONObject json = array.getJSONObject(i);
|
RuleAreaVO area = new RuleAreaVO();
|
// area.setX(Float.parseFloat(json.get("x").toString()));
|
// area.setY(Float.parseFloat(json.get("y").toString()));
|
areas.add(area);
|
}
|
}
|
// if(rule.getRuleId() == 1){
|
// rule.setLeftArea(left.substring(0, left.lastIndexOf(','))+"]");
|
// rule.setCenterArea(center.substring(0, center.lastIndexOf(','))+"]");
|
// rule.setRightArea(right.substring(0, right.lastIndexOf(','))+"]");
|
// }
|
rule.setAreas(areas);
|
}
|
aiCamera.setRules(rules);
|
return aiCamera;
|
}
|
|
/**
|
* 查询设备管理列表
|
*
|
* @param aiCamera 设备管理
|
* @return 设备管理
|
*/
|
@Override
|
public List<AiCamera> selectAiCameraList(AiCamera aiCamera)
|
{
|
List<AiCamera> aiCameras = aiCameraMapper.selectAiCameraList(aiCamera);
|
for (AiCamera camera : aiCameras) {
|
List<AiCameraRule> rules = aiCameraRuleMapper.selectAiCameraRuleListByCameraId(camera.getId());
|
for (AiCameraRule rule : rules){
|
String left = "[";
|
String center = "[";
|
String right = "[";
|
JSONArray array = JSONArray.parseArray(rule.getRuleArea());
|
List<RuleAreaVO> areas = new ArrayList<>();
|
if (array != null){
|
for (int i = 0; i < array.size(); i++) {
|
if(rule.getRuleId() == 1){
|
if(i >= 0 && i < 5){
|
left = left + array.getJSONObject(i);
|
}
|
if(i >= 5 && i < 9){
|
center = center + array.getJSONObject(i);
|
}
|
if(i >= 9){
|
right = right + array.getJSONObject(i);
|
}
|
}
|
JSONObject json = array.getJSONObject(i);
|
RuleAreaVO area = new RuleAreaVO();
|
// area.setX(Float.parseFloat(json.get("x").toString()));
|
// area.setY(Float.parseFloat(json.get("y").toString()));
|
areas.add(area);
|
}
|
}
|
rule.setLeftArea(left+"]");
|
rule.setCenterArea(center+"]");
|
rule.setRightArea(right+"]");
|
rule.setAreas(areas);
|
}
|
camera.setRules(rules);
|
}
|
return aiCameras;
|
}
|
|
/**
|
* 新增设备管理
|
*
|
* @param aiCamera 设备管理
|
* @return 结果
|
*/
|
@Override
|
public int insertAiCamera(AiCamera aiCamera)
|
{
|
aiCamera.setCreateTime(DateUtils.getNowDate());
|
aiCamera.setCreateBy(SecurityUtils.getUsername());
|
|
aiCamera.setRtspUrl("rtsp://"+aiCamera.getCameraUser()+":"+aiCamera.getCameraPassword()+
|
"@"+aiCamera.getCameraIp()+":554/h264/1/main/av_stream");
|
|
aiCameraMapper.insertAiCamera(aiCamera);
|
insertCameraRule(aiCamera);
|
aiCamera.setCameraNum("AIC0" + aiCamera.getRegion() + String.format("%03d", aiCamera.getId()));
|
return aiCameraMapper.updateAiCamera(aiCamera);
|
}
|
|
private void insertCameraRule(AiCamera aiCamera) {
|
//清除原设备算法
|
aiCameraRuleMapper.deleteAiCameraRuleByCameraId(aiCamera.getId());
|
for (AiCameraRule rule : aiCamera.getRules()){
|
rule.setCameraId(aiCamera.getId());
|
rule.setCreateBy(SecurityUtils.getUsername());
|
rule.setRuleName(aiRuleMapper.selectAiRuleById(rule.getRuleId()).getRuleName());
|
// if(rule.getRuleId() == 1){
|
// String left = rule.getLeftArea().replace("]", "");
|
// String center = rule.getCenterArea().replace("[", "");
|
// String centerArea = center.replace("]", "");
|
// String right = rule.getRightArea().replace("[", "");
|
// rule.setRuleArea(left +","+ centerArea +","+ right);
|
// }
|
aiCameraRuleMapper.insertAiCameraRule(rule);
|
}
|
}
|
|
/**
|
* 修改设备管理
|
*
|
* @param aiCamera 设备管理
|
* @return 结果
|
*/
|
@Override
|
public int updateAiCamera(AiCamera aiCamera)
|
{
|
aiCamera.setUpdateTime(DateUtils.getNowDate());
|
insertCameraRule(aiCamera);
|
aiCamera.setRtspUrl("rtsp://"+aiCamera.getCameraUser()+":"+aiCamera.getCameraPassword()+
|
"@"+aiCamera.getCameraIp()+":554/h264/1/main/av_stream");
|
return aiCameraMapper.updateAiCamera(aiCamera);
|
}
|
|
/**
|
* 批量删除设备管理
|
*
|
* @param ids 需要删除的设备管理主键
|
* @return 结果
|
*/
|
@Override
|
public int deleteAiCameraByIds(Long[] ids)
|
{
|
return aiCameraMapper.deleteAiCameraByIds(ids);
|
}
|
|
/**
|
* 删除设备管理信息
|
*
|
* @param id 设备管理主键
|
* @return 结果
|
*/
|
@Override
|
public int deleteAiCameraById(Long id)
|
{
|
return aiCameraMapper.deleteAiCameraById(id);
|
}
|
|
/**
|
* 查询要启动的AI识别设备
|
* @return
|
*/
|
@Override
|
public List<AiCamera> queryCameraToStart() {
|
List<AiCamera> cameraList = aiCameraMapper.queryCameraToStart();
|
for (AiCamera camera : cameraList){
|
List<AiCameraRule> rules = aiCameraRuleMapper.selectAiCameraRuleListByCameraId(camera.getId());
|
for (AiCameraRule rule : rules){
|
JSONArray array = JSONArray.parseArray(rule.getRuleArea());
|
List<RuleAreaVO> areas = new ArrayList<>();
|
if (array != null){
|
for (int i = 0; i < array.size(); i++) {
|
JSONObject json = array.getJSONObject(i);
|
RuleAreaVO area = new RuleAreaVO();
|
// area.setX(Float.parseFloat(json.get("x").toString()));
|
// area.setY(Float.parseFloat(json.get("y").toString()));
|
areas.add(area);
|
}
|
}
|
rule.setAreas(areas);
|
}
|
camera.setRules(rules);
|
}
|
return cameraList;
|
}
|
|
/**
|
* 变更设备状态
|
* @return
|
*/
|
@Override
|
public void updateAiCameraStatus(AiCamera aiCamera) {
|
aiCameraMapper.updateAiCamera(aiCamera);
|
}
|
|
/**
|
* 查询所有设备
|
* @return
|
*/
|
@Override
|
public List<AiCamera> selectAllCamera() {
|
AiCamera camera = new AiCamera();
|
List<AiCamera> aiCameras = aiCameraMapper.selectAiCameraList(camera);
|
for (AiCamera aiCamera : aiCameras) {
|
List<AiCameraRule> rules = aiCameraRuleMapper.selectAiCameraRuleListByCameraId(aiCamera.getId());
|
for (AiCameraRule rule : rules){
|
String left = "[";
|
String center = "[";
|
String right = "[";
|
JSONArray array = JSONArray.parseArray(rule.getRuleArea());
|
List<RuleAreaVO> areas = new ArrayList<>();
|
if (array != null){
|
for (int i = 0; i < array.size(); i++) {
|
if(rule.getRuleId() == 1){
|
if(i >= 0 && i < 4){
|
left = left + array.getJSONObject(i) + ",";
|
}
|
if(i >= 4 && i < 8){
|
center = center + array.getJSONObject(i)+ ",";
|
}
|
if(i >= 8){
|
right = right + array.getJSONObject(i)+ ",";
|
}
|
}
|
JSONObject json = array.getJSONObject(i);
|
RuleAreaVO area = new RuleAreaVO();
|
// area.setX(Float.parseFloat(json.get("x").toString()));
|
// area.setY(Float.parseFloat(json.get("y").toString()));
|
areas.add(area);
|
}
|
}
|
// if(rule.getRuleId() == 1){
|
// rule.setLeftArea(left.substring(0, left.lastIndexOf(','))+"]");
|
// rule.setCenterArea(center.substring(0, center.lastIndexOf(','))+"]");
|
// rule.setRightArea(right.substring(0, right.lastIndexOf(','))+"]");
|
// }
|
rule.setAreas(areas);
|
}
|
aiCamera.setRules(rules);
|
}
|
return aiCameras;
|
}
|
}
|
