From c96ab96cb0243e197c5d6603312dfd06b3b15ebb Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期三, 14 十二月 2022 16:01:02 +0800
Subject: [PATCH] 报警图片加红色标注框

---
 ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java |   30 +++++++++++++++++++++++++++---
 1 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java b/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java
index 3ccd611..e80d31d 100644
--- a/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java
+++ b/ycl-platform/src/main/java/com/ycl/service/video/impl/VideoAlarmReportServiceImpl.java
@@ -2,11 +2,10 @@
 
 import com.alibaba.druid.util.StringUtils;
 import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
-import com.ycl.dto.video.AlarmDataParam;
-import com.ycl.dto.video.AlarmParam;
-import com.ycl.dto.video.CameraInfParam;
+import com.ycl.dto.video.*;
 import com.ycl.entity.video.VideoAlarmReport;
 import com.ycl.mapper.video.VideoAlarmReportMapper;
 import com.ycl.service.caseHandler.IViolationsService;
@@ -15,8 +14,13 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import javax.imageio.ImageIO;
 import javax.xml.bind.DatatypeConverter;
+import java.awt.*;
+import java.awt.image.BufferedImage;
 import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
 import java.io.InputStream;
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
@@ -61,7 +65,27 @@
 
                 byte[] data = DatatypeConverter.parseBase64Binary(alarmParam.getPicData());
                 InputStream inputStream = new ByteArrayInputStream(data);
+
                 try {
+                    if (!StringUtils.isEmpty(alarmParam.getAlarmData().getAlarmObject().toString())) {
+                        String alarmObject = JSON.toJSONString(alarmParam.getAlarmData().getAlarmObject());
+                        if (!StringUtils.isEmpty(alarmObject)) {
+                            BufferedImage image = ImageIO.read(inputStream);
+
+                            Graphics2D g2 = (Graphics2D)image.getGraphics();
+                            g2.setColor(Color.RED);
+                            g2.setStroke(new BasicStroke(3.0f));
+                            List<AlarmObject> alarmObjects = JSON.parseArray(alarmObject, AlarmObject.class);
+                            alarmObjects.forEach(o -> {
+                                o.getObjectRect().forEach(rect -> {
+                                    g2.drawRect(rect.getX(), rect.getY(), rect.getW(), rect.getH());
+                                });
+                            });
+                            ByteArrayOutputStream os = new ByteArrayOutputStream();
+                            ImageIO.write(image, extension, os);
+                            inputStream = new ByteArrayInputStream(os.toByteArray());
+                        }
+                    }
                     picData = mediaFileUtil.savePicture(inputStream, extension);
                 } catch (Exception e) {
                     e.printStackTrace();

--
Gitblit v1.8.0