java中将pdf转图片后，将图片拼接下载

java中将pdf转图片后，将图片拼接下载

<!-- https://mvnrepository.com/artifact/org.apache.pdfbox/pdfbox -->
        <dependency>
            <groupId>org.apache.pdfbox</groupId>
            <artifactId>pdfbox</artifactId>
            <version>2.0.29</version>
        </dependency>

	public static void main(String[] args) throws Exception {
		//pdf路径
		String pdfPath = "D:\\opt\\data\\9\\2023-08-30\\report2023-08-26_2023-08-30-64ef31cbc09305596fc7d309.pdf";
		//生成图片保存位置
		String pngPath = "D:\\opt\\data\\9\\2023-09-05\\test.png";
		//从pdf中获取图片，并将多张图片拼接成一张，最后导出
	    PDDocument document = PDDocument.load(new File(pdfPath));
	    //根据pdf文档获取图片集合
	    List<BufferedImage> images = extractImagesFromPDF(document);
	    BufferedImage bufferedImage = mergeImage(false, images);
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        ImageIO.write(bufferedImage, "png", baos);
        byte[] imageBytes = baos.toByteArray();
        FileOutputStream fileOut = new FileOutputStream(pngPath);
        fileOut.write(imageBytes);
        fileOut.close();
	}

	/**
     * 根据pdf文档获取图片集合
     * @param document
     * @return
     * @throws IOException
     */
    private static List<BufferedImage> extractImagesFromPDF(PDDocument document) throws IOException {
        List<BufferedImage> images = new ArrayList<>();
        PDFRenderer renderer = new PDFRenderer(document);
        for (int i = 0; i < document.getNumberOfPages(); i++) {
            //PDPage page = document.getPage(i);
            BufferedImage image = renderer.renderImageWithDPI(i, 148.5f, ImageType.RGB);
            images.add(image);
        }
        return images;
    }

/**
     * 合并任数量的图片成一张图片
     *
     * @param isHorizontal true代表水平合并，false代表垂直合并
     * @param images       待合并的图片数组
     * @return BufferedImage
     */
    public static BufferedImage mergeImage(boolean isHorizontal, List<BufferedImage> images) {
        // 生成新图片
        BufferedImage destImage;
        // 计算新图片的长和高
        int allWidth = 0, allHeight = 0, allWidthMax = 0, allHeightMax = 0;
        // 获取总长、总宽、最长、最宽
        for (int i = 0; i < images.size(); i++) {
            BufferedImage img = images.get(i);
            allWidth += img.getWidth();
            if (images.size() != i + 1) {
                allHeight += img.getHeight() + 2;
            } else {
            	allHeight += img.getHeight();
            }
            if (img.getWidth() > allWidthMax) {
                allWidthMax = img.getWidth();
            }
            if (img.getHeight() > allHeightMax) {
                allHeightMax = img.getHeight();
            }
        }
        // 创建新图片
        if (isHorizontal) {
            destImage = new BufferedImage(allWidth, allHeightMax, BufferedImage.TYPE_INT_RGB);
        } else {
            destImage = new BufferedImage(allWidthMax, allHeight, BufferedImage.TYPE_INT_RGB);
        }
        // 合并所有子图片到新图片
        int wx = 0, wy = 0;
        for (BufferedImage img : images) {
            int w1 = img.getWidth();
            int h1 = img.getHeight();
            // 从图片中读取RGB
            int[] imageArrayOne = new int[w1 * h1];
            // 逐行扫描图像中各个像素的RGB到数组中
            imageArrayOne = img.getRGB(0, 0, w1, h1, imageArrayOne, 0, w1);
            if (isHorizontal) {
                // 水平方向合并
                // 设置上半部分或左半部分的RGB
                destImage.setRGB(wx, 0, w1, h1, imageArrayOne, 0, w1);
            } else {
                // 垂直方向合并
                // 设置上半部分或左半部分的RGB
                destImage.setRGB(0, wy, w1, h1, imageArrayOne, 0, w1);
            }
            wx += w1;
            wy += h1 + 2;
        }
        return destImage;
    }