generator: generate transitions

1 files changed, 4 insertions, 3 deletions

diff --git a/src/generator.hpp b/src/generator.hpp
index 880323b..b901785 100644
--- a/src/generator.hpp
+++ b/src/generator.hpp
@@ -39,17 +39,18 @@ private:
 	std::multimap<std::string, item_t> rules;
 	std::multimap<symbol_t, item_t> items;
 	std::map<std::set<item_t>, unsigned> itemsets;
+	std::map<std::pair<unsigned, symbol_t>, unsigned> transitions;
 
 	void close_set(std::set<item_t> *set);
 	std::set<item_t> get_set(const std::string &nonterm);
 
-	std::pair<bool, unsigned> add_set(const std::set<item_t> &set) {
-		auto ret = itemsets.emplace(set, itemsets.size());
-		return std::make_pair(ret.second, ret.first->second);
+	std::pair<std::map<std::set<item_t>, unsigned>::iterator, bool> add_set(const std::set<item_t> &set) {
+		return itemsets.emplace(set, itemsets.size());
 	}
 
 	void generate_itemsets();
 
+	static void print_symbol(const symbol_t &sym);
 	static void print_item(const item_t &item);
 	static void print_set(const std::set<item_t> &set);