Add ConTeXt support

Author: flying-sheep

README.md

@@ -18,10 +18,10 @@ figures like
 
 ![](https://nschloe.github.io/tikzplotlib/example.png)
 
-for native inclusion into LaTeX documents.
+for native inclusion into LaTeX or ConTeXt documents.
 
 The output of tikzplotlib is in
-[PGFPlots](http://pgfplots.sourceforge.net/pgfplots.pdf), a LaTeX library that sits on
+[PGFPlots](http://pgfplots.sourceforge.net/pgfplots.pdf), a TeX library that sits on
 top of [PGF/TikZ](https://en.wikipedia.org/wiki/PGF/TikZ) and describes graphs in terms
 of axes, data etc. Consequently, the output of tikzplotlib
 
@@ -95,7 +95,7 @@ table {%
 ```
 (Use `get_tikz_code()` instead of `save()` if you want the code as a string.)
 
-Tweaking the plot is straightforward and can be done as part of your LaTeX work flow.
+Tweaking the plot is straightforward and can be done as part of your TeX work flow.
 [The fantastic PGFPlots manual](http://pgfplots.sourceforge.net/pgfplots.pdf) contains
 great examples of how to make your plot look even better.
 
@@ -120,23 +120,23 @@ to install/update.
     ```python
     import tikzplotlib
     tikzplotlib.save("mytikz.tex")
+    # or
+    tikzplotlib.save("mytikz.tex", flavor="context")
     ```
    to store the TikZ file as `mytikz.tex`.
 
-3. Add the contents of `mytikz.tex` into your LaTeX source code. A convenient way of
+3. Add the contents of `mytikz.tex` into your TeX source code. A convenient way of
    doing so is via
     ```latex
     \input{/path/to/mytikz.tex}
     ```
    Also make sure that the packages for PGFPlots and proper Unicode support and are
-   included in the header of your document:
-    ```latex
-    \usepackage[utf8]{inputenc}
-    \usepackage{fontspec}  % optional
-    \usepackage{pgfplots}
-    \pgfplotsset{compat=newest}
-    \usepgfplotslibrary{groupplots}
-    \usepgfplotslibrary{dateplot}
+   included in the header of your document. Get the code via:
+    ```python
+    import tikzplotlib
+    tikzplotlib.Flavors.latex.preamble()
+    # or
+    tikzplotlib.Flavors.context.preamble()
     ```
 
 4. Optional: clean up the figure before exporting to tikz using the `clean_figure` command.

test/helpers.py

@@ -31,11 +31,14 @@ def _unidiff_output(expected, actual):
 
 
 def assert_equality(
-    plot, filename, assert_compilation=True, **extra_get_tikz_code_args
+    plot, filename, assert_compilation=True, flavor="latex", **extra_get_tikz_code_args
 ):
     plot()
     code = tikzplotlib.get_tikz_code(
-        include_disclaimer=False, float_format=".8g", **extra_get_tikz_code_args
+        include_disclaimer=False,
+        float_format=".8g",
+        flavor=flavor,
+        **extra_get_tikz_code_args
     )
     plt.close()
 
@@ -47,13 +50,16 @@ def assert_equality(
     if assert_compilation:
         plot()
         code = tikzplotlib.get_tikz_code(
-            include_disclaimer=False, standalone=True, **extra_get_tikz_code_args
+            include_disclaimer=False,
+            standalone=True,
+            flavor=flavor,
+            **extra_get_tikz_code_args
         )
         plt.close()
-        assert _compile(code) is not None, code
+        assert _compile(code, flavor) is not None, code
 
 
-def _compile(code):
+def _compile(code, flavor):
     _, tmp_base = tempfile.mkstemp()
 
     tex_file = tmp_base + ".tex"
@@ -64,11 +70,12 @@ def _compile(code):
     os.chdir(os.path.dirname(tex_file))
 
     # compile the output to pdf
+    cmds = dict(
+        latex=["pdflatex", "--interaction=nonstopmode"],
+        context=["context", "--nonstopmode"],
+    )
     try:
-        subprocess.check_output(
-            ["pdflatex", "--interaction=nonstopmode", tex_file],
-            stderr=subprocess.STDOUT,
-        )
+        subprocess.check_output(cmds[flavor] + [tex_file], stderr=subprocess.STDOUT)
     except subprocess.CalledProcessError as e:
         print("pdflatex output:")
         print("=" * 70)
@@ -81,15 +88,15 @@ def _compile(code):
     return output_pdf
 
 
-def compare_mpl_latex(plot):
+def compare_mpl_tex(plot, flavor="latex"):
     plot()
     code = tikzplotlib.get_tikz_code(standalone=True)
     directory = os.getcwd()
     filename = "test-0.png"
     plt.savefig(filename)
     plt.close()
 
-    pdf_file = _compile(code)
+    pdf_file = _compile(code, flavor)
     pdf_dirname = os.path.dirname(pdf_file)
 
     # Convert PDF to PNG.

test/test_annotate.py

@@ -44,5 +44,5 @@ def test():
 if __name__ == "__main__":
     import helpers
 
-    helpers.compare_mpl_latex(plot)
+    helpers.compare_mpl_tex(plot)
     # helpers.print_tree(plot())

test/test_arrows.py

@@ -70,5 +70,5 @@ def to_texstring(s):
 if __name__ == "__main__":
     import helpers
 
-    helpers.compare_mpl_latex(plot)
+    helpers.compare_mpl_tex(plot)
     # helpers.print_tree(plot())

test/test_axvline.py

@@ -22,5 +22,5 @@ def test():
 if __name__ == "__main__":
     import helpers
 
-    helpers.compare_mpl_latex(plot)
+    helpers.compare_mpl_tex(plot)
     # helpers.print_tree(plot())

test/test_barchart_legend.py

@@ -40,5 +40,5 @@ def test():
 if __name__ == "__main__":
     import helpers
 
-    helpers.compare_mpl_latex(plot)
+    helpers.compare_mpl_tex(plot)
     # helpers.print_tree(plot())

test/test_basic_sin.py

@@ -29,5 +29,5 @@ def test():
 if __name__ == "__main__":
     import helpers
 
-    helpers.compare_mpl_latex(plot)
+    helpers.compare_mpl_tex(plot)
     # helpers.print_tree(plot())

test/test_colorbars.py

@@ -81,5 +81,5 @@ def test():
 if __name__ == "__main__":
     import helpers
 
-    helpers.compare_mpl_latex(plot)
+    helpers.compare_mpl_tex(plot)
     # helpers.print_tree(plot())

test/test_context.py

@@ -0,0 +1,20 @@
+from helpers import assert_equality
+
+
+def plot():
+    import numpy as np
+    import matplotlib.pyplot as plt
+    import matplotlib.cm as cm
+
+    x, y = np.meshgrid(np.linspace(0, 1), np.linspace(0, 1))
+    z = x ** 2 - y ** 2
+
+    fig = plt.figure()
+    plt.pcolormesh(x, y, z, cmap=cm.viridis)
+
+    return fig
+
+
+def test():
+    assert_equality(plot, __file__[:-3] + "_reference.tex", flavor="context")
+    return

test/test_context_reference.tex

@@ -0,0 +1,16 @@
+\starttikzpicture
+
+\startaxis[
+tick align=outside,
+tick pos=left,
+x grid style={white!69.019608!black},
+xmin=0, xmax=1,
+xtick style={color=black},
+y grid style={white!69.019608!black},
+ymin=0, ymax=1,
+ytick style={color=black}
+]
+\addplot graphics [includegraphics cmd=\pgfimage,xmin=0, xmax=1, ymin=0, ymax=1] {tmp-000.png};
+\stopaxis
+
+\stoptikzpicture