Spaces:
Running
Running
openvoice2
/
path
/to
/venv
/lib
/python3.10
/site-packages
/fontTools
/varLib
/interpolatablePlot.py
from .interpolatableHelpers import * | |
from fontTools.ttLib import TTFont | |
from fontTools.ttLib.ttGlyphSet import LerpGlyphSet | |
from fontTools.pens.recordingPen import ( | |
RecordingPen, | |
DecomposingRecordingPen, | |
RecordingPointPen, | |
) | |
from fontTools.pens.boundsPen import ControlBoundsPen | |
from fontTools.pens.cairoPen import CairoPen | |
from fontTools.pens.pointPen import ( | |
SegmentToPointPen, | |
PointToSegmentPen, | |
ReverseContourPointPen, | |
) | |
from fontTools.varLib.interpolatableHelpers import ( | |
PerContourOrComponentPen, | |
SimpleRecordingPointPen, | |
) | |
from itertools import cycle | |
from functools import wraps | |
from io import BytesIO | |
import cairo | |
import math | |
import os | |
import logging | |
log = logging.getLogger("fontTools.varLib.interpolatable") | |
class OverridingDict(dict): | |
def __init__(self, parent_dict): | |
self.parent_dict = parent_dict | |
def __missing__(self, key): | |
return self.parent_dict[key] | |
class InterpolatablePlot: | |
width = 8.5 * 72 | |
height = 11 * 72 | |
pad = 0.1 * 72 | |
title_font_size = 24 | |
font_size = 16 | |
page_number = 1 | |
head_color = (0.3, 0.3, 0.3) | |
label_color = (0.2, 0.2, 0.2) | |
border_color = (0.9, 0.9, 0.9) | |
border_width = 0.5 | |
fill_color = (0.8, 0.8, 0.8) | |
stroke_color = (0.1, 0.1, 0.1) | |
stroke_width = 1 | |
oncurve_node_color = (0, 0.8, 0, 0.7) | |
oncurve_node_diameter = 6 | |
offcurve_node_color = (0, 0.5, 0, 0.7) | |
offcurve_node_diameter = 4 | |
handle_color = (0, 0.5, 0, 0.7) | |
handle_width = 0.5 | |
corrected_start_point_color = (0, 0.9, 0, 0.7) | |
corrected_start_point_size = 7 | |
wrong_start_point_color = (1, 0, 0, 0.7) | |
start_point_color = (0, 0, 1, 0.7) | |
start_arrow_length = 9 | |
kink_point_size = 7 | |
kink_point_color = (1, 0, 1, 0.7) | |
kink_circle_size = 15 | |
kink_circle_stroke_width = 1 | |
kink_circle_color = (1, 0, 1, 0.7) | |
contour_colors = ((1, 0, 0), (0, 0, 1), (0, 1, 0), (1, 1, 0), (1, 0, 1), (0, 1, 1)) | |
contour_alpha = 0.5 | |
weight_issue_contour_color = (0, 0, 0, 0.4) | |
no_issues_label = "Your font's good! Have a cupcake..." | |
no_issues_label_color = (0, 0.5, 0) | |
cupcake_color = (0.3, 0, 0.3) | |
cupcake = r""" | |
,@. | |
,@.@@,. | |
,@@,.@@@. @.@@@,. | |
,@@. @@@. @@. @@,. | |
,@@@.@,.@. @. @@@@,.@.@@,. | |
,@@.@. @@.@@. @,. .@' @' @@, | |
,@@. @. .@@.@@@. @@' @, | |
,@. @@. @, | |
@. @,@@,. , .@@, | |
@,. .@,@@,. .@@,. , .@@, @, @, | |
@. .@. @ @@,. , @ | |
@,.@@. @,. @@,. @. @,. @' | |
@@||@,. @'@,. @@,. @@ @,. @'@@, @' | |
\\@@@@' @,. @'@@@@' @@,. @@@' //@@@' | |
|||||||| @@,. @@' ||||||| |@@@|@|| || | |
\\\\\\\ ||@@@|| ||||||| ||||||| // | |
||||||| |||||| |||||| |||||| || | |
\\\\\\ |||||| |||||| |||||| // | |
|||||| ||||| ||||| ||||| || | |
\\\\\ ||||| ||||| ||||| // | |
||||| |||| ||||| |||| || | |
\\\\ |||| |||| |||| // | |
|||||||||||||||||||||||| | |
""" | |
emoticon_color = (0, 0.3, 0.3) | |
shrug = r"""\_(")_/""" | |
underweight = r""" | |
o | |
/|\ | |
/ \ | |
""" | |
overweight = r""" | |
o | |
/O\ | |
/ \ | |
""" | |
yay = r""" \o/ """ | |
def __init__(self, out, glyphsets, names=None, **kwargs): | |
self.out = out | |
self.glyphsets = glyphsets | |
self.names = names or [repr(g) for g in glyphsets] | |
self.toc = {} | |
for k, v in kwargs.items(): | |
if not hasattr(self, k): | |
raise TypeError("Unknown keyword argument: %s" % k) | |
setattr(self, k, v) | |
self.panel_width = self.width / 2 - self.pad * 3 | |
self.panel_height = ( | |
self.height / 2 - self.pad * 6 - self.font_size * 2 - self.title_font_size | |
) | |
def __enter__(self): | |
return self | |
def __exit__(self, type, value, traceback): | |
pass | |
def show_page(self): | |
self.page_number += 1 | |
def add_title_page( | |
self, files, *, show_tolerance=True, tolerance=None, kinkiness=None | |
): | |
pad = self.pad | |
width = self.width - 3 * self.pad | |
height = self.height - 2 * self.pad | |
x = y = pad | |
self.draw_label( | |
"Problem report for:", | |
x=x, | |
y=y, | |
bold=True, | |
width=width, | |
font_size=self.title_font_size, | |
) | |
y += self.title_font_size | |
import hashlib | |
for file in files: | |
base_file = os.path.basename(file) | |
y += self.font_size + self.pad | |
self.draw_label(base_file, x=x, y=y, bold=True, width=width) | |
y += self.font_size + self.pad | |
try: | |
h = hashlib.sha1(open(file, "rb").read()).hexdigest() | |
self.draw_label("sha1: %s" % h, x=x + pad, y=y, width=width) | |
y += self.font_size | |
except IsADirectoryError: | |
pass | |
if file.endswith(".ttf"): | |
ttFont = TTFont(file) | |
name = ttFont["name"] if "name" in ttFont else None | |
if name: | |
for what, nameIDs in ( | |
("Family name", (21, 16, 1)), | |
("Version", (5,)), | |
): | |
n = name.getFirstDebugName(nameIDs) | |
if n is None: | |
continue | |
self.draw_label( | |
"%s: %s" % (what, n), x=x + pad, y=y, width=width | |
) | |
y += self.font_size + self.pad | |
elif file.endswith((".glyphs", ".glyphspackage")): | |
from glyphsLib import GSFont | |
f = GSFont(file) | |
for what, field in ( | |
("Family name", "familyName"), | |
("VersionMajor", "versionMajor"), | |
("VersionMinor", "_versionMinor"), | |
): | |
self.draw_label( | |
"%s: %s" % (what, getattr(f, field)), | |
x=x + pad, | |
y=y, | |
width=width, | |
) | |
y += self.font_size + self.pad | |
self.draw_legend( | |
show_tolerance=show_tolerance, tolerance=tolerance, kinkiness=kinkiness | |
) | |
self.show_page() | |
def draw_legend(self, *, show_tolerance=True, tolerance=None, kinkiness=None): | |
cr = cairo.Context(self.surface) | |
x = self.pad | |
y = self.height - self.pad - self.font_size * 2 | |
width = self.width - 2 * self.pad | |
xx = x + self.pad * 2 | |
xxx = x + self.pad * 4 | |
if show_tolerance: | |
self.draw_label( | |
"Tolerance: badness; closer to zero the worse", x=xxx, y=y, width=width | |
) | |
y -= self.pad + self.font_size | |
self.draw_label("Underweight contours", x=xxx, y=y, width=width) | |
cr.rectangle(xx - self.pad * 0.7, y, 1.5 * self.pad, self.font_size) | |
cr.set_source_rgb(*self.fill_color) | |
cr.fill_preserve() | |
if self.stroke_color: | |
cr.set_source_rgb(*self.stroke_color) | |
cr.set_line_width(self.stroke_width) | |
cr.stroke_preserve() | |
cr.set_source_rgba(*self.weight_issue_contour_color) | |
cr.fill() | |
y -= self.pad + self.font_size | |
self.draw_label( | |
"Colored contours: contours with the wrong order", x=xxx, y=y, width=width | |
) | |
cr.rectangle(xx - self.pad * 0.7, y, 1.5 * self.pad, self.font_size) | |
if self.fill_color: | |
cr.set_source_rgb(*self.fill_color) | |
cr.fill_preserve() | |
if self.stroke_color: | |
cr.set_source_rgb(*self.stroke_color) | |
cr.set_line_width(self.stroke_width) | |
cr.stroke_preserve() | |
cr.set_source_rgba(*self.contour_colors[0], self.contour_alpha) | |
cr.fill() | |
y -= self.pad + self.font_size | |
self.draw_label("Kink artifact", x=xxx, y=y, width=width) | |
self.draw_circle( | |
cr, | |
x=xx, | |
y=y + self.font_size * 0.5, | |
diameter=self.kink_circle_size, | |
stroke_width=self.kink_circle_stroke_width, | |
color=self.kink_circle_color, | |
) | |
y -= self.pad + self.font_size | |
self.draw_label("Point causing kink in the contour", x=xxx, y=y, width=width) | |
self.draw_dot( | |
cr, | |
x=xx, | |
y=y + self.font_size * 0.5, | |
diameter=self.kink_point_size, | |
color=self.kink_point_color, | |
) | |
y -= self.pad + self.font_size | |
self.draw_label("Suggested new contour start point", x=xxx, y=y, width=width) | |
self.draw_dot( | |
cr, | |
x=xx, | |
y=y + self.font_size * 0.5, | |
diameter=self.corrected_start_point_size, | |
color=self.corrected_start_point_color, | |
) | |
y -= self.pad + self.font_size | |
self.draw_label( | |
"Contour start point in contours with wrong direction", | |
x=xxx, | |
y=y, | |
width=width, | |
) | |
self.draw_arrow( | |
cr, | |
x=xx - self.start_arrow_length * 0.3, | |
y=y + self.font_size * 0.5, | |
color=self.wrong_start_point_color, | |
) | |
y -= self.pad + self.font_size | |
self.draw_label( | |
"Contour start point when the first two points overlap", | |
x=xxx, | |
y=y, | |
width=width, | |
) | |
self.draw_dot( | |
cr, | |
x=xx, | |
y=y + self.font_size * 0.5, | |
diameter=self.corrected_start_point_size, | |
color=self.start_point_color, | |
) | |
y -= self.pad + self.font_size | |
self.draw_label("Contour start point and direction", x=xxx, y=y, width=width) | |
self.draw_arrow( | |
cr, | |
x=xx - self.start_arrow_length * 0.3, | |
y=y + self.font_size * 0.5, | |
color=self.start_point_color, | |
) | |
y -= self.pad + self.font_size | |
self.draw_label("Legend:", x=x, y=y, width=width, bold=True) | |
y -= self.pad + self.font_size | |
if kinkiness is not None: | |
self.draw_label( | |
"Kink-reporting aggressiveness: %g" % kinkiness, | |
x=xxx, | |
y=y, | |
width=width, | |
) | |
y -= self.pad + self.font_size | |
if tolerance is not None: | |
self.draw_label( | |
"Error tolerance: %g" % tolerance, | |
x=xxx, | |
y=y, | |
width=width, | |
) | |
y -= self.pad + self.font_size | |
self.draw_label("Parameters:", x=x, y=y, width=width, bold=True) | |
y -= self.pad + self.font_size | |
def add_summary(self, problems): | |
pad = self.pad | |
width = self.width - 3 * self.pad | |
height = self.height - 2 * self.pad | |
x = y = pad | |
self.draw_label( | |
"Summary of problems", | |
x=x, | |
y=y, | |
bold=True, | |
width=width, | |
font_size=self.title_font_size, | |
) | |
y += self.title_font_size | |
glyphs_per_problem = defaultdict(set) | |
for glyphname, problems in sorted(problems.items()): | |
for problem in problems: | |
glyphs_per_problem[problem["type"]].add(glyphname) | |
if "nothing" in glyphs_per_problem: | |
del glyphs_per_problem["nothing"] | |
for problem_type in sorted( | |
glyphs_per_problem, key=lambda x: InterpolatableProblem.severity[x] | |
): | |
y += self.font_size | |
self.draw_label( | |
"%s: %d" % (problem_type, len(glyphs_per_problem[problem_type])), | |
x=x, | |
y=y, | |
width=width, | |
bold=True, | |
) | |
y += self.font_size | |
for glyphname in sorted(glyphs_per_problem[problem_type]): | |
if y + self.font_size > height: | |
self.show_page() | |
y = self.font_size + pad | |
self.draw_label(glyphname, x=x + 2 * pad, y=y, width=width - 2 * pad) | |
y += self.font_size | |
self.show_page() | |
def _add_listing(self, title, items): | |
pad = self.pad | |
width = self.width - 2 * self.pad | |
height = self.height - 2 * self.pad | |
x = y = pad | |
self.draw_label( | |
title, x=x, y=y, bold=True, width=width, font_size=self.title_font_size | |
) | |
y += self.title_font_size + self.pad | |
last_glyphname = None | |
for page_no, (glyphname, problems) in items: | |
if glyphname == last_glyphname: | |
continue | |
last_glyphname = glyphname | |
if y + self.font_size > height: | |
self.show_page() | |
y = self.font_size + pad | |
self.draw_label(glyphname, x=x + 5 * pad, y=y, width=width - 2 * pad) | |
self.draw_label(str(page_no), x=x, y=y, width=4 * pad, align=1) | |
y += self.font_size | |
self.show_page() | |
def add_table_of_contents(self): | |
self._add_listing("Table of contents", sorted(self.toc.items())) | |
def add_index(self): | |
self._add_listing("Index", sorted(self.toc.items(), key=lambda x: x[1][0])) | |
def add_problems(self, problems, *, show_tolerance=True, show_page_number=True): | |
for glyph, glyph_problems in problems.items(): | |
last_masters = None | |
current_glyph_problems = [] | |
for p in glyph_problems: | |
masters = ( | |
p["master_idx"] | |
if "master_idx" in p | |
else (p["master_1_idx"], p["master_2_idx"]) | |
) | |
if masters == last_masters: | |
current_glyph_problems.append(p) | |
continue | |
# Flush | |
if current_glyph_problems: | |
self.add_problem( | |
glyph, | |
current_glyph_problems, | |
show_tolerance=show_tolerance, | |
show_page_number=show_page_number, | |
) | |
self.show_page() | |
current_glyph_problems = [] | |
last_masters = masters | |
current_glyph_problems.append(p) | |
if current_glyph_problems: | |
self.add_problem( | |
glyph, | |
current_glyph_problems, | |
show_tolerance=show_tolerance, | |
show_page_number=show_page_number, | |
) | |
self.show_page() | |
def add_problem( | |
self, glyphname, problems, *, show_tolerance=True, show_page_number=True | |
): | |
if type(problems) not in (list, tuple): | |
problems = [problems] | |
self.toc[self.page_number] = (glyphname, problems) | |
problem_type = problems[0]["type"] | |
problem_types = set(problem["type"] for problem in problems) | |
if not all(pt == problem_type for pt in problem_types): | |
problem_type = ", ".join(sorted({problem["type"] for problem in problems})) | |
log.info("Drawing %s: %s", glyphname, problem_type) | |
master_keys = ( | |
("master_idx",) | |
if "master_idx" in problems[0] | |
else ("master_1_idx", "master_2_idx") | |
) | |
master_indices = [problems[0][k] for k in master_keys] | |
if problem_type == InterpolatableProblem.MISSING: | |
sample_glyph = next( | |
i for i, m in enumerate(self.glyphsets) if m[glyphname] is not None | |
) | |
master_indices.insert(0, sample_glyph) | |
x = self.pad | |
y = self.pad | |
self.draw_label( | |
"Glyph name: " + glyphname, | |
x=x, | |
y=y, | |
color=self.head_color, | |
align=0, | |
bold=True, | |
font_size=self.title_font_size, | |
) | |
tolerance = min(p.get("tolerance", 1) for p in problems) | |
if tolerance < 1 and show_tolerance: | |
self.draw_label( | |
"tolerance: %.2f" % tolerance, | |
x=x, | |
y=y, | |
width=self.width - 2 * self.pad, | |
align=1, | |
bold=True, | |
) | |
y += self.title_font_size + self.pad | |
self.draw_label( | |
"Problems: " + problem_type, | |
x=x, | |
y=y, | |
width=self.width - 2 * self.pad, | |
color=self.head_color, | |
bold=True, | |
) | |
y += self.font_size + self.pad * 2 | |
scales = [] | |
for which, master_idx in enumerate(master_indices): | |
glyphset = self.glyphsets[master_idx] | |
name = self.names[master_idx] | |
self.draw_label( | |
name, | |
x=x, | |
y=y, | |
color=self.label_color, | |
width=self.panel_width, | |
align=0.5, | |
) | |
y += self.font_size + self.pad | |
if glyphset[glyphname] is not None: | |
scales.append( | |
self.draw_glyph(glyphset, glyphname, problems, which, x=x, y=y) | |
) | |
else: | |
self.draw_emoticon(self.shrug, x=x, y=y) | |
y += self.panel_height + self.font_size + self.pad | |
if any( | |
pt | |
in ( | |
InterpolatableProblem.NOTHING, | |
InterpolatableProblem.WRONG_START_POINT, | |
InterpolatableProblem.CONTOUR_ORDER, | |
InterpolatableProblem.KINK, | |
InterpolatableProblem.UNDERWEIGHT, | |
InterpolatableProblem.OVERWEIGHT, | |
) | |
for pt in problem_types | |
): | |
x = self.pad + self.panel_width + self.pad | |
y = self.pad | |
y += self.title_font_size + self.pad * 2 | |
y += self.font_size + self.pad | |
glyphset1 = self.glyphsets[master_indices[0]] | |
glyphset2 = self.glyphsets[master_indices[1]] | |
# Draw the mid-way of the two masters | |
self.draw_label( | |
"midway interpolation", | |
x=x, | |
y=y, | |
color=self.head_color, | |
width=self.panel_width, | |
align=0.5, | |
) | |
y += self.font_size + self.pad | |
midway_glyphset = LerpGlyphSet(glyphset1, glyphset2) | |
self.draw_glyph( | |
midway_glyphset, | |
glyphname, | |
[{"type": "midway"}] | |
+ [ | |
p | |
for p in problems | |
if p["type"] | |
in ( | |
InterpolatableProblem.KINK, | |
InterpolatableProblem.UNDERWEIGHT, | |
InterpolatableProblem.OVERWEIGHT, | |
) | |
], | |
None, | |
x=x, | |
y=y, | |
scale=min(scales), | |
) | |
y += self.panel_height + self.font_size + self.pad | |
if any( | |
pt | |
in ( | |
InterpolatableProblem.WRONG_START_POINT, | |
InterpolatableProblem.CONTOUR_ORDER, | |
InterpolatableProblem.KINK, | |
) | |
for pt in problem_types | |
): | |
# Draw the proposed fix | |
self.draw_label( | |
"proposed fix", | |
x=x, | |
y=y, | |
color=self.head_color, | |
width=self.panel_width, | |
align=0.5, | |
) | |
y += self.font_size + self.pad | |
overriding1 = OverridingDict(glyphset1) | |
overriding2 = OverridingDict(glyphset2) | |
perContourPen1 = PerContourOrComponentPen( | |
RecordingPen, glyphset=overriding1 | |
) | |
perContourPen2 = PerContourOrComponentPen( | |
RecordingPen, glyphset=overriding2 | |
) | |
glyphset1[glyphname].draw(perContourPen1) | |
glyphset2[glyphname].draw(perContourPen2) | |
for problem in problems: | |
if problem["type"] == InterpolatableProblem.CONTOUR_ORDER: | |
fixed_contours = [ | |
perContourPen2.value[i] for i in problems[0]["value_2"] | |
] | |
perContourPen2.value = fixed_contours | |
for problem in problems: | |
if problem["type"] == InterpolatableProblem.WRONG_START_POINT: | |
# Save the wrong contours | |
wrongContour1 = perContourPen1.value[problem["contour"]] | |
wrongContour2 = perContourPen2.value[problem["contour"]] | |
# Convert the wrong contours to point pens | |
points1 = RecordingPointPen() | |
converter = SegmentToPointPen(points1, False) | |
wrongContour1.replay(converter) | |
points2 = RecordingPointPen() | |
converter = SegmentToPointPen(points2, False) | |
wrongContour2.replay(converter) | |
proposed_start = problem["value_2"] | |
# See if we need reversing; fragile but worth a try | |
if problem["reversed"]: | |
new_points2 = RecordingPointPen() | |
reversedPen = ReverseContourPointPen(new_points2) | |
points2.replay(reversedPen) | |
points2 = new_points2 | |
proposed_start = len(points2.value) - 2 - proposed_start | |
# Rotate points2 so that the first point is the same as in points1 | |
beginPath = points2.value[:1] | |
endPath = points2.value[-1:] | |
pts = points2.value[1:-1] | |
pts = pts[proposed_start:] + pts[:proposed_start] | |
points2.value = beginPath + pts + endPath | |
# Convert the point pens back to segment pens | |
segment1 = RecordingPen() | |
converter = PointToSegmentPen(segment1, True) | |
points1.replay(converter) | |
segment2 = RecordingPen() | |
converter = PointToSegmentPen(segment2, True) | |
points2.replay(converter) | |
# Replace the wrong contours | |
wrongContour1.value = segment1.value | |
wrongContour2.value = segment2.value | |
perContourPen1.value[problem["contour"]] = wrongContour1 | |
perContourPen2.value[problem["contour"]] = wrongContour2 | |
for problem in problems: | |
# If we have a kink, try to fix it. | |
if problem["type"] == InterpolatableProblem.KINK: | |
# Save the wrong contours | |
wrongContour1 = perContourPen1.value[problem["contour"]] | |
wrongContour2 = perContourPen2.value[problem["contour"]] | |
# Convert the wrong contours to point pens | |
points1 = RecordingPointPen() | |
converter = SegmentToPointPen(points1, False) | |
wrongContour1.replay(converter) | |
points2 = RecordingPointPen() | |
converter = SegmentToPointPen(points2, False) | |
wrongContour2.replay(converter) | |
i = problem["value"] | |
# Position points to be around the same ratio | |
# beginPath / endPath dance | |
j = i + 1 | |
pt0 = points1.value[j][1][0] | |
pt1 = points2.value[j][1][0] | |
j_prev = (i - 1) % (len(points1.value) - 2) + 1 | |
pt0_prev = points1.value[j_prev][1][0] | |
pt1_prev = points2.value[j_prev][1][0] | |
j_next = (i + 1) % (len(points1.value) - 2) + 1 | |
pt0_next = points1.value[j_next][1][0] | |
pt1_next = points2.value[j_next][1][0] | |
pt0 = complex(*pt0) | |
pt1 = complex(*pt1) | |
pt0_prev = complex(*pt0_prev) | |
pt1_prev = complex(*pt1_prev) | |
pt0_next = complex(*pt0_next) | |
pt1_next = complex(*pt1_next) | |
# Find the ratio of the distance between the points | |
r0 = abs(pt0 - pt0_prev) / abs(pt0_next - pt0_prev) | |
r1 = abs(pt1 - pt1_prev) / abs(pt1_next - pt1_prev) | |
r_mid = (r0 + r1) / 2 | |
pt0 = pt0_prev + r_mid * (pt0_next - pt0_prev) | |
pt1 = pt1_prev + r_mid * (pt1_next - pt1_prev) | |
points1.value[j] = ( | |
points1.value[j][0], | |
(((pt0.real, pt0.imag),) + points1.value[j][1][1:]), | |
points1.value[j][2], | |
) | |
points2.value[j] = ( | |
points2.value[j][0], | |
(((pt1.real, pt1.imag),) + points2.value[j][1][1:]), | |
points2.value[j][2], | |
) | |
# Convert the point pens back to segment pens | |
segment1 = RecordingPen() | |
converter = PointToSegmentPen(segment1, True) | |
points1.replay(converter) | |
segment2 = RecordingPen() | |
converter = PointToSegmentPen(segment2, True) | |
points2.replay(converter) | |
# Replace the wrong contours | |
wrongContour1.value = segment1.value | |
wrongContour2.value = segment2.value | |
# Assemble | |
fixed1 = RecordingPen() | |
fixed2 = RecordingPen() | |
for contour in perContourPen1.value: | |
fixed1.value.extend(contour.value) | |
for contour in perContourPen2.value: | |
fixed2.value.extend(contour.value) | |
fixed1.draw = fixed1.replay | |
fixed2.draw = fixed2.replay | |
overriding1[glyphname] = fixed1 | |
overriding2[glyphname] = fixed2 | |
try: | |
midway_glyphset = LerpGlyphSet(overriding1, overriding2) | |
self.draw_glyph( | |
midway_glyphset, | |
glyphname, | |
{"type": "fixed"}, | |
None, | |
x=x, | |
y=y, | |
scale=min(scales), | |
) | |
except ValueError: | |
self.draw_emoticon(self.shrug, x=x, y=y) | |
y += self.panel_height + self.pad | |
else: | |
emoticon = self.shrug | |
if InterpolatableProblem.UNDERWEIGHT in problem_types: | |
emoticon = self.underweight | |
elif InterpolatableProblem.OVERWEIGHT in problem_types: | |
emoticon = self.overweight | |
elif InterpolatableProblem.NOTHING in problem_types: | |
emoticon = self.yay | |
self.draw_emoticon(emoticon, x=x, y=y) | |
if show_page_number: | |
self.draw_label( | |
str(self.page_number), | |
x=0, | |
y=self.height - self.font_size - self.pad, | |
width=self.width, | |
color=self.head_color, | |
align=0.5, | |
) | |
def draw_label( | |
self, | |
label, | |
*, | |
x=0, | |
y=0, | |
color=(0, 0, 0), | |
align=0, | |
bold=False, | |
width=None, | |
height=None, | |
font_size=None, | |
): | |
if width is None: | |
width = self.width | |
if height is None: | |
height = self.height | |
if font_size is None: | |
font_size = self.font_size | |
cr = cairo.Context(self.surface) | |
cr.select_font_face( | |
"@cairo:", | |
cairo.FONT_SLANT_NORMAL, | |
cairo.FONT_WEIGHT_BOLD if bold else cairo.FONT_WEIGHT_NORMAL, | |
) | |
cr.set_font_size(font_size) | |
font_extents = cr.font_extents() | |
font_size = font_size * font_size / font_extents[2] | |
cr.set_font_size(font_size) | |
font_extents = cr.font_extents() | |
cr.set_source_rgb(*color) | |
extents = cr.text_extents(label) | |
if extents.width > width: | |
# Shrink | |
font_size *= width / extents.width | |
cr.set_font_size(font_size) | |
font_extents = cr.font_extents() | |
extents = cr.text_extents(label) | |
# Center | |
label_x = x + (width - extents.width) * align | |
label_y = y + font_extents[0] | |
cr.move_to(label_x, label_y) | |
cr.show_text(label) | |
def draw_glyph(self, glyphset, glyphname, problems, which, *, x=0, y=0, scale=None): | |
if type(problems) not in (list, tuple): | |
problems = [problems] | |
midway = any(problem["type"] == "midway" for problem in problems) | |
problem_type = problems[0]["type"] | |
problem_types = set(problem["type"] for problem in problems) | |
if not all(pt == problem_type for pt in problem_types): | |
problem_type = "mixed" | |
glyph = glyphset[glyphname] | |
recording = RecordingPen() | |
glyph.draw(recording) | |
decomposedRecording = DecomposingRecordingPen(glyphset) | |
glyph.draw(decomposedRecording) | |
boundsPen = ControlBoundsPen(glyphset) | |
decomposedRecording.replay(boundsPen) | |
bounds = boundsPen.bounds | |
if bounds is None: | |
bounds = (0, 0, 0, 0) | |
glyph_width = bounds[2] - bounds[0] | |
glyph_height = bounds[3] - bounds[1] | |
if glyph_width: | |
if scale is None: | |
scale = self.panel_width / glyph_width | |
else: | |
scale = min(scale, self.panel_height / glyph_height) | |
if glyph_height: | |
if scale is None: | |
scale = self.panel_height / glyph_height | |
else: | |
scale = min(scale, self.panel_height / glyph_height) | |
if scale is None: | |
scale = 1 | |
cr = cairo.Context(self.surface) | |
cr.translate(x, y) | |
# Center | |
cr.translate( | |
(self.panel_width - glyph_width * scale) / 2, | |
(self.panel_height - glyph_height * scale) / 2, | |
) | |
cr.scale(scale, -scale) | |
cr.translate(-bounds[0], -bounds[3]) | |
if self.border_color: | |
cr.set_source_rgb(*self.border_color) | |
cr.rectangle(bounds[0], bounds[1], glyph_width, glyph_height) | |
cr.set_line_width(self.border_width / scale) | |
cr.stroke() | |
if self.fill_color or self.stroke_color: | |
pen = CairoPen(glyphset, cr) | |
decomposedRecording.replay(pen) | |
if self.fill_color and problem_type != InterpolatableProblem.OPEN_PATH: | |
cr.set_source_rgb(*self.fill_color) | |
cr.fill_preserve() | |
if self.stroke_color: | |
cr.set_source_rgb(*self.stroke_color) | |
cr.set_line_width(self.stroke_width / scale) | |
cr.stroke_preserve() | |
cr.new_path() | |
if ( | |
InterpolatableProblem.UNDERWEIGHT in problem_types | |
or InterpolatableProblem.OVERWEIGHT in problem_types | |
): | |
perContourPen = PerContourOrComponentPen(RecordingPen, glyphset=glyphset) | |
recording.replay(perContourPen) | |
for problem in problems: | |
if problem["type"] in ( | |
InterpolatableProblem.UNDERWEIGHT, | |
InterpolatableProblem.OVERWEIGHT, | |
): | |
contour = perContourPen.value[problem["contour"]] | |
contour.replay(CairoPen(glyphset, cr)) | |
cr.set_source_rgba(*self.weight_issue_contour_color) | |
cr.fill() | |
if any( | |
t in problem_types | |
for t in { | |
InterpolatableProblem.NOTHING, | |
InterpolatableProblem.NODE_COUNT, | |
InterpolatableProblem.NODE_INCOMPATIBILITY, | |
} | |
): | |
cr.set_line_cap(cairo.LINE_CAP_ROUND) | |
# Oncurve nodes | |
for segment, args in decomposedRecording.value: | |
if not args: | |
continue | |
x, y = args[-1] | |
cr.move_to(x, y) | |
cr.line_to(x, y) | |
cr.set_source_rgba(*self.oncurve_node_color) | |
cr.set_line_width(self.oncurve_node_diameter / scale) | |
cr.stroke() | |
# Offcurve nodes | |
for segment, args in decomposedRecording.value: | |
if not args: | |
continue | |
for x, y in args[:-1]: | |
cr.move_to(x, y) | |
cr.line_to(x, y) | |
cr.set_source_rgba(*self.offcurve_node_color) | |
cr.set_line_width(self.offcurve_node_diameter / scale) | |
cr.stroke() | |
# Handles | |
for segment, args in decomposedRecording.value: | |
if not args: | |
pass | |
elif segment in ("moveTo", "lineTo"): | |
cr.move_to(*args[0]) | |
elif segment == "qCurveTo": | |
for x, y in args: | |
cr.line_to(x, y) | |
cr.new_sub_path() | |
cr.move_to(*args[-1]) | |
elif segment == "curveTo": | |
cr.line_to(*args[0]) | |
cr.new_sub_path() | |
cr.move_to(*args[1]) | |
cr.line_to(*args[2]) | |
cr.new_sub_path() | |
cr.move_to(*args[-1]) | |
else: | |
continue | |
cr.set_source_rgba(*self.handle_color) | |
cr.set_line_width(self.handle_width / scale) | |
cr.stroke() | |
matching = None | |
for problem in problems: | |
if problem["type"] == InterpolatableProblem.CONTOUR_ORDER: | |
matching = problem["value_2"] | |
colors = cycle(self.contour_colors) | |
perContourPen = PerContourOrComponentPen( | |
RecordingPen, glyphset=glyphset | |
) | |
recording.replay(perContourPen) | |
for i, contour in enumerate(perContourPen.value): | |
if matching[i] == i: | |
continue | |
color = next(colors) | |
contour.replay(CairoPen(glyphset, cr)) | |
cr.set_source_rgba(*color, self.contour_alpha) | |
cr.fill() | |
for problem in problems: | |
if problem["type"] in ( | |
InterpolatableProblem.NOTHING, | |
InterpolatableProblem.WRONG_START_POINT, | |
): | |
idx = problem.get("contour") | |
# Draw suggested point | |
if idx is not None and which == 1 and "value_2" in problem: | |
perContourPen = PerContourOrComponentPen( | |
RecordingPen, glyphset=glyphset | |
) | |
decomposedRecording.replay(perContourPen) | |
points = SimpleRecordingPointPen() | |
converter = SegmentToPointPen(points, False) | |
perContourPen.value[ | |
idx if matching is None else matching[idx] | |
].replay(converter) | |
targetPoint = points.value[problem["value_2"]][0] | |
cr.save() | |
cr.translate(*targetPoint) | |
cr.scale(1 / scale, 1 / scale) | |
self.draw_dot( | |
cr, | |
diameter=self.corrected_start_point_size, | |
color=self.corrected_start_point_color, | |
) | |
cr.restore() | |
# Draw start-point arrow | |
if which == 0 or not problem.get("reversed"): | |
color = self.start_point_color | |
else: | |
color = self.wrong_start_point_color | |
first_pt = None | |
i = 0 | |
cr.save() | |
for segment, args in decomposedRecording.value: | |
if segment == "moveTo": | |
first_pt = args[0] | |
continue | |
if first_pt is None: | |
continue | |
if segment == "closePath": | |
second_pt = first_pt | |
else: | |
second_pt = args[0] | |
if idx is None or i == idx: | |
cr.save() | |
first_pt = complex(*first_pt) | |
second_pt = complex(*second_pt) | |
length = abs(second_pt - first_pt) | |
cr.translate(first_pt.real, first_pt.imag) | |
if length: | |
# Draw arrowhead | |
cr.rotate( | |
math.atan2( | |
second_pt.imag - first_pt.imag, | |
second_pt.real - first_pt.real, | |
) | |
) | |
cr.scale(1 / scale, 1 / scale) | |
self.draw_arrow(cr, color=color) | |
else: | |
# Draw circle | |
cr.scale(1 / scale, 1 / scale) | |
self.draw_dot( | |
cr, | |
diameter=self.corrected_start_point_size, | |
color=color, | |
) | |
cr.restore() | |
if idx is not None: | |
break | |
first_pt = None | |
i += 1 | |
cr.restore() | |
if problem["type"] == InterpolatableProblem.KINK: | |
idx = problem.get("contour") | |
perContourPen = PerContourOrComponentPen( | |
RecordingPen, glyphset=glyphset | |
) | |
decomposedRecording.replay(perContourPen) | |
points = SimpleRecordingPointPen() | |
converter = SegmentToPointPen(points, False) | |
perContourPen.value[idx if matching is None else matching[idx]].replay( | |
converter | |
) | |
targetPoint = points.value[problem["value"]][0] | |
cr.save() | |
cr.translate(*targetPoint) | |
cr.scale(1 / scale, 1 / scale) | |
if midway: | |
self.draw_circle( | |
cr, | |
diameter=self.kink_circle_size, | |
stroke_width=self.kink_circle_stroke_width, | |
color=self.kink_circle_color, | |
) | |
else: | |
self.draw_dot( | |
cr, | |
diameter=self.kink_point_size, | |
color=self.kink_point_color, | |
) | |
cr.restore() | |
return scale | |
def draw_dot(self, cr, *, x=0, y=0, color=(0, 0, 0), diameter=10): | |
cr.save() | |
cr.set_line_width(diameter) | |
cr.set_line_cap(cairo.LINE_CAP_ROUND) | |
cr.move_to(x, y) | |
cr.line_to(x, y) | |
if len(color) == 3: | |
color = color + (1,) | |
cr.set_source_rgba(*color) | |
cr.stroke() | |
cr.restore() | |
def draw_circle( | |
self, cr, *, x=0, y=0, color=(0, 0, 0), diameter=10, stroke_width=1 | |
): | |
cr.save() | |
cr.set_line_width(stroke_width) | |
cr.set_line_cap(cairo.LINE_CAP_SQUARE) | |
cr.arc(x, y, diameter / 2, 0, 2 * math.pi) | |
if len(color) == 3: | |
color = color + (1,) | |
cr.set_source_rgba(*color) | |
cr.stroke() | |
cr.restore() | |
def draw_arrow(self, cr, *, x=0, y=0, color=(0, 0, 0)): | |
cr.save() | |
if len(color) == 3: | |
color = color + (1,) | |
cr.set_source_rgba(*color) | |
cr.translate(self.start_arrow_length + x, y) | |
cr.move_to(0, 0) | |
cr.line_to( | |
-self.start_arrow_length, | |
-self.start_arrow_length * 0.4, | |
) | |
cr.line_to( | |
-self.start_arrow_length, | |
self.start_arrow_length * 0.4, | |
) | |
cr.close_path() | |
cr.fill() | |
cr.restore() | |
def draw_text(self, text, *, x=0, y=0, color=(0, 0, 0), width=None, height=None): | |
if width is None: | |
width = self.width | |
if height is None: | |
height = self.height | |
text = text.splitlines() | |
cr = cairo.Context(self.surface) | |
cr.set_source_rgb(*color) | |
cr.set_font_size(self.font_size) | |
cr.select_font_face( | |
"@cairo:monospace", cairo.FONT_SLANT_NORMAL, cairo.FONT_WEIGHT_NORMAL | |
) | |
text_width = 0 | |
text_height = 0 | |
font_extents = cr.font_extents() | |
font_font_size = font_extents[2] | |
font_ascent = font_extents[0] | |
for line in text: | |
extents = cr.text_extents(line) | |
text_width = max(text_width, extents.x_advance) | |
text_height += font_font_size | |
if not text_width: | |
return | |
cr.translate(x, y) | |
scale = min(width / text_width, height / text_height) | |
# center | |
cr.translate( | |
(width - text_width * scale) / 2, (height - text_height * scale) / 2 | |
) | |
cr.scale(scale, scale) | |
cr.translate(0, font_ascent) | |
for line in text: | |
cr.move_to(0, 0) | |
cr.show_text(line) | |
cr.translate(0, font_font_size) | |
def draw_cupcake(self): | |
self.draw_label( | |
self.no_issues_label, | |
x=self.pad, | |
y=self.pad, | |
color=self.no_issues_label_color, | |
width=self.width - 2 * self.pad, | |
align=0.5, | |
bold=True, | |
font_size=self.title_font_size, | |
) | |
self.draw_text( | |
self.cupcake, | |
x=self.pad, | |
y=self.pad + self.font_size, | |
width=self.width - 2 * self.pad, | |
height=self.height - 2 * self.pad - self.font_size, | |
color=self.cupcake_color, | |
) | |
def draw_emoticon(self, emoticon, x=0, y=0): | |
self.draw_text( | |
emoticon, | |
x=x, | |
y=y, | |
color=self.emoticon_color, | |
width=self.panel_width, | |
height=self.panel_height, | |
) | |
class InterpolatablePostscriptLike(InterpolatablePlot): | |
def __exit__(self, type, value, traceback): | |
self.surface.finish() | |
def show_page(self): | |
super().show_page() | |
self.surface.show_page() | |
class InterpolatablePS(InterpolatablePostscriptLike): | |
def __enter__(self): | |
self.surface = cairo.PSSurface(self.out, self.width, self.height) | |
return self | |
class InterpolatablePDF(InterpolatablePostscriptLike): | |
def __enter__(self): | |
self.surface = cairo.PDFSurface(self.out, self.width, self.height) | |
self.surface.set_metadata( | |
cairo.PDF_METADATA_CREATOR, "fonttools varLib.interpolatable" | |
) | |
self.surface.set_metadata(cairo.PDF_METADATA_CREATE_DATE, "") | |
return self | |
class InterpolatableSVG(InterpolatablePlot): | |
def __enter__(self): | |
self.sink = BytesIO() | |
self.surface = cairo.SVGSurface(self.sink, self.width, self.height) | |
return self | |
def __exit__(self, type, value, traceback): | |
if self.surface is not None: | |
self.show_page() | |
def show_page(self): | |
super().show_page() | |
self.surface.finish() | |
self.out.append(self.sink.getvalue()) | |
self.sink = BytesIO() | |
self.surface = cairo.SVGSurface(self.sink, self.width, self.height) | |