From 67225c1a9633e61c783560f8e205916e0e4a2721 Mon Sep 17 00:00:00 2001
From: xupei <xupei3360@163.com>
Date: Tue, 25 Mar 2025 16:20:00 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=8E=E7=BB=9F=E4=B8=80=E6=8E=A5=E5=8F=A3?=
 =?UTF-8?q?=E8=8E=B7=E5=8F=96pose=20transformation=E7=9A=84=E7=94=9F?=
 =?UTF-8?q?=E6=88=90=E8=AE=B0=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/com/ai/da/model/vo/ModuleChooseVO.java |  2 +-
 .../com/ai/da/model/vo/PoseTransformationVO.java     | 12 ++++++++++++
 src/main/java/com/ai/da/service/GenerateService.java |  2 +-
 .../com/ai/da/service/impl/GenerateServiceImpl.java  |  9 +++++++--
 .../ai/da/service/impl/UserLikeGroupServiceImpl.java |  3 +++
 5 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java b/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java
index 071cf62b..3cf15c4a 100644
--- a/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java
+++ b/src/main/java/com/ai/da/model/vo/ModuleChooseVO.java
@@ -30,7 +30,7 @@ public class ModuleChooseVO {
     private CavasModuleChooseVO canvas;
 
     private List<CollectionSketchVO> boundingBox;
-//    private moodBoardModuleChooseVO poseTransfer;
+    private List<PoseTransformationVO> poseTransfer;
 //    private moodBoardModuleChooseVO patternMaking3D;
     private SketchReconstructionVO deReconstruction;
 }
diff --git a/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java b/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java
index 55e39da1..e7349e32 100644
--- a/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java
+++ b/src/main/java/com/ai/da/model/vo/PoseTransformationVO.java
@@ -13,6 +13,8 @@ public class PoseTransformationVO {
 
     private String taskId;
 
+    private String productImage;
+
     private String gifUrl;
 
     private String videoUrl;
@@ -22,4 +24,14 @@ public class PoseTransformationVO {
     private byte isLiked;
 
     private String status;
+
+    public PoseTransformationVO(Long id, String taskId, String gifUrl, String videoUrl, String firstFrameUrl, byte isLiked, String status) {
+        this.id = id;
+        this.taskId = taskId;
+        this.gifUrl = gifUrl;
+        this.videoUrl = videoUrl;
+        this.firstFrameUrl = firstFrameUrl;
+        this.isLiked = isLiked;
+        this.status = status;
+    }
 }
diff --git a/src/main/java/com/ai/da/service/GenerateService.java b/src/main/java/com/ai/da/service/GenerateService.java
index 656e6cec..445a8eb7 100644
--- a/src/main/java/com/ai/da/service/GenerateService.java
+++ b/src/main/java/com/ai/da/service/GenerateService.java
@@ -52,7 +52,7 @@ public interface GenerateService extends IService<Generate> {
 
     PoseTransformationVO getPoseTransformationResult(String taskId);
 
-    List<PoseTransformationVO> getPoseTransformationResultList(String projectId);
+    List<PoseTransformationVO> getPoseTransformationResultList(Long projectId);
 
     Boolean disOrLikePose(Long transformedId, String likeOrDislike);
 
diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
index e98caf63..07a01e11 100644
--- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
@@ -1018,13 +1018,18 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
         }
     }
 
-    public List<PoseTransformationVO> getPoseTransformationResultList(String projectId){
+    public List<PoseTransformationVO> getPoseTransformationResultList(Long projectId){
         List<PoseTransformation> poseTransformations = poseTransformationMapper.selectList(new QueryWrapper<PoseTransformation>().eq("project_id", projectId).eq("is_liked", 1));
         List<PoseTransformationVO> vos = new ArrayList<>();
         if (poseTransformations != null && poseTransformations.size() > 1){
             poseTransformations.forEach(item -> {
-                PoseTransformationVO poseTransformationVO = CopyUtil.copyObject(item, PoseTransformationVO.class);
+                PoseTransformationVO poseTransformationVO = new PoseTransformationVO();
                 poseTransformationVO.setId(item.getId());
+                poseTransformationVO.setProductImage(minioUtil.getPreSignedUrl(item.getProductImage(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
+                poseTransformationVO.setGifUrl(minioUtil.getPreSignedUrl(item.getGifUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
+                poseTransformationVO.setVideoUrl(minioUtil.getPreSignedUrl(item.getVideoUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
+                poseTransformationVO.setFirstFrameUrl(minioUtil.getPreSignedUrl(item.getFirstFrameUrl(), CommonConstant.MINIO_IMAGE_EXPIRE_TIME));
+                poseTransformationVO.setIsLiked(item.getIsLiked());
                 vos.add(poseTransformationVO);
             });
         }
diff --git a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
index 039aeaf2..21b36dad 100644
--- a/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
+++ b/src/main/java/com/ai/da/service/impl/UserLikeGroupServiceImpl.java
@@ -1549,6 +1549,9 @@ public class UserLikeGroupServiceImpl extends ServiceImpl<UserLikeGroupMapper, U
             }else if (module.equals(Module.deReconstruction.name())){
                 SketchReconstructionVO sketchReconstruction = generateService.getSketchReconstruction(projectDTO.getId());
                 moduleChooseVO.setDeReconstruction(sketchReconstruction);
+            }else if (module.equals(Module.poseTransfer.name())){
+                List<PoseTransformationVO> poseTransformationResultList = generateService.getPoseTransformationResultList(projectDTO.getId());
+                moduleChooseVO.setPoseTransfer(poseTransformationResultList);
             }
         }
         return moduleChooseVO;