tree-sitter/src/lr/item.cpp

#include "item.h"

using namespace std;

namespace tree_sitter {
    namespace lr {
        Item::Item(const std::string &rule_name, const rules::rule_ptr rule, int consumed_sym_count) :
            rule_name(rule_name),
            rule(rule),
            consumed_sym_count(consumed_sym_count) {};
        
        TransitionMap<Item> Item::transitions() const {
            return rule->transitions().map<Item>([&](rules::rule_ptr to_rule) {
                return item_ptr(new Item(rule_name, to_rule, consumed_sym_count + 1));
            });
        };
        
        vector<rules::rule_ptr> Item::next_symbols() const {
            vector<rules::rule_ptr> result;
            for (auto pair : rule->transitions())
                result.push_back(pair.second);
            return result;
        }
    
        bool Item::operator==(const Item &other) const {
            return (
                other.rule_name == rule_name &&
                other.rule == rule &&
                other.consumed_sym_count == consumed_sym_count);
        }
        
        std::ostream& operator<<(ostream &stream, const Item &item) {
            stream <<
                string("(item '") <<
                item.rule_name <<
                string("' ") <<
                *item.rule <<
                string(")");
            return stream;
        }
    }
}