增加obj转glb工具

4dkankan-common-utils/src/main/java/com/fdkankan/common/util/OBJToGLBUtil.java

@@ -0,0 +1,141 @@
+package com.fdkankan.common.util;
+
+import cn.hutool.core.img.ImgUtil;
+import cn.hutool.core.io.FileUtil;
+import cn.hutool.core.io.file.FileReader;
+import cn.hutool.core.util.RuntimeUtil;
+import com.fdkankan.common.constant.ErrorCode;
+import com.fdkankan.common.exception.BusinessException;
+import java.awt.image.BufferedImage;
+import java.io.File;
+import java.util.LinkedHashSet;
+import java.util.List;
+import java.util.Locale;
+import lombok.extern.slf4j.Slf4j;
+
+@Slf4j
+public class OBJToGLBUtil {
+
+    /**
+     *
+     * @param objPath obj文件的目录
+     * @param glbPath glb文件的最对路径
+     */
+    public static void objToGlb(String objPath, String glbPath) {
+        OBJToGLBUtil.checkObj(objPath);
+        objPath = OBJToGLBUtil.getObj(objPath);
+        String command = "obj2gltf -i " + objPath + " -o " + glbPath;
+        log.info("开始执行obj转换gbl命令-{}", command);
+        Process exec = RuntimeUtil.exec(command);
+        log.info("结束执行obj转换gbl命令-{}", command);
+    }
+
+    static String getObj(String objPath) {
+        List<File> files = FileUtil.loopFiles(objPath);
+        if (objPath.toLowerCase(Locale.ROOT).contains(".obj")) {
+            return objPath;
+        }
+        for (File file2 : files) {
+            if (file2.isDirectory()) {
+                continue;
+            }
+            if (FileUtil.extName(file2.getName()).toLowerCase(Locale.ROOT).equals("obj")) {
+                return file2.getAbsolutePath();
+            }
+        }
+        return "";
+    }
+
+
+    private static boolean checkObj(String objPath) {
+        if (objPath.contains(".obg")) {
+            objPath = FileUtil.file(objPath).getParent();
+        }
+        File file1 = new File(objPath);
+        File[] files = file1.listFiles();
+        if (files == null || files.length <= 0) {
+            throw new BusinessException(ErrorCode.FAILURE_CODE_7014);
+        }
+        File mtlFile = null;
+        File objFile = null;
+        for (File file2 : files) {
+            if (file2.getName().endsWith(".obj")) {
+//                if (StringUtils.isChinese(file2.getName())) {
+//                    throw new ServiceException(HttpStatus.e7008.getMsg(), HttpStatus.e7008.getCode());
+//                }
+                objFile = file2;
+            }
+            if (file2.getName().endsWith(".mtl")) {
+//                if (StringUtils.isChinese(file2.getName())) {
+//                    throw new ServiceException(HttpStatus.e7008.getMsg(), HttpStatus.e7008.getCode());
+//                }
+                mtlFile = file2;
+            }
+            if (FileUtil.getType(file2).equals("jpg") || FileUtil.getType(file2).equals("png")) {
+                BufferedImage read = ImgUtil.read(file2);
+                int widthImg = read.getWidth();
+                int heightImg = read.getHeight();
+                if (widthImg > 2048 && heightImg > 2048) {
+                    OBJToGLBUtil.log.info("尺寸大2k,执行压缩");
+                    ImgUtil.scale(file2,
+                            file2,
+                            2048, 2048, null);
+                }
+            }
+        }
+        if (mtlFile == null || objFile == null) {
+            throw new BusinessException(ErrorCode.FAILURE_CODE_5059);
+        }
+        return OBJToGLBUtil.checkMtl(file1, mtlFile);
+    }
+
+    private static boolean checkMtl(File allFile, File file) {
+        if (!file.getName().endsWith(".mtl")) {
+            return false;
+        }
+        LinkedHashSet<String> imgName = new LinkedHashSet<>();
+        if (allFile == null || allFile.length() <= 0) {
+            return false;
+        }
+        File[] files = allFile.listFiles();
+        if (files == null || files.length <= 0) {
+            return false;
+        }
+        for (File listFile : files) {
+            String modelName = listFile.getName();
+            imgName.add(modelName);
+        }
+        LinkedHashSet<String> imgMtl = OBJToGLBUtil.readMtlFile(file.getPath());
+        for (String mtlName : imgMtl) {
+            if (!imgName.contains(mtlName)) {
+                throw new BusinessException(-1 ,mtlName + "图片缺失!");
+            }
+        }
+        return true;
+    }
+
+    private static LinkedHashSet<String> readMtlFile(String mtlPath) {
+        LinkedHashSet<String> imgName = new LinkedHashSet<>();
+        FileReader fileReader = new FileReader(mtlPath);
+        try {
+            List<String> lines = fileReader.readLines();
+            for (String line : lines) {
+                String[] tempsa = line.split("[ ]+");
+                if (tempsa[0].trim().equals("map_Ka")) {
+                    imgName.add(tempsa[1]);
+                }
+                if (tempsa[0].trim().equals("map_Kd")) {
+                    imgName.add(tempsa[1]);
+                }
+            }
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return imgName;
+    }
+
+
+    public static void main(String[] args) {
+        System.out.println(OBJToGLBUtil.checkObj("F:\\test\\新建文件夹\\police模型"));
+    }
+}