-
Notifications
You must be signed in to change notification settings - Fork 366
/
change_environment.py
170 lines (141 loc) · 5.12 KB
/
change_environment.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
import sublime
import sublime_plugin
import re
from .deprecated_command import deprecate
__all__ = [
"LatextoolsChangeEnvironmentCommand",
"LatextoolsToggleEnvironmentStarCommand",
]
class LatextoolsChangeEnvironmentCommand(sublime_plugin.TextCommand):
def run(self, edit):
view = self.view
new_regions = _find_env_regions(view)
if not new_regions:
return
view.sel().clear()
for r in new_regions:
view.sel().add(r)
class LatextoolsToggleEnvironmentStarCommand(sublime_plugin.TextCommand):
def run(self, edit):
view = self.view
new_regions = _find_env_regions(view)
if not new_regions:
return
# replace '*' with '' or vice versa for each region
for r in reversed(new_regions):
if view.substr(r).endswith('*'):
view.replace(edit, r, view.substr(r)[:-1])
else:
view.replace(edit, r, view.substr(r) + "*")
def _find_env_regions(view):
"""returns the regions corresponding to nearest matching environments"""
begin_re = r"\\begin(?:\[[^\]]*\])?\{([^\}]*)\}"
end_re = r"\\end\{([^\}]*)\}"
begins = view.find_all(begin_re, sublime.IGNORECASE)
ends = view.find_all(end_re, sublime.IGNORECASE)
# compile the begin_re (findall does not work if its compiled)
begin_re = re.compile(begin_re)
comment_line_re = re.compile(r"\s*%.*")
def is_comment(reg):
line_str = view.substr(view.line(reg))
return comment_line_re.match(line_str) is not None
begins = [b for b in begins if not is_comment(b)]
ends = [e for e in ends if not is_comment(e)]
def extract_begin_region(region):
"""creates a sublime.Region: \\begin{|text|}"""
s = view.substr(region)
boffset = len("\\begin{")
m = begin_re.search(s)
if m:
boffset = m.regs[1][0]
return sublime.Region(region.begin() + boffset, region.end() - 1)
def extract_end_region(region):
"""creates a sublime.Region: \\end{|text|}"""
boffset = len("\\end{")
return sublime.Region(region.begin() + boffset, region.end() - 1)
new_regions = []
one_sel = len(view.sel()) == 1
for sel in view.sel():
# partition the open and closed environments
begin_before, begin_after =\
_partition(begins, lambda b: b.begin() <= sel.begin())
end_before, end_after =\
_partition(ends, lambda e: e.end() < sel.begin())
# get the nearest open environments
try:
begin = _get_closest_begin(begin_before, end_before)
end = _get_closest_end(end_after, begin_after)
except NoEnvError as e:
if one_sel:
sublime.status_message(e.args[0])
return []
else:
continue
# extract the regions for the environments
begin_region = extract_begin_region(begin)
end_region = extract_end_region(end)
# validity check: matching env name
if view.substr(begin_region) == view.substr(end_region):
new_regions.append(begin_region)
new_regions.append(end_region)
elif one_sel:
sublime.status_message(
"The environment begin and end does not match:"
"'{0}' and '{1}'"
.format(view.substr(begin_region), view.substr(end_region))
)
if not new_regions:
sublime.status_message("Environment detection failed")
return new_regions
def _partition(env_list, is_before):
"""partition the list in the list items before and after the sel"""
before, after = [], []
iterator = iter(env_list)
while True:
try:
item = next(iterator)
except Exception:
break
if is_before(item):
before.append(item)
else:
after.append(item)
after.extend(iterator)
break
return before, after
class NoEnvError(Exception):
pass
def _get_closest_begin(begin_before, end_before):
"""returns the closest \\begin, that is open"""
end_iter = reversed(end_before)
begin_iter = reversed(begin_before)
while True:
try:
b = next(begin_iter)
except Exception:
raise NoEnvError("No open environment detected")
try:
e = next(end_iter)
except Exception:
break
if not b.begin() < e.begin():
break
return b
def _get_closest_end(end_after, begin_after):
"""returns the closest \\end, that is open"""
end_iter = iter(end_after)
begin_iter = iter(begin_after)
while True:
try:
e = next(end_iter)
except Exception:
raise NoEnvError("No closing environment detected")
try:
b = next(begin_iter)
except Exception:
break
if not e.begin() > b.begin():
break
return e
deprecate(globals(), 'LatexChangeEnvironmentCommand', LatextoolsChangeEnvironmentCommand)
deprecate(globals(), 'LatexToggleEnvironmentStarCommand', LatextoolsToggleEnvironmentStarCommand)