[2906157] Let sphinx reference matplotlib-created plots

mdboom · mdboom · commit 39933e4c3754 · 2009-12-01T18:58:49.000Z
svn path=/trunk/matplotlib/; revision=7997
diff --git a/lib/matplotlib/sphinxext/plot_directive.py b/lib/matplotlib/sphinxext/plot_directive.py
@@ -308,8 +308,8 @@ def _plot_directive(plot_path, basedir, function_name, plot_code, caption,
     # treated as relative to the root of the documentation tree.  We
     # need to support both methods here.
     tmpdir = os.path.join('build', outdir)
+    tmpdir = os.path.abspath(tmpdir)
     if sphinx_version < (0, 6):
-        tmpdir = os.path.abspath(tmpdir)
         prefix = ''
     else:
         prefix = '/'
@@ -421,6 +421,36 @@ def plot_directive(name, arguments, options, content, lineno,
         return _plot_directive(plot_path, 'inline', None, plot_code, '', options,
                                state_machine)
 
+def mark_plot_labels(app, document):
+    """
+    To make plots referenceable, we need to move the reference from
+    the "htmlonly" (or "latexonly") node to the actual figure node
+    itself.
+    """
+    for name, explicit in document.nametypes.iteritems():
+        if not explicit:
+            continue
+        labelid = document.nameids[name]
+        if labelid is None:
+            continue
+        node = document.ids[labelid]
+        if node.tagname in ('html_only', 'latex_only'):
+            for n in node:
+                if n.tagname == 'figure':
+                    sectname = name
+                    for c in n:
+                        if c.tagname == 'caption':
+                            sectname = c.astext()
+                            break
+
+                    node['ids'].remove(labelid)
+                    node['names'].remove(name)
+                    n['ids'].append(labelid)
+                    n['names'].append(name)
+                    document.settings.env.labels[name] = \
+                        document.settings.env.docname, labelid, sectname
+                    break
+
 def setup(app):
     setup.app = app
     setup.config = app.config
@@ -441,3 +471,4 @@ def setup(app):
         [('png', 80), ('hires.png', 200), ('pdf', 50)],
         True)
 
+    app.connect('doctree-read', mark_plot_labels)
