2025 - 科研神器 - 批量处理 PDF、SVG、PNG 和 JPG 文件，将它们转换为彩色 TIFF 文件，并保存到指定的 tiff 文件夹中

'''
Created on 2024-11-22

@author: wangyang(<[email protected]>)
批量处理 PDF、SVG、PNG 和 JPG 文件，将它们转换为彩色 TIFF 文件，并保存到指定的 tiff 文件夹中。
使用 Ghostscript 和 cairosvg 库进行转换。

使用方法：
1. 打开命令行窗口，切换到脚本所在目录。
2. 输入命令：python 批量处理tiff.py
3. 按照提示输入要处理的目录路径和 DPI 值（默认 300）。
4. 等待脚本处理完毕，所有文件将被转换为彩色 TIFF 文件并保存到 tiff 文件夹中。
5. 脚本会自动删除空的 tiff 文件夹。

注意：
1. 脚本依赖 Ghostscript 和 cairosvg 库，请先安装它们。
2. 脚本仅处理 PDF、SVG、PNG 和 JPG 文件，其他文件将被忽略。
'''

import os
import subprocess
import shutil
from PIL import Image
import cairosvg
import io  # 修复未定义的 io 模块问题


def convert_pdf_to_tiff(pdf_path, tiff_folder, dpi=300):
    """将 PDF 文件转换为彩色 TIFF 文件，并保存到指定的 tiff 文件夹中"""
    try:
        base_name = os.path.splitext(os.path.basename(pdf_path))[0]
        tiff_path = os.path.join(tiff_folder, f"{base_name}.tiff")

        subprocess.run([
            "gs", "-dBATCH", "-dNOPAUSE", "-sDEVICE=tiff24nc",
            f"-r{dpi}", "-sCompression=lzw",
            "-sOutputFile=" + tiff_path, pdf_path
        ], check=True)

        print(f"成功将 PDF {pdf_path} 转换为 {tiff_path}")
    except subprocess.CalledProcessError as e:
        print(f"处理 PDF 文件 {pdf_path} 时出错: {e}")


def convert_svg_to_tiff(svg_path, tiff_folder, dpi=300):
    """将 SVG 文件转换为彩色 TIFF 文件"""
    try:
        base_name = os.path.splitext(os.path.basename(svg_path))[0]
        tiff_path = os.path.join(tiff_folder, f"{base_name}.tiff")

        # 使用 cairosvg 将 SVG 转为 TIFF
        png_data = cairosvg.svg2png(url=svg_path, dpi=dpi)
        with open(tiff_path, "wb") as tiff_file:
            img = Image.open(io.BytesIO(png_data))
            img.save(tiff_file, format="TIFF", dpi=(dpi, dpi))

        print(f"成功将 SVG {svg_path} 转换为 {tiff_path}")
    except Exception as e:
        print(f"处理 SVG 文件 {svg_path} 时出错: {e}")


def convert_image_to_tiff(image_path, tiff_folder, dpi=300):
    """将 PNG 或 JPG 文件转换为彩色 TIFF 文件"""
    try:
        base_name = os.path.splitext(os.path.basename(image_path))[0]
        tiff_path = os.path.join(tiff_folder, f"{base_name}.tiff")

        with Image.open(image_path) as img:
            img.save(tiff_path, format="TIFF", dpi=(dpi, dpi))

        print(f"成功将 {image_path} 转换为 {tiff_path}")
    except Exception as e:
        print(f"处理图像文件 {image_path} 时出错: {e}")


def clean_empty_tiff_folders(input_dir):
    """递归删除空的 TIFF 文件夹"""
    for root, dirs, _ in os.walk(input_dir, topdown=False):
        for dir_name in dirs:
            dir_path = os.path.join(root, dir_name)
            if dir_name == "tiff" and not os.listdir(dir_path):  # 如果 tiff 文件夹为空
                shutil.rmtree(dir_path)
                print(f"删除空文件夹: {dir_path}")


def process_directory_recursive(input_dir, dpi=300):
    """递归处理目录中的 PDF、SVG、PNG 和 JPG 文件"""
    for root, _, files in os.walk(input_dir):
        # 在每个目录中创建 tiff 文件夹
        tiff_folder = os.path.join(root, "tiff")
        tiff_folder_created = False  # 标记是否创建了 tiff 文件夹

        # 遍历文件，处理 PDF、SVG、PNG 和 JPG
        for file in files:
            file_path = os.path.join(root, file)
            if file.lower().endswith((".pdf", ".svg", ".png", ".jpg", ".jpeg")):
                if not tiff_folder_created:
                    os.makedirs(tiff_folder, exist_ok=True)
                    tiff_folder_created = True

                if file.lower().endswith(".pdf"):
                    convert_pdf_to_tiff(file_path, tiff_folder, dpi)
                elif file.lower().endswith(".svg"):
                    convert_svg_to_tiff(file_path, tiff_folder, dpi)
                elif file.lower().endswith((".png", ".jpg", ".jpeg")):
                    convert_image_to_tiff(file_path, tiff_folder, dpi)

    # 清理空的 TIFF 文件夹
    clean_empty_tiff_folders(input_dir)


if __name__ == "__main__":
    # 提示用户输入目录
    input_dir = input("请输入要处理的目录路径: ").strip()
    dpi_input = input("请输入 DPI 值（默认 300）: ").strip()

    try:
        dpi = int(dpi_input) if dpi_input else 300
    except ValueError:
        print("无效的 DPI 值，使用默认值 300")
        dpi = 300

    if os.path.isdir(input_dir):
        process_directory_recursive(input_dir, dpi)
        print("所有文件已成功转换为彩色 TIFF 文件！")
    else:
        print(f"路径无效：{input_dir}")