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.IOException;
|
import java.io.InputStreamReader;
|
import java.nio.charset.StandardCharsets;
|
|
import javax.imageio.ImageIO;
|
|
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 {
|
|
/**
|
* 根据文本文件创建 PDF（使用 PDFBox）
|
*
|
* @param textFilePath 输入文本文件路径
|
* @param pdfFilePath 输出 PDF 文件路径
|
* @throws IOException
|
*/
|
public static void createPdfFromTextFile(String textFilePath, String pdfFilePath) throws IOException {
|
String content = readTextFile(textFilePath);
|
|
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）");
|
}
|
|
// 加载字体
|
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\\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");
|
|
// 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();
|
}
|
}
|
}
|
