Java实现文档视频缩略图：Spire插件实战

1
<!-- Spire.Doc -->
2
<dependency>
3
    <groupId>e-iceblue</groupId>
4
    <artifactId>spire.doc.free</artifactId>
5
    <version>5.4.2</version>
6
</dependency>
7

8
<!-- Spire.Presentation -->
9
<dependency>
10
    <groupId>e-iceblue</groupId>
11
    <artifactId>spire.presentation.free</artifactId>
12
    <version>5.4.0</version>
13
</dependency>
14

15
<!-- Spire.Pdf -->
16
<dependency>
17
    <groupId>e-iceblue</groupId>
18
    <artifactId>spire.pdf.free</artifactId>
19
    <version>8.6.0</version>
20
</dependency>
21

22
<!-- JavaCV（视频截帧） -->
23
<dependency>
24
    <groupId>org.bytedeco</groupId>
25
    <artifactId>javacv-platform</artifactId>
26
    <version>1.5.9</version>
27
</dependency>

1
private static String videoImage(String videoFile, String outputImgPath) throws FrameGrabber.Exception {
2
    FFmpegFrameGrabber ff = FFmpegFrameGrabber.createDefault(videoFile);
3
    ff.start();
4
    Frame f = ff.grabImage();
5
    String pngPath = getFileName(videoFile) + ".png";
6
    String resPath = outputImgPath + pngPath;
7
    String imageMat = "PNG";
8
    if (null == f || null == f.image) {
9
        return "";
10
    }
11
    Java2DFrameConverter converter = new Java2DFrameConverter();
12
    BufferedImage bufferedImage = converter.getBufferedImage(f);
13
    File output = new File(resPath);
14
    try {
15
        ImageIO.write(bufferedImage, imageMat, output);
16
    } catch (IOException e) {
17
        e.printStackTrace();
18
    }
19
    log.info("video缩略图获取完成，图片目录：[{}]", resPath);
20
    ff.stop();
21
    return resPath;
22
}

1
public static String generateThumbnail(String file, String outputImgPath) throws Exception {
2
    if (!outputImgPath.endsWith(File.separator)) {
3
        outputImgPath += File.separator;
4
    }
5
    String filePath = "";
6
    if (file.endsWith("doc") | file.endsWith("docx")) {
7
        filePath = wordToImage(file, outputImgPath);
8
    } else if (file.endsWith("ppt") | file.endsWith("pptx")) {
9
        filePath = pptToImage(file, outputImgPath);
10
    } else if (file.endsWith("pdf")) {
11
        filePath = pdfToImage(file, outputImgPath);
12
    } else if (file.endsWith("mp4")) {
13
        filePath = videoImage(file, outputImgPath);
14
    }
15
    return filePath;
16
}

1
import com.spire.doc.Document;
2
import com.spire.doc.documents.ImageType;
3
import com.spire.pdf.PdfDocument;
4
import com.spire.pdf.graphics.PdfImageType;
5
import com.spire.presentation.Presentation;
6
import lombok.extern.slf4j.Slf4j;
7
import org.bytedeco.javacv.FFmpegFrameGrabber;
8
import org.bytedeco.javacv.Frame;
9
import org.bytedeco.javacv.FrameGrabber;
10
import org.bytedeco.javacv.Java2DFrameConverter;
11

12
import javax.imageio.ImageIO;
13
import java.awt.image.BufferedImage;
14
import java.io.File;
15
import java.io.IOException;
16

17
@Slf4j
18
public class ThumbnailUtil {
19

20
    private static String getFileName(String filePath) {
21
        String fileName = new File(filePath).getName();
22
        return System.currentTimeMillis() + "-" + fileName.substring(0, fileName.lastIndexOf("."));
23
    }
24

25
    private static String wordToImage(String wordFile, String outputImgPath) throws Exception {
26
        Document word = new Document();
27
        word.loadFromFile(wordFile);
28
        String fileName = getFileName(wordFile);
29
        BufferedImage image = word.saveToImages(0, ImageType.Bitmap);
30
        String imgUrl = outputImgPath + fileName + (0 + 1) + ".png";
31
        File file = new File(imgUrl);
32
        ImageIO.write(image, "PNG", file);
33
        log.info("word缩略图获取完成，图片目录：[{}]", imgUrl);
34
        return imgUrl;
35
    }
36

37
    private static String pptToImage(String pptFile, String outputImgPath) throws Exception {
38
        Presentation ppt = new Presentation();
39
        ppt.loadFromFile(pptFile);
40
        String fileName = getFileName(pptFile);
41
        BufferedImage image = ppt.getSlides().get(0).saveAsImage();
42
        String imgUrl = outputImgPath + fileName + (0 + 1) + ".png";
43
        ImageIO.write(image, "PNG", new File(imgUrl));
44
        ppt.dispose();
45
        log.info("ppt缩略图获取完成，图片目录：[{}]", imgUrl);
46
        return imgUrl;
47
    }
48

49
    private static String pdfToImage(String pdfFile, String outputImgPath) throws Exception {
50
        PdfDocument pdf = new PdfDocument();
51
        pdf.loadFromFile(pdfFile);
52
        String fileName = getFileName(pdfFile);
53
        BufferedImage image = pdf.saveAsImage(0, PdfImageType.Bitmap);
54
        String imgUrl = outputImgPath + fileName + (0 + 1) + ".png";
55
        File file = new File(imgUrl);
56
        ImageIO.write(image, "PNG", file);
57
        log.info("pdf缩略图获取完成，图片目录：[{}]", imgUrl);
58
        return imgUrl;
59
    }
60

61
    private static String videoImage(String videoFile, String outputImgPath) throws FrameGrabber.Exception {
62
        FFmpegFrameGrabber ff = FFmpegFrameGrabber.createDefault(videoFile);
63
        ff.start();
64
        Frame f = ff.grabImage();
65
        String pngPath = getFileName(videoFile) + ".png";
66
        String resPath = outputImgPath + pngPath;
67
        if (null == f || null == f.image) {
68
            return "";
69
        }
70
        Java2DFrameConverter converter = new Java2DFrameConverter();
71
        BufferedImage bufferedImage = converter.getBufferedImage(f);
72
        File output = new File(resPath);
73
        try {
74
            ImageIO.write(bufferedImage, "PNG", output);
75
        } catch (IOException e) {
76
            e.printStackTrace();
77
        }
78
        log.info("video缩略图获取完成，图片目录：[{}]", resPath);
79
        ff.stop();
80
        return resPath;
81
    }
82

83
    public static String generateThumbnail(String file, String outputImgPath) throws Exception {
84
        if (!outputImgPath.endsWith(File.separator)) {
85
            outputImgPath += File.separator;
86
        }
87
        String filePath = "";
88
        if (file.endsWith("doc") | file.endsWith("docx")) {
89
            filePath = wordToImage(file, outputImgPath);
90
        } else if (file.endsWith("ppt") | file.endsWith("pptx")) {
91
            filePath = pptToImage(file, outputImgPath);
92
        } else if (file.endsWith("pdf")) {
93
            filePath = pdfToImage(file, outputImgPath);
94
        } else if (file.endsWith("mp4")) {
95
            filePath = videoImage(file, outputImgPath);
96
        }
97
        return filePath;
98
    }
99

100
    public static void main(String[] args) throws Exception {
101
        String path = generateThumbnail("D:\\desktop\\test\\video.mp4",
102
                "D:\\desktop\\test\\img");
103
        System.out.println(path);
104
    }
105
}