62 lines
1.5 KiB
Ruby
62 lines
1.5 KiB
Ruby
require "marisa"
|
|
|
|
keyset = Marisa::Keyset.new
|
|
keyset.push_back("cake")
|
|
keyset.push_back("cookie")
|
|
keyset.push_back("ice")
|
|
keyset.push_back("ice-cream")
|
|
|
|
trie = Marisa::Trie.new
|
|
trie.build(keyset)
|
|
print("no. keys: ", trie.num_keys(), "\n")
|
|
print("no. tries: ", trie.num_tries(), "\n")
|
|
print("no. nodes: ", trie.num_nodes(), "\n")
|
|
print("size: ", trie.io_size(), "\n")
|
|
|
|
agent = Marisa::Agent.new
|
|
|
|
agent.set_query("cake")
|
|
trie.lookup(agent)
|
|
print(agent.query_str(), ": ", agent.key_id(), "\n")
|
|
|
|
agent.set_query("cookie")
|
|
trie.lookup(agent)
|
|
print(agent.query_str(), ": ", agent.key_id(), "\n")
|
|
|
|
agent.set_query("cockoo")
|
|
if not trie.lookup(agent)
|
|
print(agent.query_str(), ": not found\n")
|
|
end
|
|
|
|
print("ice: ", trie.lookup("ice"), "\n")
|
|
print("ice-cream: ", trie.lookup("ice-cream"), "\n")
|
|
if trie.lookup("ice-age") == Marisa::INVALID_KEY_ID
|
|
print("ice-age: not found\n")
|
|
end
|
|
|
|
trie.save("sample.dic")
|
|
trie.load("sample.dic")
|
|
|
|
agent.set_query(0)
|
|
trie.reverse_lookup(agent)
|
|
print(agent.query_id(), ": ", agent.key_str(), "\n")
|
|
|
|
agent.set_query(1)
|
|
trie.reverse_lookup(agent)
|
|
print(agent.query_id(), ": ", agent.key_str(), "\n")
|
|
|
|
print("2: ", trie.reverse_lookup(2), "\n")
|
|
print("3: ", trie.reverse_lookup(3), "\n")
|
|
|
|
trie.mmap("sample.dic")
|
|
|
|
agent.set_query("ice-cream soda")
|
|
while trie.common_prefix_search(agent)
|
|
print(agent.query_str(), ": ", agent.key_str(), " (", agent.key_id(), ")\n")
|
|
end
|
|
|
|
agent.set_query("ic")
|
|
while trie.predictive_search(agent)
|
|
print(agent.query_str(), ": ", agent.key_str(), " (", agent.key_id(), ")\n")
|
|
end
|