JAVA利用pdfbox将pdf转图片

1.引入依赖

<dependency>
    <groupId>cn.hutoolgroupId>
    <artifactId>hutool-allartifactId>
    <version>5.8.12version>
dependency>

<dependency>
    <groupId>org.apache.pdfboxgroupId>
    <artifactId>pdfboxartifactId>
    <version>2.0.26version>
dependency>

2.编写工具类

public class PdfUtil {
    /**
     * 经过测试,dpi为96,100,105,120,150,200中, 120,150,200显示效果较为清晰,体积稳定,dpi越高图片体积越大,耗时越久
     */
    public static final int DEFAULT_DPI = 150;

    /**
     * pdf按页转图片 pdf有多少页就会生成多少张图片
     * @param inputStream pdf文件流列表
     */
    public static List<BufferedImage> pdfConvertImageList(InputStream inputStream) {
        PDDocument pdf = null;
        try {
            pdf = PDDocument.load(inputStream);
            PDFRenderer pdfRenderer = new PDFRenderer(pdf);
            int actSize = pdf.getNumberOfPages();
            List<BufferedImage> picList = new ArrayList<>();
            for (int i = 0; i < actSize; i++) {
                BufferedImage image = pdfRenderer.renderImageWithDPI(i, DEFAULT_DPI, ImageType.RGB);
                picList.add(image);
            }
            return picList;
        } catch (IOException e) {
            System.out.println("pdf转图片失败");
        } finally {
            IoUtil.close(pdf);
        }
        return Collections.emptyList();
    }


    /**
     * pdf按页转图片流 pdf有多少页就会生成多少图片流
     * @param inputStream pdf流
     * @param imgType 转后的图片类型 例: png jpg ImgUtil.IMAGE_TYPE_PNG
     */
    public static List<InputStream> pdfConvertImageStreamList(InputStream inputStream,String imgType){
        return pdfConvertImageList(inputStream).stream().map( bi ->ImgUtil.toStream(bi, imgType)).collect(Collectors.toList());
    }


    /**
     * pdf转图片 整个pdf会转成一张图片
     * @param inputStream pdf文件流
     */
    public static BufferedImage pdfConvertImage(InputStream inputStream){
        return pdfConvertImageList(inputStream).stream().reduce( (bi1,bi2)->mergeImage(bi1,bi2,false)).get();
    }

    /**
     * pdf转图片 整个pdf会转成一张图片流
     * @param inputStream pdf流
     * @param imgType 转后的图片类型 例: png jpg ImgUtil.IMAGE_TYPE_PNG
     */
    public static InputStream pdfConvertImageStream(InputStream inputStream,String imgType){
        return ImgUtil.toStream(pdfConvertImage(inputStream),imgType);
    }


    /**
     * 图片拼接
     * @param img1 图片1
     * @param img2 图片2
     * @param isHorizontal 是否水平拼接 true水平拼接 false垂直拼接
     * @return 拼接后的图片
     */
    private static BufferedImage mergeImage(BufferedImage img1, BufferedImage img2, boolean isHorizontal) {
        int w1 = img1.getWidth();
        int h1 = img1.getHeight();
        int w2 = img2.getWidth();
        int h2 = img2.getHeight();

        // 从图片中读取RGB
        int[] ImageArrayOne = new int[w1 * h1];
        ImageArrayOne = img1.getRGB(0, 0, w1, h1, ImageArrayOne, 0, w1); // 逐行扫描图像中各个像素的RGB到数组中
        int[] ImageArrayTwo = new int[w2 * h2];
        ImageArrayTwo = img2.getRGB(0, 0, w2, h2, ImageArrayTwo, 0, w2);

        // 生成新图片
        BufferedImage DestImage = null;
        if (isHorizontal) { // 水平方向合并
            DestImage = new BufferedImage(w1+w2, h1, BufferedImage.TYPE_INT_RGB);
            DestImage.setRGB(0, 0, w1, h1, ImageArrayOne, 0, w1); // 设置上半部分或左半部分的RGB
            DestImage.setRGB(w1, 0, w2, h2, ImageArrayTwo, 0, w2);
        } else { // 垂直方向合并
            DestImage = new BufferedImage(w1, h1 + h2, BufferedImage.TYPE_INT_RGB);
            DestImage.setRGB(0, 0, w1, h1, ImageArrayOne, 0, w1); // 设置上半部分或左半部分的RGB
            DestImage.setRGB(0, h1, w2, h2, ImageArrayTwo, 0, w2); // 设置下半部分的RGB
        }

        return DestImage;
    }
}

3.测试

public static void main(String[] args) {
        InputStream inputStream = PdfUtil.pdfConvertImageStream(FileUtil.getInputStream("E:\\test.pdf"), ImgUtil.IMAGE_TYPE_PNG);
        FileUtil.writeFromStream(inputStream,"E:\\test\\test.png");

}

你可能感兴趣的:(bug及常见问题汇总,java,pdf)