io_mesh_uv_layout/export_uv_eps.py

   1 # SPDX-License-Identifier: GPL-2.0-or-later
   2
   3 import bpy
   4
   5
   6 def export(filepath, face_data, colors, width, height, opacity):
   7     with open(filepath, 'w', encoding='utf-8') as file:
   8         for text in get_file_parts(face_data, colors, width, height, opacity):
   9             file.write(text)
  10
  11
  12 def get_file_parts(face_data, colors, width, height, opacity):
  13     yield from header(width, height)
  14     if opacity > 0.0:
  15         name_by_color = {}
  16         yield from prepare_colors(colors, name_by_color)
  17         yield from draw_colored_polygons(face_data, name_by_color, width, height)
  18     yield from draw_lines(face_data, width, height)
  19     yield from footer()
  20
  21
  22 def header(width, height):
  23     yield "%!PS-Adobe-3.0 EPSF-3.0\n"
  24     yield f"%%Creator: Blender {bpy.app.version_string}\n"
  25     yield "%%Pages: 1\n"
  26     yield "%%Orientation: Portrait\n"
  27     yield f"%%BoundingBox: 0 0 {width} {height}\n"
  28     yield f"%%HiResBoundingBox: 0.0 0.0 {width:.4f} {height:.4f}\n"
  29     yield "%%EndComments\n"
  30     yield "%%Page: 1 1\n"
  31     yield "0 0 translate\n"
  32     yield "1.0 1.0 scale\n"
  33     yield "0 0 0 setrgbcolor\n"
  34     yield "[] 0 setdash\n"
  35     yield "1 setlinewidth\n"
  36     yield "1 setlinejoin\n"
  37     yield "1 setlinecap\n"
  38
  39
  40 def prepare_colors(colors, out_name_by_color):
  41     for i, color in enumerate(colors):
  42         name = f"COLOR_{i}"
  43         yield "/%s {" % name
  44         out_name_by_color[color] = name
  45
  46         yield "gsave\n"
  47         yield "%.3g %.3g %.3g setrgbcolor\n" % color
  48         yield "fill\n"
  49         yield "grestore\n"
  50         yield "0 setgray\n"
  51         yield "} def\n"
  52
  53
  54 def draw_colored_polygons(face_data, name_by_color, width, height):
  55     for uvs, color in face_data:
  56         yield from draw_polygon_path(uvs, width, height)
  57         yield "closepath\n"
  58         yield "%s\n" % name_by_color[color]
  59
  60
  61 def draw_lines(face_data, width, height):
  62     for uvs, _ in face_data:
  63         yield from draw_polygon_path(uvs, width, height)
  64         yield "closepath\n"
  65         yield "stroke\n"
  66
  67
  68 def draw_polygon_path(uvs, width, height):
  69     yield "newpath\n"
  70     for j, uv in enumerate(uvs):
  71         uv_scale = (uv[0] * width, uv[1] * height)
  72         if j == 0:
  73             yield "%.5f %.5f moveto\n" % uv_scale
  74         else:
  75             yield "%.5f %.5f lineto\n" % uv_scale
  76
  77
  78 def footer():
  79     yield "showpage\n"
  80     yield "%%EOF\n"