Test.java

/*
 * Copyright 2012 Takao Nakaguchi
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.trie4j.bytes;

import java.io.UnsupportedEncodingException;

import org.trie4j.bytes.Node;
import org.trie4j.bytes.PatriciaTrie;
import org.trie4j.bytes.TrieVisitor;

public class Test {
	public static void main(String[] args) throws Exception{
		System.out.println("--- patricia trie ---");
		go(new PatriciaTrie());
//		System.out.println("--- hash trie ---");
//		go(new HashSetTrie());
	}

	private static void go(PatriciaTrie trie) throws Exception{
		String[] words = {
				"apple", "appear", "a", "orange"
				, "applejuice", "appletea", "appleshower"
				, "orangejuice"
				};
		trie.insert("".getBytes("UTF-8"));
		for(String w : words){
			System.out.println("insert \"" + w + "\"");
			trie.insert(w.getBytes("UTF-8"));
			System.out.println("--dump--");
			trie.visit(new TrieVisitor() {
				@Override
				public void accept(Node node, int nest) {
					for(int i = 0; i < nest; i++){
						System.out.print(" ");
					}
					byte[] letters = node.getLetters();
					if(letters == null || letters.length == 0){
						System.out.print("<empty>");
					} else{
						System.out.print(fromUTF8(letters));
					}
					if(node.isTerminate()){
						System.out.println("*");
					} else{
						System.out.println("");
					}
				}
			});
		}
		System.out.println(trie.contains(""));

		System.out.println("--test contains--");
		for(String w : words){
			System.out.print(w + ": ");
			System.out.println(trie.contains(w));
		}
		System.out.println("--test not contains--");
		for(String w : new String[]{"banana", "app", "applebeer", "applejuice2"}){
			System.out.println(w + ": " + trie.contains(w));
		}
		System.out.println("-- test common prefix search --");
		System.out.println("query: applejuicebar");
		for(byte[] w : trie.commonPrefixSearch("applejuicebar".getBytes("UTF-8"))){
			System.out.println(new String(w, "UTF-8"));
		}
	}

	private static String fromUTF8(byte[] bytes){
		try {
			return new String(bytes, "UTF-8");
		} catch (UnsupportedEncodingException e) {
			throw new RuntimeException(e);
		}
	}
}