Add mpl template function

Author: joelostblom

Diff for: packages/python/plotly/templategen/definitions.py

@@ -2,6 +2,7 @@
 from templategen.utils import initialize_template
 from .utils.colors import colors
 import colorcet as cc
+import plotly.express as px
 
 # dict of template builder functions
 # This way we can loop over definitions in __init__.py
@@ -73,6 +74,96 @@ def ggplot2():
 builders["ggplot2"] = ggplot2
 
 
+def mpl():
+    # Set colorbar_common
+    colorbar_common = dict(
+        outlinewidth=10,
+        tickcolor=colors["gray14"],
+        ticks="outside",
+        tickwidth=2,
+        ticklen=8,
+    )
+
+    # Common axis common properties
+    axis_common = dict(
+        showgrid=False,
+        gridcolor=plotly_clrs["Rhino Light 2"],
+        linecolor=colors["gray14"],
+        ticks="outside",
+        showline=True,
+    )
+
+    # semi-transparent black and no outline
+    # annotation_clr = "rgb(67,103,167)"
+    # shape_defaults = dict(fillcolor=annotation_clr, line={"width": 0}, opacity=0.5)
+    # Near black line color, no fill
+    annotation_clr = plotly_clrs["Rhino Core"]
+    shape_defaults = dict(line_color=annotation_clr)
+
+    # Remove arrow head and make line thinner
+    # annotation_defaults = {"arrowcolor": annotation_clr}
+
+    # Remove arrow head and make line thinner
+    annotation_defaults = {
+        "arrowcolor": annotation_clr,
+        "arrowhead": 0,
+        "arrowwidth": 1,
+    }
+
+    template = initialize_template(
+        paper_clr="white",
+        font_clr=colors["gray14"],
+        panel_background_clr="white",
+        panel_grid_clr="white",
+        axis_ticks_clr=colors["gray14"],
+        zerolinecolor_clr=colors["gray14"],
+        table_cell_clr="white",
+        table_header_clr="white",
+        table_line_clr="white",
+        colorway=px.colors.qualitative.D3,
+        colorbar_common=colorbar_common,
+        colorscale=px.colors.sequential.Viridis,
+        colorscale_diverging=px.colors.diverging.RdBu,
+        axis_common=axis_common,
+        annotation_defaults=annotation_defaults,
+        shape_defaults=shape_defaults,
+    )
+
+    # Left align title
+    template.layout.title.x = 0.05
+
+    # Increase grid width for 3d plots
+    opts = dict(gridwidth=2, gridcolor=plotly_clrs["Rhino Light 1"], zeroline=False)
+    template.layout.scene.xaxis.update(opts)
+    template.layout.scene.yaxis.update(opts)
+    template.layout.scene.zaxis.update(opts)
+
+    # Darken ternary
+    opts = dict(
+        linecolor=plotly_clrs["Rhino Medium 1"], gridcolor=plotly_clrs["Rhino Light 1"]
+    )
+    template.layout.ternary.aaxis.update(opts)
+    template.layout.ternary.baxis.update(opts)
+    template.layout.ternary.caxis.update(opts)
+
+    # Remove lines through the origin
+    template.layout.xaxis.update(zeroline=False)
+    template.layout.yaxis.update(zeroline=False)
+
+    # Separate histogram bins wit ha white line
+    opts = {"marker": {"line": {"width": 1, "color": "white"}}}
+    template.data.histogram = [opts]
+
+    # Mapbox light style
+    template.layout.mapbox.style = "light"
+
+    # Set table header font color to white
+    return template
+
+
+builders["mpl"] = mpl
+
+
 def seaborn():
     # Define colors
     # -------------