From c19666de02a21d62fb74c993a481141466111484 Mon Sep 17 00:00:00 2001
From: xiangpei <xiangpei@timesnew.cn>
Date: 星期三, 28 八月 2024 21:52:25 +0800
Subject: [PATCH] 工单获取图片ok

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java |   76 ++++++++++++++++++++++++--------------
 1 files changed, 48 insertions(+), 28 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
index 9d519a6..a389e0d 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/WorkOrderServiceImpl.java
@@ -1,6 +1,7 @@
 package com.ycl.platform.service.impl;
 
 import com.alibaba.fastjson2.JSON;
+import com.alibaba.fastjson2.JSONObject;
 import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
@@ -44,9 +45,9 @@
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.ObjectUtils;
-import org.bytedeco.javacv.FFmpegFrameGrabber;
-import org.bytedeco.javacv.Frame;
-import org.bytedeco.javacv.Java2DFrameConverter;
+import org.bytedeco.javacv.*;
+import org.bytedeco.opencv.global.opencv_imgcodecs;
+import org.bytedeco.opencv.opencv_core.Mat;
 import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -57,6 +58,7 @@
 import org.springframework.util.StringUtils;
 
 import javax.imageio.ImageIO;
+import javax.swing.*;
 import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
@@ -661,40 +663,58 @@
 
 
     @Override
-    public String getFrameImgByDevice(String deviceId, String channelId) {
-        String url = String.format(this.rtspServer + "/start/%s/%s", deviceId, channelId);
+    public String getFrameImgByDevice(String deviceId, String channelId) throws FFmpegFrameGrabber.Exception {
+        String url = String.format(this.rtspServer + "/api/play/start/%s/%s", deviceId, channelId);
         String result = HttpUtils.sendGet(url);
-        log.error("鎷垮埌鍙栨祦鍝嶅簲缁撴灉锛�" + result);
+// 鍋囪WVPResult鏄纭В鏋愬搷搴旂殑绫�
         WVPResult wvpResult = JSON.parseObject(result, WVPResult.class);
-        if (0 == wvpResult.getCode()) {
-            log.error("璇锋眰鎴愬姛");
-            StreamContent data = (StreamContent) wvpResult.getData();
-            log.error("鎷垮埌鍙栨祦鍝嶅簲缁撴灉锛�" + data);
-            String rtspUrl = data.getRtsp();
+        if (wvpResult.getCode() == 0) {
+            System.out.println("璇锋眰鎴愬姛");
+            JSONObject data = (JSONObject) wvpResult.getData();
+            String rtspUrl = data.getString("fmp4"); // 娉ㄦ剰锛氶�氬父RTSP URL涓嶄細浠�"fmp4"缁撳熬锛岃繖閲屽彲鑳芥槸鐗瑰畾API鐨勮繑鍥炴牸寮�
             if (StringUtils.hasText(rtspUrl)) {
-                try {
-                    // 鍒涘缓FFmpegFrameGrabber瀵硅薄
-                    FFmpegFrameGrabber grabber = new FFmpegFrameGrabber(rtspUrl);
-                    grabber.start(); // 寮�濮嬫崟鑾�
+                System.out.println("鐩爣鍦板潃锛�" + rtspUrl);
 
-                    Frame frame;
-                    if ((frame = grabber.grabFrame()) != null) {
-                        // 灏嗗抚杞崲涓哄浘鐗�
-                        Java2DFrameConverter converter = new Java2DFrameConverter();
-                        BufferedImage image = converter.convert(frame);
-                        File outputfile = new File(serverConfig.getUrl() + PlatformConfig.getUploadPath() + deviceId + IdUtils.fastSimpleUUID() + ".png");
-                        ImageIO.write(image, "png", outputfile);
-                        String imgPath = outputfile.getAbsolutePath();
-                        System.out.println("Saved " + imgPath);
-                        return imgPath;
+                FFmpegFrameGrabber grabber = null;
+                try {
+                    grabber = FFmpegFrameGrabber.createDefault(rtspUrl);
+                    grabber.start();
+
+                    Frame frame = grabber.grabImage(); // 鐩存帴鎹曡幏涓�甯�
+                    if (frame != null) {
+                        System.out.println("鎴愬姛鎹曡幏涓�甯�");
+
+                        // 灏咶rame杞崲涓篗at
+                        OpenCVFrameConverter.ToMat converter = new OpenCVFrameConverter.ToMat();
+                        Mat mat = converter.convertToMat(frame);
+
+                        // 鐢熸垚鍥剧墖璺緞
+                        String imgPath = PlatformConfig.getUploadPath() + deviceId + IdUtils.fastSimpleUUID() + ".png";
+                        System.out.println("鍥剧墖鍦板潃锛�" + imgPath);
+
+                        // 淇濆瓨鍥剧墖
+                        opencv_imgcodecs.imwrite(imgPath, mat);
+                    } else {
+                        System.out.println("鏈崟鑾峰埌甯�");
                     }
-                    grabber.stop(); // 鍋滄鎹曡幏
-                    grabber.release(); // 閲婃斁璧勬簮
-                } catch (IOException e) {
+                } catch (FrameGrabber.Exception e) {
                     e.printStackTrace();
+                } finally {
+                    if (grabber != null) {
+                        try {
+                            grabber.stop(); // 鍋滄鎹曡幏
+                        } catch (FrameGrabber.Exception e) {
+                            e.printStackTrace();
+                        }
+                        // 閫氬父涓嶉渶瑕佽皟鐢╮elease()锛屽洜涓簊top()浼氬鐞嗚祫婧愰噴鏀�
+                        // grabber.release(); // 閲婃斁璧勬簮
+                    }
                 }
             }
+        } else {
+            System.out.println("璇锋眰澶辫触锛岄敊璇爜锛�" + wvpResult.getCode());
         }
+
         return null;
     }
 

--
Gitblit v1.8.0