package com.product.file.util;
|
|
import java.awt.image.BufferedImage;
|
import java.io.BufferedReader;
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.FileReader;
|
import java.io.IOException;
|
import java.io.InputStreamReader;
|
import java.nio.charset.StandardCharsets;
|
|
import javax.imageio.ImageIO;
|
|
import org.apache.fontbox.ttf.TrueTypeCollection;
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
import org.apache.pdfbox.pdmodel.PDPage;
|
import org.apache.pdfbox.pdmodel.PDPageContentStream;
|
import org.apache.pdfbox.pdmodel.common.PDRectangle;
|
import org.apache.pdfbox.pdmodel.font.PDType0Font;
|
import org.apache.pdfbox.rendering.PDFRenderer;
|
import org.apache.pdfbox.text.PDFTextStripper;
|
|
public class PDFOperateUtil {
|
|
|
public static void createPdfWithCode(String javaFilePath, String outputPath) throws IOException {
|
String javaCode = readJavaFileWithComments(javaFilePath);
|
|
try (PDDocument document = new PDDocument()) {
|
PDPage page = new PDPage(PDRectangle.A4);
|
document.addPage(page);
|
|
// 加载字体（如 SimSun）
|
File fontFile = getSystemFontFile();
|
if (!fontFile.exists()) {
|
throw new IOException("未找到中文字体文件");
|
}
|
TrueTypeCollection ttc = new TrueTypeCollection(fontFile);
|
// 通过字体名称获取（如 "SimSun"）
|
PDType0Font font = PDType0Font.load(document, ttc.getFontByName("SimSun"), true);
|
// PDType0Font font = PDType0Font.load(document, fontFile);
|
float fontSize = 10;
|
float margin = 50;
|
float leading = 1.2f * fontSize; // 行距
|
|
// 可用宽度 = 页面宽度 - 左右边距
|
PDRectangle mediaBox = page.getMediaBox();
|
float availableWidth = mediaBox.getWidth() - 2 * margin;
|
|
// 初始 Y 位置（从顶部开始）
|
float yPosition = mediaBox.getHeight() - margin;
|
float startX = margin;
|
|
// 分割代码为行
|
String[] lines = javaCode.split("\n");
|
|
PDPageContentStream contentStream = null;
|
try {
|
contentStream = new PDPageContentStream(document, page);
|
contentStream.setFont(font, fontSize);
|
|
for (String line : lines) {
|
// 如果当前行超出页面底部，换页
|
if (yPosition < margin) {
|
contentStream.close();
|
page = new PDPage(PDRectangle.A4);
|
document.addPage(page);
|
yPosition = mediaBox.getHeight() - margin;
|
contentStream = new PDPageContentStream(document, page);
|
contentStream.setFont(font, fontSize);
|
}
|
|
// 处理超宽行：按字符拆分并换行
|
while (!line.isEmpty()) {
|
// 计算当前行能容纳的最大字符数
|
int maxChars = calculateMaxChars(font, fontSize, line, availableWidth);
|
String subLine = line.substring(0, Math.min(maxChars, line.length()));
|
line = line.substring(subLine.length());
|
|
// 写入当前子行
|
writeTextLine(contentStream, subLine, startX, yPosition);
|
yPosition -= leading;
|
|
// 如果剩余内容仍超出页面底部，换页
|
if (yPosition < margin && !line.isEmpty()) {
|
contentStream.close();
|
page = new PDPage(PDRectangle.A4);
|
document.addPage(page);
|
yPosition = mediaBox.getHeight() - margin;
|
contentStream = new PDPageContentStream(document, page);
|
contentStream.setFont(font, fontSize);
|
}
|
}
|
}
|
} finally {
|
if (contentStream != null) {
|
contentStream.close();
|
}
|
}
|
|
document.save(outputPath);
|
}
|
}
|
|
// 计算一行能容纳的最大字符数
|
private static int calculateMaxChars(PDType0Font font, float fontSize, String text, float availableWidth) throws IOException {
|
float width = 0;
|
int i = 0;
|
for (; i < text.length(); i++) {
|
width += font.getWidth(text.charAt(i)) / 1000 * fontSize;
|
if (width > availableWidth) {
|
break;
|
}
|
}
|
return i;
|
}
|
|
// 写入单行文本（左对齐）
|
private static void writeTextLine(PDPageContentStream contentStream, String text, float x, float y) throws IOException {
|
contentStream.beginText();
|
contentStream.newLineAtOffset(x, y);
|
contentStream.showText(text);
|
contentStream.endText();
|
}
|
|
/**
|
* 读取 Java 文件内容（保留注释和空行）
|
*/
|
private static String readJavaFileWithComments(String filePath) throws IOException {
|
StringBuilder codeBuilder = new StringBuilder();
|
try (BufferedReader reader = new BufferedReader(new FileReader(filePath))) {
|
String line;
|
while ((line = reader.readLine()) != null) {
|
line = line.replace("\t", " ");
|
// 保留所有行（包括注释和空行）
|
codeBuilder.append(line).append("\n");
|
}
|
}
|
return codeBuilder.toString();
|
}
|
|
/**
|
* 计算字体高度（基于字体的 Ascent 和 Descent）
|
*/
|
private static float calculateFontHeight(PDType0Font font, float fontSize) {
|
float ascent = font.getFontDescriptor().getAscent();
|
float descent = font.getFontDescriptor().getDescent();
|
return (ascent - descent) / 1000 * fontSize;
|
}
|
|
/**
|
* 根据文本文件创建 PDF（使用 PDFBox）
|
*
|
* @param textFilePath 输入文本文件路径
|
* @param pdfFilePath 输出 PDF 文件路径
|
* @throws IOException
|
*/
|
public static void createPdfFromTextFile(String textFilePath, String pdfFilePath) throws IOException {
|
String content = readTextFile(textFilePath);
|
content = content.replace("\t", " ");
|
try (PDDocument document = new PDDocument()) {
|
PDPage page = new PDPage(PDRectangle.A4);
|
document.addPage(page);
|
|
// 获取字体文件（仅支持 Windows/Linux）
|
File fontFile = getSystemFontFile();
|
if (!fontFile.exists()) {
|
throw new IOException("未找到支持的中文字体文件（Windows/Linux）");
|
}
|
TrueTypeCollection ttc = new TrueTypeCollection(fontFile);
|
|
// 通过字体名称获取（如 "SimSun"）
|
PDType0Font font = PDType0Font.load(document, ttc.getFontByName("SimSun"), true);
|
|
// 加载字体
|
// PDType0Font font;
|
// try (FileInputStream fontStream = new FileInputStream(fontFile)) {
|
// font = PDType0Font.load(document, fontStream);
|
// }
|
|
// 写入 PDF 内容
|
try (PDPageContentStream contentStream = new PDPageContentStream(document, page)) {
|
contentStream.beginText();
|
contentStream.setFont(font, 12);
|
contentStream.newLineAtOffset(50, 700);
|
|
String[] lines = content.split("\n");
|
for (String line : lines) {
|
contentStream.showText(line);
|
contentStream.newLineAtOffset(0, -15);
|
}
|
contentStream.endText();
|
}
|
|
document.save(pdfFilePath);
|
}
|
}
|
|
/**
|
* 读取 PDF 内容（使用 PDFBox）
|
*
|
* @param pdfFilePath PDF 文件路径
|
* @return 提取的文本内容
|
* @throws IOException
|
*/
|
public static String readPdfContent(String pdfFilePath) throws IOException {
|
try (PDDocument document = PDDocument.load(new File(pdfFilePath))) {
|
PDFTextStripper stripper = new PDFTextStripper();
|
return stripper.getText(document);
|
}
|
}
|
|
/**
|
* 将 PDF 转换为图片（使用 PDFBox）
|
*
|
* @param pdfFilePath PDF 文件路径
|
* @param outputFolder 输出图片文件夹
|
* @param imageName 输出图片文件名
|
* @param imageFormat 图片格式（如 "png", "jpg"）
|
* @throws IOException
|
*/
|
public static void convertPdfToImages(String pdfFilePath, String outputFolder, String imageName, String imageFormat) throws IOException {
|
try (PDDocument document = PDDocument.load(new File(pdfFilePath))) {
|
PDFRenderer renderer = new PDFRenderer(document);
|
|
// 确保输出文件夹存在
|
new File(outputFolder).mkdirs();
|
|
// 逐页渲染为图片（300 DPI）
|
for (int page = 0; page < document.getNumberOfPages(); page++) {
|
BufferedImage image = renderer.renderImageWithDPI(page, 300);
|
String outputFileName = String.format("%s/%s%d.%s", outputFolder, imageName, page + 1, imageFormat);
|
ImageIO.write(image, imageFormat, new File(outputFileName));
|
}
|
}
|
}
|
|
/**
|
* 辅助方法：读取文本文件内容
|
*
|
* @param filePath 文本文件路径
|
* @return 文件内容
|
* @throws IOException
|
*/
|
private static String readTextFile(String filePath) throws IOException {
|
StringBuilder content = new StringBuilder();
|
try (BufferedReader reader = new BufferedReader(
|
new InputStreamReader(new FileInputStream(filePath), StandardCharsets.UTF_8))) {
|
String line;
|
while ((line = reader.readLine()) != null) {
|
content.append(line).append("\n");
|
}
|
}
|
return content.toString();
|
}
|
|
/**
|
* 获取系统字体文件（仅 Windows/Linux）
|
*/
|
private static File getSystemFontFile() {
|
String os = System.getProperty("os.name").toLowerCase();
|
String[] possiblePaths;
|
|
if (os.contains("win")) {
|
// Windows 字体目录（优先检查常见中文字体）
|
possiblePaths = new String[]{
|
// System.getenv("WINDIR") + "\\Fonts\\arial.ttf", //
|
// System.getenv("WINDIR") + "\\Fonts\\simkai.ttf", // 楷体
|
// System.getenv("WINDIR") + "\\Fonts\\simfang.ttf", // 宋体
|
System.getenv("WINDIR") + "\\Fonts\\simsun.ttc", // 宋体
|
System.getenv("WINDIR") + "\\Fonts\\simhei.ttf", // 黑体
|
System.getenv("WINDIR") + "\\Fonts\\msyh.ttc", // 微软雅黑
|
System.getenv("WINDIR") + "\\Fonts\\mingliu.ttc" // 明体（备用）
|
};
|
} else {
|
// Linux 字体目录（优先检查开源中文字体）
|
possiblePaths = new String[]{
|
"/usr/share/fonts/truetype/wqy/wqy-microhei.ttc", // 文泉驿微米黑
|
"/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc", // 思源黑体
|
"/usr/share/fonts/truetype/arphic/ukai.ttc" // 文鼎字体（备用）
|
};
|
}
|
|
// 检查所有可能的路径
|
for (String path : possiblePaths) {
|
File fontFile = new File(path);
|
if (fontFile.exists()) {
|
return fontFile;
|
}
|
}
|
|
// 如果系统字体未找到，尝试从项目资源目录加载（需手动放入字体文件）
|
File embeddedFont = new File("fonts/simhei.ttf"); // 假设字体文件放在项目根目录的 fonts/ 文件夹下
|
return embeddedFont.exists() ? embeddedFont : null;
|
}
|
|
// 示例用法
|
public static void main(String[] args) {
|
try {
|
// 1. 从文本文件创建 PDF
|
// createPdfFromTextFile("E://新建1.txt", "E://新建1.pdf");
|
createPdfWithCode("E:\\PROJECT\\Product\\后端\\product\\product-server\\product-server-file\\src\\main\\java\\com\\product\\file\\util\\FileUtil.java", "E://新建1.pdf");
|
|
// 2. 读取 PDF 内容
|
String content = readPdfContent("E://新建1.pdf");
|
System.out.println("PDF 内容:\n" + content);
|
|
// 3. 将 PDF 转换为图片
|
convertPdfToImages("E://新建1.pdf", "E://", "新建1", "png");
|
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
