|
|
const sharp = require('sharp');
|
|
|
const fs = require('fs');
|
|
|
const path = require('path');
|
|
|
const { getInputDir } = require('../utils/index');
|
|
|
const { merge, cover } = require('../config/index');
|
|
|
|
|
|
//工作目录文件夹路径
|
|
|
const inputDir = getInputDir();
|
|
|
if (!inputDir) {
|
|
|
console.log('未执行ppt拼图功能，工作目录不存在');
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
//需要拼图的文件夹
|
|
|
const merageImgDir = `${inputDir}/img`;
|
|
|
const pdfDir = `${inputDir}/pdf`;
|
|
|
if (!merageImgDir) {
|
|
|
console.log('未执行ppt拼图功能，工作目录下的图片文件夹不存在');
|
|
|
return;
|
|
|
}
|
|
|
|
|
|
//创建预览图输出文件夹
|
|
|
const previewDir = `${inputDir}/预览图`;
|
|
|
if (!fs.existsSync(previewDir)) {
|
|
|
fs.mkdirSync(previewDir);
|
|
|
}
|
|
|
|
|
|
async function stitchImages(imagePaths, { row, bgColor, width, margin, padding }, outputName) {
|
|
|
if (imagePaths.length >= 25) {
|
|
|
row = 3; // 超过25张图片时，每行最多3张图片
|
|
|
}
|
|
|
|
|
|
// 获取第一张图片的信息
|
|
|
const firstImageInfo = await sharp(imagePaths[0]).metadata();
|
|
|
const firstImageHeight = Math.floor((firstImageInfo.height / firstImageInfo.width) * width); // 根据比例计算高度
|
|
|
|
|
|
// 处理剩余的图片
|
|
|
const remainingImagesInfo = await Promise.all(
|
|
|
imagePaths.slice(1).map(async (imagePath) => {
|
|
|
const { data, info } = await sharp(imagePath)
|
|
|
.resize({ width: Math.floor((width - padding * 2 * row) / row), fit: 'inside' })
|
|
|
.toBuffer({ resolveWithObject: true });
|
|
|
return { data, info }; // 返回图片数据及其高度
|
|
|
})
|
|
|
);
|
|
|
|
|
|
const rows = Math.ceil(remainingImagesInfo.length / row); // 计算剩余图片的行数
|
|
|
const totalHeight = margin * 2 + (padding * 2 + firstImageHeight) + (remainingImagesInfo[0]?.info.height + padding * 2) * rows; // 总高度
|
|
|
const totalWidth = width + 2 * margin; // 总宽度，包括外层间隙
|
|
|
|
|
|
// 创建拼接后的图片
|
|
|
const outputImage = await sharp({
|
|
|
create: {
|
|
|
width: totalWidth,
|
|
|
height: totalHeight,
|
|
|
channels: 4,
|
|
|
background: bgColor,
|
|
|
},
|
|
|
})
|
|
|
.composite([
|
|
|
{
|
|
|
input: await sharp(imagePaths[0])
|
|
|
.resize(width - 2 * padding, firstImageHeight)
|
|
|
.toBuffer(),
|
|
|
top: margin + padding, // 第一张图片顶部留出外层间隙
|
|
|
left: margin + padding, // 第一张图片左侧留出外层间隙
|
|
|
},
|
|
|
...remainingImagesInfo.map(({ data, info }, index) => {
|
|
|
const currentRow = Math.floor(index / row);
|
|
|
const currentCol = index % row;
|
|
|
return {
|
|
|
input: data,
|
|
|
top: margin + padding + firstImageHeight + 2 * padding + currentRow * (info.height + padding * 2), // 计算纵向位置
|
|
|
left: margin + padding + currentCol * padding * 2 + currentCol * info.width, // 计算横向位置并取整
|
|
|
};
|
|
|
}),
|
|
|
])
|
|
|
.toFile(outputName);
|
|
|
|
|
|
console.log('拼接完成，输出文件为:', outputName);
|
|
|
}
|
|
|
|
|
|
const stitchPromises = []; // 存储所有拼图任务的 Promise
|
|
|
// 读取输入目录下所有文件夹，将每个文件夹内的图片拼图
|
|
|
fs.readdirSync(merageImgDir, { withFileTypes: true }).forEach((imgDir) => {
|
|
|
if (imgDir.isDirectory()) {
|
|
|
const inputPath = path.join(merageImgDir, imgDir.name);
|
|
|
const imagePaths = [];
|
|
|
fs.readdirSync(inputPath).forEach((file) => {
|
|
|
const ext = file.split('.').pop().toLowerCase();
|
|
|
if (cover.imgFormat.has(ext)) {
|
|
|
const imgPath = path.join(inputPath, file);
|
|
|
imagePaths.push(imgPath);
|
|
|
}
|
|
|
});
|
|
|
const sortImagePaths = imagePaths.sort((a, b) => {
|
|
|
// 提取数字部分
|
|
|
const numA = parseInt(a.match(/幻灯片(\d+)\.png/)[1], 10);
|
|
|
const numB = parseInt(b.match(/幻灯片(\d+)\.png/)[1], 10);
|
|
|
|
|
|
return numA - numB; // 按数字排序
|
|
|
});
|
|
|
const outputName = path.join(previewDir, `${imgDir.name}.jpg`);
|
|
|
stitchPromises.push(stitchImages(sortImagePaths, merge, outputName));
|
|
|
}
|
|
|
});
|
|
|
|
|
|
// 等待所有拼图任务完成
|
|
|
Promise.all(stitchPromises)
|
|
|
.then(() => {
|
|
|
console.log('所有拼图任务完成');
|
|
|
fs.rmSync(merageImgDir, { recursive: true, force: true });
|
|
|
fs.rmSync(pdfDir, { recursive: true, force: true });
|
|
|
console.log('删除img和pdf文件夹任务完成');
|
|
|
})
|
|
|
.catch((err) => {
|
|
|
console.error('发生错误:', err);
|
|
|
});
|