Add helper for dynamic-casting to rule subclasses

This commit is contained in:
Max Brunsfeld 2015-10-12 15:26:50 -07:00
parent db9966b57c
commit 5c67f58a4b
14 changed files with 47 additions and 28 deletions

View file

@ -2,6 +2,7 @@
#include <string>
#include "compiler/rules/visitor.h"
#include "compiler/rules/blank.h"
#include "compiler/rules/metadata.h"
namespace tree_sitter {
namespace rules {
@ -14,14 +15,25 @@ Seq::Seq(rule_ptr left, rule_ptr right) : left(left), right(right) {}
rule_ptr Seq::build(const std::vector<rule_ptr> &rules) {
rule_ptr result = make_shared<Blank>();
for (auto &rule : rules)
result = (typeid(*result) != typeid(Blank)) ? make_shared<Seq>(result, rule)
: rule;
for (auto &rule : rules) {
auto blank = rule->as<Blank>();
if (blank)
continue;
auto metadata = rule->as<Metadata>();
if (metadata && metadata->rule->as<Blank>())
continue;
if (result->as<Blank>())
result = rule;
else
result = make_shared<Seq>(result, rule);
}
return result;
}
bool Seq::operator==(const Rule &rule) const {
const Seq *other = dynamic_cast<const Seq *>(&rule);
const Seq *other = rule.as<Seq>();
return other && (*other->left == *left) && (*other->right == *right);
}