AIR/cpp-peglib
1
0
Fork 0
mirror of https://github.com/yhirose/cpp-peglib.git synced 2024-12-22 20:05:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changed ast_to_s to be customizable.

Browse Source
This commit is contained in:
yhirose 2017-08-17 01:39:30 -04:00
parent 463ed17967
commit dcdfd9fdb7
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
peglib.h
View File

@ -2224,7 +2224,12 @@ struct AstBase : public Annotation
};
template <typename T>
void ast_to_s(const std::shared_ptr<T>& ptr, std::string& s, int level = 0) {
void ast_to_s_core(
const std::shared_ptr<T>& ptr,
std::string& s,
int level,
std::function<std::string (const T& ast, int level)> fn) {
const auto& ast = *ptr;
for (auto i = 0; i < level; i++) {
s += " ";
@ -2240,15 +2245,21 @@ void ast_to_s(const std::shared_ptr<T>& ptr, std::string& s, int level = 0) {
} else {
s += "+ " + name + "\n";
}
if (fn) {
s += fn(ast, level + 1);
}
for (auto node : ast.nodes) {
ast_to_s(node, s, level + 1);
ast_to_s_core(node, s, level + 1, fn);
}
}
template <typename T>
std::string ast_to_s(const std::shared_ptr<T>& ptr) {
std::string ast_to_s(
const std::shared_ptr<T>& ptr,
std::function<std::string (const T& ast, int level)> fn = nullptr) {
std::string s;
ast_to_s(ptr, s);
ast_to_s_core(ptr, s, 0, fn);
return s;
}