134 lines
5.5 KiB
TypeScript
134 lines
5.5 KiB
TypeScript
import JSZip from "jszip";
|
||
import * as THREE from "three";
|
||
import Loader from "@/core/loader/Loader.ts";
|
||
import App from "@/core/app/App.ts";
|
||
|
||
/**
|
||
* 解析材质zip包
|
||
*/
|
||
export function parseMaterialZip(zipFile: File): Promise<THREE.MeshStandardMaterial> {
|
||
return new Promise(async (resolve, reject) => {
|
||
const zip = new JSZip();
|
||
|
||
const zipContent = await zip.loadAsync(zipFile);
|
||
|
||
// 强制检查根目录下是否存在material.json
|
||
let materialJson: any = zipContent.file('material.json');
|
||
if (!materialJson) {
|
||
materialJson = {
|
||
textures: {},
|
||
properties: {}
|
||
}
|
||
|
||
// 获取所有文件路径列表（JSZip存储结构为 {路径: 文件对象}）
|
||
const filePaths = Object.keys(zipContent.files);
|
||
for (let i = 0; i < filePaths.length; i++) {
|
||
const relativePath: string = filePaths[i];
|
||
const file = zipContent.file(relativePath);
|
||
|
||
if (!file) continue;
|
||
|
||
// 判断是否为文件（JSZip中目录路径以 '/' 结尾）
|
||
if (file.dir || relativePath.endsWith('/')) continue;
|
||
|
||
if (relativePath.includes("baseColor")) {
|
||
materialJson.textures.baseColor = relativePath;
|
||
} else if (relativePath.includes("normal")) {
|
||
materialJson.textures.normal = relativePath;
|
||
} else if (relativePath.includes("bump")) {
|
||
materialJson.textures.bump = relativePath;
|
||
} else if (relativePath.includes("displacement")) {
|
||
materialJson.textures.displacement = relativePath;
|
||
} else if (relativePath.includes("emissive")) {
|
||
materialJson.textures.emissive = relativePath;
|
||
} else if (relativePath.includes("alpha")) {
|
||
materialJson.textures.alpha = relativePath;
|
||
} else if (relativePath.includes("env")) {
|
||
materialJson.textures.env = relativePath;
|
||
} else if (relativePath.includes("light")) {
|
||
materialJson.textures.light = relativePath;
|
||
} else {
|
||
// arm可能在一张图上各占一个通道
|
||
if (relativePath.includes("roughness")) {
|
||
materialJson.textures.roughness = relativePath;
|
||
}
|
||
if (relativePath.includes("metalness")) {
|
||
materialJson.textures.metalness = relativePath;
|
||
}
|
||
if (relativePath.includes("ao")) {
|
||
materialJson.textures.ao = relativePath;
|
||
}
|
||
}
|
||
}
|
||
} else {
|
||
try {
|
||
materialJson = JSON.parse(await materialJson.async('text'));
|
||
} catch (error) {
|
||
reject(error);
|
||
}
|
||
}
|
||
|
||
// 并行加载所有纹理
|
||
const texturePromises = Object.entries(materialJson.textures).map(
|
||
async ([type, path]:any) => {
|
||
const textureFile = zipContent.file(path);
|
||
if (!textureFile) {
|
||
console.warn(`Texture file not found: ${path}`);
|
||
return { type, texture: null };
|
||
}
|
||
|
||
const extension = path.toString().split(".").pop()?.toLowerCase() || "jpg";
|
||
let textureBlob: Blob;
|
||
|
||
try {
|
||
// 特殊处理 EXR 格式
|
||
if (extension === "exr") {
|
||
const buffer = await textureFile.async("arraybuffer");
|
||
textureBlob = new Blob([buffer], { type: "image/x-exr" });
|
||
} else {
|
||
textureBlob = await textureFile.async("blob");
|
||
}
|
||
} catch (err) {
|
||
console.error(`Failed to load texture (${type}):`, err);
|
||
return { type, texture: null };
|
||
}
|
||
|
||
const textureUrl = URL.createObjectURL(textureBlob);
|
||
|
||
return new Promise<{ type: string; texture: THREE.Texture | null }>(
|
||
(resolve) => {
|
||
Loader.loadUrlTexture(
|
||
extension,
|
||
textureUrl,
|
||
(texture: THREE.Texture) => {
|
||
URL.revokeObjectURL(textureUrl); // 清理资源
|
||
resolve({ type, texture });
|
||
},
|
||
(error: Error) => {
|
||
URL.revokeObjectURL(textureUrl);
|
||
console.error(`Texture load error (${type}):`, error);
|
||
resolve({ type, texture: null });
|
||
}
|
||
);
|
||
}
|
||
);
|
||
}
|
||
);
|
||
|
||
// 等待所有纹理加载完成
|
||
const textureResults = await Promise.all(texturePromises);
|
||
const textures = textureResults.reduce((acc, { type, texture }) => {
|
||
if (texture) acc[type] = texture;
|
||
return acc;
|
||
}, {} as Record<string, THREE.Texture>);
|
||
|
||
// 处理无有效纹理的情况
|
||
if (Object.keys(textures).length === 0) {
|
||
throw new Error("No valid textures found in the zip file");
|
||
}
|
||
|
||
const material = await App.createPBRMaterial(textures, materialJson.properties || {});
|
||
|
||
resolve(material);
|
||
})
|
||
} |