Initial commit.

Author: anntzer

.gitignore

@@ -0,0 +1,16 @@
+**/.ipynb_checkpoints/
+*.egg-info/
+.cache/
+build/
+dist/
+doc/_build/
+doc/_static/
+doc/_templates/
+htmlcov/
+oprofile_data/
+.*.swp
+*.o
+*.pyc
+*.so
+.coverage
+.gdb_history

mpl_cairo/__init__.py

@@ -0,0 +1,66 @@
+__all__ = ["GraphicsContextRendererCairo", "install", "uninstall"]
+
+from matplotlib.backend_bases import GraphicsContextBase, RendererBase
+from matplotlib.backends import backend_cairo
+from matplotlib.font_manager import ttfFontProperty
+from matplotlib.mathtext import MathtextBackendCairo, MathTextParser
+from . import _mpl_cairo
+
+
+class MathtextBackendCairo2(MathtextBackendCairo):
+    def render_glyph(self, ox, oy, info):
+        self.glyphs.append(
+            # Convert to ttfFontProperty here.
+            (ttfFontProperty(info.font),
+             info.fontsize,
+             chr(info.num),
+             ox,
+             oy - info.offset - self.height))
+
+
+MathTextParser._backend_mapping["cairo2"] = MathtextBackendCairo2
+
+
+class GraphicsContextRendererCairo(
+        _mpl_cairo.GraphicsContextRendererCairo,
+        GraphicsContextBase,
+        RendererBase):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.mathtext_parser = MathTextParser("cairo2")
+
+
+# Keep the two classes separate, so that we can figure out who inherits what.
+
+
+class GraphicsContextCairo(GraphicsContextRendererCairo):
+    pass
+
+
+class RendererCairo(GraphicsContextRendererCairo):
+    pass
+
+
+_class_pairs = [
+    (backend_cairo.GraphicsContextCairo, GraphicsContextRendererCairo),
+    (backend_cairo.RendererCairo, RendererCairo)]
+
+
+def _swap(class_pairs):
+    for old, new in class_pairs:
+        for cls in old.__subclasses__():
+            idx = cls.__bases__.index(old)
+            cls.__bases__ = cls.__bases__[:idx] + (new,) + cls.__bases__[idx + 1:]
+        setattr(backend_cairo, new.__name__, new)
+
+
+def install():
+    """Switch to new cairo backend implementation.
+    """
+    _swap(_class_pairs)
+
+
+def uninstall():
+    """Switch back to upstream cairo backend implementation.
+    """
+    _swap([(new, old) for old, new in _class_pairs])

setup.py

@@ -0,0 +1,65 @@
+from setuptools import setup, Extension
+from setuptools.command.build_ext import build_ext
+import sys
+import setuptools
+
+
+__version__ = "0.0"
+
+
+class get_pybind_include(object):
+    """Helper class to determine the pybind11 include path.
+
+    The purpose of this class is to postpone importing pybind11 until it is
+    actually installed, so that the ``get_include()`` method can be invoked.
+    """
+
+    def __init__(self, user=False):
+        self.user = user
+
+    def __str__(self):
+        import pybind11
+        return pybind11.get_include(self.user)
+
+
+ext_modules = [
+    Extension(
+        "mpl_cairo._mpl_cairo",
+        ["src/_mpl_cairo.cpp"],
+        language="c++",
+        include_dirs=[
+            get_pybind_include(), get_pybind_include(user=True)
+        ],
+        libraries=["cairo"],
+    ),
+]
+
+
+class BuildExt(build_ext):
+    """A custom build extension for adding compiler-specific options."""
+    c_opts = {
+        "msvc": ["/EHsc"],
+        "unix": ["-std=c++17", "-fvisibility=hidden"],
+    }
+
+    if sys.platform == "darwin":
+        c_opts["unix"] += ["-stdlib=libc++", "-mmacosx-version-min=10.7"]
+
+    def build_extensions(self):
+        ct = self.compiler.compiler_type
+        opts = self.c_opts.get(ct, [])
+        for ext in self.extensions:
+            ext.extra_compile_args = opts
+        build_ext.build_extensions(self)
+
+setup(
+    name="mpl_cairo",
+    version=__version__,
+    author="Antony Lee",
+    description="A cairo backend for matplotlib.",
+    long_description="",
+    ext_modules=ext_modules,
+    install_requires=["pybind11"],
+    cmdclass={"build_ext": BuildExt},
+    zip_safe=False,
+)