JAVA根据PDF文件生成图片

心已赠人 2023-10-05 16:50 3阅读 0赞

### PDF文件生成图片 ###

##### 实现功能 #####

根据上传的PDF文件,生成图片文件,
    单页PDF,生成图片文件.
    多页PDF,则生成zip压缩包.

##### 一 : 文件生成效果 #####

![单页PDF文件][PDF]  
![多页PDF文件][PDF 1]

##### 二 : 引入所需maven依赖 #####

项目采用springboot框架

<dependency>
            <groupId>org.icepdf.os</groupId>
            <artifactId>icepdf-core</artifactId>
            <version>6.2.2</version>
            <exclusions>
                <exclusion>
                    <groupId>javax.media</groupId>
                    <artifactId>jai_core</artifactId>
                </exclusion>
            </exclusions>
        </dependency>

##### 三 : JAVA代码工具类 #####

生成图片工具类 PdfToImageUtil.java

package com.util;
    
    import com.lowagie.text.pdf.PdfReader;
    import lombok.extern.slf4j.Slf4j;
    import org.icepdf.core.pobjects.Document;
    import org.icepdf.core.pobjects.Page;
    import org.icepdf.core.util.GraphicsRenderingHints;
    import org.springframework.web.multipart.MultipartFile;
    import javax.imageio.ImageIO;
    import javax.servlet.http.HttpServletResponse;
    import java.awt.image.BufferedImage;
    import java.io.BufferedInputStream;
    import java.io.File;
    import java.io.FileInputStream;
    import java.io.FileOutputStream;
    import java.io.IOException;
    import java.io.OutputStream;
    import java.net.URLEncoder;
    import java.util.ArrayList;
    import java.util.Arrays;
    import java.util.List;
    import java.util.zip.ZipEntry;
    import java.util.zip.ZipOutputStream;
    
    @Slf4j
    public class PdfToImageUtil {
        
    
        //支持文件格式
        public static final String SUPPORT_FILE = "pdf";
        //图片文件格式
        public static final String IMAGE_SUFFIX = "jpg"; //png
        //压缩文件格式
        public static final String ZIP_SUFFIX = "zip";
        //PDF是否为一页
        private static boolean isImage;
    
        /**
         * 对外的开放接口，用于将PDF文件转换为图片文件压缩包进行下载
         * @param file SpringMVC获取的图片文件
         */
        public static synchronized void pdfToTransformation(MultipartFile file, HttpServletResponse response) throws Exception {
        
            String fileName = file.getOriginalFilename();
            if(null == fileName) return;
            String suffix = fileName.substring(fileName.lastIndexOf(".") + 1);
            log.info("文件名称：" + fileName + ",文件后缀：" + suffix);
            if(!SUPPORT_FILE.equals(suffix)) return;
            isImage = false;
            File imageFile = generateFile(file);
            log.info("文件生成成功!");
            downloadFile(imageFile, response);
        }
    
        /**
         * 将PDF文件转换为多张图片并放入一个压缩包中
         * @param file SpringMVC获取的图片文件
         * @return 图片文件压缩包
         * @throws Exception 抛出异常
         */
        private static File generateFile(MultipartFile file) throws Exception {
        
            String fileName = file.getOriginalFilename();
            if(null == fileName) return null;
            Document document = new Document();
            document.setByteArray(file.getBytes(), 0, file.getBytes().length, fileName);
            log.info("PDF页数：" + document.getNumberOfPages());
            isImage = 1 == document.getNumberOfPages();
            File imageReturnFile = null;
            List<File> fileList = new ArrayList<>();
            for (int i = 0; i < document.getNumberOfPages(); i++) {
        
                BufferedImage image = (BufferedImage) document.getPageImage(i, GraphicsRenderingHints.SCREEN,
                        Page.BOUNDARY_CROPBOX, 0F, 2.5F);
                File imageFile = new File((i + 1) + "." + IMAGE_SUFFIX);
                ImageIO.write(image, IMAGE_SUFFIX, imageFile);
                image.flush();
                if(isImage){
        
                    imageReturnFile = imageFile;
                    break;
                }
                fileList.add(imageFile);
            }
            document.dispose();
            if(isImage) return imageReturnFile;
            //压缩图片文件
            String directoryName = fileName.substring(0, fileName.lastIndexOf("."));
            File zipFile = new File(directoryName + "." + ZIP_SUFFIX);
            ZipOutputStream zipOutputStream = new ZipOutputStream(new FileOutputStream(zipFile));
            zipFile(fileList, zipOutputStream);
            zipOutputStream.close();
            return zipFile;
        }
    
        /**
         * 下载image/zip文件
         * @param downloadFile 文件
         * @param response HttpServletResponse
         * @throws IOException IO异常
         */
        private static void downloadFile(File downloadFile, HttpServletResponse response) throws IOException {
        
            FileInputStream fileInputStream = new FileInputStream(downloadFile);
            byte[] bytes = new byte[fileInputStream.available()];
            fileInputStream.read(bytes);
            fileInputStream.close();
            //设置response参数
            response.reset();
            if(isImage){
        
                response.setContentType("image/jpeg");
            } else {
        
                response.setContentType("application/zip");
            }
            response.setCharacterEncoding("UTF-8");
            response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(downloadFile.getName(), "UTF-8"));
            OutputStream outputStream = response.getOutputStream();
            outputStream.write(bytes);
            outputStream.flush();
            outputStream.close();
            if(!isImage) downloadFile.delete();
        }
    
        /**
         * 压缩文件
         * @param inputFiles 具体需要压缩的文件集合
         * @param zipOutputStream ZipOutputStream对象
         * @throws IOException IO异常
         */
        private static void zipFile(List<File> inputFiles, ZipOutputStream zipOutputStream) throws IOException {
        
            byte[] buffer = new byte[1024];
            for (File file : inputFiles) {
        
                if (file.exists()) {
        
                    if (file.isFile()) {
        
                        BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
                        zipOutputStream.putNextEntry(new ZipEntry(file.getName()));
                        int size = 0;
                        while ((size = bis.read(buffer)) > 0) {
        
                            zipOutputStream.write(buffer, 0, size);
                        }
                        zipOutputStream.closeEntry();
                        bis.close();
                        file.delete();
                    } else {
        
                        File[] files = file.listFiles();
                        if(null == files) continue;
                        List<File> childrenFileList = Arrays.asList(files);
                        zipFile(childrenFileList, zipOutputStream);
                    }
                }
            }
        }
    
        /**
         * 获取PDF页数
         * @throws IOException
         */
        public static String getPdfPageSize(MultipartFile file) throws IOException {
        
            PdfReader pdfReader = new PdfReader(file.getBytes());
            int pages = pdfReader.getNumberOfPages();
            return String.valueOf(pages);
        }
    }

