files_processor/pdf_jpg.py at master · linbeyoung/files_processor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# -*- coding: utf-8 -*-
# @Time   : 2022/1/6 23:15
# @Author : beyoung
# @Email  : linbeyoung@stu.pku.edu.cn
# @File   : pdf_cut.py

from PyPDF2 import PdfFileWriter, PdfFileReader
import os
import time

import PyPDF4
import fitz
import pikepdf
from PIL import Image


# 将每一页转化为图片并保存
def pdf_image(pdf_name, st, ed, Gray=True, n=3):
    img_paths = []
    pdf = fitz.Document(pdf_name)
    # for i, pg in enumerate(range(0, pdf.pageCount)):
    for i, pg in enumerate(range(st, ed)):
        page = pdf[pg]  # 获得每一页的对象
        trans = fitz.Matrix(3.0, 3.0).preRotate(0)
        pm = page.getPixmap(matrix=trans, alpha=False)  # 获得每一页的流对象
        # pm.writePNG(dir_name + os.sep + base_name[:-4] + '_' + '{:0>3d}.png'.format(pg + 1))  # 保存图片
        # if not os.path.exists(pdf_name[:-4]):
        #     os.makedirs(pdf_name[:-4])

        img_folder_name = pdf_name.split('/')[-1][:-4]
        # print(os.path.dirname(pdf_name), img_folder_name)
        img_folder_pth = os.path.join(os.path.dirname(pdf_name), img_folder_name)
        if not os.path.exists(img_folder_pth):
            os.mkdir(img_folder_pth)
        print(img_folder_pth)
        # img_path = os.path.join(img_folder_pth, pdf_name[:-4] + '_' + str(pg + 1) + '.jpg')
        img_path = img_folder_pth + '/' + pdf_name.split('/')[-1][:-4] + '_' + str((pg + 1)).zfill(n) + '.png'
        print(img_path)
        # print(pdf_name[:-4], pdf_name[:-4] + '_' + str(pg + 1) + '.jpg')
        pm.writePNG(img_path)  # 保存图片
        img_paths.append(img_path)

        if Gray:  # 是否转为灰度
            img = Image.open(img_path)
            # img.show()
            low = img.convert('L')
            low.save(img_path)

            '''
            这种模式转换的灰度图片size比较大
            img = cv2.imread(img_path, 0)
            # cv2.imshow("img", img)
            cv2.imwrite(img_path, img)
            '''
    pdf.close()
    return img_paths


def cut_pdf(start_page, end_page, ip_pdf_pth, op_pdf_pth=''):
    # 开始页
    # start_page = 0

    # 截止页
    # end_page = 5
    if op_pdf_pth == '':
        op_pdf_pth = ip_pdf_pth[:-4] + '_cut.pdf'
    output = PdfFileWriter()
    pdf_file = PdfFileReader(open(ip_pdf_pth, "rb"))
    pdf_pages_len = pdf_file.getNumPages()

    # 保存input.pdf中的1-5页到output.pdf
    for i in range(start_page - 1, end_page):
        output.addPage(pdf_file.getPage(i))

    outputStream = open(op_pdf_pth, "wb")
    output.write(outputStream)


def cut_pdf_img(sp, ep, ipdf_pth, opdf_pth=''):

    if opdf_pth == '':
        opdf_pth = ipdf_pth[:-4] + '_png.pdf'
    cut_pdf(sp, ep, ipdf_pth, opdf_pth)
    # pdf_image(opdf_pth, Gray=True)  # 是否灰度读取
    pdf_image(ipdf_pth, sp - 1, ep - 1, Gray=False)


if __name__ == '__main__':
    cut_pdf_img(10, 29, '../corpus/K877.5-2015-清华简5-清华大学出土文献研究与保护中心.pdf')  # 9
    # print(os.path.dirname(__file__))  # /Users/Beyoung/Desktop/Projects/files_processor