140 lines
2.6 KiB
Lua
140 lines
2.6 KiB
Lua
local ls = require("luasnip")
|
|
local s = ls.snippet
|
|
local sn = ls.snippet_node
|
|
local t = ls.text_node
|
|
local i = ls.insert_node
|
|
local f = ls.function_node
|
|
local d = ls.dynamic_node
|
|
local fmta = require("luasnip.extras.fmt").fmta
|
|
local rep = require("luasnip.extras").rep
|
|
local line_begin = require("luasnip.extras.expand_conditions").line_begin
|
|
local tex = require("util.latex")
|
|
local get_visual = function(args, parent)
|
|
if #parent.snippet.env.SELECT_RAW > 0 then
|
|
return sn(nil, t(parent.snippet.env.SELECT_RAW))
|
|
else -- If SELECT_RAW is empty, return a blank insert node
|
|
return sn(nil, i(1))
|
|
end
|
|
end
|
|
|
|
return {
|
|
s(
|
|
{ trig = "bal", snippetType = "autosnippet" },
|
|
fmta(
|
|
[[
|
|
\begin{algorithm}
|
|
\caption{<>}
|
|
\label{algo:<>}
|
|
\begin{algorithmic}[1]
|
|
\Require <>
|
|
\end{algorithmic}
|
|
|
|
\end{algorithm}
|
|
]],
|
|
{
|
|
i(1),
|
|
rep(1),
|
|
i(0),
|
|
}
|
|
),
|
|
{ condition = tex.in_text * line_begin }
|
|
),
|
|
s({ trig = "ss", snippetType = "autosnippet" }, {
|
|
t("\\State "),
|
|
}, { condition = tex.in_algo * line_begin }),
|
|
s({ trig = "sx", snippetType = "autosnippet" }, {
|
|
t("\\Statex "),
|
|
}, { condition = tex.in_algo * line_begin }),
|
|
s({ trig = "return", snippetType = "autosnippet" }, {
|
|
t("\\Return"),
|
|
}, { condition = tex.in_algo }),
|
|
s(
|
|
{ trig = "while", snippetType = "autosnippet" },
|
|
fmta(
|
|
[[
|
|
\While{<>}
|
|
<>
|
|
\EndWhile
|
|
]],
|
|
{
|
|
i(1),
|
|
i(0),
|
|
}
|
|
),
|
|
{ condition = tex.in_algo * line_begin }
|
|
),
|
|
s(
|
|
{ trig = "for", snippetType = "autosnippet" },
|
|
fmta(
|
|
[[
|
|
\For{<>}
|
|
<>
|
|
\EndFor
|
|
]],
|
|
{
|
|
i(1),
|
|
i(0),
|
|
}
|
|
),
|
|
{ condition = tex.in_algo * line_begin }
|
|
),
|
|
s(
|
|
{ trig = "if", snippetType = "autosnippet" },
|
|
fmta(
|
|
[[
|
|
\If{<>}
|
|
<>
|
|
\EndIf
|
|
]],
|
|
{
|
|
i(1),
|
|
i(0),
|
|
}
|
|
),
|
|
{ condition = tex.in_algo * line_begin }
|
|
),
|
|
s(
|
|
{ trig = "elif", snippetType = "autosnippet" },
|
|
fmta(
|
|
[[
|
|
\ElsIf{<>}
|
|
<>
|
|
]],
|
|
{
|
|
i(1),
|
|
i(0),
|
|
}
|
|
),
|
|
{ condition = tex.in_algo * line_begin }
|
|
),
|
|
s(
|
|
{ trig = "else", snippetType = "autosnippet" },
|
|
fmta(
|
|
[[
|
|
\Else
|
|
<>
|
|
]],
|
|
{
|
|
i(0),
|
|
}
|
|
),
|
|
{ condition = tex.in_algo * line_begin }
|
|
),
|
|
s(
|
|
{ trig = "fn", snippetType = "autosnippet" },
|
|
fmta(
|
|
[[
|
|
\Function{<>}{<>}
|
|
<>
|
|
\EndFunction
|
|
]],
|
|
{
|
|
i(1),
|
|
i(2),
|
|
i(0),
|
|
}
|
|
),
|
|
{ condition = tex.in_algo * line_begin }
|
|
),
|
|
}
|