io_mesh_uv_layout/export_uv_eps.py

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