Updated vimtex and snippets

LuaSnip/python/mpl.lua

@@ -0,0 +1,125 @@
+local helpers = require('personal.luasnip-helper-funcs')
+local get_visual = helpers.get_visual
+
+local line_begin = require("luasnip.extras.expand_conditions").line_begin
+
+return
+  {
+    -- NEW FIGURE, AXES
+    s({trig="fx", snippetType="autosnippet"},
+      fmta(
+        [[fig, ax = plt.subplots(<>)]],
+        {
+          d(1, get_visual),
+        }
+      ),
+      {condition = line_begin}
+    ),
+    -- plt.
+    s({trig=";p", snippetType="autosnippet"},
+      {t("plt.")}
+    ),
+    -- plt.plot()
+    s({trig="pll", snippetType="autosnippet"},
+      fmta(
+        [[plt.plot(<>)]],
+        {
+          d(1, get_visual),
+        }
+      ),
+      {condition = line_begin}
+    ),
+    -- plt.show()
+    s({trig="pss", snippetType="autosnippet"},
+      fmta(
+        [[plt.show()]],
+        { }
+      ),
+      {condition = line_begin}
+    ),
+    -- AXIS PLOT
+    s({trig="xp", snippetType="autosnippet"},
+      fmta(
+        [[ax.plot(<>)]],
+        {
+          d(1, get_visual),
+        }
+      ),
+      {condition = line_begin}
+    ),
+    -- AXIS SET_XLABEL
+    s({trig="xxl", snippetType="autosnippet"},
+      fmta(
+        [[ax.set_xlabel(<>)]],
+        {
+          d(1, get_visual),
+        }
+      ),
+      {condition = line_begin}
+    ),
+    -- AXIS SET_YLABEL
+    s({trig="xyl", snippetType="autosnippet"},
+      fmta(
+        [[ax.set_ylabel(<>)]],
+        {
+          d(1, get_visual),
+        }
+      ),
+      {condition = line_begin}
+    ),
+    -- AXIS SET_TITLE
+    s({trig="xt", snippetType="autosnippet"},
+      fmta(
+        [[ax.set_title(<>)]],
+        {
+          d(1, get_visual),
+        }
+      ),
+      {condition = line_begin}
+    ),
+    -- TIGHT LAYOUT
+    s({trig="ttl", snippetType="autosnippet"},
+      {t("plt.tight_layout()")},
+      {condition = line_begin}
+    ),
+    -- STEM PLOT
+    s({trig="stem", snippetType="autosnippet"},
+      fmta(
+        [[
+          (markers, stemlines, baseline) = ax.stem(<>)
+          plt.setp(markers, marker='o', markerfacecolor=<>, markeredgecolor="none", markersize=6)
+          plt.setp(baseline, color=<>, linestyle="-")
+          plt.setp(stemlines, linestyle="--", color=<>, linewidth=2)
+        ]],
+        {
+          i(1),
+          i(2),
+          i(3),
+          i(4)
+        }
+      ),
+      {condition = line_begin}
+    ),
+    -- REMOVE SPINE FUNCTION
+    s({trig="spines"},
+      fmta(
+        [[
+        def remove_spines(ax):
+            """
+            Removes top and right spines from the inputted Matplotlib axis. This is for
+            aesthetic purposes only and has no practical function.
+            """
+            ax.spines['top'].set_visible(False)
+            ax.spines['right'].set_visible(False)
+            ax.get_xaxis().tick_bottom()
+            ax.get_yaxis().tick_left()
+        ]],
+        {
+        }
+      ),
+      {condition = line_begin}
+    ),
+  }
+
+
+