pdf: Support setting URLs on Text objects.

QuLogic · QuLogic · commit b47fa78773d4 · 2020-10-02T23:31:32.000-04:00
diff --git a/doc/users/next_whats_new/pdf_urls.rst b/doc/users/next_whats_new/pdf_urls.rst
@@ -0,0 +1,5 @@
+PDF supports URLs on ``Text`` artists
+-------------------------------------
+
+URLs on `.text.Text` artists (i.e., from `.Artist.set_url`) will now be saved
+in PDF files.
diff --git a/lib/matplotlib/backends/backend_pdf.py b/lib/matplotlib/backends/backend_pdf.py
@@ -2106,6 +2106,19 @@ def draw_mathtext(self, gc, x, y, s, prop, angle):
         width, height, descent, glyphs, rects = \
             self._text2path.mathtext_parser.parse(s, 72, prop)
 
+        if gc.get_url() is not None:
+            link_annotation = {
+                'Type': Name('Annot'),
+                'Subtype': Name('Link'),
+                'Rect': (x, y, x + width, y + height),
+                'Border': [0, 0, 0],
+                'A': {
+                    'S': Name('URI'),
+                    'URI': gc.get_url(),
+                },
+            }
+            self.file._annotations[-1][1].append(link_annotation)
+
         global_fonttype = mpl.rcParams['pdf.fonttype']
 
         # Set up a global transformation matrix for the whole math expression
@@ -2162,6 +2175,19 @@ def draw_tex(self, gc, x, y, s, prop, angle, ismath='TeX!', mtext=None):
         with dviread.Dvi(dvifile, 72) as dvi:
             page, = dvi
 
+        if gc.get_url() is not None:
+            link_annotation = {
+                'Type': Name('Annot'),
+                'Subtype': Name('Link'),
+                'Rect': (x, y, x + page.width, y + page.height),
+                'Border': [0, 0, 0],
+                'A': {
+                    'S': Name('URI'),
+                    'URI': gc.get_url(),
+                },
+            }
+            self.file._annotations[-1][1].append(link_annotation)
+
         # Gather font information and do some setup for combining
         # characters into strings. The variable seq will contain a
         # sequence of font and text entries. A font entry is a list
@@ -2261,6 +2287,21 @@ def draw_text(self, gc, x, y, s, prop, angle, ismath=False, mtext=None):
             if is_opentype_cff_font(font.fname):
                 fonttype = 42
 
+        if gc.get_url() is not None:
+            font.set_text(s)
+            width, height = font.get_width_height()
+            link_annotation = {
+                'Type': Name('Annot'),
+                'Subtype': Name('Link'),
+                'Rect': (x, y, x + width / 64, y + height / 64),
+                'Border': [0, 0, 0],
+                'A': {
+                    'S': Name('URI'),
+                    'URI': gc.get_url(),
+                },
+            }
+            self.file._annotations[-1][1].append(link_annotation)
+
         # If fonttype != 3 or there are no multibyte characters, emit the whole
         # string at once.
         if fonttype != 3 or all(ord(char) <= 255 for char in s):
diff --git a/lib/matplotlib/tests/test_backend_pdf.py b/lib/matplotlib/tests/test_backend_pdf.py
@@ -1,4 +1,5 @@
 import datetime
+import decimal
 import io
 import os
 from pathlib import Path
@@ -212,6 +213,53 @@ def test_multipage_metadata(monkeypatch):
     }
 
 
+def test_text_urls():
+    pikepdf = pytest.importorskip('pikepdf')
+
+    test_url = 'https://test_text_urls.matplotlib.org/'
+
+    fig = plt.figure(figsize=(2, 1))
+    fig.text(0.1, 0.1, 'test plain 123', url=f'{test_url}plain')
+    fig.text(0.1, 0.4, 'test mathtext $123$', url=f'{test_url}mathtext')
+
+    with io.BytesIO() as fd:
+        fig.savefig(fd, format='pdf')
+
+        with pikepdf.Pdf.open(fd) as pdf:
+            annots = pdf.pages[0].Annots
+
+    for y, fragment in [('0.1', 'plain'), ('0.4', 'mathtext')]:
+        annot = next(
+            (a for a in annots if a.A.URI == f'{test_url}{fragment}'),
+            None)
+        assert annot is not None
+        # Positions in points (72 per inch.)
+        assert annot.Rect[1] == decimal.Decimal(y) * 72
+
+
+@needs_usetex
+def test_text_urls_tex():
+    pikepdf = pytest.importorskip('pikepdf')
+
+    test_url = 'https://test_text_urls.matplotlib.org/'
+
+    fig = plt.figure(figsize=(2, 1))
+    fig.text(0.1, 0.7, 'test tex $123$', usetex=True, url=f'{test_url}tex')
+
+    with io.BytesIO() as fd:
+        fig.savefig(fd, format='pdf')
+
+        with pikepdf.Pdf.open(fd) as pdf:
+            annots = pdf.pages[0].Annots
+
+    annot = next(
+        (a for a in annots if a.A.URI == f'{test_url}tex'),
+        None)
+    assert annot is not None
+    # Positions in points (72 per inch.)
+    assert annot.Rect[1] == decimal.Decimal('0.7') * 72
+
+
 def test_pdfpages_fspath():
     with PdfPages(Path(os.devnull)) as pdf:
         pdf.savefig(plt.figure())
