190 lines
6.6 KiB
TypeScript
190 lines
6.6 KiB
TypeScript
import fs, { writeFileSync } from "fs";
|
||
import path from "path";
|
||
import ExcelJS from "exceljs";
|
||
import { PDFDocument, StandardFonts, rgb } from "pdf-lib";
|
||
|
||
/**
|
||
* 读取指定路径下的所有.stl文件
|
||
* @param dirPath 要搜索的目录路径
|
||
* @returns 返回包含所有.stl文件名的数组
|
||
*/
|
||
export const findSTLFiles = (dirPath: string): Promise<string[]> => {
|
||
return new Promise((resolve) => {
|
||
// 检查路径是否存在
|
||
if (!fs.existsSync(dirPath)) {
|
||
resolve([]); // 路径不存在时返回空数组
|
||
return;
|
||
}
|
||
|
||
// 异步读取目录内容
|
||
fs.readdir(dirPath, (err, files) => {
|
||
if (err) {
|
||
resolve([]); // 读取错误时返回空数组
|
||
} else {
|
||
// 过滤出.stl文件
|
||
const stlFiles = files.filter(
|
||
(file) => path.extname(file).toLowerCase() === ".stl"
|
||
);
|
||
resolve(stlFiles);
|
||
}
|
||
});
|
||
});
|
||
};
|
||
|
||
export const createPDF = async (savePath: string) => {
|
||
// 创建一个新的 PDF 文档
|
||
const pdfDoc = await PDFDocument.create();
|
||
|
||
// 自定义纸张大小，例如：宽 600，高 800
|
||
const page = pdfDoc.addPage([600, 800]);
|
||
|
||
// 自定义边距，例如左右上下都是 50
|
||
const margin = 50;
|
||
const pageWidth = page.getWidth();
|
||
const pageHeight = page.getHeight();
|
||
|
||
// 设置字体
|
||
const font = await pdfDoc.embedFont(StandardFonts.Helvetica);
|
||
|
||
// 绘制透明色块
|
||
// 设置颜色为半透明蓝色，透明度为 0.5
|
||
const color = rgb(0.11, 0.62, 0.92, 0.5); // 最后一个值是透明度
|
||
page.drawRectangle({
|
||
x: margin,
|
||
y: margin,
|
||
width: pageWidth - 2 * margin,
|
||
height: pageHeight - 2 * margin,
|
||
color: color,
|
||
opacity: 0.5,
|
||
});
|
||
|
||
// 添加文本
|
||
page.drawText("这里是自定义边距内的文本。", {
|
||
x: margin,
|
||
y: pageHeight - margin - 30, // 文本的位置考虑了顶部边距
|
||
size: 12,
|
||
font: font,
|
||
color: rgb(0, 0, 0),
|
||
});
|
||
|
||
// 保存 PDF
|
||
const pdfBytes = await pdfDoc.save();
|
||
writeFileSync(path.join(savePath, "报告.pdf"), pdfBytes);
|
||
};
|
||
|
||
/**
|
||
* json存excel
|
||
*/
|
||
export const createExcelFile = async (savePath: string, measurement) => {
|
||
// 创建一个新的工作簿
|
||
const workbook = new ExcelJS.Workbook();
|
||
// 添加一个工作表
|
||
const worksheet = workbook.addWorksheet("My Sheet");
|
||
|
||
// 添加列标题
|
||
worksheet.columns = [
|
||
{ key: "1", header: "瓣环平面周长", width: 10 },
|
||
{ key: "2", header: "瓣环平面周长导出径", width: 10 },
|
||
{ key: "3", header: "瓣环平面面积", width: 10 },
|
||
{ key: "4", header: "瓣环平面面积导出径", width: 10 },
|
||
{ key: "5", header: "瓣环平面长径", width: 10 },
|
||
{ key: "6", header: "瓣环平面短径", width: 10 },
|
||
{ key: "7", header: "瓣环平面平均径", width: 10 },
|
||
//
|
||
{ key: "8", header: "窦管交界周长", width: 10 },
|
||
{ key: "9", header: "窦管交界周长导出径", width: 10 },
|
||
{ key: "10", header: "窦管交界面积", width: 10 },
|
||
{ key: "11", header: "窦管交界面积导出径", width: 10 },
|
||
{ key: "12", header: "窦管交界长径", width: 10 },
|
||
{ key: "13", header: "窦管交界短径", width: 10 },
|
||
{ key: "14", header: "窦管交界平均径", width: 10 },
|
||
//
|
||
{ key: "15", header: "左室流出道周长", width: 10 },
|
||
{ key: "16", header: "左室流出道周长导出径", width: 10 },
|
||
{ key: "17", header: "左室流出道面积", width: 10 },
|
||
{ key: "18", header: "左室流出道面积导出径", width: 10 },
|
||
{ key: "19", header: "左室流出道长径", width: 10 },
|
||
{ key: "20", header: "左室流出道短径", width: 10 },
|
||
{ key: "21", header: "左室流出道平均径", width: 10 },
|
||
//
|
||
{ key: "22", header: "升主动脉（瓣环上4cm）周长", width: 10 },
|
||
{ key: "23", header: "升主动脉（瓣环上4cm）周长导出径", width: 10 },
|
||
{ key: "24", header: "升主动脉（瓣环上4cm）面积", width: 10 },
|
||
{ key: "25", header: "升主动脉（瓣环上4cm）面积导出径", width: 10 },
|
||
{ key: "26", header: "升主动脉（瓣环上4cm）长径", width: 10 },
|
||
{ key: "27", header: "升主动脉（瓣环上4cm）短径", width: 10 },
|
||
{ key: "28", header: "升主动脉（瓣环上4cm）平均径", width: 10 },
|
||
//
|
||
{ key: "29", header: "瓦氏窦周长", width: 10 },
|
||
{ key: "30", header: "瓦氏窦周长导出径", width: 10 },
|
||
{ key: "31", header: "瓦氏窦面积", width: 10 },
|
||
{ key: "32", header: "瓦氏窦面积导出径", width: 10 },
|
||
{ key: "33", header: "瓦氏窦长径", width: 10 },
|
||
{ key: "34", header: "瓦氏窦短径", width: 10 },
|
||
{ key: "35", header: "瓦氏窦平均径", width: 10 },
|
||
//
|
||
{ key: "36", header: "左侧冠脉开口高度", width: 10 },
|
||
{ key: "37", header: "右侧冠脉开口高度", width: 10 },
|
||
{ key: "38", header: "钙化信息", width: 10 },
|
||
{ key: "39", header: "主动脉夹角", width: 10 },
|
||
];
|
||
|
||
const annulus_plane = measurement["annulus_plane"];
|
||
const STJ_plane = measurement["STJ_plane"];
|
||
const LVOT_plane = measurement["LVOT_plane"];
|
||
const nominal_aorta_plane = measurement["nominal_aorta_plane"];
|
||
const SOV_plane = measurement["SOV_plane"];
|
||
const calcification_analysis =
|
||
measurement["calcification_analysis"]["HU_statistic"];
|
||
|
||
worksheet.addRow({
|
||
1: annulus_plane["perimeter"],
|
||
2: annulus_plane["PED"],
|
||
3: annulus_plane["area"],
|
||
4: annulus_plane["AED"],
|
||
5: annulus_plane["max_dist"],
|
||
6: annulus_plane["min_dist"],
|
||
7: annulus_plane["average_dist"],
|
||
//
|
||
8: STJ_plane["perimeter"],
|
||
9: STJ_plane["PED"],
|
||
10: STJ_plane["area"],
|
||
11: STJ_plane["AED"],
|
||
12: STJ_plane["max_dist"],
|
||
13: STJ_plane["min_dist"],
|
||
14: STJ_plane["average_dist"],
|
||
//
|
||
15: LVOT_plane["perimeter"],
|
||
16: LVOT_plane["PED"],
|
||
17: LVOT_plane["area"],
|
||
18: LVOT_plane["AED"],
|
||
19: LVOT_plane["max_dist"],
|
||
20: LVOT_plane["min_dist"],
|
||
21: LVOT_plane["average_dist"],
|
||
//
|
||
22: nominal_aorta_plane["perimeter"],
|
||
23: nominal_aorta_plane["PED"],
|
||
24: nominal_aorta_plane["area"],
|
||
25: nominal_aorta_plane["AED"],
|
||
26: nominal_aorta_plane["max_dist"],
|
||
27: nominal_aorta_plane["min_dist"],
|
||
28: nominal_aorta_plane["average_dist"],
|
||
//
|
||
29: SOV_plane["perimeter"],
|
||
30: SOV_plane["PED"],
|
||
31: SOV_plane["area"],
|
||
32: SOV_plane["AED"],
|
||
33: SOV_plane["max_dist"],
|
||
34: SOV_plane["min_dist"],
|
||
35: SOV_plane["average_dist"],
|
||
//
|
||
38: calcification_analysis.map((i) => `HU${i.key}:${i.value}`).join(";"),
|
||
39: annulus_plane["horizontal_angle"],
|
||
});
|
||
|
||
// 设置样式
|
||
worksheet.getRow(1).font = { bold: true };
|
||
// 保存到文件
|
||
await workbook.xlsx.writeFile(path.join(savePath));
|
||
};
|