VocabTreeManager.java

package com.alibaba.simpleimage.analyze.search.tree;

import java.util.List;

import com.alibaba.simpleimage.analyze.search.cluster.ClusterBuilder;
import com.alibaba.simpleimage.analyze.search.cluster.Clusterable;
import com.alibaba.simpleimage.analyze.search.util.SerializationUtils;

public class VocabTreeManager {
	private static KMeansTree tree;

	public static KMeansTree loadVocabTree(String file) {
		if (tree == null) {
			tree = (KMeansTree) SerializationUtils.loadObject(file);
		}
		return tree;
	}
	
	public static KMeansTree makeTree(List<Clusterable> points,
			ClusterBuilder clusterer, int groupId) {
		return makeSampledTree(points, clusterer, 1);
	}

	public static KMeansTree makeSampledTree(List<Clusterable> points,
	                                         ClusterBuilder clusterer, int useEvery) {

		KMeansTree tree = new KMeansTree(points, 10, 5, clusterer);
		return tree;
	}
	
}