88 lines
3.7 KiB
Python
88 lines
3.7 KiB
Python
import gdstk
|
||
import cairosvg
|
||
import argparse
|
||
import os
|
||
|
||
def convert_layout_to_png_via_svg(layout_path, png_path, cell_name=None, pixels_per_unit=10):
|
||
"""
|
||
通过先生成 SVG 再转换为 PNG 的方式，将 GDSII 或 OASIS 文件光栅化。
|
||
此版本修正了 write_svg 的参数错误，兼容性更强。
|
||
|
||
参数:
|
||
layout_path (str): 输入的版图文件路径（.gds 或 .oas）。
|
||
png_path (str): 输出的 PNG 文件路径。
|
||
cell_name (str, optional): 需要转换的单元名称。如果为 None，则使用顶层单元。
|
||
pixels_per_unit (int, optional): 版图数据库单位到像素的转换比例，控制图像分辨率。
|
||
"""
|
||
print(f"正在从 '{layout_path}' 读取版图文件...")
|
||
|
||
# 1. 加载版图文件
|
||
_, extension = os.path.splitext(layout_path)
|
||
extension = extension.lower()
|
||
|
||
if extension == '.gds':
|
||
lib = gdstk.read_gds(layout_path)
|
||
elif extension == '.oas':
|
||
lib = gdstk.read_oas(layout_path)
|
||
else:
|
||
raise ValueError(f"不支持的文件类型: '{extension}'。请输入 .gds 或 .oas 文件。")
|
||
|
||
if cell_name:
|
||
cell = lib.cells[cell_name]
|
||
else:
|
||
top_cells = lib.top_level()
|
||
if not top_cells:
|
||
raise ValueError("错误：版图文件中没有找到顶层单元。")
|
||
cell = top_cells[0]
|
||
print(f"未指定单元名称，自动选择顶层单元: '{cell.name}'")
|
||
|
||
# 2. 将版图单元写入临时的 SVG 文件 (已移除无效的 padding 参数)
|
||
temp_svg_path = png_path + ".temp.svg"
|
||
print(f"步骤 1/2: 正在将单元 '{cell.name}' 转换为临时 SVG 文件...")
|
||
cell.write_svg(
|
||
temp_svg_path # 隐藏默认字体，避免影响边界
|
||
)
|
||
|
||
# 3. 使用 cairosvg 将 SVG 文件转换为 PNG
|
||
print(f"步骤 2/2: 正在将 SVG 转换为 PNG...")
|
||
# 获取单元的精确边界框
|
||
bb = cell.bb()
|
||
if bb is None:
|
||
raise ValueError(f"单元 '{cell.name}' 为空或无法获取其边界框。")
|
||
|
||
# 根据边界框和分辨率计算输出图像的宽度
|
||
width, height = bb[1] - bb[0]
|
||
output_width = width * pixels_per_unit
|
||
|
||
cairosvg.svg2png(url=temp_svg_path, write_to=png_path, output_width=output_width)
|
||
|
||
# 4. 清理临时的 SVG 文件
|
||
os.remove(temp_svg_path)
|
||
|
||
print(f"成功！图像已保存至: '{png_path}'")
|
||
|
||
if __name__ == "__main__":
|
||
parser = argparse.ArgumentParser(
|
||
description="将 GDSII (.gds) 或 OASIS (.oas) 版图文件转换为 PNG 图像 (通过SVG)。",
|
||
epilog="示例: python rasterize.py -i my_chip.oas -o my_chip.png -ppu 20"
|
||
)
|
||
parser.add_argument('-i', '--input', type=str, required=True, help="输入的版图文件路径 (.gds 或 .oas)。")
|
||
parser.add_argument('-o', '--output', type=str, help="输出的 PNG 文件路径。如果未提供，将使用输入文件名并替换扩展名为 .png。")
|
||
parser.add_argument('-c', '--cell', type=str, default=None, help="要转换的特定单元的名称。默认为顶层单元。")
|
||
parser.add_argument('-ppu', '--pixels_per_unit', type=int, default=10, help="每微米（um）的像素数，用于控制输出图像的分辨率。")
|
||
args = parser.parse_args()
|
||
|
||
if not args.output:
|
||
base_name = os.path.splitext(os.path.basename(args.input))[0]
|
||
args.output = f"{base_name}.png"
|
||
print(f"未指定输出路径，将自动保存为: '{args.output}'")
|
||
|
||
try:
|
||
convert_layout_to_png_via_svg(
|
||
layout_path=args.input,
|
||
png_path=args.output,
|
||
cell_name=args.cell,
|
||
pixels_per_unit=args.pixels_per_unit
|
||
)
|
||
except Exception as e:
|
||
print(f"\n处理失败: {e}") |