kakoune/src/highlighters.cc

419 lines
14 KiB
C++
Raw Normal View History

#include "highlighters.hh"
#include "assert.hh"
#include "color_registry.hh"
2012-12-31 14:28:32 +01:00
#include "context.hh"
#include "option_types.hh"
#include "register_manager.hh"
#include "string.hh"
#include "utf8.hh"
#include "utf8_iterator.hh"
#include "window.hh"
#include <sstream>
#include <locale>
2011-09-30 21:16:23 +02:00
namespace Kakoune
{
using namespace std::placeholders;
typedef boost::regex_iterator<BufferIterator> RegexIterator;
template<typename T>
void highlight_range(DisplayBuffer& display_buffer,
BufferCoord begin, BufferCoord end,
bool skip_replaced, T func)
2011-09-30 21:16:23 +02:00
{
2012-08-10 14:20:15 +02:00
if (begin == end or end <= display_buffer.range().first
or begin >= display_buffer.range().second)
return;
for (auto& line : display_buffer.lines())
2011-09-30 21:16:23 +02:00
{
auto& range = line.range();
if (range.second <= begin or end < range.first)
2012-12-12 19:33:29 +01:00
continue;
for (auto atom_it = line.begin(); atom_it != line.end(); ++atom_it)
{
2013-07-24 14:55:57 +02:00
bool is_replaced = atom_it->type() == DisplayAtom::ReplacedBufferRange;
2013-07-24 14:55:57 +02:00
if (not atom_it->has_buffer_range() or
2012-12-12 19:33:29 +01:00
(skip_replaced and is_replaced))
continue;
2011-09-30 21:16:23 +02:00
2013-07-24 14:55:57 +02:00
if (end <= atom_it->begin() or begin >= atom_it->end())
2012-12-12 19:33:29 +01:00
continue;
2013-07-24 14:55:57 +02:00
if (not is_replaced and begin > atom_it->begin())
2012-12-12 19:33:29 +01:00
atom_it = ++line.split(atom_it, begin);
2013-07-24 14:55:57 +02:00
if (not is_replaced and end < atom_it->end())
2012-12-12 19:33:29 +01:00
{
atom_it = line.split(atom_it, end);
func(*atom_it);
++atom_it;
}
2012-12-12 19:33:29 +01:00
else
func(*atom_it);
}
2011-09-30 21:16:23 +02:00
}
}
typedef std::unordered_map<size_t, const ColorPair*> ColorSpec;
class RegexColorizer
{
public:
RegexColorizer(Regex regex, ColorSpec colors)
: m_regex(std::move(regex)), m_colors(std::move(colors))
{
}
void operator()(const Window& window, DisplayBuffer& display_buffer)
{
const Buffer& buffer = window.buffer();
update_cache_ifn(buffer, display_buffer.range());
for (auto& match : m_caches[&buffer].m_matches)
{
for (size_t n = 0; n < match.size(); ++n)
{
auto col_it = m_colors.find(n);
if (col_it == m_colors.end())
continue;
highlight_range(display_buffer, match[n].first, match[n].second, true,
[&](DisplayAtom& atom) { atom.colors = *col_it->second; });
}
}
}
private:
struct MatchesCache
{
BufferRange m_range;
size_t m_timestamp;
std::vector<std::vector<std::pair<BufferCoord, BufferCoord>>> m_matches;
};
std::unordered_map<const Buffer*, MatchesCache> m_caches;
Regex m_regex;
ColorSpec m_colors;
void update_cache_ifn(const Buffer& buffer, const BufferRange& range)
{
MatchesCache& cache = m_caches[&buffer];
if (buffer.timestamp() == cache.m_timestamp and
range.first >= cache.m_range.first and
range.second <= cache.m_range.second)
return;
cache.m_range.first = buffer.clamp({range.first.line - 10, 0});
cache.m_range.second = buffer.next(buffer.clamp({range.second.line + 10, INT_MAX}));
cache.m_timestamp = buffer.timestamp();
cache.m_matches.clear();
RegexIterator re_it{buffer.iterator_at(cache.m_range.first),
buffer.iterator_at(cache.m_range.second), m_regex};
RegexIterator re_end;
for (; re_it != re_end; ++re_it)
{
cache.m_matches.emplace_back();
auto& match = cache.m_matches.back();
for (auto& sub : *re_it)
match.emplace_back(sub.first.coord(), sub.second.coord());
}
}
};
2013-07-26 01:17:12 +02:00
HighlighterAndId colorize_regex_factory(HighlighterParameters params, const Window&)
{
if (params.size() < 2)
throw runtime_error("wrong parameter count");
try
{
static Regex color_spec_ex(R"((\d+):(\w+(,\w+)?))");
ColorSpec colors;
for (auto it = params.begin() + 1; it != params.end(); ++it)
{
boost::smatch res;
if (not boost::regex_match(it->begin(), it->end(), res, color_spec_ex))
throw runtime_error("wrong colorspec: '" + *it +
"' expected <capture>:<fgcolor>[,<bgcolor>]");
int capture = str_to_int(res[1].str());
const ColorPair*& color = colors[capture];
2013-05-13 14:23:07 +02:00
color = &get_color(res[2].str());
}
String id = "colre'" + params[0] + "'";
Regex ex(params[0].begin(), params[0].end(),
boost::regex::perl | boost::regex::optimize);
return HighlighterAndId(id, RegexColorizer(std::move(ex),
std::move(colors)));
}
catch (boost::regex_error& err)
{
throw runtime_error(String("regex error: ") + err.what());
}
}
template<typename RegexGetter>
class DynamicRegexHighlighter
2012-12-31 14:28:32 +01:00
{
public:
DynamicRegexHighlighter(const ColorSpec& colors, RegexGetter getter)
: m_regex_getter(getter), m_colors(colors), m_colorizer(Regex(), m_colors) {}
2012-12-31 14:28:32 +01:00
2013-05-23 13:39:00 +02:00
void operator()(const Window& window, DisplayBuffer& display_buffer)
2012-12-31 14:28:32 +01:00
{
Regex regex = m_regex_getter(window);
if (regex != m_last_regex)
2012-12-31 14:28:32 +01:00
{
m_last_regex = regex;
if (not m_last_regex.empty())
m_colorizer = RegexColorizer{m_last_regex, m_colors};
2012-12-31 14:28:32 +01:00
}
if (not m_last_regex.empty())
2013-05-23 13:39:00 +02:00
m_colorizer(window, display_buffer);
2012-12-31 14:28:32 +01:00
}
private:
Regex m_last_regex;
2013-01-04 18:39:13 +01:00
ColorSpec m_colors;
RegexColorizer m_colorizer;
RegexGetter m_regex_getter;
2012-12-31 14:28:32 +01:00
};
2013-07-26 01:17:12 +02:00
HighlighterAndId highlight_search_factory(HighlighterParameters params, const Window&)
2012-12-31 14:28:32 +01:00
{
if (params.size() != 1)
throw runtime_error("wrong parameter count");
try
{
2013-04-04 18:47:34 +02:00
ColorSpec colors { { 0, &get_color(params[0]) } };
auto get_regex = [](const Window&){
auto s = RegisterManager::instance()['/'].values(Context{});
return s.empty() ? Regex{} : Regex{s[0].begin(), s[0].end()};
};
return {"hlsearch", DynamicRegexHighlighter<decltype(get_regex)>{colors, get_regex}};
2012-12-31 14:28:32 +01:00
}
catch (boost::regex_error& err)
{
throw runtime_error(String("regex error: ") + err.what());
}
}
2012-12-31 14:28:32 +01:00
HighlighterAndId highlight_regex_option_factory(HighlighterParameters params, const Window&)
{
if (params.size() != 2)
throw runtime_error("wrong parameter count");
2013-04-04 18:47:34 +02:00
ColorSpec colors { { 0, &get_color(params[1]) } };
String option_name = params[0];
// verify option type now
GlobalOptions::instance()[option_name].get<Regex>();
auto get_regex = [option_name](const Window& window){ return window.options()[option_name].get<Regex>(); };
return {"hloption_" + option_name, DynamicRegexHighlighter<decltype(get_regex)>{colors, get_regex}};
}
2013-05-23 13:39:00 +02:00
void expand_tabulations(const Window& window, DisplayBuffer& display_buffer)
{
2013-05-23 13:39:00 +02:00
const int tabstop = window.options()["tabstop"].get<int>();
auto& buffer = window.buffer();
for (auto& line : display_buffer.lines())
{
for (auto atom_it = line.begin(); atom_it != line.end(); ++atom_it)
{
2013-07-24 14:55:57 +02:00
if (atom_it->type() != DisplayAtom::BufferRange)
continue;
2013-07-24 14:55:57 +02:00
auto begin = buffer.iterator_at(atom_it->begin());
auto end = buffer.iterator_at(atom_it->end());
for (BufferIterator it = begin; it != end; ++it)
{
if (*it == '\t')
{
if (it != begin)
atom_it = ++line.split(atom_it, it.coord());
if (it+1 != end)
atom_it = line.split(atom_it, (it+1).coord());
int column = 0;
for (auto line_it = buffer.iterator_at(it.coord().line);
line_it != it; ++line_it)
{
kak_assert(*line_it != '\n');
if (*line_it == '\t')
column += tabstop - (column % tabstop);
else
++column;
}
int count = tabstop - (column % tabstop);
String padding;
for (int i = 0; i < count; ++i)
padding += ' ';
2013-07-24 14:55:57 +02:00
atom_it->replace(padding);
break;
}
}
}
}
}
2013-05-23 13:39:00 +02:00
void show_line_numbers(const Window& window, DisplayBuffer& display_buffer)
{
2013-05-23 13:39:00 +02:00
LineCount last_line = window.buffer().line_count();
int digit_count = 0;
for (LineCount c = last_line; c > 0; c /= 10)
++digit_count;
2013-05-24 18:39:03 +02:00
char format[] = "%?d│";
format[1] = '0' + digit_count;
2013-04-04 18:47:34 +02:00
auto& colors = get_color("LineNumbers");
for (auto& line : display_buffer.lines())
{
char buffer[10];
snprintf(buffer, 10, format, (int)line.range().first.line + 1);
2013-07-24 14:55:57 +02:00
DisplayAtom atom{buffer};
atom.colors = colors;
line.insert(line.begin(), std::move(atom));
}
}
void highlight_selections(const Window& window, DisplayBuffer& display_buffer)
{
const bool only_cursor = window.is_editing() and window.options()["insert_hide_sel"].get<bool>();
const auto& buffer = window.buffer();
for (size_t i = 0; i < window.selections().size(); ++i)
{
auto& sel = window.selections()[i];
const bool forward = sel.first() <= sel.last();
BufferCoord begin = forward ? sel.first() : buffer.char_next(sel.last());
BufferCoord end = forward ? sel.last() : buffer.char_next(sel.first());
const bool primary = (i == window.main_selection_index());
if (not only_cursor)
{
2013-04-04 18:47:34 +02:00
ColorPair sel_colors = get_color(primary ? "PrimarySelection" : "SecondarySelection");
highlight_range(display_buffer, begin, end, false,
[&](DisplayAtom& atom) { atom.colors = sel_colors; });
}
2013-04-04 18:47:34 +02:00
ColorPair cur_colors = get_color(primary ? "PrimaryCursor" : "SecondaryCursor");
highlight_range(display_buffer, sel.last(), buffer.char_next(sel.last()), false,
[&](DisplayAtom& atom) { atom.colors = cur_colors; });
}
}
void expand_unprintable(const Window& window, DisplayBuffer& display_buffer)
{
auto& buffer = window.buffer();
for (auto& line : display_buffer.lines())
{
for (auto atom_it = line.begin(); atom_it != line.end(); ++atom_it)
{
2013-07-24 14:55:57 +02:00
if (atom_it->type() == DisplayAtom::BufferRange)
{
using Utf8It = utf8::utf8_iterator<BufferIterator, utf8::InvalidBytePolicy::Pass>;
2013-07-24 14:55:57 +02:00
for (Utf8It it = buffer.iterator_at(atom_it->begin()),
end = buffer.iterator_at(atom_it->end()); it != end; ++it)
{
Codepoint cp = *it;
if (cp != '\n' and iscntrl((int)cp))
{
std::ostringstream oss;
oss << "U+" << std::hex << cp;
String str = oss.str();
2013-07-24 14:55:57 +02:00
if (it.base().coord() != atom_it->begin())
atom_it = ++line.split(atom_it, it.base().coord());
2013-07-24 14:55:57 +02:00
if ((it+1).base().coord() != atom_it->end())
atom_it = line.split(atom_it, (it+1).base().coord());
2013-07-24 14:55:57 +02:00
atom_it->replace(str);
atom_it->colors = { Colors::Red, Colors::Black };
break;
}
}
}
}
}
}
2013-07-26 01:17:12 +02:00
HighlighterAndId flag_lines_factory(HighlighterParameters params, Window& window)
{
if (params.size() != 2)
throw runtime_error("wrong parameter count");
const String& option_name = params[1];
Color bg = str_to_color(params[0]);
// throw if wrong option type
GlobalOptions::instance()[option_name].get<std::vector<LineAndFlag>>();
return {"hlflags_" + params[1],
[=](const Window& window, DisplayBuffer& display_buffer)
{
auto& lines_opt = window.options()[option_name];
auto& lines = lines_opt.get<std::vector<LineAndFlag>>();
CharCount width = 0;
for (auto& l : lines)
width = std::max(width, std::get<2>(l).char_length());
const String empty{' ', width};
for (auto& line : display_buffer.lines())
{
int line_num = (int)line.range().first.line + 1;
auto it = find_if(lines,
[&](const LineAndFlag& l)
{ return std::get<0>(l) == line_num; });
String content = it != lines.end() ? std::get<2>(*it) : empty;
content += String(' ', width - content.char_length());
DisplayAtom atom{std::move(content)};
atom.colors = { it != lines.end() ? std::get<1>(*it) : Colors::Default , bg };
line.insert(line.begin(), std::move(atom));
}
}};
}
2013-05-23 13:39:00 +02:00
template<void (*highlighter_func)(const Window&, DisplayBuffer&)>
class SimpleHighlighterFactory
{
public:
SimpleHighlighterFactory(const String& id) : m_id(id) {}
2013-07-26 01:17:12 +02:00
HighlighterAndId operator()(HighlighterParameters params, const Window&) const
{
return HighlighterAndId(m_id, HighlighterFunc(highlighter_func));
}
private:
String m_id;
};
2013-07-26 01:17:12 +02:00
HighlighterAndId highlighter_group_factory(HighlighterParameters params, const Window&)
{
if (params.size() != 1)
throw runtime_error("wrong parameter count");
return HighlighterAndId(params[0], HighlighterGroup());
}
void register_highlighters()
{
HighlighterRegistry& registry = HighlighterRegistry::instance();
registry.register_func("number_lines", SimpleHighlighterFactory<show_line_numbers>("number_lines"));
registry.register_func("regex", colorize_regex_factory);
registry.register_func("regex_option", highlight_regex_option_factory);
2012-12-31 14:28:32 +01:00
registry.register_func("search", highlight_search_factory);
registry.register_func("group", highlighter_group_factory);
registry.register_func("flag_lines", flag_lines_factory);
}
2011-09-30 21:16:23 +02:00
}