生成图片PdfController.java

package com.web.transformation.controller;
    
    import com.util.PdfToImageUtil;
    import org.springframework.stereotype.Controller;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.ResponseBody;
    import org.springframework.web.multipart.MultipartFile;
    import javax.servlet.http.HttpServletResponse;
    import java.io.IOException;
    
    @Controller
    public class PdfController {
        
    
        //PDF转图片或者ZIP
        @RequestMapping("/pdfToImage")
        @ResponseBody
        public void pdfToImage(MultipartFile file, HttpServletResponse response) throws Exception{
        
            PdfToImageUtil.pdfToTransformation(file,response);
        }
    
        //获取PDF页数
        @RequestMapping("/getPdfPageSize")
        @ResponseBody
        public String pdfToImage(MultipartFile file) throws IOException {
        
            return PdfToImageUtil.getPdfPageSize(file);
        }
    
    }

##### 四 : 说明 #####

生成的图片文件格式,可手动修改,此处设置为jpg,也可设置为png  
也可将该配置,放置在yml配置文件中  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4MjU0NjM1_size_16_color_FFFFFF_t_70]

##### 五 : 相关链接 #####

JAVA生成Zip文件并导出：  
[https://blog.csdn.net/qq\_38254635/article/details/127364398][https_blog.csdn.net_qq_38254635_article_details_127364398]

JAVA导出Excel文件：  
[https://blog.csdn.net/qq\_38254635/article/details/126691175][https_blog.csdn.net_qq_38254635_article_details_126691175]

JAVA根据模板生成WORD文件并导出：  
[https://blog.csdn.net/qq\_38254635/article/details/103952823][https_blog.csdn.net_qq_38254635_article_details_103952823]

JAVA根据模板生成PDF文件并导出：  
[https://blog.csdn.net/qq\_38254635/article/details/103919024][https_blog.csdn.net_qq_38254635_article_details_103919024]

[PDF]: https://img-blog.csdnimg.cn/2020123116531658.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4MjU0NjM1,size_16,color_FFFFFF,t_70
[PDF 1]: https://img-blog.csdnimg.cn/20201231165455339.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4MjU0NjM1,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4MjU0NjM1_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20201231170155719.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4MjU0NjM1,size_16,color_FFFFFF,t_70
[https_blog.csdn.net_qq_38254635_article_details_127364398]: https://blog.csdn.net/qq_38254635/article/details/127364398
[https_blog.csdn.net_qq_38254635_article_details_126691175]: https://blog.csdn.net/qq_38254635/article/details/126691175
[https_blog.csdn.net_qq_38254635_article_details_103952823]: https://blog.csdn.net/qq_38254635/article/details/103952823
[https_blog.csdn.net_qq_38254635_article_details_103919024]: https://blog.csdn.net/qq_38254635/article/details/103919024