#!/usr/bin/env python3 """ 胡汉三千年项目 - 图像转换工具 支持 PPM -> PNG/JPG/SVG 转换 """ import os import sys from pathlib import Path import argparse def install_requirements(): """安装必要的依赖包""" import subprocess packages = [ 'Pillow>=10.0.0', # PIL的现代版本 'opencv-python>=4.8.0', # OpenCV 'svgwrite>=1.4.0', # SVG生成 'numpy>=1.24.0', # 数值计算 ] print("🔧 安装必要的Python包...") for package in packages: try: subprocess.check_call([sys.executable, '-m', 'pip', 'install', package]) print(f"✅ {package} 安装成功") except subprocess.CalledProcessError as e: print(f"❌ {package} 安装失败: {e}") return False return True def convert_ppm_to_png(ppm_path, output_path=None, quality=95): """将PPM文件转换为PNG格式""" try: from PIL import Image if output_path is None: output_path = str(Path(ppm_path).with_suffix('.png')) # 打开PPM文件 with Image.open(ppm_path) as img: # 转换为RGB模式（PPM通常是RGB） if img.mode != 'RGB': img = img.convert('RGB') # 保存为PNG img.save(output_path, 'PNG', optimize=True) print(f"✅ PPM -> PNG: {ppm_path} -> {output_path}") return output_path except ImportError: print("❌ 需要安装 Pillow: pip install Pillow") return None except Exception as e: print(f"❌ 转换失败: {e}") return None def convert_ppm_to_jpg(ppm_path, output_path=None, quality=95): """将PPM文件转换为JPG格式""" try: from PIL import Image if output_path is None: output_path = str(Path(ppm_path).with_suffix('.jpg')) with Image.open(ppm_path) as img: if img.mode != 'RGB': img = img.convert('RGB') img.save(output_path, 'JPEG', quality=quality, optimize=True) print(f"✅ PPM -> JPG: {ppm_path} -> {output_path}") return output_path except ImportError: print("❌ 需要安装 Pillow: pip install Pillow") return None except Exception as e: print(f"❌ 转换失败: {e}") return None def create_svg_template(image_path, output_path=None): """为图像创建SVG模板""" try: import svgwrite from PIL import Image if output_path is None: output_path = str(Path(image_path).with_suffix('.svg')) # 获取图像尺寸 with Image.open(image_path) as img: width, height = img.size # 创建SVG文档 dwg = svgwrite.Drawing(output_path, size=(f"{width}px", f"{height}px")) # 添加背景矩形 dwg.add(dwg.rect(insert=(0, 0), size=(width, height), fill='white', stroke='black', stroke_width=1)) # 添加标题 dwg.add(dwg.text('胡汉三千年 - 图像模板', insert=(width//2, 30), text_anchor='middle', font_size=16, font_family='Arial')) # 添加说明文字 dwg.add(dwg.text('此SVG模板需要手动添加具体内容', insert=(width//2, height-30), text_anchor='middle', font_size=12, font_family='Arial')) dwg.save() print(f"✅ SVG模板创建: {output_path}") return output_path except ImportError: print("❌ 需要安装 svgwrite: pip install svgwrite") return None except Exception as e: print(f"❌ SVG创建失败: {e}") return None def batch_convert_directory(directory_path, formats=['png', 'jpg']): """批量转换目录中的所有PPM文件""" directory = Path(directory_path) if not directory.exists(): print(f"❌ 目录不存在: {directory_path}") return ppm_files = list(directory.rglob('*.ppm')) if not ppm_files: print(f"❌ 在 {directory_path} 中未找到PPM文件") return print(f"🔍 找到 {len(ppm_files)} 个PPM文件") converted_count = 0 for ppm_file in ppm_files: print(f"\n📁 处理: {ppm_file}") for format_type in formats: if format_type == 'png': result = convert_ppm_to_png(str(ppm_file)) elif format_type == 'jpg': result = convert_ppm_to_jpg(str(ppm_file)) elif format_type == 'svg': result = create_svg_template(str(ppm_file)) if result: converted_count += 1 print(f"\n🎉 批量转换完成! 成功转换 {converted_count} 个文件") def analyze_image_content(image_path): """分析图像内容并生成描述""" try: from PIL import Image import numpy as np with Image.open(image_path) as img: width, height = img.size mode = img.mode # 转换为numpy数组进行分析 img_array = np.array(img) print(f"📊 图像分析: {image_path}") print(f" 尺寸: {width} x {height}") print(f" 模式: {mode}") print(f" 数据类型: {img_array.dtype}") print(f" 形状: {img_array.shape}") # 分析颜色分布 if len(img_array.shape) == 3: # RGB图像 unique_colors = len(np.unique(img_array.reshape(-1, img_array.shape[-1]), axis=0)) print(f" 唯一颜色数: {unique_colors}") return { 'width': width, 'height': height, 'mode': mode, 'shape': img_array.shape } except Exception as e: print(f"❌ 图像分析失败: {e}") return None def main(): parser = argparse.ArgumentParser(description='胡汉三千年项目 - 图像转换工具') parser.add_argument('--install', action='store_true', help='安装必要的依赖包') parser.add_argument('--convert', type=str, help='转换单个PPM文件') parser.add_argument('--batch', type=str, help='批量转换目录中的所有PPM文件') parser.add_argument('--analyze', type=str, help='分析图像内容') parser.add_argument('--formats', nargs='+', default=['png', 'jpg'], help='转换格式 (png, jpg, svg)') args = parser.parse_args() if args.install: if install_requirements(): print("🎉 所有依赖包安装完成!") else: print("❌ 依赖包安装失败") sys.exit(1) elif args.convert: ppm_path = args.convert if not os.path.exists(ppm_path): print(f"❌ 文件不存在: {ppm_path}") sys.exit(1) print(f"🔄 转换文件: {ppm_path}") for format_type in args.formats: if format_type == 'png': convert_ppm_to_png(ppm_path) elif format_type == 'jpg': convert_ppm_to_jpg(ppm_path) elif format_type == 'svg': create_svg_template(ppm_path) elif args.batch: print(f"🔄 批量转换目录: {args.batch}") batch_convert_directory(args.batch, args.formats) elif args.analyze: image_path = args.analyze if not os.path.exists(image_path): print(f"❌ 文件不存在: {image_path}") sys.exit(1) analyze_image_content(image_path) else: print("🎯 胡汉三千年项目 - 图像转换工具") print("\n使用方法:") print(" python image_converter.py --install # 安装依赖") print(" python image_converter.py --convert file.ppm # 转换单个文件") print(" python image_converter.py --batch images/ # 批量转换目录") print(" python image_converter.py --analyze file.png # 分析图像") print(" python image_converter.py --formats png jpg svg # 指定转换格式") if __name__ == '__main__': main()